xnu-4903.231.4.tar.gz

[apple/xnu.git] / iokit / Kernel / IOUserClient.cpp

diff --git a/iokit/Kernel/IOUserClient.cpp b/iokit/Kernel/IOUserClient.cpp
index d331cb2b0a3af0e86dcd791abe62137a82af9c00..b99f027ddd6fa28a952469428e29dd759f473360 100644 (file)
--- a/iokit/Kernel/IOUserClient.cpp
+++ b/iokit/Kernel/IOUserClient.cpp
@@ -383,6 +383,7 @@ public:
     virtual void reset() APPLE_KEXT_OVERRIDE;
     virtual bool isValid() APPLE_KEXT_OVERRIDE;
     virtual OSObject * getNextObject() APPLE_KEXT_OVERRIDE;
     virtual void reset() APPLE_KEXT_OVERRIDE;
     virtual bool isValid() APPLE_KEXT_OVERRIDE;
     virtual OSObject * getNextObject() APPLE_KEXT_OVERRIDE;

+    virtual OSObject * copyNextObject();

 };
 
 /* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
 };
 
 /* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */


@@ -471,11 +472,20 @@ IOUserIterator::isValid()
 OSObject *
 IOUserIterator::getNextObject()
 {
 OSObject *
 IOUserIterator::getNextObject()
 {

-    OSObject * ret;
+    assert(false);
+    return (NULL);
+}
+
+OSObject *
+IOUserIterator::copyNextObject()
+{
+    OSObject * ret = NULL;

 
     IOLockLock(lock);
 
     IOLockLock(lock);

-    assert(OSDynamicCast(OSIterator, userIteratorObject));
-    ret = ((OSIterator *)userIteratorObject)->getNextObject();
+    if (userIteratorObject) {
+        ret = ((OSIterator *)userIteratorObject)->getNextObject();
+        if (ret) ret->retain();
+    }

     IOLockUnlock(lock);
 
     return (ret);
     IOLockUnlock(lock);
 
     return (ret);


@@ -616,7 +626,6 @@ class IOServiceUserNotification : public IOUserNotification
     PingMsg    *       pingMsg;
     vm_size_t          msgSize;
     OSArray    *       newSet;
     PingMsg    *       pingMsg;
     vm_size_t          msgSize;
     OSArray    *       newSet;

-    OSObject   *       lastEntry;

     bool               armed;
     bool                ipcLogged;
 
     bool               armed;
     bool                ipcLogged;
 


@@ -633,6 +642,7 @@ public:
     virtual bool handler( void * ref, IOService * newService );
 
     virtual OSObject * getNextObject() APPLE_KEXT_OVERRIDE;
     virtual bool handler( void * ref, IOService * newService );
 
     virtual OSObject * getNextObject() APPLE_KEXT_OVERRIDE;

+    virtual OSObject * copyNextObject() APPLE_KEXT_OVERRIDE;

 };
 
 class IOServiceMessageUserNotification : public IOUserNotification
 };
 
 class IOServiceMessageUserNotification : public IOUserNotification


@@ -670,6 +680,7 @@ public:
                               void * messageArgument, vm_size_t argSize );
 
     virtual OSObject * getNextObject() APPLE_KEXT_OVERRIDE;
                               void * messageArgument, vm_size_t argSize );
 
     virtual OSObject * getNextObject() APPLE_KEXT_OVERRIDE;

+    virtual OSObject * copyNextObject() APPLE_KEXT_OVERRIDE;

 };
 
 /* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
 };
 
 /* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */


@@ -776,11 +787,9 @@ void IOServiceUserNotification::free( void )
     PingMsg   *        _pingMsg;
     vm_size_t  _msgSize;
     OSArray   *        _newSet;
     PingMsg   *        _pingMsg;
     vm_size_t  _msgSize;
     OSArray   *        _newSet;

-    OSObject  *        _lastEntry;

 
     _pingMsg   = pingMsg;
     _msgSize   = msgSize;
 
     _pingMsg   = pingMsg;
     _msgSize   = msgSize;

-    _lastEntry = lastEntry;

     _newSet    = newSet;
 
     super::free();
     _newSet    = newSet;
 
     super::free();


@@ -792,9 +801,6 @@ void IOServiceUserNotification::free( void )
         IOFree(_pingMsg, _msgSize);
        }
 
         IOFree(_pingMsg, _msgSize);
        }
 

-    if( _lastEntry)
-        _lastEntry->release();
-

     if( _newSet)
         _newSet->release();
 }
     if( _newSet)
         _newSet->release();
 }


@@ -850,16 +856,19 @@ bool IOServiceUserNotification::handler( void * ref,
 
     return( true );
 }
 
     return( true );
 }

-

 OSObject * IOServiceUserNotification::getNextObject()
 OSObject * IOServiceUserNotification::getNextObject()

+{
+    assert(false);
+    return (NULL);
+}
+
+OSObject * IOServiceUserNotification::copyNextObject()

 {
     unsigned int       count;
     OSObject *         result;
 {
     unsigned int       count;
     OSObject *         result;

-    OSObject *         releaseEntry;

 
     IOLockLock(lock);
 
 
     IOLockLock(lock);
 

-    releaseEntry = lastEntry;

     count = newSet->getCount();
     if( count ) {
         result = newSet->getObject( count - 1 );
     count = newSet->getCount();
     if( count ) {
         result = newSet->getObject( count - 1 );


@@ -869,12 +878,9 @@ OSObject * IOServiceUserNotification::getNextObject()
         result = 0;
         armed = true;
     }
         result = 0;
         armed = true;
     }

-    lastEntry = result;

 
     IOLockUnlock(lock);
 
 
     IOLockUnlock(lock);
 

-    if (releaseEntry) releaseEntry->release();
-

     return( result );
 }
 
     return( result );
 }
 


@@ -1068,6 +1074,11 @@ OSObject * IOServiceMessageUserNotification::getNextObject()
     return( 0 );
 }
 
     return( 0 );
 }
 

+OSObject * IOServiceMessageUserNotification::copyNextObject()
+{
+    return( NULL );
+}
+

 /* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
 
 #undef super
 /* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
 
 #undef super


@@ -1438,7 +1449,7 @@ IOUserClient::registerOwner(task_t task)
     if (newOwner)
     {
         owner = IONew(IOUserClientOwner, 1);
     if (newOwner)
     {
         owner = IONew(IOUserClientOwner, 1);

-        if (!newOwner) ret = kIOReturnNoMemory;
+        if (!owner) ret = kIOReturnNoMemory;

         else
         {
             owner->task = task;
         else
         {
             owner->task = task;


@@ -2049,12 +2060,24 @@ kern_return_t is_io_iterator_next(
 {
     IOReturn    ret;
     OSObject * obj;
 {
     IOReturn    ret;
     OSObject * obj;

+    OSIterator * iter;
+    IOUserIterator * uiter;

 
 

-    CHECK( OSIterator, iterator, iter );
+       if ((uiter = OSDynamicCast(IOUserIterator, iterator)))
+       {
+               obj = uiter->copyNextObject();
+       }
+       else if ((iter = OSDynamicCast(OSIterator, iterator)))
+       {
+               obj = iter->getNextObject();
+               if (obj) obj->retain();
+       }
+       else
+       {
+           return( kIOReturnBadArgument );
+       }

 
 

-    obj = iter->getNextObject();

     if( obj) {
     if( obj) {

-       obj->retain();

        *object = obj;
         ret = kIOReturnSuccess;
     } else
        *object = obj;
         ret = kIOReturnSuccess;
     } else


@@ -3292,8 +3315,8 @@ kern_return_t is_io_registry_entry_get_child_iterator(
 {
     CHECK( IORegistryEntry, registry_entry, entry );
 
 {
     CHECK( IORegistryEntry, registry_entry, entry );
 

-    *iterator = entry->getChildIterator(
-    IORegistryEntry::getPlane( plane ));
+    *iterator = IOUserIterator::withIterator(entry->getChildIterator(
+    IORegistryEntry::getPlane( plane )));

 
     return( kIOReturnSuccess );
 }
 
     return( kIOReturnSuccess );
 }


@@ -3306,8 +3329,8 @@ kern_return_t is_io_registry_entry_get_parent_iterator(
 {
     CHECK( IORegistryEntry, registry_entry, entry );
 
 {
     CHECK( IORegistryEntry, registry_entry, entry );
 

-    *iterator = entry->getParentIterator(
-       IORegistryEntry::getPlane( plane ));
+    *iterator = IOUserIterator::withIterator(entry->getParentIterator(
+       IORegistryEntry::getPlane( plane )));

 
     return( kIOReturnSuccess );
 }
 
     return( kIOReturnSuccess );
 }


@@ -4944,7 +4967,7 @@ kern_return_t is_io_catalog_send_data(
         return kIOReturnBadArgument;
     }
 
         return kIOReturnBadArgument;
     }
 

-    if (!IOTaskHasEntitlement(current_task(), "com.apple.rootless.kext-management"))
+    if (!IOTaskHasEntitlement(current_task(), "com.apple.rootless.kext-secure-management"))

     {
         OSString * taskName = IOCopyLogNameForPID(proc_selfpid());
         IOLog("IOCatalogueSendData(%s): Not entitled\n", taskName ? taskName->getCStringNoCopy() : "");
     {
         OSString * taskName = IOCopyLogNameForPID(proc_selfpid());
         IOLog("IOCatalogueSendData(%s): Not entitled\n", taskName ? taskName->getCStringNoCopy() : "");