X-Git-Url: https://git.saurik.com/apple/xnu.git/blobdiff_plain/cb3231590a3c94ab4375e2228bd5e86b0cf1ad7e..bca245acd4c03fd752d1a45f011ad495e60fe53d:/iokit/Kernel/IOUserClient.cpp?ds=inline

diff --git a/iokit/Kernel/IOUserClient.cpp b/iokit/Kernel/IOUserClient.cpp
index bbe9448fd..e98f41f36 100644
--- a/iokit/Kernel/IOUserClient.cpp
+++ b/iokit/Kernel/IOUserClient.cpp
@@ -548,7 +548,31 @@ extern "C" {
 // functions called from osfmk/device/iokit_rpc.c
 
 void
-iokit_add_reference( io_object_t obj, ipc_kobject_type_t type )
+iokit_port_object_description(io_object_t obj, kobject_description_t desc)
+{
+	IORegistryEntry    * regEntry;
+	IOUserNotification * __unused noti;
+	_IOServiceNotifier * __unused serviceNoti;
+	OSSerialize        * __unused s;
+
+	if ((regEntry = OSDynamicCast(IORegistryEntry, obj))) {
+		snprintf(desc, KOBJECT_DESCRIPTION_LENGTH, "%s(0x%qx)", obj->getMetaClass()->getClassName(), regEntry->getRegistryEntryID());
+#if DEVELOPMENT || DEBUG
+	} else if ((noti = OSDynamicCast(IOUserNotification, obj))
+	    && ((serviceNoti = OSDynamicCast(_IOServiceNotifier, noti->holdNotify)))) {
+		s = OSSerialize::withCapacity(page_size);
+		if (s && serviceNoti->matching->serialize(s)) {
+			snprintf(desc, KOBJECT_DESCRIPTION_LENGTH, "%s(%s)", obj->getMetaClass()->getClassName(), s->text());
+		}
+		OSSafeReleaseNULL(s);
+#endif /* DEVELOPMENT || DEBUG */
+	} else {
+		snprintf(desc, KOBJECT_DESCRIPTION_LENGTH, "%s", obj->getMetaClass()->getClassName());
+	}
+}
+
+void
+iokit_add_reference( io_object_t obj, natural_t type )
 {
 	IOUserClient * uc;
 
@@ -2072,7 +2096,8 @@ IOUserClient::_sendAsyncResult64(OSAsyncReference64 reference,
 		replyMsg.m.msg64.notifyHdr.size = sizeof(IOAsyncCompletionContent)
 		    + numArgs * sizeof(io_user_reference_t);
 		replyMsg.m.msg64.notifyHdr.type = kIOAsyncCompletionNotificationType;
-		bcopy(reference, replyMsg.m.msg64.notifyHdr.reference, sizeof(OSAsyncReference64));
+		/* Copy reference except for reference[0], which is left as 0 from the earlier bzero */
+		bcopy(&reference[1], &replyMsg.m.msg64.notifyHdr.reference[1], sizeof(OSAsyncReference64) - sizeof(reference[0]));
 
 		replyMsg.m.msg64.asyncContent.result = result;
 		if (numArgs) {
@@ -2089,7 +2114,8 @@ IOUserClient::_sendAsyncResult64(OSAsyncReference64 reference,
 		    + numArgs * sizeof(uint32_t);
 		replyMsg.m.msg32.notifyHdr.type = kIOAsyncCompletionNotificationType;
 
-		for (idx = 0; idx < kOSAsyncRefCount; idx++) {
+		/* Skip reference[0] which is left as 0 from the earlier bzero */
+		for (idx = 1; idx < kOSAsyncRefCount; idx++) {
 			replyMsg.m.msg32.notifyHdr.reference[idx] = REF32(reference[idx]);
 		}
 
@@ -3548,10 +3574,12 @@ GetPropertiesEditor(void                  * reference,
 
 #endif /* CONFIG_MACF */
 
-/* Routine io_registry_entry_get_properties */
+/* Routine io_registry_entry_get_properties_bin_buf */
 kern_return_t
-is_io_registry_entry_get_properties_bin(
+is_io_registry_entry_get_properties_bin_buf(
 	io_object_t registry_entry,
+	mach_vm_address_t buf,
+	mach_vm_size_t *bufsize,
 	io_buf_ptr_t *properties,
 	mach_msg_type_number_t *propertiesCnt)
 {
@@ -3585,21 +3613,48 @@ is_io_registry_entry_get_properties_bin(
 
 	if (kIOReturnSuccess == err) {
 		len = s->getLength();
-		*propertiesCnt = len;
-		err = copyoutkdata(s->text(), len, properties);
+		if (buf && bufsize && len <= *bufsize) {
+			*bufsize = len;
+			*propertiesCnt = 0;
+			*properties = nullptr;
+			if (copyout(s->text(), buf, len)) {
+				err = kIOReturnVMError;
+			} else {
+				err = kIOReturnSuccess;
+			}
+		} else {
+			if (bufsize) {
+				*bufsize = 0;
+			}
+			*propertiesCnt = len;
+			err = copyoutkdata( s->text(), len, properties );
+		}
 	}
 	s->release();
 
 	return err;
 }
 
-/* Routine io_registry_entry_get_property_bin */
+/* Routine io_registry_entry_get_properties_bin */
 kern_return_t
-is_io_registry_entry_get_property_bin(
+is_io_registry_entry_get_properties_bin(
+	io_object_t registry_entry,
+	io_buf_ptr_t *properties,
+	mach_msg_type_number_t *propertiesCnt)
+{
+	return is_io_registry_entry_get_properties_bin_buf(registry_entry,
+	           0, NULL, properties, propertiesCnt);
+}
+
+/* Routine io_registry_entry_get_property_bin_buf */
+kern_return_t
+is_io_registry_entry_get_property_bin_buf(
 	io_object_t registry_entry,
 	io_name_t plane,
 	io_name_t property_name,
 	uint32_t options,
+	mach_vm_address_t buf,
+	mach_vm_size_t *bufsize,
 	io_buf_ptr_t *properties,
 	mach_msg_type_number_t *propertiesCnt )
 {
@@ -3648,8 +3703,22 @@ is_io_registry_entry_get_property_bin(
 
 	if (obj->serialize( s )) {
 		len = s->getLength();
-		*propertiesCnt = len;
-		err = copyoutkdata( s->text(), len, properties );
+		if (buf && bufsize && len <= *bufsize) {
+			*bufsize = len;
+			*propertiesCnt = 0;
+			*properties = nullptr;
+			if (copyout(s->text(), buf, len)) {
+				err = kIOReturnVMError;
+			} else {
+				err = kIOReturnSuccess;
+			}
+		} else {
+			if (bufsize) {
+				*bufsize = 0;
+			}
+			*propertiesCnt = len;
+			err = copyoutkdata( s->text(), len, properties );
+		}
 	} else {
 		err = kIOReturnUnsupported;
 	}
@@ -3660,6 +3729,20 @@ is_io_registry_entry_get_property_bin(
 	return err;
 }
 
+/* Routine io_registry_entry_get_property_bin */
+kern_return_t
+is_io_registry_entry_get_property_bin(
+	io_object_t registry_entry,
+	io_name_t plane,
+	io_name_t property_name,
+	uint32_t options,
+	io_buf_ptr_t *properties,
+	mach_msg_type_number_t *propertiesCnt )
+{
+	return is_io_registry_entry_get_property_bin_buf(registry_entry, plane,
+	           property_name, options, 0, NULL, properties, propertiesCnt);
+}
+
 
 /* Routine io_registry_entry_set_properties */
 kern_return_t
@@ -5586,28 +5669,7 @@ is_io_catalog_terminate(
 	switch (flag) {
 #if !defined(SECURE_KERNEL)
 	case kIOCatalogServiceTerminate:
-		OSIterator *        iter;
-		IOService *         service;
-
-		iter = IORegistryIterator::iterateOver(gIOServicePlane,
-		    kIORegistryIterateRecursively);
-		if (!iter) {
-			return kIOReturnNoMemory;
-		}
-
-		do {
-			iter->reset();
-			while ((service = (IOService *)iter->getNextObject())) {
-				if (service->metaCast(name)) {
-					if (!service->terminate( kIOServiceRequired
-					    | kIOServiceSynchronous)) {
-						kr = kIOReturnUnsupported;
-						break;
-					}
-				}
-			}
-		} while (!service && !iter->isValid());
-		iter->release();
+		kr = gIOCatalogue->terminateDrivers(NULL, name);
 		break;
 
 	case kIOCatalogModuleUnload: