Loading...
iokit/Kernel/RootDomainUserClient.cpp xnu-3248.50.21 xnu-1699.26.8
--- xnu/xnu-3248.50.21/iokit/Kernel/RootDomainUserClient.cpp
+++ xnu/xnu-1699.26.8/iokit/Kernel/RootDomainUserClient.cpp
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 1998-2012 Apple Computer, Inc. All rights reserved.
+ * Copyright (c) 1998-2000 Apple Computer, Inc. All rights reserved.
  *
  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
  * 
@@ -36,8 +36,6 @@
 #include <IOKit/IOBufferMemoryDescriptor.h>
 #include "RootDomainUserClient.h"
 #include <IOKit/pwr_mgt/IOPMLibDefs.h>
-#include <IOKit/pwr_mgt/IOPMPrivate.h>
-#include <sys/proc.h>
 
 #define super IOUserClient
 
@@ -47,17 +45,17 @@
 
 /* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
 
-bool RootDomainUserClient::initWithTask(task_t owningTask, void *security_id,
-                    UInt32 type, OSDictionary * properties)
+bool RootDomainUserClient::initWithTask(task_t owningTask, void *security_id, 
+					UInt32 type, OSDictionary * properties)
 {
     if (properties)
-    properties->setObject(kIOUserClientCrossEndianCompatibleKey, kOSBooleanTrue);
+	properties->setObject(kIOUserClientCrossEndianCompatibleKey, kOSBooleanTrue);
 
     if (!super::initWithTask(owningTask, security_id, type, properties))
-    return false;
+	return false;
 
     fOwningTask = owningTask;
-    task_reference (fOwningTask);
+    task_reference (fOwningTask);    
     return true;
 }
 
@@ -78,9 +76,9 @@
     return secureSleepSystemOptions(NULL, 0, return_code);
 }
 
-IOReturn RootDomainUserClient::secureSleepSystemOptions(
-    const void      *inOptions,
-    IOByteCount     inOptionsSize,
+IOReturn RootDomainUserClient::secureSleepSystemOptions( 
+    const void      *inOptions, 
+    IOByteCount     inOptionsSize __unused,
     uint32_t        *returnCode)
 {
 
@@ -92,44 +90,39 @@
 
     ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeLocalUser);
     local_priv = (kIOReturnSuccess == ret);
-
+    
     ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeAdministrator);
     admin_priv = (kIOReturnSuccess == ret);
-
-
+    
+    
     if (inOptions)
     {
         unserializedOptions = OSDynamicCast( OSDictionary,
-                                             OSUnserializeXML((const char *)inOptions, inOptionsSize, &unserializeErrorString));
-
+                                             OSUnserializeXML((const char *)inOptions, &unserializeErrorString));
+    
         if (!unserializedOptions) {
-            IOLog("IOPMRootDomain SleepSystem unserialization failure: %s\n",
+            IOLog("IOPMRootDomain SleepSystem unserialization failure: %s\n", 
                 unserializeErrorString ? unserializeErrorString->getCStringNoCopy() : "Unknown");
         }
     }
 
-    if ( (local_priv || admin_priv) && fOwner )
+    if ( (local_priv || admin_priv) 
+          && fOwner ) 
     {
-        proc_t p;
-        p = (proc_t)get_bsdtask_info(fOwningTask);
-        if (p) {
-            fOwner->setProperty("SleepRequestedByPID", proc_pid(p), 32);
-        }
-
-        if (unserializedOptions)
+        if (unserializedOptions) 
         {
             // Publish Sleep Options in registry under root_domain
-            fOwner->setProperty( kRootDomainSleepOptionsKey, unserializedOptions);
+            fOwner->setProperty( kRootDomainSleepOptionsKey, unserializedOptions);            
 
             *returnCode = fOwner->sleepSystemOptions( unserializedOptions );
 
-            unserializedOptions->release();
+            unserializedOptions->release();        
         } else {
             // No options
             // Clear any pre-existing options
             fOwner->removeProperty( kRootDomainSleepOptionsKey );
 
-            *returnCode = fOwner->sleepSystemOptions( NULL );
+            *returnCode = fOwner->sleepSystemOptions( NULL );        
         }
 
     } else {
@@ -139,7 +132,7 @@
     return kIOReturnSuccess;
 }
 
-IOReturn RootDomainUserClient::secureSetAggressiveness(
+IOReturn RootDomainUserClient::secureSetAggressiveness( 
     unsigned long   type,
     unsigned long   newLevel,
     int             *return_code )
@@ -150,7 +143,7 @@
 
     ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeLocalUser);
     local_priv = (kIOReturnSuccess == ret);
-
+    
     ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeAdministrator);
     admin_priv = (kIOReturnSuccess == ret);
 
@@ -168,7 +161,7 @@
 {
     int                     admin_priv = 0;
     IOReturn                ret = kIOReturnNotPrivileged;
-
+    
     ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeAdministrator);
     admin_priv = (kIOReturnSuccess == ret);
 
@@ -185,7 +178,7 @@
 {
     int                     admin_priv = 0;
     IOReturn                ret = kIOReturnNotPrivileged;
-
+    
     ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeAdministrator);
     admin_priv = (kIOReturnSuccess == ret);
 
@@ -197,44 +190,50 @@
     return kIOReturnSuccess;
 }
 
-IOReturn RootDomainUserClient::secureGetSystemSleepType(
-    uint32_t    *outSleepType)
-{
-    int                     admin_priv = 0;
-    IOReturn                ret;
-
-    ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeAdministrator);
-    admin_priv = (kIOReturnSuccess == ret);
-
-    if (admin_priv && fOwner) {
-        ret = fOwner->getSystemSleepType(outSleepType);
-    } else {
-        ret = kIOReturnNotPrivileged;
-    }
-    return ret;
-}
-
 IOReturn RootDomainUserClient::clientClose( void )
 {
     detach(fOwner);
-
+    
     if(fOwningTask) {
         task_deallocate(fOwningTask);
         fOwningTask = 0;
-    }
-
-    return kIOReturnSuccess;
+    }   
+    
+    return kIOReturnSuccess;
+}
+
+IOReturn RootDomainUserClient::clientMemoryForType(
+    UInt32 type,
+    IOOptionBits *options,
+    IOMemoryDescriptor ** memory)
+{
+    if (!fOwner)
+        return kIOReturnNotReady;
+
+    if (kPMRootDomainMapTraceBuffer == type)
+    {
+        *memory = fOwner->getPMTraceMemoryDescriptor();
+        if (*memory) {
+            (*memory)->retain();
+            *options = 0;
+            return kIOReturnSuccess;
+        } else {
+            return kIOReturnNotFound;
+        }
+
+    }
+    return kIOReturnUnsupported;
 }
 
 IOReturn RootDomainUserClient::externalMethod(
-    uint32_t selector,
+    uint32_t selector, 
     IOExternalMethodArguments * arguments,
-    IOExternalMethodDispatch * dispatch __unused,
+    IOExternalMethodDispatch * dispatch __unused, 
     OSObject * target __unused,
     void * reference __unused )
 {
-    IOReturn    ret = kIOReturnBadArgument;
-
+    IOReturn    ret = kIOReturnBadArgument;    
+    
     switch (selector)
     {
         case kPMSetAggressiveness:
@@ -247,7 +246,7 @@
                                 (int *)&arguments->scalarOutput[0]);
             }
             break;
-
+        
         case kPMGetAggressiveness:
             if ((1 == arguments->scalarInputCount)
                 && (1 == arguments->scalarOutputCount))
@@ -257,12 +256,12 @@
                                 (unsigned long *)&arguments->scalarOutput[0]);
             }
             break;
-
+        
         case kPMSleepSystem:
             if (1 == arguments->scalarOutputCount)
             {
                 ret = this->secureSleepSystem(
-                                (uint32_t *)&arguments->scalarOutput[0]);
+                                (uint32_t *)&arguments->scalarOutput[0]);            
             }
             break;
 
@@ -300,62 +299,70 @@
             break;
         case kPMSetMaintenanceWakeCalendar:
             ret = this->secureSetMaintenanceWakeCalendar(
-                    (IOPMCalendarStruct *)arguments->structureInput,
+                    (IOPMCalendarStruct *)arguments->structureInput, 
                     (uint32_t *)&arguments->structureOutput);
             arguments->structureOutputSize = sizeof(uint32_t);
             break;
-
+            
         case kPMSetUserAssertionLevels:
             ret = this->secureSetUserAssertionLevels(
                         (uint32_t)arguments->scalarInput[0]);
             break;
-
+            
         case kPMActivityTickle:
-            if ( fOwner->checkSystemCanSustainFullWake() )
-            {
-               fOwner->reportUserInput( );
-               fOwner->setProperty(kIOPMRootDomainWakeTypeKey, "UserActivity Assertion");
-            }
+            fOwner->reportUserInput( );
             ret = kIOReturnSuccess;
             break;
-
-        case kPMSetClamshellSleepState:
-            fOwner->setDisableClamShellSleep(arguments->scalarInput[0] ? true : false);
-            ret = kIOReturnSuccess;
-            break;
-
-        case kPMGetSystemSleepType:
-            if (1 == arguments->scalarOutputCount)
-            {
-                ret = this->secureGetSystemSleepType(
-                        (uint32_t *) &arguments->scalarOutput[0]);
-            }
-            break;
-
-#if defined(__i386__) || defined(__x86_64__)
-        case kPMSleepWakeWatchdogEnable:
-            ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeAdministrator);
-            if (ret == kIOReturnSuccess)
-               fOwner->sleepWakeDebugEnableWdog();
-            break;
-
-
-        case kPMSleepWakeDebugTrig:
-            ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeAdministrator);
-            if (ret == kIOReturnSuccess)
-               fOwner->sleepWakeDebugTrig(false);
-            break;
-#endif
-
-        case kPMSetDisplayPowerOn:
-            if (1 == arguments->scalarInputCount)
-            {
-                ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeAdministrator);
-                if (ret == kIOReturnSuccess)
-                    fOwner->setDisplayPowerOn((uint32_t)arguments->scalarInput[0]);
-            }
-            break;
-
+            
+/*
+        case kPMMethodCopySystemTimeline:
+            // intentional fallthrough
+        case kPMMethodCopyDetailedTimeline:
+
+            if (!arguments->structureOutputDescriptor)
+            {
+                // TODO: Force IOKit.framework to always send this data out
+                // of line; so I don't have to create a MemoryDescriptor here.
+                mem_size = arguments->structureOutputSize;
+                mem = IOMemoryDescriptor::withAddressRange(
+                                    (mach_vm_address_t)arguments->structureOutput, 
+                                    (mach_vm_size_t)mem_size,
+                                    kIODirectionIn, current_task());
+            } else {
+                mem_size = arguments->structureOutputDescriptorSize;
+                if (( mem = arguments->structureOutputDescriptor ))
+                    mem->retain();   
+            }
+            
+            if (mem)
+            {
+                mem->prepare(kIODirectionNone);
+    
+                if (kPMMethodCopySystemTimeline == selector) {
+                    arguments->scalarOutput[0] = fOwner->copySystemTimeline(
+                                    mem, &mem_size);
+                } 
+                else
+                if (kPMMethodCopyDetailedTimeline == selector) {
+                    arguments->scalarOutput[0] = fOwner->copyDetailedTimeline(
+                                    mem, &mem_size);
+                }
+            
+                if (arguments->structureOutputDescriptor) {
+                    arguments->structureOutputDescriptorSize = mem_size;
+                } else {
+                    arguments->structureOutputSize = mem_size;
+                }
+            
+                mem->release();
+
+                ret = kIOReturnSuccess;
+            } else {
+                ret = kIOReturnCannotWire;
+            }
+            
+            break;
+*/
         default:
             // bad selector
             return kIOReturnBadArgument;
@@ -369,7 +376,7 @@
  * We maintain getTargetAndExternalMethod since it's an exported symbol,
  * and only for that reason.
  */
-IOExternalMethod * RootDomainUserClient::getTargetAndMethodForIndex(
+IOExternalMethod * RootDomainUserClient::getTargetAndMethodForIndex( 
     IOService ** targetP, UInt32 index )
 {
     // DO NOT EDIT
@@ -379,6 +386,6 @@
 /* setPreventative
  * Does nothing. Exists only for exported symbol compatibility.
  */
-void
+void 
 RootDomainUserClient::setPreventative(UInt32 on_off, UInt32 types_of_sleep)
 { return; } // DO NOT EDIT