command_line.h 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250
  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.
  4. // This class works with command lines: building and parsing.
  5. // Arguments with prefixes ('--', '-', and on Windows, '/') are switches.
  6. // Switches will precede all other arguments without switch prefixes.
  7. // Switches can optionally have values, delimited by '=', e.g., "-switch=value".
  8. // If a switch is specified multiple times, only the last value is used.
  9. // An argument of "--" will terminate switch parsing during initialization,
  10. // interpreting subsequent tokens as non-switch arguments, regardless of prefix.
  11. // There is a singleton read-only CommandLine that represents the command line
  12. // that the current process was started with. It must be initialized in main().
  13. #ifndef BASE_COMMAND_LINE_H_
  14. #define BASE_COMMAND_LINE_H_
  15. #include <stddef.h>
  16. #include <map>
  17. #include <string>
  18. #include <vector>
  19. #include "base/base_export.h"
  20. #include "base/strings/string16.h"
  21. #include "base/strings/string_piece.h"
  22. #include "build/build_config.h"
  23. namespace base {
  24. class FilePath;
  25. class BASE_EXPORT CommandLine {
  26. public:
  27. #if defined(OS_WIN)
  28. // The native command line string type.
  29. using StringType = std::wstring;
  30. #elif defined(OS_POSIX) || defined(OS_FUCHSIA)
  31. using StringType = std::string;
  32. #endif
  33. using StringPieceType = base::BasicStringPiece<StringType>;
  34. using CharType = StringType::value_type;
  35. using StringVector = std::vector<StringType>;
  36. using SwitchMap = std::map<std::string, StringType, std::less<>>;
  37. // A constructor for CommandLines that only carry switches and arguments.
  38. enum NoProgram { NO_PROGRAM };
  39. explicit CommandLine(NoProgram no_program);
  40. // Construct a new command line with |program| as argv[0].
  41. explicit CommandLine(const FilePath& program);
  42. // Construct a new command line from an argument list.
  43. CommandLine(int argc, const CharType* const* argv);
  44. explicit CommandLine(const StringVector& argv);
  45. // Override copy and assign to ensure |switches_by_stringpiece_| is valid.
  46. CommandLine(const CommandLine& other);
  47. CommandLine& operator=(const CommandLine& other);
  48. ~CommandLine();
  49. #if defined(OS_WIN)
  50. // By default this class will treat command-line arguments beginning with
  51. // slashes as switches on Windows, but not other platforms.
  52. //
  53. // If this behavior is inappropriate for your application, you can call this
  54. // function BEFORE initializing the current process' global command line
  55. // object and the behavior will be the same as Posix systems (only hyphens
  56. // begin switches, everything else will be an arg).
  57. static void set_slash_is_not_a_switch();
  58. // Normally when the CommandLine singleton is initialized it gets the command
  59. // line via the GetCommandLineW API and then uses the shell32 API
  60. // CommandLineToArgvW to parse the command line and convert it back to
  61. // argc and argv. Tests who don't want this dependency on shell32 and need
  62. // to honor the arguments passed in should use this function.
  63. static void InitUsingArgvForTesting(int argc, const char* const* argv);
  64. #endif
  65. // Initialize the current process CommandLine singleton. On Windows, ignores
  66. // its arguments (we instead parse GetCommandLineW() directly) because we
  67. // don't trust the CRT's parsing of the command line, but it still must be
  68. // called to set up the command line. Returns false if initialization has
  69. // already occurred, and true otherwise. Only the caller receiving a 'true'
  70. // return value should take responsibility for calling Reset.
  71. static bool Init(int argc, const char* const* argv);
  72. // Destroys the current process CommandLine singleton. This is necessary if
  73. // you want to reset the base library to its initial state (for example, in an
  74. // outer library that needs to be able to terminate, and be re-initialized).
  75. // If Init is called only once, as in main(), Reset() is not necessary.
  76. // Do not call this in tests. Use base::test::ScopedCommandLine instead.
  77. static void Reset();
  78. // Get the singleton CommandLine representing the current process's
  79. // command line. Note: returned value is mutable, but not thread safe;
  80. // only mutate if you know what you're doing!
  81. static CommandLine* ForCurrentProcess();
  82. // Returns true if the CommandLine has been initialized for the given process.
  83. static bool InitializedForCurrentProcess();
  84. #if defined(OS_WIN)
  85. static CommandLine FromString(StringPieceType command_line);
  86. #endif
  87. // Initialize from an argv vector.
  88. void InitFromArgv(int argc, const CharType* const* argv);
  89. void InitFromArgv(const StringVector& argv);
  90. // Constructs and returns the represented command line string.
  91. // CAUTION! This should be avoided on POSIX because quoting behavior is
  92. // unclear.
  93. // CAUTION! If writing a command line to the Windows registry, use
  94. // GetCommandLineStringForShell() instead.
  95. StringType GetCommandLineString() const;
  96. #if defined(OS_WIN)
  97. // Returns the command-line string in the proper format for the Windows shell,
  98. // ending with the argument placeholder "--single-argument=%1". The single-
  99. // argument switch prevents unexpected parsing of arguments from other
  100. // software that cannot be trusted to escape double quotes when substituting
  101. // into a placeholder (e.g., "%1" placeholders populated by the Windows
  102. // shell).
  103. // NOTE: this must be used to generate the command-line string for the shell
  104. // even if this command line was parsed from a string with the proper syntax,
  105. // because the --single-argument switch is not preserved during parsing.
  106. StringType GetCommandLineStringForShell() const;
  107. #endif
  108. // Constructs and returns the represented arguments string.
  109. // CAUTION! This should be avoided on POSIX because quoting behavior is
  110. // unclear.
  111. StringType GetArgumentsString() const;
  112. // Returns the original command line string as a vector of strings.
  113. const StringVector& argv() const { return argv_; }
  114. // Get and Set the program part of the command line string (the first item).
  115. FilePath GetProgram() const;
  116. void SetProgram(const FilePath& program);
  117. // Returns true if this command line contains the given switch.
  118. // Switch names must be lowercase.
  119. // The second override provides an optimized version to avoid inlining codegen
  120. // at every callsite to find the length of the constant and construct a
  121. // StringPiece.
  122. bool HasSwitch(const StringPiece& switch_string) const;
  123. bool HasSwitch(const char switch_constant[]) const;
  124. // Returns the value associated with the given switch. If the switch has no
  125. // value or isn't present, this method returns the empty string.
  126. // Switch names must be lowercase.
  127. std::string GetSwitchValueASCII(const StringPiece& switch_string) const;
  128. FilePath GetSwitchValuePath(const StringPiece& switch_string) const;
  129. StringType GetSwitchValueNative(const StringPiece& switch_string) const;
  130. // Get a copy of all switches, along with their values.
  131. const SwitchMap& GetSwitches() const { return switches_; }
  132. // Append a switch [with optional value] to the command line.
  133. // Note: Switches will precede arguments regardless of appending order.
  134. void AppendSwitch(const std::string& switch_string);
  135. void AppendSwitchPath(const std::string& switch_string,
  136. const FilePath& path);
  137. void AppendSwitchNative(const std::string& switch_string,
  138. const StringType& value);
  139. void AppendSwitchASCII(const std::string& switch_string,
  140. const std::string& value);
  141. // Removes the switch that matches |switch_key_without_prefix|, regardless of
  142. // prefix and value. If no such switch is present, this has no effect.
  143. void RemoveSwitch(const base::StringPiece switch_key_without_prefix);
  144. // Copy a set of switches (and any values) from another command line.
  145. // Commonly used when launching a subprocess.
  146. void CopySwitchesFrom(const CommandLine& source,
  147. const char* const switches[],
  148. size_t count);
  149. // Get the remaining arguments to the command.
  150. StringVector GetArgs() const;
  151. // Append an argument to the command line. Note that the argument is quoted
  152. // properly such that it is interpreted as one argument to the target command.
  153. // AppendArg is primarily for ASCII; non-ASCII input is interpreted as UTF-8.
  154. // Note: Switches will precede arguments regardless of appending order.
  155. void AppendArg(const std::string& value);
  156. void AppendArgPath(const FilePath& value);
  157. void AppendArgNative(const StringType& value);
  158. // Append the switches and arguments from another command line to this one.
  159. // If |include_program| is true, include |other|'s program as well.
  160. void AppendArguments(const CommandLine& other, bool include_program);
  161. // Insert a command before the current command.
  162. // Common for debuggers, like "gdb --args".
  163. void PrependWrapper(const StringType& wrapper);
  164. #if defined(OS_WIN)
  165. // Initialize by parsing the given command line string.
  166. // The program name is assumed to be the first item in the string.
  167. void ParseFromString(StringPieceType command_line);
  168. #endif
  169. private:
  170. // Disallow default constructor; a program name must be explicitly specified.
  171. CommandLine() = delete;
  172. // Allow the copy constructor. A common pattern is to copy of the current
  173. // process's command line and then add some flags to it. For example:
  174. // CommandLine cl(*CommandLine::ForCurrentProcess());
  175. // cl.AppendSwitch(...);
  176. // Append switches and arguments, keeping switches before arguments.
  177. void AppendSwitchesAndArguments(const StringVector& argv);
  178. #if defined(OS_WIN)
  179. // Initializes by parsing |raw_command_line_string_|, treating everything
  180. // after |single_arg_switch_string| + "=" as the command line's single
  181. // argument, and dropping any arguments previously parsed. The command line
  182. // must contain |single_arg_switch_string| followed by "=".
  183. // NOTE: the single-argument switch is not preserved after parsing;
  184. // GetCommandLineStringForShell() must be used to reproduce the original
  185. // command-line string with single-argument switch.
  186. void ParseAsSingleArgument(const StringType& single_arg_switch_string);
  187. // The string returned by GetCommandLineW(), to be parsed via
  188. // ParseFromString(). Empty if this command line was not parsed from a string,
  189. // or if ParseFromString() has finished executing.
  190. StringPieceType raw_command_line_string_;
  191. #endif
  192. // The singleton CommandLine representing the current process's command line.
  193. static CommandLine* current_process_commandline_;
  194. // The argv array: { program, [(--|-|/)switch[=value]]*, [--], [argument]* }
  195. StringVector argv_;
  196. // Parsed-out switch keys and values.
  197. SwitchMap switches_;
  198. // The index after the program and switches, any arguments start here.
  199. size_t begin_args_;
  200. };
  201. } // namespace base
  202. #endif // BASE_COMMAND_LINE_H_