register.c File Reference

#include "w32time.h"
#include <debug.h>
#include <strsafe.h>
#include "resource.h"

Include dependency graph for register.c:

Go to the source code of this file.

Functions
static PWSTR	ReadString (_In_ UINT uID)
static HRESULT	RegisterService (VOID)
static HRESULT	SetParametersValues (VOID)
static HRESULT	SetNtpClientValues (VOID)
HRESULT WINAPI	DllRegisterServer (VOID)
static UINT	UnregisterService (VOID)
HRESULT WINAPI	DllUnregisterServer (VOID)

Variables
PCWSTR	g_pszServiceName = L"W32Time"

Function Documentation

DllRegisterServer()

HRESULT WINAPI DllRegisterServer

(

VOID

)

Definition at line 269 of file register.c.

{
    HRESULT hresult;
 
    hresult = RegisterService();
    if (FAILED(hresult))
        return hresult;
 
    hresult = SetParametersValues();
    if (FAILED(hresult))
        return hresult;
 
    hresult = SetNtpClientValues();
 
    return hresult;
}

DllUnregisterServer()

HRESULT WINAPI DllUnregisterServer

(

VOID

)

Definition at line 327 of file register.c.

{
    UINT error = UnregisterService();
    return HRESULT_FROM_WIN32(error);
}

ReadString()

static PWSTR ReadString ( _In_ UINT  uID )

static

Definition at line 18 of file register.c.

{
    HINSTANCE hInstance;
    int nLength;
    PWSTR pszString, pszPtr;
 
    hInstance = GetModuleHandleW(L"w32time");
    if (hInstance == NULL)
        return NULL;
 
    nLength = LoadStringW(hInstance, uID, (LPWSTR)&pszPtr, 0);
    if (nLength == 0)
        return NULL;
 
    pszString = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, (nLength + 1) * sizeof(WCHAR));
    if (pszString == NULL)
        return NULL;
 
    wcsncpy(pszString, pszPtr, nLength);
 
    return pszString;
}

Referenced by RegisterService().

RegisterService()

static HRESULT RegisterService ( VOID  )

static

Definition at line 45 of file register.c.

{
    SC_HANDLE hServiceManager = NULL;
    SC_HANDLE hService = NULL;
    HKEY hKey = NULL;
    PWSTR pszValue;
    PWSTR pszDisplayName = NULL, pszDescription = NULL;
    DWORD dwDisposition, dwError;
    SERVICE_DESCRIPTIONW ServiceDescription;
    HRESULT hresult = S_OK;
 
    DPRINT("RegisterService()\n");
 
    hServiceManager = OpenSCManagerW(NULL, NULL, SC_MANAGER_CREATE_SERVICE);
    if (hServiceManager == NULL)
    {
        DPRINT1("OpenSCManager() failed!\n");
        hresult = E_FAIL;
        goto done;
    }
 
    pszDisplayName = ReadString(IDS_DISPLAYNAME);
    if (pszDisplayName == NULL)
    {
        DPRINT1("ReadString(IDS_DISPLAYNAME) failed!\n");
        hresult = E_FAIL;
        goto done;
    }
 
    pszDescription = ReadString(IDS_DESCRIPTION);
    if (pszDescription == NULL)
    {
        DPRINT1("ReadString(IDS_DESCRIPTION) failed!\n");
        hresult = E_FAIL;
        goto done;
    }
 
    hService = CreateServiceW(hServiceManager,
                              g_pszServiceName,
                              pszDisplayName,
                              GENERIC_WRITE,
                              SERVICE_WIN32_SHARE_PROCESS,
                              SERVICE_AUTO_START,
                              SERVICE_ERROR_NORMAL,
                              L"%SystemRoot%\\system32\\svchost.exe -k netsvcs",
                              L"Time", NULL, NULL, L"LocalSystem", NULL);
 
    if (!hService && GetLastError() == ERROR_SERVICE_EXISTS)
    {
        hService = OpenServiceW(hServiceManager, g_pszServiceName, SERVICE_CHANGE_CONFIG);
    }
    if (hService == NULL)
    {
        DPRINT1("CreateService() failed!\n");
        hresult = E_FAIL;
        goto done;
    }
 
    ServiceDescription.lpDescription = pszDescription;
    if (ChangeServiceConfig2W(hService,
                              SERVICE_CONFIG_DESCRIPTION,
                              &ServiceDescription) == FALSE)
    {
        DPRINT1("ChangeServiceConfig2() failed!\n");
        hresult = E_FAIL;
        goto done;
    }
 
    dwError = RegCreateKeyExW(HKEY_LOCAL_MACHINE,
                              L"System\\CurrentControlSet\\Services\\W32Time\\Parameters",
                              0,
                              NULL,
                              REG_OPTION_NON_VOLATILE,
                              KEY_WRITE,
                              NULL,
                              &hKey,
                              &dwDisposition);
    if (dwError != ERROR_SUCCESS)
    {
        DPRINT1("RegCreateKeyEx() failed!\n");
        hresult = E_FAIL;
        goto done;
    }
 
    pszValue = L"%SystemRoot%\\system32\\w32time.dll";
    dwError = RegSetValueExW(hKey,
                             L"ServiceDll",
                             0,
                             REG_EXPAND_SZ,
                             (LPBYTE)pszValue,
                             (wcslen(pszValue) + 1) * sizeof(WCHAR));
    if (dwError != ERROR_SUCCESS)
    {
        DPRINT1("RegSetValueEx() failed!\n");
        hresult = E_FAIL;
        goto done;
    }
 
    pszValue = L"SvchostEntry_W32Time";
    dwError = RegSetValueExW(hKey,
                             L"ServiceMain",
                             0,
                             REG_SZ,
                             (LPBYTE)pszValue,
                             (wcslen(pszValue) + 1) * sizeof(WCHAR));
    if (dwError != ERROR_SUCCESS)
    {
        DPRINT1("RegSetValueEx() failed!\n");
        hresult = E_FAIL;
        goto done;
    }
 
done:
    if (hKey)
        RegCloseKey(hKey);
 
    if (hService)
        CloseServiceHandle(hService);
 
    if (hServiceManager)
        CloseServiceHandle(hServiceManager);
 
    if (pszDescription)
        HeapFree(GetProcessHeap(), 0, pszDescription);
 
    if (pszDisplayName)
        HeapFree(GetProcessHeap(), 0, pszDisplayName);
 
    return hresult;
}

Referenced by DllRegisterServer().

SetNtpClientValues()

static HRESULT SetNtpClientValues ( VOID  )

static

Definition at line 224 of file register.c.

{
    HKEY hKey = NULL;
    DWORD dwValue;
    DWORD dwDisposition, dwError;
    HRESULT hresult = S_OK;
 
    dwError = RegCreateKeyExW(HKEY_LOCAL_MACHINE,
                              L"System\\CurrentControlSet\\Services\\W32Time\\TimeProviders\\NtpClient",
                              0,
                              NULL,
                              REG_OPTION_NON_VOLATILE,
                              KEY_WRITE,
                              NULL,
                              &hKey,
                              &dwDisposition);
    if (dwError != ERROR_SUCCESS)
    {
        DPRINT1("RegCreateKeyEx() failed!\n");
        goto done;
    }
 
    dwValue = W32TIME_POLL_INTERVAL;
    dwError = RegSetValueExW(hKey,
                             L"SpecialPollInterval",
                             0,
                             REG_DWORD,
                             (LPBYTE)&dwValue,
                             sizeof(dwValue));
    if (dwError != ERROR_SUCCESS)
    {
        DPRINT1("RegSetValueEx() failed!\n");
        goto done;
    }
 
done:
    if (hKey)
        RegCloseKey(hKey);
 
    return hresult;
}

Referenced by DllRegisterServer().

SetParametersValues()

static HRESULT SetParametersValues ( VOID  )

static

Definition at line 179 of file register.c.

{
    HKEY hKey = NULL;
    PWSTR pszValue;
    DWORD dwDisposition, dwError;
    HRESULT hresult = S_OK;
 
    dwError = RegCreateKeyExW(HKEY_LOCAL_MACHINE,
                              L"System\\CurrentControlSet\\Services\\W32Time\\Parameters",
                              0,
                              NULL,
                              REG_OPTION_NON_VOLATILE,
                              KEY_WRITE,
                              NULL,
                              &hKey,
                              &dwDisposition);
    if (dwError != ERROR_SUCCESS)
    {
        DPRINT1("RegCreateKeyEx() failed!\n");
        goto done;
    }
 
    pszValue = L"NTP";
    dwError = RegSetValueExW(hKey,
                             L"Type",
                             0,
                             REG_SZ,
                             (LPBYTE)pszValue,
                             (wcslen(pszValue) + 1) * sizeof(WCHAR));
    if (dwError != ERROR_SUCCESS)
    {
        DPRINT1("RegSetValueEx() failed!\n");
        goto done;
    }
 
done:
    if (hKey)
        RegCloseKey(hKey);
 
    return hresult;
}

Referenced by DllRegisterServer().

UnregisterService()

static UINT UnregisterService ( VOID  )

static

Definition at line 289 of file register.c.

{
    UINT error;
    SC_HANDLE hServiceManager, hService;
 
    hServiceManager = OpenSCManagerW(NULL, NULL, SC_MANAGER_CREATE_SERVICE);
    if (!hServiceManager)
    {
        error = GetLastError();
        DPRINT1("OpenSCManager() failed, error %lu\n", error);
        return error;
    }
 
    hService = OpenServiceW(hServiceManager, g_pszServiceName, DELETE);
    error = GetLastError();
    CloseServiceHandle(hServiceManager);
    if (!hService)
    {
        if (error == ERROR_SERVICE_DOES_NOT_EXIST)
            return ERROR_SUCCESS;
        DPRINT1("OpenService() failed, error %lu\n", error);
    }
    else
    {
        error = DeleteService(hService) ? ERROR_SUCCESS : GetLastError();
        if (error == ERROR_SERVICE_MARKED_FOR_DELETE)
            error = ERROR_SUCCESS;
        if (error)
            DPRINT1("DeleteService() failed, error %lu\n", error);
 
        CloseServiceHandle(hService);
    }
    return error;
}

Referenced by DllUnregisterServer().

Variable Documentation

g_pszServiceName

PCWSTR g_pszServiceName = L"W32Time"

Definition at line 14 of file register.c.

Referenced by RegisterService(), and UnregisterService().