diff --git a/src/desktopshellrun.cpp b/src/desktopshellrun.cpp index 76c49928..4ef8a9e5 100644 --- a/src/desktopshellrun.cpp +++ b/src/desktopshellrun.cpp @@ -1,341 +1,303 @@ /* Copyright (C) 2014 by Bundesamt für Sicherheit in der Informationstechnik * Copyright (C) 2016 Intevation GmbH * Software engineering by Intevation GmbH * * This file is Free Software under the GNU GPL (v>=2) * and comes with ABSOLUTELY NO WARRANTY! */ #include #include #include #include #include #include #include #include #include #include #include #include "exdll.h" #ifndef INITGUID #define INITGUID #endif /* Some declarations missing in mingw-w64 3.1.0 taken from msdn */ __CRT_UUID_DECL(IShellWindows, 0x85CB6900, 0x4D95, 0x11CF, 0x96, 0x0C, 0x00, 0x80, 0xC7, 0xF4, 0xEE, 0x85); DEFINE_GUID(IID_IShellWindows, 0x85CB6900, 0x4D95, 0x11CF, 0x96, 0x0C, 0x00, 0x80, 0xC7, 0xF4, 0xEE, 0x85); DEFINE_GUID(CLSID_ShellWindows, 0x9BA05972, 0xF6A8, 0x11CF, 0xA4, 0x42, 0x00, 0xA0, 0xC9, 0x0A, 0x8F, 0x39); __CRT_UUID_DECL(IShellDispatch2, 0xA4C6892C, 0x3BA9, 0x11d2, 0x9D, 0xEA, 0x00, 0xC0, 0x4F, 0xB1, 0x61, 0x62); __CRT_UUID_DECL(IShellFolderViewDual, 0xe7a1af80, 0x4d96, 0x11cf, 0x96, 0x0c, 0x00, 0x80, 0xc7, 0xf4, 0xee, 0x85); #ifndef SWC_DESKTOP /* Will probably be addedd in future mingw */ #define SWC_DESKTOP 0x00000008 /* from http://msdn.microsoft.com/en-us/library/windows/desktop/cc836581%28v=vs.85%29.aspx */ #endif #define UNUSED(x) (void)(x) -#define ERRORPRINTF(fmt, ...) \ - { \ - char buf[512]; \ - snprintf(buf, 511, "ERROR: " fmt, ##__VA_ARGS__); \ - buf[511] = '\0'; \ - OutputDebugStringA(buf); \ - } /** @brief the actual execuation call on the shell dispatcher * * @param[in] disp The shell dispatcher to use for shell execute. * @param[in] fName The file that should be exectued. * @param[in] param Optinal parameters to add. * * @returns true on success. */ static bool shellexecute(IShellDispatch2 *disp, wchar_t *fName, wchar_t *param) { BSTR bName = NULL, bParam = NULL, bDir = NULL, bOp = NULL; VARIANT vParams[4]; HRESULT hr; if (!fName || !disp) { ERRORPRINTF ("Invalid call to shellexecute."); return false; } bName = SysAllocString(fName); bParam = SysAllocString(param ? param : L""); bDir = SysAllocString(L""); bOp = SysAllocString(L""); if (!bName || !bParam || !bDir || !bOp) { /* Out of memory */ ERRORPRINTF ("Failed to allocate bstr values "); return false; } vParams[0].vt = VT_BSTR; vParams[0].bstrVal = bParam; vParams[1].vt = VT_BSTR; vParams[1].bstrVal = bDir; vParams[2].vt = VT_BSTR; vParams[2].bstrVal = bOp; vParams[3].vt = VT_INT; vParams[3].intVal = SW_SHOWNORMAL; hr = disp->ShellExecute(bName, vParams[0], vParams[1], vParams[2], vParams[3]); SysFreeString(bName); SysFreeString(bParam); SysFreeString(bOp); SysFreeString(bDir); if (FAILED(hr)) { ERRORPRINTF ("Failed to execute."); return false; } return true; } -wchar_t -*acp_to_wchar (const char *string, size_t len) -{ - int n, ilen; - wchar_t *result; - - ilen = (int) len; - if (ilen < 0) - return NULL; - - n = MultiByteToWideChar (CP_ACP, 0, string, ilen, NULL, 0); - if (n < 0 || n + 1 < 0) - return NULL; - - result = (wchar_t *) malloc ((size_t)(n+1) * sizeof *result); - if (!result) - { - ERRORPRINTF("Out of core"); - exit(1); - } - n = MultiByteToWideChar (CP_ACP, 0, string, ilen, result, n); - if (n < 0) - { - if (result) - free (result); - return NULL; - } - result[n] = 0; - return result; -} - #ifdef __cplusplus extern "C" { #endif /** @brief Execute a command with the current running shell. * * This function is intended to be called when you want to * make sure that your application is not executed with higher * privileges then the normal desktop session. * * The code is based on the idea: * http://blogs.msdn.com/b/oldnewthing/archive/2013/11/18/10468726.aspx * * The function signature is explained by NSIS. */ void __declspec(dllexport) __cdecl DesktopShellRun(HWND hwndParent, int string_size, char *variables, stack_t **stacktop) { UNUSED(hwndParent); UNUSED(string_size); HRESULT hr; wchar_t *wbuf = NULL, *params = NULL; IShellWindows *shellWindows = NULL; IShellBrowser *shellBrowser = NULL; IShellView *shellView = NULL; IShellFolderViewDual *folderView = NULL; IShellDispatch2 *shellDispatch = NULL; IServiceProvider *serviceProv = NULL; HWND hwnd; IDispatch *disp = NULL, *bgDisp = NULL, *sDisp = NULL; VARIANT vEmpty = {}; if (!stacktop || !*stacktop || !(*stacktop)->text) { ERRORPRINTF ("Invalid call to exec :"); return; } /* Initialize com ctx */ hr = CoInitialize(NULL); if(FAILED(hr)) { ERRORPRINTF ("CoInitializeEx failed. error = 0x%lx.", hr); return; } /* Get the shell interface */ hr = CoCreateInstance(CLSID_ShellWindows, NULL, CLSCTX_LOCAL_SERVER, IID_PPV_ARGS(&shellWindows)); if (FAILED(hr)) { ERRORPRINTF ("Failed to get shell interface."); goto done; } /* Get the desktop shell window */ hr = shellWindows->FindWindowSW(&vEmpty, &vEmpty, SWC_DESKTOP, (long*)&hwnd, SWFO_NEEDDISPATCH, &disp); if (FAILED(hr)) { ERRORPRINTF ("Failed to find the desktop dispatcher."); goto done; } hr = disp->QueryInterface(IID_PPV_ARGS(&serviceProv)); if (FAILED(hr)) { ERRORPRINTF ("Failed to get the service provider."); goto done; } /* Get the shell browser */ hr = serviceProv->QueryService(SID_STopLevelBrowser, IID_PPV_ARGS(&shellBrowser)); if (FAILED(hr)) { ERRORPRINTF ("Failed to find the top level browser."); goto done; } hr = shellBrowser->QueryActiveShellView(&shellView); if (FAILED(hr)) { ERRORPRINTF ("Failed to find the active view."); goto done; } hr = shellView->GetItemObject(SVGIO_BACKGROUND, IID_PPV_ARGS(&bgDisp)); if (FAILED(hr)) { ERRORPRINTF ("Failed to get the views background."); goto done; } hr = bgDisp->QueryInterface(IID_PPV_ARGS(&folderView)); if (FAILED(hr)) { ERRORPRINTF ("Failed to get the folder view."); goto done; } hr = folderView->get_Application(&sDisp); if (FAILED(hr)) { ERRORPRINTF ("Failed to get the shell dispatch."); goto done; } hr = sDisp->QueryInterface(IID_PPV_ARGS(&shellDispatch)); if (FAILED(hr)) { ERRORPRINTF ("Failed to get the shell dispatch interface."); goto done; } /* For unicodensis this has to be utf8 to wchar */ wbuf = acp_to_wchar((*stacktop)->text, strlen((*stacktop)->text)); if (!wbuf) { ERRORPRINTF ("Failed to convert argument to wchar. error = 0x%lx.", hr); goto done; } if ((*stacktop)->next && (*stacktop)->next->text) { params = acp_to_wchar((*stacktop)->next->text, strlen((*stacktop)->next->text)); } if (!shellexecute(shellDispatch, wbuf, params)) { ERRORPRINTF ("Failed to execute."); } if (wbuf) free(wbuf); if (params) free (params); done: if (folderView) { folderView->Release(); } if (disp) { disp->Release(); } if (shellBrowser) { shellBrowser->Release(); } if (shellWindows) { shellWindows->Release(); } if (shellView) { shellView->Release(); } if (sDisp) { sDisp->Release(); } if (shellDispatch) { shellDispatch->Release(); } if (serviceProv) { serviceProv->Release(); } CoUninitialize(); return; } #ifdef __cplusplus } #endif diff --git a/src/exdll.h b/src/exdll.h index a32ed204..0c36f28a 100644 --- a/src/exdll.h +++ b/src/exdll.h @@ -1,151 +1,189 @@ /* exdll.h for use with gpg4win * Copyright (C) 1999-2005 Nullsoft, Inc. * * This license applies to everything in the NSIS package, except * where otherwise noted. * * This software is provided 'as-is', without any express or implied * warranty. In no event will the authors be held liable for any * damages arising from the use of this software. * * Permission is granted to anyone to use this software for any * purpose, including commercial applications, and to alter it and * redistribute it freely, subject to the following restrictions: * * 1. The origin of this software must not be misrepresented; you must * not claim that you wrote the original software. If you use this * software in a product, an acknowledgment in the product * documentation would be appreciated but is not required. * * 2. Altered source versions must be plainly marked as such, and must * not be misrepresented as being the original software. * * 3. This notice may not be removed or altered from any source * distribution. ************************************************************ * 2005-11-14 wk Applied license text to orginal exdll.h file from * NSIS 2.0.4 and did some formatting changes. */ #ifndef _EXDLL_H_ #define _EXDLL_H_ /* only include this file from one place in your DLL. (it is all static, if you use it in two places it will fail) */ #define EXDLL_INIT() { \ g_stringsize=string_size; \ g_stacktop=stacktop; \ g_variables=variables; } /* For page showing plug-ins */ #define WM_NOTIFY_OUTER_NEXT (WM_USER+0x8) #define WM_NOTIFY_CUSTOM_READY (WM_USER+0xd) #define NOTIFY_BYE_BYE 'x' typedef struct _stack_t { struct _stack_t *next; char text[1]; /* This should be the length of string_size. */ } stack_t; static unsigned int g_stringsize; static stack_t **g_stacktop; static char *g_variables; static int __stdcall popstring(char *str, size_t maxlen); /* 0 on success, 1 on empty stack */ static void __stdcall pushstring(const char *str); enum { INST_0, // $0 INST_1, // $1 INST_2, // $2 INST_3, // $3 INST_4, // $4 INST_5, // $5 INST_6, // $6 INST_7, // $7 INST_8, // $8 INST_9, // $9 INST_R0, // $R0 INST_R1, // $R1 INST_R2, // $R2 INST_R3, // $R3 INST_R4, // $R4 INST_R5, // $R5 INST_R6, // $R6 INST_R7, // $R7 INST_R8, // $R8 INST_R9, // $R9 INST_CMDLINE, // $CMDLINE INST_INSTDIR, // $INSTDIR INST_OUTDIR, // $OUTDIR INST_EXEDIR, // $EXEDIR INST_LANG, // $LANGUAGE __INST_LAST }; typedef struct { int autoclose; int all_user_var; int exec_error; int abort; int exec_reboot; int reboot_called; int XXX_cur_insttype; /* deprecated */ int XXX_insttype_changed; /* deprecated */ int silent; int instdir_error; int rtl; int errlvl; } exec_flags_t; typedef struct { exec_flags_t *exec_flags; int (__stdcall *ExecuteCodeSegment)(int, HWND); } extra_parameters_t; /* Utility functions (not required but often useful). */ static int __stdcall popstring(char *str, size_t maxlen) { stack_t *th; if (!g_stacktop || !*g_stacktop) return 1; th=(*g_stacktop); lstrcpyn (str, th->text, maxlen); *g_stacktop = th->next; GlobalFree((HGLOBAL)th); return 0; } static void __stdcall pushstring(const char *str) { stack_t *th; if (!g_stacktop) return; th=(stack_t*)GlobalAlloc(GPTR,sizeof(stack_t)+g_stringsize); lstrcpyn(th->text,str,g_stringsize); th->next=*g_stacktop; *g_stacktop=th; } static char * __stdcall getuservariable(const int varnum) { if (varnum < 0 || varnum >= __INST_LAST) return NULL; return g_variables+varnum*g_stringsize; } static void __stdcall setuservariable(const int varnum, const char *var) { if (var != NULL && varnum >= 0 && varnum < __INST_LAST) lstrcpy(g_variables + varnum*g_stringsize, var); } +#define ERRORPRINTF(fmt, ...) \ + { \ + char buf[512]; \ + snprintf(buf, 511, "ERROR: " fmt, ##__VA_ARGS__); \ + buf[511] = '\0'; \ + OutputDebugStringA(buf); \ + } + +static wchar_t +*acp_to_wchar (const char *string, size_t len) +{ + int n, ilen; + wchar_t *result; + + ilen = (int) len; + if (ilen < 0) + return NULL; + + n = MultiByteToWideChar (CP_ACP, 0, string, ilen, NULL, 0); + if (n < 0 || n + 1 < 0) + return NULL; + + result = (wchar_t *) malloc ((size_t)(n+1) * sizeof *result); + if (!result) + { + ERRORPRINTF("Out of core"); + exit(1); + } + n = MultiByteToWideChar (CP_ACP, 0, string, ilen, result, n); + if (n < 0) + { + if (result) + free (result); + return NULL; + } + result[n] = 0; + return result; +} #endif//_EXDLL_H_ diff --git a/src/g4wihelp.c b/src/g4wihelp.c index 7cb0679e..dd2bc83f 100644 --- a/src/g4wihelp.c +++ b/src/g4wihelp.c @@ -1,1135 +1,1200 @@ /* g4wihelp.c - NSIS Helper DLL used with gpg4win. -*- coding: latin-1; -*- * Copyright (C) 2005 g10 Code GmbH * Copyright (C) 2001 Justin Frankel + * Copyright (C) 2016 Intevation GmbH * * This software is provided 'as-is', without any express or implied * warranty. In no event will the authors be held liable for any * damages arising from the use of this software. * * Permission is granted to anyone to use this software for any * purpose, including commercial applications, and to alter it and * redistribute it freely, subject to the following restrictions: * * 1. The origin of this software must not be misrepresented; you must * not claim that you wrote the original software. If you use this * software in a product, an acknowledgment in the product * documentation would be appreciated but is not required. * * 2. Altered source versions must be plainly marked as such, and must * not be misrepresented as being the original software. * * 3. This notice may not be removed or altered from any source * distribution. ************************************************************ * The code for the splash screen has been taken from the Splash * plugin of the NSIS 2.04 distribution. That code comes without * explicit copyright notices in tyhe source files or author names, it * seems that it has been written by Justin Frankel; not sure about * the year, though. [wk 2005-11-28] */ #include #include +#include +#include #include "exdll.h" static HINSTANCE g_hInstance; /* Our Instance. */ static HWND g_hwndParent; /* Handle of parent window or NULL. */ static HBITMAP g_hbm; /* Handle of the splash image. */ static int sleepint; /* Milliseconds to show the spals image. */ /* Standard entry point for DLLs. */ int WINAPI DllMain (HANDLE hinst, DWORD reason, LPVOID reserved) { if (reason == DLL_PROCESS_ATTACH) g_hInstance = hinst; return TRUE; } /* Dummy function for testing. */ void __declspec(dllexport) dummy (HWND hwndParent, int string_size, char *variables, stack_t **stacktop, extra_parameters_t *extra) { g_hwndParent = hwndParent; EXDLL_INIT(); // note if you want parameters from the stack, pop them off in order. // i.e. if you are called via exdll::myFunction file.dat poop.dat // calling popstring() the first time would give you file.dat, // and the second time would give you poop.dat. // you should empty the stack of your parameters, and ONLY your // parameters. // do your stuff here { char buf[1024]; sprintf(buf,"$R0=%s\r\n$R1=%s\r\n", getuservariable(INST_R0), getuservariable(INST_R1)); MessageBox(g_hwndParent,buf,0,MB_OK); sprintf (buf, "autoclose =%d\r\n" "all_user_var =%d\r\n" "exec_error =%d\r\n" "abort =%d\r\n" "exec_reboot =%d\r\n" "reboot_called=%d\r\n" "silent =%d\r\n" "instdir_error=%d\r\n" "rtl =%d\r\n" "errlvl =%d\r\n", extra->exec_flags->autoclose, extra->exec_flags->all_user_var, extra->exec_flags->exec_error, extra->exec_flags->abort, extra->exec_flags->exec_reboot, extra->exec_flags->reboot_called, extra->exec_flags->silent, extra->exec_flags->instdir_error, extra->exec_flags->rtl, extra->exec_flags->errlvl); MessageBox(g_hwndParent,buf,0,MB_OK); } } void __declspec(dllexport) runonce (HWND hwndParent, int string_size, char *variables, stack_t **stacktop, extra_parameters_t *extra) { const char *result; g_hwndParent = hwndParent; EXDLL_INIT(); CreateMutexA (NULL, 0, getuservariable(INST_R0)); result = GetLastError ()? "1":"0"; setuservariable (INST_R0, result); } void __declspec(dllexport) playsound (HWND hwndParent, int string_size, char *variables, stack_t **stacktop, extra_parameters_t *extra) { char fname[MAX_PATH]; g_hwndParent = hwndParent; EXDLL_INIT(); if (popstring(fname, sizeof fname)) return; PlaySound (fname, NULL, SND_ASYNC|SND_FILENAME|SND_NODEFAULT); } void __declspec(dllexport) stopsound (HWND hwndParent, int string_size, char *variables, stack_t **stacktop, extra_parameters_t *extra) { g_hwndParent = hwndParent; EXDLL_INIT(); PlaySound (NULL, NULL, 0); } /* Windows procedure to control the splashimage. This one pauses the execution until the sleep time is over or the user closes this windows. */ static LRESULT CALLBACK splash_wndproc (HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam) { LRESULT result = 0; switch (uMsg) { case WM_CREATE: { BITMAP bm; RECT vp; GetObject(g_hbm, sizeof(bm), (LPSTR)&bm); SystemParametersInfo(SPI_GETWORKAREA, 0, &vp, 0); SetWindowLong(hwnd,GWL_STYLE,0); SetWindowPos(hwnd,NULL, vp.left+(vp.right-vp.left-bm.bmWidth)/2, vp.top+(vp.bottom-vp.top-bm.bmHeight)/2, bm.bmWidth,bm.bmHeight, SWP_NOZORDER); ShowWindow(hwnd,SW_SHOW); SetTimer(hwnd,1,sleepint,NULL); } break; case WM_PAINT: { PAINTSTRUCT ps; RECT r; HDC curdc=BeginPaint(hwnd,&ps); HDC hdc=CreateCompatibleDC(curdc); HBITMAP oldbm; GetClientRect(hwnd,&r); oldbm=(HBITMAP)SelectObject(hdc,g_hbm); BitBlt(curdc,r.left,r.top,r.right-r.left,r.bottom-r.top, hdc,0,0,SRCCOPY); SelectObject(hdc,oldbm); DeleteDC(hdc); EndPaint(hwnd,&ps); } break; case WM_CLOSE: break; case WM_TIMER: case WM_LBUTTONDOWN: DestroyWindow(hwnd); /*(fall through)*/ default: result = DefWindowProc (hwnd, uMsg, wParam, lParam); } return result; } /* Display a splash screen. Call as g4wihelp::showsplash SLEEP FNAME With SLEEP being the time in milliseconds to show the splashscreen and FNAME the complete filename of the image. As of now only BMP is supported. */ void __declspec(dllexport) showsplash (HWND hwndParent, int string_size, char *variables, stack_t **stacktop, extra_parameters_t *extra) { static WNDCLASS wc; char sleepstr[30]; char fname[MAX_PATH]; int err = 0; char *p; char classname[] = "_sp"; g_hwndParent = hwndParent; EXDLL_INIT(); if (popstring(sleepstr, sizeof sleepstr)) err = 1; if (popstring(fname, sizeof fname)) err = 1; if (err) return; if (!*fname) return; /* Nothing to do. */ for (sleepint=0, p=sleepstr; *p >= '0' && *p <= '9'; p++) { sleepint *= 10; sleepint += *p - '0'; } if (sleepint <= 0) return; /* Nothing to do. */ wc.lpfnWndProc = splash_wndproc; wc.hInstance = g_hInstance; wc.hCursor = LoadCursor(NULL,IDC_ARROW); wc.lpszClassName = classname; if (!RegisterClass(&wc)) return; /* Error. */ g_hbm = LoadImage (NULL, fname, IMAGE_BITMAP, 0, 0 , LR_CREATEDIBSECTION|LR_LOADFROMFILE); if (g_hbm) { MSG msg; HWND hwnd; hwnd = CreateWindowEx (WS_EX_TOOLWINDOW, classname, classname, 0, 0, 0, 0, 0, (HWND)hwndParent, NULL, g_hInstance, NULL); while (IsWindow(hwnd) && GetMessage ( &msg, hwnd, 0, 0)) { DispatchMessage (&msg); } DeleteObject (g_hbm); g_hbm = NULL; } UnregisterClass (classname, g_hInstance); } /* Service Management. */ /* Use this to report unexpected errors. FIXME: This is really not very descriptive. */ void service_error (const char *str) { char buf[1024]; snprintf (buf, sizeof (buf) - 1, "error: %s: ec=%d\r\n", str, GetLastError ()); MessageBox(g_hwndParent, buf, 0, MB_OK); setuservariable (INST_R0, "1"); } void __declspec(dllexport) service_create (HWND hwndParent, int string_size, char *variables, stack_t **stacktop, extra_parameters_t *extra) { SC_HANDLE sc; SC_HANDLE service; const char *result = NULL; char service_name[256]; char display_name[256]; char program[256]; int err = 0; g_hwndParent = hwndParent; EXDLL_INIT(); /* The expected stack layout: service_name, display_name, program. */ if (popstring (service_name, sizeof (service_name))) err = 1; if (!err && popstring (display_name, sizeof (display_name))) err = 1; if (!err && popstring (program, sizeof (program))) err = 1; if (err) { setuservariable (INST_R0, "1"); return; } sc = OpenSCManager (NULL, NULL, SC_MANAGER_ALL_ACCESS); if (sc == NULL) { service_error ("OpenSCManager"); return; } service = CreateService (sc, service_name, display_name, SERVICE_ALL_ACCESS, SERVICE_WIN32_OWN_PROCESS, /* Use SERVICE_DEMAND_START for testing. FIXME: Currently not configurable by caller. */ SERVICE_AUTO_START, SERVICE_ERROR_NORMAL, program, NULL, NULL, NULL, /* FIXME: Currently not configurable by caller. */ /* FIXME: LocalService or NetworkService don't work for dirmngr right now. NOTE! If you change it here, you also should adjust make-msi.pl for the msi installer. In the future, this should be an argument to the function and then the make-msi.pl script can extract it from the invocation. */ NULL /* "NT AUTHORITY\\LocalService" */, NULL); if (service == NULL) { service_error ("CreateService"); CloseServiceHandle (sc); return; } CloseServiceHandle (service); result = GetLastError () ? "1":"0"; setuservariable (INST_R0, result); return; } /* Requires g_hwndParent to be set! */ SC_HANDLE service_lookup (char *service_name) { SC_HANDLE sc; SC_HANDLE service; sc = OpenSCManager (NULL, NULL, SC_MANAGER_ALL_ACCESS); if (sc == NULL) { service_error ("OpenSCManager"); return NULL; } service = OpenService (sc, service_name, SC_MANAGER_ALL_ACCESS); if (service == NULL) { /* Fail silently here. */ CloseServiceHandle (sc); return NULL; } CloseServiceHandle (sc); return service; } /* Returns status. */ void __declspec(dllexport) service_query (HWND hwndParent, int string_size, char *variables, stack_t **stacktop, extra_parameters_t *extra) { SC_HANDLE service; const char *result = NULL; char service_name[256]; int err = 0; SERVICE_STATUS status; g_hwndParent = hwndParent; EXDLL_INIT(); /* The expected stack layout: service_name argc [argv]. */ if (popstring (service_name, sizeof (service_name))) err = 1; if (err) { setuservariable (INST_R0, "ERROR"); return; } service = service_lookup (service_name); if (service == NULL) if (err == 0) { setuservariable (INST_R0, "MISSING"); return; } err = QueryServiceStatus (service, &status); if (err == 0) { setuservariable (INST_R0, "ERROR"); CloseServiceHandle (service); return; } CloseServiceHandle (service); switch (status.dwCurrentState) { case SERVICE_START_PENDING: result = "START_PENDING"; break; case SERVICE_RUNNING: result = "RUNNING"; break; case SERVICE_PAUSE_PENDING: result = "PAUSE_PENDING"; break; case SERVICE_PAUSED: result = "PAUSED"; break; case SERVICE_CONTINUE_PENDING: result = "CONTINUE_PENDING"; break; case SERVICE_STOP_PENDING: result = "STOP_PENDING"; break; case SERVICE_STOPPED: result = "STOPPED"; break; default: result = "UNKNOWN"; } setuservariable (INST_R0, result); return; } void __declspec(dllexport) service_start (HWND hwndParent, int string_size, char *variables, stack_t **stacktop, extra_parameters_t *extra) { SC_HANDLE service; const char *result = NULL; char service_name[256]; char argc_str[256]; #define NR_ARGS 10 #define ARG_MAX 256 char argv_str[NR_ARGS][ARG_MAX]; char *argv[NR_ARGS + 1]; int argc; int i; int err = 0; g_hwndParent = hwndParent; EXDLL_INIT(); /* The expected stack layout: service_name argc [argv]. */ if (popstring (service_name, sizeof (service_name))) err = 1; if (!err && popstring (argc_str, sizeof (argc_str))) err = 1; if (!err) { argc = atoi (argc_str); for (i = 0; i < argc; i++) { if (popstring (argv_str[i], ARG_MAX)) { err = 1; break; } argv[i] = argv_str[i]; } argv[i] = NULL; } if (err) { setuservariable (INST_R0, "1"); return; } service = service_lookup (service_name); if (service == NULL) return; err = StartService (service, argc, argc == 0 ? NULL : argv); if (err == 0) { service_error ("StartService"); CloseServiceHandle (service); return; } CloseServiceHandle (service); setuservariable (INST_R0, "0"); return; } void __declspec(dllexport) service_stop (HWND hwndParent, int string_size, char *variables, stack_t **stacktop, extra_parameters_t *extra) { SC_HANDLE service; const char *result = NULL; char service_name[256]; int err = 0; SERVICE_STATUS status; DWORD timeout = 10000; /* 10 seconds. */ DWORD start_time; g_hwndParent = hwndParent; EXDLL_INIT(); /* The expected stack layout: service_name argc [argv]. */ if (popstring (service_name, sizeof (service_name))) err = 1; if (err) { setuservariable (INST_R0, "1"); return; } service = service_lookup (service_name); if (service == NULL) return; err = QueryServiceStatus (service, &status); if (err == 0) { service_error ("QueryService"); CloseServiceHandle (service); return; } if (status.dwCurrentState != SERVICE_STOPPED && status.dwCurrentState != SERVICE_STOP_PENDING) { err = ControlService (service, SERVICE_CONTROL_STOP, &status); if (err == 0) { service_error ("ControlService"); CloseServiceHandle (service); return; } } start_time = GetTickCount (); while (status.dwCurrentState != SERVICE_STOPPED) { Sleep (1000); /* One second. */ if (!QueryServiceStatus (service, &status)) { service_error ("QueryService"); CloseServiceHandle (service); return; } if (status.dwCurrentState == SERVICE_STOPPED) break; if (GetTickCount () - start_time > timeout) { char buf[1024]; snprintf (buf, sizeof (buf) - 1, "time out waiting for service %s to stop\r\n", service_name); MessageBox (g_hwndParent, buf, 0, MB_OK); setuservariable (INST_R0, "1"); return; } } CloseServiceHandle (service); setuservariable (INST_R0, "0"); return; } void __declspec(dllexport) service_delete (HWND hwndParent, int string_size, char *variables, stack_t **stacktop, extra_parameters_t *extra) { SC_HANDLE service; const char *result = NULL; char service_name[256]; int err = 0; g_hwndParent = hwndParent; EXDLL_INIT(); /* The expected stack layout: service_name argc [argv]. */ if (popstring (service_name, sizeof (service_name))) err = 1; if (err) { setuservariable (INST_R0, "1"); return; } service = service_lookup (service_name); if (service == NULL) return; err = DeleteService (service); if (err == 0) { service_error ("DeleteService"); CloseServiceHandle (service); return; } CloseServiceHandle (service); setuservariable (INST_R0, "0"); return; } #include /* Extract config file parameters. FIXME: Not particularly robust. We expect some reasonable formatting. The parser below is very limited. It expects a command line option /c=FILE or /C=FILE, where FILE must be enclosed in double-quotes if it contains spaces. That file should contain a single section [gpg4win] and KEY=VALUE pairs for each additional configuration file to install. Comments are supported only on lines by themselves. VALUE can be quoted in double-quotes, but does not need to be, unless it has whitespace at the beginning or end. KEY can, for example, be "gpg.conf" (without the quotes). */ void config_init (char **keys, char **values, int max) { /* First, parse the command line. */ char *cmdline; char *begin = NULL; char *end = NULL; char mark; char *fname; char *ptr; FILE *conf; *keys = NULL; *values = NULL; cmdline = getuservariable (INST_CMDLINE); mark = (*cmdline == '"') ? (cmdline++, '"') : ' '; while (*cmdline && *cmdline != mark) cmdline++; if (mark == '"' && *cmdline) cmdline++; while (*cmdline && *cmdline == ' ') cmdline++; while (*cmdline) { /* We are at the beginning of a new argument. */ if (cmdline[0] == '/' && (cmdline[1] == 'C' || cmdline[1] == 'c') && cmdline[2] == '=') { cmdline += 3; begin = cmdline; } while (*cmdline && *cmdline != ' ') { /* Skip over quoted parts. */ if (*cmdline == '"') { cmdline++; while (*cmdline && *cmdline != '"') cmdline++; if (*cmdline) cmdline++; } else cmdline++; } if (begin && !end) { end = cmdline - 1; break; } while (*cmdline && *cmdline == ' ') cmdline++; } if (!begin || begin > end) return; /* Strip quotes. */ if (*begin == '"' && *end == '"') { begin++; end--; } if (begin > end) return; fname = malloc (end - begin + 2); if (!fname) return; ptr = fname; while (begin <= end) *(ptr++) = *(begin++); *ptr = '\0'; conf = fopen (fname, "r"); free (fname); if (!conf) return; while (max - 1 > 0) { char line[256]; char *ptr2; if (fgets (line, sizeof (line), conf) == NULL) break; ptr = &line[strlen (line)]; while (ptr > line && (ptr[-1] == '\n' || ptr[-1] == '\r' || ptr[-1] == ' ' || ptr[-1] == '\t')) ptr--; *ptr = '\0'; ptr = line; while (*ptr && (*ptr == ' ' || *ptr == '\t')) ptr++; /* Ignore comment lines. */ /* FIXME: Ignore section markers. */ if (*ptr == '\0' || *ptr == ';' || *ptr == '[') continue; begin = ptr; while (*ptr && *ptr != '=' && *ptr != ' ' && *ptr != '\t') ptr++; end = ptr - 1; while (*ptr && (*ptr == ' ' || *ptr == '\t')) ptr++; if (*ptr != '=') continue; ptr++; if (begin > end) continue; /* We found a key. */ *keys = malloc (end - begin + 2); if (!keys) return; ptr2 = *keys; while (begin <= end) *(ptr2++) = *(begin++); *ptr2 = '\0'; *values = NULL; while (*ptr && (*ptr == ' ' || *ptr == '\t')) ptr++; begin = ptr; /* In this case, end points to the byte after the value, which is OK because that is '\0'. */ end = &line[strlen (line)]; if (begin > end) begin = end; /* Strip quotes. */ if (*begin == '"' && end[-1] == '"') { begin++; end--; *end = '\0'; } if (begin > end) return; *values = malloc (end - begin + 1); ptr2 = *values; while (begin <= end) *(ptr2++) = *(begin++); keys++; values++; max--; } fclose (conf); *keys = NULL; *values = NULL; } char * config_lookup (char *key) { #define MAX_KEYS 128 static int initialised = 0; static char *keys[MAX_KEYS]; static char *values[MAX_KEYS]; int i; if (initialised == 0) { initialised = 1; config_init (keys, values, MAX_KEYS); #if 0 MessageBox(g_hwndParent, "Configuration File:", 0, MB_OK); i = 0; while (keys[i]) { char buf[256]; sprintf (buf, "%s=%s\r\n", keys[i], values[i]); MessageBox (g_hwndParent, buf, 0, MB_OK); i++; } #endif } i = 0; while (keys[i]) { if (!strcmp (keys[i], key)) return values[i]; i++; } return NULL; } void __declspec(dllexport) config_fetch (HWND hwndParent, int string_size, char *variables, stack_t **stacktop, extra_parameters_t *extra) { char key[256]; int err = 0; char *value; g_hwndParent = hwndParent; EXDLL_INIT(); /* The expected stack layout: key. */ if (popstring (key, sizeof (key))) err = 1; if (err) { setuservariable (INST_R0, ""); return; } value = config_lookup (key); setuservariable (INST_R0, value == NULL ? "" : value); return; } void __declspec(dllexport) config_fetch_bool (HWND hwndParent, int string_size, char *variables, stack_t **stacktop, extra_parameters_t *extra) { char key[256]; int err = 0; char *value; int result; g_hwndParent = hwndParent; EXDLL_INIT(); /* The expected stack layout: key. */ if (popstring (key, sizeof (key))) err = 1; if (err) { setuservariable (INST_R0, ""); return; } value = config_lookup (key); if (value == NULL || *value == '\0') { setuservariable (INST_R0, ""); return; } result = 0; if (!strcasecmp (value, "true") || !strcasecmp (value, "yes") || atoi (value) != 0) result = 1; setuservariable (INST_R0, result == 0 ? "0" : "1"); return; } /* Return a string from the Win32 Registry or NULL in case of error. Caller must release the return value. A NULL for root is an alias for HKEY_CURRENT_USER, HKEY_LOCAL_MACHINE in turn. */ char * read_w32_registry_string (HKEY root, const char *dir, const char *name) { HKEY root_key; HKEY key_handle; DWORD n1, nbytes, type; char *result = NULL; root_key = root; if (! root_key) root_key = HKEY_CURRENT_USER; if( RegOpenKeyEx( root_key, dir, 0, KEY_READ, &key_handle ) ) { if (root) return NULL; /* no need for a RegClose, so return direct */ /* It seems to be common practise to fall back to HKLM. */ if (RegOpenKeyEx (HKEY_LOCAL_MACHINE, dir, 0, KEY_READ, &key_handle) ) return NULL; /* still no need for a RegClose, so return direct */ } nbytes = 1; if( RegQueryValueEx( key_handle, name, 0, NULL, NULL, &nbytes ) ) { if (root) goto leave; /* Try to fallback to HKLM also vor a missing value. */ RegCloseKey (key_handle); if (RegOpenKeyEx (HKEY_LOCAL_MACHINE, dir, 0, KEY_READ, &key_handle) ) return NULL; /* Nope. */ if (RegQueryValueEx( key_handle, name, 0, NULL, NULL, &nbytes)) goto leave; } result = malloc( (n1=nbytes+1) ); if( !result ) goto leave; if( RegQueryValueEx( key_handle, name, 0, &type, result, &n1 ) ) { free(result); result = NULL; goto leave; } result[nbytes] = 0; /* make sure it is really a string */ leave: RegCloseKey( key_handle ); return result; } #define ENV_HK HKEY_LOCAL_MACHINE #define ENV_REG "SYSTEM\\CurrentControlSet\\Control\\" \ "Session Manager\\Environment" /* The following setting can be used for a per-user setting. */ #if 0 #define ENV_HK HKEY_CURRENT_USER #define ENV_REG "Environment" #endif /* Due to a bug in Windows7 (kb 2685893) we better but a lower limit than 8191 on the maximum length of the PATH variable. Note, that depending on the used toolchain we used to have a 259 byte limit in the past. */ #define PATH_LENGTH_LIMIT 2047 void __declspec(dllexport) path_add (HWND hwndParent, int string_size, char *variables, stack_t **stacktop, extra_parameters_t *extra) { char dir[PATH_LENGTH_LIMIT]; char *path; char *path_new; int path_new_size; char *comp; const char delims[] = ";"; HKEY key_handle = 0; g_hwndParent = hwndParent; EXDLL_INIT(); setuservariable (INST_R0, "0"); /* MessageBox (g_hwndParent, "XXX 1", 0, MB_OK); */ /* The expected stack layout: path component. */ if (popstring (dir, sizeof (dir))) return; /* MessageBox (g_hwndParent, "XXX 2", 0, MB_OK); */ path = read_w32_registry_string (ENV_HK, ENV_REG, "Path"); if (! path) { MessageBox (g_hwndParent, "No PATH variable found", 0, MB_OK); return; } /* MessageBox (g_hwndParent, "XXX 3", 0, MB_OK); */ /* Old path plus semicolon plus dir plus terminating nul. */ path_new_size = strlen (path) + 1 + strlen (dir) + 1; if (path_new_size > PATH_LENGTH_LIMIT) { MessageBox (g_hwndParent, "PATH env variable too big", 0, MB_OK); free (path); return; } /* MessageBox (g_hwndParent, "XXX 4", 0, MB_OK); */ path_new = malloc (path_new_size); if (!path_new) { free (path); return; } /* MessageBox (g_hwndParent, "XXX 5", 0, MB_OK); */ strcpy (path_new, path); strcat (path_new, ";"); strcat (path_new, dir); /* MessageBox (g_hwndParent, "XXX 6", 0, MB_OK); */ /* MessageBox (g_hwndParent, dir, 0, MB_OK); */ /* MessageBox (g_hwndParent, "XXX 7", 0, MB_OK); */ /* Check if the directory already exists in the path. */ comp = strtok (path, delims); do { /* MessageBox (g_hwndParent, comp, 0, MB_OK); */ if (!strcmp (comp, dir)) { free (path); free (path_new); return; } comp = strtok (NULL, delims); } while (comp); free (path); /* MessageBox (g_hwndParent, "XXX 8", 0, MB_OK); */ /* Set a key for our CLSID. */ RegCreateKey (ENV_HK, ENV_REG, &key_handle); RegSetValueEx (key_handle, "Path", 0, REG_EXPAND_SZ, path_new, path_new_size); RegCloseKey (key_handle); SetEnvironmentVariable("PATH", path_new); free (path_new); /* MessageBox (g_hwndParent, "XXX 9", 0, MB_OK); */ setuservariable (INST_R0, "1"); } void __declspec(dllexport) path_remove (HWND hwndParent, int string_size, char *variables, stack_t **stacktop, extra_parameters_t *extra) { char dir[PATH_LENGTH_LIMIT]; char *path; char *path_new; int path_new_size; char *comp; const char delims[] = ";"; HKEY key_handle = 0; int changed = 0; int count = 0; g_hwndParent = hwndParent; EXDLL_INIT(); setuservariable (INST_R0, "0"); /* The expected stack layout: path component. */ if (popstring (dir, sizeof (dir))) return; path = read_w32_registry_string (ENV_HK, ENV_REG, "Path"); /* Old path plus semicolon plus dir plus terminating nul. */ path_new_size = strlen (path) + 1; path_new = malloc (path_new_size); if (!path_new) { free (path); return; } path_new[0] = '\0'; /* Compose the new path. */ comp = strtok (path, delims); do { if (strcmp (comp, dir)) { if (count != 0) strcat (path_new, ";"); strcat (path_new, comp); count++; } else changed = 1; comp = strtok (NULL, delims); } while (comp); free (path); if (! changed) return; /* Set a key for our CLSID. */ RegCreateKey (ENV_HK, ENV_REG, &key_handle); RegSetValueEx (key_handle, "Path", 0, REG_EXPAND_SZ, path_new, path_new_size); RegCloseKey (key_handle); free (path_new); setuservariable (INST_R0, "1"); } + +/** @brief Kill processes with the name name. + * + * This function tries to kill a process using ExitProcess. + * + * If it does not work it does not work. No return values. + * The intention is to make an effort to kill something during + * installation / uninstallation. + * + * The function signature is explained by NSIS. + */ +void __declspec(dllexport) __cdecl KillProc(HWND hwndParent, + int string_size, + char *variables, + stack_t **stacktop) +{ + HANDLE h; + PROCESSENTRY32 pe32; + + if (!stacktop || !*stacktop || !(*stacktop)->text) + { + ERRORPRINTF ("Invalid call to KillProc."); + return; + } + + + h = CreateToolhelp32Snapshot (TH32CS_SNAPPROCESS, 0); + if (h == INVALID_HANDLE_VALUE) + { + ERRORPRINTF ("Failed to create Toolhelp snapshot"); + return; + } + pe32.dwSize = sizeof (PROCESSENTRY32); + + if (!Process32First (h, &pe32)) + { + ERRORPRINTF ("Failed to get first process"); + CloseHandle (h); + return; + } + + do + { + if (!strcmp ((*stacktop)->text, pe32.szExeFile)) + { + HANDLE hProc = OpenProcess (PROCESS_ALL_ACCESS, FALSE, + pe32.th32ProcessID); + if (!hProc) + { + ERRORPRINTF ("Failed to open process handle."); + continue; + } + if (!TerminateProcess (hProc, 1)) + { + ERRORPRINTF ("Failed to terminate process."); + } + CloseHandle (hProc); + } + } + while (Process32Next (h, &pe32)); + CloseHandle (h); +}