1*08ab5237SOystein Eftevaag // Copyright (c) 1999, Google Inc.
2*08ab5237SOystein Eftevaag // All rights reserved.
3*08ab5237SOystein Eftevaag //
4*08ab5237SOystein Eftevaag // Redistribution and use in source and binary forms, with or without
5*08ab5237SOystein Eftevaag // modification, are permitted provided that the following conditions are
6*08ab5237SOystein Eftevaag // met:
7*08ab5237SOystein Eftevaag //
8*08ab5237SOystein Eftevaag // * Redistributions of source code must retain the above copyright
9*08ab5237SOystein Eftevaag // notice, this list of conditions and the following disclaimer.
10*08ab5237SOystein Eftevaag // * Redistributions in binary form must reproduce the above
11*08ab5237SOystein Eftevaag // copyright notice, this list of conditions and the following disclaimer
12*08ab5237SOystein Eftevaag // in the documentation and/or other materials provided with the
13*08ab5237SOystein Eftevaag // distribution.
14*08ab5237SOystein Eftevaag // * Neither the name of Google Inc. nor the names of its
15*08ab5237SOystein Eftevaag // contributors may be used to endorse or promote products derived from
16*08ab5237SOystein Eftevaag // this software without specific prior written permission.
17*08ab5237SOystein Eftevaag //
18*08ab5237SOystein Eftevaag // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19*08ab5237SOystein Eftevaag // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20*08ab5237SOystein Eftevaag // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21*08ab5237SOystein Eftevaag // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22*08ab5237SOystein Eftevaag // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23*08ab5237SOystein Eftevaag // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24*08ab5237SOystein Eftevaag // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25*08ab5237SOystein Eftevaag // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26*08ab5237SOystein Eftevaag // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27*08ab5237SOystein Eftevaag // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28*08ab5237SOystein Eftevaag // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29*08ab5237SOystein Eftevaag
30*08ab5237SOystein Eftevaag // ---
31*08ab5237SOystein Eftevaag // Revamped and reorganized by Craig Silverstein
32*08ab5237SOystein Eftevaag //
33*08ab5237SOystein Eftevaag // This file contains the implementation of all our command line flags
34*08ab5237SOystein Eftevaag // stuff. Here's how everything fits together
35*08ab5237SOystein Eftevaag //
36*08ab5237SOystein Eftevaag // * FlagRegistry owns CommandLineFlags owns FlagValue.
37*08ab5237SOystein Eftevaag // * FlagSaver holds a FlagRegistry (saves it at construct time,
38*08ab5237SOystein Eftevaag // restores it at destroy time).
39*08ab5237SOystein Eftevaag // * CommandLineFlagParser lives outside that hierarchy, but works on
40*08ab5237SOystein Eftevaag // CommandLineFlags (modifying the FlagValues).
41*08ab5237SOystein Eftevaag // * Free functions like SetCommandLineOption() work via one of the
42*08ab5237SOystein Eftevaag // above (such as CommandLineFlagParser).
43*08ab5237SOystein Eftevaag //
44*08ab5237SOystein Eftevaag // In more detail:
45*08ab5237SOystein Eftevaag //
46*08ab5237SOystein Eftevaag // -- The main classes that hold flag data:
47*08ab5237SOystein Eftevaag //
48*08ab5237SOystein Eftevaag // FlagValue holds the current value of a flag. It's
49*08ab5237SOystein Eftevaag // pseudo-templatized: every operation on a FlagValue is typed. It
50*08ab5237SOystein Eftevaag // also deals with storage-lifetime issues (so flag values don't go
51*08ab5237SOystein Eftevaag // away in a destructor), which is why we need a whole class to hold a
52*08ab5237SOystein Eftevaag // variable's value.
53*08ab5237SOystein Eftevaag //
54*08ab5237SOystein Eftevaag // CommandLineFlag is all the information about a single command-line
55*08ab5237SOystein Eftevaag // flag. It has a FlagValue for the flag's current value, but also
56*08ab5237SOystein Eftevaag // the flag's name, type, etc.
57*08ab5237SOystein Eftevaag //
58*08ab5237SOystein Eftevaag // FlagRegistry is a collection of CommandLineFlags. There's the
59*08ab5237SOystein Eftevaag // global registry, which is where flags defined via DEFINE_foo()
60*08ab5237SOystein Eftevaag // live. But it's possible to define your own flag, manually, in a
61*08ab5237SOystein Eftevaag // different registry you create. (In practice, multiple registries
62*08ab5237SOystein Eftevaag // are used only by FlagSaver).
63*08ab5237SOystein Eftevaag //
64*08ab5237SOystein Eftevaag // A given FlagValue is owned by exactly one CommandLineFlag. A given
65*08ab5237SOystein Eftevaag // CommandLineFlag is owned by exactly one FlagRegistry. FlagRegistry
66*08ab5237SOystein Eftevaag // has a lock; any operation that writes to a FlagValue or
67*08ab5237SOystein Eftevaag // CommandLineFlag owned by that registry must acquire the
68*08ab5237SOystein Eftevaag // FlagRegistry lock before doing so.
69*08ab5237SOystein Eftevaag //
70*08ab5237SOystein Eftevaag // --- Some other classes and free functions:
71*08ab5237SOystein Eftevaag //
72*08ab5237SOystein Eftevaag // CommandLineFlagInfo is a client-exposed version of CommandLineFlag.
73*08ab5237SOystein Eftevaag // Once it's instantiated, it has no dependencies or relationships
74*08ab5237SOystein Eftevaag // with any other part of this file.
75*08ab5237SOystein Eftevaag //
76*08ab5237SOystein Eftevaag // FlagRegisterer is the helper class used by the DEFINE_* macros to
77*08ab5237SOystein Eftevaag // allow work to be done at global initialization time.
78*08ab5237SOystein Eftevaag //
79*08ab5237SOystein Eftevaag // CommandLineFlagParser is the class that reads from the commandline
80*08ab5237SOystein Eftevaag // and instantiates flag values based on that. It needs to poke into
81*08ab5237SOystein Eftevaag // the innards of the FlagValue->CommandLineFlag->FlagRegistry class
82*08ab5237SOystein Eftevaag // hierarchy to do that. It's careful to acquire the FlagRegistry
83*08ab5237SOystein Eftevaag // lock before doing any writing or other non-const actions.
84*08ab5237SOystein Eftevaag //
85*08ab5237SOystein Eftevaag // GetCommandLineOption is just a hook into registry routines to
86*08ab5237SOystein Eftevaag // retrieve a flag based on its name. SetCommandLineOption, on the
87*08ab5237SOystein Eftevaag // other hand, hooks into CommandLineFlagParser. Other API functions
88*08ab5237SOystein Eftevaag // are, similarly, mostly hooks into the functionality described above.
89*08ab5237SOystein Eftevaag
90*08ab5237SOystein Eftevaag #include "config.h"
91*08ab5237SOystein Eftevaag #include "gflags/gflags.h"
92*08ab5237SOystein Eftevaag
93*08ab5237SOystein Eftevaag #include <assert.h>
94*08ab5237SOystein Eftevaag #include <ctype.h>
95*08ab5237SOystein Eftevaag #include <errno.h>
96*08ab5237SOystein Eftevaag #if defined(HAVE_FNMATCH_H)
97*08ab5237SOystein Eftevaag # include <fnmatch.h>
98*08ab5237SOystein Eftevaag #elif defined(HAVE_SHLWAPI_H)
99*08ab5237SOystein Eftevaag # define NO_SHLWAPI_ISOS
100*08ab5237SOystein Eftevaag # include <shlwapi.h>
101*08ab5237SOystein Eftevaag #endif
102*08ab5237SOystein Eftevaag #include <stdarg.h> // For va_list and related operations
103*08ab5237SOystein Eftevaag #include <stdio.h>
104*08ab5237SOystein Eftevaag #include <string.h>
105*08ab5237SOystein Eftevaag
106*08ab5237SOystein Eftevaag #include <algorithm>
107*08ab5237SOystein Eftevaag #include <map>
108*08ab5237SOystein Eftevaag #include <string>
109*08ab5237SOystein Eftevaag #include <utility> // for pair<>
110*08ab5237SOystein Eftevaag #include <vector>
111*08ab5237SOystein Eftevaag
112*08ab5237SOystein Eftevaag #include "mutex.h"
113*08ab5237SOystein Eftevaag #include "util.h"
114*08ab5237SOystein Eftevaag
115*08ab5237SOystein Eftevaag using namespace MUTEX_NAMESPACE;
116*08ab5237SOystein Eftevaag
117*08ab5237SOystein Eftevaag
118*08ab5237SOystein Eftevaag // Special flags, type 1: the 'recursive' flags. They set another flag's val.
119*08ab5237SOystein Eftevaag DEFINE_string(flagfile, "", "load flags from file");
120*08ab5237SOystein Eftevaag DEFINE_string(fromenv, "", "set flags from the environment"
121*08ab5237SOystein Eftevaag " [use 'export FLAGS_flag1=value']");
122*08ab5237SOystein Eftevaag DEFINE_string(tryfromenv, "", "set flags from the environment if present");
123*08ab5237SOystein Eftevaag
124*08ab5237SOystein Eftevaag // Special flags, type 2: the 'parsing' flags. They modify how we parse.
125*08ab5237SOystein Eftevaag DEFINE_string(undefok, "", "comma-separated list of flag names that it is okay to specify "
126*08ab5237SOystein Eftevaag "on the command line even if the program does not define a flag "
127*08ab5237SOystein Eftevaag "with that name. IMPORTANT: flags in this list that have "
128*08ab5237SOystein Eftevaag "arguments MUST use the flag=value format");
129*08ab5237SOystein Eftevaag
130*08ab5237SOystein Eftevaag namespace GFLAGS_NAMESPACE {
131*08ab5237SOystein Eftevaag
132*08ab5237SOystein Eftevaag using std::map;
133*08ab5237SOystein Eftevaag using std::pair;
134*08ab5237SOystein Eftevaag using std::sort;
135*08ab5237SOystein Eftevaag using std::string;
136*08ab5237SOystein Eftevaag using std::vector;
137*08ab5237SOystein Eftevaag
138*08ab5237SOystein Eftevaag // This is used by the unittest to test error-exit code
139*08ab5237SOystein Eftevaag void GFLAGS_DLL_DECL (*gflags_exitfunc)(int) = &exit; // from stdlib.h
140*08ab5237SOystein Eftevaag
141*08ab5237SOystein Eftevaag
142*08ab5237SOystein Eftevaag // The help message indicating that the commandline flag has been
143*08ab5237SOystein Eftevaag // 'stripped'. It will not show up when doing "-help" and its
144*08ab5237SOystein Eftevaag // variants. The flag is stripped if STRIP_FLAG_HELP is set to 1
145*08ab5237SOystein Eftevaag // before including base/gflags.h
146*08ab5237SOystein Eftevaag
147*08ab5237SOystein Eftevaag // This is used by this file, and also in gflags_reporting.cc
148*08ab5237SOystein Eftevaag const char kStrippedFlagHelp[] = "\001\002\003\004 (unknown) \004\003\002\001";
149*08ab5237SOystein Eftevaag
150*08ab5237SOystein Eftevaag namespace {
151*08ab5237SOystein Eftevaag
152*08ab5237SOystein Eftevaag // There are also 'reporting' flags, in gflags_reporting.cc.
153*08ab5237SOystein Eftevaag
154*08ab5237SOystein Eftevaag static const char kError[] = "ERROR: ";
155*08ab5237SOystein Eftevaag
156*08ab5237SOystein Eftevaag // Indicates that undefined options are to be ignored.
157*08ab5237SOystein Eftevaag // Enables deferred processing of flags in dynamically loaded libraries.
158*08ab5237SOystein Eftevaag static bool allow_command_line_reparsing = false;
159*08ab5237SOystein Eftevaag
160*08ab5237SOystein Eftevaag static bool logging_is_probably_set_up = false;
161*08ab5237SOystein Eftevaag
162*08ab5237SOystein Eftevaag // This is a 'prototype' validate-function. 'Real' validate
163*08ab5237SOystein Eftevaag // functions, take a flag-value as an argument: ValidateFn(bool) or
164*08ab5237SOystein Eftevaag // ValidateFn(uint64). However, for easier storage, we strip off this
165*08ab5237SOystein Eftevaag // argument and then restore it when actually calling the function on
166*08ab5237SOystein Eftevaag // a flag value.
167*08ab5237SOystein Eftevaag typedef bool (*ValidateFnProto)();
168*08ab5237SOystein Eftevaag
169*08ab5237SOystein Eftevaag // Whether we should die when reporting an error.
170*08ab5237SOystein Eftevaag enum DieWhenReporting { DIE, DO_NOT_DIE };
171*08ab5237SOystein Eftevaag
172*08ab5237SOystein Eftevaag // Report Error and exit if requested.
ReportError(DieWhenReporting should_die,const char * format,...)173*08ab5237SOystein Eftevaag static void ReportError(DieWhenReporting should_die, const char* format, ...) {
174*08ab5237SOystein Eftevaag va_list ap;
175*08ab5237SOystein Eftevaag va_start(ap, format);
176*08ab5237SOystein Eftevaag vfprintf(stderr, format, ap);
177*08ab5237SOystein Eftevaag va_end(ap);
178*08ab5237SOystein Eftevaag fflush(stderr); // should be unnecessary, but cygwin's rxvt buffers stderr
179*08ab5237SOystein Eftevaag if (should_die == DIE) gflags_exitfunc(1);
180*08ab5237SOystein Eftevaag }
181*08ab5237SOystein Eftevaag
182*08ab5237SOystein Eftevaag
183*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
184*08ab5237SOystein Eftevaag // FlagValue
185*08ab5237SOystein Eftevaag // This represent the value a single flag might have. The major
186*08ab5237SOystein Eftevaag // functionality is to convert from a string to an object of a
187*08ab5237SOystein Eftevaag // given type, and back. Thread-compatible.
188*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
189*08ab5237SOystein Eftevaag
190*08ab5237SOystein Eftevaag class CommandLineFlag;
191*08ab5237SOystein Eftevaag class FlagValue {
192*08ab5237SOystein Eftevaag public:
193*08ab5237SOystein Eftevaag enum ValueType {
194*08ab5237SOystein Eftevaag FV_BOOL = 0,
195*08ab5237SOystein Eftevaag FV_INT32 = 1,
196*08ab5237SOystein Eftevaag FV_UINT32 = 2,
197*08ab5237SOystein Eftevaag FV_INT64 = 3,
198*08ab5237SOystein Eftevaag FV_UINT64 = 4,
199*08ab5237SOystein Eftevaag FV_DOUBLE = 5,
200*08ab5237SOystein Eftevaag FV_STRING = 6,
201*08ab5237SOystein Eftevaag FV_MAX_INDEX = 6,
202*08ab5237SOystein Eftevaag };
203*08ab5237SOystein Eftevaag
204*08ab5237SOystein Eftevaag template <typename FlagType>
205*08ab5237SOystein Eftevaag FlagValue(FlagType* valbuf, bool transfer_ownership_of_value);
206*08ab5237SOystein Eftevaag ~FlagValue();
207*08ab5237SOystein Eftevaag
208*08ab5237SOystein Eftevaag bool ParseFrom(const char* spec);
209*08ab5237SOystein Eftevaag string ToString() const;
210*08ab5237SOystein Eftevaag
Type() const211*08ab5237SOystein Eftevaag ValueType Type() const { return static_cast<ValueType>(type_); }
212*08ab5237SOystein Eftevaag
213*08ab5237SOystein Eftevaag private:
214*08ab5237SOystein Eftevaag friend class CommandLineFlag; // for many things, including Validate()
215*08ab5237SOystein Eftevaag friend class GFLAGS_NAMESPACE::FlagSaverImpl; // calls New()
216*08ab5237SOystein Eftevaag friend class FlagRegistry; // checks value_buffer_ for flags_by_ptr_ map
217*08ab5237SOystein Eftevaag template <typename T> friend T GetFromEnv(const char*, T);
218*08ab5237SOystein Eftevaag friend bool TryParseLocked(const CommandLineFlag*, FlagValue*,
219*08ab5237SOystein Eftevaag const char*, string*); // for New(), CopyFrom()
220*08ab5237SOystein Eftevaag
221*08ab5237SOystein Eftevaag template <typename FlagType>
222*08ab5237SOystein Eftevaag struct FlagValueTraits;
223*08ab5237SOystein Eftevaag
224*08ab5237SOystein Eftevaag const char* TypeName() const;
225*08ab5237SOystein Eftevaag bool Equal(const FlagValue& x) const;
226*08ab5237SOystein Eftevaag FlagValue* New() const; // creates a new one with default value
227*08ab5237SOystein Eftevaag void CopyFrom(const FlagValue& x);
228*08ab5237SOystein Eftevaag
229*08ab5237SOystein Eftevaag // Calls the given validate-fn on value_buffer_, and returns
230*08ab5237SOystein Eftevaag // whatever it returns. But first casts validate_fn_proto to a
231*08ab5237SOystein Eftevaag // function that takes our value as an argument (eg void
232*08ab5237SOystein Eftevaag // (*validate_fn)(bool) for a bool flag).
233*08ab5237SOystein Eftevaag bool Validate(const char* flagname, ValidateFnProto validate_fn_proto) const;
234*08ab5237SOystein Eftevaag
235*08ab5237SOystein Eftevaag void* const value_buffer_; // points to the buffer holding our data
236*08ab5237SOystein Eftevaag const int8 type_; // how to interpret value_
237*08ab5237SOystein Eftevaag const bool owns_value_; // whether to free value on destruct
238*08ab5237SOystein Eftevaag
239*08ab5237SOystein Eftevaag FlagValue(const FlagValue&); // no copying!
240*08ab5237SOystein Eftevaag void operator=(const FlagValue&);
241*08ab5237SOystein Eftevaag };
242*08ab5237SOystein Eftevaag
243*08ab5237SOystein Eftevaag // Map the given C++ type to a value of the ValueType enum at compile time.
244*08ab5237SOystein Eftevaag #define DEFINE_FLAG_TRAITS(type, value) \
245*08ab5237SOystein Eftevaag template <> \
246*08ab5237SOystein Eftevaag struct FlagValue::FlagValueTraits<type> { \
247*08ab5237SOystein Eftevaag static const ValueType kValueType = value; \
248*08ab5237SOystein Eftevaag }
249*08ab5237SOystein Eftevaag
250*08ab5237SOystein Eftevaag // Define full template specializations of the FlagValueTraits template
251*08ab5237SOystein Eftevaag // for all supported flag types.
252*08ab5237SOystein Eftevaag DEFINE_FLAG_TRAITS(bool, FV_BOOL);
253*08ab5237SOystein Eftevaag DEFINE_FLAG_TRAITS(int32, FV_INT32);
254*08ab5237SOystein Eftevaag DEFINE_FLAG_TRAITS(uint32, FV_UINT32);
255*08ab5237SOystein Eftevaag DEFINE_FLAG_TRAITS(int64, FV_INT64);
256*08ab5237SOystein Eftevaag DEFINE_FLAG_TRAITS(uint64, FV_UINT64);
257*08ab5237SOystein Eftevaag DEFINE_FLAG_TRAITS(double, FV_DOUBLE);
258*08ab5237SOystein Eftevaag DEFINE_FLAG_TRAITS(std::string, FV_STRING);
259*08ab5237SOystein Eftevaag
260*08ab5237SOystein Eftevaag #undef DEFINE_FLAG_TRAITS
261*08ab5237SOystein Eftevaag
262*08ab5237SOystein Eftevaag
263*08ab5237SOystein Eftevaag // This could be a templated method of FlagValue, but doing so adds to the
264*08ab5237SOystein Eftevaag // size of the .o. Since there's no type-safety here anyway, macro is ok.
265*08ab5237SOystein Eftevaag #define VALUE_AS(type) *reinterpret_cast<type*>(value_buffer_)
266*08ab5237SOystein Eftevaag #define OTHER_VALUE_AS(fv, type) *reinterpret_cast<type*>(fv.value_buffer_)
267*08ab5237SOystein Eftevaag #define SET_VALUE_AS(type, value) VALUE_AS(type) = (value)
268*08ab5237SOystein Eftevaag
269*08ab5237SOystein Eftevaag template <typename FlagType>
FlagValue(FlagType * valbuf,bool transfer_ownership_of_value)270*08ab5237SOystein Eftevaag FlagValue::FlagValue(FlagType* valbuf,
271*08ab5237SOystein Eftevaag bool transfer_ownership_of_value)
272*08ab5237SOystein Eftevaag : value_buffer_(valbuf),
273*08ab5237SOystein Eftevaag type_(FlagValueTraits<FlagType>::kValueType),
274*08ab5237SOystein Eftevaag owns_value_(transfer_ownership_of_value) {
275*08ab5237SOystein Eftevaag }
276*08ab5237SOystein Eftevaag
~FlagValue()277*08ab5237SOystein Eftevaag FlagValue::~FlagValue() {
278*08ab5237SOystein Eftevaag if (!owns_value_) {
279*08ab5237SOystein Eftevaag return;
280*08ab5237SOystein Eftevaag }
281*08ab5237SOystein Eftevaag switch (type_) {
282*08ab5237SOystein Eftevaag case FV_BOOL: delete reinterpret_cast<bool*>(value_buffer_); break;
283*08ab5237SOystein Eftevaag case FV_INT32: delete reinterpret_cast<int32*>(value_buffer_); break;
284*08ab5237SOystein Eftevaag case FV_UINT32: delete reinterpret_cast<uint32*>(value_buffer_); break;
285*08ab5237SOystein Eftevaag case FV_INT64: delete reinterpret_cast<int64*>(value_buffer_); break;
286*08ab5237SOystein Eftevaag case FV_UINT64: delete reinterpret_cast<uint64*>(value_buffer_); break;
287*08ab5237SOystein Eftevaag case FV_DOUBLE: delete reinterpret_cast<double*>(value_buffer_); break;
288*08ab5237SOystein Eftevaag case FV_STRING: delete reinterpret_cast<string*>(value_buffer_); break;
289*08ab5237SOystein Eftevaag }
290*08ab5237SOystein Eftevaag }
291*08ab5237SOystein Eftevaag
ParseFrom(const char * value)292*08ab5237SOystein Eftevaag bool FlagValue::ParseFrom(const char* value) {
293*08ab5237SOystein Eftevaag if (type_ == FV_BOOL) {
294*08ab5237SOystein Eftevaag const char* kTrue[] = { "1", "t", "true", "y", "yes" };
295*08ab5237SOystein Eftevaag const char* kFalse[] = { "0", "f", "false", "n", "no" };
296*08ab5237SOystein Eftevaag COMPILE_ASSERT(sizeof(kTrue) == sizeof(kFalse), true_false_equal);
297*08ab5237SOystein Eftevaag for (size_t i = 0; i < sizeof(kTrue)/sizeof(*kTrue); ++i) {
298*08ab5237SOystein Eftevaag if (strcasecmp(value, kTrue[i]) == 0) {
299*08ab5237SOystein Eftevaag SET_VALUE_AS(bool, true);
300*08ab5237SOystein Eftevaag return true;
301*08ab5237SOystein Eftevaag } else if (strcasecmp(value, kFalse[i]) == 0) {
302*08ab5237SOystein Eftevaag SET_VALUE_AS(bool, false);
303*08ab5237SOystein Eftevaag return true;
304*08ab5237SOystein Eftevaag }
305*08ab5237SOystein Eftevaag }
306*08ab5237SOystein Eftevaag return false; // didn't match a legal input
307*08ab5237SOystein Eftevaag
308*08ab5237SOystein Eftevaag } else if (type_ == FV_STRING) {
309*08ab5237SOystein Eftevaag SET_VALUE_AS(string, value);
310*08ab5237SOystein Eftevaag return true;
311*08ab5237SOystein Eftevaag }
312*08ab5237SOystein Eftevaag
313*08ab5237SOystein Eftevaag // OK, it's likely to be numeric, and we'll be using a strtoXXX method.
314*08ab5237SOystein Eftevaag if (value[0] == '\0') // empty-string is only allowed for string type.
315*08ab5237SOystein Eftevaag return false;
316*08ab5237SOystein Eftevaag char* end;
317*08ab5237SOystein Eftevaag // Leading 0x puts us in base 16. But leading 0 does not put us in base 8!
318*08ab5237SOystein Eftevaag // It caused too many bugs when we had that behavior.
319*08ab5237SOystein Eftevaag int base = 10; // by default
320*08ab5237SOystein Eftevaag if (value[0] == '0' && (value[1] == 'x' || value[1] == 'X'))
321*08ab5237SOystein Eftevaag base = 16;
322*08ab5237SOystein Eftevaag errno = 0;
323*08ab5237SOystein Eftevaag
324*08ab5237SOystein Eftevaag switch (type_) {
325*08ab5237SOystein Eftevaag case FV_INT32: {
326*08ab5237SOystein Eftevaag const int64 r = strto64(value, &end, base);
327*08ab5237SOystein Eftevaag if (errno || end != value + strlen(value)) return false; // bad parse
328*08ab5237SOystein Eftevaag if (static_cast<int32>(r) != r) // worked, but number out of range
329*08ab5237SOystein Eftevaag return false;
330*08ab5237SOystein Eftevaag SET_VALUE_AS(int32, static_cast<int32>(r));
331*08ab5237SOystein Eftevaag return true;
332*08ab5237SOystein Eftevaag }
333*08ab5237SOystein Eftevaag case FV_UINT32: {
334*08ab5237SOystein Eftevaag while (*value == ' ') value++;
335*08ab5237SOystein Eftevaag if (*value == '-') return false; // negative number
336*08ab5237SOystein Eftevaag const uint64 r = strtou64(value, &end, base);
337*08ab5237SOystein Eftevaag if (errno || end != value + strlen(value)) return false; // bad parse
338*08ab5237SOystein Eftevaag if (static_cast<uint32>(r) != r) // worked, but number out of range
339*08ab5237SOystein Eftevaag return false;
340*08ab5237SOystein Eftevaag SET_VALUE_AS(uint32, static_cast<uint32>(r));
341*08ab5237SOystein Eftevaag return true;
342*08ab5237SOystein Eftevaag }
343*08ab5237SOystein Eftevaag case FV_INT64: {
344*08ab5237SOystein Eftevaag const int64 r = strto64(value, &end, base);
345*08ab5237SOystein Eftevaag if (errno || end != value + strlen(value)) return false; // bad parse
346*08ab5237SOystein Eftevaag SET_VALUE_AS(int64, r);
347*08ab5237SOystein Eftevaag return true;
348*08ab5237SOystein Eftevaag }
349*08ab5237SOystein Eftevaag case FV_UINT64: {
350*08ab5237SOystein Eftevaag while (*value == ' ') value++;
351*08ab5237SOystein Eftevaag if (*value == '-') return false; // negative number
352*08ab5237SOystein Eftevaag const uint64 r = strtou64(value, &end, base);
353*08ab5237SOystein Eftevaag if (errno || end != value + strlen(value)) return false; // bad parse
354*08ab5237SOystein Eftevaag SET_VALUE_AS(uint64, r);
355*08ab5237SOystein Eftevaag return true;
356*08ab5237SOystein Eftevaag }
357*08ab5237SOystein Eftevaag case FV_DOUBLE: {
358*08ab5237SOystein Eftevaag const double r = strtod(value, &end);
359*08ab5237SOystein Eftevaag if (errno || end != value + strlen(value)) return false; // bad parse
360*08ab5237SOystein Eftevaag SET_VALUE_AS(double, r);
361*08ab5237SOystein Eftevaag return true;
362*08ab5237SOystein Eftevaag }
363*08ab5237SOystein Eftevaag default: {
364*08ab5237SOystein Eftevaag assert(false); // unknown type
365*08ab5237SOystein Eftevaag return false;
366*08ab5237SOystein Eftevaag }
367*08ab5237SOystein Eftevaag }
368*08ab5237SOystein Eftevaag }
369*08ab5237SOystein Eftevaag
ToString() const370*08ab5237SOystein Eftevaag string FlagValue::ToString() const {
371*08ab5237SOystein Eftevaag char intbuf[64]; // enough to hold even the biggest number
372*08ab5237SOystein Eftevaag switch (type_) {
373*08ab5237SOystein Eftevaag case FV_BOOL:
374*08ab5237SOystein Eftevaag return VALUE_AS(bool) ? "true" : "false";
375*08ab5237SOystein Eftevaag case FV_INT32:
376*08ab5237SOystein Eftevaag snprintf(intbuf, sizeof(intbuf), "%" PRId32, VALUE_AS(int32));
377*08ab5237SOystein Eftevaag return intbuf;
378*08ab5237SOystein Eftevaag case FV_UINT32:
379*08ab5237SOystein Eftevaag snprintf(intbuf, sizeof(intbuf), "%" PRIu32, VALUE_AS(uint32));
380*08ab5237SOystein Eftevaag return intbuf;
381*08ab5237SOystein Eftevaag case FV_INT64:
382*08ab5237SOystein Eftevaag snprintf(intbuf, sizeof(intbuf), "%" PRId64, VALUE_AS(int64));
383*08ab5237SOystein Eftevaag return intbuf;
384*08ab5237SOystein Eftevaag case FV_UINT64:
385*08ab5237SOystein Eftevaag snprintf(intbuf, sizeof(intbuf), "%" PRIu64, VALUE_AS(uint64));
386*08ab5237SOystein Eftevaag return intbuf;
387*08ab5237SOystein Eftevaag case FV_DOUBLE:
388*08ab5237SOystein Eftevaag snprintf(intbuf, sizeof(intbuf), "%.17g", VALUE_AS(double));
389*08ab5237SOystein Eftevaag return intbuf;
390*08ab5237SOystein Eftevaag case FV_STRING:
391*08ab5237SOystein Eftevaag return VALUE_AS(string);
392*08ab5237SOystein Eftevaag default:
393*08ab5237SOystein Eftevaag assert(false);
394*08ab5237SOystein Eftevaag return ""; // unknown type
395*08ab5237SOystein Eftevaag }
396*08ab5237SOystein Eftevaag }
397*08ab5237SOystein Eftevaag
Validate(const char * flagname,ValidateFnProto validate_fn_proto) const398*08ab5237SOystein Eftevaag bool FlagValue::Validate(const char* flagname,
399*08ab5237SOystein Eftevaag ValidateFnProto validate_fn_proto) const {
400*08ab5237SOystein Eftevaag switch (type_) {
401*08ab5237SOystein Eftevaag case FV_BOOL:
402*08ab5237SOystein Eftevaag return reinterpret_cast<bool (*)(const char*, bool)>(
403*08ab5237SOystein Eftevaag validate_fn_proto)(flagname, VALUE_AS(bool));
404*08ab5237SOystein Eftevaag case FV_INT32:
405*08ab5237SOystein Eftevaag return reinterpret_cast<bool (*)(const char*, int32)>(
406*08ab5237SOystein Eftevaag validate_fn_proto)(flagname, VALUE_AS(int32));
407*08ab5237SOystein Eftevaag case FV_UINT32:
408*08ab5237SOystein Eftevaag return reinterpret_cast<bool (*)(const char*, uint32)>(
409*08ab5237SOystein Eftevaag validate_fn_proto)(flagname, VALUE_AS(uint32));
410*08ab5237SOystein Eftevaag case FV_INT64:
411*08ab5237SOystein Eftevaag return reinterpret_cast<bool (*)(const char*, int64)>(
412*08ab5237SOystein Eftevaag validate_fn_proto)(flagname, VALUE_AS(int64));
413*08ab5237SOystein Eftevaag case FV_UINT64:
414*08ab5237SOystein Eftevaag return reinterpret_cast<bool (*)(const char*, uint64)>(
415*08ab5237SOystein Eftevaag validate_fn_proto)(flagname, VALUE_AS(uint64));
416*08ab5237SOystein Eftevaag case FV_DOUBLE:
417*08ab5237SOystein Eftevaag return reinterpret_cast<bool (*)(const char*, double)>(
418*08ab5237SOystein Eftevaag validate_fn_proto)(flagname, VALUE_AS(double));
419*08ab5237SOystein Eftevaag case FV_STRING:
420*08ab5237SOystein Eftevaag return reinterpret_cast<bool (*)(const char*, const string&)>(
421*08ab5237SOystein Eftevaag validate_fn_proto)(flagname, VALUE_AS(string));
422*08ab5237SOystein Eftevaag default:
423*08ab5237SOystein Eftevaag assert(false); // unknown type
424*08ab5237SOystein Eftevaag return false;
425*08ab5237SOystein Eftevaag }
426*08ab5237SOystein Eftevaag }
427*08ab5237SOystein Eftevaag
TypeName() const428*08ab5237SOystein Eftevaag const char* FlagValue::TypeName() const {
429*08ab5237SOystein Eftevaag static const char types[] =
430*08ab5237SOystein Eftevaag "bool\0xx"
431*08ab5237SOystein Eftevaag "int32\0x"
432*08ab5237SOystein Eftevaag "uint32\0"
433*08ab5237SOystein Eftevaag "int64\0x"
434*08ab5237SOystein Eftevaag "uint64\0"
435*08ab5237SOystein Eftevaag "double\0"
436*08ab5237SOystein Eftevaag "string";
437*08ab5237SOystein Eftevaag if (type_ > FV_MAX_INDEX) {
438*08ab5237SOystein Eftevaag assert(false);
439*08ab5237SOystein Eftevaag return "";
440*08ab5237SOystein Eftevaag }
441*08ab5237SOystein Eftevaag // Directly indexing the strings in the 'types' string, each of them is 7 bytes long.
442*08ab5237SOystein Eftevaag return &types[type_ * 7];
443*08ab5237SOystein Eftevaag }
444*08ab5237SOystein Eftevaag
Equal(const FlagValue & x) const445*08ab5237SOystein Eftevaag bool FlagValue::Equal(const FlagValue& x) const {
446*08ab5237SOystein Eftevaag if (type_ != x.type_)
447*08ab5237SOystein Eftevaag return false;
448*08ab5237SOystein Eftevaag switch (type_) {
449*08ab5237SOystein Eftevaag case FV_BOOL: return VALUE_AS(bool) == OTHER_VALUE_AS(x, bool);
450*08ab5237SOystein Eftevaag case FV_INT32: return VALUE_AS(int32) == OTHER_VALUE_AS(x, int32);
451*08ab5237SOystein Eftevaag case FV_UINT32: return VALUE_AS(uint32) == OTHER_VALUE_AS(x, uint32);
452*08ab5237SOystein Eftevaag case FV_INT64: return VALUE_AS(int64) == OTHER_VALUE_AS(x, int64);
453*08ab5237SOystein Eftevaag case FV_UINT64: return VALUE_AS(uint64) == OTHER_VALUE_AS(x, uint64);
454*08ab5237SOystein Eftevaag case FV_DOUBLE: return VALUE_AS(double) == OTHER_VALUE_AS(x, double);
455*08ab5237SOystein Eftevaag case FV_STRING: return VALUE_AS(string) == OTHER_VALUE_AS(x, string);
456*08ab5237SOystein Eftevaag default: assert(false); return false; // unknown type
457*08ab5237SOystein Eftevaag }
458*08ab5237SOystein Eftevaag }
459*08ab5237SOystein Eftevaag
New() const460*08ab5237SOystein Eftevaag FlagValue* FlagValue::New() const {
461*08ab5237SOystein Eftevaag switch (type_) {
462*08ab5237SOystein Eftevaag case FV_BOOL: return new FlagValue(new bool(false), true);
463*08ab5237SOystein Eftevaag case FV_INT32: return new FlagValue(new int32(0), true);
464*08ab5237SOystein Eftevaag case FV_UINT32: return new FlagValue(new uint32(0), true);
465*08ab5237SOystein Eftevaag case FV_INT64: return new FlagValue(new int64(0), true);
466*08ab5237SOystein Eftevaag case FV_UINT64: return new FlagValue(new uint64(0), true);
467*08ab5237SOystein Eftevaag case FV_DOUBLE: return new FlagValue(new double(0.0), true);
468*08ab5237SOystein Eftevaag case FV_STRING: return new FlagValue(new string, true);
469*08ab5237SOystein Eftevaag default: assert(false); return NULL; // unknown type
470*08ab5237SOystein Eftevaag }
471*08ab5237SOystein Eftevaag }
472*08ab5237SOystein Eftevaag
CopyFrom(const FlagValue & x)473*08ab5237SOystein Eftevaag void FlagValue::CopyFrom(const FlagValue& x) {
474*08ab5237SOystein Eftevaag assert(type_ == x.type_);
475*08ab5237SOystein Eftevaag switch (type_) {
476*08ab5237SOystein Eftevaag case FV_BOOL: SET_VALUE_AS(bool, OTHER_VALUE_AS(x, bool)); break;
477*08ab5237SOystein Eftevaag case FV_INT32: SET_VALUE_AS(int32, OTHER_VALUE_AS(x, int32)); break;
478*08ab5237SOystein Eftevaag case FV_UINT32: SET_VALUE_AS(uint32, OTHER_VALUE_AS(x, uint32)); break;
479*08ab5237SOystein Eftevaag case FV_INT64: SET_VALUE_AS(int64, OTHER_VALUE_AS(x, int64)); break;
480*08ab5237SOystein Eftevaag case FV_UINT64: SET_VALUE_AS(uint64, OTHER_VALUE_AS(x, uint64)); break;
481*08ab5237SOystein Eftevaag case FV_DOUBLE: SET_VALUE_AS(double, OTHER_VALUE_AS(x, double)); break;
482*08ab5237SOystein Eftevaag case FV_STRING: SET_VALUE_AS(string, OTHER_VALUE_AS(x, string)); break;
483*08ab5237SOystein Eftevaag default: assert(false); // unknown type
484*08ab5237SOystein Eftevaag }
485*08ab5237SOystein Eftevaag }
486*08ab5237SOystein Eftevaag
487*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
488*08ab5237SOystein Eftevaag // CommandLineFlag
489*08ab5237SOystein Eftevaag // This represents a single flag, including its name, description,
490*08ab5237SOystein Eftevaag // default value, and current value. Mostly this serves as a
491*08ab5237SOystein Eftevaag // struct, though it also knows how to register itself.
492*08ab5237SOystein Eftevaag // All CommandLineFlags are owned by a (exactly one)
493*08ab5237SOystein Eftevaag // FlagRegistry. If you wish to modify fields in this class, you
494*08ab5237SOystein Eftevaag // should acquire the FlagRegistry lock for the registry that owns
495*08ab5237SOystein Eftevaag // this flag.
496*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
497*08ab5237SOystein Eftevaag
498*08ab5237SOystein Eftevaag class CommandLineFlag {
499*08ab5237SOystein Eftevaag public:
500*08ab5237SOystein Eftevaag // Note: we take over memory-ownership of current_val and default_val.
501*08ab5237SOystein Eftevaag CommandLineFlag(const char* name, const char* help, const char* filename,
502*08ab5237SOystein Eftevaag FlagValue* current_val, FlagValue* default_val);
503*08ab5237SOystein Eftevaag ~CommandLineFlag();
504*08ab5237SOystein Eftevaag
name() const505*08ab5237SOystein Eftevaag const char* name() const { return name_; }
help() const506*08ab5237SOystein Eftevaag const char* help() const { return help_; }
filename() const507*08ab5237SOystein Eftevaag const char* filename() const { return file_; }
508*08ab5237SOystein Eftevaag const char* CleanFileName() const; // nixes irrelevant prefix such as homedir
current_value() const509*08ab5237SOystein Eftevaag string current_value() const { return current_->ToString(); }
default_value() const510*08ab5237SOystein Eftevaag string default_value() const { return defvalue_->ToString(); }
type_name() const511*08ab5237SOystein Eftevaag const char* type_name() const { return defvalue_->TypeName(); }
validate_function() const512*08ab5237SOystein Eftevaag ValidateFnProto validate_function() const { return validate_fn_proto_; }
flag_ptr() const513*08ab5237SOystein Eftevaag const void* flag_ptr() const { return current_->value_buffer_; }
514*08ab5237SOystein Eftevaag
Type() const515*08ab5237SOystein Eftevaag FlagValue::ValueType Type() const { return defvalue_->Type(); }
516*08ab5237SOystein Eftevaag
517*08ab5237SOystein Eftevaag void FillCommandLineFlagInfo(struct CommandLineFlagInfo* result);
518*08ab5237SOystein Eftevaag
519*08ab5237SOystein Eftevaag // If validate_fn_proto_ is non-NULL, calls it on value, returns result.
520*08ab5237SOystein Eftevaag bool Validate(const FlagValue& value) const;
ValidateCurrent() const521*08ab5237SOystein Eftevaag bool ValidateCurrent() const { return Validate(*current_); }
Modified() const522*08ab5237SOystein Eftevaag bool Modified() const { return modified_; }
523*08ab5237SOystein Eftevaag
524*08ab5237SOystein Eftevaag private:
525*08ab5237SOystein Eftevaag // for SetFlagLocked() and setting flags_by_ptr_
526*08ab5237SOystein Eftevaag friend class FlagRegistry;
527*08ab5237SOystein Eftevaag friend class GFLAGS_NAMESPACE::FlagSaverImpl; // for cloning the values
528*08ab5237SOystein Eftevaag // set validate_fn
529*08ab5237SOystein Eftevaag friend bool AddFlagValidator(const void*, ValidateFnProto);
530*08ab5237SOystein Eftevaag
531*08ab5237SOystein Eftevaag // This copies all the non-const members: modified, processed, defvalue, etc.
532*08ab5237SOystein Eftevaag void CopyFrom(const CommandLineFlag& src);
533*08ab5237SOystein Eftevaag
534*08ab5237SOystein Eftevaag void UpdateModifiedBit();
535*08ab5237SOystein Eftevaag
536*08ab5237SOystein Eftevaag const char* const name_; // Flag name
537*08ab5237SOystein Eftevaag const char* const help_; // Help message
538*08ab5237SOystein Eftevaag const char* const file_; // Which file did this come from?
539*08ab5237SOystein Eftevaag bool modified_; // Set after default assignment?
540*08ab5237SOystein Eftevaag FlagValue* defvalue_; // Default value for flag
541*08ab5237SOystein Eftevaag FlagValue* current_; // Current value for flag
542*08ab5237SOystein Eftevaag // This is a casted, 'generic' version of validate_fn, which actually
543*08ab5237SOystein Eftevaag // takes a flag-value as an arg (void (*validate_fn)(bool), say).
544*08ab5237SOystein Eftevaag // When we pass this to current_->Validate(), it will cast it back to
545*08ab5237SOystein Eftevaag // the proper type. This may be NULL to mean we have no validate_fn.
546*08ab5237SOystein Eftevaag ValidateFnProto validate_fn_proto_;
547*08ab5237SOystein Eftevaag
548*08ab5237SOystein Eftevaag CommandLineFlag(const CommandLineFlag&); // no copying!
549*08ab5237SOystein Eftevaag void operator=(const CommandLineFlag&);
550*08ab5237SOystein Eftevaag };
551*08ab5237SOystein Eftevaag
CommandLineFlag(const char * name,const char * help,const char * filename,FlagValue * current_val,FlagValue * default_val)552*08ab5237SOystein Eftevaag CommandLineFlag::CommandLineFlag(const char* name, const char* help,
553*08ab5237SOystein Eftevaag const char* filename,
554*08ab5237SOystein Eftevaag FlagValue* current_val, FlagValue* default_val)
555*08ab5237SOystein Eftevaag : name_(name), help_(help), file_(filename), modified_(false),
556*08ab5237SOystein Eftevaag defvalue_(default_val), current_(current_val), validate_fn_proto_(NULL) {
557*08ab5237SOystein Eftevaag }
558*08ab5237SOystein Eftevaag
~CommandLineFlag()559*08ab5237SOystein Eftevaag CommandLineFlag::~CommandLineFlag() {
560*08ab5237SOystein Eftevaag delete current_;
561*08ab5237SOystein Eftevaag delete defvalue_;
562*08ab5237SOystein Eftevaag }
563*08ab5237SOystein Eftevaag
CleanFileName() const564*08ab5237SOystein Eftevaag const char* CommandLineFlag::CleanFileName() const {
565*08ab5237SOystein Eftevaag // This function has been used to strip off a common prefix from
566*08ab5237SOystein Eftevaag // flag source file names. Because flags can be defined in different
567*08ab5237SOystein Eftevaag // shared libraries, there may not be a single common prefix.
568*08ab5237SOystein Eftevaag // Further, this functionality hasn't been active for many years.
569*08ab5237SOystein Eftevaag // Need a better way to produce more user friendly help output or
570*08ab5237SOystein Eftevaag // "anonymize" file paths in help output, respectively.
571*08ab5237SOystein Eftevaag // Follow issue at: https://github.com/gflags/gflags/issues/86
572*08ab5237SOystein Eftevaag return filename();
573*08ab5237SOystein Eftevaag }
574*08ab5237SOystein Eftevaag
FillCommandLineFlagInfo(CommandLineFlagInfo * result)575*08ab5237SOystein Eftevaag void CommandLineFlag::FillCommandLineFlagInfo(
576*08ab5237SOystein Eftevaag CommandLineFlagInfo* result) {
577*08ab5237SOystein Eftevaag result->name = name();
578*08ab5237SOystein Eftevaag result->type = type_name();
579*08ab5237SOystein Eftevaag result->description = help();
580*08ab5237SOystein Eftevaag result->current_value = current_value();
581*08ab5237SOystein Eftevaag result->default_value = default_value();
582*08ab5237SOystein Eftevaag result->filename = CleanFileName();
583*08ab5237SOystein Eftevaag UpdateModifiedBit();
584*08ab5237SOystein Eftevaag result->is_default = !modified_;
585*08ab5237SOystein Eftevaag result->has_validator_fn = validate_function() != NULL;
586*08ab5237SOystein Eftevaag result->flag_ptr = flag_ptr();
587*08ab5237SOystein Eftevaag }
588*08ab5237SOystein Eftevaag
UpdateModifiedBit()589*08ab5237SOystein Eftevaag void CommandLineFlag::UpdateModifiedBit() {
590*08ab5237SOystein Eftevaag // Update the "modified" bit in case somebody bypassed the
591*08ab5237SOystein Eftevaag // Flags API and wrote directly through the FLAGS_name variable.
592*08ab5237SOystein Eftevaag if (!modified_ && !current_->Equal(*defvalue_)) {
593*08ab5237SOystein Eftevaag modified_ = true;
594*08ab5237SOystein Eftevaag }
595*08ab5237SOystein Eftevaag }
596*08ab5237SOystein Eftevaag
CopyFrom(const CommandLineFlag & src)597*08ab5237SOystein Eftevaag void CommandLineFlag::CopyFrom(const CommandLineFlag& src) {
598*08ab5237SOystein Eftevaag // Note we only copy the non-const members; others are fixed at construct time
599*08ab5237SOystein Eftevaag if (modified_ != src.modified_) modified_ = src.modified_;
600*08ab5237SOystein Eftevaag if (!current_->Equal(*src.current_)) current_->CopyFrom(*src.current_);
601*08ab5237SOystein Eftevaag if (!defvalue_->Equal(*src.defvalue_)) defvalue_->CopyFrom(*src.defvalue_);
602*08ab5237SOystein Eftevaag if (validate_fn_proto_ != src.validate_fn_proto_)
603*08ab5237SOystein Eftevaag validate_fn_proto_ = src.validate_fn_proto_;
604*08ab5237SOystein Eftevaag }
605*08ab5237SOystein Eftevaag
Validate(const FlagValue & value) const606*08ab5237SOystein Eftevaag bool CommandLineFlag::Validate(const FlagValue& value) const {
607*08ab5237SOystein Eftevaag
608*08ab5237SOystein Eftevaag if (validate_function() == NULL)
609*08ab5237SOystein Eftevaag return true;
610*08ab5237SOystein Eftevaag else
611*08ab5237SOystein Eftevaag return value.Validate(name(), validate_function());
612*08ab5237SOystein Eftevaag }
613*08ab5237SOystein Eftevaag
614*08ab5237SOystein Eftevaag
615*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
616*08ab5237SOystein Eftevaag // FlagRegistry
617*08ab5237SOystein Eftevaag // A FlagRegistry singleton object holds all flag objects indexed
618*08ab5237SOystein Eftevaag // by their names so that if you know a flag's name (as a C
619*08ab5237SOystein Eftevaag // string), you can access or set it. If the function is named
620*08ab5237SOystein Eftevaag // FooLocked(), you must own the registry lock before calling
621*08ab5237SOystein Eftevaag // the function; otherwise, you should *not* hold the lock, and
622*08ab5237SOystein Eftevaag // the function will acquire it itself if needed.
623*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
624*08ab5237SOystein Eftevaag
625*08ab5237SOystein Eftevaag struct StringCmp { // Used by the FlagRegistry map class to compare char*'s
operator ()GFLAGS_NAMESPACE::__anonab228c690111::StringCmp626*08ab5237SOystein Eftevaag bool operator() (const char* s1, const char* s2) const {
627*08ab5237SOystein Eftevaag return (strcmp(s1, s2) < 0);
628*08ab5237SOystein Eftevaag }
629*08ab5237SOystein Eftevaag };
630*08ab5237SOystein Eftevaag
631*08ab5237SOystein Eftevaag
632*08ab5237SOystein Eftevaag class FlagRegistry {
633*08ab5237SOystein Eftevaag public:
FlagRegistry()634*08ab5237SOystein Eftevaag FlagRegistry() {
635*08ab5237SOystein Eftevaag }
~FlagRegistry()636*08ab5237SOystein Eftevaag ~FlagRegistry() {
637*08ab5237SOystein Eftevaag // Not using STLDeleteElements as that resides in util and this
638*08ab5237SOystein Eftevaag // class is base.
639*08ab5237SOystein Eftevaag for (FlagMap::iterator p = flags_.begin(), e = flags_.end(); p != e; ++p) {
640*08ab5237SOystein Eftevaag CommandLineFlag* flag = p->second;
641*08ab5237SOystein Eftevaag delete flag;
642*08ab5237SOystein Eftevaag }
643*08ab5237SOystein Eftevaag }
644*08ab5237SOystein Eftevaag
DeleteGlobalRegistry()645*08ab5237SOystein Eftevaag static void DeleteGlobalRegistry() {
646*08ab5237SOystein Eftevaag delete global_registry_;
647*08ab5237SOystein Eftevaag global_registry_ = NULL;
648*08ab5237SOystein Eftevaag }
649*08ab5237SOystein Eftevaag
650*08ab5237SOystein Eftevaag // Store a flag in this registry. Takes ownership of the given pointer.
651*08ab5237SOystein Eftevaag void RegisterFlag(CommandLineFlag* flag);
652*08ab5237SOystein Eftevaag
Lock()653*08ab5237SOystein Eftevaag void Lock() { lock_.Lock(); }
Unlock()654*08ab5237SOystein Eftevaag void Unlock() { lock_.Unlock(); }
655*08ab5237SOystein Eftevaag
656*08ab5237SOystein Eftevaag // Returns the flag object for the specified name, or NULL if not found.
657*08ab5237SOystein Eftevaag CommandLineFlag* FindFlagLocked(const char* name);
658*08ab5237SOystein Eftevaag
659*08ab5237SOystein Eftevaag // Returns the flag object whose current-value is stored at flag_ptr.
660*08ab5237SOystein Eftevaag // That is, for whom current_->value_buffer_ == flag_ptr
661*08ab5237SOystein Eftevaag CommandLineFlag* FindFlagViaPtrLocked(const void* flag_ptr);
662*08ab5237SOystein Eftevaag
663*08ab5237SOystein Eftevaag // A fancier form of FindFlag that works correctly if name is of the
664*08ab5237SOystein Eftevaag // form flag=value. In that case, we set key to point to flag, and
665*08ab5237SOystein Eftevaag // modify v to point to the value (if present), and return the flag
666*08ab5237SOystein Eftevaag // with the given name. If the flag does not exist, returns NULL
667*08ab5237SOystein Eftevaag // and sets error_message.
668*08ab5237SOystein Eftevaag CommandLineFlag* SplitArgumentLocked(const char* argument,
669*08ab5237SOystein Eftevaag string* key, const char** v,
670*08ab5237SOystein Eftevaag string* error_message);
671*08ab5237SOystein Eftevaag
672*08ab5237SOystein Eftevaag // Set the value of a flag. If the flag was successfully set to
673*08ab5237SOystein Eftevaag // value, set msg to indicate the new flag-value, and return true.
674*08ab5237SOystein Eftevaag // Otherwise, set msg to indicate the error, leave flag unchanged,
675*08ab5237SOystein Eftevaag // and return false. msg can be NULL.
676*08ab5237SOystein Eftevaag bool SetFlagLocked(CommandLineFlag* flag, const char* value,
677*08ab5237SOystein Eftevaag FlagSettingMode set_mode, string* msg);
678*08ab5237SOystein Eftevaag
679*08ab5237SOystein Eftevaag static FlagRegistry* GlobalRegistry(); // returns a singleton registry
680*08ab5237SOystein Eftevaag
681*08ab5237SOystein Eftevaag private:
682*08ab5237SOystein Eftevaag friend class GFLAGS_NAMESPACE::FlagSaverImpl; // reads all the flags in order to copy them
683*08ab5237SOystein Eftevaag friend class CommandLineFlagParser; // for ValidateUnmodifiedFlags
684*08ab5237SOystein Eftevaag friend void GFLAGS_NAMESPACE::GetAllFlags(vector<CommandLineFlagInfo>*);
685*08ab5237SOystein Eftevaag
686*08ab5237SOystein Eftevaag // The map from name to flag, for FindFlagLocked().
687*08ab5237SOystein Eftevaag typedef map<const char*, CommandLineFlag*, StringCmp> FlagMap;
688*08ab5237SOystein Eftevaag typedef FlagMap::iterator FlagIterator;
689*08ab5237SOystein Eftevaag typedef FlagMap::const_iterator FlagConstIterator;
690*08ab5237SOystein Eftevaag FlagMap flags_;
691*08ab5237SOystein Eftevaag
692*08ab5237SOystein Eftevaag // The map from current-value pointer to flag, fo FindFlagViaPtrLocked().
693*08ab5237SOystein Eftevaag typedef map<const void*, CommandLineFlag*> FlagPtrMap;
694*08ab5237SOystein Eftevaag FlagPtrMap flags_by_ptr_;
695*08ab5237SOystein Eftevaag
696*08ab5237SOystein Eftevaag static FlagRegistry* global_registry_; // a singleton registry
697*08ab5237SOystein Eftevaag
698*08ab5237SOystein Eftevaag Mutex lock_;
699*08ab5237SOystein Eftevaag
700*08ab5237SOystein Eftevaag static void InitGlobalRegistry();
701*08ab5237SOystein Eftevaag
702*08ab5237SOystein Eftevaag // Disallow
703*08ab5237SOystein Eftevaag FlagRegistry(const FlagRegistry&);
704*08ab5237SOystein Eftevaag FlagRegistry& operator=(const FlagRegistry&);
705*08ab5237SOystein Eftevaag };
706*08ab5237SOystein Eftevaag
707*08ab5237SOystein Eftevaag class FlagRegistryLock {
708*08ab5237SOystein Eftevaag public:
FlagRegistryLock(FlagRegistry * fr)709*08ab5237SOystein Eftevaag explicit FlagRegistryLock(FlagRegistry* fr) : fr_(fr) { fr_->Lock(); }
~FlagRegistryLock()710*08ab5237SOystein Eftevaag ~FlagRegistryLock() { fr_->Unlock(); }
711*08ab5237SOystein Eftevaag private:
712*08ab5237SOystein Eftevaag FlagRegistry *const fr_;
713*08ab5237SOystein Eftevaag };
714*08ab5237SOystein Eftevaag
715*08ab5237SOystein Eftevaag
RegisterFlag(CommandLineFlag * flag)716*08ab5237SOystein Eftevaag void FlagRegistry::RegisterFlag(CommandLineFlag* flag) {
717*08ab5237SOystein Eftevaag Lock();
718*08ab5237SOystein Eftevaag pair<FlagIterator, bool> ins =
719*08ab5237SOystein Eftevaag flags_.insert(pair<const char*, CommandLineFlag*>(flag->name(), flag));
720*08ab5237SOystein Eftevaag if (ins.second == false) { // means the name was already in the map
721*08ab5237SOystein Eftevaag if (strcmp(ins.first->second->filename(), flag->filename()) != 0) {
722*08ab5237SOystein Eftevaag ReportError(DIE, "ERROR: flag '%s' was defined more than once "
723*08ab5237SOystein Eftevaag "(in files '%s' and '%s').\n",
724*08ab5237SOystein Eftevaag flag->name(),
725*08ab5237SOystein Eftevaag ins.first->second->filename(),
726*08ab5237SOystein Eftevaag flag->filename());
727*08ab5237SOystein Eftevaag } else {
728*08ab5237SOystein Eftevaag ReportError(DIE, "ERROR: something wrong with flag '%s' in file '%s'. "
729*08ab5237SOystein Eftevaag "One possibility: file '%s' is being linked both statically "
730*08ab5237SOystein Eftevaag "and dynamically into this executable.\n",
731*08ab5237SOystein Eftevaag flag->name(),
732*08ab5237SOystein Eftevaag flag->filename(), flag->filename());
733*08ab5237SOystein Eftevaag }
734*08ab5237SOystein Eftevaag }
735*08ab5237SOystein Eftevaag // Also add to the flags_by_ptr_ map.
736*08ab5237SOystein Eftevaag flags_by_ptr_[flag->current_->value_buffer_] = flag;
737*08ab5237SOystein Eftevaag Unlock();
738*08ab5237SOystein Eftevaag }
739*08ab5237SOystein Eftevaag
FindFlagLocked(const char * name)740*08ab5237SOystein Eftevaag CommandLineFlag* FlagRegistry::FindFlagLocked(const char* name) {
741*08ab5237SOystein Eftevaag FlagConstIterator i = flags_.find(name);
742*08ab5237SOystein Eftevaag if (i == flags_.end()) {
743*08ab5237SOystein Eftevaag // If the name has dashes in it, try again after replacing with
744*08ab5237SOystein Eftevaag // underscores.
745*08ab5237SOystein Eftevaag if (strchr(name, '-') == NULL) return NULL;
746*08ab5237SOystein Eftevaag string name_rep = name;
747*08ab5237SOystein Eftevaag std::replace(name_rep.begin(), name_rep.end(), '-', '_');
748*08ab5237SOystein Eftevaag return FindFlagLocked(name_rep.c_str());
749*08ab5237SOystein Eftevaag } else {
750*08ab5237SOystein Eftevaag return i->second;
751*08ab5237SOystein Eftevaag }
752*08ab5237SOystein Eftevaag }
753*08ab5237SOystein Eftevaag
FindFlagViaPtrLocked(const void * flag_ptr)754*08ab5237SOystein Eftevaag CommandLineFlag* FlagRegistry::FindFlagViaPtrLocked(const void* flag_ptr) {
755*08ab5237SOystein Eftevaag FlagPtrMap::const_iterator i = flags_by_ptr_.find(flag_ptr);
756*08ab5237SOystein Eftevaag if (i == flags_by_ptr_.end()) {
757*08ab5237SOystein Eftevaag return NULL;
758*08ab5237SOystein Eftevaag } else {
759*08ab5237SOystein Eftevaag return i->second;
760*08ab5237SOystein Eftevaag }
761*08ab5237SOystein Eftevaag }
762*08ab5237SOystein Eftevaag
SplitArgumentLocked(const char * arg,string * key,const char ** v,string * error_message)763*08ab5237SOystein Eftevaag CommandLineFlag* FlagRegistry::SplitArgumentLocked(const char* arg,
764*08ab5237SOystein Eftevaag string* key,
765*08ab5237SOystein Eftevaag const char** v,
766*08ab5237SOystein Eftevaag string* error_message) {
767*08ab5237SOystein Eftevaag // Find the flag object for this option
768*08ab5237SOystein Eftevaag const char* flag_name;
769*08ab5237SOystein Eftevaag const char* value = strchr(arg, '=');
770*08ab5237SOystein Eftevaag if (value == NULL) {
771*08ab5237SOystein Eftevaag key->assign(arg);
772*08ab5237SOystein Eftevaag *v = NULL;
773*08ab5237SOystein Eftevaag } else {
774*08ab5237SOystein Eftevaag // Strip out the "=value" portion from arg
775*08ab5237SOystein Eftevaag key->assign(arg, value-arg);
776*08ab5237SOystein Eftevaag *v = ++value; // advance past the '='
777*08ab5237SOystein Eftevaag }
778*08ab5237SOystein Eftevaag flag_name = key->c_str();
779*08ab5237SOystein Eftevaag
780*08ab5237SOystein Eftevaag CommandLineFlag* flag = FindFlagLocked(flag_name);
781*08ab5237SOystein Eftevaag
782*08ab5237SOystein Eftevaag if (flag == NULL) {
783*08ab5237SOystein Eftevaag // If we can't find the flag-name, then we should return an error.
784*08ab5237SOystein Eftevaag // The one exception is if 1) the flag-name is 'nox', 2) there
785*08ab5237SOystein Eftevaag // exists a flag named 'x', and 3) 'x' is a boolean flag.
786*08ab5237SOystein Eftevaag // In that case, we want to return flag 'x'.
787*08ab5237SOystein Eftevaag if (!(flag_name[0] == 'n' && flag_name[1] == 'o')) {
788*08ab5237SOystein Eftevaag // flag-name is not 'nox', so we're not in the exception case.
789*08ab5237SOystein Eftevaag *error_message = StringPrintf("%sunknown command line flag '%s'\n",
790*08ab5237SOystein Eftevaag kError, key->c_str());
791*08ab5237SOystein Eftevaag return NULL;
792*08ab5237SOystein Eftevaag }
793*08ab5237SOystein Eftevaag flag = FindFlagLocked(flag_name+2);
794*08ab5237SOystein Eftevaag if (flag == NULL) {
795*08ab5237SOystein Eftevaag // No flag named 'x' exists, so we're not in the exception case.
796*08ab5237SOystein Eftevaag *error_message = StringPrintf("%sunknown command line flag '%s'\n",
797*08ab5237SOystein Eftevaag kError, key->c_str());
798*08ab5237SOystein Eftevaag return NULL;
799*08ab5237SOystein Eftevaag }
800*08ab5237SOystein Eftevaag if (flag->Type() != FlagValue::FV_BOOL) {
801*08ab5237SOystein Eftevaag // 'x' exists but is not boolean, so we're not in the exception case.
802*08ab5237SOystein Eftevaag *error_message = StringPrintf(
803*08ab5237SOystein Eftevaag "%sboolean value (%s) specified for %s command line flag\n",
804*08ab5237SOystein Eftevaag kError, key->c_str(), flag->type_name());
805*08ab5237SOystein Eftevaag return NULL;
806*08ab5237SOystein Eftevaag }
807*08ab5237SOystein Eftevaag // We're in the exception case!
808*08ab5237SOystein Eftevaag // Make up a fake value to replace the "no" we stripped out
809*08ab5237SOystein Eftevaag key->assign(flag_name+2); // the name without the "no"
810*08ab5237SOystein Eftevaag *v = "0";
811*08ab5237SOystein Eftevaag }
812*08ab5237SOystein Eftevaag
813*08ab5237SOystein Eftevaag // Assign a value if this is a boolean flag
814*08ab5237SOystein Eftevaag if (*v == NULL && flag->Type() == FlagValue::FV_BOOL) {
815*08ab5237SOystein Eftevaag *v = "1"; // the --nox case was already handled, so this is the --x case
816*08ab5237SOystein Eftevaag }
817*08ab5237SOystein Eftevaag
818*08ab5237SOystein Eftevaag return flag;
819*08ab5237SOystein Eftevaag }
820*08ab5237SOystein Eftevaag
TryParseLocked(const CommandLineFlag * flag,FlagValue * flag_value,const char * value,string * msg)821*08ab5237SOystein Eftevaag bool TryParseLocked(const CommandLineFlag* flag, FlagValue* flag_value,
822*08ab5237SOystein Eftevaag const char* value, string* msg) {
823*08ab5237SOystein Eftevaag // Use tenative_value, not flag_value, until we know value is valid.
824*08ab5237SOystein Eftevaag FlagValue* tentative_value = flag_value->New();
825*08ab5237SOystein Eftevaag if (!tentative_value->ParseFrom(value)) {
826*08ab5237SOystein Eftevaag if (msg) {
827*08ab5237SOystein Eftevaag StringAppendF(msg,
828*08ab5237SOystein Eftevaag "%sillegal value '%s' specified for %s flag '%s'\n",
829*08ab5237SOystein Eftevaag kError, value,
830*08ab5237SOystein Eftevaag flag->type_name(), flag->name());
831*08ab5237SOystein Eftevaag }
832*08ab5237SOystein Eftevaag delete tentative_value;
833*08ab5237SOystein Eftevaag return false;
834*08ab5237SOystein Eftevaag } else if (!flag->Validate(*tentative_value)) {
835*08ab5237SOystein Eftevaag if (msg) {
836*08ab5237SOystein Eftevaag StringAppendF(msg,
837*08ab5237SOystein Eftevaag "%sfailed validation of new value '%s' for flag '%s'\n",
838*08ab5237SOystein Eftevaag kError, tentative_value->ToString().c_str(),
839*08ab5237SOystein Eftevaag flag->name());
840*08ab5237SOystein Eftevaag }
841*08ab5237SOystein Eftevaag delete tentative_value;
842*08ab5237SOystein Eftevaag return false;
843*08ab5237SOystein Eftevaag } else {
844*08ab5237SOystein Eftevaag flag_value->CopyFrom(*tentative_value);
845*08ab5237SOystein Eftevaag if (msg) {
846*08ab5237SOystein Eftevaag StringAppendF(msg, "%s set to %s\n",
847*08ab5237SOystein Eftevaag flag->name(), flag_value->ToString().c_str());
848*08ab5237SOystein Eftevaag }
849*08ab5237SOystein Eftevaag delete tentative_value;
850*08ab5237SOystein Eftevaag return true;
851*08ab5237SOystein Eftevaag }
852*08ab5237SOystein Eftevaag }
853*08ab5237SOystein Eftevaag
SetFlagLocked(CommandLineFlag * flag,const char * value,FlagSettingMode set_mode,string * msg)854*08ab5237SOystein Eftevaag bool FlagRegistry::SetFlagLocked(CommandLineFlag* flag,
855*08ab5237SOystein Eftevaag const char* value,
856*08ab5237SOystein Eftevaag FlagSettingMode set_mode,
857*08ab5237SOystein Eftevaag string* msg) {
858*08ab5237SOystein Eftevaag flag->UpdateModifiedBit();
859*08ab5237SOystein Eftevaag switch (set_mode) {
860*08ab5237SOystein Eftevaag case SET_FLAGS_VALUE: {
861*08ab5237SOystein Eftevaag // set or modify the flag's value
862*08ab5237SOystein Eftevaag if (!TryParseLocked(flag, flag->current_, value, msg))
863*08ab5237SOystein Eftevaag return false;
864*08ab5237SOystein Eftevaag flag->modified_ = true;
865*08ab5237SOystein Eftevaag break;
866*08ab5237SOystein Eftevaag }
867*08ab5237SOystein Eftevaag case SET_FLAG_IF_DEFAULT: {
868*08ab5237SOystein Eftevaag // set the flag's value, but only if it hasn't been set by someone else
869*08ab5237SOystein Eftevaag if (!flag->modified_) {
870*08ab5237SOystein Eftevaag if (!TryParseLocked(flag, flag->current_, value, msg))
871*08ab5237SOystein Eftevaag return false;
872*08ab5237SOystein Eftevaag flag->modified_ = true;
873*08ab5237SOystein Eftevaag } else {
874*08ab5237SOystein Eftevaag *msg = StringPrintf("%s set to %s",
875*08ab5237SOystein Eftevaag flag->name(), flag->current_value().c_str());
876*08ab5237SOystein Eftevaag }
877*08ab5237SOystein Eftevaag break;
878*08ab5237SOystein Eftevaag }
879*08ab5237SOystein Eftevaag case SET_FLAGS_DEFAULT: {
880*08ab5237SOystein Eftevaag // modify the flag's default-value
881*08ab5237SOystein Eftevaag if (!TryParseLocked(flag, flag->defvalue_, value, msg))
882*08ab5237SOystein Eftevaag return false;
883*08ab5237SOystein Eftevaag if (!flag->modified_) {
884*08ab5237SOystein Eftevaag // Need to set both defvalue *and* current, in this case
885*08ab5237SOystein Eftevaag TryParseLocked(flag, flag->current_, value, NULL);
886*08ab5237SOystein Eftevaag }
887*08ab5237SOystein Eftevaag break;
888*08ab5237SOystein Eftevaag }
889*08ab5237SOystein Eftevaag default: {
890*08ab5237SOystein Eftevaag // unknown set_mode
891*08ab5237SOystein Eftevaag assert(false);
892*08ab5237SOystein Eftevaag return false;
893*08ab5237SOystein Eftevaag }
894*08ab5237SOystein Eftevaag }
895*08ab5237SOystein Eftevaag
896*08ab5237SOystein Eftevaag return true;
897*08ab5237SOystein Eftevaag }
898*08ab5237SOystein Eftevaag
899*08ab5237SOystein Eftevaag // Get the singleton FlagRegistry object
900*08ab5237SOystein Eftevaag FlagRegistry* FlagRegistry::global_registry_ = NULL;
901*08ab5237SOystein Eftevaag
GlobalRegistry()902*08ab5237SOystein Eftevaag FlagRegistry* FlagRegistry::GlobalRegistry() {
903*08ab5237SOystein Eftevaag static Mutex lock(Mutex::LINKER_INITIALIZED);
904*08ab5237SOystein Eftevaag MutexLock acquire_lock(&lock);
905*08ab5237SOystein Eftevaag if (!global_registry_) {
906*08ab5237SOystein Eftevaag global_registry_ = new FlagRegistry;
907*08ab5237SOystein Eftevaag }
908*08ab5237SOystein Eftevaag return global_registry_;
909*08ab5237SOystein Eftevaag }
910*08ab5237SOystein Eftevaag
911*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
912*08ab5237SOystein Eftevaag // CommandLineFlagParser
913*08ab5237SOystein Eftevaag // Parsing is done in two stages. In the first, we go through
914*08ab5237SOystein Eftevaag // argv. For every flag-like arg we can make sense of, we parse
915*08ab5237SOystein Eftevaag // it and set the appropriate FLAGS_* variable. For every flag-
916*08ab5237SOystein Eftevaag // like arg we can't make sense of, we store it in a vector,
917*08ab5237SOystein Eftevaag // along with an explanation of the trouble. In stage 2, we
918*08ab5237SOystein Eftevaag // handle the 'reporting' flags like --help and --mpm_version.
919*08ab5237SOystein Eftevaag // (This is via a call to HandleCommandLineHelpFlags(), in
920*08ab5237SOystein Eftevaag // gflags_reporting.cc.)
921*08ab5237SOystein Eftevaag // An optional stage 3 prints out the error messages.
922*08ab5237SOystein Eftevaag // This is a bit of a simplification. For instance, --flagfile
923*08ab5237SOystein Eftevaag // is handled as soon as it's seen in stage 1, not in stage 2.
924*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
925*08ab5237SOystein Eftevaag
926*08ab5237SOystein Eftevaag class CommandLineFlagParser {
927*08ab5237SOystein Eftevaag public:
928*08ab5237SOystein Eftevaag // The argument is the flag-registry to register the parsed flags in
CommandLineFlagParser(FlagRegistry * reg)929*08ab5237SOystein Eftevaag explicit CommandLineFlagParser(FlagRegistry* reg) : registry_(reg) {}
~CommandLineFlagParser()930*08ab5237SOystein Eftevaag ~CommandLineFlagParser() {}
931*08ab5237SOystein Eftevaag
932*08ab5237SOystein Eftevaag // Stage 1: Every time this is called, it reads all flags in argv.
933*08ab5237SOystein Eftevaag // However, it ignores all flags that have been successfully set
934*08ab5237SOystein Eftevaag // before. Typically this is only called once, so this 'reparsing'
935*08ab5237SOystein Eftevaag // behavior isn't important. It can be useful when trying to
936*08ab5237SOystein Eftevaag // reparse after loading a dll, though.
937*08ab5237SOystein Eftevaag uint32 ParseNewCommandLineFlags(int* argc, char*** argv, bool remove_flags);
938*08ab5237SOystein Eftevaag
939*08ab5237SOystein Eftevaag // Stage 2: print reporting info and exit, if requested.
940*08ab5237SOystein Eftevaag // In gflags_reporting.cc:HandleCommandLineHelpFlags().
941*08ab5237SOystein Eftevaag
942*08ab5237SOystein Eftevaag // Stage 3: validate all the commandline flags that have validators
943*08ab5237SOystein Eftevaag // registered and were not set/modified by ParseNewCommandLineFlags.
944*08ab5237SOystein Eftevaag void ValidateFlags(bool all);
945*08ab5237SOystein Eftevaag void ValidateUnmodifiedFlags();
946*08ab5237SOystein Eftevaag
947*08ab5237SOystein Eftevaag // Stage 4: report any errors and return true if any were found.
948*08ab5237SOystein Eftevaag bool ReportErrors();
949*08ab5237SOystein Eftevaag
950*08ab5237SOystein Eftevaag // Set a particular command line option. "newval" is a string
951*08ab5237SOystein Eftevaag // describing the new value that the option has been set to. If
952*08ab5237SOystein Eftevaag // option_name does not specify a valid option name, or value is not
953*08ab5237SOystein Eftevaag // a valid value for option_name, newval is empty. Does recursive
954*08ab5237SOystein Eftevaag // processing for --flagfile and --fromenv. Returns the new value
955*08ab5237SOystein Eftevaag // if everything went ok, or empty-string if not. (Actually, the
956*08ab5237SOystein Eftevaag // return-string could hold many flag/value pairs due to --flagfile.)
957*08ab5237SOystein Eftevaag // NB: Must have called registry_->Lock() before calling this function.
958*08ab5237SOystein Eftevaag string ProcessSingleOptionLocked(CommandLineFlag* flag,
959*08ab5237SOystein Eftevaag const char* value,
960*08ab5237SOystein Eftevaag FlagSettingMode set_mode);
961*08ab5237SOystein Eftevaag
962*08ab5237SOystein Eftevaag // Set a whole batch of command line options as specified by contentdata,
963*08ab5237SOystein Eftevaag // which is in flagfile format (and probably has been read from a flagfile).
964*08ab5237SOystein Eftevaag // Returns the new value if everything went ok, or empty-string if
965*08ab5237SOystein Eftevaag // not. (Actually, the return-string could hold many flag/value
966*08ab5237SOystein Eftevaag // pairs due to --flagfile.)
967*08ab5237SOystein Eftevaag // NB: Must have called registry_->Lock() before calling this function.
968*08ab5237SOystein Eftevaag string ProcessOptionsFromStringLocked(const string& contentdata,
969*08ab5237SOystein Eftevaag FlagSettingMode set_mode);
970*08ab5237SOystein Eftevaag
971*08ab5237SOystein Eftevaag // These are the 'recursive' flags, defined at the top of this file.
972*08ab5237SOystein Eftevaag // Whenever we see these flags on the commandline, we must take action.
973*08ab5237SOystein Eftevaag // These are called by ProcessSingleOptionLocked and, similarly, return
974*08ab5237SOystein Eftevaag // new values if everything went ok, or the empty-string if not.
975*08ab5237SOystein Eftevaag string ProcessFlagfileLocked(const string& flagval, FlagSettingMode set_mode);
976*08ab5237SOystein Eftevaag // diff fromenv/tryfromenv
977*08ab5237SOystein Eftevaag string ProcessFromenvLocked(const string& flagval, FlagSettingMode set_mode,
978*08ab5237SOystein Eftevaag bool errors_are_fatal);
979*08ab5237SOystein Eftevaag
980*08ab5237SOystein Eftevaag private:
981*08ab5237SOystein Eftevaag FlagRegistry* const registry_;
982*08ab5237SOystein Eftevaag map<string, string> error_flags_; // map from name to error message
983*08ab5237SOystein Eftevaag // This could be a set<string>, but we reuse the map to minimize the .o size
984*08ab5237SOystein Eftevaag map<string, string> undefined_names_; // --[flag] name was not registered
985*08ab5237SOystein Eftevaag };
986*08ab5237SOystein Eftevaag
987*08ab5237SOystein Eftevaag
988*08ab5237SOystein Eftevaag // Parse a list of (comma-separated) flags.
ParseFlagList(const char * value,vector<string> * flags)989*08ab5237SOystein Eftevaag static void ParseFlagList(const char* value, vector<string>* flags) {
990*08ab5237SOystein Eftevaag for (const char *p = value; p && *p; value = p) {
991*08ab5237SOystein Eftevaag p = strchr(value, ',');
992*08ab5237SOystein Eftevaag size_t len;
993*08ab5237SOystein Eftevaag if (p) {
994*08ab5237SOystein Eftevaag len = p - value;
995*08ab5237SOystein Eftevaag p++;
996*08ab5237SOystein Eftevaag } else {
997*08ab5237SOystein Eftevaag len = strlen(value);
998*08ab5237SOystein Eftevaag }
999*08ab5237SOystein Eftevaag
1000*08ab5237SOystein Eftevaag if (len == 0)
1001*08ab5237SOystein Eftevaag ReportError(DIE, "ERROR: empty flaglist entry\n");
1002*08ab5237SOystein Eftevaag if (value[0] == '-')
1003*08ab5237SOystein Eftevaag ReportError(DIE, "ERROR: flag \"%*s\" begins with '-'\n", len, value);
1004*08ab5237SOystein Eftevaag
1005*08ab5237SOystein Eftevaag flags->push_back(string(value, len));
1006*08ab5237SOystein Eftevaag }
1007*08ab5237SOystein Eftevaag }
1008*08ab5237SOystein Eftevaag
1009*08ab5237SOystein Eftevaag // Snarf an entire file into a C++ string. This is just so that we
1010*08ab5237SOystein Eftevaag // can do all the I/O in one place and not worry about it everywhere.
1011*08ab5237SOystein Eftevaag // Plus, it's convenient to have the whole file contents at hand.
1012*08ab5237SOystein Eftevaag // Adds a newline at the end of the file.
1013*08ab5237SOystein Eftevaag #define PFATAL(s) do { perror(s); gflags_exitfunc(1); } while (0)
1014*08ab5237SOystein Eftevaag
ReadFileIntoString(const char * filename)1015*08ab5237SOystein Eftevaag static string ReadFileIntoString(const char* filename) {
1016*08ab5237SOystein Eftevaag const int kBufSize = 8092;
1017*08ab5237SOystein Eftevaag char buffer[kBufSize];
1018*08ab5237SOystein Eftevaag string s;
1019*08ab5237SOystein Eftevaag FILE* fp;
1020*08ab5237SOystein Eftevaag if ((errno = SafeFOpen(&fp, filename, "r")) != 0) PFATAL(filename);
1021*08ab5237SOystein Eftevaag size_t n;
1022*08ab5237SOystein Eftevaag while ( (n=fread(buffer, 1, kBufSize, fp)) > 0 ) {
1023*08ab5237SOystein Eftevaag if (ferror(fp)) PFATAL(filename);
1024*08ab5237SOystein Eftevaag s.append(buffer, n);
1025*08ab5237SOystein Eftevaag }
1026*08ab5237SOystein Eftevaag fclose(fp);
1027*08ab5237SOystein Eftevaag return s;
1028*08ab5237SOystein Eftevaag }
1029*08ab5237SOystein Eftevaag
ParseNewCommandLineFlags(int * argc,char *** argv,bool remove_flags)1030*08ab5237SOystein Eftevaag uint32 CommandLineFlagParser::ParseNewCommandLineFlags(int* argc, char*** argv,
1031*08ab5237SOystein Eftevaag bool remove_flags) {
1032*08ab5237SOystein Eftevaag int first_nonopt = *argc; // for non-options moved to the end
1033*08ab5237SOystein Eftevaag
1034*08ab5237SOystein Eftevaag registry_->Lock();
1035*08ab5237SOystein Eftevaag for (int i = 1; i < first_nonopt; i++) {
1036*08ab5237SOystein Eftevaag char* arg = (*argv)[i];
1037*08ab5237SOystein Eftevaag
1038*08ab5237SOystein Eftevaag // Like getopt(), we permute non-option flags to be at the end.
1039*08ab5237SOystein Eftevaag if (arg[0] != '-' || arg[1] == '\0') { // must be a program argument: "-" is an argument, not a flag
1040*08ab5237SOystein Eftevaag memmove((*argv) + i, (*argv) + i+1, (*argc - (i+1)) * sizeof((*argv)[i]));
1041*08ab5237SOystein Eftevaag (*argv)[*argc-1] = arg; // we go last
1042*08ab5237SOystein Eftevaag first_nonopt--; // we've been pushed onto the stack
1043*08ab5237SOystein Eftevaag i--; // to undo the i++ in the loop
1044*08ab5237SOystein Eftevaag continue;
1045*08ab5237SOystein Eftevaag }
1046*08ab5237SOystein Eftevaag arg++; // skip leading '-'
1047*08ab5237SOystein Eftevaag if (arg[0] == '-') arg++; // or leading '--'
1048*08ab5237SOystein Eftevaag
1049*08ab5237SOystein Eftevaag // -- alone means what it does for GNU: stop options parsing
1050*08ab5237SOystein Eftevaag if (*arg == '\0') {
1051*08ab5237SOystein Eftevaag first_nonopt = i+1;
1052*08ab5237SOystein Eftevaag break;
1053*08ab5237SOystein Eftevaag }
1054*08ab5237SOystein Eftevaag
1055*08ab5237SOystein Eftevaag // Find the flag object for this option
1056*08ab5237SOystein Eftevaag string key;
1057*08ab5237SOystein Eftevaag const char* value;
1058*08ab5237SOystein Eftevaag string error_message;
1059*08ab5237SOystein Eftevaag CommandLineFlag* flag = registry_->SplitArgumentLocked(arg, &key, &value,
1060*08ab5237SOystein Eftevaag &error_message);
1061*08ab5237SOystein Eftevaag if (flag == NULL) {
1062*08ab5237SOystein Eftevaag undefined_names_[key] = ""; // value isn't actually used
1063*08ab5237SOystein Eftevaag error_flags_[key] = error_message;
1064*08ab5237SOystein Eftevaag continue;
1065*08ab5237SOystein Eftevaag }
1066*08ab5237SOystein Eftevaag
1067*08ab5237SOystein Eftevaag if (value == NULL) {
1068*08ab5237SOystein Eftevaag // Boolean options are always assigned a value by SplitArgumentLocked()
1069*08ab5237SOystein Eftevaag assert(flag->Type() != FlagValue::FV_BOOL);
1070*08ab5237SOystein Eftevaag if (i+1 >= first_nonopt) {
1071*08ab5237SOystein Eftevaag // This flag needs a value, but there is nothing available
1072*08ab5237SOystein Eftevaag error_flags_[key] = (string(kError) + "flag '" + (*argv)[i] + "'"
1073*08ab5237SOystein Eftevaag + " is missing its argument");
1074*08ab5237SOystein Eftevaag if (flag->help() && flag->help()[0] > '\001') {
1075*08ab5237SOystein Eftevaag // Be useful in case we have a non-stripped description.
1076*08ab5237SOystein Eftevaag error_flags_[key] += string("; flag description: ") + flag->help();
1077*08ab5237SOystein Eftevaag }
1078*08ab5237SOystein Eftevaag error_flags_[key] += "\n";
1079*08ab5237SOystein Eftevaag break; // we treat this as an unrecoverable error
1080*08ab5237SOystein Eftevaag } else {
1081*08ab5237SOystein Eftevaag value = (*argv)[++i]; // read next arg for value
1082*08ab5237SOystein Eftevaag
1083*08ab5237SOystein Eftevaag // Heuristic to detect the case where someone treats a string arg
1084*08ab5237SOystein Eftevaag // like a bool:
1085*08ab5237SOystein Eftevaag // --my_string_var --foo=bar
1086*08ab5237SOystein Eftevaag // We look for a flag of string type, whose value begins with a
1087*08ab5237SOystein Eftevaag // dash, and where the flag-name and value are separated by a
1088*08ab5237SOystein Eftevaag // space rather than an '='.
1089*08ab5237SOystein Eftevaag // To avoid false positives, we also require the word "true"
1090*08ab5237SOystein Eftevaag // or "false" in the help string. Without this, a valid usage
1091*08ab5237SOystein Eftevaag // "-lat -30.5" would trigger the warning. The common cases we
1092*08ab5237SOystein Eftevaag // want to solve talk about true and false as values.
1093*08ab5237SOystein Eftevaag if (value[0] == '-'
1094*08ab5237SOystein Eftevaag && flag->Type() == FlagValue::FV_STRING
1095*08ab5237SOystein Eftevaag && (strstr(flag->help(), "true")
1096*08ab5237SOystein Eftevaag || strstr(flag->help(), "false"))) {
1097*08ab5237SOystein Eftevaag LOG(WARNING) << "Did you really mean to set flag '"
1098*08ab5237SOystein Eftevaag << flag->name() << "' to the value '"
1099*08ab5237SOystein Eftevaag << value << "'?";
1100*08ab5237SOystein Eftevaag }
1101*08ab5237SOystein Eftevaag }
1102*08ab5237SOystein Eftevaag }
1103*08ab5237SOystein Eftevaag
1104*08ab5237SOystein Eftevaag // TODO(csilvers): only set a flag if we hadn't set it before here
1105*08ab5237SOystein Eftevaag ProcessSingleOptionLocked(flag, value, SET_FLAGS_VALUE);
1106*08ab5237SOystein Eftevaag }
1107*08ab5237SOystein Eftevaag registry_->Unlock();
1108*08ab5237SOystein Eftevaag
1109*08ab5237SOystein Eftevaag if (remove_flags) { // Fix up argc and argv by removing command line flags
1110*08ab5237SOystein Eftevaag (*argv)[first_nonopt-1] = (*argv)[0];
1111*08ab5237SOystein Eftevaag (*argv) += (first_nonopt-1);
1112*08ab5237SOystein Eftevaag (*argc) -= (first_nonopt-1);
1113*08ab5237SOystein Eftevaag first_nonopt = 1; // because we still don't count argv[0]
1114*08ab5237SOystein Eftevaag }
1115*08ab5237SOystein Eftevaag
1116*08ab5237SOystein Eftevaag logging_is_probably_set_up = true; // because we've parsed --logdir, etc.
1117*08ab5237SOystein Eftevaag
1118*08ab5237SOystein Eftevaag return first_nonopt;
1119*08ab5237SOystein Eftevaag }
1120*08ab5237SOystein Eftevaag
ProcessFlagfileLocked(const string & flagval,FlagSettingMode set_mode)1121*08ab5237SOystein Eftevaag string CommandLineFlagParser::ProcessFlagfileLocked(const string& flagval,
1122*08ab5237SOystein Eftevaag FlagSettingMode set_mode) {
1123*08ab5237SOystein Eftevaag if (flagval.empty())
1124*08ab5237SOystein Eftevaag return "";
1125*08ab5237SOystein Eftevaag
1126*08ab5237SOystein Eftevaag string msg;
1127*08ab5237SOystein Eftevaag vector<string> filename_list;
1128*08ab5237SOystein Eftevaag ParseFlagList(flagval.c_str(), &filename_list); // take a list of filenames
1129*08ab5237SOystein Eftevaag for (size_t i = 0; i < filename_list.size(); ++i) {
1130*08ab5237SOystein Eftevaag const char* file = filename_list[i].c_str();
1131*08ab5237SOystein Eftevaag msg += ProcessOptionsFromStringLocked(ReadFileIntoString(file), set_mode);
1132*08ab5237SOystein Eftevaag }
1133*08ab5237SOystein Eftevaag return msg;
1134*08ab5237SOystein Eftevaag }
1135*08ab5237SOystein Eftevaag
ProcessFromenvLocked(const string & flagval,FlagSettingMode set_mode,bool errors_are_fatal)1136*08ab5237SOystein Eftevaag string CommandLineFlagParser::ProcessFromenvLocked(const string& flagval,
1137*08ab5237SOystein Eftevaag FlagSettingMode set_mode,
1138*08ab5237SOystein Eftevaag bool errors_are_fatal) {
1139*08ab5237SOystein Eftevaag if (flagval.empty())
1140*08ab5237SOystein Eftevaag return "";
1141*08ab5237SOystein Eftevaag
1142*08ab5237SOystein Eftevaag string msg;
1143*08ab5237SOystein Eftevaag vector<string> flaglist;
1144*08ab5237SOystein Eftevaag ParseFlagList(flagval.c_str(), &flaglist);
1145*08ab5237SOystein Eftevaag
1146*08ab5237SOystein Eftevaag for (size_t i = 0; i < flaglist.size(); ++i) {
1147*08ab5237SOystein Eftevaag const char* flagname = flaglist[i].c_str();
1148*08ab5237SOystein Eftevaag CommandLineFlag* flag = registry_->FindFlagLocked(flagname);
1149*08ab5237SOystein Eftevaag if (flag == NULL) {
1150*08ab5237SOystein Eftevaag error_flags_[flagname] =
1151*08ab5237SOystein Eftevaag StringPrintf("%sunknown command line flag '%s' "
1152*08ab5237SOystein Eftevaag "(via --fromenv or --tryfromenv)\n",
1153*08ab5237SOystein Eftevaag kError, flagname);
1154*08ab5237SOystein Eftevaag undefined_names_[flagname] = "";
1155*08ab5237SOystein Eftevaag continue;
1156*08ab5237SOystein Eftevaag }
1157*08ab5237SOystein Eftevaag
1158*08ab5237SOystein Eftevaag const string envname = string("FLAGS_") + string(flagname);
1159*08ab5237SOystein Eftevaag string envval;
1160*08ab5237SOystein Eftevaag if (!SafeGetEnv(envname.c_str(), envval)) {
1161*08ab5237SOystein Eftevaag if (errors_are_fatal) {
1162*08ab5237SOystein Eftevaag error_flags_[flagname] = (string(kError) + envname +
1163*08ab5237SOystein Eftevaag " not found in environment\n");
1164*08ab5237SOystein Eftevaag }
1165*08ab5237SOystein Eftevaag continue;
1166*08ab5237SOystein Eftevaag }
1167*08ab5237SOystein Eftevaag
1168*08ab5237SOystein Eftevaag // Avoid infinite recursion.
1169*08ab5237SOystein Eftevaag if (envval == "fromenv" || envval == "tryfromenv") {
1170*08ab5237SOystein Eftevaag error_flags_[flagname] =
1171*08ab5237SOystein Eftevaag StringPrintf("%sinfinite recursion on environment flag '%s'\n",
1172*08ab5237SOystein Eftevaag kError, envval.c_str());
1173*08ab5237SOystein Eftevaag continue;
1174*08ab5237SOystein Eftevaag }
1175*08ab5237SOystein Eftevaag
1176*08ab5237SOystein Eftevaag msg += ProcessSingleOptionLocked(flag, envval.c_str(), set_mode);
1177*08ab5237SOystein Eftevaag }
1178*08ab5237SOystein Eftevaag return msg;
1179*08ab5237SOystein Eftevaag }
1180*08ab5237SOystein Eftevaag
ProcessSingleOptionLocked(CommandLineFlag * flag,const char * value,FlagSettingMode set_mode)1181*08ab5237SOystein Eftevaag string CommandLineFlagParser::ProcessSingleOptionLocked(
1182*08ab5237SOystein Eftevaag CommandLineFlag* flag, const char* value, FlagSettingMode set_mode) {
1183*08ab5237SOystein Eftevaag string msg;
1184*08ab5237SOystein Eftevaag if (value && !registry_->SetFlagLocked(flag, value, set_mode, &msg)) {
1185*08ab5237SOystein Eftevaag error_flags_[flag->name()] = msg;
1186*08ab5237SOystein Eftevaag return "";
1187*08ab5237SOystein Eftevaag }
1188*08ab5237SOystein Eftevaag
1189*08ab5237SOystein Eftevaag // The recursive flags, --flagfile and --fromenv and --tryfromenv,
1190*08ab5237SOystein Eftevaag // must be dealt with as soon as they're seen. They will emit
1191*08ab5237SOystein Eftevaag // messages of their own.
1192*08ab5237SOystein Eftevaag if (strcmp(flag->name(), "flagfile") == 0) {
1193*08ab5237SOystein Eftevaag msg += ProcessFlagfileLocked(FLAGS_flagfile, set_mode);
1194*08ab5237SOystein Eftevaag
1195*08ab5237SOystein Eftevaag } else if (strcmp(flag->name(), "fromenv") == 0) {
1196*08ab5237SOystein Eftevaag // last arg indicates envval-not-found is fatal (unlike in --tryfromenv)
1197*08ab5237SOystein Eftevaag msg += ProcessFromenvLocked(FLAGS_fromenv, set_mode, true);
1198*08ab5237SOystein Eftevaag
1199*08ab5237SOystein Eftevaag } else if (strcmp(flag->name(), "tryfromenv") == 0) {
1200*08ab5237SOystein Eftevaag msg += ProcessFromenvLocked(FLAGS_tryfromenv, set_mode, false);
1201*08ab5237SOystein Eftevaag }
1202*08ab5237SOystein Eftevaag
1203*08ab5237SOystein Eftevaag return msg;
1204*08ab5237SOystein Eftevaag }
1205*08ab5237SOystein Eftevaag
ValidateFlags(bool all)1206*08ab5237SOystein Eftevaag void CommandLineFlagParser::ValidateFlags(bool all) {
1207*08ab5237SOystein Eftevaag FlagRegistryLock frl(registry_);
1208*08ab5237SOystein Eftevaag for (FlagRegistry::FlagConstIterator i = registry_->flags_.begin();
1209*08ab5237SOystein Eftevaag i != registry_->flags_.end(); ++i) {
1210*08ab5237SOystein Eftevaag if ((all || !i->second->Modified()) && !i->second->ValidateCurrent()) {
1211*08ab5237SOystein Eftevaag // only set a message if one isn't already there. (If there's
1212*08ab5237SOystein Eftevaag // an error message, our job is done, even if it's not exactly
1213*08ab5237SOystein Eftevaag // the same error.)
1214*08ab5237SOystein Eftevaag if (error_flags_[i->second->name()].empty()) {
1215*08ab5237SOystein Eftevaag error_flags_[i->second->name()] =
1216*08ab5237SOystein Eftevaag string(kError) + "--" + i->second->name() +
1217*08ab5237SOystein Eftevaag " must be set on the commandline";
1218*08ab5237SOystein Eftevaag if (!i->second->Modified()) {
1219*08ab5237SOystein Eftevaag error_flags_[i->second->name()] += " (default value fails validation)";
1220*08ab5237SOystein Eftevaag }
1221*08ab5237SOystein Eftevaag error_flags_[i->second->name()] += "\n";
1222*08ab5237SOystein Eftevaag }
1223*08ab5237SOystein Eftevaag }
1224*08ab5237SOystein Eftevaag }
1225*08ab5237SOystein Eftevaag }
1226*08ab5237SOystein Eftevaag
ValidateUnmodifiedFlags()1227*08ab5237SOystein Eftevaag void CommandLineFlagParser::ValidateUnmodifiedFlags() {
1228*08ab5237SOystein Eftevaag ValidateFlags(false);
1229*08ab5237SOystein Eftevaag }
1230*08ab5237SOystein Eftevaag
ReportErrors()1231*08ab5237SOystein Eftevaag bool CommandLineFlagParser::ReportErrors() {
1232*08ab5237SOystein Eftevaag // error_flags_ indicates errors we saw while parsing.
1233*08ab5237SOystein Eftevaag // But we ignore undefined-names if ok'ed by --undef_ok
1234*08ab5237SOystein Eftevaag if (!FLAGS_undefok.empty()) {
1235*08ab5237SOystein Eftevaag vector<string> flaglist;
1236*08ab5237SOystein Eftevaag ParseFlagList(FLAGS_undefok.c_str(), &flaglist);
1237*08ab5237SOystein Eftevaag for (size_t i = 0; i < flaglist.size(); ++i) {
1238*08ab5237SOystein Eftevaag // We also deal with --no<flag>, in case the flagname was boolean
1239*08ab5237SOystein Eftevaag const string no_version = string("no") + flaglist[i];
1240*08ab5237SOystein Eftevaag if (undefined_names_.find(flaglist[i]) != undefined_names_.end()) {
1241*08ab5237SOystein Eftevaag error_flags_[flaglist[i]] = ""; // clear the error message
1242*08ab5237SOystein Eftevaag } else if (undefined_names_.find(no_version) != undefined_names_.end()) {
1243*08ab5237SOystein Eftevaag error_flags_[no_version] = "";
1244*08ab5237SOystein Eftevaag }
1245*08ab5237SOystein Eftevaag }
1246*08ab5237SOystein Eftevaag }
1247*08ab5237SOystein Eftevaag // Likewise, if they decided to allow reparsing, all undefined-names
1248*08ab5237SOystein Eftevaag // are ok; we just silently ignore them now, and hope that a future
1249*08ab5237SOystein Eftevaag // parse will pick them up somehow.
1250*08ab5237SOystein Eftevaag if (allow_command_line_reparsing) {
1251*08ab5237SOystein Eftevaag for (map<string, string>::const_iterator it = undefined_names_.begin();
1252*08ab5237SOystein Eftevaag it != undefined_names_.end(); ++it)
1253*08ab5237SOystein Eftevaag error_flags_[it->first] = ""; // clear the error message
1254*08ab5237SOystein Eftevaag }
1255*08ab5237SOystein Eftevaag
1256*08ab5237SOystein Eftevaag bool found_error = false;
1257*08ab5237SOystein Eftevaag string error_message;
1258*08ab5237SOystein Eftevaag for (map<string, string>::const_iterator it = error_flags_.begin();
1259*08ab5237SOystein Eftevaag it != error_flags_.end(); ++it) {
1260*08ab5237SOystein Eftevaag if (!it->second.empty()) {
1261*08ab5237SOystein Eftevaag error_message.append(it->second.data(), it->second.size());
1262*08ab5237SOystein Eftevaag found_error = true;
1263*08ab5237SOystein Eftevaag }
1264*08ab5237SOystein Eftevaag }
1265*08ab5237SOystein Eftevaag if (found_error)
1266*08ab5237SOystein Eftevaag ReportError(DO_NOT_DIE, "%s", error_message.c_str());
1267*08ab5237SOystein Eftevaag return found_error;
1268*08ab5237SOystein Eftevaag }
1269*08ab5237SOystein Eftevaag
ProcessOptionsFromStringLocked(const string & contentdata,FlagSettingMode set_mode)1270*08ab5237SOystein Eftevaag string CommandLineFlagParser::ProcessOptionsFromStringLocked(
1271*08ab5237SOystein Eftevaag const string& contentdata, FlagSettingMode set_mode) {
1272*08ab5237SOystein Eftevaag string retval;
1273*08ab5237SOystein Eftevaag const char* flagfile_contents = contentdata.c_str();
1274*08ab5237SOystein Eftevaag bool flags_are_relevant = true; // set to false when filenames don't match
1275*08ab5237SOystein Eftevaag bool in_filename_section = false;
1276*08ab5237SOystein Eftevaag
1277*08ab5237SOystein Eftevaag const char* line_end = flagfile_contents;
1278*08ab5237SOystein Eftevaag // We read this file a line at a time.
1279*08ab5237SOystein Eftevaag for (; line_end; flagfile_contents = line_end + 1) {
1280*08ab5237SOystein Eftevaag while (*flagfile_contents && isspace(*flagfile_contents))
1281*08ab5237SOystein Eftevaag ++flagfile_contents;
1282*08ab5237SOystein Eftevaag // Windows uses "\r\n"
1283*08ab5237SOystein Eftevaag line_end = strchr(flagfile_contents, '\r');
1284*08ab5237SOystein Eftevaag if (line_end == NULL)
1285*08ab5237SOystein Eftevaag line_end = strchr(flagfile_contents, '\n');
1286*08ab5237SOystein Eftevaag
1287*08ab5237SOystein Eftevaag size_t len = line_end ? line_end - flagfile_contents
1288*08ab5237SOystein Eftevaag : strlen(flagfile_contents);
1289*08ab5237SOystein Eftevaag string line(flagfile_contents, len);
1290*08ab5237SOystein Eftevaag
1291*08ab5237SOystein Eftevaag // Each line can be one of four things:
1292*08ab5237SOystein Eftevaag // 1) A comment line -- we skip it
1293*08ab5237SOystein Eftevaag // 2) An empty line -- we skip it
1294*08ab5237SOystein Eftevaag // 3) A list of filenames -- starts a new filenames+flags section
1295*08ab5237SOystein Eftevaag // 4) A --flag=value line -- apply if previous filenames match
1296*08ab5237SOystein Eftevaag if (line.empty() || line[0] == '#') {
1297*08ab5237SOystein Eftevaag // comment or empty line; just ignore
1298*08ab5237SOystein Eftevaag
1299*08ab5237SOystein Eftevaag } else if (line[0] == '-') { // flag
1300*08ab5237SOystein Eftevaag in_filename_section = false; // instead, it was a flag-line
1301*08ab5237SOystein Eftevaag if (!flags_are_relevant) // skip this flag; applies to someone else
1302*08ab5237SOystein Eftevaag continue;
1303*08ab5237SOystein Eftevaag
1304*08ab5237SOystein Eftevaag const char* name_and_val = line.c_str() + 1; // skip the leading -
1305*08ab5237SOystein Eftevaag if (*name_and_val == '-')
1306*08ab5237SOystein Eftevaag name_and_val++; // skip second - too
1307*08ab5237SOystein Eftevaag string key;
1308*08ab5237SOystein Eftevaag const char* value;
1309*08ab5237SOystein Eftevaag string error_message;
1310*08ab5237SOystein Eftevaag CommandLineFlag* flag = registry_->SplitArgumentLocked(name_and_val,
1311*08ab5237SOystein Eftevaag &key, &value,
1312*08ab5237SOystein Eftevaag &error_message);
1313*08ab5237SOystein Eftevaag // By API, errors parsing flagfile lines are silently ignored.
1314*08ab5237SOystein Eftevaag if (flag == NULL) {
1315*08ab5237SOystein Eftevaag // "WARNING: flagname '" + key + "' not found\n"
1316*08ab5237SOystein Eftevaag } else if (value == NULL) {
1317*08ab5237SOystein Eftevaag // "WARNING: flagname '" + key + "' missing a value\n"
1318*08ab5237SOystein Eftevaag } else {
1319*08ab5237SOystein Eftevaag retval += ProcessSingleOptionLocked(flag, value, set_mode);
1320*08ab5237SOystein Eftevaag }
1321*08ab5237SOystein Eftevaag
1322*08ab5237SOystein Eftevaag } else { // a filename!
1323*08ab5237SOystein Eftevaag if (!in_filename_section) { // start over: assume filenames don't match
1324*08ab5237SOystein Eftevaag in_filename_section = true;
1325*08ab5237SOystein Eftevaag flags_are_relevant = false;
1326*08ab5237SOystein Eftevaag }
1327*08ab5237SOystein Eftevaag
1328*08ab5237SOystein Eftevaag // Split the line up at spaces into glob-patterns
1329*08ab5237SOystein Eftevaag const char* space = line.c_str(); // just has to be non-NULL
1330*08ab5237SOystein Eftevaag for (const char* word = line.c_str(); *space; word = space+1) {
1331*08ab5237SOystein Eftevaag if (flags_are_relevant) // we can stop as soon as we match
1332*08ab5237SOystein Eftevaag break;
1333*08ab5237SOystein Eftevaag space = strchr(word, ' ');
1334*08ab5237SOystein Eftevaag if (space == NULL)
1335*08ab5237SOystein Eftevaag space = word + strlen(word);
1336*08ab5237SOystein Eftevaag const string glob(word, space - word);
1337*08ab5237SOystein Eftevaag // We try matching both against the full argv0 and basename(argv0)
1338*08ab5237SOystein Eftevaag if (glob == ProgramInvocationName() // small optimization
1339*08ab5237SOystein Eftevaag || glob == ProgramInvocationShortName()
1340*08ab5237SOystein Eftevaag #if defined(HAVE_FNMATCH_H)
1341*08ab5237SOystein Eftevaag || fnmatch(glob.c_str(), ProgramInvocationName(), FNM_PATHNAME) == 0
1342*08ab5237SOystein Eftevaag || fnmatch(glob.c_str(), ProgramInvocationShortName(), FNM_PATHNAME) == 0
1343*08ab5237SOystein Eftevaag #elif defined(HAVE_SHLWAPI_H)
1344*08ab5237SOystein Eftevaag || PathMatchSpecA(glob.c_str(), ProgramInvocationName())
1345*08ab5237SOystein Eftevaag || PathMatchSpecA(glob.c_str(), ProgramInvocationShortName())
1346*08ab5237SOystein Eftevaag #endif
1347*08ab5237SOystein Eftevaag ) {
1348*08ab5237SOystein Eftevaag flags_are_relevant = true;
1349*08ab5237SOystein Eftevaag }
1350*08ab5237SOystein Eftevaag }
1351*08ab5237SOystein Eftevaag }
1352*08ab5237SOystein Eftevaag }
1353*08ab5237SOystein Eftevaag return retval;
1354*08ab5237SOystein Eftevaag }
1355*08ab5237SOystein Eftevaag
1356*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1357*08ab5237SOystein Eftevaag // GetFromEnv()
1358*08ab5237SOystein Eftevaag // AddFlagValidator()
1359*08ab5237SOystein Eftevaag // These are helper functions for routines like BoolFromEnv() and
1360*08ab5237SOystein Eftevaag // RegisterFlagValidator, defined below. They're defined here so
1361*08ab5237SOystein Eftevaag // they can live in the unnamed namespace (which makes friendship
1362*08ab5237SOystein Eftevaag // declarations for these classes possible).
1363*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1364*08ab5237SOystein Eftevaag
1365*08ab5237SOystein Eftevaag template<typename T>
GetFromEnv(const char * varname,T dflt)1366*08ab5237SOystein Eftevaag T GetFromEnv(const char *varname, T dflt) {
1367*08ab5237SOystein Eftevaag std::string valstr;
1368*08ab5237SOystein Eftevaag if (SafeGetEnv(varname, valstr)) {
1369*08ab5237SOystein Eftevaag FlagValue ifv(new T, true);
1370*08ab5237SOystein Eftevaag if (!ifv.ParseFrom(valstr.c_str())) {
1371*08ab5237SOystein Eftevaag ReportError(DIE, "ERROR: error parsing env variable '%s' with value '%s'\n",
1372*08ab5237SOystein Eftevaag varname, valstr.c_str());
1373*08ab5237SOystein Eftevaag }
1374*08ab5237SOystein Eftevaag return OTHER_VALUE_AS(ifv, T);
1375*08ab5237SOystein Eftevaag } else return dflt;
1376*08ab5237SOystein Eftevaag }
1377*08ab5237SOystein Eftevaag
AddFlagValidator(const void * flag_ptr,ValidateFnProto validate_fn_proto)1378*08ab5237SOystein Eftevaag bool AddFlagValidator(const void* flag_ptr, ValidateFnProto validate_fn_proto) {
1379*08ab5237SOystein Eftevaag // We want a lock around this routine, in case two threads try to
1380*08ab5237SOystein Eftevaag // add a validator (hopefully the same one!) at once. We could use
1381*08ab5237SOystein Eftevaag // our own thread, but we need to loook at the registry anyway, so
1382*08ab5237SOystein Eftevaag // we just steal that one.
1383*08ab5237SOystein Eftevaag FlagRegistry* const registry = FlagRegistry::GlobalRegistry();
1384*08ab5237SOystein Eftevaag FlagRegistryLock frl(registry);
1385*08ab5237SOystein Eftevaag // First, find the flag whose current-flag storage is 'flag'.
1386*08ab5237SOystein Eftevaag // This is the CommandLineFlag whose current_->value_buffer_ == flag
1387*08ab5237SOystein Eftevaag CommandLineFlag* flag = registry->FindFlagViaPtrLocked(flag_ptr);
1388*08ab5237SOystein Eftevaag if (!flag) {
1389*08ab5237SOystein Eftevaag LOG(WARNING) << "Ignoring RegisterValidateFunction() for flag pointer "
1390*08ab5237SOystein Eftevaag << flag_ptr << ": no flag found at that address";
1391*08ab5237SOystein Eftevaag return false;
1392*08ab5237SOystein Eftevaag } else if (validate_fn_proto == flag->validate_function()) {
1393*08ab5237SOystein Eftevaag return true; // ok to register the same function over and over again
1394*08ab5237SOystein Eftevaag } else if (validate_fn_proto != NULL && flag->validate_function() != NULL) {
1395*08ab5237SOystein Eftevaag LOG(WARNING) << "Ignoring RegisterValidateFunction() for flag '"
1396*08ab5237SOystein Eftevaag << flag->name() << "': validate-fn already registered";
1397*08ab5237SOystein Eftevaag return false;
1398*08ab5237SOystein Eftevaag } else {
1399*08ab5237SOystein Eftevaag flag->validate_fn_proto_ = validate_fn_proto;
1400*08ab5237SOystein Eftevaag return true;
1401*08ab5237SOystein Eftevaag }
1402*08ab5237SOystein Eftevaag }
1403*08ab5237SOystein Eftevaag
1404*08ab5237SOystein Eftevaag } // end unnamed namespaces
1405*08ab5237SOystein Eftevaag
1406*08ab5237SOystein Eftevaag
1407*08ab5237SOystein Eftevaag // Now define the functions that are exported via the .h file
1408*08ab5237SOystein Eftevaag
1409*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1410*08ab5237SOystein Eftevaag // FlagRegisterer
1411*08ab5237SOystein Eftevaag // This class exists merely to have a global constructor (the
1412*08ab5237SOystein Eftevaag // kind that runs before main(), that goes an initializes each
1413*08ab5237SOystein Eftevaag // flag that's been declared. Note that it's very important we
1414*08ab5237SOystein Eftevaag // don't have a destructor that deletes flag_, because that would
1415*08ab5237SOystein Eftevaag // cause us to delete current_storage/defvalue_storage as well,
1416*08ab5237SOystein Eftevaag // which can cause a crash if anything tries to access the flag
1417*08ab5237SOystein Eftevaag // values in a global destructor.
1418*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1419*08ab5237SOystein Eftevaag
1420*08ab5237SOystein Eftevaag namespace {
RegisterCommandLineFlag(const char * name,const char * help,const char * filename,FlagValue * current,FlagValue * defvalue)1421*08ab5237SOystein Eftevaag void RegisterCommandLineFlag(const char* name,
1422*08ab5237SOystein Eftevaag const char* help,
1423*08ab5237SOystein Eftevaag const char* filename,
1424*08ab5237SOystein Eftevaag FlagValue* current,
1425*08ab5237SOystein Eftevaag FlagValue* defvalue) {
1426*08ab5237SOystein Eftevaag if (help == NULL)
1427*08ab5237SOystein Eftevaag help = "";
1428*08ab5237SOystein Eftevaag // Importantly, flag_ will never be deleted, so storage is always good.
1429*08ab5237SOystein Eftevaag CommandLineFlag* flag =
1430*08ab5237SOystein Eftevaag new CommandLineFlag(name, help, filename, current, defvalue);
1431*08ab5237SOystein Eftevaag FlagRegistry::GlobalRegistry()->RegisterFlag(flag); // default registry
1432*08ab5237SOystein Eftevaag }
1433*08ab5237SOystein Eftevaag }
1434*08ab5237SOystein Eftevaag
1435*08ab5237SOystein Eftevaag template <typename FlagType>
FlagRegisterer(const char * name,const char * help,const char * filename,FlagType * current_storage,FlagType * defvalue_storage)1436*08ab5237SOystein Eftevaag FlagRegisterer::FlagRegisterer(const char* name,
1437*08ab5237SOystein Eftevaag const char* help,
1438*08ab5237SOystein Eftevaag const char* filename,
1439*08ab5237SOystein Eftevaag FlagType* current_storage,
1440*08ab5237SOystein Eftevaag FlagType* defvalue_storage) {
1441*08ab5237SOystein Eftevaag FlagValue* const current = new FlagValue(current_storage, false);
1442*08ab5237SOystein Eftevaag FlagValue* const defvalue = new FlagValue(defvalue_storage, false);
1443*08ab5237SOystein Eftevaag RegisterCommandLineFlag(name, help, filename, current, defvalue);
1444*08ab5237SOystein Eftevaag }
1445*08ab5237SOystein Eftevaag
1446*08ab5237SOystein Eftevaag // Force compiler to generate code for the given template specialization.
1447*08ab5237SOystein Eftevaag #define INSTANTIATE_FLAG_REGISTERER_CTOR(type) \
1448*08ab5237SOystein Eftevaag template GFLAGS_DLL_DECL FlagRegisterer::FlagRegisterer( \
1449*08ab5237SOystein Eftevaag const char* name, const char* help, const char* filename, \
1450*08ab5237SOystein Eftevaag type* current_storage, type* defvalue_storage)
1451*08ab5237SOystein Eftevaag
1452*08ab5237SOystein Eftevaag // Do this for all supported flag types.
1453*08ab5237SOystein Eftevaag INSTANTIATE_FLAG_REGISTERER_CTOR(bool);
1454*08ab5237SOystein Eftevaag INSTANTIATE_FLAG_REGISTERER_CTOR(int32);
1455*08ab5237SOystein Eftevaag INSTANTIATE_FLAG_REGISTERER_CTOR(uint32);
1456*08ab5237SOystein Eftevaag INSTANTIATE_FLAG_REGISTERER_CTOR(int64);
1457*08ab5237SOystein Eftevaag INSTANTIATE_FLAG_REGISTERER_CTOR(uint64);
1458*08ab5237SOystein Eftevaag INSTANTIATE_FLAG_REGISTERER_CTOR(double);
1459*08ab5237SOystein Eftevaag INSTANTIATE_FLAG_REGISTERER_CTOR(std::string);
1460*08ab5237SOystein Eftevaag
1461*08ab5237SOystein Eftevaag #undef INSTANTIATE_FLAG_REGISTERER_CTOR
1462*08ab5237SOystein Eftevaag
1463*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1464*08ab5237SOystein Eftevaag // GetAllFlags()
1465*08ab5237SOystein Eftevaag // The main way the FlagRegistry class exposes its data. This
1466*08ab5237SOystein Eftevaag // returns, as strings, all the info about all the flags in
1467*08ab5237SOystein Eftevaag // the main registry, sorted first by filename they are defined
1468*08ab5237SOystein Eftevaag // in, and then by flagname.
1469*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1470*08ab5237SOystein Eftevaag
1471*08ab5237SOystein Eftevaag struct FilenameFlagnameCmp {
operator ()GFLAGS_NAMESPACE::FilenameFlagnameCmp1472*08ab5237SOystein Eftevaag bool operator()(const CommandLineFlagInfo& a,
1473*08ab5237SOystein Eftevaag const CommandLineFlagInfo& b) const {
1474*08ab5237SOystein Eftevaag int cmp = strcmp(a.filename.c_str(), b.filename.c_str());
1475*08ab5237SOystein Eftevaag if (cmp == 0)
1476*08ab5237SOystein Eftevaag cmp = strcmp(a.name.c_str(), b.name.c_str()); // secondary sort key
1477*08ab5237SOystein Eftevaag return cmp < 0;
1478*08ab5237SOystein Eftevaag }
1479*08ab5237SOystein Eftevaag };
1480*08ab5237SOystein Eftevaag
GetAllFlags(vector<CommandLineFlagInfo> * OUTPUT)1481*08ab5237SOystein Eftevaag void GetAllFlags(vector<CommandLineFlagInfo>* OUTPUT) {
1482*08ab5237SOystein Eftevaag FlagRegistry* const registry = FlagRegistry::GlobalRegistry();
1483*08ab5237SOystein Eftevaag registry->Lock();
1484*08ab5237SOystein Eftevaag for (FlagRegistry::FlagConstIterator i = registry->flags_.begin();
1485*08ab5237SOystein Eftevaag i != registry->flags_.end(); ++i) {
1486*08ab5237SOystein Eftevaag CommandLineFlagInfo fi;
1487*08ab5237SOystein Eftevaag i->second->FillCommandLineFlagInfo(&fi);
1488*08ab5237SOystein Eftevaag OUTPUT->push_back(fi);
1489*08ab5237SOystein Eftevaag }
1490*08ab5237SOystein Eftevaag registry->Unlock();
1491*08ab5237SOystein Eftevaag // Now sort the flags, first by filename they occur in, then alphabetically
1492*08ab5237SOystein Eftevaag sort(OUTPUT->begin(), OUTPUT->end(), FilenameFlagnameCmp());
1493*08ab5237SOystein Eftevaag }
1494*08ab5237SOystein Eftevaag
1495*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1496*08ab5237SOystein Eftevaag // SetArgv()
1497*08ab5237SOystein Eftevaag // GetArgvs()
1498*08ab5237SOystein Eftevaag // GetArgv()
1499*08ab5237SOystein Eftevaag // GetArgv0()
1500*08ab5237SOystein Eftevaag // ProgramInvocationName()
1501*08ab5237SOystein Eftevaag // ProgramInvocationShortName()
1502*08ab5237SOystein Eftevaag // SetUsageMessage()
1503*08ab5237SOystein Eftevaag // ProgramUsage()
1504*08ab5237SOystein Eftevaag // Functions to set and get argv. Typically the setter is called
1505*08ab5237SOystein Eftevaag // by ParseCommandLineFlags. Also can get the ProgramUsage string,
1506*08ab5237SOystein Eftevaag // set by SetUsageMessage.
1507*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1508*08ab5237SOystein Eftevaag
1509*08ab5237SOystein Eftevaag // These values are not protected by a Mutex because they are normally
1510*08ab5237SOystein Eftevaag // set only once during program startup.
1511*08ab5237SOystein Eftevaag static string argv0("UNKNOWN"); // just the program name
1512*08ab5237SOystein Eftevaag static string cmdline; // the entire command-line
1513*08ab5237SOystein Eftevaag static string program_usage;
1514*08ab5237SOystein Eftevaag static vector<string> argvs;
1515*08ab5237SOystein Eftevaag static uint32 argv_sum = 0;
1516*08ab5237SOystein Eftevaag
SetArgv(int argc,const char ** argv)1517*08ab5237SOystein Eftevaag void SetArgv(int argc, const char** argv) {
1518*08ab5237SOystein Eftevaag static bool called_set_argv = false;
1519*08ab5237SOystein Eftevaag if (called_set_argv) return;
1520*08ab5237SOystein Eftevaag called_set_argv = true;
1521*08ab5237SOystein Eftevaag
1522*08ab5237SOystein Eftevaag assert(argc > 0); // every program has at least a name
1523*08ab5237SOystein Eftevaag argv0 = argv[0];
1524*08ab5237SOystein Eftevaag
1525*08ab5237SOystein Eftevaag cmdline.clear();
1526*08ab5237SOystein Eftevaag for (int i = 0; i < argc; i++) {
1527*08ab5237SOystein Eftevaag if (i != 0) cmdline += " ";
1528*08ab5237SOystein Eftevaag cmdline += argv[i];
1529*08ab5237SOystein Eftevaag argvs.push_back(argv[i]);
1530*08ab5237SOystein Eftevaag }
1531*08ab5237SOystein Eftevaag
1532*08ab5237SOystein Eftevaag // Compute a simple sum of all the chars in argv
1533*08ab5237SOystein Eftevaag argv_sum = 0;
1534*08ab5237SOystein Eftevaag for (string::const_iterator c = cmdline.begin(); c != cmdline.end(); ++c) {
1535*08ab5237SOystein Eftevaag argv_sum += *c;
1536*08ab5237SOystein Eftevaag }
1537*08ab5237SOystein Eftevaag }
1538*08ab5237SOystein Eftevaag
GetArgvs()1539*08ab5237SOystein Eftevaag const vector<string>& GetArgvs() { return argvs; }
GetArgv()1540*08ab5237SOystein Eftevaag const char* GetArgv() { return cmdline.c_str(); }
GetArgv0()1541*08ab5237SOystein Eftevaag const char* GetArgv0() { return argv0.c_str(); }
GetArgvSum()1542*08ab5237SOystein Eftevaag uint32 GetArgvSum() { return argv_sum; }
ProgramInvocationName()1543*08ab5237SOystein Eftevaag const char* ProgramInvocationName() { // like the GNU libc fn
1544*08ab5237SOystein Eftevaag return GetArgv0();
1545*08ab5237SOystein Eftevaag }
ProgramInvocationShortName()1546*08ab5237SOystein Eftevaag const char* ProgramInvocationShortName() { // like the GNU libc fn
1547*08ab5237SOystein Eftevaag size_t pos = argv0.rfind('/');
1548*08ab5237SOystein Eftevaag #ifdef OS_WINDOWS
1549*08ab5237SOystein Eftevaag if (pos == string::npos) pos = argv0.rfind('\\');
1550*08ab5237SOystein Eftevaag #endif
1551*08ab5237SOystein Eftevaag return (pos == string::npos ? argv0.c_str() : (argv0.c_str() + pos + 1));
1552*08ab5237SOystein Eftevaag }
1553*08ab5237SOystein Eftevaag
SetUsageMessage(const string & usage)1554*08ab5237SOystein Eftevaag void SetUsageMessage(const string& usage) {
1555*08ab5237SOystein Eftevaag program_usage = usage;
1556*08ab5237SOystein Eftevaag }
1557*08ab5237SOystein Eftevaag
ProgramUsage()1558*08ab5237SOystein Eftevaag const char* ProgramUsage() {
1559*08ab5237SOystein Eftevaag if (program_usage.empty()) {
1560*08ab5237SOystein Eftevaag return "Warning: SetUsageMessage() never called";
1561*08ab5237SOystein Eftevaag }
1562*08ab5237SOystein Eftevaag return program_usage.c_str();
1563*08ab5237SOystein Eftevaag }
1564*08ab5237SOystein Eftevaag
1565*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1566*08ab5237SOystein Eftevaag // SetVersionString()
1567*08ab5237SOystein Eftevaag // VersionString()
1568*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1569*08ab5237SOystein Eftevaag
1570*08ab5237SOystein Eftevaag static string version_string;
1571*08ab5237SOystein Eftevaag
SetVersionString(const string & version)1572*08ab5237SOystein Eftevaag void SetVersionString(const string& version) {
1573*08ab5237SOystein Eftevaag version_string = version;
1574*08ab5237SOystein Eftevaag }
1575*08ab5237SOystein Eftevaag
VersionString()1576*08ab5237SOystein Eftevaag const char* VersionString() {
1577*08ab5237SOystein Eftevaag return version_string.c_str();
1578*08ab5237SOystein Eftevaag }
1579*08ab5237SOystein Eftevaag
1580*08ab5237SOystein Eftevaag
1581*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1582*08ab5237SOystein Eftevaag // GetCommandLineOption()
1583*08ab5237SOystein Eftevaag // GetCommandLineFlagInfo()
1584*08ab5237SOystein Eftevaag // GetCommandLineFlagInfoOrDie()
1585*08ab5237SOystein Eftevaag // SetCommandLineOption()
1586*08ab5237SOystein Eftevaag // SetCommandLineOptionWithMode()
1587*08ab5237SOystein Eftevaag // The programmatic way to set a flag's value, using a string
1588*08ab5237SOystein Eftevaag // for its name rather than the variable itself (that is,
1589*08ab5237SOystein Eftevaag // SetCommandLineOption("foo", x) rather than FLAGS_foo = x).
1590*08ab5237SOystein Eftevaag // There's also a bit more flexibility here due to the various
1591*08ab5237SOystein Eftevaag // set-modes, but typically these are used when you only have
1592*08ab5237SOystein Eftevaag // that flag's name as a string, perhaps at runtime.
1593*08ab5237SOystein Eftevaag // All of these work on the default, global registry.
1594*08ab5237SOystein Eftevaag // For GetCommandLineOption, return false if no such flag
1595*08ab5237SOystein Eftevaag // is known, true otherwise. We clear "value" if a suitable
1596*08ab5237SOystein Eftevaag // flag is found.
1597*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1598*08ab5237SOystein Eftevaag
1599*08ab5237SOystein Eftevaag
GetCommandLineOption(const char * name,string * value)1600*08ab5237SOystein Eftevaag bool GetCommandLineOption(const char* name, string* value) {
1601*08ab5237SOystein Eftevaag if (NULL == name)
1602*08ab5237SOystein Eftevaag return false;
1603*08ab5237SOystein Eftevaag assert(value);
1604*08ab5237SOystein Eftevaag
1605*08ab5237SOystein Eftevaag FlagRegistry* const registry = FlagRegistry::GlobalRegistry();
1606*08ab5237SOystein Eftevaag FlagRegistryLock frl(registry);
1607*08ab5237SOystein Eftevaag CommandLineFlag* flag = registry->FindFlagLocked(name);
1608*08ab5237SOystein Eftevaag if (flag == NULL) {
1609*08ab5237SOystein Eftevaag return false;
1610*08ab5237SOystein Eftevaag } else {
1611*08ab5237SOystein Eftevaag *value = flag->current_value();
1612*08ab5237SOystein Eftevaag return true;
1613*08ab5237SOystein Eftevaag }
1614*08ab5237SOystein Eftevaag }
1615*08ab5237SOystein Eftevaag
GetCommandLineFlagInfo(const char * name,CommandLineFlagInfo * OUTPUT)1616*08ab5237SOystein Eftevaag bool GetCommandLineFlagInfo(const char* name, CommandLineFlagInfo* OUTPUT) {
1617*08ab5237SOystein Eftevaag if (NULL == name) return false;
1618*08ab5237SOystein Eftevaag FlagRegistry* const registry = FlagRegistry::GlobalRegistry();
1619*08ab5237SOystein Eftevaag FlagRegistryLock frl(registry);
1620*08ab5237SOystein Eftevaag CommandLineFlag* flag = registry->FindFlagLocked(name);
1621*08ab5237SOystein Eftevaag if (flag == NULL) {
1622*08ab5237SOystein Eftevaag return false;
1623*08ab5237SOystein Eftevaag } else {
1624*08ab5237SOystein Eftevaag assert(OUTPUT);
1625*08ab5237SOystein Eftevaag flag->FillCommandLineFlagInfo(OUTPUT);
1626*08ab5237SOystein Eftevaag return true;
1627*08ab5237SOystein Eftevaag }
1628*08ab5237SOystein Eftevaag }
1629*08ab5237SOystein Eftevaag
GetCommandLineFlagInfoOrDie(const char * name)1630*08ab5237SOystein Eftevaag CommandLineFlagInfo GetCommandLineFlagInfoOrDie(const char* name) {
1631*08ab5237SOystein Eftevaag CommandLineFlagInfo info;
1632*08ab5237SOystein Eftevaag if (!GetCommandLineFlagInfo(name, &info)) {
1633*08ab5237SOystein Eftevaag fprintf(stderr, "FATAL ERROR: flag name '%s' doesn't exist\n", name);
1634*08ab5237SOystein Eftevaag gflags_exitfunc(1); // almost certainly gflags_exitfunc()
1635*08ab5237SOystein Eftevaag }
1636*08ab5237SOystein Eftevaag return info;
1637*08ab5237SOystein Eftevaag }
1638*08ab5237SOystein Eftevaag
SetCommandLineOptionWithMode(const char * name,const char * value,FlagSettingMode set_mode)1639*08ab5237SOystein Eftevaag string SetCommandLineOptionWithMode(const char* name, const char* value,
1640*08ab5237SOystein Eftevaag FlagSettingMode set_mode) {
1641*08ab5237SOystein Eftevaag string result;
1642*08ab5237SOystein Eftevaag FlagRegistry* const registry = FlagRegistry::GlobalRegistry();
1643*08ab5237SOystein Eftevaag FlagRegistryLock frl(registry);
1644*08ab5237SOystein Eftevaag CommandLineFlag* flag = registry->FindFlagLocked(name);
1645*08ab5237SOystein Eftevaag if (flag) {
1646*08ab5237SOystein Eftevaag CommandLineFlagParser parser(registry);
1647*08ab5237SOystein Eftevaag result = parser.ProcessSingleOptionLocked(flag, value, set_mode);
1648*08ab5237SOystein Eftevaag if (!result.empty()) { // in the error case, we've already logged
1649*08ab5237SOystein Eftevaag // Could consider logging this change
1650*08ab5237SOystein Eftevaag }
1651*08ab5237SOystein Eftevaag }
1652*08ab5237SOystein Eftevaag // The API of this function is that we return empty string on error
1653*08ab5237SOystein Eftevaag return result;
1654*08ab5237SOystein Eftevaag }
1655*08ab5237SOystein Eftevaag
SetCommandLineOption(const char * name,const char * value)1656*08ab5237SOystein Eftevaag string SetCommandLineOption(const char* name, const char* value) {
1657*08ab5237SOystein Eftevaag return SetCommandLineOptionWithMode(name, value, SET_FLAGS_VALUE);
1658*08ab5237SOystein Eftevaag }
1659*08ab5237SOystein Eftevaag
1660*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1661*08ab5237SOystein Eftevaag // FlagSaver
1662*08ab5237SOystein Eftevaag // FlagSaverImpl
1663*08ab5237SOystein Eftevaag // This class stores the states of all flags at construct time,
1664*08ab5237SOystein Eftevaag // and restores all flags to that state at destruct time.
1665*08ab5237SOystein Eftevaag // Its major implementation challenge is that it never modifies
1666*08ab5237SOystein Eftevaag // pointers in the 'main' registry, so global FLAG_* vars always
1667*08ab5237SOystein Eftevaag // point to the right place.
1668*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1669*08ab5237SOystein Eftevaag
1670*08ab5237SOystein Eftevaag class FlagSaverImpl {
1671*08ab5237SOystein Eftevaag public:
1672*08ab5237SOystein Eftevaag // Constructs an empty FlagSaverImpl object.
FlagSaverImpl(FlagRegistry * main_registry)1673*08ab5237SOystein Eftevaag explicit FlagSaverImpl(FlagRegistry* main_registry)
1674*08ab5237SOystein Eftevaag : main_registry_(main_registry) { }
~FlagSaverImpl()1675*08ab5237SOystein Eftevaag ~FlagSaverImpl() {
1676*08ab5237SOystein Eftevaag // reclaim memory from each of our CommandLineFlags
1677*08ab5237SOystein Eftevaag vector<CommandLineFlag*>::const_iterator it;
1678*08ab5237SOystein Eftevaag for (it = backup_registry_.begin(); it != backup_registry_.end(); ++it)
1679*08ab5237SOystein Eftevaag delete *it;
1680*08ab5237SOystein Eftevaag }
1681*08ab5237SOystein Eftevaag
1682*08ab5237SOystein Eftevaag // Saves the flag states from the flag registry into this object.
1683*08ab5237SOystein Eftevaag // It's an error to call this more than once.
1684*08ab5237SOystein Eftevaag // Must be called when the registry mutex is not held.
SaveFromRegistry()1685*08ab5237SOystein Eftevaag void SaveFromRegistry() {
1686*08ab5237SOystein Eftevaag FlagRegistryLock frl(main_registry_);
1687*08ab5237SOystein Eftevaag assert(backup_registry_.empty()); // call only once!
1688*08ab5237SOystein Eftevaag for (FlagRegistry::FlagConstIterator it = main_registry_->flags_.begin();
1689*08ab5237SOystein Eftevaag it != main_registry_->flags_.end();
1690*08ab5237SOystein Eftevaag ++it) {
1691*08ab5237SOystein Eftevaag const CommandLineFlag* main = it->second;
1692*08ab5237SOystein Eftevaag // Sets up all the const variables in backup correctly
1693*08ab5237SOystein Eftevaag CommandLineFlag* backup = new CommandLineFlag(
1694*08ab5237SOystein Eftevaag main->name(), main->help(), main->filename(),
1695*08ab5237SOystein Eftevaag main->current_->New(), main->defvalue_->New());
1696*08ab5237SOystein Eftevaag // Sets up all the non-const variables in backup correctly
1697*08ab5237SOystein Eftevaag backup->CopyFrom(*main);
1698*08ab5237SOystein Eftevaag backup_registry_.push_back(backup); // add it to a convenient list
1699*08ab5237SOystein Eftevaag }
1700*08ab5237SOystein Eftevaag }
1701*08ab5237SOystein Eftevaag
1702*08ab5237SOystein Eftevaag // Restores the saved flag states into the flag registry. We
1703*08ab5237SOystein Eftevaag // assume no flags were added or deleted from the registry since
1704*08ab5237SOystein Eftevaag // the SaveFromRegistry; if they were, that's trouble! Must be
1705*08ab5237SOystein Eftevaag // called when the registry mutex is not held.
RestoreToRegistry()1706*08ab5237SOystein Eftevaag void RestoreToRegistry() {
1707*08ab5237SOystein Eftevaag FlagRegistryLock frl(main_registry_);
1708*08ab5237SOystein Eftevaag vector<CommandLineFlag*>::const_iterator it;
1709*08ab5237SOystein Eftevaag for (it = backup_registry_.begin(); it != backup_registry_.end(); ++it) {
1710*08ab5237SOystein Eftevaag CommandLineFlag* main = main_registry_->FindFlagLocked((*it)->name());
1711*08ab5237SOystein Eftevaag if (main != NULL) { // if NULL, flag got deleted from registry(!)
1712*08ab5237SOystein Eftevaag main->CopyFrom(**it);
1713*08ab5237SOystein Eftevaag }
1714*08ab5237SOystein Eftevaag }
1715*08ab5237SOystein Eftevaag }
1716*08ab5237SOystein Eftevaag
1717*08ab5237SOystein Eftevaag private:
1718*08ab5237SOystein Eftevaag FlagRegistry* const main_registry_;
1719*08ab5237SOystein Eftevaag vector<CommandLineFlag*> backup_registry_;
1720*08ab5237SOystein Eftevaag
1721*08ab5237SOystein Eftevaag FlagSaverImpl(const FlagSaverImpl&); // no copying!
1722*08ab5237SOystein Eftevaag void operator=(const FlagSaverImpl&);
1723*08ab5237SOystein Eftevaag };
1724*08ab5237SOystein Eftevaag
FlagSaver()1725*08ab5237SOystein Eftevaag FlagSaver::FlagSaver()
1726*08ab5237SOystein Eftevaag : impl_(new FlagSaverImpl(FlagRegistry::GlobalRegistry())) {
1727*08ab5237SOystein Eftevaag impl_->SaveFromRegistry();
1728*08ab5237SOystein Eftevaag }
1729*08ab5237SOystein Eftevaag
~FlagSaver()1730*08ab5237SOystein Eftevaag FlagSaver::~FlagSaver() {
1731*08ab5237SOystein Eftevaag impl_->RestoreToRegistry();
1732*08ab5237SOystein Eftevaag delete impl_;
1733*08ab5237SOystein Eftevaag }
1734*08ab5237SOystein Eftevaag
1735*08ab5237SOystein Eftevaag
1736*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1737*08ab5237SOystein Eftevaag // CommandlineFlagsIntoString()
1738*08ab5237SOystein Eftevaag // ReadFlagsFromString()
1739*08ab5237SOystein Eftevaag // AppendFlagsIntoFile()
1740*08ab5237SOystein Eftevaag // ReadFromFlagsFile()
1741*08ab5237SOystein Eftevaag // These are mostly-deprecated routines that stick the
1742*08ab5237SOystein Eftevaag // commandline flags into a file/string and read them back
1743*08ab5237SOystein Eftevaag // out again. I can see a use for CommandlineFlagsIntoString,
1744*08ab5237SOystein Eftevaag // for creating a flagfile, but the rest don't seem that useful
1745*08ab5237SOystein Eftevaag // -- some, I think, are a poor-man's attempt at FlagSaver --
1746*08ab5237SOystein Eftevaag // and are included only until we can delete them from callers.
1747*08ab5237SOystein Eftevaag // Note they don't save --flagfile flags (though they do save
1748*08ab5237SOystein Eftevaag // the result of having called the flagfile, of course).
1749*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1750*08ab5237SOystein Eftevaag
TheseCommandlineFlagsIntoString(const vector<CommandLineFlagInfo> & flags)1751*08ab5237SOystein Eftevaag static string TheseCommandlineFlagsIntoString(
1752*08ab5237SOystein Eftevaag const vector<CommandLineFlagInfo>& flags) {
1753*08ab5237SOystein Eftevaag vector<CommandLineFlagInfo>::const_iterator i;
1754*08ab5237SOystein Eftevaag
1755*08ab5237SOystein Eftevaag size_t retval_space = 0;
1756*08ab5237SOystein Eftevaag for (i = flags.begin(); i != flags.end(); ++i) {
1757*08ab5237SOystein Eftevaag // An (over)estimate of how much space it will take to print this flag
1758*08ab5237SOystein Eftevaag retval_space += i->name.length() + i->current_value.length() + 5;
1759*08ab5237SOystein Eftevaag }
1760*08ab5237SOystein Eftevaag
1761*08ab5237SOystein Eftevaag string retval;
1762*08ab5237SOystein Eftevaag retval.reserve(retval_space);
1763*08ab5237SOystein Eftevaag for (i = flags.begin(); i != flags.end(); ++i) {
1764*08ab5237SOystein Eftevaag retval += "--";
1765*08ab5237SOystein Eftevaag retval += i->name;
1766*08ab5237SOystein Eftevaag retval += "=";
1767*08ab5237SOystein Eftevaag retval += i->current_value;
1768*08ab5237SOystein Eftevaag retval += "\n";
1769*08ab5237SOystein Eftevaag }
1770*08ab5237SOystein Eftevaag return retval;
1771*08ab5237SOystein Eftevaag }
1772*08ab5237SOystein Eftevaag
CommandlineFlagsIntoString()1773*08ab5237SOystein Eftevaag string CommandlineFlagsIntoString() {
1774*08ab5237SOystein Eftevaag vector<CommandLineFlagInfo> sorted_flags;
1775*08ab5237SOystein Eftevaag GetAllFlags(&sorted_flags);
1776*08ab5237SOystein Eftevaag return TheseCommandlineFlagsIntoString(sorted_flags);
1777*08ab5237SOystein Eftevaag }
1778*08ab5237SOystein Eftevaag
ReadFlagsFromString(const string & flagfilecontents,const char *,bool errors_are_fatal)1779*08ab5237SOystein Eftevaag bool ReadFlagsFromString(const string& flagfilecontents,
1780*08ab5237SOystein Eftevaag const char* /*prog_name*/, // TODO(csilvers): nix this
1781*08ab5237SOystein Eftevaag bool errors_are_fatal) {
1782*08ab5237SOystein Eftevaag FlagRegistry* const registry = FlagRegistry::GlobalRegistry();
1783*08ab5237SOystein Eftevaag FlagSaverImpl saved_states(registry);
1784*08ab5237SOystein Eftevaag saved_states.SaveFromRegistry();
1785*08ab5237SOystein Eftevaag
1786*08ab5237SOystein Eftevaag CommandLineFlagParser parser(registry);
1787*08ab5237SOystein Eftevaag registry->Lock();
1788*08ab5237SOystein Eftevaag parser.ProcessOptionsFromStringLocked(flagfilecontents, SET_FLAGS_VALUE);
1789*08ab5237SOystein Eftevaag registry->Unlock();
1790*08ab5237SOystein Eftevaag // Should we handle --help and such when reading flags from a string? Sure.
1791*08ab5237SOystein Eftevaag HandleCommandLineHelpFlags();
1792*08ab5237SOystein Eftevaag if (parser.ReportErrors()) {
1793*08ab5237SOystein Eftevaag // Error. Restore all global flags to their previous values.
1794*08ab5237SOystein Eftevaag if (errors_are_fatal)
1795*08ab5237SOystein Eftevaag gflags_exitfunc(1);
1796*08ab5237SOystein Eftevaag saved_states.RestoreToRegistry();
1797*08ab5237SOystein Eftevaag return false;
1798*08ab5237SOystein Eftevaag }
1799*08ab5237SOystein Eftevaag return true;
1800*08ab5237SOystein Eftevaag }
1801*08ab5237SOystein Eftevaag
1802*08ab5237SOystein Eftevaag // TODO(csilvers): nix prog_name in favor of ProgramInvocationShortName()
AppendFlagsIntoFile(const string & filename,const char * prog_name)1803*08ab5237SOystein Eftevaag bool AppendFlagsIntoFile(const string& filename, const char *prog_name) {
1804*08ab5237SOystein Eftevaag FILE *fp;
1805*08ab5237SOystein Eftevaag if (SafeFOpen(&fp, filename.c_str(), "a") != 0) {
1806*08ab5237SOystein Eftevaag return false;
1807*08ab5237SOystein Eftevaag }
1808*08ab5237SOystein Eftevaag
1809*08ab5237SOystein Eftevaag if (prog_name)
1810*08ab5237SOystein Eftevaag fprintf(fp, "%s\n", prog_name);
1811*08ab5237SOystein Eftevaag
1812*08ab5237SOystein Eftevaag vector<CommandLineFlagInfo> flags;
1813*08ab5237SOystein Eftevaag GetAllFlags(&flags);
1814*08ab5237SOystein Eftevaag // But we don't want --flagfile, which leads to weird recursion issues
1815*08ab5237SOystein Eftevaag vector<CommandLineFlagInfo>::iterator i;
1816*08ab5237SOystein Eftevaag for (i = flags.begin(); i != flags.end(); ++i) {
1817*08ab5237SOystein Eftevaag if (strcmp(i->name.c_str(), "flagfile") == 0) {
1818*08ab5237SOystein Eftevaag flags.erase(i);
1819*08ab5237SOystein Eftevaag break;
1820*08ab5237SOystein Eftevaag }
1821*08ab5237SOystein Eftevaag }
1822*08ab5237SOystein Eftevaag fprintf(fp, "%s", TheseCommandlineFlagsIntoString(flags).c_str());
1823*08ab5237SOystein Eftevaag
1824*08ab5237SOystein Eftevaag fclose(fp);
1825*08ab5237SOystein Eftevaag return true;
1826*08ab5237SOystein Eftevaag }
1827*08ab5237SOystein Eftevaag
ReadFromFlagsFile(const string & filename,const char * prog_name,bool errors_are_fatal)1828*08ab5237SOystein Eftevaag bool ReadFromFlagsFile(const string& filename, const char* prog_name,
1829*08ab5237SOystein Eftevaag bool errors_are_fatal) {
1830*08ab5237SOystein Eftevaag return ReadFlagsFromString(ReadFileIntoString(filename.c_str()),
1831*08ab5237SOystein Eftevaag prog_name, errors_are_fatal);
1832*08ab5237SOystein Eftevaag }
1833*08ab5237SOystein Eftevaag
1834*08ab5237SOystein Eftevaag
1835*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1836*08ab5237SOystein Eftevaag // BoolFromEnv()
1837*08ab5237SOystein Eftevaag // Int32FromEnv()
1838*08ab5237SOystein Eftevaag // Uint32FromEnv()
1839*08ab5237SOystein Eftevaag // Int64FromEnv()
1840*08ab5237SOystein Eftevaag // Uint64FromEnv()
1841*08ab5237SOystein Eftevaag // DoubleFromEnv()
1842*08ab5237SOystein Eftevaag // StringFromEnv()
1843*08ab5237SOystein Eftevaag // Reads the value from the environment and returns it.
1844*08ab5237SOystein Eftevaag // We use an FlagValue to make the parsing easy.
1845*08ab5237SOystein Eftevaag // Example usage:
1846*08ab5237SOystein Eftevaag // DEFINE_bool(myflag, BoolFromEnv("MYFLAG_DEFAULT", false), "whatever");
1847*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1848*08ab5237SOystein Eftevaag
BoolFromEnv(const char * v,bool dflt)1849*08ab5237SOystein Eftevaag bool BoolFromEnv(const char *v, bool dflt) {
1850*08ab5237SOystein Eftevaag return GetFromEnv(v, dflt);
1851*08ab5237SOystein Eftevaag }
Int32FromEnv(const char * v,int32 dflt)1852*08ab5237SOystein Eftevaag int32 Int32FromEnv(const char *v, int32 dflt) {
1853*08ab5237SOystein Eftevaag return GetFromEnv(v, dflt);
1854*08ab5237SOystein Eftevaag }
Uint32FromEnv(const char * v,uint32 dflt)1855*08ab5237SOystein Eftevaag uint32 Uint32FromEnv(const char *v, uint32 dflt) {
1856*08ab5237SOystein Eftevaag return GetFromEnv(v, dflt);
1857*08ab5237SOystein Eftevaag }
Int64FromEnv(const char * v,int64 dflt)1858*08ab5237SOystein Eftevaag int64 Int64FromEnv(const char *v, int64 dflt) {
1859*08ab5237SOystein Eftevaag return GetFromEnv(v, dflt);
1860*08ab5237SOystein Eftevaag }
Uint64FromEnv(const char * v,uint64 dflt)1861*08ab5237SOystein Eftevaag uint64 Uint64FromEnv(const char *v, uint64 dflt) {
1862*08ab5237SOystein Eftevaag return GetFromEnv(v, dflt);
1863*08ab5237SOystein Eftevaag }
DoubleFromEnv(const char * v,double dflt)1864*08ab5237SOystein Eftevaag double DoubleFromEnv(const char *v, double dflt) {
1865*08ab5237SOystein Eftevaag return GetFromEnv(v, dflt);
1866*08ab5237SOystein Eftevaag }
1867*08ab5237SOystein Eftevaag
1868*08ab5237SOystein Eftevaag #ifdef _MSC_VER
1869*08ab5237SOystein Eftevaag # pragma warning(push)
1870*08ab5237SOystein Eftevaag # pragma warning(disable: 4996) // ignore getenv security warning
1871*08ab5237SOystein Eftevaag #endif
StringFromEnv(const char * varname,const char * dflt)1872*08ab5237SOystein Eftevaag const char *StringFromEnv(const char *varname, const char *dflt) {
1873*08ab5237SOystein Eftevaag const char* const val = getenv(varname);
1874*08ab5237SOystein Eftevaag return val ? val : dflt;
1875*08ab5237SOystein Eftevaag }
1876*08ab5237SOystein Eftevaag #ifdef _MSC_VER
1877*08ab5237SOystein Eftevaag # pragma warning(pop)
1878*08ab5237SOystein Eftevaag #endif
1879*08ab5237SOystein Eftevaag
1880*08ab5237SOystein Eftevaag
1881*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1882*08ab5237SOystein Eftevaag // RegisterFlagValidator()
1883*08ab5237SOystein Eftevaag // RegisterFlagValidator() is the function that clients use to
1884*08ab5237SOystein Eftevaag // 'decorate' a flag with a validation function. Once this is
1885*08ab5237SOystein Eftevaag // done, every time the flag is set (including when the flag
1886*08ab5237SOystein Eftevaag // is parsed from argv), the validator-function is called.
1887*08ab5237SOystein Eftevaag // These functions return true if the validator was added
1888*08ab5237SOystein Eftevaag // successfully, or false if not: the flag already has a validator,
1889*08ab5237SOystein Eftevaag // (only one allowed per flag), the 1st arg isn't a flag, etc.
1890*08ab5237SOystein Eftevaag // This function is not thread-safe.
1891*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1892*08ab5237SOystein Eftevaag
RegisterFlagValidator(const bool * flag,bool (* validate_fn)(const char *,bool))1893*08ab5237SOystein Eftevaag bool RegisterFlagValidator(const bool* flag,
1894*08ab5237SOystein Eftevaag bool (*validate_fn)(const char*, bool)) {
1895*08ab5237SOystein Eftevaag return AddFlagValidator(flag, reinterpret_cast<ValidateFnProto>(validate_fn));
1896*08ab5237SOystein Eftevaag }
RegisterFlagValidator(const int32 * flag,bool (* validate_fn)(const char *,int32))1897*08ab5237SOystein Eftevaag bool RegisterFlagValidator(const int32* flag,
1898*08ab5237SOystein Eftevaag bool (*validate_fn)(const char*, int32)) {
1899*08ab5237SOystein Eftevaag return AddFlagValidator(flag, reinterpret_cast<ValidateFnProto>(validate_fn));
1900*08ab5237SOystein Eftevaag }
RegisterFlagValidator(const uint32 * flag,bool (* validate_fn)(const char *,uint32))1901*08ab5237SOystein Eftevaag bool RegisterFlagValidator(const uint32* flag,
1902*08ab5237SOystein Eftevaag bool (*validate_fn)(const char*, uint32)) {
1903*08ab5237SOystein Eftevaag return AddFlagValidator(flag, reinterpret_cast<ValidateFnProto>(validate_fn));
1904*08ab5237SOystein Eftevaag }
RegisterFlagValidator(const int64 * flag,bool (* validate_fn)(const char *,int64))1905*08ab5237SOystein Eftevaag bool RegisterFlagValidator(const int64* flag,
1906*08ab5237SOystein Eftevaag bool (*validate_fn)(const char*, int64)) {
1907*08ab5237SOystein Eftevaag return AddFlagValidator(flag, reinterpret_cast<ValidateFnProto>(validate_fn));
1908*08ab5237SOystein Eftevaag }
RegisterFlagValidator(const uint64 * flag,bool (* validate_fn)(const char *,uint64))1909*08ab5237SOystein Eftevaag bool RegisterFlagValidator(const uint64* flag,
1910*08ab5237SOystein Eftevaag bool (*validate_fn)(const char*, uint64)) {
1911*08ab5237SOystein Eftevaag return AddFlagValidator(flag, reinterpret_cast<ValidateFnProto>(validate_fn));
1912*08ab5237SOystein Eftevaag }
RegisterFlagValidator(const double * flag,bool (* validate_fn)(const char *,double))1913*08ab5237SOystein Eftevaag bool RegisterFlagValidator(const double* flag,
1914*08ab5237SOystein Eftevaag bool (*validate_fn)(const char*, double)) {
1915*08ab5237SOystein Eftevaag return AddFlagValidator(flag, reinterpret_cast<ValidateFnProto>(validate_fn));
1916*08ab5237SOystein Eftevaag }
RegisterFlagValidator(const string * flag,bool (* validate_fn)(const char *,const string &))1917*08ab5237SOystein Eftevaag bool RegisterFlagValidator(const string* flag,
1918*08ab5237SOystein Eftevaag bool (*validate_fn)(const char*, const string&)) {
1919*08ab5237SOystein Eftevaag return AddFlagValidator(flag, reinterpret_cast<ValidateFnProto>(validate_fn));
1920*08ab5237SOystein Eftevaag }
1921*08ab5237SOystein Eftevaag
1922*08ab5237SOystein Eftevaag
1923*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1924*08ab5237SOystein Eftevaag // ParseCommandLineFlags()
1925*08ab5237SOystein Eftevaag // ParseCommandLineNonHelpFlags()
1926*08ab5237SOystein Eftevaag // HandleCommandLineHelpFlags()
1927*08ab5237SOystein Eftevaag // This is the main function called from main(), to actually
1928*08ab5237SOystein Eftevaag // parse the commandline. It modifies argc and argv as described
1929*08ab5237SOystein Eftevaag // at the top of gflags.h. You can also divide this
1930*08ab5237SOystein Eftevaag // function into two parts, if you want to do work between
1931*08ab5237SOystein Eftevaag // the parsing of the flags and the printing of any help output.
1932*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1933*08ab5237SOystein Eftevaag
ParseCommandLineFlagsInternal(int * argc,char *** argv,bool remove_flags,bool do_report)1934*08ab5237SOystein Eftevaag static uint32 ParseCommandLineFlagsInternal(int* argc, char*** argv,
1935*08ab5237SOystein Eftevaag bool remove_flags, bool do_report) {
1936*08ab5237SOystein Eftevaag SetArgv(*argc, const_cast<const char**>(*argv)); // save it for later
1937*08ab5237SOystein Eftevaag
1938*08ab5237SOystein Eftevaag FlagRegistry* const registry = FlagRegistry::GlobalRegistry();
1939*08ab5237SOystein Eftevaag CommandLineFlagParser parser(registry);
1940*08ab5237SOystein Eftevaag
1941*08ab5237SOystein Eftevaag // When we parse the commandline flags, we'll handle --flagfile,
1942*08ab5237SOystein Eftevaag // --tryfromenv, etc. as we see them (since flag-evaluation order
1943*08ab5237SOystein Eftevaag // may be important). But sometimes apps set FLAGS_tryfromenv/etc.
1944*08ab5237SOystein Eftevaag // manually before calling ParseCommandLineFlags. We want to evaluate
1945*08ab5237SOystein Eftevaag // those too, as if they were the first flags on the commandline.
1946*08ab5237SOystein Eftevaag registry->Lock();
1947*08ab5237SOystein Eftevaag parser.ProcessFlagfileLocked(FLAGS_flagfile, SET_FLAGS_VALUE);
1948*08ab5237SOystein Eftevaag // Last arg here indicates whether flag-not-found is a fatal error or not
1949*08ab5237SOystein Eftevaag parser.ProcessFromenvLocked(FLAGS_fromenv, SET_FLAGS_VALUE, true);
1950*08ab5237SOystein Eftevaag parser.ProcessFromenvLocked(FLAGS_tryfromenv, SET_FLAGS_VALUE, false);
1951*08ab5237SOystein Eftevaag registry->Unlock();
1952*08ab5237SOystein Eftevaag
1953*08ab5237SOystein Eftevaag // Now get the flags specified on the commandline
1954*08ab5237SOystein Eftevaag const int r = parser.ParseNewCommandLineFlags(argc, argv, remove_flags);
1955*08ab5237SOystein Eftevaag
1956*08ab5237SOystein Eftevaag if (do_report)
1957*08ab5237SOystein Eftevaag HandleCommandLineHelpFlags(); // may cause us to exit on --help, etc.
1958*08ab5237SOystein Eftevaag
1959*08ab5237SOystein Eftevaag // See if any of the unset flags fail their validation checks
1960*08ab5237SOystein Eftevaag parser.ValidateUnmodifiedFlags();
1961*08ab5237SOystein Eftevaag
1962*08ab5237SOystein Eftevaag if (parser.ReportErrors()) // may cause us to exit on illegal flags
1963*08ab5237SOystein Eftevaag gflags_exitfunc(1);
1964*08ab5237SOystein Eftevaag return r;
1965*08ab5237SOystein Eftevaag }
1966*08ab5237SOystein Eftevaag
ParseCommandLineFlags(int * argc,char *** argv,bool remove_flags)1967*08ab5237SOystein Eftevaag uint32 ParseCommandLineFlags(int* argc, char*** argv, bool remove_flags) {
1968*08ab5237SOystein Eftevaag return ParseCommandLineFlagsInternal(argc, argv, remove_flags, true);
1969*08ab5237SOystein Eftevaag }
1970*08ab5237SOystein Eftevaag
ParseCommandLineNonHelpFlags(int * argc,char *** argv,bool remove_flags)1971*08ab5237SOystein Eftevaag uint32 ParseCommandLineNonHelpFlags(int* argc, char*** argv,
1972*08ab5237SOystein Eftevaag bool remove_flags) {
1973*08ab5237SOystein Eftevaag return ParseCommandLineFlagsInternal(argc, argv, remove_flags, false);
1974*08ab5237SOystein Eftevaag }
1975*08ab5237SOystein Eftevaag
1976*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1977*08ab5237SOystein Eftevaag // AllowCommandLineReparsing()
1978*08ab5237SOystein Eftevaag // ReparseCommandLineNonHelpFlags()
1979*08ab5237SOystein Eftevaag // This is most useful for shared libraries. The idea is if
1980*08ab5237SOystein Eftevaag // a flag is defined in a shared library that is dlopen'ed
1981*08ab5237SOystein Eftevaag // sometime after main(), you can ParseCommandLineFlags before
1982*08ab5237SOystein Eftevaag // the dlopen, then ReparseCommandLineNonHelpFlags() after the
1983*08ab5237SOystein Eftevaag // dlopen, to get the new flags. But you have to explicitly
1984*08ab5237SOystein Eftevaag // Allow() it; otherwise, you get the normal default behavior
1985*08ab5237SOystein Eftevaag // of unrecognized flags calling a fatal error.
1986*08ab5237SOystein Eftevaag // TODO(csilvers): this isn't used. Just delete it?
1987*08ab5237SOystein Eftevaag // --------------------------------------------------------------------
1988*08ab5237SOystein Eftevaag
AllowCommandLineReparsing()1989*08ab5237SOystein Eftevaag void AllowCommandLineReparsing() {
1990*08ab5237SOystein Eftevaag allow_command_line_reparsing = true;
1991*08ab5237SOystein Eftevaag }
1992*08ab5237SOystein Eftevaag
ReparseCommandLineNonHelpFlags()1993*08ab5237SOystein Eftevaag void ReparseCommandLineNonHelpFlags() {
1994*08ab5237SOystein Eftevaag // We make a copy of argc and argv to pass in
1995*08ab5237SOystein Eftevaag const vector<string>& argvs = GetArgvs();
1996*08ab5237SOystein Eftevaag int tmp_argc = static_cast<int>(argvs.size());
1997*08ab5237SOystein Eftevaag char** tmp_argv = new char* [tmp_argc + 1];
1998*08ab5237SOystein Eftevaag for (int i = 0; i < tmp_argc; ++i)
1999*08ab5237SOystein Eftevaag tmp_argv[i] = strdup(argvs[i].c_str()); // TODO(csilvers): don't dup
2000*08ab5237SOystein Eftevaag
2001*08ab5237SOystein Eftevaag ParseCommandLineNonHelpFlags(&tmp_argc, &tmp_argv, false);
2002*08ab5237SOystein Eftevaag
2003*08ab5237SOystein Eftevaag for (int i = 0; i < tmp_argc; ++i)
2004*08ab5237SOystein Eftevaag free(tmp_argv[i]);
2005*08ab5237SOystein Eftevaag delete[] tmp_argv;
2006*08ab5237SOystein Eftevaag }
2007*08ab5237SOystein Eftevaag
ShutDownCommandLineFlags()2008*08ab5237SOystein Eftevaag void ShutDownCommandLineFlags() {
2009*08ab5237SOystein Eftevaag FlagRegistry::DeleteGlobalRegistry();
2010*08ab5237SOystein Eftevaag }
2011*08ab5237SOystein Eftevaag
2012*08ab5237SOystein Eftevaag
2013*08ab5237SOystein Eftevaag } // namespace GFLAGS_NAMESPACE
2014