1//===-- xray_flags.cpp ------------------------------------------*- C++ -*-===// 2// 3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4// See https://llvm.org/LICENSE.txt for license information. 5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6// 7//===----------------------------------------------------------------------===// 8// 9// This file is a part of XRay, a dynamic runtime instrumentation system. 10// 11// XRay flag parsing logic. 12//===----------------------------------------------------------------------===// 13 14#include "xray_flags.h" 15#include "sanitizer_common/sanitizer_common.h" 16#include "sanitizer_common/sanitizer_flag_parser.h" 17#include "sanitizer_common/sanitizer_libc.h" 18#include "xray_defs.h" 19 20using namespace __sanitizer; 21 22namespace __xray { 23 24Flags xray_flags_dont_use_directly; // use via flags(). 25 26void Flags::setDefaults() XRAY_NEVER_INSTRUMENT { 27#define XRAY_FLAG(Type, Name, DefaultValue, Description) Name = DefaultValue; 28#include "xray_flags.inc" 29#undef XRAY_FLAG 30} 31 32void registerXRayFlags(FlagParser *P, Flags *F) XRAY_NEVER_INSTRUMENT { 33#define XRAY_FLAG(Type, Name, DefaultValue, Description) \ 34 RegisterFlag(P, #Name, Description, &F->Name); 35#include "xray_flags.inc" 36#undef XRAY_FLAG 37} 38 39// This function, as defined with the help of a macro meant to be introduced at 40// build time of the XRay runtime, passes in a statically defined list of 41// options that control XRay. This means users/deployments can tweak the 42// defaults that override the hard-coded defaults in the xray_flags.inc at 43// compile-time using the XRAY_DEFAULT_OPTIONS macro. 44const char *useCompilerDefinedFlags() XRAY_NEVER_INSTRUMENT { 45#ifdef XRAY_DEFAULT_OPTIONS 46 // Do the double-layered string conversion to prevent badly crafted strings 47 // provided through the XRAY_DEFAULT_OPTIONS from causing compilation issues 48 // (or changing the semantics of the implementation through the macro). This 49 // ensures that we convert whatever XRAY_DEFAULT_OPTIONS is defined as a 50 // string literal. 51 return SANITIZER_STRINGIFY(XRAY_DEFAULT_OPTIONS); 52#else 53 return ""; 54#endif 55} 56 57void initializeFlags() XRAY_NEVER_INSTRUMENT { 58 SetCommonFlagsDefaults(); 59 auto *F = flags(); 60 F->setDefaults(); 61 62 FlagParser XRayParser; 63 registerXRayFlags(&XRayParser, F); 64 RegisterCommonFlags(&XRayParser); 65 66 // Use options defaulted at compile-time for the runtime. 67 const char *XRayCompileFlags = useCompilerDefinedFlags(); 68 XRayParser.ParseString(XRayCompileFlags); 69 70 // Override from environment variables. 71 XRayParser.ParseStringFromEnv("XRAY_OPTIONS"); 72 73 // Override from command line. 74 InitializeCommonFlags(); 75 76 if (Verbosity()) 77 ReportUnrecognizedFlags(); 78 79 if (common_flags()->help) { 80 XRayParser.PrintFlagDescriptions(); 81 } 82} 83 84} // namespace __xray 85