| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #ifndef SANDBOX_SANDBOX_POC_MAIN_UI_WINDOW_H__ | |
| 6 #define SANDBOX_SANDBOX_POC_MAIN_UI_WINDOW_H__ | |
| 7 | |
| 8 #include <string> | |
| 9 | |
| 10 #include "base/basictypes.h" | |
| 11 | |
| 12 namespace sandbox { | |
| 13 class BrokerServices; | |
| 14 enum ResultCode; | |
| 15 } | |
| 16 | |
| 17 // Header file for the MainUIWindow, a simple window with a menu bar that | |
| 18 // can pop up a dialog (accessible through the menu bar), to specify a path and | |
| 19 // filename of any DLL to load and choose an entry point of his/her choice | |
| 20 // (note: only entry points with no parameters are expected to work). | |
| 21 // | |
| 22 // The purpose of this is to be able to spawn an EXE inside a SandBox, have it | |
| 23 // load a DLL and call the entry point on it to test how it behaves inside the | |
| 24 // sandbox. This is useful for developer debugging and for security testing. | |
| 25 // | |
| 26 // The MainUIWindow also has a listview that displays debugging information to | |
| 27 // the user. | |
| 28 // | |
| 29 // Sample usage: | |
| 30 // | |
| 31 // MainUIWindow window; | |
| 32 // unsigned int ret = window.CreateMainWindowAndLoop( | |
| 33 // handle_to_current_instance, | |
| 34 // ::GetCommandLineW(), | |
| 35 // show_command, | |
| 36 // broker); | |
| 37 // | |
| 38 // The CreateMainWindowAndLoop() contains a message loop that ends when the | |
| 39 // user closes the MainUIWindow. | |
| 40 | |
| 41 // This class encapsulates the Main UI window for the broker application. | |
| 42 // It simply shows a menu that gives the user the ability (through a dialog) to | |
| 43 // specify a DLL and what entry point to call in the DLL. | |
| 44 class MainUIWindow { | |
| 45 public: | |
| 46 MainUIWindow(); | |
| 47 ~MainUIWindow(); | |
| 48 | |
| 49 // Creates the main window, displays it and starts the message pump. This | |
| 50 // call will not return until user closes the main UI window that appears | |
| 51 // as a result. Arguments 'instance', 'command_line' and 'show_cmd' can be | |
| 52 // passed in directly from winmain. The 'broker' argument is a pointer to a | |
| 53 // BrokerService that will launch a new EXE inside the sandbox and load the | |
| 54 // DLL of the user's choice. | |
| 55 unsigned int CreateMainWindowAndLoop(HINSTANCE instance, | |
| 56 wchar_t* command_line, | |
| 57 int show_command, | |
| 58 sandbox::BrokerServices* broker); | |
| 59 | |
| 60 private: | |
| 61 // The default value DLL name to add to the edit box. | |
| 62 static const wchar_t kDefaultDll_[]; | |
| 63 | |
| 64 // The default value to show in the entry point. | |
| 65 static const wchar_t kDefaultEntryPoint_[]; | |
| 66 | |
| 67 // The default value to show in the log file. | |
| 68 static const wchar_t kDefaultLogFile_[]; | |
| 69 | |
| 70 // Handles the messages sent to the main UI window. The return value is the | |
| 71 // result of the message processing and depends on the message. | |
| 72 static LRESULT CALLBACK WndProc(HWND window, | |
| 73 UINT message_id, | |
| 74 WPARAM wparam, | |
| 75 LPARAM lparam); | |
| 76 | |
| 77 // Handles the messages sent to the SpawnTarget dialog. The return value is | |
| 78 // the result of the message processing and depends on the message. | |
| 79 static INT_PTR CALLBACK SpawnTargetWndProc(HWND dialog, | |
| 80 UINT message_id, | |
| 81 WPARAM wparam, | |
| 82 LPARAM lparam); | |
| 83 | |
| 84 // Retrieves a pointer to the MainWindow from a value stored along with the | |
| 85 // window handle (passed in as hwnd). Return value is a pointer to the | |
| 86 // MainUIWindow previously stored with SetWindowLong() during WM_CREATE. | |
| 87 static MainUIWindow* FromWindow(HWND main_window); | |
| 88 | |
| 89 // Handles the WM_CREATE message for the main UI window. Returns TRUE on | |
| 90 // success. | |
| 91 static BOOL OnCreate(HWND parent_window, LPCREATESTRUCT); | |
| 92 | |
| 93 // Handles the WM_DESTROY message for the main UI window. | |
| 94 void OnDestroy(HWND window); | |
| 95 | |
| 96 // Handles the WM_SIZE message for the main UI window. | |
| 97 void OnSize(HWND window, UINT state, int cx, int cy); | |
| 98 | |
| 99 // Handles the WM_PAINT message for the main UI window. | |
| 100 void OnPaint(HWND window); | |
| 101 | |
| 102 // Handles the menu command File \ Exit for the main UI window. | |
| 103 void OnFileExit(); | |
| 104 | |
| 105 // Handles the menu command Commands \ Launch for the main UI window. | |
| 106 void OnCommandsLaunch(HWND window); | |
| 107 | |
| 108 // Handles the Launch button in the SpawnTarget dialog (normally clicked | |
| 109 // after selecting DLL and entry point). OnLaunchDll will retrieve the | |
| 110 // values entered by the user and store it in the members of the class. | |
| 111 // Returns true if user selected a non-zero values for DLL filename | |
| 112 // (possibly including path also) and entry point. | |
| 113 bool OnLaunchDll(HWND dialog); | |
| 114 | |
| 115 // Spawns a target EXE inside the sandbox (with the help of the | |
| 116 // BrokerServices passed in to CreateMainWindowAndLoop), and passes to it | |
| 117 // (as command line arguments) the DLL path and the entry point function | |
| 118 // name. The EXE is expected to parse the command line and load the DLL. | |
| 119 // NOTE: The broker does not know if the target EXE successfully loaded the | |
| 120 // DLL, for that you have to rely on the EXE providing a log. | |
| 121 // Returns true if the broker reports that it was successful in creating | |
| 122 // the target and false if not. | |
| 123 bool SpawnTarget(); | |
| 124 | |
| 125 // Shows a standard File Open dialog and returns the DLL filename selected or | |
| 126 // blank string if the user cancelled (or an error occurred). | |
| 127 std::wstring OnShowBrowseForDllDlg(HWND owner); | |
| 128 | |
| 129 // Shows a standard Save As dialog and returns the log filename selected or | |
| 130 // blank string if the user cancelled (or an error occurred). | |
| 131 std::wstring OnShowBrowseForLogFileDlg(HWND owner); | |
| 132 | |
| 133 // Formats a message using the supplied format string and prints it in the | |
| 134 // listview in the main UI window. Passing a NULL param in 'fmt' results in | |
| 135 // no action being performed. Maximum message length is 1K. | |
| 136 void AddDebugMessage(const wchar_t* format, ...); | |
| 137 | |
| 138 // Assists AddDebugMessage in displaying a message in the ListView. It | |
| 139 // simply wraps ListView_InsertItem to insert a debugging message to the | |
| 140 // top of the list view. Passing a NULL param in 'fmt' results in no action | |
| 141 // being performed. | |
| 142 void InsertLineInListView(wchar_t* debug_message); | |
| 143 | |
| 144 // Calls ListenPipe using the class instance received in parameter. This is | |
| 145 // used to create new threads executing ListenPipe | |
| 146 static DWORD WINAPI ListenPipeThunk(void *param); | |
| 147 | |
| 148 // Calls WaitForTargetThunk using the class instance received in parameter | |
| 149 // This is used to create new threads executing WaitForTarget. | |
| 150 static DWORD WINAPI WaitForTargetThunk(void *param); | |
| 151 | |
| 152 // Listens on a pipe and output the data received to a file and to the UI. | |
| 153 DWORD ListenPipe(); | |
| 154 | |
| 155 // Waits for the target to dies and display a message in the UI. | |
| 156 DWORD WaitForTarget(); | |
| 157 | |
| 158 // The BrokerServices will be used to spawn an EXE in a sandbox and ask | |
| 159 // it to load a DLL. | |
| 160 sandbox::BrokerServices* broker_; | |
| 161 | |
| 162 // Contains the information about the running target. | |
| 163 PROCESS_INFORMATION target_; | |
| 164 | |
| 165 // This is essentially a command line to a target executable that the | |
| 166 // broker will spawn and ask to load the DLL. | |
| 167 std::wstring spawn_target_; | |
| 168 | |
| 169 // A handle to the current instance of the app. Passed in to this class | |
| 170 // through CreateMainWindowAndLoop. | |
| 171 HINSTANCE instance_handle_; | |
| 172 | |
| 173 // A path to the DLL that the target should load once it executes. | |
| 174 std::wstring dll_path_; | |
| 175 | |
| 176 // The name of the entry point the target should call after it loads the DLL. | |
| 177 std::wstring entry_point_; | |
| 178 | |
| 179 // The name of the log file to use. | |
| 180 std::wstring log_file_; | |
| 181 | |
| 182 // This is a static handle to the list view that fills up the entire main | |
| 183 // UI window. The list view is used to display debugging information to the | |
| 184 // user. | |
| 185 static HWND list_view_; | |
| 186 | |
| 187 // Pipe used to communicate the logs between the target and the broker. | |
| 188 HANDLE pipe_handle_; | |
| 189 | |
| 190 DISALLOW_COPY_AND_ASSIGN(MainUIWindow); | |
| 191 }; | |
| 192 | |
| 193 #endif // SANDBOX_SANDBOX_POC_MAIN_UI_WINDOW_H__ | |
| OLD | NEW |