/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include <windows.h>
#include "../../../../toolkit/mozapps/update/common/pathhash.h"
#pragma comment(lib,
"advapi32.lib")
typedef struct _stack_t {
struct _stack_t *next;
TCHAR text[MAX_PATH];
} stack_t;
int popstring(stack_t **stacktop, LPTSTR str,
int len);
void pushstring(stack_t **stacktop, LPCTSTR str,
int len);
/**
* Determines if the specified service exists or not
*
* @param serviceName The name of the service to check
* @param exists Whether or not the service exists
* @return TRUE if there were no errors
*/
static BOOL
IsServiceInstalled(LPCWSTR serviceName,
BOOL &exists)
{
exists =
FALSE;
// Get a handle to the local computer SCM database with full access rights.
SC_HANDLE serviceManager = OpenSCManager(NULL, NULL,
SC_MANAGER_ENUMERATE_SERVICE);
if (!serviceManager) {
return FALSE;
}
SC_HANDLE serviceHandle = OpenServiceW(serviceManager,
serviceName,
SERVICE_QUERY_CONFIG);
if (!serviceHandle && GetLastError() != ERROR_SERVICE_DOES_NOT_EXIST) {
CloseServiceHandle(serviceManager);
return FALSE;
}
if (serviceHandle) {
CloseServiceHandle(serviceHandle);
exists =
TRUE;
}
CloseServiceHandle(serviceManager);
return TRUE;
}
/**
* Determines if the specified service is installed or not
*
* @param stacktop A pointer to the top of the stack
* @param variables A pointer to the NSIS variables
* @return 0 if the service does not exist
* 1 if the service does exist
* -1 if there was an error.
*/
extern "C" void __declspec(dllexport)
IsInstalled(HWND hwndParent,
int string_size,
TCHAR *variables, stack_t **stacktop,
void *extra)
{
TCHAR tmp[MAX_PATH] = { L
'\0' };
WCHAR serviceName[MAX_PATH] = {
'\0' };
popstring(stacktop, tmp, MAX_PATH);
#if !
defined(UNICODE)
MultiByteToWideChar(CP_ACP, 0, tmp, -1, serviceName, MAX_PATH);
#else
wcscpy(serviceName, tmp);
#endif
BOOL serviceInstalled;
if (!IsServiceInstalled(serviceName, serviceInstalled)) {
pushstring(stacktop, TEXT(
"-1"), 3);
}
else {
pushstring(stacktop, serviceInstalled ? TEXT(
"1") : TEXT(
"0"), 2);
}
}
/**
* Stops the specified service.
*
* @param serviceName The name of the service to stop
* @return TRUE if the operation was successful
*/
static BOOL
StopService(LPCWSTR serviceName)
{
// Get a handle to the local computer SCM database with full access rights.
SC_HANDLE serviceManager = OpenSCManager(NULL, NULL,
SC_MANAGER_ENUMERATE_SERVICE);
if (!serviceManager) {
return FALSE;
}
SC_HANDLE serviceHandle = OpenServiceW(serviceManager,
serviceName,
SERVICE_STOP);
if (!serviceHandle) {
CloseServiceHandle(serviceManager);
return FALSE;
}
//Stop the service so it deletes faster and so the uninstaller
// can actually delete its EXE.
DWORD totalWaitTime = 0;
SERVICE_STATUS status;
static const int maxWaitTime = 1000 * 60;
// Never wait more than a minute
BOOL stopped =
FALSE;
if (ControlService(serviceHandle, SERVICE_CONTROL_STOP, &status)) {
do {
Sleep(status.dwWaitHint);
// + 10 milliseconds to make sure we always approach maxWaitTime
totalWaitTime += (status.dwWaitHint + 10);
if (status.dwCurrentState == SERVICE_STOPPED) {
stopped =
true;
break;
}
else if (totalWaitTime > maxWaitTime) {
break;
}
}
while (QueryServiceStatus(serviceHandle, &status));
}
CloseServiceHandle(serviceHandle);
CloseServiceHandle(serviceManager);
return stopped;
}
/**
* Stops the specified service
*
* @param stacktop A pointer to the top of the stack
* @param variables A pointer to the NSIS variables
* @return 1 if the service was stopped, 0 on error
*/
extern "C" void __declspec(dllexport)
Stop(HWND hwndParent,
int string_size,
TCHAR *variables, stack_t **stacktop,
void *extra)
{
TCHAR tmp[MAX_PATH] = { L
'\0' };
WCHAR serviceName[MAX_PATH] = {
'\0' };
popstring(stacktop, tmp, MAX_PATH);
#if !
defined(UNICODE)
MultiByteToWideChar(CP_ACP, 0, tmp, -1, serviceName, MAX_PATH);
#else
wcscpy(serviceName, tmp);
#endif
if (StopService(serviceName)) {
pushstring(stacktop, TEXT(
"1"), 2);
}
else {
pushstring(stacktop, TEXT(
"0"), 2);
}
}
/**
* Determines a unique registry path from a file or directory path
*
* @param stacktop A pointer to the top of the stack
* @param variables A pointer to the NSIS variables
* @return The unique registry path or an empty string on error
*/
extern "C" void __declspec(dllexport)
PathToUniqueRegistryPath(HWND hwndParent,
int string_size,
TCHAR *variables, stack_t **stacktop,
void *extra)
{
TCHAR tmp[MAX_PATH] = { L
'\0' };
WCHAR installBasePath[MAX_PATH] = {
'\0' };
popstring(stacktop, tmp, MAX_PATH);
#if !
defined(UNICODE)
MultiByteToWideChar(CP_ACP, 0, tmp, -1, installBasePath, MAX_PATH);
#else
wcscpy(installBasePath, tmp);
#endif
WCHAR registryPath[MAX_PATH + 1] = {
'\0' };
if (CalculateRegistryPathFromFilePath(installBasePath, registryPath)) {
pushstring(stacktop, registryPath, wcslen(registryPath) + 1);
}
else {
pushstring(stacktop, TEXT(
""), 1);
}
}
BOOL WINAPI
DllMain(HANDLE hInst, ULONG ul_reason_for_call, LPVOID lpReserved)
{
return TRUE;
}
/**
* Removes an element from the top of the NSIS stack
*
* @param stacktop A pointer to the top of the stack
* @param str The string to pop to
* @param len The max length
* @return 0 on success
*/
int popstring(stack_t **stacktop, TCHAR *str,
int len)
{
// Removes the element from the top of the stack and puts it in the buffer
stack_t *th;
if (!stacktop || !*stacktop) {
return 1;
}
th = (*stacktop);
lstrcpyn(str,th->text, len);
*stacktop = th->next;
GlobalFree((HGLOBAL)th);
return 0;
}
/**
* Adds an element to the top of the NSIS stack
*
* @param stacktop A pointer to the top of the stack
* @param str The string to push on the stack
* @param len The length of the string to push on the stack
* @return 0 on success
*/
void pushstring(stack_t **stacktop,
const TCHAR *str,
int len)
{
stack_t *th;
if (!stacktop) {
return;
}
th = (stack_t*)GlobalAlloc(GPTR,
sizeof(stack_t) + len);
lstrcpyn(th->text, str, len);
th->next = *stacktop;
*stacktop = th;
}
