Avi Drissman | e4622aa | 2022-09-08 20:36:06 | [diff] [blame] | 1 | // Copyright 2012 The Chromium Authors |
license.bot | bf09a50 | 2008-08-24 00:55:55 | [diff] [blame] | 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
deanm@chromium.org | 02c8796 | 2008-10-06 10:25:35 | [diff] [blame] | 4 | |
evan@chromium.org | bb97536 | 2009-01-21 01:00:22 | [diff] [blame] | 5 | // This class works with command lines: building and parsing. |
msw@chromium.org | a40ca430 | 2011-05-14 01:10:24 | [diff] [blame] | 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". |
John Rummell | b1d5fcb | 2019-04-27 01:13:33 | [diff] [blame] | 9 | // If a switch is specified multiple times, only the last value is used. |
msw@chromium.org | a40ca430 | 2011-05-14 01:10:24 | [diff] [blame] | 10 | // An argument of "--" will terminate switch parsing during initialization, |
| 11 | // interpreting subsequent tokens as non-switch arguments, regardless of prefix. |
evan@chromium.org | bb97536 | 2009-01-21 01:00:22 | [diff] [blame] | 12 | |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 13 | // There is a singleton read-only CommandLine that represents the command line |
| 14 | // that the current process was started with. It must be initialized in main(). |
initial.commit | d7cae12 | 2008-07-26 21:49:38 | [diff] [blame] | 15 | |
deanm@chromium.org | 02c8796 | 2008-10-06 10:25:35 | [diff] [blame] | 16 | #ifndef BASE_COMMAND_LINE_H_ |
| 17 | #define BASE_COMMAND_LINE_H_ |
initial.commit | d7cae12 | 2008-07-26 21:49:38 | [diff] [blame] | 18 | |
jhawkins@chromium.org | 2edc286 | 2011-04-04 18:04:37 | [diff] [blame] | 19 | #include <stddef.h> |
Sumaid Syed | 22f60eeb | 2021-08-26 05:16:26 | [diff] [blame] | 20 | #include <functional> |
initial.commit | d7cae12 | 2008-07-26 21:49:38 | [diff] [blame] | 21 | #include <map> |
Stephan Hartmann | 4185978 | 2021-10-12 18:49:23 | [diff] [blame] | 22 | #include <memory> |
initial.commit | d7cae12 | 2008-07-26 21:49:38 | [diff] [blame] | 23 | #include <string> |
Takuto Ikuta | 9dcb26dd | 2023-12-19 08:02:08 | [diff] [blame] | 24 | #include <string_view> |
initial.commit | d7cae12 | 2008-07-26 21:49:38 | [diff] [blame] | 25 | #include <vector> |
| 26 | |
darin@chromium.org | 0bea725 | 2011-08-05 15:34:00 | [diff] [blame] | 27 | #include "base/base_export.h" |
Lei Zhang | 24dcd3e | 2023-07-07 20:08:36 | [diff] [blame] | 28 | #include "base/containers/span.h" |
Will Harris | ee9f0a2 | 2023-03-02 21:38:58 | [diff] [blame] | 29 | #include "base/debug/debugging_buildflags.h" |
jackhou | 1bd9da9 | 2015-05-21 04:48:00 | [diff] [blame] | 30 | #include "base/strings/string_piece.h" |
brettw@chromium.org | 74e9fa2 | 2010-12-29 21:06:43 | [diff] [blame] | 31 | #include "build/build_config.h" |
initial.commit | d7cae12 | 2008-07-26 21:49:38 | [diff] [blame] | 32 | |
Will Harris | ee9f0a2 | 2023-03-02 21:38:58 | [diff] [blame] | 33 | #if BUILDFLAG(ENABLE_COMMANDLINE_SEQUENCE_CHECKS) |
| 34 | #include "base/sequence_checker.h" |
| 35 | #endif // BUILDFLAG(ENABLE_COMMANDLINE_SEQUENCE_CHECKS) |
| 36 | |
brettw@chromium.org | a3ef483 | 2013-02-02 05:12:33 | [diff] [blame] | 37 | namespace base { |
brettw@chromium.org | 2f3b1cc | 2014-03-17 23:07:15 | [diff] [blame] | 38 | |
Greg Guterman | fe16560d | 2021-10-07 19:58:15 | [diff] [blame] | 39 | class DuplicateSwitchHandler; |
Greg Guterman | e605119 | 2021-10-15 21:14:36 | [diff] [blame] | 40 | class FilePath; |
sky@google.com | d4515eb | 2009-01-30 00:40:43 | [diff] [blame] | 41 | |
darin@chromium.org | 0bea725 | 2011-08-05 15:34:00 | [diff] [blame] | 42 | class BASE_EXPORT CommandLine { |
initial.commit | d7cae12 | 2008-07-26 21:49:38 | [diff] [blame] | 43 | public: |
Xiaohan Wang | 38e4ebb | 2022-01-19 06:57:43 | [diff] [blame] | 44 | #if BUILDFLAG(IS_WIN) |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 45 | // The native command line string type. |
Jan Wilken Dörrie | da77fd43 | 2019-10-24 21:40:34 | [diff] [blame] | 46 | using StringType = std::wstring; |
Xiaohan Wang | 38e4ebb | 2022-01-19 06:57:43 | [diff] [blame] | 47 | #elif BUILDFLAG(IS_POSIX) || BUILDFLAG(IS_FUCHSIA) |
thestig | fd08568 | 2016-07-15 02:50:16 | [diff] [blame] | 48 | using StringType = std::string; |
erg@google.com | a502bbe7 | 2011-01-07 18:06:45 | [diff] [blame] | 49 | #endif |
| 50 | |
thestig | fd08568 | 2016-07-15 02:50:16 | [diff] [blame] | 51 | using CharType = StringType::value_type; |
Takuto Ikuta | 9dcb26dd | 2023-12-19 08:02:08 | [diff] [blame] | 52 | using StringPieceType = std::basic_string_view<CharType>; |
thestig | fd08568 | 2016-07-15 02:50:16 | [diff] [blame] | 53 | using StringVector = std::vector<StringType>; |
Jeremy Roman | 863386d | 2017-10-31 19:25:38 | [diff] [blame] | 54 | using SwitchMap = std::map<std::string, StringType, std::less<>>; |
erg@google.com | a502bbe7 | 2011-01-07 18:06:45 | [diff] [blame] | 55 | |
mayuko aiba | fa08692f | 2024-01-12 03:58:38 | [diff] [blame] | 56 | // Returns CommandLine object constructed with switches and keys alone. |
| 57 | // NOTE: `argv` must NOT include the program path, and the switch arguments |
| 58 | // must start from the index 0. |
| 59 | static CommandLine FromArgvWithoutProgram(const StringVector& argv); |
| 60 | |
| 61 | #if BUILDFLAG(IS_WIN) |
| 62 | static CommandLine FromString(StringPieceType command_line); |
| 63 | #endif |
| 64 | |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 65 | // A constructor for CommandLines that only carry switches and arguments. |
mattm@chromium.org | 947446b | 2010-10-21 03:36:31 | [diff] [blame] | 66 | enum NoProgram { NO_PROGRAM }; |
| 67 | explicit CommandLine(NoProgram no_program); |
erg@google.com | a502bbe7 | 2011-01-07 18:06:45 | [diff] [blame] | 68 | |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 69 | // Construct a new command line with |program| as argv[0]. |
brettw@chromium.org | 2f3b1cc | 2014-03-17 23:07:15 | [diff] [blame] | 70 | explicit CommandLine(const FilePath& program); |
erg@google.com | a502bbe7 | 2011-01-07 18:06:45 | [diff] [blame] | 71 | |
msw@chromium.org | a40ca430 | 2011-05-14 01:10:24 | [diff] [blame] | 72 | // Construct a new command line from an argument list. |
| 73 | CommandLine(int argc, const CharType* const* argv); |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 74 | explicit CommandLine(const StringVector& argv); |
erg@google.com | a502bbe7 | 2011-01-07 18:06:45 | [diff] [blame] | 75 | |
Will Harris | ee9f0a2 | 2023-03-02 21:38:58 | [diff] [blame] | 76 | // Allow the copy constructor. A common pattern is to copy of the current |
| 77 | // process's command line and then add some flags to it. For example: |
| 78 | // CommandLine cl(*CommandLine::ForCurrentProcess()); |
| 79 | // cl.AppendSwitch(...); |
jackhou | 1bd9da9 | 2015-05-21 04:48:00 | [diff] [blame] | 80 | CommandLine(const CommandLine& other); |
| 81 | CommandLine& operator=(const CommandLine& other); |
| 82 | |
msw@chromium.org | acbeb3d | 2011-03-01 20:47:58 | [diff] [blame] | 83 | ~CommandLine(); |
| 84 | |
Xiaohan Wang | 38e4ebb | 2022-01-19 06:57:43 | [diff] [blame] | 85 | #if BUILDFLAG(IS_WIN) |
brettw@chromium.org | bf98a0e1 | 2013-09-25 23:36:00 | [diff] [blame] | 86 | // By default this class will treat command-line arguments beginning with |
| 87 | // slashes as switches on Windows, but not other platforms. |
| 88 | // |
| 89 | // If this behavior is inappropriate for your application, you can call this |
| 90 | // function BEFORE initializing the current process' global command line |
| 91 | // object and the behavior will be the same as Posix systems (only hyphens |
| 92 | // begin switches, everything else will be an arg). |
| 93 | static void set_slash_is_not_a_switch(); |
ananta | d936bc1 | 2016-06-22 21:40:31 | [diff] [blame] | 94 | |
| 95 | // Normally when the CommandLine singleton is initialized it gets the command |
| 96 | // line via the GetCommandLineW API and then uses the shell32 API |
| 97 | // CommandLineToArgvW to parse the command line and convert it back to |
| 98 | // argc and argv. Tests who don't want this dependency on shell32 and need |
| 99 | // to honor the arguments passed in should use this function. |
| 100 | static void InitUsingArgvForTesting(int argc, const char* const* argv); |
brettw@chromium.org | bf98a0e1 | 2013-09-25 23:36:00 | [diff] [blame] | 101 | #endif |
| 102 | |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 103 | // Initialize the current process CommandLine singleton. On Windows, ignores |
| 104 | // its arguments (we instead parse GetCommandLineW() directly) because we |
| 105 | // don't trust the CRT's parsing of the command line, but it still must be |
erikwright@chromium.org | 72e2e242 | 2012-02-27 18:38:12 | [diff] [blame] | 106 | // called to set up the command line. Returns false if initialization has |
| 107 | // already occurred, and true otherwise. Only the caller receiving a 'true' |
| 108 | // return value should take responsibility for calling Reset. |
| 109 | static bool Init(int argc, const char* const* argv); |
evan@chromium.org | bb97536 | 2009-01-21 01:00:22 | [diff] [blame] | 110 | |
evan@chromium.org | a2318cda | 2009-02-25 21:13:53 | [diff] [blame] | 111 | // Destroys the current process CommandLine singleton. This is necessary if |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 112 | // you want to reset the base library to its initial state (for example, in an |
evan@chromium.org | a2318cda | 2009-02-25 21:13:53 | [diff] [blame] | 113 | // outer library that needs to be able to terminate, and be re-initialized). |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 114 | // If Init is called only once, as in main(), Reset() is not necessary. |
thestig | fd08568 | 2016-07-15 02:50:16 | [diff] [blame] | 115 | // Do not call this in tests. Use base::test::ScopedCommandLine instead. |
evan@chromium.org | 02fb75ab | 2009-10-12 16:11:40 | [diff] [blame] | 116 | static void Reset(); |
evan@chromium.org | a2318cda | 2009-02-25 21:13:53 | [diff] [blame] | 117 | |
evan@chromium.org | bb97536 | 2009-01-21 01:00:22 | [diff] [blame] | 118 | // Get the singleton CommandLine representing the current process's |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 119 | // command line. Note: returned value is mutable, but not thread safe; |
evan@chromium.org | 0189bbd | 2009-10-12 22:50:39 | [diff] [blame] | 120 | // only mutate if you know what you're doing! |
erg@google.com | dcd869c | 2010-08-30 20:15:25 | [diff] [blame] | 121 | static CommandLine* ForCurrentProcess(); |
evanm@google.com | 1a48f31 | 2008-08-12 01:14:37 | [diff] [blame] | 122 | |
vollick@chromium.org | 2bf64a9 | 2013-07-11 23:10:40 | [diff] [blame] | 123 | // Returns true if the CommandLine has been initialized for the given process. |
| 124 | static bool InitializedForCurrentProcess(); |
| 125 | |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 126 | // Initialize from an argv vector. |
msw@chromium.org | a40ca430 | 2011-05-14 01:10:24 | [diff] [blame] | 127 | void InitFromArgv(int argc, const CharType* const* argv); |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 128 | void InitFromArgv(const StringVector& argv); |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 129 | |
msw@chromium.org | a40ca430 | 2011-05-14 01:10:24 | [diff] [blame] | 130 | // Constructs and returns the represented command line string. |
gab@chromium.org | 45f982e | 2012-10-29 21:31:31 | [diff] [blame] | 131 | // CAUTION! This should be avoided on POSIX because quoting behavior is |
| 132 | // unclear. |
Jesse McKenna | 036150c | 2020-07-17 21:11:17 | [diff] [blame] | 133 | // CAUTION! If writing a command line to the Windows registry, use |
| 134 | // GetCommandLineStringForShell() instead. |
| 135 | StringType GetCommandLineString() const; |
mgiuca | c974d51 | 2014-10-01 09:24:51 | [diff] [blame] | 136 | |
Xiaohan Wang | 38e4ebb | 2022-01-19 06:57:43 | [diff] [blame] | 137 | #if BUILDFLAG(IS_WIN) |
S. Ganesh | f2c7cb73 | 2022-12-16 22:54:42 | [diff] [blame] | 138 | // Quotes and escapes `arg` if necessary so that it will be interpreted as a |
| 139 | // single command-line parameter according to the following rules in line with |
| 140 | // `::CommandLineToArgvW` and C++ `main`: |
| 141 | // * Returns `arg` unchanged if `arg` does not include any characters that may |
| 142 | // need encoding, which is spaces, tabs, backslashes, and double-quotes. |
| 143 | // * Otherwise, double-quotes `arg` and in addition: |
| 144 | // * Escapes any double-quotes in `arg` with backslashes. |
| 145 | // * Escapes backslashes in `arg` if: |
| 146 | // * `arg` ends with backslashes , or |
| 147 | // * the backslashes end in a pre-existing double quote. |
| 148 | // |
| 149 | // https://learn.microsoft.com/en-us/search/?terms=CommandLineToArgvW and |
| 150 | // http://msdn.microsoft.com/en-us/library/17w5ykft.aspx#parsing-c-command-line-arguments. |
| 151 | static std::wstring QuoteForCommandLineToArgvW(const std::wstring& arg); |
| 152 | |
Jesse McKenna | 036150c | 2020-07-17 21:11:17 | [diff] [blame] | 153 | // Returns the command-line string in the proper format for the Windows shell, |
| 154 | // ending with the argument placeholder "--single-argument %1". The single- |
| 155 | // argument switch prevents unexpected parsing of arguments from other |
| 156 | // software that cannot be trusted to escape double quotes when substituting |
Jesse McKenna | 185ceda2 | 2021-03-10 06:47:55 | [diff] [blame] | 157 | // into a placeholder (e.g., "%1" insert sequences populated by the Windows |
Jesse McKenna | 036150c | 2020-07-17 21:11:17 | [diff] [blame] | 158 | // shell). |
| 159 | // NOTE: this must be used to generate the command-line string for the shell |
| 160 | // even if this command line was parsed from a string with the proper syntax, |
| 161 | // because the --single-argument switch is not preserved during parsing. |
| 162 | StringType GetCommandLineStringForShell() const; |
Jesse McKenna | 185ceda2 | 2021-03-10 06:47:55 | [diff] [blame] | 163 | |
| 164 | // Returns the represented command-line string. Allows the use of unsafe |
| 165 | // Windows insert sequences like "%1". Only use this method if |
| 166 | // GetCommandLineStringForShell() is not adequate AND the processor inserting |
| 167 | // the arguments is known to do so securely (i.e., is not the Windows shell). |
| 168 | // If in doubt, do not use. |
| 169 | StringType GetCommandLineStringWithUnsafeInsertSequences() const; |
mgiuca | c974d51 | 2014-10-01 09:24:51 | [diff] [blame] | 170 | #endif |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 171 | |
gab@chromium.org | 45f982e | 2012-10-29 21:31:31 | [diff] [blame] | 172 | // Constructs and returns the represented arguments string. |
| 173 | // CAUTION! This should be avoided on POSIX because quoting behavior is |
| 174 | // unclear. |
Jesse McKenna | 036150c | 2020-07-17 21:11:17 | [diff] [blame] | 175 | StringType GetArgumentsString() const; |
gab@chromium.org | 45f982e | 2012-10-29 21:31:31 | [diff] [blame] | 176 | |
estade@chromium.org | 10e42bf | 2008-10-15 21:59:08 | [diff] [blame] | 177 | // Returns the original command line string as a vector of strings. |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 178 | const StringVector& argv() const { return argv_; } |
estade@chromium.org | 10e42bf | 2008-10-15 21:59:08 | [diff] [blame] | 179 | |
msw@chromium.org | a40ca430 | 2011-05-14 01:10:24 | [diff] [blame] | 180 | // Get and Set the program part of the command line string (the first item). |
brettw@chromium.org | 2f3b1cc | 2014-03-17 23:07:15 | [diff] [blame] | 181 | FilePath GetProgram() const; |
| 182 | void SetProgram(const FilePath& program); |
evan@chromium.org | 0189bbd | 2009-10-12 22:50:39 | [diff] [blame] | 183 | |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 184 | // Returns true if this command line contains the given switch. |
jackhou | 1bd9da9 | 2015-05-21 04:48:00 | [diff] [blame] | 185 | // Switch names must be lowercase. |
| 186 | // The second override provides an optimized version to avoid inlining codegen |
| 187 | // at every callsite to find the length of the constant and construct a |
| 188 | // StringPiece. |
Wez | 65018d0 | 2021-05-20 15:47:45 | [diff] [blame] | 189 | bool HasSwitch(StringPiece switch_string) const; |
tapted | 009a1dc8 | 2015-03-30 03:57:10 | [diff] [blame] | 190 | bool HasSwitch(const char switch_constant[]) const; |
initial.commit | d7cae12 | 2008-07-26 21:49:38 | [diff] [blame] | 191 | |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 192 | // Returns the value associated with the given switch. If the switch has no |
| 193 | // value or isn't present, this method returns the empty string. |
jackhou | 1bd9da9 | 2015-05-21 04:48:00 | [diff] [blame] | 194 | // Switch names must be lowercase. |
Wez | 65018d0 | 2021-05-20 15:47:45 | [diff] [blame] | 195 | std::string GetSwitchValueASCII(StringPiece switch_string) const; |
| 196 | FilePath GetSwitchValuePath(StringPiece switch_string) const; |
| 197 | StringType GetSwitchValueNative(StringPiece switch_string) const; |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 198 | |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 199 | // Get a copy of all switches, along with their values. |
| 200 | const SwitchMap& GetSwitches() const { return switches_; } |
| 201 | |
| 202 | // Append a switch [with optional value] to the command line. |
msw@chromium.org | a40ca430 | 2011-05-14 01:10:24 | [diff] [blame] | 203 | // Note: Switches will precede arguments regardless of appending order. |
Wez | 65018d0 | 2021-05-20 15:47:45 | [diff] [blame] | 204 | void AppendSwitch(StringPiece switch_string); |
| 205 | void AppendSwitchPath(StringPiece switch_string, const FilePath& path); |
| 206 | void AppendSwitchNative(StringPiece switch_string, StringPieceType value); |
| 207 | void AppendSwitchASCII(StringPiece switch_string, StringPiece value); |
evan@chromium.org | 4f08c83f | 2010-07-29 23:02:34 | [diff] [blame] | 208 | |
Pavol Marko | bf16b81 | 2019-06-14 00:53:12 | [diff] [blame] | 209 | // Removes the switch that matches |switch_key_without_prefix|, regardless of |
| 210 | // prefix and value. If no such switch is present, this has no effect. |
| 211 | void RemoveSwitch(const base::StringPiece switch_key_without_prefix); |
Avi Drissman | 1aa6cb9 | 2019-01-23 15:58:38 | [diff] [blame] | 212 | |
Lei Zhang | 233b3cd | 2023-07-19 02:25:10 | [diff] [blame] | 213 | // Copies a set of switches (and any values) from another command line. |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 214 | // Commonly used when launching a subprocess. |
Lei Zhang | 233b3cd | 2023-07-19 02:25:10 | [diff] [blame] | 215 | // If an entry in `switches` does not exist in `source`, then it is ignored. |
Lei Zhang | 24dcd3e | 2023-07-07 20:08:36 | [diff] [blame] | 216 | void CopySwitchesFrom(const CommandLine& source, |
| 217 | span<const char* const> switches); |
| 218 | |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 219 | // Get the remaining arguments to the command. |
msw@chromium.org | 75f1c78 | 2011-07-13 23:41:22 | [diff] [blame] | 220 | StringVector GetArgs() const; |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 221 | |
| 222 | // Append an argument to the command line. Note that the argument is quoted |
| 223 | // properly such that it is interpreted as one argument to the target command. |
| 224 | // AppendArg is primarily for ASCII; non-ASCII input is interpreted as UTF-8. |
msw@chromium.org | a40ca430 | 2011-05-14 01:10:24 | [diff] [blame] | 225 | // Note: Switches will precede arguments regardless of appending order. |
Wez | 65018d0 | 2021-05-20 15:47:45 | [diff] [blame] | 226 | void AppendArg(StringPiece value); |
brettw@chromium.org | 2f3b1cc | 2014-03-17 23:07:15 | [diff] [blame] | 227 | void AppendArgPath(const FilePath& value); |
Wez | 65018d0 | 2021-05-20 15:47:45 | [diff] [blame] | 228 | void AppendArgNative(StringPieceType value); |
evan@chromium.org | bb97536 | 2009-01-21 01:00:22 | [diff] [blame] | 229 | |
msw@chromium.org | a40ca430 | 2011-05-14 01:10:24 | [diff] [blame] | 230 | // Append the switches and arguments from another command line to this one. |
mayuko aiba | fa08692f | 2024-01-12 03:58:38 | [diff] [blame] | 231 | // If `include_program` is true, program will be overwritten by other's. |
msw@chromium.org | a40ca430 | 2011-05-14 01:10:24 | [diff] [blame] | 232 | void AppendArguments(const CommandLine& other, bool include_program); |
initial.commit | d7cae12 | 2008-07-26 21:49:38 | [diff] [blame] | 233 | |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 234 | // Insert a command before the current command. |
Mostyn Bramley-Moore | d0ecd6a | 2017-12-06 19:13:21 | [diff] [blame] | 235 | // Common for debuggers, like "gdb --args". |
Wez | 65018d0 | 2021-05-20 15:47:45 | [diff] [blame] | 236 | void PrependWrapper(StringPieceType wrapper); |
agl@chromium.org | 052f1d48 | 2009-02-10 00:52:14 | [diff] [blame] | 237 | |
Xiaohan Wang | 38e4ebb | 2022-01-19 06:57:43 | [diff] [blame] | 238 | #if BUILDFLAG(IS_WIN) |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 239 | // Initialize by parsing the given command line string. |
| 240 | // The program name is assumed to be the first item in the string. |
Jan Wilken Dörrie | da77fd43 | 2019-10-24 21:40:34 | [diff] [blame] | 241 | void ParseFromString(StringPieceType command_line); |
David Bienvenu | edbb4ce | 2022-09-28 16:02:16 | [diff] [blame] | 242 | |
| 243 | // Returns true if the command line had the --single-argument switch, and |
| 244 | // thus likely came from a Windows shell registration. This is only set if the |
| 245 | // command line is parsed, and is not changed after it is parsed. |
| 246 | bool HasSingleArgumentSwitch() const { return has_single_argument_switch_; } |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 247 | #endif |
evan@chromium.org | 4f08c83f | 2010-07-29 23:02:34 | [diff] [blame] | 248 | |
Will Harris | ee9f0a2 | 2023-03-02 21:38:58 | [diff] [blame] | 249 | // Detaches this object from the current sequence in preparation for a move to |
| 250 | // a different sequence. |
| 251 | void DetachFromCurrentSequence(); |
| 252 | |
Greg Guterman | fe16560d | 2021-10-07 19:58:15 | [diff] [blame] | 253 | // Sets a delegate that's called when we encounter a duplicate switch |
| 254 | static void SetDuplicateSwitchHandler( |
| 255 | std::unique_ptr<DuplicateSwitchHandler>); |
| 256 | |
initial.commit | d7cae12 | 2008-07-26 21:49:38 | [diff] [blame] | 257 | private: |
Will Harris | ee9f0a2 | 2023-03-02 21:38:58 | [diff] [blame] | 258 | #if BUILDFLAG(ENABLE_COMMANDLINE_SEQUENCE_CHECKS) |
| 259 | // A helper class that encapsulates a SEQUENCE_CHECKER but allows copy. |
| 260 | // Copying this class will detach the sequence checker from the owning object. |
| 261 | class InstanceBoundSequenceChecker { |
| 262 | public: |
| 263 | InstanceBoundSequenceChecker() = default; |
| 264 | |
| 265 | InstanceBoundSequenceChecker(const InstanceBoundSequenceChecker& other) {} |
| 266 | |
| 267 | InstanceBoundSequenceChecker& operator=( |
| 268 | const InstanceBoundSequenceChecker& other) { |
| 269 | return *this; |
| 270 | } |
| 271 | |
| 272 | // Disallow move. |
| 273 | InstanceBoundSequenceChecker(InstanceBoundSequenceChecker&&) = delete; |
| 274 | InstanceBoundSequenceChecker& operator=(InstanceBoundSequenceChecker&&) = |
| 275 | delete; |
| 276 | |
| 277 | void Detach() { DETACH_FROM_SEQUENCE(sequence_checker_); } |
| 278 | void Check() { DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); } |
| 279 | |
| 280 | private: |
| 281 | SEQUENCE_CHECKER(sequence_checker_); |
| 282 | }; |
| 283 | #endif // BUILDFLAG(ENABLE_COMMANDLINE_SEQUENCE_CHECKS) |
| 284 | |
msw@chromium.org | a40ca430 | 2011-05-14 01:10:24 | [diff] [blame] | 285 | // Disallow default constructor; a program name must be explicitly specified. |
Chris Watkins | 091d629 | 2017-12-13 04:25:58 | [diff] [blame] | 286 | CommandLine() = delete; |
sky@google.com | d4515eb | 2009-01-30 00:40:43 | [diff] [blame] | 287 | |
Jesse McKenna | 036150c | 2020-07-17 21:11:17 | [diff] [blame] | 288 | // Append switches and arguments, keeping switches before arguments. |
mayuko aiba | fa08692f | 2024-01-12 03:58:38 | [diff] [blame] | 289 | // NOTE: `argv` should not include the "program" element. |
| 290 | void AppendSwitchesAndArguments(span<const StringType> argv); |
mgiuca | c974d51 | 2014-10-01 09:24:51 | [diff] [blame] | 291 | |
Jesse McKenna | 185ceda2 | 2021-03-10 06:47:55 | [diff] [blame] | 292 | // Internal version of GetArgumentsString to support allowing unsafe insert |
| 293 | // sequences in rare cases (see |
| 294 | // GetCommandLineStringWithUnsafeInsertSequences). |
| 295 | StringType GetArgumentsStringInternal( |
| 296 | bool allow_unsafe_insert_sequences) const; |
| 297 | |
Xiaohan Wang | 38e4ebb | 2022-01-19 06:57:43 | [diff] [blame] | 298 | #if BUILDFLAG(IS_WIN) |
Jesse McKenna | 036150c | 2020-07-17 21:11:17 | [diff] [blame] | 299 | // Initializes by parsing |raw_command_line_string_|, treating everything |
| 300 | // after |single_arg_switch_string| + <a single character> as the command |
| 301 | // line's single argument, and dropping any arguments previously parsed. The |
| 302 | // command line must contain |single_arg_switch_string|, and the argument, if |
| 303 | // present, must be separated from |single_arg_switch_string| by one |
| 304 | // character. |
| 305 | // NOTE: the single-argument switch is not preserved after parsing; |
| 306 | // GetCommandLineStringForShell() must be used to reproduce the original |
| 307 | // command-line string with single-argument switch. |
| 308 | void ParseAsSingleArgument(const StringType& single_arg_switch_string); |
| 309 | |
| 310 | // The string returned by GetCommandLineW(), to be parsed via |
| 311 | // ParseFromString(). Empty if this command line was not parsed from a string, |
| 312 | // or if ParseFromString() has finished executing. |
| 313 | StringPieceType raw_command_line_string_; |
David Bienvenu | edbb4ce | 2022-09-28 16:02:16 | [diff] [blame] | 314 | |
| 315 | // Set to true if the command line had --single-argument when initially |
| 316 | // parsed. It does not change if the command line mutates after initial |
| 317 | // parsing. |
| 318 | bool has_single_argument_switch_ = false; |
Jesse McKenna | 036150c | 2020-07-17 21:11:17 | [diff] [blame] | 319 | #endif |
mgiuca | c974d51 | 2014-10-01 09:24:51 | [diff] [blame] | 320 | |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 321 | // The singleton CommandLine representing the current process's command line. |
evan@chromium.org | bb97536 | 2009-01-21 01:00:22 | [diff] [blame] | 322 | static CommandLine* current_process_commandline_; |
initial.commit | d7cae12 | 2008-07-26 21:49:38 | [diff] [blame] | 323 | |
msw@chromium.org | a40ca430 | 2011-05-14 01:10:24 | [diff] [blame] | 324 | // The argv array: { program, [(--|-|/)switch[=value]]*, [--], [argument]* } |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 325 | StringVector argv_; |
evan@chromium.org | bb97536 | 2009-01-21 01:00:22 | [diff] [blame] | 326 | |
msw@chromium.org | 06cc083a | 2011-03-01 02:28:42 | [diff] [blame] | 327 | // Parsed-out switch keys and values. |
kinuko@chromium.org | f6c3483 | 2010-05-24 10:39:59 | [diff] [blame] | 328 | SwitchMap switches_; |
evan@chromium.org | bb97536 | 2009-01-21 01:00:22 | [diff] [blame] | 329 | |
msw@chromium.org | a40ca430 | 2011-05-14 01:10:24 | [diff] [blame] | 330 | // The index after the program and switches, any arguments start here. |
Peter Kasting | de85e74 | 2022-06-01 17:41:54 | [diff] [blame] | 331 | ptrdiff_t begin_args_; |
Will Harris | ee9f0a2 | 2023-03-02 21:38:58 | [diff] [blame] | 332 | |
| 333 | #if BUILDFLAG(ENABLE_COMMANDLINE_SEQUENCE_CHECKS) |
| 334 | InstanceBoundSequenceChecker sequence_checker_; |
| 335 | #endif |
initial.commit | d7cae12 | 2008-07-26 21:49:38 | [diff] [blame] | 336 | }; |
| 337 | |
Greg Guterman | fe16560d | 2021-10-07 19:58:15 | [diff] [blame] | 338 | class BASE_EXPORT DuplicateSwitchHandler { |
| 339 | public: |
| 340 | // out_value contains the existing value of the switch |
| 341 | virtual void ResolveDuplicate(base::StringPiece key, |
| 342 | CommandLine::StringPieceType new_value, |
| 343 | CommandLine::StringType& out_value) = 0; |
| 344 | virtual ~DuplicateSwitchHandler() = default; |
| 345 | }; |
| 346 | |
brettw@chromium.org | 2f3b1cc | 2014-03-17 23:07:15 | [diff] [blame] | 347 | } // namespace base |
| 348 | |
deanm@chromium.org | 02c8796 | 2008-10-06 10:25:35 | [diff] [blame] | 349 | #endif // BASE_COMMAND_LINE_H_ |