OLD | NEW |
1 // Copyright 2015 The Crashpad Authors. All rights reserved. | 1 // Copyright 2015 The Crashpad Authors. All rights reserved. |
2 // | 2 // |
3 // Licensed under the Apache License, Version 2.0 (the "License"); | 3 // Licensed under the Apache License, Version 2.0 (the "License"); |
4 // you may not use this file except in compliance with the License. | 4 // you may not use this file except in compliance with the License. |
5 // You may obtain a copy of the License at | 5 // You may obtain a copy of the License at |
6 // | 6 // |
7 // http://www.apache.org/licenses/LICENSE-2.0 | 7 // http://www.apache.org/licenses/LICENSE-2.0 |
8 // | 8 // |
9 // Unless required by applicable law or agreed to in writing, software | 9 // Unless required by applicable law or agreed to in writing, software |
10 // distributed under the License is distributed on an "AS IS" BASIS, | 10 // distributed under the License is distributed on an "AS IS" BASIS, |
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
12 // See the License for the specific language governing permissions and | 12 // See the License for the specific language governing permissions and |
13 // limitations under the License. | 13 // limitations under the License. |
14 | 14 |
15 #include "test/win/child_launcher.h" | 15 #include "test/win/child_launcher.h" |
16 | 16 |
17 #include "gtest/gtest.h" | 17 #include "gtest/gtest.h" |
| 18 #include "util/win/command_line.h" |
18 | 19 |
19 namespace crashpad { | 20 namespace crashpad { |
20 namespace test { | 21 namespace test { |
21 | 22 |
22 ChildLauncher::ChildLauncher(const std::wstring& executable, | 23 ChildLauncher::ChildLauncher(const std::wstring& executable, |
23 const std::wstring& command_line) | 24 const std::wstring& command_line) |
24 : executable_(executable), | 25 : executable_(executable), |
25 command_line_(command_line), | 26 command_line_(command_line), |
26 process_handle_(), | 27 process_handle_(), |
27 main_thread_handle_(), | 28 main_thread_handle_(), |
(...skipping 61 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
89 DWORD ChildLauncher::WaitForExit() { | 90 DWORD ChildLauncher::WaitForExit() { |
90 EXPECT_TRUE(process_handle_.is_valid()); | 91 EXPECT_TRUE(process_handle_.is_valid()); |
91 EXPECT_EQ(WAIT_OBJECT_0, | 92 EXPECT_EQ(WAIT_OBJECT_0, |
92 WaitForSingleObject(process_handle_.get(), INFINITE)); | 93 WaitForSingleObject(process_handle_.get(), INFINITE)); |
93 DWORD exit_code = 0; | 94 DWORD exit_code = 0; |
94 EXPECT_TRUE(GetExitCodeProcess(process_handle_.get(), &exit_code)); | 95 EXPECT_TRUE(GetExitCodeProcess(process_handle_.get(), &exit_code)); |
95 process_handle_.reset(); | 96 process_handle_.reset(); |
96 return exit_code; | 97 return exit_code; |
97 } | 98 } |
98 | 99 |
99 // Ref: http://blogs.msdn.com/b/twistylittlepassagesallalike/archive/2011/04/23/
everyone-quotes-arguments-the-wrong-way.aspx | |
100 void AppendCommandLineArgument(const std::wstring& argument, | |
101 std::wstring* command_line) { | |
102 // Don't bother quoting if unnecessary. | |
103 if (!argument.empty() && | |
104 argument.find_first_of(L" \t\n\v\"") == std::wstring::npos) { | |
105 command_line->append(argument); | |
106 } else { | |
107 command_line->push_back(L'"'); | |
108 for (std::wstring::const_iterator i = argument.begin();; ++i) { | |
109 size_t backslash_count = 0; | |
110 while (i != argument.end() && *i == L'\\') { | |
111 ++i; | |
112 ++backslash_count; | |
113 } | |
114 if (i == argument.end()) { | |
115 // Escape all backslashes, but let the terminating double quotation mark | |
116 // we add below be interpreted as a metacharacter. | |
117 command_line->append(backslash_count * 2, L'\\'); | |
118 break; | |
119 } else if (*i == L'"') { | |
120 // Escape all backslashes and the following double quotation mark. | |
121 command_line->append(backslash_count * 2 + 1, L'\\'); | |
122 command_line->push_back(*i); | |
123 } else { | |
124 // Backslashes aren't special here. | |
125 command_line->append(backslash_count, L'\\'); | |
126 command_line->push_back(*i); | |
127 } | |
128 } | |
129 command_line->push_back(L'"'); | |
130 } | |
131 } | |
132 | |
133 } // namespace test | 100 } // namespace test |
134 } // namespace crashpad | 101 } // namespace crashpad |
OLD | NEW |