1 /*
2 * Copyright 2006 The WebRTC Project Authors. All rights reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
11 #include <stdio.h>
12 #include <stdlib.h>
13 #include <string.h>
14
15
16 #if defined(WEBRTC_WIN)
17 #include "webrtc/base/win32.h"
18 #include <shellapi.h>
19 #endif
20
21 #include "webrtc/base/flags.h"
22
23 namespace rtc {
24 // -----------------------------------------------------------------------------
25 // Implementation of Flag
26
Flag(const char * file,const char * name,const char * comment,Type type,void * variable,FlagValue default__)27 Flag::Flag(const char* file, const char* name, const char* comment,
28 Type type, void* variable, FlagValue default__)
29 : file_(file),
30 name_(name),
31 comment_(comment),
32 type_(type),
33 variable_(reinterpret_cast<FlagValue*>(variable)),
34 default_(default__) {
35 FlagList::Register(this);
36 }
37
38
SetToDefault()39 void Flag::SetToDefault() {
40 // Note that we cannot simply do '*variable_ = default_;' since
41 // flag variables are not really of type FlagValue and thus may
42 // be smaller! The FlagValue union is simply 'overlayed' on top
43 // of a flag variable for convenient access. Since union members
44 // are guarantee to be aligned at the beginning, this works.
45 switch (type_) {
46 case Flag::BOOL:
47 variable_->b = default_.b;
48 return;
49 case Flag::INT:
50 variable_->i = default_.i;
51 return;
52 case Flag::FLOAT:
53 variable_->f = default_.f;
54 return;
55 case Flag::STRING:
56 variable_->s = default_.s;
57 return;
58 }
59 UNREACHABLE();
60 }
61
62
Type2String(Flag::Type type)63 static const char* Type2String(Flag::Type type) {
64 switch (type) {
65 case Flag::BOOL: return "bool";
66 case Flag::INT: return "int";
67 case Flag::FLOAT: return "float";
68 case Flag::STRING: return "string";
69 }
70 UNREACHABLE();
71 return NULL;
72 }
73
74
PrintFlagValue(Flag::Type type,FlagValue * p)75 static void PrintFlagValue(Flag::Type type, FlagValue* p) {
76 switch (type) {
77 case Flag::BOOL:
78 printf("%s", (p->b ? "true" : "false"));
79 return;
80 case Flag::INT:
81 printf("%d", p->i);
82 return;
83 case Flag::FLOAT:
84 printf("%f", p->f);
85 return;
86 case Flag::STRING:
87 printf("%s", p->s);
88 return;
89 }
90 UNREACHABLE();
91 }
92
93
Print(bool print_current_value)94 void Flag::Print(bool print_current_value) {
95 printf(" --%s (%s) type: %s default: ", name_, comment_,
96 Type2String(type_));
97 PrintFlagValue(type_, &default_);
98 if (print_current_value) {
99 printf(" current value: ");
100 PrintFlagValue(type_, variable_);
101 }
102 printf("\n");
103 }
104
105
106 // -----------------------------------------------------------------------------
107 // Implementation of FlagList
108
109 Flag* FlagList::list_ = NULL;
110
111
FlagList()112 FlagList::FlagList() {
113 list_ = NULL;
114 }
115
Print(const char * file,bool print_current_value)116 void FlagList::Print(const char* file, bool print_current_value) {
117 // Since flag registration is likely by file (= C++ file),
118 // we don't need to sort by file and still get grouped output.
119 const char* current = NULL;
120 for (Flag* f = list_; f != NULL; f = f->next()) {
121 if (file == NULL || file == f->file()) {
122 if (current != f->file()) {
123 printf("Flags from %s:\n", f->file());
124 current = f->file();
125 }
126 f->Print(print_current_value);
127 }
128 }
129 }
130
131
Lookup(const char * name)132 Flag* FlagList::Lookup(const char* name) {
133 Flag* f = list_;
134 while (f != NULL && strcmp(name, f->name()) != 0)
135 f = f->next();
136 return f;
137 }
138
139
SplitArgument(const char * arg,char * buffer,int buffer_size,const char ** name,const char ** value,bool * is_bool)140 void FlagList::SplitArgument(const char* arg,
141 char* buffer, int buffer_size,
142 const char** name, const char** value,
143 bool* is_bool) {
144 *name = NULL;
145 *value = NULL;
146 *is_bool = false;
147
148 if (*arg == '-') {
149 // find the begin of the flag name
150 arg++; // remove 1st '-'
151 if (*arg == '-')
152 arg++; // remove 2nd '-'
153 if (arg[0] == 'n' && arg[1] == 'o') {
154 arg += 2; // remove "no"
155 *is_bool = true;
156 }
157 *name = arg;
158
159 // find the end of the flag name
160 while (*arg != '\0' && *arg != '=')
161 arg++;
162
163 // get the value if any
164 if (*arg == '=') {
165 // make a copy so we can NUL-terminate flag name
166 int n = static_cast<int>(arg - *name);
167 if (n >= buffer_size)
168 Fatal(__FILE__, __LINE__, "CHECK(%s) failed", "n < buffer_size");
169 memcpy(buffer, *name, n * sizeof(char));
170 buffer[n] = '\0';
171 *name = buffer;
172 // get the value
173 *value = arg + 1;
174 }
175 }
176 }
177
178
SetFlagsFromCommandLine(int * argc,const char ** argv,bool remove_flags)179 int FlagList::SetFlagsFromCommandLine(int* argc, const char** argv,
180 bool remove_flags) {
181 // parse arguments
182 for (int i = 1; i < *argc; /* see below */) {
183 int j = i; // j > 0
184 const char* arg = argv[i++];
185
186 // split arg into flag components
187 char buffer[1024];
188 const char* name;
189 const char* value;
190 bool is_bool;
191 SplitArgument(arg, buffer, sizeof buffer, &name, &value, &is_bool);
192
193 if (name != NULL) {
194 // lookup the flag
195 Flag* flag = Lookup(name);
196 if (flag == NULL) {
197 fprintf(stderr, "Error: unrecognized flag %s\n", arg);
198 return j;
199 }
200
201 // if we still need a flag value, use the next argument if available
202 if (flag->type() != Flag::BOOL && value == NULL) {
203 if (i < *argc) {
204 value = argv[i++];
205 } else {
206 fprintf(stderr, "Error: missing value for flag %s of type %s\n",
207 arg, Type2String(flag->type()));
208 return j;
209 }
210 }
211
212 // set the flag
213 char empty[] = { '\0' };
214 char* endp = empty;
215 switch (flag->type()) {
216 case Flag::BOOL:
217 *flag->bool_variable() = !is_bool;
218 break;
219 case Flag::INT:
220 *flag->int_variable() = strtol(value, &endp, 10);
221 break;
222 case Flag::FLOAT:
223 *flag->float_variable() = strtod(value, &endp);
224 break;
225 case Flag::STRING:
226 *flag->string_variable() = value;
227 break;
228 }
229
230 // handle errors
231 if ((flag->type() == Flag::BOOL && value != NULL) ||
232 (flag->type() != Flag::BOOL && is_bool) ||
233 *endp != '\0') {
234 fprintf(stderr, "Error: illegal value for flag %s of type %s\n",
235 arg, Type2String(flag->type()));
236 return j;
237 }
238
239 // remove the flag & value from the command
240 if (remove_flags)
241 while (j < i)
242 argv[j++] = NULL;
243 }
244 }
245
246 // shrink the argument list
247 if (remove_flags) {
248 int j = 1;
249 for (int i = 1; i < *argc; i++) {
250 if (argv[i] != NULL)
251 argv[j++] = argv[i];
252 }
253 *argc = j;
254 }
255
256 // parsed all flags successfully
257 return 0;
258 }
259
Register(Flag * flag)260 void FlagList::Register(Flag* flag) {
261 assert(flag != NULL && strlen(flag->name()) > 0);
262 if (Lookup(flag->name()) != NULL)
263 Fatal(flag->file(), 0, "flag %s declared twice", flag->name());
264 flag->next_ = list_;
265 list_ = flag;
266 }
267
268 #if defined(WEBRTC_WIN)
WindowsCommandLineArguments()269 WindowsCommandLineArguments::WindowsCommandLineArguments() {
270 // start by getting the command line.
271 LPTSTR command_line = ::GetCommandLine();
272 // now, convert it to a list of wide char strings.
273 LPWSTR *wide_argv = ::CommandLineToArgvW(command_line, &argc_);
274 // now allocate an array big enough to hold that many string pointers.
275 argv_ = new char*[argc_];
276
277 // iterate over the returned wide strings;
278 for(int i = 0; i < argc_; ++i) {
279 std::string s = rtc::ToUtf8(wide_argv[i], wcslen(wide_argv[i]));
280 char *buffer = new char[s.length() + 1];
281 rtc::strcpyn(buffer, s.length() + 1, s.c_str());
282
283 // make sure the argv array has the right string at this point.
284 argv_[i] = buffer;
285 }
286 LocalFree(wide_argv);
287 }
288
~WindowsCommandLineArguments()289 WindowsCommandLineArguments::~WindowsCommandLineArguments() {
290 // need to free each string in the array, and then the array.
291 for(int i = 0; i < argc_; i++) {
292 delete[] argv_[i];
293 }
294
295 delete[] argv_;
296 }
297 #endif // WEBRTC_WIN
298
299 } // namespace rtc
300