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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
|
// Copyright (c) 2010 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.
//
// chrome_frame_helper_main.cc : The .exe that bootstraps the
// chrome_frame_helper.dll.
//
// This is a small exe that loads the hook dll to set the event hooks and then
// waits in a message loop. It is intended to be shut down by looking for a
// window with the class and title
// kChromeFrameHelperWindowClassName and kChromeFrameHelperWindowName and then
// sending that window a WM_CLOSE message.
//
#include <crtdbg.h>
#include <windows.h>
#include "chrome_frame/crash_server_init.h"
// Window class and window names.
const wchar_t kChromeFrameHelperWindowClassName[] =
L"ChromeFrameHelperWindowClass";
const wchar_t kChromeFrameHelperWindowName[] =
L"ChromeFrameHelperWindowName";
// Small helper class that assists in loading the DLL that contains code
// to register our event hook callback. Automatically removes the hook and
// unloads the DLL on destruction.
class HookDllLoader {
public:
HookDllLoader() : dll_(NULL), start_proc_(NULL), stop_proc_(NULL) {}
~HookDllLoader() {
if (dll_) {
Unload();
}
}
bool Load() {
dll_ = LoadLibrary(L"chrome_frame_helper.dll");
if (dll_) {
start_proc_ = GetProcAddress(dll_, "StartUserModeBrowserInjection");
stop_proc_ = GetProcAddress(dll_, "StopUserModeBrowserInjection");
}
bool result = true;
if (!start_proc_ || !stop_proc_) {
_ASSERTE(L"failed to load hook dll.");
result = false;
} else {
if (FAILED(start_proc_())) {
_ASSERTE(L"failed to initialize hook dll.");
result = false;
}
}
return result;
}
void Unload() {
if (stop_proc_) {
stop_proc_();
}
if (dll_) {
FreeLibrary(dll_);
}
}
private:
HMODULE dll_;
PROC start_proc_;
PROC stop_proc_;
};
// Checks the window title and then class of hwnd. If they match with that
// of a chrome_frame_helper.exe window, then add it to the list of windows
// pointed to by lparam.
BOOL CALLBACK CloseHelperWindowsEnumProc(HWND hwnd, LPARAM lparam) {
_ASSERTE(lparam != 0);
wchar_t title_buffer[MAX_PATH] = {0};
if (GetWindowText(hwnd, title_buffer, MAX_PATH)) {
if (lstrcmpiW(title_buffer, kChromeFrameHelperWindowName) == 0) {
wchar_t class_buffer[MAX_PATH] = {0};
if (GetClassName(hwnd, class_buffer, MAX_PATH)) {
if (lstrcmpiW(class_buffer, kChromeFrameHelperWindowClassName) == 0) {
if (hwnd != reinterpret_cast<HWND>(lparam)) {
PostMessage(hwnd, WM_CLOSE, 0, 0);
}
}
}
}
}
return TRUE;
}
// Enumerates all top level windows, looking for those that look like a
// Chrome Frame helper window. It then closes all of them except for
// except_me.
void CloseAllHelperWindowsApartFrom(HWND except_me) {
EnumWindows(CloseHelperWindowsEnumProc, reinterpret_cast<LPARAM>(except_me));
}
LRESULT CALLBACK ChromeFrameHelperWndProc(HWND hwnd,
UINT message,
WPARAM wparam,
LPARAM lparam) {
switch (message) {
case WM_CREATE:
CloseAllHelperWindowsApartFrom(hwnd);
break;
case WM_DESTROY:
PostQuitMessage(0);
break;
default:
return ::DefWindowProc(hwnd, message, wparam, lparam);
}
return 0;
}
HWND RegisterAndCreateWindow(HINSTANCE hinstance) {
WNDCLASSEX wcex = {0};
wcex.cbSize = sizeof(WNDCLASSEX);
wcex.lpfnWndProc = ChromeFrameHelperWndProc;
wcex.hInstance = GetModuleHandle(NULL);
wcex.hbrBackground = reinterpret_cast<HBRUSH>(COLOR_WINDOW+1);
wcex.lpszClassName = kChromeFrameHelperWindowClassName;
RegisterClassEx(&wcex);
HWND hwnd = CreateWindow(kChromeFrameHelperWindowClassName,
kChromeFrameHelperWindowName, WS_OVERLAPPEDWINDOW, CW_USEDEFAULT, 0,
CW_USEDEFAULT, 0, NULL, NULL, hinstance, NULL);
return hwnd;
}
int APIENTRY wWinMain(HINSTANCE hinstance, HINSTANCE, wchar_t*, int show_cmd) {
const wchar_t* cmd_line = ::GetCommandLine();
google_breakpad::scoped_ptr<google_breakpad::ExceptionHandler> breakpad(
InitializeCrashReporting(cmd_line));
// Create a window with a known class and title just to listen for WM_CLOSE
// messages that will shut us down.
HWND hwnd = RegisterAndCreateWindow(hinstance);
_ASSERTE(hwnd);
// Load the hook dll, and set the event hooks.
HookDllLoader loader;
bool loaded = loader.Load();
_ASSERTE(loaded);
if (loaded) {
MSG msg;
BOOL ret;
// Main message loop:
while ((ret = GetMessage(&msg, NULL, 0, 0))) {
if (ret == -1) {
break;
} else {
TranslateMessage(&msg);
DispatchMessage(&msg);
}
}
}
return 0;
}
|
