如何在Windows服务程序中添加U盘插拔的消息

研究了下这个问题，主要要在一般的windows服务程序中修改两个地方：

一、调用RegisterServiceCtrlHandlerEx

VOID WINAPI SvcMain( DWORD dwArgc, LPTSTR *lpszArgv )
{
    // Register the handler function for the service.

    gSvcStatusHandle = RegisterServiceCtrlHandlerEx(
        SVCNAME,
        SvcCtrlHandler,
        NULL);

    if( !gSvcStatusHandle )
    {
        printf("Failed to register service control handler");
        return;
    }

    // These SERVICE_STATUS members remain as set here.

    gSvcStatus.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
    gSvcStatus.dwServiceSpecificExitCode = 0;    

    // Report initial status to the SCM.

    ReportSvcStatus( SERVICE_START_PENDING, NO_ERROR, 3000 );

    // Perform service-specific initialization and work.

    HDEVNOTIFY hDevNotify;

    if(!DoRegisterDeviceInterface(GUID_DEVINTERFACE_VOLUME, &hDevNotify))
    {
        printf("DoRegisterDeviceInterface failed ");
        ReportSvcStatus( SERVICE_STOPPED, NO_ERROR, 0 );
        return;
    }

    SvcInit( dwArgc, lpszArgv );

    UnregisterDeviceNotification(hDevNotify);
}

二、注册设备响应接口

/*
Routine Description:
    Registers for notification of changes in the device interfaces for
    the specified interface class GUID.

Parameters:
    InterfaceClassGuid - The interface class GUID for the device
        interfaces.

    hDevNotify - Receives the device notification handle. On failure,
        this value is NULL.

Return Value:
    If the function succeeds, the return value is TRUE.
    If the function fails, the return value is FALSE.
*/

BOOL DoRegisterDeviceInterface(
    GUID InterfaceClassGuid,
    HDEVNOTIFY *hDevNotify
)

{
    DEV_BROADCAST_DEVICEINTERFACE NotificationFilter;

    ZeroMemory( &NotificationFilter, sizeof(NotificationFilter) );
    NotificationFilter.dbcc_size =
        sizeof(DEV_BROADCAST_DEVICEINTERFACE);
    NotificationFilter.dbcc_devicetype = DBT_DEVTYP_DEVICEINTERFACE;
    NotificationFilter.dbcc_classguid = InterfaceClassGuid;

    *hDevNotify = RegisterDeviceNotification(
        gSvcStatusHandle,
        &NotificationFilter,
        DEVICE_NOTIFY_SERVICE_HANDLE
    );

    if(!*hDevNotify)
    {
        printf("RegisterDeviceNotification failed: %d ",
                GetLastError());
        return FALSE;
    }

    return TRUE;
}

三、处理U盘插拔消息

DWORD WINAPI SvcCtrlHandler(DWORD dwCtrl, DWORD dwEventType, LPVOID lpEventData, LPVOID lpContext)
{
    // Handle the requested control code.
    switch(dwCtrl)
    {  
    case SERVICE_CONTROL_STOP:
        ReportSvcStatus(SERVICE_STOP_PENDING, NO_ERROR, 0);

        // Signal the service to stop.

        SetEvent(ghSvcStopEvent);
        
        return NO_ERROR;

    case SERVICE_CONTROL_DEVICEEVENT:
        if (DBT_DEVICEARRIVAL  == dwEventType)
        {
            FILE *fp;
            
            DEV_BROADCAST_HDR * pHdr = (DEV_BROADCAST_HDR *)lpEventData;
            PDEV_BROADCAST_DEVICEINTERFACE pDevInf;
            PDEV_BROADCAST_HANDLE pDevHnd;
            PDEV_BROADCAST_OEM pDevOem;
            PDEV_BROADCAST_PORT pDevPort;
            PDEV_BROADCAST_VOLUME pDevVolume;

            switch( pHdr->dbch_devicetype )
            {
            case DBT_DEVTYP_DEVICEINTERFACE:
                pDevInf = (PDEV_BROADCAST_DEVICEINTERFACE)pHdr;

                fp = fopen("C:\log.txt", "a+");
                if (fp)
                {
                    _ftprintf(fp, _T("%s "), pDevInf->dbcc_name);
                    fclose(fp);
                }
                break;
            case DBT_DEVTYP_HANDLE:
                pDevHnd = (PDEV_BROADCAST_HANDLE)pHdr;
                break;
            case DBT_DEVTYP_OEM:
                pDevOem = (PDEV_BROADCAST_OEM)pHdr;
                break;
            case DBT_DEVTYP_PORT:
                pDevPort = (PDEV_BROADCAST_PORT)pHdr;
                break;
            case DBT_DEVTYP_VOLUME:
                pDevVolume = (PDEV_BROADCAST_VOLUME)pHdr;
                break;
            }
        }
        break;

    case SERVICE_CONTROL_INTERROGATE:
        // Fall through to send current status.
        break;

    default:
        break;
    }

    ReportSvcStatus(gSvcStatus.dwCurrentState, NO_ERROR, 0);
    return NO_ERROR;
}