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
|
// 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>
// 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_;
};
LRESULT CALLBACK ChromeFrameHelperWndProc(HWND hwnd,
UINT message,
WPARAM wparam,
LPARAM lparam) {
switch (message) {
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) {
// TODO(robertshield): Before this actually gets used, add breakpad
// integration.
// 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;
}
|
