X-Git-Url: https://git.saurik.com/apple/xnu.git/blobdiff_plain/f427ee49d309d8fc33ebf3042c3a775f2f530ded..refs/heads/master:/iokit/Kernel/IOService.cpp

diff --git a/iokit/Kernel/IOService.cpp b/iokit/Kernel/IOService.cpp
index c99de8858..04b3faf94 100644
--- a/iokit/Kernel/IOService.cpp
+++ b/iokit/Kernel/IOService.cpp
@@ -370,6 +370,7 @@ IOService           * fSystemPowerAckTo;
 uint32_t              fSystemPowerAckRef;
 uint8_t               fSystemOff;
 uint8_t               fUserServerOff;
+uint8_t               fWaitingUserServers;
 
 void lock();
 void unlock();
@@ -4150,6 +4151,12 @@ IOServicePH::serverRemove(IOUserServer * server)
 	if (idx != -1U) {
 		fUserServers->removeObject(idx);
 	}
+
+	if (fWaitingUserServers) {
+		fWaitingUserServers = false;
+		IOLockWakeup(gJobsLock, &fWaitingUserServers, /* one-thread */ false);
+	}
+
 	unlock();
 }
 
@@ -4275,6 +4282,41 @@ IOServicePH::matchingEnd(IOService * service)
 	serverAck(NULL);
 }
 
+
+void
+IOServicePH::systemHalt(void)
+{
+	OSArray * notifyServers;
+	uint64_t  deadline;
+
+	lock();
+	notifyServers = OSArray::withArray(fUserServers);
+	unlock();
+
+	if (notifyServers) {
+		notifyServers->iterateObjects(^bool (OSObject * obj) {
+			IOUserServer * us;
+			us = (typeof(us))obj;
+			us->systemHalt();
+			return false;
+		});
+		OSSafeReleaseNULL(notifyServers);
+	}
+
+	lock();
+	clock_interval_to_deadline(1000, kMillisecondScale, &deadline);
+	while (0 < fUserServers->getCount()) {
+		fWaitingUserServers = true;
+		__assert_only int waitResult =
+		    IOLockSleepDeadline(gJobsLock, &fWaitingUserServers, deadline, THREAD_UNINT);
+		assert((THREAD_AWAKENED == waitResult) || (THREAD_TIMED_OUT == waitResult));
+		if (THREAD_TIMED_OUT == waitResult) {
+			break;
+		}
+	}
+	unlock();
+}
+
 bool
 IOServicePH::serverSlept(void)
 {