如何在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\n");
      
 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\n",

                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\n"),
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;
}