+++ /dev/null
-/*
- * Copyright (c) 2004-2007,2011 Apple Inc. All Rights Reserved.
- *
- * @APPLE_LICENSE_HEADER_START@
- *
- * This file contains Original Code and/or Modifications of Original Code
- * as defined in and that are subject to the Apple Public Source License
- * Version 2.0 (the 'License'). You may not use this file except in
- * compliance with the License. Please obtain a copy of the License at
- * http://www.opensource.apple.com/apsl/ and read it before using this
- * file.
- *
- * The Original Code and all software distributed under the License are
- * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
- * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
- * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
- * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
- * Please see the License for the specific language governing rights and
- * limitations under the License.
- *
- * @APPLE_LICENSE_HEADER_END@
- */
-
-
-//
-// iodevices - code for finding and tracking devices via IOKit
-//
-#include "iodevices.h"
-#include <security_utilities/cfutilities.h>
-#include <IOKit/IOMessage.h>
-#include <IOKit/usb/IOUSBLib.h>
-
-using namespace MachPlusPlus;
-
-
-namespace Security {
-namespace IOKit {
-
-
-//
-// Master Ports.
-// Note that multiple MasterPort objects may refer to the same Mach port.
-//
-MasterPort::MasterPort()
-{
- Error::check(::IOMasterPort(MACH_PORT_NULL, &port()));
-}
-
-
-//
-// IOKit Devices (a small subset of functionality)
-//
-Device::~Device()
-{
- ::IOObjectRelease(mService);
-}
-
-string Device::name() const
-{
- io_name_t tName;
- Error::check(::IORegistryEntryGetName(mService, tName));
- return tName;
-}
-
-string Device::name(const char *plane) const
-{
- io_name_t tName;
- Error::check(::IORegistryEntryGetNameInPlane(mService, plane, tName));
- return tName;
-}
-
-string Device::path(const char *plane) const
-{
- io_string_t tPath;
- Error::check(::IORegistryEntryGetPath(mService, plane, tPath));
- return tPath;
-}
-
-CFDictionaryRef Device::properties() const
-{
- CFMutableDictionaryRef dict;
- Error::check(::IORegistryEntryCreateCFProperties(mService, &dict, NULL, 0));
- return dict;
-}
-
-CFTypeRef Device::property(const char *name) const
-{
- return ::IORegistryEntryCreateCFProperty(mService, CFTempString(name), NULL, 0);
-}
-
-
-//
-// DeviceIterators
-//
-DeviceIterator::~DeviceIterator()
-{
- // drain the iterator to avoid port leakage
- while (Device dev = (*this)())
- ;
-}
-
-
-io_service_t DeviceIterator::operator () ()
-{
- io_service_t dev = ::IOIteratorNext(mIterator);
- mAtEnd = !dev;
- return dev;
-}
-
-
-//
-// DeviceMatches
-//
-DeviceMatch::DeviceMatch()
- : CFRef<CFMutableDictionaryRef>(makeCFMutableDictionary())
-{
- CFError::check(*this);
-}
-
-DeviceMatch::DeviceMatch(const char *cls)
- : CFRef<CFMutableDictionaryRef>(::IOServiceMatching(cls))
-{
- CFError::check(*this);
-}
-
-DeviceMatch::DeviceMatch(const char *cls, const char *name, uint32_t value, ...)
- : CFRef<CFMutableDictionaryRef>(::IOServiceMatching(cls))
-{
- CFError::check(*this);
-
- va_list args;
- va_start(args, value);
- while (name) {
- add(name, value);
- name = va_arg(args, const char *);
- if (!name)
- break;
- value = va_arg(args, uint32_t);
- }
-}
-
-void DeviceMatch::add(const char *name, uint32_t value)
-{
- CFRef<CFNumberRef> number = CFNumberCreate(NULL, kCFNumberSInt32Type, &value);
- CFDictionarySetValue(*this, CFTempString(name), number);
-}
-
-
-//
-// NotificationPorts
-//
-NotificationPort::NotificationPort()
-{
- CFError::check(mPortRef = ::IONotificationPortCreate(MasterPort()));
-}
-
-NotificationPort::NotificationPort(const MasterPort &master)
-{
- CFError::check(mPortRef = ::IONotificationPortCreate(master));
-}
-
-NotificationPort::~NotificationPort()
-{
- ::IONotificationPortDestroy(mPortRef);
-}
-
-
-mach_port_t NotificationPort::port() const
-{
- mach_port_t p = ::IONotificationPortGetMachPort(mPortRef);
- CFError::check(p);
- return p;
-}
-
-CFRunLoopSourceRef NotificationPort::source() const
-{
- CFRunLoopSourceRef rls = ::IONotificationPortGetRunLoopSource(mPortRef);
- CFError::check(rls);
- return rls;
-}
-
-
-void NotificationPort::add(const DeviceMatch &match, Receiver &receiver, const char *type)
-{
- io_iterator_t iterator;
- CFRetain(match); // compensate for IOSAMN not retaining its argument
- Error::check(::IOServiceAddMatchingNotification(mPortRef, type,
- match,
- ioNotify, &receiver,
- &iterator));
-
- // run initial iterator to process existing devices
- secdebug("iokit", "dispatching initial device match iterator %d", iterator);
- DeviceIterator it(iterator);
- receiver.ioChange(it);
-}
-
-void NotificationPort::addInterestNotification(Receiver &receiver, io_service_t service,
- const io_name_t interestType)
-{
- io_iterator_t iterator;
- mach_port_t pp = NotificationPort::port();
-
- secdebug("iokit", "NotificationPort::addInterest - type: %s [port: %p (0x%08X), service: 0x%08X]",
- interestType, mPortRef, pp, service);
-
- // We cannot throw if we get an error here since we will receive notifications
- // from each plane, and not all planes have the necessary information to be
- // able to add an interest notification
- kern_return_t kr = ::IOServiceAddInterestNotification(mPortRef,
- service, interestType, ioDeviceNotification, &receiver, &iterator);
- const char *msgstr = mach_error_string(kr);
- const char *msgtyp = mach_error_type(kr);
- if (msgstr && msgtyp)
- secdebug("iokit", " msg: %s, typ: %s", msgstr, msgtyp);
-}
-
-void NotificationPort::ioNotify(void *refCon, io_iterator_t iterator)
-{
- secdebug("iokit", "dispatching new device match iterator %d", iterator);
- DeviceIterator it(iterator);
- try {
- reinterpret_cast<Receiver *>(refCon)->ioChange(it);
- } catch (...) {
- secdebug("iokit", "ioChange callback threw an exception (ignored)");
- }
-}
-
-void NotificationPort::ioDeviceNotification(void *refCon, io_service_t service,
- natural_t messageType, void *messageArgument)
-{
- secdebug("iokit", "dispatching NEW device notification iterator, service 0x%08X, msg: 0x%04X, arg: %p",
- service, messageType, messageArgument);
-
- const char *msgstr = mach_error_string(messageType);
- const char *msgtyp = mach_error_type(messageType);
- if (msgstr && msgtyp)
- secdebug("iokit", " msg: %s, typ: %s", msgstr, msgtyp);
-
- if (service!=io_service_t(-1))
- reinterpret_cast<Receiver *>(refCon)->ioServiceChange(refCon, service, messageType, messageArgument);
-}
-
-//
-// Abstract NotificationPort::Receivers
-//
-NotificationPort::Receiver::~Receiver()
-{ /* virtual */ }
-
-
-//
-// MachPortNotificationPorts
-//
-MachPortNotificationPort::MachPortNotificationPort()
-{
- NoReplyHandler::port(NotificationPort::port());
-}
-
-MachPortNotificationPort::~MachPortNotificationPort()
-{
-}
-
-boolean_t MachPortNotificationPort::handle(mach_msg_header_t *in)
-{
- ::IODispatchCalloutFromMessage(NULL, in, mPortRef);
- return TRUE;
-}
-
-
-
-} // end namespace IOKit
-} // end namespace Security
projects / apple / security.git / blobdiff