dnsrslvr.c File Reference

#include "precomp.h"
#include <debug.h>

Include dependency graph for dnsrslvr.c:

Go to the source code of this file.

Macros
#define	NDEBUG

Functions
DWORD WINAPI	RpcThreadRoutine (LPVOID lpParameter)
static VOID	UpdateServiceStatus (HANDLE hServiceStatus, DWORD NewStatus, DWORD Increment)
static DWORD WINAPI	ServiceControlHandler (DWORD dwControl, DWORD dwEventType, LPVOID lpEventData, LPVOID lpContext)
VOID WINAPI	ServiceMain (DWORD argc, LPWSTR *argv)
BOOL WINAPI	DllMain (_In_ HINSTANCE hinstDLL, _In_ DWORD fdwReason, _In_ PVOID pvReserved)

Variables
HINSTANCE	hDllInstance
SERVICE_STATUS_HANDLE	ServiceStatusHandle
SERVICE_STATUS	SvcStatus
static WCHAR	ServiceName [] = L"Dnscache"

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 13 of file dnsrslvr.c.

Function Documentation

DllMain()

BOOL WINAPI DllMain	(	_In_ HINSTANCE	hinstDLL,
		_In_ DWORD	fdwReason,
		_In_ PVOID	pvReserved
	)

Definition at line 135 of file dnsrslvr.c.

{
    UNREFERENCED_PARAMETER(pvReserved);
 
    switch (fdwReason)
    {
        case DLL_PROCESS_ATTACH:
            DisableThreadLibraryCalls(hinstDLL);
            hDllInstance = hinstDLL;
            break;
 
        case DLL_PROCESS_DETACH:
            break;
    }
 
    return TRUE;
}

RpcThreadRoutine()

DWORD WINAPI RpcThreadRoutine

(

LPVOID

lpParameter

)

Definition at line 20 of file rpcserver.c.

{
    RPC_STATUS Status;
 
    Status = RpcServerUseProtseqEpW(L"ncacn_np", 20, L"\\pipe\\browser", NULL);
    if (Status != RPC_S_OK)
    {
        ERR("RpcServerUseProtseqEpW() failed (Status %lx)\n", Status);
        return 0;
    }
 
    Status = RpcServerRegisterIf(browser_v0_0_s_ifspec, NULL, NULL);
    if (Status != RPC_S_OK)
    {
        ERR("RpcServerRegisterIf() failed (Status %lx)\n", Status);
        return 0;
    }
 
    Status = RpcServerListen(1, RPC_C_LISTEN_MAX_CALLS_DEFAULT, FALSE);
    if (Status != RPC_S_OK)
    {
        ERR("RpcServerListen() failed (Status %lx)\n", Status);
    }
 
    return 0;
}

Referenced by ServiceMain().

ServiceControlHandler()

static DWORD WINAPI ServiceControlHandler ( DWORD  dwControl, DWORD  dwEventType, LPVOID  lpEventData, LPVOID  lpContext  )

static

Definition at line 46 of file dnsrslvr.c.

{
    switch (dwControl)
    {
        case SERVICE_CONTROL_SHUTDOWN:
        case SERVICE_CONTROL_STOP:
            UpdateServiceStatus(ServiceStatusHandle, SERVICE_STOP_PENDING, 1);
            RpcMgmtStopServerListening(NULL);
            DnsIntCacheFree();
            UpdateServiceStatus(ServiceStatusHandle, SERVICE_STOPPED, 0);
            break;
 
        case SERVICE_CONTROL_PARAMCHANGE:
            DPRINT1("SERVICE_CONTROL_PARAMCHANGE!\n");
            return NO_ERROR;
 
        case SERVICE_CONTROL_INTERROGATE:
            return NO_ERROR;
 
        default:
            return ERROR_CALL_NOT_IMPLEMENTED;
    }
    return NO_ERROR;
}

Referenced by ServiceMain().

ServiceMain()

VOID WINAPI ServiceMain	(	DWORD	argc,
		LPWSTR *	argv
	)

Definition at line 77 of file dnsrslvr.c.

{
    HANDLE hThread;
 
    UNREFERENCED_PARAMETER(argc);
    UNREFERENCED_PARAMETER(argv);
 
    DPRINT("ServiceMain() called\n");
 
    SvcStatus.dwServiceType             = SERVICE_WIN32_OWN_PROCESS;
    SvcStatus.dwCurrentState            = SERVICE_START_PENDING;
    SvcStatus.dwControlsAccepted        = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN | SERVICE_ACCEPT_PARAMCHANGE;
    SvcStatus.dwCheckPoint              = 0;
    SvcStatus.dwWin32ExitCode           = NO_ERROR;
    SvcStatus.dwServiceSpecificExitCode = 0;
    SvcStatus.dwWaitHint                = 4000;
 
    ServiceStatusHandle = RegisterServiceCtrlHandlerExW(ServiceName,
                                                        ServiceControlHandler,
                                                        NULL);
 
    if (!ServiceStatusHandle)
    {
        DPRINT1("DNSRSLVR: Unable to register service control handler (%lx)\n", GetLastError());
        return; // FALSE
    }
 
    DnsIntCacheInitialize();
 
    ReadHostsFile();
 
    hThread = CreateThread(NULL,
                           0,
                           (LPTHREAD_START_ROUTINE)
                           RpcThreadRoutine,
                           NULL,
                           0,
                           NULL);
 
    if (!hThread)
    {
        DnsIntCacheFree();
        DPRINT("Can't create RpcThread\n");
        UpdateServiceStatus(ServiceStatusHandle, SERVICE_STOPPED, 0);
    }
    else
    {
        CloseHandle(hThread);
    }
 
    DPRINT("ServiceMain() done\n");
    UpdateServiceStatus(ServiceStatusHandle, SERVICE_RUNNING, 0);
}

UpdateServiceStatus()

static VOID UpdateServiceStatus ( HANDLE  hServiceStatus, DWORD  NewStatus, DWORD  Increment  )

static

Definition at line 29 of file dnsrslvr.c.

{
    if (Increment > 0)
        SvcStatus.dwCheckPoint += Increment;
    else
        SvcStatus.dwCheckPoint = 0;
 
    SvcStatus.dwCurrentState = NewStatus;
    SetServiceStatus(hServiceStatus, &SvcStatus);
}

Referenced by ServiceControlHandler(), and ServiceMain().

Variable Documentation

hDllInstance

HINSTANCE hDllInstance

Definition at line 18 of file dnsrslvr.c.

Referenced by DllMain().

ServiceName

WCHAR ServiceName[] = L"Dnscache"

static

Definition at line 21 of file dnsrslvr.c.

Referenced by ServiceMain().

ServiceStatusHandle

SERVICE_STATUS_HANDLE ServiceStatusHandle

Definition at line 19 of file dnsrslvr.c.

Referenced by ServiceControlHandler(), and ServiceMain().

SvcStatus

SERVICE_STATUS SvcStatus

Definition at line 20 of file dnsrslvr.c.

Referenced by ServiceMain(), and UpdateServiceStatus().