mirror of
https://github.com/Wind4/vlmcsd.git
synced 2025-05-28 05:34:09 -04:00
vlmcsd-svn812-2015-08-30-Hotbird64
This commit is contained in:
Wind4
commit
1af203d2a8
133 changed files with 102338 additions and 0 deletions
346
ntservice.c
Normal file
346
ntservice.c
Normal file
|
|
@ -0,0 +1,346 @@
|
|||
#ifndef CONFIG
|
||||
#define CONFIG "config.h"
|
||||
#endif // CONFIG
|
||||
#include CONFIG
|
||||
|
||||
#include "ntservice.h"
|
||||
#include "shared_globals.h"
|
||||
#include "vlmcsd.h"
|
||||
#include "output.h"
|
||||
#include "helpers.h"
|
||||
|
||||
#ifdef _NTSERVICE
|
||||
|
||||
SERVICE_STATUS gSvcStatus;
|
||||
SERVICE_STATUS_HANDLE gSvcStatusHandle;
|
||||
|
||||
static VOID WINAPI ServiceCtrlHandler(const DWORD dwCtrl)
|
||||
{
|
||||
// Handle the requested control code.
|
||||
|
||||
switch(dwCtrl)
|
||||
{
|
||||
case SERVICE_CONTROL_STOP:
|
||||
case SERVICE_CONTROL_SHUTDOWN:
|
||||
|
||||
ServiceShutdown = TRUE;
|
||||
ReportServiceStatus(SERVICE_STOP_PENDING, NO_ERROR, 0);
|
||||
|
||||
// Remove PID file and free ressources
|
||||
cleanup();
|
||||
# ifdef USE_MSRPC
|
||||
ReportServiceStatus(SERVICE_STOPPED, NO_ERROR, 0);
|
||||
# endif // !USE_MSRPC
|
||||
return;
|
||||
|
||||
/*case SERVICE_CONTROL_INTERROGATE:
|
||||
break;*/
|
||||
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
static VOID WINAPI ServiceMain(const int argc_unused, CARGV argv_unused)
|
||||
{
|
||||
// Register the handler function for the service
|
||||
|
||||
gSvcStatusHandle = RegisterServiceCtrlHandler(
|
||||
NT_SERVICE_NAME,
|
||||
ServiceCtrlHandler
|
||||
);
|
||||
|
||||
if(!gSvcStatusHandle)
|
||||
{
|
||||
//ServiceReportEvent(RegisterServiceCtrlHandler);
|
||||
return;
|
||||
}
|
||||
|
||||
// These SERVICE_STATUS members remain as set here
|
||||
|
||||
gSvcStatus.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
|
||||
gSvcStatus.dwServiceSpecificExitCode = 0;
|
||||
|
||||
// Run the actual program
|
||||
ReportServiceStatus(SERVICE_STOPPED, newmain(), 3000);
|
||||
}
|
||||
|
||||
SERVICE_TABLE_ENTRY NTServiceDispatchTable[] = {
|
||||
{
|
||||
(LPSTR)NT_SERVICE_NAME,
|
||||
(LPSERVICE_MAIN_FUNCTION) ServiceMain
|
||||
},
|
||||
{
|
||||
NULL,
|
||||
NULL
|
||||
}
|
||||
};
|
||||
|
||||
VOID ReportServiceStatus(const DWORD dwCurrentState, const DWORD dwWin32ExitCode, const DWORD dwWaitHint)
|
||||
{
|
||||
static DWORD dwCheckPoint = 1;
|
||||
|
||||
// Fill in the SERVICE_STATUS structure.
|
||||
|
||||
gSvcStatus.dwCurrentState = dwCurrentState;
|
||||
gSvcStatus.dwWin32ExitCode = dwWin32ExitCode;
|
||||
gSvcStatus.dwWaitHint = dwWaitHint;
|
||||
|
||||
if (dwCurrentState == SERVICE_START_PENDING)
|
||||
gSvcStatus.dwControlsAccepted = 0;
|
||||
else
|
||||
gSvcStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP;
|
||||
|
||||
if ( (dwCurrentState == SERVICE_RUNNING) ||
|
||||
(dwCurrentState == SERVICE_STOPPED) )
|
||||
gSvcStatus.dwCheckPoint = 0;
|
||||
else
|
||||
gSvcStatus.dwCheckPoint = dwCheckPoint++;
|
||||
|
||||
// Report the status of the service to the SCM.
|
||||
SetServiceStatus(gSvcStatusHandle, &gSvcStatus);
|
||||
}
|
||||
|
||||
/*VOID ServiceReportEvent(char *szFunction)
|
||||
{
|
||||
HANDLE hEventSource;
|
||||
const char *eventStrings[2];
|
||||
TCHAR Buffer[80];
|
||||
|
||||
hEventSource = RegisterEventSource(NULL, NT_SERVICE_NAME);
|
||||
|
||||
if (hEventSource)
|
||||
{
|
||||
snprintf(Buffer, 80, "%s failed with %d", szFunction, GetLastError());
|
||||
|
||||
eventStrings[0] = NT_SERVICE_NAME;
|
||||
eventStrings[1] = Buffer;
|
||||
|
||||
ReportEvent(hEventSource, // event log handle
|
||||
EVENTLOG_ERROR_TYPE, // event type
|
||||
0, // event category
|
||||
00, // event identifier
|
||||
NULL, // no security identifier
|
||||
2, // size of lpszStrings array
|
||||
0, // no binary data
|
||||
eventStrings, // array of strings
|
||||
NULL); // no binary data
|
||||
|
||||
DeregisterEventSource(hEventSource);
|
||||
}
|
||||
}*/
|
||||
|
||||
//Returns 0=Error, 1=Success, 2=Doesn't exist
|
||||
|
||||
|
||||
static uint_fast8_t OpenAndRemoveService(DWORD *dwPreviousState, SC_HANDLE *schSCManager)
|
||||
{
|
||||
SERVICE_STATUS status;
|
||||
uint_fast8_t i;
|
||||
SC_HANDLE installedService;
|
||||
uint_fast8_t result = 1;
|
||||
BOOL closeManager = FALSE;
|
||||
|
||||
// Allow NULL for both Arguments
|
||||
if (!dwPreviousState) dwPreviousState = (DWORD*)alloca(sizeof(*dwPreviousState));
|
||||
if (!schSCManager)
|
||||
{
|
||||
schSCManager = (SC_HANDLE*)alloca(sizeof(*schSCManager));
|
||||
closeManager = TRUE;
|
||||
}
|
||||
|
||||
*schSCManager = OpenSCManager(
|
||||
NULL, // local computer
|
||||
NULL, // ServicesActive database
|
||||
SC_MANAGER_ALL_ACCESS); // full access rights
|
||||
|
||||
if (!*schSCManager) return 0;
|
||||
|
||||
if (!(installedService = OpenService(*schSCManager, NT_SERVICE_NAME, SERVICE_ALL_ACCESS)))
|
||||
{
|
||||
result = 2;
|
||||
}
|
||||
else
|
||||
{
|
||||
*dwPreviousState = SERVICE_STOPPED;
|
||||
if (QueryServiceStatus(installedService, &status)) *dwPreviousState = status.dwCurrentState;
|
||||
|
||||
ControlService(installedService, SERVICE_CONTROL_STOP, &status);
|
||||
|
||||
for (i = 0; i < 10; i++)
|
||||
{
|
||||
QueryServiceStatus(installedService, &status);
|
||||
// Give it 100 ms after it reported SERVICE_STOPPED. Subsequent CreateService will fail otherwise
|
||||
Sleep(100);
|
||||
if (status.dwCurrentState == SERVICE_STOPPED) break;
|
||||
}
|
||||
|
||||
if (!DeleteService(installedService)) result = 0;
|
||||
CloseServiceHandle(installedService);
|
||||
}
|
||||
|
||||
if (closeManager) CloseServiceHandle(*schSCManager);
|
||||
return result;
|
||||
}
|
||||
|
||||
static VOID ServiceInstaller(const char *restrict ServiceUser, const char *const ServicePassword)
|
||||
{
|
||||
SC_HANDLE schSCManager;
|
||||
SC_HANDLE schService;
|
||||
char szPath[MAX_PATH] = "\"";
|
||||
|
||||
if (!GetModuleFileName(NULL, szPath + sizeof(char), MAX_PATH - 1))
|
||||
{
|
||||
errorout("Cannot install service (%d)\n", (uint32_t)GetLastError());
|
||||
return;
|
||||
}
|
||||
|
||||
strcat(szPath,"\"");
|
||||
|
||||
int i;
|
||||
for (i = 1; i < global_argc; i ++)
|
||||
{
|
||||
// Strip unneccessary parameters, especially the password
|
||||
if (!strcmp(global_argv[i], "-s")) continue;
|
||||
|
||||
if (!strcmp(global_argv[i], "-W") ||
|
||||
!strcmp(global_argv[i], "-U"))
|
||||
{
|
||||
i++;
|
||||
continue;
|
||||
}
|
||||
|
||||
strcat(szPath, " ");
|
||||
|
||||
if (strchr(global_argv[i], ' '))
|
||||
{
|
||||
strcat(szPath, "\"");
|
||||
strcat(szPath, global_argv[i]);
|
||||
strcat(szPath, "\"");
|
||||
}
|
||||
else
|
||||
strcat(szPath, global_argv[i]);
|
||||
}
|
||||
|
||||
// Get a handle to the SCM database.
|
||||
|
||||
SERVICE_STATUS status;
|
||||
DWORD dwPreviousState;
|
||||
|
||||
if (!OpenAndRemoveService(&dwPreviousState, &schSCManager))
|
||||
{
|
||||
errorout("Service removal failed (%d)\n", (uint32_t)GetLastError());
|
||||
return;
|
||||
}
|
||||
|
||||
char *tempUser = NULL;
|
||||
|
||||
if (ServiceUser)
|
||||
{
|
||||
// Shortcuts for some well known users
|
||||
if (!strcasecmp(ServiceUser, "/l")) ServiceUser="NT AUTHORITY\\LocalService";
|
||||
if (!strcasecmp(ServiceUser, "/n")) ServiceUser="NT AUTHORITY\\NetworkService";
|
||||
|
||||
// Allow Local Users without .\ , e.g. "johndoe" instead of ".\johndoe"
|
||||
if (!strchr(ServiceUser, '\\'))
|
||||
{
|
||||
tempUser = (char*)vlmcsd_malloc(strlen(ServiceUser) + 3);
|
||||
strcpy(tempUser, ".\\");
|
||||
strcat(tempUser, ServiceUser);
|
||||
ServiceUser = tempUser;
|
||||
}
|
||||
}
|
||||
|
||||
schService = CreateService(
|
||||
schSCManager, // SCM database
|
||||
NT_SERVICE_NAME, // name of service
|
||||
NT_SERVICE_DISPLAY_NAME, // service name to display
|
||||
SERVICE_ALL_ACCESS, // desired access
|
||||
SERVICE_WIN32_OWN_PROCESS, // service type
|
||||
SERVICE_AUTO_START, // start type
|
||||
SERVICE_ERROR_NORMAL, // error control type
|
||||
szPath, // path to service's binary
|
||||
NULL, // no load ordering group
|
||||
NULL, // no tag identifier
|
||||
"tcpip\0", // depends on TCP/IP
|
||||
ServiceUser, // LocalSystem account
|
||||
ServicePassword); // no password
|
||||
|
||||
# if __clang__ && (__CYGWIN__ || __MINGW64__ )
|
||||
// Workaround for clang not understanding some GCC asm syntax used in <w32api/psdk_inc/intrin-impl.h>
|
||||
ZeroMemory((char*)ServicePassword, strlen(ServicePassword));
|
||||
# else
|
||||
SecureZeroMemory((char*)ServicePassword, strlen(ServicePassword));
|
||||
# endif
|
||||
if (tempUser) free(tempUser);
|
||||
|
||||
if (schService == NULL)
|
||||
{
|
||||
errorout("CreateService failed (%u)\n", (uint32_t)GetLastError());
|
||||
CloseServiceHandle(schSCManager);
|
||||
return;
|
||||
}
|
||||
else
|
||||
{
|
||||
errorout("Service installed successfully\n");
|
||||
|
||||
if (dwPreviousState == SERVICE_RUNNING)
|
||||
{
|
||||
printf("Restarting " NT_SERVICE_NAME " service => ");
|
||||
status.dwCurrentState = SERVICE_STOPPED;
|
||||
|
||||
if (StartService(schService, 0, NULL))
|
||||
{
|
||||
for (i = 0; i < 10; i++)
|
||||
{
|
||||
if (!QueryServiceStatus(schService, &status) || status.dwCurrentState != SERVICE_START_PENDING) break;
|
||||
Sleep(100);
|
||||
}
|
||||
|
||||
if (status.dwCurrentState == SERVICE_RUNNING)
|
||||
printf("Success\n");
|
||||
else if (status.dwCurrentState == SERVICE_START_PENDING)
|
||||
printf("Not ready within a second\n");
|
||||
else
|
||||
errorout("Error\n");
|
||||
}
|
||||
else
|
||||
errorout("Error %u\n", (uint32_t)GetLastError());
|
||||
}
|
||||
}
|
||||
|
||||
CloseServiceHandle(schService);
|
||||
CloseServiceHandle(schSCManager);
|
||||
}
|
||||
|
||||
int NtServiceInstallation(const int_fast8_t installService, const char *restrict ServiceUser, const char *const ServicePassword)
|
||||
{
|
||||
if (IsNTService) return 0;
|
||||
|
||||
if (installService == 1) // Install
|
||||
{
|
||||
ServiceInstaller(ServiceUser, ServicePassword);
|
||||
return(0);
|
||||
}
|
||||
|
||||
if (installService == 2) // Remove
|
||||
{
|
||||
switch(OpenAndRemoveService(NULL, NULL))
|
||||
{
|
||||
case 0:
|
||||
errorout("Error removing service %s\n", NT_SERVICE_NAME);
|
||||
return(!0);
|
||||
case 1:
|
||||
printf("Service %s removed successfully\n", NT_SERVICE_NAME);
|
||||
return(0);
|
||||
default:
|
||||
errorout("Service %s does not exist.\n", NT_SERVICE_NAME);
|
||||
return(!0);
|
||||
}
|
||||
}
|
||||
|
||||
// Do nothing
|
||||
|
||||
return(0);
|
||||
}
|
||||
#endif // _NTSERVICE
|
||||
Loading…
Add table
Add a link
Reference in a new issue