| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 // This class works with command lines: building and parsing. | 5 // This class works with command lines: building and parsing. |
| 6 // Arguments with prefixes ('--', '-', and on Windows, '/') are switches. | 6 // Arguments with prefixes ('--', '-', and on Windows, '/') are switches. |
| 7 // Switches will precede all other arguments without switch prefixes. | 7 // Switches will precede all other arguments without switch prefixes. |
| 8 // Switches can optionally have values, delimited by '=', e.g., "-switch=value". | 8 // Switches can optionally have values, delimited by '=', e.g., "-switch=value". |
| 9 // An argument of "--" will terminate switch parsing during initialization, | 9 // An argument of "--" will terminate switch parsing during initialization, |
| 10 // interpreting subsequent tokens as non-switch arguments, regardless of prefix. | 10 // interpreting subsequent tokens as non-switch arguments, regardless of prefix. |
| 11 | 11 |
| 12 // There is a singleton read-only CommandLine that represents the command line | 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(). | 13 // that the current process was started with. It must be initialized in main(). |
| 14 | 14 |
| 15 #ifndef BASE_COMMAND_LINE_H_ | 15 #ifndef BASE_COMMAND_LINE_H_ |
| 16 #define BASE_COMMAND_LINE_H_ | 16 #define BASE_COMMAND_LINE_H_ |
| 17 | 17 |
| 18 #include <stddef.h> | 18 #include <stddef.h> |
| 19 #include <stdint.h> |
| 19 #include <map> | 20 #include <map> |
| 20 #include <string> | 21 #include <string> |
| 21 #include <vector> | 22 #include <vector> |
| 22 | 23 |
| 23 #include "base/base_export.h" | 24 #include "base/base_export.h" |
| 24 #include "build/build_config.h" | 25 #include "build/build_config.h" |
| 25 | 26 |
| 26 namespace base { | 27 namespace base { |
| 27 class FilePath; | 28 class FilePath; |
| 28 } | 29 } |
| 29 | 30 |
| 30 class BASE_EXPORT CommandLine { | 31 class BASE_EXPORT CommandLine { |
| 31 public: | 32 public: |
| 32 #if defined(OS_WIN) | 33 #if defined(OS_WIN) |
| 33 // The native command line string type. | 34 // The native command line string type. |
| 34 typedef std::wstring StringType; | 35 typedef std::wstring StringType; |
| 35 #elif defined(OS_POSIX) | 36 #elif defined(OS_POSIX) |
| 36 typedef std::string StringType; | 37 typedef std::string StringType; |
| 37 #endif | 38 #endif |
| 38 | 39 |
| 39 typedef StringType::value_type CharType; | 40 typedef StringType::value_type CharType; |
| 40 typedef std::vector<StringType> StringVector; | 41 typedef std::vector<StringType> StringVector; |
| 41 typedef std::map<std::string, StringType> SwitchMap; | 42 typedef std::map<std::string, StringType> SwitchMap; |
| 42 | 43 |
| 44 struct ExplicitStdString { |
| 45 ExplicitStdString(const std::string& str) : s(str) {} |
| 46 const std::string& s; |
| 47 }; |
| 48 |
| 43 // A constructor for CommandLines that only carry switches and arguments. | 49 // A constructor for CommandLines that only carry switches and arguments. |
| 44 enum NoProgram { NO_PROGRAM }; | 50 enum NoProgram { NO_PROGRAM }; |
| 45 explicit CommandLine(NoProgram no_program); | 51 explicit CommandLine(NoProgram no_program); |
| 46 | 52 |
| 47 // Construct a new command line with |program| as argv[0]. | 53 // Construct a new command line with |program| as argv[0]. |
| 48 explicit CommandLine(const base::FilePath& program); | 54 explicit CommandLine(const base::FilePath& program); |
| 49 | 55 |
| 50 // Construct a new command line from an argument list. | 56 // Construct a new command line from an argument list. |
| 51 CommandLine(int argc, const CharType* const* argv); | 57 CommandLine(int argc, const CharType* const* argv); |
| 52 explicit CommandLine(const StringVector& argv); | 58 explicit CommandLine(const StringVector& argv); |
| (...skipping 42 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 95 | 101 |
| 96 // Returns the original command line string as a vector of strings. | 102 // Returns the original command line string as a vector of strings. |
| 97 const StringVector& argv() const { return argv_; } | 103 const StringVector& argv() const { return argv_; } |
| 98 | 104 |
| 99 // Get and Set the program part of the command line string (the first item). | 105 // Get and Set the program part of the command line string (the first item). |
| 100 base::FilePath GetProgram() const; | 106 base::FilePath GetProgram() const; |
| 101 void SetProgram(const base::FilePath& program); | 107 void SetProgram(const base::FilePath& program); |
| 102 | 108 |
| 103 // Returns true if this command line contains the given switch. | 109 // Returns true if this command line contains the given switch. |
| 104 // (Switch names are case-insensitive). | 110 // (Switch names are case-insensitive). |
| 105 bool HasSwitch(const std::string& switch_string) const; | 111 bool HasStringSwitch(const std::string& switch_string) const; |
| 112 bool HasSwitch(const ExplicitStdString& switch_string) const { |
| 113 return HasStringSwitch(switch_string.s); |
| 114 } |
| 115 bool HasSwitch(const char* switch_string, size_t string_length) const; |
| 116 |
| 117 template <size_t N> |
| 118 bool HasSwitch(const char (&constant_switch_string)[N]) const { |
| 119 // Note that N includes a null terminator, so this will generate a compile |
| 120 // error if called with an empty string constant. |
| 121 if (N <= 65 && |
| 122 (switch_lengths_mask_ & (static_cast<uint64_t>(1) << (N - 2))) == 0) { |
| 123 return false; |
| 124 } |
| 125 |
| 126 return HasSwitch(constant_switch_string, N - 1); |
| 127 } |
| 106 | 128 |
| 107 // Returns the value associated with the given switch. If the switch has no | 129 // Returns the value associated with the given switch. If the switch has no |
| 108 // value or isn't present, this method returns the empty string. | 130 // value or isn't present, this method returns the empty string. |
| 109 std::string GetSwitchValueASCII(const std::string& switch_string) const; | 131 std::string GetSwitchValueASCII(const std::string& switch_string) const; |
| 110 base::FilePath GetSwitchValuePath(const std::string& switch_string) const; | 132 base::FilePath GetSwitchValuePath(const std::string& switch_string) const; |
| 111 StringType GetSwitchValueNative(const std::string& switch_string) const; | 133 StringType GetSwitchValueNative(const std::string& switch_string) const; |
| 112 | 134 |
| 113 // Get a copy of all switches, along with their values. | 135 // Get a copy of all switches, along with their values. |
| 114 const SwitchMap& GetSwitches() const { return switches_; } | 136 const SwitchMap& GetSwitches() const { return switches_; } |
| 115 | 137 |
| (...skipping 50 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 166 static CommandLine* current_process_commandline_; | 188 static CommandLine* current_process_commandline_; |
| 167 | 189 |
| 168 // The argv array: { program, [(--|-|/)switch[=value]]*, [--], [argument]* } | 190 // The argv array: { program, [(--|-|/)switch[=value]]*, [--], [argument]* } |
| 169 StringVector argv_; | 191 StringVector argv_; |
| 170 | 192 |
| 171 // Parsed-out switch keys and values. | 193 // Parsed-out switch keys and values. |
| 172 SwitchMap switches_; | 194 SwitchMap switches_; |
| 173 | 195 |
| 174 // The index after the program and switches, any arguments start here. | 196 // The index after the program and switches, any arguments start here. |
| 175 size_t begin_args_; | 197 size_t begin_args_; |
| 198 |
| 199 // A bitset, with a 1 at bit postion |i| indicating that the command line |
| 200 // includes a switch of string length |i| + 1 ascii characters. |
| 201 uint64_t switch_lengths_mask_; |
| 176 }; | 202 }; |
| 177 | 203 |
| 178 #endif // BASE_COMMAND_LINE_H_ | 204 #endif // BASE_COMMAND_LINE_H_ |
| OLD | NEW |