OLD | NEW |
1 /** | 1 /** |
2 * \file close_application.cpp | 2 * \file close_application.cpp |
3 */ | 3 */ |
4 | 4 |
5 #include <algorithm> | 5 #include <algorithm> |
6 | 6 |
7 #include "session.h" | 7 #include "session.h" |
8 #include "property.h" | 8 #include "property.h" |
9 #include "database.h" | 9 #include "database.h" |
10 #include "process.h" | 10 #include "process.h" |
11 #include "interaction.h" | 11 #include "interaction.h" |
12 #include "custom-i18n.h" | 12 #include "custom-i18n.h" |
13 | 13 |
14 const wchar_t * ie_names[] = { L"IExplore.exe" } ; | 14 const wchar_t * browserNames[] = { L"IExplore.exe" } ; |
15 const wchar_t * engine_names[] = { L"AdblockPlusEngine.exe" } ; | 15 const wchar_t * engineNames[] = { L"AdblockPlusEngine.exe" } ; |
16 | 16 |
17 //------------------------------------------------------- | 17 //------------------------------------------------------- |
18 //------------------------------------------------------- | 18 //------------------------------------------------------- |
19 class IE_Closer | 19 class InternetExplorerCloser |
20 { | 20 { |
21 Process_Snapshot snapshot ; | 21 ProcessSnapshot snapshot ; |
22 | 22 |
23 Process_Closer ie_closer ; | 23 ProcessCloser browserCloser ; |
24 | 24 |
25 Process_Closer engine_closer ; | 25 ProcessCloser engineCloser ; |
26 | 26 |
27 public: | 27 public: |
28 IE_Closer() | 28 InternetExplorerCloser() |
29 : snapshot(), ie_closer( snapshot, ie_names), engine_closer( snapshot, engin
e_names ) | 29 : snapshot(), browserCloser( snapshot, browserNames ), engineCloser( snapsho
t, engineNames ) |
30 {} | 30 {} |
31 | 31 |
32 void refresh() | 32 void Refresh() |
33 { | 33 { |
34 snapshot.refresh() ; | 34 snapshot.Refresh() ; |
35 ie_closer.refresh() ; | 35 browserCloser.Refresh() ; |
36 engine_closer.refresh() ; | 36 engineCloser.Refresh() ; |
37 } | 37 } |
38 | 38 |
39 bool is_running() | 39 bool IsRunning() |
40 { | 40 { |
41 return ie_closer.is_running() || engine_closer.is_running() ; | 41 return browserCloser.IsRunning() || engineCloser.IsRunning() ; |
42 } | 42 } |
43 | 43 |
44 bool shut_down() | 44 bool ShutDown() |
45 { | 45 { |
46 if ( ie_closer.is_running() && ! ie_closer.shut_down() ) | 46 if ( browserCloser.IsRunning() && ! browserCloser.ShutDown() ) |
47 { | 47 { |
48 // Assert IE is still running | 48 // Assert IE is still running |
49 // This is after we've tried to shut it down, so we fail | 49 // This is after we've tried to shut it down, so we fail |
50 return false ; | 50 return false ; |
51 } | 51 } |
52 if ( engine_closer.is_running() && ! engine_closer.shut_down() ) | 52 if ( engineCloser.IsRunning() && ! engineCloser.ShutDown() ) |
53 { | 53 { |
54 // Assert the engine is still running | 54 // Assert the engine is still running |
55 // This is after IE has shut down itself and after we've tried to shut dow
n the engine. Whatever. | 55 // This is after IE has shut down itself and after we've tried to shut dow
n the engine. Whatever. |
56 return false ; | 56 return false ; |
57 } | 57 } |
58 return true ; | 58 return true ; |
59 } | 59 } |
60 } ; | 60 } ; |
61 | 61 |
62 | 62 |
63 //------------------------------------------------------- | 63 //------------------------------------------------------- |
64 // abp_close_ie | 64 // AbpCloseIe |
65 //------------------------------------------------------- | 65 //------------------------------------------------------- |
66 /** | 66 /** |
67 * Exposed DLL entry point for custom action. | 67 * Exposed DLL entry point for custom action. |
68 * The function signature matches the calling convention used by Windows Install
er. | 68 * The function signature matches the calling convention used by Windows Install
er. |
69 * | 69 * |
70 * This function supports four policy stances with respect to a running IE proce
ss. | 70 * This function supports four policy stances with respect to a running IE proce
ss. |
71 * | 71 * |
72 * - Allow reboot. | 72 * - Allow reboot. |
73 * IE is running, so what? I'm willing to reboot after installation. | 73 * IE is running, so what? I'm willing to reboot after installation. |
74 * - Avoid reboot passively. | 74 * - Avoid reboot passively. |
75 * I don't want to affect any other running processes, but I don't want to reb
oot. I'll abort the installation. | 75 * I don't want to affect any other running processes, but I don't want to reb
oot. I'll abort the installation. |
76 * - Avoid reboot actively. | 76 * - Avoid reboot actively. |
77 * I want to shut down IE in order to avoid a reboot. | 77 * I want to shut down IE in order to avoid a reboot. |
78 * I'll do it manually when the time is right. | 78 * I'll do it manually when the time is right. |
79 * - Avoid reboot automatically. | 79 * - Avoid reboot automatically. |
80 * I want to shut down IE automatically in order to avoid a reboot. | 80 * I want to shut down IE automatically in order to avoid a reboot. |
81 * | 81 * |
82 * In a silent installation, the default stance is "allow reboot", which is to s
ay, to act like most installers. | 82 * In a silent installation, the default stance is "allow reboot", which is to s
ay, to act like most installers. |
83 * In an interactive installation, the stance is gathered from the user through
dialog boxes. | 83 * In an interactive installation, the stance is gathered from the user through
dialog boxes. |
84 * If the MSI property AVOIDREBOOT is set to one of the values NO, PASSIVE, ACTI
VE, or AUTOMATIC, the policy is set accordingly. | 84 * If the MSI property AVOIDREBOOT is set to one of the values NO, PASSIVE, ACTI
VE, or AUTOMATIC, the policy is set accordingly. |
85 * In a silent installation, this is the only way getting a stance other than th
e default. | 85 * In a silent installation, this is the only way getting a stance other than th
e default. |
86 * In an interactive installation, AVOIDREBOOT skips the dialogs. | 86 * In an interactive installation, AVOIDREBOOT skips the dialogs. |
87 * | 87 * |
88 * \param[in] session_handle | 88 * \param[in] sessionHandle |
89 * Windows installer session handle | 89 * Windows installer session handle |
90 * | 90 * |
91 * \return | 91 * \return |
92 * An integer interpreted as a custom action return value. | 92 * An integer interpreted as a custom action return value. |
93 * | 93 * |
94 * \post | 94 * \post |
95 * + The return value is one of the following: | 95 * + The return value is one of the following: |
96 * - ERROR_INSTALL_USEREXIT if the user cancelled installation. | 96 * - ERROR_INSTALL_USEREXIT if the user cancelled installation. |
97 * - ERROR_INSTALL_FAILURE if something unexpected happened, usually if the
top level try-block caught an exception. | 97 * - ERROR_INSTALL_FAILURE if something unexpected happened, usually if the
top level try-block caught an exception. |
98 * - ERROR_SUCCESS otherwise. | 98 * - ERROR_SUCCESS otherwise. |
99 * + The function performed at least one check that Internet Explorer was runn
ing. | 99 * + The function performed at least one check that Internet Explorer was runn
ing. |
100 * + If ERROR_SUCCESS, the MSI property RUNNINGBROWSER is set and has one of t
he following values: | 100 * + If ERROR_SUCCESS, the MSI property RUNNINGBROWSER is set and has one of t
he following values: |
101 * - 1 if Internet Explorer was running upon the last check. | 101 * - 1 if Internet Explorer was running upon the last check. |
102 * - 0 otherwise. | 102 * - 0 otherwise. |
103 * \post | 103 * \post |
104 * Note that this function cannot provide any assurance that Internet Explorer
stays either stays running or stays not running. | 104 * Note that this function cannot provide any assurance that Internet Explorer
stays either stays running or stays not running. |
105 * | 105 * |
106 * \sa | 106 * \sa |
107 * - MSDN [Custom Action Return Values](http://msdn.microsoft.com/en-us/librar
y/aa368072%28v=vs.85%29.aspx) | 107 * - MSDN [Custom Action Return Values](http://msdn.microsoft.com/en-us/librar
y/aa368072%28v=vs.85%29.aspx) |
108 */ | 108 */ |
109 extern "C" UINT __stdcall | 109 extern "C" UINT __stdcall |
110 abp_close_ie( MSIHANDLE session_handle ) | 110 AbpCloseIe( MSIHANDLE sessionHandle ) |
111 { | 111 { |
112 // Utility typedef to shorten the class name. | 112 // Utility typedef to shorten the class name. |
113 typedef Installer_Message_Box IMB ; | 113 typedef InstallerMessageBox IMB ; |
114 | 114 |
115 /* | 115 /* |
116 * Immediate_Session cannot throw, so it can go outside the try-block. | 116 * ImmediateSession cannot throw, so it can go outside the try-block. |
117 * It's needed in the catch-all block to write an error message to the log. | 117 * It's needed in the catch-all block to write an error message to the log. |
118 */ | 118 */ |
119 Immediate_Session session( session_handle, "abp_close_ie" ) ; | 119 ImmediateSession session( sessionHandle, "AbpCloseIe" ) ; |
120 | 120 |
121 // The body of an entry point function must have a catch-all. | 121 // The body of an entry point function must have a catch-all. |
122 try { | 122 try { |
123 | 123 |
124 // MSI property BROWSERRUNNING is one of the return values of this function. | 124 // MSI property BROWSERRUNNING is one of the return values of this function. |
125 Property browser_running( session, L"BROWSERRUNNING" ) ; | 125 Property browserRunning( session, L"BROWSERRUNNING" ) ; |
126 Property browser_closed( session, L"BROWSERCLOSED" ) ; | 126 Property browserClosed( session, L"BROWSERCLOSED" ) ; |
127 | 127 |
128 // Instantiation of Process_Closer takes a snapshot. | 128 // Instantiation of ProcessCloser takes a snapshot. |
129 IE_Closer iec ; | 129 InternetExplorerCloser iec ; |
130 | 130 |
131 /* | 131 /* |
132 * We take the short path through this function if neither IE nor engine is
not running at the outset. | 132 * We take the short path through this function if neither IE nor engine is
not running at the outset. |
133 */ | 133 */ |
134 if ( ! iec.is_running() ) | 134 if ( ! iec.IsRunning() ) |
135 { | 135 { |
136 browser_running = L"0" ;» // The browser is not running. | 136 browserRunning = L"0" ;» // The browser is not running. |
137 browser_closed = L"0" ;» // We didn't close the browser (and we could
n't have). | 137 browserClosed = L"0" ;» // We didn't close the browser (and we could
n't have). |
138 session.log( "IE not running. No issue with reboot policy." ) ; | 138 session.Log( "IE not running. No issue with reboot policy." ) ; |
139 return ERROR_SUCCESS ; | 139 return ERROR_SUCCESS ; |
140 } | 140 } |
141 | 141 |
142 /* | 142 /* |
143 * As a (potentially) user-driven function, a state machine manages control
flow. | 143 * As a (potentially) user-driven function, a state machine manages control
flow. |
144 * The states are organized around the policy stances. | 144 * The states are organized around the policy stances. |
145 */ | 145 */ |
146 enum Policy_State { | 146 enum PolicyState { |
147 // Non-terminal states | 147 // Non-terminal states |
148 not_known,» // We don't know the user's stance at all | 148 notKnown,» // We don't know the user's stance at all |
149 part_known,» // The user has indicated either ACTIVE or AUTOMATIC | 149 partKnown,» // The user has indicated either ACTIVE or AUTOMATIC |
150 active, // Actively avoid reboot | 150 active, // Actively avoid reboot |
151 automatic, // Automatically avoid reboot | 151 automatic, // Automatically avoid reboot |
152 // Terminal states | 152 // Terminal states |
153 success, | 153 success, |
154 abort, | 154 abort, |
155 // Aliases for what would ordinarily be non-terminal states. | 155 // Aliases for what would ordinarily be non-terminal states. |
156 // They're terminal because of implementation details. | 156 // They're terminal because of implementation details. |
157 allow = success, // Allow reboot. | 157 allow = success, // Allow reboot. |
158 passive = abort, // Passively avoid reboot, that is, don't try to close
IE. | 158 passive = abort, // Passively avoid reboot, that is, don't try to close
IE. |
159 }; | 159 }; |
160 Policy_State state ; | 160 PolicyState state ; |
161 | 161 |
162 /* | 162 /* |
163 * Use the AVOIDREBOOT property, if present, to set an initial state. | 163 * Use the AVOIDREBOOT property, if present, to set an initial state. |
164 */ | 164 */ |
165 std::wstring avoid_reboot = Property( session, L"AVOIDREBOOT" ) ; | 165 std::wstring avoidReboot = Property( session, L"AVOIDREBOOT" ) ; |
166 std::transform( avoid_reboot.begin(), avoid_reboot.end(), avoid_reboot.begin
(), ::towupper ) ; | 166 std::transform( avoidReboot.begin(), avoidReboot.end(), avoidReboot.begin(),
::towupper ) ; |
167 if ( avoid_reboot == L"" ) | 167 if ( avoidReboot == L"" ) |
168 { | 168 { |
169 state = not_known ; | 169 state = notKnown ; |
170 } | 170 } |
171 else if ( avoid_reboot == L"NO" ) | 171 else if ( avoidReboot == L"NO" ) |
172 { | 172 { |
173 state = allow ; | 173 state = allow ; |
174 session.log( "Reboot allowed on command line." ) ; | 174 session.Log( "Reboot allowed on command line." ) ; |
175 } | 175 } |
176 else if ( avoid_reboot == L"PASSIVE" ) | 176 else if ( avoidReboot == L"PASSIVE" ) |
177 { | 177 { |
178 state = passive ; | 178 state = passive ; |
179 session.log( "Reboot avoided on command line." ) ; | 179 session.Log( "Reboot avoided on command line." ) ; |
180 } | 180 } |
181 else if ( avoid_reboot == L"ACTIVE" ) | 181 else if ( avoidReboot == L"ACTIVE" ) |
182 { | 182 { |
183 state = active ; | 183 state = active ; |
184 } | 184 } |
185 else if ( avoid_reboot == L"AUTOMATIC" ) | 185 else if ( avoidReboot == L"AUTOMATIC" ) |
186 { | 186 { |
187 state = automatic ; | 187 state = automatic ; |
188 } | 188 } |
189 else | 189 else |
190 { | 190 { |
191 // It's an error to specify an unrecognized value for AVOIDREBOOT. | 191 // It's an error to specify an unrecognized value for AVOIDREBOOT. |
192 throw std::runtime_error( "unrecognized value for AVOIDREBOOT" ) ; | 192 throw std::runtime_error( "unrecognized value for AVOIDREBOOT" ) ; |
193 } | 193 } |
194 | 194 |
195 /* | 195 /* |
196 * When running as an update (see Updater.cpp), this installer receives the
command line option "/qb", | 196 * When running as an update (see Updater.cpp), this installer receives the
command line option "/qb", |
197 * which sets the UI level to "Basic UI". | 197 * which sets the UI level to "Basic UI". |
198 * When running as an initial install, we cannot expect what command line op
tions this installer receives. | 198 * When running as an initial install, we cannot expect what command line op
tions this installer receives. |
199 */ | 199 */ |
200 /* | 200 /* |
201 * The UILevel property indicates whether we have the ability to put dialog
boxes up. | 201 * The UILevel property indicates whether we have the ability to put dialog
boxes up. |
202 * Levels 2 (silent) and 3 (basic) do not have this ability. | 202 * Levels 2 (silent) and 3 (basic) do not have this ability. |
203 * Levels 4 (reduced) and 5 (full) do. | 203 * Levels 4 (reduced) and 5 (full) do. |
204 * | 204 * |
205 * MSDN [UILevel property](http://msdn.microsoft.com/en-us/library/windows/d
esktop/aa372096%28v=vs.85%29.aspx) | 205 * MSDN [UILevel property](http://msdn.microsoft.com/en-us/library/windows/d
esktop/aa372096%28v=vs.85%29.aspx) |
206 */ | 206 */ |
207 std::wstring uilevel = Property( session, L"UILevel" ) ; | 207 std::wstring uilevel = Property( session, L"UILevel" ) ; |
208 bool interactive ; | 208 bool interactive ; |
209 if ( uilevel == L"5" || uilevel == L"4" ) | 209 if ( uilevel == L"5" || uilevel == L"4" ) |
210 { | 210 { |
211 interactive = true ; | 211 interactive = true ; |
212 // Assert state is one of { not_known, allow, passive, active, automatic } | 212 // Assert state is one of { notKnown, allow, passive, active, automatic } |
213 } | 213 } |
214 else if ( uilevel == L"3" || uilevel == L"2" ) | 214 else if ( uilevel == L"3" || uilevel == L"2" ) |
215 { | 215 { |
216 // Assert installer is running without user interaction. | 216 // Assert installer is running without user interaction. |
217 interactive = false ; | 217 interactive = false ; |
218 if ( state == not_known ) | 218 if ( state == notKnown ) |
219 { | 219 { |
220 // Assert AVOIDREBOOT was not specified | 220 // Assert AVOIDREBOOT was not specified |
221 /* | 221 /* |
222 * This is where we specify default behavior for non-interactive operati
on. | 222 * This is where we specify default behavior for non-interactive operati
on. |
223 * The choice of "allow" makes it act like other installers, which is to
make no effort to avoid a reboot after installation. | 223 * The choice of "allow" makes it act like other installers, which is to
make no effort to avoid a reboot after installation. |
224 */ | 224 */ |
225 state = allow ; | 225 state = allow ; |
226 » session.log( "Reboot allowed by default in non-interactive session." ) ; | 226 » session.Log( "Reboot allowed by default in non-interactive session." ) ; |
227 } | 227 } |
228 else if ( state == active ) | 228 else if ( state == active ) |
229 { | 229 { |
230 throw std::runtime_error( "AVOIDREBOOT=ACTIVE in non-interative session
is not consistent" ) ; | 230 throw std::runtime_error( "AVOIDREBOOT=ACTIVE in non-interative session
is not consistent" ) ; |
231 } | 231 } |
232 // Assert state is one of { allow, passive, automatic } | 232 // Assert state is one of { allow, passive, automatic } |
233 } | 233 } |
234 else | 234 else |
235 { | 235 { |
236 throw std::runtime_error( "unrecognized value for UILevel" ) ; | 236 throw std::runtime_error( "unrecognized value for UILevel" ) ; |
237 } | 237 } |
238 | 238 |
239 /* | 239 /* |
240 * Now that preliminaries are over, we set up the accessors for UI text. | 240 * Now that preliminaries are over, we set up the accessors for UI text. |
241 * We only use the object 'message_text' for interactive sessions, but it's
cheap to set up and a hassle to conditionalize. | 241 * We only use the object 'messageText' for interactive sessions, but it's c
heap to set up and a hassle to conditionalize. |
242 * | 242 * |
243 * The key "close_ie" is the component name within the file "close_ie.wxi" t
hat defines rows in the localization table. | 243 * The key "close_ie" is the component name within the file "close_ie.wxi" t
hat defines rows in the localization table. |
244 * The identifiers for the message_text.text() function are defined within t
hat file. | 244 * The identifiers for the messageText.text() function are defined within th
at file. |
245 */ | 245 */ |
246 Installation_Database db( session ) ; | 246 InstallationDatabase db( session ) ; |
247 custom_message_text message_text( db, L"close_ie" ) ; | 247 CustomMessageText messageText( db, L"close_ie" ) ; |
248 | 248 |
249 /* | 249 /* |
250 * State machine: Loop through non-terminal states. | 250 * State machine: Loop through non-terminal states. |
251 * | 251 * |
252 * Loop invariant: IE was running at last check, that is, iec.is_running() w
ould return true. | 252 * Loop invariant: IE was running at last check, that is, iec.IsRunning() wo
uld return true. |
253 */ | 253 */ |
254 while ( state <= automatic ) // "automatic" is the non-terminal sta
te with the highest value | 254 while ( state <= automatic ) // "automatic" is the non-terminal sta
te with the highest value |
255 { | 255 { |
256 switch ( state ) | 256 switch ( state ) |
257 { | 257 { |
258 case not_known: | 258 case notKnown: |
259 /* | 259 /* |
260 * Precondition: interactive session | 260 * Precondition: interactive session |
261 * | 261 * |
262 * Ask the user "Would you like to close IE and avoid reboot?" | 262 * Ask the user "Would you like to close IE and avoid reboot?" |
263 » * Yes -> Close IE somehow. Goto part_known. | 263 » * Yes -> Close IE somehow. Goto partKnown. |
264 * No -> Install with reboot. Goto allow. | 264 * No -> Install with reboot. Goto allow. |
265 * Cancel -> terminate installation. Goto abort. | 265 * Cancel -> terminate installation. Goto abort. |
266 */ | 266 */ |
267 » { | 267 { |
268 » int x = session.write_message( IMB( message_text.text( L"dialog_unknow
n" ), IMB::warning_box, IMB::yes_no_cancel, IMB::default_button_three ) ) ; | 268 int x = session.WriteMessage(IMB( |
| 269 messageText.Text(L"dialog_unknown"), |
| 270 IMB::Box::warning, IMB::ButtonSet::yesNoCancel, IMB::DefaultButton::
three |
| 271 )) ; |
269 switch ( x ) | 272 switch ( x ) |
270 { | 273 { |
271 case IDYES: | 274 case IDYES: |
272 » state = part_known ; | 275 » state = partKnown ; |
273 break ; | 276 break ; |
274 case IDNO: | 277 case IDNO: |
275 state = allow ; | 278 state = allow ; |
276 » session.log( "User chose to allow reboot" ) ; | 279 » session.Log( "User chose to allow reboot" ) ; |
277 break ; | 280 break ; |
278 case IDCANCEL: | 281 case IDCANCEL: |
279 state = abort ; | 282 state = abort ; |
280 » session.log( "User cancelled installation" ) ; | 283 » session.Log( "User cancelled installation" ) ; |
281 break ; | 284 break ; |
282 default: | 285 default: |
283 » throw unexpected_return_value_from_message_box() ; | 286 » throw UnexpectedReturnValueFromMessageBox() ; |
284 } | 287 } |
285 } | 288 } |
286 break ; | 289 break ; |
287 | 290 |
288 case part_known: | 291 case partKnown: |
289 /* | 292 /* |
290 * Precondition: interactive session | 293 * Precondition: interactive session |
291 * | 294 * |
292 * Ask the user "Would you like the installer to close IE for you?" | 295 * Ask the user "Would you like the installer to close IE for you?" |
293 * Yes -> Goto automatic | 296 * Yes -> Goto automatic |
294 * No -> Goto active | 297 * No -> Goto active |
295 » * Cancel -> Goto not_known | 298 » * Cancel -> Goto notKnown |
296 */ | 299 */ |
297 { | 300 { |
298 » int x = session.write_message( IMB( message_text.text( L"dialog_part_k
nown" ), IMB::warning_box, IMB::yes_no_cancel, IMB::default_button_three ) ) ; | 301 » int x = session.WriteMessage(IMB( |
| 302 messageText.Text(L"dialog_part_known"), |
| 303 IMB::Box::warning, IMB::ButtonSet::yesNoCancel, IMB::DefaultButton::
three |
| 304 )) ; |
299 switch ( x ) | 305 switch ( x ) |
300 { | 306 { |
301 case IDYES: | 307 case IDYES: |
302 state = automatic ; | 308 state = automatic ; |
303 break ; | 309 break ; |
304 case IDNO: | 310 case IDNO: |
305 state = active ; | 311 state = active ; |
306 break ; | 312 break ; |
307 case IDCANCEL: | 313 case IDCANCEL: |
308 » state = not_known ; | 314 » state = notKnown ; |
309 break ; | 315 break ; |
310 default: | 316 default: |
311 » throw unexpected_return_value_from_message_box() ; | 317 » throw UnexpectedReturnValueFromMessageBox() ; |
312 } | 318 } |
313 } | 319 } |
314 break ; | 320 break ; |
315 | 321 |
316 case active: | 322 case active: |
317 /* | 323 /* |
318 * Precondition: interactive session | 324 * Precondition: interactive session |
319 * | 325 * |
320 * IE is no longer running -> Goto success | 326 * IE is no longer running -> Goto success |
321 * IE is still running -> | 327 * IE is still running -> |
322 * Ask the user to close IE manually | 328 * Ask the user to close IE manually |
323 * OK -> re-enter this state | 329 * OK -> re-enter this state |
324 » * Cancel -> Goto not_known | 330 » * Cancel -> Goto notKnown |
325 */ | 331 */ |
326 { | 332 { |
327 » int x = session.write_message( IMB( message_text.text( L"dialog_active
_retry" ), IMB::warning_box, IMB::ok_cancel, IMB::default_button_one ) ) ; | 333 » int x = session.WriteMessage(IMB( |
| 334 messageText.Text(L"dialog_active_retry"), |
| 335 IMB::Box::warning, IMB::ButtonSet::okCancel, IMB::DefaultButton::one |
| 336 )) ; |
328 switch ( x ) | 337 switch ( x ) |
329 { | 338 { |
330 case IDOK: | 339 case IDOK: |
331 /* | 340 /* |
332 * Refresh our knowledge of whether IE is running. | 341 * Refresh our knowledge of whether IE is running. |
333 * If it is, we display the dialog again. The state doesn't change,
so we just iterate again. | 342 * If it is, we display the dialog again. The state doesn't change,
so we just iterate again. |
334 * If it's not, then the user has closed IE and we're done. | 343 * If it's not, then the user has closed IE and we're done. |
335 */ | 344 */ |
336 » iec.refresh() ; | 345 » iec.Refresh() ; |
337 » if ( ! iec.is_running() ) | 346 » if ( ! iec.IsRunning() ) |
338 { | 347 { |
339 state = success ; | 348 state = success ; |
340 » session.log( "User shut down IE manually." ) ; | 349 » session.Log( "User shut down IE manually." ) ; |
341 } | 350 } |
342 break ; | 351 break ; |
343 case IDCANCEL: | 352 case IDCANCEL: |
344 » state = not_known ; | 353 » state = notKnown ; |
345 break ; | 354 break ; |
346 default: | 355 default: |
347 » throw unexpected_return_value_from_message_box() ; | 356 » throw UnexpectedReturnValueFromMessageBox() ; |
348 } | 357 } |
349 } | 358 } |
350 break ; | 359 break ; |
351 | 360 |
352 case automatic: | 361 case automatic: |
353 /* | 362 /* |
354 * Close all known IE instances. | 363 * Close all known IE instances. |
355 * Unlike other cases, this state starts with an action and not a user q
uery. | 364 * Unlike other cases, this state starts with an action and not a user q
uery. |
356 * We first shut down IE, or at least attempt to. | 365 * We first shut down IE, or at least attempt to. |
357 * | 366 * |
358 * Succeeded -> Goto success | 367 * Succeeded -> Goto success |
359 * Failed && interactive -> | 368 * Failed && interactive -> |
360 * Ask user if they would like to try again | 369 * Ask user if they would like to try again |
361 * Retry -> re-enter this state | 370 * Retry -> re-enter this state |
362 » * Cancel -> Goto not_known | 371 » * Cancel -> Goto notKnown |
363 * Failed && not interactive -> Goto abort | 372 * Failed && not interactive -> Goto abort |
364 */ | 373 */ |
365 { | 374 { |
366 » bool IE_was_closed = iec.shut_down() ; | 375 » bool ieWasClosed = iec.ShutDown() ; |
367 » if ( iec.is_running() ) | 376 » if ( iec.IsRunning() ) |
368 { | 377 { |
369 » session.log( "Attempt to shut down IE automatically failed." ) ; | 378 » session.Log( "Attempt to shut down IE automatically failed." ) ; |
370 if ( interactive ) | 379 if ( interactive ) |
371 { | 380 { |
372 // Assert Interactive session and IE did not shut down. | 381 // Assert Interactive session and IE did not shut down. |
373 » int x = session.write_message( IMB( message_text.text( L"dialog_au
tomatic_retry" ), IMB::warning_box, IMB::retry_cancel, IMB::default_button_one )
) ; | 382 » int x = session.WriteMessage(IMB( |
| 383 messageText.Text(L"dialog_automatic_retry"), |
| 384 IMB::Box::warning, IMB::ButtonSet::retryCancel, IMB::DefaultButt
on::one |
| 385 )) ; |
374 switch ( x ) | 386 switch ( x ) |
375 { | 387 { |
376 case IDRETRY: | 388 case IDRETRY: |
377 // Don't change the state. Iterate again. | 389 // Don't change the state. Iterate again. |
378 break ; | 390 break ; |
379 case IDCANCEL: | 391 case IDCANCEL: |
380 » » state = not_known ; | 392 » » state = notKnown ; |
381 break ; | 393 break ; |
382 default: | 394 default: |
383 » » throw unexpected_return_value_from_message_box() ; | 395 » » throw UnexpectedReturnValueFromMessageBox() ; |
384 } | 396 } |
385 } | 397 } |
386 else | 398 else |
387 { | 399 { |
388 // Assert Non-interactive session and IE did not shut down. | 400 // Assert Non-interactive session and IE did not shut down. |
389 state = abort ; | 401 state = abort ; |
390 » session.log( "Failed to shut down IE automatically." ) ; | 402 » session.Log( "Failed to shut down IE automatically." ) ; |
391 } | 403 } |
392 } | 404 } |
393 else | 405 else |
394 { | 406 { |
395 » // Assert IE is not running, so shut_down() succeeded. | 407 » // Assert IE is not running, so ShutDown() succeeded. |
396 state = success ; | 408 state = success ; |
397 » session.log( "Automatically shut down IE." ) ; | 409 » session.Log( "Automatically shut down IE." ) ; |
398 } | 410 } |
399 } | 411 } |
400 break; | 412 break; |
401 } | 413 } |
402 } | 414 } |
403 /* | 415 /* |
404 * State machine: Actions for terminal states. | 416 * State machine: Actions for terminal states. |
405 */ | 417 */ |
406 switch ( state ) | 418 switch ( state ) |
407 { | 419 { |
408 case success: | 420 case success: |
409 » if ( iec.is_running() ) | 421 » if ( iec.IsRunning() ) |
410 { | 422 { |
411 » browser_running = L"1" ; | 423 » browserRunning = L"1" ; |
412 » browser_closed = L"0" ; | 424 » browserClosed = L"0" ; |
413 } | 425 } |
414 else | 426 else |
415 { | 427 { |
416 » browser_running = L"0" ; | 428 » browserRunning = L"0" ; |
417 » browser_closed = L"1" ; | 429 » browserClosed = L"1" ; |
418 } | 430 } |
419 return ERROR_SUCCESS ; | 431 return ERROR_SUCCESS ; |
420 break; | 432 break; |
421 case abort: | 433 case abort: |
422 return ERROR_INSTALL_USEREXIT ; | 434 return ERROR_INSTALL_USEREXIT ; |
423 break; | 435 break; |
424 } | 436 } |
425 } | 437 } |
426 catch( std::exception & e ) | 438 catch( std::exception & e ) |
427 { | 439 { |
428 session.log_noexcept( "terminated by exception: " + std::string( e.what() )
) ; | 440 session.LogNoexcept( "terminated by exception: " + std::string( e.what() ) )
; |
429 return ERROR_INSTALL_FAILURE ; | 441 return ERROR_INSTALL_FAILURE ; |
430 } | 442 } |
431 catch( ... ) | 443 catch( ... ) |
432 { | 444 { |
433 session.log_noexcept( "terminated by unknown exception" ) ; | 445 session.LogNoexcept( "terminated by unknown exception" ) ; |
434 return ERROR_INSTALL_FAILURE ; | 446 return ERROR_INSTALL_FAILURE ; |
435 } | 447 } |
436 // Should be unreachable. | 448 // Should be unreachable. |
437 return ERROR_INSTALL_FAILURE ; | 449 return ERROR_INSTALL_FAILURE ; |
438 } | 450 } |
439 | |
440 /* | |
441 * EnumWindows system call: http://msdn.microsoft.com/en-us/library/windows/desk
top/ms633497%28v=vs.85%29.aspx | |
442 */ | |
443 /** | |
444 * | |
445 * Callback function for EnumWindows. | |
446 */ | |
447 BOOL CALLBACK EnumWindowsProc(HWND hwnd, LPARAM lParam) | |
448 { | |
449 return TRUE ; | |
450 } | |
451 | |
452 /** | |
453 * Windows_List | |
454 */ | |
455 class Window_List { | |
456 public: | |
457 void enumerate_top_level(); | |
458 }; | |
OLD | NEW |