自动检测USB设备连接/断开连接

时间:2016-11-02 20:20:35

标签: macos cocoa notifications usb

我有一个Cocoa应用程序需要在设备连接或断开USB端口时得到通知。我可以让DeviceConnected回调工作,但断开USB设备时不会调用DeviceDisconnected函数。

以下是我的代码:

+ (void)listenForUSBEvents
{
   io_iterator_t  portIterator = 0;
   CFMutableDictionaryRef  matchingDict = IOServiceMatching( kIOUSBDeviceClassName );
   IONotificationPortRef  notifyPort = IONotificationPortCreate( kIOMasterPortDefault );
   CFRunLoopSourceRef  runLoopSource = IONotificationPortGetRunLoopSource( notifyPort );
   CFRunLoopRef  runLoop = CFRunLoopGetCurrent();

   CFRunLoopAddSource( runLoop, runLoopSource, kCFRunLoopDefaultMode);
   CFRetain( matchingDict );

   kern_return_t  returnCode = IOServiceAddMatchingNotification( notifyPort, kIOMatchedNotification, matchingDict, DeviceConnected, NULL, &portIterator );

   if ( returnCode == 0 )
   {
      DeviceConnected( nil, portIterator );
   }

   returnCode = IOServiceAddMatchingNotification( notifyPort, kIOMatchedNotification, matchingDict, DeviceDisconnected, NULL, &portIterator );

   if ( returnCode == 0 )
   {
      DeviceDisconnected( nil, portIterator );
   }
}

@end


void DeviceConnected( void *refCon, io_iterator_t iterator )
{
   kern_return_t  returnCode = KERN_FAILURE;
   io_object_t  usbDevice;

   while ( ( usbDevice = IOIteratorNext( iterator ) ) )
   {
     io_name_t name;

     returnCode = IORegistryEntryGetName( usbDevice, name );

     if ( returnCode != KERN_SUCCESS )
     {
        return;
     }

     [[NSNotificationCenter defaultCenter] postNotificationName:deviceConnectedNotification object:nil userInfo:nil];
   }
}

void DeviceDisconnected( void *refCon, io_iterator_t iterator )
{
   [[NSNotificationCenter defaultCenter] postNotificationName:deviceDiconnectedNotification object:nil userInfo:nil];
}

1 个答案:

答案 0 :(得分:0)

我弄清楚我做错了什么。

首先,deviceDisconnected的IOServiceAddMatchingNotification应如下所示:

returnCode = IOServiceAddMatchingNotification( notifyPort, kIOTerminatedNotification, matchingDict, DeviceDisconnected, NULL, &portIterator );

其次,DeviceDisconnected函数应如下所示:

void DeviceDisconnected( void *refCon, io_iterator_t iterator )
{
    kern_return_t    returnCode = KERN_FAILURE;
    io_object_t      usbDevice;

    while ( ( usbDevice = ioIteratorNext( iterator ) ) )
    {
        returnCode = IOObjectRelease( usbDevice );

        if ( returnCode != kIOReturnSuccess )
        {
            NSLog( @"Couldn't release raw device object: %08x.", returnCode );
        }
    }

    [[NSNotificationCenter defaultCenter] postNotificationName:deviceDiconnectedNotification object:nil userInfo:nil];
}