Left: | ||
Right: |
OLD | NEW |
---|---|
1 /** | 1 /** |
2 * \file test-installer-lib-sandbox.cpp | 2 * \file test-installer-lib-sandbox.cpp |
3 * | 3 * |
4 * Automatic testing of many of the units within the custom action is infeasible . | 4 * Automatic testing of many of the units within the custom action is infeasible . |
5 * In one case, they rely on the execution environment within an installation se ssion. | 5 * In one case, they rely on the execution environment within an installation se ssion. |
6 * In another, they rely on the operation system environment as a whole. | 6 * In another, they rely on the operation system environment as a whole. |
7 * In these cases, it's easier to verify behavior manually. | 7 * In these cases, it's easier to verify behavior manually. |
8 * | 8 * |
9 * This file contains a custom action function sandbox() as well as a number of test functions. | 9 * This file contains a custom action function sandbox() as well as a number of test functions. |
10 * At any given time, not all of the test functions need to be referenced within the body of custom action. | 10 * At any given time, not all of the test functions need to be referenced within the body of custom action. |
11 */ | 11 */ |
12 | 12 |
13 #include <sstream> | 13 #include <sstream> |
14 #include <functional> | 14 #include <functional> |
15 | 15 |
16 #include "session.h" | 16 #include "session.h" |
17 #include "property.h" | 17 #include "property.h" |
18 #include "database.h" | 18 #include "database.h" |
19 #include "process.h" | 19 #include "process.h" |
20 #include "interaction.h" | 20 #include "interaction.h" |
21 | 21 |
22 //------------------------------------------------------- | 22 //------------------------------------------------------- |
23 // log_all_window_handles | 23 // LogAllWindowHandles |
24 //------------------------------------------------------- | 24 //------------------------------------------------------- |
25 class log_single_window_handle | 25 class LogSingleWindowHandle |
26 { | 26 { |
27 ImmediateSession& session; | 27 ImmediateSession& session; |
28 | 28 |
29 public: | 29 public: |
30 log_single_window_handle(ImmediateSession& session) | 30 LogSingleWindowHandle(ImmediateSession& session) |
31 : session(session) | 31 : session(session) |
32 { | 32 { |
33 } | 33 } |
34 | 34 |
35 bool operator()(HWND window) | 35 bool operator()(HWND window) |
36 { | 36 { |
37 std::stringstream s; | 37 std::stringstream s; |
38 s << "Window handle 0x" << std::hex << window; | 38 s << "Window handle 0x" << std::hex << window; |
39 session.Log(s.str()); | 39 session.Log(s.str()); |
40 return true; | 40 return true; |
41 } | 41 } |
42 }; | 42 }; |
43 | 43 |
44 void log_all_window_handles(ImmediateSession& session) | 44 void LogAllWindowHandles(ImmediateSession& session) |
45 { | 45 { |
46 session.Log("log_all_window_handles"); | 46 session.Log("LogAllWindowHandles"); |
47 log_single_window_handle lp(session); | 47 LogSingleWindowHandle lp(session); |
48 EnumerateWindows(lp); | 48 EnumerateWindows(lp); |
49 } | 49 } |
50 | 50 |
51 //------------------------------------------------------- | 51 //------------------------------------------------------- |
52 // log_IE_window_handles | 52 // LogIEWindowHandles |
53 //------------------------------------------------------- | 53 //------------------------------------------------------- |
54 class log_single_window_handle_only_if_IE | 54 class LogSingleWindowHandleOnlyIfIE |
Oleksandr
2015/10/29 23:26:22
Nit: LogSingleWindowHandleOnlyIfIe
Eric
2015/11/14 18:01:07
Done.
| |
55 { | 55 { |
56 ImmediateSession& session; | 56 ImmediateSession& session; |
57 | 57 |
58 ProcessCloser& pc; | 58 ProcessCloser& pc; |
59 | 59 |
60 public: | 60 public: |
61 log_single_window_handle_only_if_IE(ImmediateSession& session, ProcessCloser& pc) | 61 LogSingleWindowHandleOnlyIfIE(ImmediateSession& session, ProcessCloser& pc) |
62 : session(session), pc(pc) | 62 : session(session), pc(pc) |
63 { | 63 { |
64 } | 64 } |
65 | 65 |
66 bool operator()(HWND window) | 66 bool operator()(HWND window) |
67 { | 67 { |
68 DWORD pid = CreatorProcess(window); | 68 DWORD pid = CreatorProcess(window); |
69 if (pc.Contains(pid)) | 69 if (pc.Contains(pid)) |
70 { | 70 { |
71 std::stringstream s; | 71 std::stringstream s; |
72 s << "Window handle 0x" << std::hex << window; | 72 s << "Window handle 0x" << std::hex << window; |
73 session.Log(s.str()); | 73 session.Log(s.str()); |
74 } | 74 } |
75 return true; | 75 return true; |
76 } | 76 } |
77 }; | 77 }; |
78 | 78 |
79 void log_IE_window_handles(ImmediateSession& session) | 79 void LogIEWindowHandles(ImmediateSession& session) |
80 { | 80 { |
81 session.Log("log_IE_window_handles"); | 81 session.Log("LogIEWindowHandles"); |
82 const wchar_t* IE_names[] = { L"IExplore.exe", L"AdblockPlusEngine.exe" }; | 82 const wchar_t* IENames[] = {L"IExplore.exe", L"AdblockPlusEngine.exe"}; |
Oleksandr
2015/10/29 23:26:22
Nit: IeNames?
Eric
2015/11/14 18:01:07
Done.
| |
83 ProcessSnapshot snapshot; | 83 ProcessSnapshot snapshot; |
84 ProcessCloser iec(snapshot, IE_names); | 84 ProcessCloser iec(snapshot, IENames); |
85 log_single_window_handle_only_if_IE lp(session, iec); | 85 LogSingleWindowHandleOnlyIfIE lp(session, iec); |
86 EnumerateWindows(lp); | 86 EnumerateWindows(lp); |
87 } | 87 } |
88 | 88 |
89 //------------------------------------------------------- | 89 //------------------------------------------------------- |
90 // log_only_window_handle_in_closer | 90 // LogOnlyWindowHandleInCloser |
91 //------------------------------------------------------- | 91 //------------------------------------------------------- |
92 void log_only_window_handle_in_closer(ImmediateSession& session) | 92 void LogOnlyWindowHandleInCloser(ImmediateSession& session) |
93 { | 93 { |
94 session.Log("log_only_window_handle_in_closer"); | 94 session.Log("LogOnlyWindowHandleInCloser"); |
95 const wchar_t* IE_names[] = { L"IExplore.exe", L"AdblockPlusEngine.exe" }; | 95 const wchar_t* IENames[] = {L"IExplore.exe", L"AdblockPlusEngine.exe"}; |
96 ProcessSnapshot snapshot; | 96 ProcessSnapshot snapshot; |
97 ProcessCloser iec(snapshot, IE_names); | 97 ProcessCloser iec(snapshot, IENames); |
98 iec.IterateOurWindows(log_single_window_handle(session)); | 98 iec.IterateOurWindows(LogSingleWindowHandle(session)); |
99 } | 99 } |
100 | 100 |
101 //------------------------------------------------------- | 101 //------------------------------------------------------- |
102 // sandbox | 102 // sandbox |
103 //------------------------------------------------------- | 103 //------------------------------------------------------- |
104 /** | 104 /** |
105 * Exposed DLL entry point for custom action. | 105 * Exposed DLL entry point for custom action. |
106 * The function signature matches the calling convention used by Windows Install er. | 106 * The function signature matches the calling convention used by Windows Install er. |
107 | 107 |
108 * \param[in] session_handle | 108 * \param[in] sessionHandle |
109 * Windows installer session handle | 109 * Windows installer session handle |
110 * | 110 * |
111 * \return | 111 * \return |
112 * An integer interpreted as a custom action return value. | 112 * An integer interpreted as a custom action return value. |
113 * | 113 * |
114 * \sa | 114 * \sa |
115 * - MSDN [Custom Action Return Values](http://msdn.microsoft.com/en-us/librar y/aa368072%28v=vs.85%29.aspx) | 115 * - MSDN [Custom Action Return Values](http://msdn.microsoft.com/en-us/librar y/aa368072%28v=vs.85%29.aspx) |
116 */ | 116 */ |
117 extern "C" UINT __stdcall | 117 extern "C" UINT __stdcall |
118 sandbox(MSIHANDLE session_handle) | 118 sandbox(MSIHANDLE sessionHandle) |
119 { | 119 { |
120 ImmediateSession session(session_handle, "sandbox"); | 120 ImmediateSession session(sessionHandle, "sandbox"); |
121 | 121 |
122 try | 122 try |
123 { | 123 { |
124 session.Log("Sandbox timestamp " __TIMESTAMP__); | 124 session.Log("Sandbox timestamp " __TIMESTAMP__); |
125 log_only_window_handle_in_closer(session); | 125 LogOnlyWindowHandleInCloser(session); |
126 } | 126 } |
127 catch (std::exception& e) | 127 catch (std::exception& e) |
128 { | 128 { |
129 session.LogNoexcept("terminated by exception: " + std::string(e.what())); | 129 session.LogNoexcept("terminated by exception: " + std::string(e.what())); |
130 return ERROR_INSTALL_FAILURE; | 130 return ERROR_INSTALL_FAILURE; |
131 } | 131 } |
132 catch (...) | 132 catch (...) |
133 { | 133 { |
134 session.LogNoexcept("Caught an exception"); | 134 session.LogNoexcept("Caught an exception"); |
135 return ERROR_INSTALL_FAILURE; | 135 return ERROR_INSTALL_FAILURE; |
136 } | 136 } |
137 | 137 |
138 return ERROR_SUCCESS; | 138 return ERROR_SUCCESS; |
139 } | 139 } |
OLD | NEW |