base/command_line.cc

   1 // Copyright (c) 2011 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
   5 #include "base/command_line.h"
   6
   7 #include <algorithm>
   8 #include <ostream>
   9
  10 #include "base/basictypes.h"
  11 #include "base/file_path.h"
  12 #include "base/logging.h"
  13 #include "base/string_split.h"
  14 #include "base/string_util.h"
  15 #include "base/utf_string_conversions.h"
  16 #include "build/build_config.h"
  17
  18 #if defined(OS_WIN)
  19 #include <windows.h>
  20 #include <shellapi.h>
  21 #endif
  22
  23 CommandLine* CommandLine::current_process_commandline_ = NULL;
  24
  25 namespace {
  26 const CommandLine::CharType kSwitchTerminator[] = FILE_PATH_LITERAL("--");
  27 const CommandLine::CharType kSwitchValueSeparator[] = FILE_PATH_LITERAL("=");
  28 // Since we use a lazy match, make sure that longer versions (like "--") are
  29 // listed before shorter versions (like "-") of similar prefixes.
  30 #if defined(OS_WIN)
  31 const CommandLine::CharType* const kSwitchPrefixes[] = {L"--", L"-", L"/"};
  32 #elif defined(OS_POSIX)
  33 // Unixes don't use slash as a switch.
  34 const CommandLine::CharType* const kSwitchPrefixes[] = {"--", "-"};
  35 #endif
  36
  37 size_t GetSwitchPrefixLength(const CommandLine::StringType& string) {
  38   for (size_t i = 0; i < arraysize(kSwitchPrefixes); ++i) {
  39     CommandLine::StringType prefix(kSwitchPrefixes[i]);
  40     if (string.compare(0, prefix.length(), prefix) == 0)
  41       return prefix.length();
  42   }
  43   return 0;
  44 }
  45
  46 // Fills in |switch_string| and |switch_value| if |string| is a switch.
  47 // This will preserve the input switch prefix in the output |switch_string|.
  48 bool IsSwitch(const CommandLine::StringType& string,
  49               CommandLine::StringType* switch_string,
  50               CommandLine::StringType* switch_value) {
  51   switch_string->clear();
  52   switch_value->clear();
  53   if (GetSwitchPrefixLength(string) == 0)
  54     return false;
  55
  56   const size_t equals_position = string.find(kSwitchValueSeparator);
  57   *switch_string = string.substr(0, equals_position);
  58   if (equals_position != CommandLine::StringType::npos)
  59     *switch_value = string.substr(equals_position + 1);
  60   return true;
  61 }
  62
  63 // Append switches and arguments, keeping switches before arguments.
  64 void AppendSwitchesAndArguments(CommandLine& command_line,
  65                                 const CommandLine::StringVector& argv) {
  66   bool parse_switches = true;
  67   for (size_t i = 1; i < argv.size(); ++i) {
  68     CommandLine::StringType arg = argv[i];
  69     TrimWhitespace(arg, TRIM_ALL, &arg);
  70
  71     CommandLine::StringType switch_string;
  72     CommandLine::StringType switch_value;
  73     parse_switches &= (arg != kSwitchTerminator);
  74     if (parse_switches && IsSwitch(arg, &switch_string, &switch_value)) {
  75 #if defined(OS_WIN)
  76       command_line.AppendSwitchNative(WideToASCII(switch_string), switch_value);
  77 #elif defined(OS_POSIX)
  78       command_line.AppendSwitchNative(switch_string, switch_value);
  79 #endif
  80     } else {
  81       command_line.AppendArgNative(arg);
  82     }
  83   }
  84 }
  85
  86 // Lowercase switches for backwards compatiblity *on Windows*.
  87 std::string LowerASCIIOnWindows(const std::string& string) {
  88 #if defined(OS_WIN)
  89   return StringToLowerASCII(string);
  90 #elif defined(OS_POSIX)
  91   return string;
  92 #endif
  93 }
  94
  95
  96 #if defined(OS_WIN)
  97 // Quote a string as necessary for CommandLineToArgvW compatiblity *on Windows*.
  98 std::wstring QuoteForCommandLineToArgvW(const std::wstring& arg) {
  99   // We follow the quoting rules of CommandLineToArgvW.
 100   // http://msdn.microsoft.com/en-us/library/17w5ykft.aspx
 101   if (arg.find_first_of(L" \\\"") == std::wstring::npos) {
 102     // No quoting necessary.
 103     return arg;
 104   }
 105
 106   std::wstring out;
 107   out.push_back(L'"');
 108   for (size_t i = 0; i < arg.size(); ++i) {
 109     if (arg[i] == '\\') {
 110       // Find the extent of this run of backslashes.
 111       size_t start = i, end = start + 1;
 112       for (; end < arg.size() && arg[end] == '\\'; ++end)
 113         /* empty */;
 114       size_t backslash_count = end - start;
 115
 116       // Backslashes are escapes only if the run is followed by a double quote.
 117       // Since we also will end the string with a double quote, we escape for
 118       // either a double quote or the end of the string.
 119       if (end == arg.size() || arg[end] == '"') {
 120         // To quote, we need to output 2x as many backslashes.
 121         backslash_count *= 2;
 122       }
 123       for (size_t j = 0; j < backslash_count; ++j)
 124         out.push_back('\\');
 125
 126       // Advance i to one before the end to balance i++ in loop.
 127       i = end - 1;
 128     } else if (arg[i] == '"') {
 129       out.push_back('\\');
 130       out.push_back('"');
 131     } else {
 132       out.push_back(arg[i]);
 133     }
 134   }
 135   out.push_back('"');
 136
 137   return out;
 138 }
 139 #endif
 140
 141 }  // namespace
 142
 143 CommandLine::CommandLine(NoProgram no_program)
 144     : argv_(1),
 145       begin_args_(1) {
 146 }
 147
 148 CommandLine::CommandLine(const FilePath& program)
 149     : argv_(1),
 150       begin_args_(1) {
 151   SetProgram(program);
 152 }
 153
 154 CommandLine::CommandLine(int argc, const CommandLine::CharType* const* argv)
 155     : argv_(1),
 156       begin_args_(1) {
 157   InitFromArgv(argc, argv);
 158 }
 159
 160 CommandLine::CommandLine(const StringVector& argv)
 161     : argv_(1),
 162       begin_args_(1) {
 163   InitFromArgv(argv);
 164 }
 165
 166 CommandLine::~CommandLine() {
 167 }
 168
 169 // static
 170 void CommandLine::Init(int argc, const char* const* argv) {
 171   if (current_process_commandline_) {
 172     // If this is intentional, Reset() must be called first. If we are using
 173     // the shared build mode, we have to share a single object across multiple
 174     // shared libraries.
 175     return;
 176   }
 177
 178   current_process_commandline_ = new CommandLine(NO_PROGRAM);
 179 #if defined(OS_WIN)
 180   current_process_commandline_->ParseFromString(::GetCommandLineW());
 181 #elif defined(OS_POSIX)
 182   current_process_commandline_->InitFromArgv(argc, argv);
 183 #endif
 184 }
 185
 186 // static
 187 void CommandLine::Reset() {
 188   DCHECK(current_process_commandline_);
 189   delete current_process_commandline_;
 190   current_process_commandline_ = NULL;
 191 }
 192
 193 // static
 194 CommandLine* CommandLine::ForCurrentProcess() {
 195   DCHECK(current_process_commandline_);
 196   return current_process_commandline_;
 197 }
 198
 199 #if defined(OS_WIN)
 200 // static
 201 CommandLine CommandLine::FromString(const std::wstring& command_line) {
 202   CommandLine cmd(NO_PROGRAM);
 203   cmd.ParseFromString(command_line);
 204   return cmd;
 205 }
 206 #endif
 207
 208 void CommandLine::InitFromArgv(int argc,
 209                                const CommandLine::CharType* const* argv) {
 210   StringVector new_argv;
 211   for (int i = 0; i < argc; ++i)
 212     new_argv.push_back(argv[i]);
 213   InitFromArgv(new_argv);
 214 }
 215
 216 void CommandLine::InitFromArgv(const StringVector& argv) {
 217   argv_ = StringVector(1);
 218   begin_args_ = 1;
 219   SetProgram(argv.empty() ? FilePath() : FilePath(argv[0]));
 220   AppendSwitchesAndArguments(*this, argv);
 221 }
 222
 223 CommandLine::StringType CommandLine::GetCommandLineString() const {
 224   StringType string(argv_[0]);
 225 #if defined(OS_WIN)
 226   string = QuoteForCommandLineToArgvW(string);
 227 #endif
 228   // Append switches and arguments.
 229   bool parse_switches = true;
 230   for (size_t i = 1; i < argv_.size(); ++i) {
 231     CommandLine::StringType arg = argv_[i];
 232     CommandLine::StringType switch_string;
 233     CommandLine::StringType switch_value;
 234     parse_switches &= arg != kSwitchTerminator;
 235     string.append(StringType(FILE_PATH_LITERAL(" ")));
 236     if (parse_switches && IsSwitch(arg, &switch_string, &switch_value)) {
 237       string.append(switch_string);
 238       if (!switch_value.empty()) {
 239 #if defined(OS_WIN)
 240         switch_value = QuoteForCommandLineToArgvW(switch_value);
 241 #endif
 242         string.append(kSwitchValueSeparator + switch_value);
 243       }
 244     }
 245     else {
 246 #if defined(OS_WIN)
 247       arg = QuoteForCommandLineToArgvW(arg);
 248 #endif
 249       string.append(arg);
 250     }
 251   }
 252   return string;
 253 }
 254
 255 FilePath CommandLine::GetProgram() const {
 256   return FilePath(argv_[0]);
 257 }
 258
 259 void CommandLine::SetProgram(const FilePath& program) {
 260   TrimWhitespace(program.value(), TRIM_ALL, &argv_[0]);
 261 }
 262
 263 bool CommandLine::HasSwitch(const std::string& switch_string) const {
 264   return switches_.find(LowerASCIIOnWindows(switch_string)) != switches_.end();
 265 }
 266
 267 std::string CommandLine::GetSwitchValueASCII(
 268     const std::string& switch_string) const {
 269   StringType value = GetSwitchValueNative(switch_string);
 270   if (!IsStringASCII(value)) {
 271     DLOG(WARNING) << "Value of switch (" << switch_string << ") must be ASCII.";
 272     return std::string();
 273   }
 274 #if defined(OS_WIN)
 275   return WideToASCII(value);
 276 #else
 277   return value;
 278 #endif
 279 }
 280
 281 FilePath CommandLine::GetSwitchValuePath(
 282     const std::string& switch_string) const {
 283   return FilePath(GetSwitchValueNative(switch_string));
 284 }
 285
 286 CommandLine::StringType CommandLine::GetSwitchValueNative(
 287     const std::string& switch_string) const {
 288   SwitchMap::const_iterator result = switches_.end();
 289   result = switches_.find(LowerASCIIOnWindows(switch_string));
 290   return result == switches_.end() ? StringType() : result->second;
 291 }
 292
 293 void CommandLine::AppendSwitch(const std::string& switch_string) {
 294   AppendSwitchNative(switch_string, StringType());
 295 }
 296
 297 void CommandLine::AppendSwitchPath(const std::string& switch_string,
 298                                    const FilePath& path) {
 299   AppendSwitchNative(switch_string, path.value());
 300 }
 301
 302 void CommandLine::AppendSwitchNative(const std::string& switch_string,
 303                                      const CommandLine::StringType& value) {
 304   std::string switch_key(LowerASCIIOnWindows(switch_string));
 305 #if defined(OS_WIN)
 306   StringType combined_switch_string(ASCIIToWide(switch_key));
 307 #elif defined(OS_POSIX)
 308   StringType combined_switch_string(switch_string);
 309 #endif
 310   size_t prefix_length = GetSwitchPrefixLength(combined_switch_string);
 311   switches_[switch_key.substr(prefix_length)] = value;
 312   // Preserve existing switch prefixes in |argv_|; only append one if necessary.
 313   if (prefix_length == 0)
 314     combined_switch_string = kSwitchPrefixes[0] + combined_switch_string;
 315   if (!value.empty())
 316     combined_switch_string += kSwitchValueSeparator + value;
 317   // Append the switch and update the switches/arguments divider |begin_args_|.
 318   argv_.insert(argv_.begin() + begin_args_++, combined_switch_string);
 319 }
 320
 321 void CommandLine::AppendSwitchASCII(const std::string& switch_string,
 322                                     const std::string& value_string) {
 323 #if defined(OS_WIN)
 324   AppendSwitchNative(switch_string, ASCIIToWide(value_string));
 325 #elif defined(OS_POSIX)
 326   AppendSwitchNative(switch_string, value_string);
 327 #endif
 328 }
 329
 330 void CommandLine::CopySwitchesFrom(const CommandLine& source,
 331                                    const char* const switches[],
 332                                    size_t count) {
 333   for (size_t i = 0; i < count; ++i) {
 334     if (source.HasSwitch(switches[i]))
 335       AppendSwitchNative(switches[i], source.GetSwitchValueNative(switches[i]));
 336   }
 337 }
 338
 339 CommandLine::StringVector CommandLine::GetArgs() const {
 340   // Gather all arguments after the last switch (may include kSwitchTerminator).
 341   StringVector args(argv_.begin() + begin_args_, argv_.end());
 342   // Erase only the first kSwitchTerminator (maybe "--" is a legitimate page?)
 343   StringVector::iterator switch_terminator =
 344       std::find(args.begin(), args.end(), kSwitchTerminator);
 345   if (switch_terminator != args.end())
 346     args.erase(switch_terminator);
 347   return args;
 348 }
 349
 350 void CommandLine::AppendArg(const std::string& value) {
 351 #if defined(OS_WIN)
 352   DCHECK(IsStringUTF8(value));
 353   AppendArgNative(UTF8ToWide(value));
 354 #elif defined(OS_POSIX)
 355   AppendArgNative(value);
 356 #endif
 357 }
 358
 359 void CommandLine::AppendArgPath(const FilePath& path) {
 360   AppendArgNative(path.value());
 361 }
 362
 363 void CommandLine::AppendArgNative(const CommandLine::StringType& value) {
 364   argv_.push_back(value);
 365 }
 366
 367 void CommandLine::AppendArguments(const CommandLine& other,
 368                                   bool include_program) {
 369   if (include_program)
 370     SetProgram(other.GetProgram());
 371   AppendSwitchesAndArguments(*this, other.argv());
 372 }
 373
 374 void CommandLine::PrependWrapper(const CommandLine::StringType& wrapper) {
 375   if (wrapper.empty())
 376     return;
 377   // The wrapper may have embedded arguments (like "gdb --args"). In this case,
 378   // we don't pretend to do anything fancy, we just split on spaces.
 379   StringVector wrapper_argv;
 380   base::SplitString(wrapper, FILE_PATH_LITERAL(' '), &wrapper_argv);
 381   // Prepend the wrapper and update the switches/arguments |begin_args_|.
 382   argv_.insert(argv_.begin(), wrapper_argv.begin(), wrapper_argv.end());
 383   begin_args_ += wrapper_argv.size();
 384 }
 385
 386 #if defined(OS_WIN)
 387 void CommandLine::ParseFromString(const std::wstring& command_line) {
 388   std::wstring command_line_string;
 389   TrimWhitespace(command_line, TRIM_ALL, &command_line_string);
 390   if (command_line_string.empty())
 391     return;
 392
 393   int num_args = 0;
 394   wchar_t** args = NULL;
 395   args = ::CommandLineToArgvW(command_line_string.c_str(), &num_args);
 396
 397   DPLOG_IF(FATAL, !args) << "CommandLineToArgvW failed on command line: "
 398                          << command_line;
 399   InitFromArgv(num_args, args);
 400   LocalFree(args);
 401 }
 402 #endif