diff options
author | mgiuca <mgiuca@chromium.org> | 2014-10-01 02:24:51 -0700 |
---|---|---|
committer | Commit bot <commit-bot@chromium.org> | 2014-10-01 09:25:00 +0000 |
commit | c974d5100e52cdbc33c0d687f80498f34b6f3e35 (patch) | |
tree | 84cb856151d969bb5ed3bb10e85dc04caa4fe120 /base/command_line.cc | |
parent | 9b4a861e8438048e06bd8d89da8e6ab635553e19 (diff) | |
download | chromium_src-c974d5100e52cdbc33c0d687f80498f34b6f3e35.zip chromium_src-c974d5100e52cdbc33c0d687f80498f34b6f3e35.tar.gz chromium_src-c974d5100e52cdbc33c0d687f80498f34b6f3e35.tar.bz2 |
base::CommandLine: Added optional quoting of placeholder arguments.
Added an optional parameter |quote_placeholder| to GetCommandLineString
and GetArgumentsString. If true, placeholder command-line arguments such
as "%1" will also be quoted. These need quoting in case the substituted
argument has a space in it. It was previously not possible to build a
CommandLine that would contain a quoted "%1" when converted to a string.
Only affects Windows.
BUG=416785
Review URL: https://codereview.chromium.org/595803002
Cr-Commit-Position: refs/heads/master@{#297617}
Diffstat (limited to 'base/command_line.cc')
-rw-r--r-- | base/command_line.cc | 99 |
1 files changed, 54 insertions, 45 deletions
diff --git a/base/command_line.cc b/base/command_line.cc index 9bad539..1f5edce 100644 --- a/base/command_line.cc +++ b/base/command_line.cc @@ -105,10 +105,16 @@ std::string LowerASCIIOnWindows(const std::string& string) { #if defined(OS_WIN) // Quote a string as necessary for CommandLineToArgvW compatiblity *on Windows*. -base::string16 QuoteForCommandLineToArgvW(const base::string16& arg) { +base::string16 QuoteForCommandLineToArgvW(const base::string16& arg, + bool quote_placeholders) { // We follow the quoting rules of CommandLineToArgvW. // http://msdn.microsoft.com/en-us/library/17w5ykft.aspx - if (arg.find_first_of(L" \\\"") == base::string16::npos) { + base::string16 quotable_chars(L" \\\""); + // We may also be required to quote '%', which is commonly used in a command + // line as a placeholder. (It may be substituted for a string with spaces.) + if (quote_placeholders) + quotable_chars.push_back(L'%'); + if (arg.find_first_of(quotable_chars) == base::string16::npos) { // No quoting necessary. return arg; } @@ -247,49 +253,6 @@ void CommandLine::InitFromArgv(const StringVector& argv) { AppendSwitchesAndArguments(*this, argv); } -CommandLine::StringType CommandLine::GetCommandLineString() const { - StringType string(argv_[0]); -#if defined(OS_WIN) - string = QuoteForCommandLineToArgvW(string); -#endif - StringType params(GetArgumentsString()); - if (!params.empty()) { - string.append(StringType(FILE_PATH_LITERAL(" "))); - string.append(params); - } - return string; -} - -CommandLine::StringType CommandLine::GetArgumentsString() const { - StringType params; - // Append switches and arguments. - bool parse_switches = true; - for (size_t i = 1; i < argv_.size(); ++i) { - StringType arg = argv_[i]; - StringType switch_string; - StringType switch_value; - parse_switches &= arg != kSwitchTerminator; - if (i > 1) - params.append(StringType(FILE_PATH_LITERAL(" "))); - if (parse_switches && IsSwitch(arg, &switch_string, &switch_value)) { - params.append(switch_string); - if (!switch_value.empty()) { -#if defined(OS_WIN) - switch_value = QuoteForCommandLineToArgvW(switch_value); -#endif - params.append(kSwitchValueSeparator + switch_value); - } - } - else { -#if defined(OS_WIN) - arg = QuoteForCommandLineToArgvW(arg); -#endif - params.append(arg); - } - } - return params; -} - FilePath CommandLine::GetProgram() const { return FilePath(argv_[0]); } @@ -439,4 +402,50 @@ void CommandLine::ParseFromString(const base::string16& command_line) { } #endif +CommandLine::StringType CommandLine::GetCommandLineStringInternal( + bool quote_placeholders) const { + StringType string(argv_[0]); +#if defined(OS_WIN) + string = QuoteForCommandLineToArgvW(string, quote_placeholders); +#endif + StringType params(GetArgumentsStringInternal(quote_placeholders)); + if (!params.empty()) { + string.append(StringType(FILE_PATH_LITERAL(" "))); + string.append(params); + } + return string; +} + +CommandLine::StringType CommandLine::GetArgumentsStringInternal( + bool quote_placeholders) const { + StringType params; + // Append switches and arguments. + bool parse_switches = true; + for (size_t i = 1; i < argv_.size(); ++i) { + StringType arg = argv_[i]; + StringType switch_string; + StringType switch_value; + parse_switches &= arg != kSwitchTerminator; + if (i > 1) + params.append(StringType(FILE_PATH_LITERAL(" "))); + if (parse_switches && IsSwitch(arg, &switch_string, &switch_value)) { + params.append(switch_string); + if (!switch_value.empty()) { +#if defined(OS_WIN) + switch_value = + QuoteForCommandLineToArgvW(switch_value, quote_placeholders); +#endif + params.append(kSwitchValueSeparator + switch_value); + } + } + else { +#if defined(OS_WIN) + arg = QuoteForCommandLineToArgvW(arg, quote_placeholders); +#endif + params.append(arg); + } + } + return params; +} + } // namespace base |