| 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 // log_all_window_handles |
| 24 //------------------------------------------------------- | 24 //------------------------------------------------------- |
| 25 class log_single_window_handle | 25 class log_single_window_handle |
| 26 { | 26 { |
| 27 ImmediateSession & session ; | 27 ImmediateSession& session; |
| 28 | 28 |
| 29 public: | 29 public: |
| 30 log_single_window_handle( ImmediateSession & session ) | 30 log_single_window_handle(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 log_all_window_handles(ImmediateSession& session) |
| 45 { | 45 { |
| 46 session.Log( "log_all_window_handles" ) ; | 46 session.Log("log_all_window_handles"); |
| 47 log_single_window_handle lp( session ) ; | 47 log_single_window_handle lp(session); |
| 48 EnumerateWindows( lp ) ; | 48 EnumerateWindows(lp); |
| 49 } | 49 } |
| 50 | 50 |
| 51 //------------------------------------------------------- | 51 //------------------------------------------------------- |
| 52 // log_IE_window_handles | 52 // log_IE_window_handles |
| 53 //------------------------------------------------------- | 53 //------------------------------------------------------- |
| 54 class log_single_window_handle_only_if_IE | 54 class log_single_window_handle_only_if_IE |
| 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 log_single_window_handle_only_if_IE(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 log_IE_window_handles(ImmediateSession& session) |
| 80 { | 80 { |
| 81 session.Log( "log_IE_window_handles" ) ; | 81 session.Log("log_IE_window_handles"); |
| 82 const wchar_t * IE_names[] = { L"IExplore.exe", L"AdblockPlusEngine.exe" } ; | 82 const wchar_t* IE_names[] = { L"IExplore.exe", L"AdblockPlusEngine.exe" }; |
| 83 ProcessSnapshot snapshot ; | 83 ProcessSnapshot snapshot; |
| 84 ProcessCloser iec(snapshot, IE_names) ; | 84 ProcessCloser iec(snapshot, IE_names); |
| 85 log_single_window_handle_only_if_IE lp( session, iec ) ; | 85 log_single_window_handle_only_if_IE lp(session, iec); |
| 86 EnumerateWindows( lp ) ; | 86 EnumerateWindows(lp); |
| 87 } | 87 } |
| 88 | 88 |
| 89 //------------------------------------------------------- | 89 //------------------------------------------------------- |
| 90 // log_only_window_handle_in_closer | 90 // log_only_window_handle_in_closer |
| 91 //------------------------------------------------------- | 91 //------------------------------------------------------- |
| 92 void log_only_window_handle_in_closer( ImmediateSession & session ) | 92 void log_only_window_handle_in_closer(ImmediateSession& session) |
| 93 { | 93 { |
| 94 session.Log( "log_only_window_handle_in_closer" ) ; | 94 session.Log("log_only_window_handle_in_closer"); |
| 95 const wchar_t * IE_names[] = { L"IExplore.exe", L"AdblockPlusEngine.exe" } ; | 95 const wchar_t* IE_names[] = { L"IExplore.exe", L"AdblockPlusEngine.exe" }; |
| 96 ProcessSnapshot snapshot ; | 96 ProcessSnapshot snapshot; |
| 97 ProcessCloser iec( snapshot, IE_names) ; | 97 ProcessCloser iec(snapshot, IE_names); |
| 98 iec.IterateOurWindows( log_single_window_handle( session ) ) ; | 98 iec.IterateOurWindows(log_single_window_handle(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] session_handle |
| 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 session_handle) |
| 119 { | 119 { |
| 120 ImmediateSession session( session_handle, "sandbox" ) ; | 120 ImmediateSession session(session_handle, "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 log_only_window_handle_in_closer(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 } |
| 140 | |
| OLD | NEW |