OLD | NEW |
1 // Copyright (c) 2006-2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2006-2010 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 #include <windows.h> | 5 #include <windows.h> |
6 #include <tchar.h> | 6 #include <tchar.h> |
7 #include <shellapi.h> | 7 #include <shellapi.h> |
8 #include "sandbox/win/sandbox_poc/sandbox.h" | 8 #include "sandbox/win/sandbox_poc/sandbox.h" |
9 #include "base/logging.h" | 9 #include "base/logging.h" |
10 #include "sandbox/win/sandbox_poc/main_ui_window.h" | 10 #include "sandbox/win/sandbox_poc/main_ui_window.h" |
11 #include "sandbox/win/src/sandbox.h" | 11 #include "sandbox/win/src/sandbox.h" |
12 #include "sandbox/win/src/sandbox_factory.h" | 12 #include "sandbox/win/src/sandbox_factory.h" |
13 | 13 |
14 // Prototype allowed for functions to be called in the POC | 14 // Prototype allowed for functions to be called in the POC |
15 typedef void(__cdecl *lpfnInit)(HANDLE); | 15 typedef void(__cdecl *lpfnInit)(HANDLE); |
16 | 16 |
17 bool ParseCommandLine(wchar_t * command_line, | 17 bool ParseCommandLine(wchar_t * command_line, |
18 std::string * dll_name, | 18 std::string * dll_name, |
19 std::string * entry_point, | 19 std::string * entry_point, |
20 std::wstring * log_file) { | 20 base::string16 * log_file) { |
21 DCHECK(dll_name); | 21 DCHECK(dll_name); |
22 DCHECK(entry_point); | 22 DCHECK(entry_point); |
23 DCHECK(log_file); | 23 DCHECK(log_file); |
24 if (!dll_name || !entry_point || !log_file) | 24 if (!dll_name || !entry_point || !log_file) |
25 return false; | 25 return false; |
26 | 26 |
27 LPWSTR *arg_list; | 27 LPWSTR *arg_list; |
28 int arg_count; | 28 int arg_count; |
29 | 29 |
30 // We expect the command line to contain: EntryPointName "DLLPath" "LogPath" | 30 // We expect the command line to contain: EntryPointName "DLLPath" "LogPath" |
31 // NOTE: Double quotes are required, even if long path name not used | 31 // NOTE: Double quotes are required, even if long path name not used |
32 // NOTE: LogPath can be blank, but still requires the double quotes | 32 // NOTE: LogPath can be blank, but still requires the double quotes |
33 arg_list = CommandLineToArgvW(command_line, &arg_count); | 33 arg_list = CommandLineToArgvW(command_line, &arg_count); |
34 if (NULL == arg_list || arg_count < 4) { | 34 if (NULL == arg_list || arg_count < 4) { |
35 return false; | 35 return false; |
36 } | 36 } |
37 | 37 |
38 std::wstring entry_point_wide = arg_list[1]; | 38 base::string16 entry_point_wide = arg_list[1]; |
39 std::wstring dll_name_wide = arg_list[2]; | 39 base::string16 dll_name_wide = arg_list[2]; |
40 *entry_point = std::string(entry_point_wide.begin(), entry_point_wide.end()); | 40 *entry_point = std::string(entry_point_wide.begin(), entry_point_wide.end()); |
41 *dll_name = std::string(dll_name_wide.begin(), dll_name_wide.end()); | 41 *dll_name = std::string(dll_name_wide.begin(), dll_name_wide.end()); |
42 *log_file = arg_list[3]; | 42 *log_file = arg_list[3]; |
43 | 43 |
44 // Free memory allocated for CommandLineToArgvW arguments. | 44 // Free memory allocated for CommandLineToArgvW arguments. |
45 LocalFree(arg_list); | 45 LocalFree(arg_list); |
46 | 46 |
47 return true; | 47 return true; |
48 } | 48 } |
49 | 49 |
(...skipping 69 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
119 | 119 |
120 // Sleep(10000); | 120 // Sleep(10000); |
121 | 121 |
122 if (sandbox::SBOX_ALL_OK != (result = target_service->Init())) { | 122 if (sandbox::SBOX_ALL_OK != (result = target_service->Init())) { |
123 // TODO(finnur): write the initialization error to the log file | 123 // TODO(finnur): write the initialization error to the log file |
124 return -2; | 124 return -2; |
125 } | 125 } |
126 | 126 |
127 // Parse the command line to find out what we need to call | 127 // Parse the command line to find out what we need to call |
128 std::string dll_name, entry_point; | 128 std::string dll_name, entry_point; |
129 std::wstring log_file; | 129 base::string16 log_file; |
130 if (!ParseCommandLine(GetCommandLineW(), | 130 if (!ParseCommandLine(GetCommandLineW(), |
131 &dll_name, | 131 &dll_name, |
132 &entry_point, | 132 &entry_point, |
133 &log_file)) { | 133 &log_file)) { |
134 // TODO(finnur): write the failure to the log file | 134 // TODO(finnur): write the failure to the log file |
135 return -3; | 135 return -3; |
136 } | 136 } |
137 | 137 |
138 // Open the pipe to transfert the log output | 138 // Open the pipe to transfert the log output |
139 HANDLE pipe = ::CreateFile(log_file.c_str(), | 139 HANDLE pipe = ::CreateFile(log_file.c_str(), |
(...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
173 | 173 |
174 CloseHandle(pipe); | 174 CloseHandle(pipe); |
175 Sleep(1000); // Give a change to the debug output to arrive before the | 175 Sleep(1000); // Give a change to the debug output to arrive before the |
176 // end of the process | 176 // end of the process |
177 | 177 |
178 ::FreeLibrary(dll_module); | 178 ::FreeLibrary(dll_module); |
179 } | 179 } |
180 | 180 |
181 return 0; | 181 return 0; |
182 } | 182 } |
OLD | NEW |