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 #include "base/process/launch.h" | 5 #include "base/process/launch.h" |
6 | 6 |
7 #include <fcntl.h> | 7 #include <fcntl.h> |
8 #include <io.h> | 8 #include <io.h> |
| 9 #include <shellapi.h> |
9 #include <windows.h> | 10 #include <windows.h> |
10 #include <userenv.h> | 11 #include <userenv.h> |
11 #include <psapi.h> | 12 #include <psapi.h> |
12 | 13 |
13 #include <ios> | 14 #include <ios> |
14 #include <limits> | 15 #include <limits> |
15 | 16 |
16 #include "base/bind.h" | 17 #include "base/bind.h" |
17 #include "base/bind_helpers.h" | 18 #include "base/bind_helpers.h" |
18 #include "base/command_line.h" | 19 #include "base/command_line.h" |
(...skipping 213 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
232 ProcessHandle* process_handle) { | 233 ProcessHandle* process_handle) { |
233 if (!process_handle) | 234 if (!process_handle) |
234 return LaunchProcess(cmdline.GetCommandLineString(), options, NULL); | 235 return LaunchProcess(cmdline.GetCommandLineString(), options, NULL); |
235 | 236 |
236 win::ScopedHandle process; | 237 win::ScopedHandle process; |
237 bool rv = LaunchProcess(cmdline.GetCommandLineString(), options, &process); | 238 bool rv = LaunchProcess(cmdline.GetCommandLineString(), options, &process); |
238 *process_handle = process.Take(); | 239 *process_handle = process.Take(); |
239 return rv; | 240 return rv; |
240 } | 241 } |
241 | 242 |
| 243 bool LaunchElevatedProcess(const CommandLine& cmdline, |
| 244 const LaunchOptions& options, |
| 245 ProcessHandle* process_handle) { |
| 246 const string16 file = cmdline.GetProgram().value(); |
| 247 const string16 arguments = cmdline.GetArgumentsString(); |
| 248 |
| 249 SHELLEXECUTEINFO shex_info = {0}; |
| 250 shex_info.cbSize = sizeof(shex_info); |
| 251 shex_info.fMask = SEE_MASK_NOCLOSEPROCESS; |
| 252 shex_info.hwnd = GetActiveWindow(); |
| 253 shex_info.lpVerb = L"runas"; |
| 254 shex_info.lpFile = file.c_str(); |
| 255 shex_info.lpParameters = arguments.c_str(); |
| 256 shex_info.lpDirectory = NULL; |
| 257 shex_info.nShow = options.start_hidden ? SW_HIDE : SW_SHOW; |
| 258 shex_info.hInstApp = NULL; |
| 259 |
| 260 if (!ShellExecuteEx(&shex_info)) { |
| 261 DPLOG(ERROR); |
| 262 return false; |
| 263 } |
| 264 |
| 265 if (options.wait) |
| 266 WaitForSingleObject(shex_info.hProcess, INFINITE); |
| 267 |
| 268 // If the caller wants the process handle give it to them, otherwise just |
| 269 // close it. Closing it does not terminate the process. |
| 270 if (process_handle) |
| 271 *process_handle = shex_info.hProcess; |
| 272 else |
| 273 CloseHandle(shex_info.hProcess); |
| 274 |
| 275 return true; |
| 276 } |
| 277 |
242 bool SetJobObjectLimitFlags(HANDLE job_object, DWORD limit_flags) { | 278 bool SetJobObjectLimitFlags(HANDLE job_object, DWORD limit_flags) { |
243 JOBOBJECT_EXTENDED_LIMIT_INFORMATION limit_info = {0}; | 279 JOBOBJECT_EXTENDED_LIMIT_INFORMATION limit_info = {0}; |
244 limit_info.BasicLimitInformation.LimitFlags = limit_flags; | 280 limit_info.BasicLimitInformation.LimitFlags = limit_flags; |
245 return 0 != SetInformationJobObject( | 281 return 0 != SetInformationJobObject( |
246 job_object, | 282 job_object, |
247 JobObjectExtendedLimitInformation, | 283 JobObjectExtendedLimitInformation, |
248 &limit_info, | 284 &limit_info, |
249 sizeof(limit_info)); | 285 sizeof(limit_info)); |
250 } | 286 } |
251 | 287 |
(...skipping 71 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
323 WaitForSingleObject(proc_info.process_handle(), INFINITE); | 359 WaitForSingleObject(proc_info.process_handle(), INFINITE); |
324 | 360 |
325 return true; | 361 return true; |
326 } | 362 } |
327 | 363 |
328 void RaiseProcessToHighPriority() { | 364 void RaiseProcessToHighPriority() { |
329 SetPriorityClass(GetCurrentProcess(), HIGH_PRIORITY_CLASS); | 365 SetPriorityClass(GetCurrentProcess(), HIGH_PRIORITY_CLASS); |
330 } | 366 } |
331 | 367 |
332 } // namespace base | 368 } // namespace base |
OLD | NEW |