X-Git-Url: https://git.saurik.com/apple/xnu.git/blobdiff_plain/7ee9d059c4eecf68ae4f8b0fb99ae2471eda79af..ebb1b9f42b62218f29061826217bb0f71cd375a6:/iokit/Kernel/IOService.cpp

diff --git a/iokit/Kernel/IOService.cpp b/iokit/Kernel/IOService.cpp
index 6ef0b3413..f08348272 100644
--- a/iokit/Kernel/IOService.cpp
+++ b/iokit/Kernel/IOService.cpp
@@ -115,6 +115,7 @@ const OSSymbol *		gIOConsoleSessionUIDKey;
 const OSSymbol *		gIOConsoleSessionAuditIDKey;
 const OSSymbol *		gIOConsoleUsersSeedKey;
 const OSSymbol *		gIOConsoleSessionOnConsoleKey;
+const OSSymbol *		gIOConsoleSessionLoginDoneKey;
 const OSSymbol *		gIOConsoleSessionSecureInputPIDKey;
 const OSSymbol *		gIOConsoleSessionScreenLockedTimeKey;
 
@@ -322,6 +323,7 @@ void IOService::initialize( void )
 
     gIOConsoleUsersSeedKey	         = OSSymbol::withCStringNoCopy(kIOConsoleUsersSeedKey);
     gIOConsoleSessionOnConsoleKey        = OSSymbol::withCStringNoCopy(kIOConsoleSessionOnConsoleKey);
+    gIOConsoleSessionLoginDoneKey        = OSSymbol::withCStringNoCopy(kIOConsoleSessionLoginDoneKey);
     gIOConsoleSessionSecureInputPIDKey   = OSSymbol::withCStringNoCopy(kIOConsoleSessionSecureInputPIDKey);
     gIOConsoleSessionScreenLockedTimeKey = OSSymbol::withCStringNoCopy(kIOConsoleSessionScreenLockedTimeKey);
 
@@ -361,6 +363,8 @@ void IOService::initialize( void )
 
     gIOConsoleLockCallout = thread_call_allocate(&IOService::consoleLockTimer, NULL);
 
+    IORegistryEntry::getRegistryRoot()->setProperty(gIOConsoleLockedKey, kOSBooleanTrue);
+
     assert( gIOServiceBusyLock && gJobs && gJobsLock && gIOConsoleUsersLock
     		&& gIOConsoleLockCallout && (err == KERN_SUCCESS) );
 
@@ -4234,6 +4238,7 @@ void IOService::updateConsoleUsers(OSArray * consoleUsers, IOMessage systemMessa
     IORegistryEntry * regEntry;
     OSObject *        locked = kOSBooleanFalse;
     uint32_t          idx;
+    bool              loggedIn;
     bool              publish;
     OSDictionary *    user;
     static IOMessage  sSystemPower;
@@ -4246,38 +4251,45 @@ void IOService::updateConsoleUsers(OSArray * consoleUsers, IOMessage systemMessa
     {
         sSystemPower = systemMessage;
     }
+    loggedIn = false;
     if (consoleUsers)
     {
         OSNumber * num = 0;
 	for (idx = 0; 
-	      (!num) && (user = OSDynamicCast(OSDictionary, consoleUsers->getObject(idx))); 
+	      (user = OSDynamicCast(OSDictionary, consoleUsers->getObject(idx))); 
 	      idx++)
 	{
-	    num = OSDynamicCast(OSNumber, user->getObject(gIOConsoleSessionScreenLockedTimeKey));
+	    loggedIn |= ((kOSBooleanTrue == user->getObject(gIOConsoleSessionOnConsoleKey))
+	      		&& (kOSBooleanTrue == user->getObject(gIOConsoleSessionLoginDoneKey)));
+	    if (!num)
+	    {
+   	        num = OSDynamicCast(OSNumber, user->getObject(gIOConsoleSessionScreenLockedTimeKey));
+	    }
 	}
         gIOConsoleLockTime = num ? num->unsigned32BitValue() : 0;
     }
 
-    if (gIOConsoleLockTime)
+    if (!loggedIn 
+     || (kIOMessageSystemWillSleep == sSystemPower)
+     || (kIOMessageSystemPagingOff == sSystemPower))
     {
-	if (kIOMessageSystemWillSleep == sSystemPower)
-	    locked = kOSBooleanTrue;
+	locked = kOSBooleanTrue;
+    }
+    else if (gIOConsoleLockTime)
+    {
+	clock_sec_t  now;
+	clock_usec_t microsecs;
+
+	clock_get_calendar_microtime(&now, &microsecs);
+	if (gIOConsoleLockTime > now)
+	{
+	    AbsoluteTime deadline;
+	    clock_interval_to_deadline(gIOConsoleLockTime - now, kSecondScale, &deadline);
+	    thread_call_enter_delayed(gIOConsoleLockCallout, deadline);
+	}
 	else
 	{
-	    clock_sec_t  now;
-	    clock_usec_t microsecs;
-
-	    clock_get_calendar_microtime(&now, &microsecs);
-	    if (gIOConsoleLockTime > now)
-	    {
-		AbsoluteTime deadline;
-		clock_interval_to_deadline(gIOConsoleLockTime - now, kSecondScale, &deadline);
-		thread_call_enter_delayed(gIOConsoleLockCallout, deadline);
-	    }
-	    else
-	    {
-		locked = kOSBooleanTrue;
-	    }
+	    locked = kOSBooleanTrue;
 	}
     }