1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
// Copyright (c) 2011 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome_frame/update_launcher.h"
#include <windows.h>
#include <Shellapi.h>
#include "google_update_idl.h" // NOLINT
namespace {
const wchar_t kChromeFrameGuid[] = L"{8BA986DA-5100-405E-AA35-86F34A02ACBF}";
const DWORD kLaunchFailureExitCode = 0xFF;
const wchar_t kUpdateCommandFlag[] = L"--update-cmd";
// Waits indefinitely for the provided process to exit. Returns the process's
// exit code, or kLaunchFailureExitCode if an error occurs in the waiting.
DWORD WaitForProcessExitCode(HANDLE handle) {
DWORD exit_code = 0;
DWORD wait_result = ::WaitForSingleObject(handle, INFINITE);
if (wait_result == WAIT_OBJECT_0 && ::GetExitCodeProcess(handle, &exit_code))
return exit_code;
return kLaunchFailureExitCode;
}
} // namespace
namespace update_launcher {
std::wstring GetUpdateCommandFromArguments(const wchar_t* command_line) {
std::wstring command;
if (command_line != NULL) {
int num_args = 0;
wchar_t** args = NULL;
args = ::CommandLineToArgvW(command_line, &num_args);
if (args) {
if (num_args == 3 && _wcsicmp(args[1], kUpdateCommandFlag) == 0)
command = args[2];
::LocalFree(args);
}
}
return command;
}
// Because we do not have 'base' and all of its pretty RAII helpers, please
// ensure that this function only ever contains a single 'return', in order
// to reduce the chance of introducing a leak.
DWORD LaunchUpdateCommand(const std::wstring& command) {
DWORD exit_code = kLaunchFailureExitCode;
HRESULT hr = ::CoInitialize(NULL);
if (SUCCEEDED(hr)) {
IProcessLauncher* ipl = NULL;
HANDLE process = NULL;
hr = ::CoCreateInstance(__uuidof(ProcessLauncherClass), NULL,
CLSCTX_ALL, __uuidof(IProcessLauncher),
reinterpret_cast<void**>(&ipl));
if (SUCCEEDED(hr)) {
ULONG_PTR phandle = NULL;
DWORD id = ::GetCurrentProcessId();
hr = ipl->LaunchCmdElevated(kChromeFrameGuid,
command.c_str(), id, &phandle);
if (SUCCEEDED(hr)) {
process = reinterpret_cast<HANDLE>(phandle);
exit_code = WaitForProcessExitCode(process);
}
}
if (process)
::CloseHandle(process);
if (ipl)
ipl->Release();
::CoUninitialize();
}
return exit_code;
}
} // namespace process_launcher
|