mss.c

#define UNICODE
#define WINVER 0x501

#include <stdio.h>
#include <windows.h>
#include <dbt.h>

#define SERVICE_NAME L"MediaSpy Service"
#define SLEEP_TIME   5000

SERVICE_STATUS Status;
SERVICE_STATUS_HANDLE hStatus;
HDEVNOTIFY hDeviceNotify;
FILE* Log;

void ServiceMain();
DWORD ControlHandler(DWORD, DWORD, LPVOID, LPVOID);

void main()
{
    Log = fopen("C:\\mss.log", "w");
    fprintf(Log, "Logging started...\n");

    fprintf(Log, "Service executable started...\n");
    SERVICE_TABLE_ENTRY ServiceTable[1];
    ServiceTable[0].lpServiceName = SERVICE_NAME;
    ServiceTable[0].lpServiceProc = (LPSERVICE_MAIN_FUNCTION)ServiceMain;

    fprintf(Log, "Invoking StartServiceCtrlDispatcher...\n");
    StartServiceCtrlDispatcher(ServiceTable);

    fprintf(Log, "Logging finished!\n");
    fclose(Log);
}

void ServiceMain()
{
    fprintf(Log, "Entering ServiceMain function...\n");
  Status.dwServiceType       = SERVICE_WIN32_OWN_PROCESS;
  Status.dwCurrentState      = SERVICE_START_PENDING;
  Status.dwControlsAccepted  = SERVICE_ACCEPT_STOP |
                                      SERVICE_ACCEPT_SHUTDOWN;
  Status.dwWin32ExitCode     = 0;
  Status.dwServiceSpecificExitCode = 0;
  Status.dwCheckPoint = 0;
  Status.dwWaitHint   = 0;

  hStatus =
    RegisterServiceCtrlHandlerEx( SERVICE_NAME,
                                  (LPHANDLER_FUNCTION_EX)ControlHandler,
                                  0 );
  fprintf(Log, "RegisterServiceCtrlHandlerEx returned %p\n", hStatus);

    if(hStatus == (SERVICE_STATUS_HANDLE)0)
        ; // Error

    SetServiceStatus(hStatus, &Status);

    // Some initialization
    fprintf(Log, "Registering device notification...\n");
    DEV_BROADCAST_DEVICEINTERFACE NotificationFilter;
    hDeviceNotify = NULL;



    static const GUID GuidDevInterfaceDisk =
    { 0x53F56307, 0xB6BF, 0x11D0,
        { 0x94, 0xF2, 0x00, 0xA0, 0xC9, 0x1E, 0xFB, 0x8B } };

    ZeroMemory(&NotificationFilter, sizeof(NotificationFilter));

    NotificationFilter.dbcc_size = sizeof(DEV_BROADCAST_DEVICEINTERFACE);
    NotificationFilter.dbcc_devicetype = DBT_DEVTYP_DEVICEINTERFACE;
    // NotificationFilter.dbcc_classguid = GuidDevInterfaceDisk;

    hDeviceNotify =
        RegisterDeviceNotification( (HANDLE)hStatus,
                                    &NotificationFilter,
                                    DEVICE_NOTIFY_SERVICE_HANDLE |
                                    DEVICE_NOTIFY_ALL_INTERFACE_CLASSES );

    fprintf(Log, "RegisterDeviceNotification returned %p\n", hDeviceNotify);

    if(hDeviceNotify == NULL)
        ; // Error


    Status.dwCurrentState = SERVICE_RUNNING;
  SetServiceStatus(hStatus, &Status);

    fprintf(Log, "Running...\n");
    while(Status.dwCurrentState == SERVICE_RUNNING)
    {
        Sleep(SLEEP_TIME);
    }
}

DWORD ControlHandler( DWORD dwControl, DWORD dwEventType,
                             LPVOID lpEventData, LPVOID lpContext )
{
    fprintf(Log, "ControlHandler call with %d %d\n", dwControl, dwEventType);
    switch(dwControl)
    {
    case SERVICE_CONTROL_STOP:
        UnregisterDeviceNotification(hDeviceNotify);
        Status.dwCurrentState = SERVICE_STOPPED;
        SetServiceStatus(hStatus, &Status);
        return NO_ERROR;
    case SERVICE_CONTROL_SHUTDOWN:
        UnregisterDeviceNotification(hDeviceNotify);
        Status.dwCurrentState = SERVICE_STOPPED;
        SetServiceStatus(hStatus, &Status);
        return NO_ERROR;
    case SERVICE_CONTROL_DEVICEEVENT:
        fprintf(Log, "SERVICE_CONTROL_DEVICEEVENT! ");
        switch(dwEventType)
        {
        case DBT_DEVICEARRIVAL:
            // Handle
            fprintf(Log, "DBT_DEVICEARRIVAL\n");
            break;
        case DBT_DEVICEREMOVECOMPLETE:
            fprintf(Log, "DBT_DEVICEREMOVECOMPLETE\n");
            // Handle
            break;
        }
    default:
        fprintf(Log, "Unknown dwControl: %d\n", dwControl);
        SetServiceStatus(hStatus, &Status);
        break;
    }
    return NO_ERROR;
}