研究了下这个问题,主要要在一般的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;
}