1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
5 // This class works with command lines: building and parsing.
6 // Arguments with prefixes ('--', '-', and on Windows, '/') are switches.
7 // Switches will precede all other arguments without switch prefixes.
8 // Switches can optionally have values, delimited by '=', e.g., "-switch=value".
9 // An argument of "--" will terminate switch parsing during initialization,
10 // interpreting subsequent tokens as non-switch arguments, regardless of prefix.
12 // There is a singleton read-only CommandLine that represents the command line
13 // that the current process was started with. It must be initialized in main().
15 #ifndef BASE_COMMAND_LINE_H_
16 #define BASE_COMMAND_LINE_H_
23 #include "base/base_export.h"
24 #include "base/strings/string16.h"
25 #include "build/build_config.h"
31 class BASE_EXPORT CommandLine
{
34 // The native command line string type.
35 typedef base::string16 StringType
;
36 #elif defined(OS_POSIX)
37 typedef std::string StringType
;
40 typedef StringType::value_type CharType
;
41 typedef std::vector
<StringType
> StringVector
;
42 typedef std::map
<std::string
, StringType
> SwitchMap
;
44 // A constructor for CommandLines that only carry switches and arguments.
45 enum NoProgram
{ NO_PROGRAM
};
46 explicit CommandLine(NoProgram no_program
);
48 // Construct a new command line with |program| as argv[0].
49 explicit CommandLine(const FilePath
& program
);
51 // Construct a new command line from an argument list.
52 CommandLine(int argc
, const CharType
* const* argv
);
53 explicit CommandLine(const StringVector
& argv
);
58 // By default this class will treat command-line arguments beginning with
59 // slashes as switches on Windows, but not other platforms.
61 // If this behavior is inappropriate for your application, you can call this
62 // function BEFORE initializing the current process' global command line
63 // object and the behavior will be the same as Posix systems (only hyphens
64 // begin switches, everything else will be an arg).
65 static void set_slash_is_not_a_switch();
68 // Initialize the current process CommandLine singleton. On Windows, ignores
69 // its arguments (we instead parse GetCommandLineW() directly) because we
70 // don't trust the CRT's parsing of the command line, but it still must be
71 // called to set up the command line. Returns false if initialization has
72 // already occurred, and true otherwise. Only the caller receiving a 'true'
73 // return value should take responsibility for calling Reset.
74 static bool Init(int argc
, const char* const* argv
);
76 // Destroys the current process CommandLine singleton. This is necessary if
77 // you want to reset the base library to its initial state (for example, in an
78 // outer library that needs to be able to terminate, and be re-initialized).
79 // If Init is called only once, as in main(), Reset() is not necessary.
82 // Get the singleton CommandLine representing the current process's
83 // command line. Note: returned value is mutable, but not thread safe;
84 // only mutate if you know what you're doing!
85 static CommandLine
* ForCurrentProcess();
87 // Returns true if the CommandLine has been initialized for the given process.
88 static bool InitializedForCurrentProcess();
91 static CommandLine
FromString(const base::string16
& command_line
);
94 // Initialize from an argv vector.
95 void InitFromArgv(int argc
, const CharType
* const* argv
);
96 void InitFromArgv(const StringVector
& argv
);
98 // Constructs and returns the represented command line string.
99 // CAUTION! This should be avoided on POSIX because quoting behavior is
101 StringType
GetCommandLineString() const {
102 return GetCommandLineStringInternal(false);
106 // Constructs and returns the represented command line string. Assumes the
107 // command line contains placeholders (eg, %1) and quotes any program or
108 // argument with a '%' in it. This should be avoided unless the placeholder is
109 // required by an external interface (eg, the Windows registry), because it is
110 // not generally safe to replace it with an arbitrary string. If possible,
111 // placeholders should be replaced *before* converting the command line to a
113 StringType
GetCommandLineStringWithPlaceholders() const {
114 return GetCommandLineStringInternal(true);
118 // Constructs and returns the represented arguments string.
119 // CAUTION! This should be avoided on POSIX because quoting behavior is
121 StringType
GetArgumentsString() const {
122 return GetArgumentsStringInternal(false);
126 // Constructs and returns the represented arguments string. Assumes the
127 // command line contains placeholders (eg, %1) and quotes any argument with a
128 // '%' in it. This should be avoided unless the placeholder is required by an
129 // external interface (eg, the Windows registry), because it is not generally
130 // safe to replace it with an arbitrary string. If possible, placeholders
131 // should be replaced *before* converting the arguments to a string.
132 StringType
GetArgumentsStringWithPlaceholders() const {
133 return GetArgumentsStringInternal(true);
137 // Returns the original command line string as a vector of strings.
138 const StringVector
& argv() const { return argv_
; }
140 // Get and Set the program part of the command line string (the first item).
141 FilePath
GetProgram() const;
142 void SetProgram(const FilePath
& program
);
144 // Returns true if this command line contains the given switch.
145 // (Switch names are case-insensitive).
146 // The second override provides an optimized version to avoid inlining the
147 // codegen for the string allocation.
148 bool HasSwitch(const std::string
& switch_string
) const;
149 bool HasSwitch(const char switch_constant
[]) const;
151 // Returns the value associated with the given switch. If the switch has no
152 // value or isn't present, this method returns the empty string.
153 std::string
GetSwitchValueASCII(const std::string
& switch_string
) const;
154 FilePath
GetSwitchValuePath(const std::string
& switch_string
) const;
155 StringType
GetSwitchValueNative(const std::string
& switch_string
) const;
157 // Get a copy of all switches, along with their values.
158 const SwitchMap
& GetSwitches() const { return switches_
; }
160 // Append a switch [with optional value] to the command line.
161 // Note: Switches will precede arguments regardless of appending order.
162 void AppendSwitch(const std::string
& switch_string
);
163 void AppendSwitchPath(const std::string
& switch_string
,
164 const FilePath
& path
);
165 void AppendSwitchNative(const std::string
& switch_string
,
166 const StringType
& value
);
167 void AppendSwitchASCII(const std::string
& switch_string
,
168 const std::string
& value
);
170 // Copy a set of switches (and any values) from another command line.
171 // Commonly used when launching a subprocess.
172 void CopySwitchesFrom(const CommandLine
& source
,
173 const char* const switches
[],
176 // Get the remaining arguments to the command.
177 StringVector
GetArgs() const;
179 // Append an argument to the command line. Note that the argument is quoted
180 // properly such that it is interpreted as one argument to the target command.
181 // AppendArg is primarily for ASCII; non-ASCII input is interpreted as UTF-8.
182 // Note: Switches will precede arguments regardless of appending order.
183 void AppendArg(const std::string
& value
);
184 void AppendArgPath(const FilePath
& value
);
185 void AppendArgNative(const StringType
& value
);
187 // Append the switches and arguments from another command line to this one.
188 // If |include_program| is true, include |other|'s program as well.
189 void AppendArguments(const CommandLine
& other
, bool include_program
);
191 // Insert a command before the current command.
192 // Common for debuggers, like "valgrind" or "gdb --args".
193 void PrependWrapper(const StringType
& wrapper
);
196 // Initialize by parsing the given command line string.
197 // The program name is assumed to be the first item in the string.
198 void ParseFromString(const base::string16
& command_line
);
202 // Disallow default constructor; a program name must be explicitly specified.
204 // Allow the copy constructor. A common pattern is to copy of the current
205 // process's command line and then add some flags to it. For example:
206 // CommandLine cl(*CommandLine::ForCurrentProcess());
207 // cl.AppendSwitch(...);
209 // Internal version of GetCommandLineString. If |quote_placeholders| is true,
210 // also quotes parts with '%' in them.
211 StringType
GetCommandLineStringInternal(bool quote_placeholders
) const;
213 // Internal version of GetArgumentsString. If |quote_placeholders| is true,
214 // also quotes parts with '%' in them.
215 StringType
GetArgumentsStringInternal(bool quote_placeholders
) const;
217 // The singleton CommandLine representing the current process's command line.
218 static CommandLine
* current_process_commandline_
;
220 // The argv array: { program, [(--|-|/)switch[=value]]*, [--], [argument]* }
223 // Parsed-out switch keys and values.
226 // The index after the program and switches, any arguments start here.
232 #endif // BASE_COMMAND_LINE_H_