service.c

Go to the documentation of this file.

/*
 * PROJECT:     Local Security Authority Server DLL
 * LICENSE:     GPL - See COPYING in the top level directory
 * FILE:        dll/win32/lsasrv/service.c
 * PURPOSE:     Security service
 * COPYRIGHT:   Copyright 2016, 2019 Eric Kohl <eric.kohl@reactos.org>
 */
 
/* INCLUDES ****************************************************************/
 
#include "lsasrv.h"
#include <winsvc.h>
#include <winsvc_undoc.h>
 
typedef VOID (WINAPI *PNETLOGONMAIN)(INT ArgCount, PWSTR *ArgVector);
 
static VOID WINAPI NetlogonServiceMain(DWORD dwArgc, PWSTR *pszArgv);
static VOID WINAPI SamSsServiceMain(DWORD dwArgc, PWSTR *pszArgv);
 
SERVICE_TABLE_ENTRYW ServiceTable[] =
{
    {L"NETLOGON", NetlogonServiceMain},
    {L"SAMSS", SamSsServiceMain},
    {NULL, NULL}
};
 
 
/* FUNCTIONS ***************************************************************/
 
static
VOID
WINAPI
NetlogonServiceMain(
    _In_ DWORD dwArgc,
    _In_ PWSTR *pszArgv)
{
    HINSTANCE hNetlogon = NULL;
    PNETLOGONMAIN pNetlogonMain = NULL;
 
    TRACE("NetlogonServiceMain(%lu %p)\n", dwArgc, pszArgv);
 
    hNetlogon = LoadLibraryW(L"Netlogon.dll");
    if (hNetlogon == NULL)
    {
        ERR("LoadLibrary() failed!\n");
        return;
    }
 
    pNetlogonMain = (PNETLOGONMAIN)GetProcAddress(hNetlogon, "NlNetlogonMain");
    if (pNetlogonMain == NULL)
    {
        ERR("GetProcAddress(NlNetlogonMain) failed!\n");
        FreeLibrary(hNetlogon);
        return;
    }
 
    TRACE("NlNetlogonMain %p\n", pNetlogonMain);
 
    pNetlogonMain(dwArgc, pszArgv);
}
 
 
static
VOID
WINAPI
SamSsControlHandler(
    _In_ DWORD fdwControl)
{
    TRACE("SamSsControlHandler(%lu)\n", fdwControl);
}
 
 
static
VOID
WINAPI
SamSsServiceMain(
    _In_ DWORD dwArgc,
    _In_ PWSTR *pszArgv)
{
    SERVICE_STATUS_HANDLE hStatus;
    SERVICE_STATUS ServiceStatus;
 
    TRACE("SamSsServiceMain(%lu %p)\n", dwArgc, pszArgv);
 
    hStatus = RegisterServiceCtrlHandlerW(L"SAMSS",
                                          SamSsControlHandler);
    if (hStatus == NULL)
        return;
 
    ServiceStatus.dwServiceType = SERVICE_WIN32_SHARE_PROCESS;
    ServiceStatus.dwCurrentState = SERVICE_START_PENDING;
    ServiceStatus.dwControlsAccepted = 0;
    ServiceStatus.dwWin32ExitCode = ERROR_SUCCESS;
    ServiceStatus.dwServiceSpecificExitCode = ERROR_SUCCESS;
    ServiceStatus.dwCheckPoint = 1;
    ServiceStatus.dwWaitHint = 0x7530;
 
    SetServiceStatus(hStatus, &ServiceStatus);
 
    ServiceStatus.dwCurrentState = SERVICE_RUNNING;
    ServiceStatus.dwCheckPoint = 0;
    ServiceStatus.dwWaitHint = 0;
 
    SetServiceStatus(hStatus, &ServiceStatus);
}
 
 
static
DWORD
WINAPI
DispatcherThread(
    _In_ PVOID pParameter)
{
    HANDLE hEvent;
    DWORD dwError;
 
    TRACE("DispatcherThread(%p)\n", pParameter);
 
    /* Create or open the SECURITY_SERVICES_STARTED event */
    hEvent = CreateEventW(NULL,
                          TRUE,
                          FALSE,
                          L"SECURITY_SERVICES_STARTED");
    if (hEvent == NULL)
    {
        dwError = GetLastError();
        if (dwError != ERROR_ALREADY_EXISTS)
            return dwError;
 
        hEvent = OpenEventW(SYNCHRONIZE,
                            FALSE,
                            L"SECURITY_SERVICES_STARTED");
        if (hEvent == NULL)
            return GetLastError();
    }
 
    /* Wait for the SECURITY_SERVICES_STARTED event to be signaled */
    TRACE("Waiting for the SECURITY_SERVICES_STARTED event!\n");
    dwError = WaitForSingleObject(hEvent, INFINITE);
    TRACE("WaitForSingleObject returned %lu\n", dwError);
 
    /* Close the event handle */
    CloseHandle(hEvent);
 
    /* Fail, if the event was not signaled */
    if (dwError != WAIT_OBJECT_0)
    {
        ERR("Wait failed!\n");
        return dwError;
    }
 
    /* This is the security process */
    I_ScIsSecurityProcess();
 
    /* Start the services */
    TRACE("Start the security services!\n");
    if (!StartServiceCtrlDispatcherW(ServiceTable))
        return GetLastError();
 
    TRACE("Done!\n");
 
    return ERROR_SUCCESS;
}
 
 
NTSTATUS
WINAPI
ServiceInit(VOID)
{
    HANDLE hThread;
    DWORD dwThreadId;
 
    TRACE("ServiceInit()\n");
 
    hThread = CreateThread(NULL,
                           0,
                           DispatcherThread,
                           NULL,
                           0,
                           &dwThreadId);
    if (hThread == NULL)
       return (NTSTATUS)GetLastError();
 
    return STATUS_SUCCESS;
}
 
/* EOF */