如何在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;
}

巴特西

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

最新文章

热门文章