Loading...
--- xnu/xnu-1228/iokit/Kernel/RootDomainUserClient.cpp
+++ xnu/xnu-792.21.3/iokit/Kernel/RootDomainUserClient.cpp
@@ -32,7 +32,6 @@
#include <IOKit/assert.h>
#include <IOKit/IOLib.h>
-#include <IOKit/IOKitKeys.h>
#include <IOKit/IOBufferMemoryDescriptor.h>
#include "RootDomainUserClient.h"
#include <IOKit/pwr_mgt/IOPMLibDefs.h>
@@ -45,13 +44,9 @@
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-bool RootDomainUserClient::initWithTask(task_t owningTask, void *security_id,
- UInt32 type, OSDictionary * properties)
+bool RootDomainUserClient::initWithTask(task_t owningTask, void *security_id, UInt32)
{
- if (properties)
- properties->setObject(kIOUserClientCrossEndianCompatibleKey, kOSBooleanTrue);
-
- if (!super::initWithTask(owningTask, security_id, type, properties))
+ if (!super::init())
return false;
fOwningTask = owningTask;
@@ -71,75 +66,26 @@
return true;
}
-IOReturn RootDomainUserClient::secureSleepSystem( uint32_t *return_code )
+IOReturn RootDomainUserClient::secureSleepSystem( int *return_code )
{
- IOByteCount return_code_size = 1;
-
- return secureSleepSystemOptions( NULL, // inOptions
- (void *)return_code, // returnCode
- (void *)0, // inSize
- (void *)&return_code_size, // returnSize
- NULL, NULL);
-}
-
-IOReturn RootDomainUserClient::secureSleepSystemOptions(
- void * p1, void * p2, void * p3,
- void * p4, void * p5, void * p6 )
-{
- void *inOptions = (void *)p1;
- uint32_t *returnCode = (uint32_t *)p2;
- IOByteCount inOptionsSize = (IOByteCount)p3;
- IOByteCount *returnCodeSize = (IOByteCount *)p4;
-
int local_priv = 0;
int admin_priv = 0;
IOReturn ret = kIOReturnNotPrivileged;
- OSDictionary *unserializedOptions = NULL;
- OSString *unserializeErrorString = NULL;
ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeLocalUser);
local_priv = (kIOReturnSuccess == ret);
ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeAdministrator);
admin_priv = (kIOReturnSuccess == ret);
-
- *returnCodeSize = sizeof(uint32_t);
-
- if (inOptions)
- {
- unserializedOptions = OSDynamicCast( OSDictionary,
- OSUnserializeXML((const char *)inOptions, &unserializeErrorString));
-
- if (!unserializedOptions) {
- IOLog("IOPMRootDomain SleepSystem unserialization failure: %s\n",
- unserializeErrorString ? unserializeErrorString->getCStringNoCopy() : "Unknown");
- }
+
+ if((local_priv || admin_priv) && fOwner) {
+ *return_code = fOwner->sleepSystem();
+ return kIOReturnSuccess;
+ } else {
+ *return_code = kIOReturnNotPrivileged;
+ return kIOReturnSuccess;
}
- if ( (local_priv || admin_priv)
- && fOwner )
- {
- if (unserializedOptions)
- {
- // Publish Sleep Options in registry under root_domain
- fOwner->setProperty( kRootDomainSleepOptionsKey, unserializedOptions);
-
- *returnCode = fOwner->sleepSystemOptions( unserializedOptions );
-
- unserializedOptions->release();
- } else {
- // No options
- // Clear any pre-existing options
- fOwner->removeProperty( kRootDomainSleepOptionsKey );
-
- *returnCode = fOwner->sleepSystemOptions( NULL );
- }
-
- } else {
- *returnCode = kIOReturnNotPrivileged;
- }
-
- return kIOReturnSuccess;
}
IOReturn RootDomainUserClient::secureSetAggressiveness(
@@ -183,59 +129,15 @@
IOExternalMethod *
RootDomainUserClient::getTargetAndMethodForIndex( IOService ** targetP, UInt32 index )
{
- static const IOExternalMethod sMethods[] = {
+ static IOExternalMethod sMethods[] = {
{ // kPMSetAggressiveness, 0
- (IOService *)1, (IOMethod)&RootDomainUserClient::secureSetAggressiveness, kIOUCScalarIScalarO, 2, 1
+ 1, (IOMethod)&RootDomainUserClient::secureSetAggressiveness, kIOUCScalarIScalarO, 2, 1
},
{ // kPMGetAggressiveness, 1
0, (IOMethod)&IOPMrootDomain::getAggressiveness, kIOUCScalarIScalarO, 1, 1
},
{ // kPMSleepSystem, 2
- (IOService *)1, (IOMethod)&RootDomainUserClient::secureSleepSystem, kIOUCScalarIScalarO, 0, 1
- },
- { // kPMAllowPowerChange, 3
- 0, (IOMethod)&IOPMrootDomain::allowPowerChange, kIOUCScalarIScalarO, 1, 0
- },
- { // kPMCancelPowerChange, 4
- 0, (IOMethod)&IOPMrootDomain::cancelPowerChange, kIOUCScalarIScalarO, 1, 0
- },
- { // kPMShutdownSystem, 5
- 0, (IOMethod)&IOPMrootDomain::shutdownSystem, kIOUCScalarIScalarO, 0, 0
- },
- { // kPMRestartSystem, 6
- 0, (IOMethod)&IOPMrootDomain::restartSystem, kIOUCScalarIScalarO, 0, 0
- },
- { // kPMSleepSystemOptions, 7
- (IOService *)1, (IOMethod)&RootDomainUserClient::secureSleepSystemOptions,
- kIOUCStructIStructO, kIOUCVariableStructureSize, sizeof(uint32_t)
- }
- };
-
- if(index >= kNumPMMethods)
- return NULL;
- else {
- if (sMethods[index].object)
- *targetP = this;
- else
- *targetP = fOwner;
-
- return (IOExternalMethod *)&sMethods[index];
- }
-}
-
-#if 0
-IOReturn RootDomainUserClient::externalMethod( uint32_t selector, IOExternalMethodArguments * args,
- IOExternalMethodDispatch * dispatch, OSObject * target, void * reference )
-{
- static const IOExternalMethodDispatch sMethods[] = {
- { // kPMSetAggressiveness, 0
- (IOService *)1, (IOMethod)&RootDomainUserClient::secureSetAggressiveness, kIOUCScalarIScalarO, 2, 1
- },
- { // kPMGetAggressiveness, 1
- 0, (IOMethod)&IOPMrootDomain::getAggressiveness, kIOUCScalarIScalarO, 1, 1
- },
- { // kPMSleepSystem, 2
- (IOService *)1, (IOMethod)&RootDomainUserClient::secureSleepSystem, kIOUCScalarIScalarO, 0, 1
+ 1, (IOMethod)&RootDomainUserClient::secureSleepSystem, kIOUCScalarIScalarO, 0, 1
},
{ // kPMAllowPowerChange, 3
0, (IOMethod)&IOPMrootDomain::allowPowerChange, kIOUCScalarIScalarO, 1, 0
@@ -250,21 +152,21 @@
0, (IOMethod)&IOPMrootDomain::restartSystem, kIOUCScalarIScalarO, 0, 0
},
{ // kPMSetPreventative, 7
- (IOService *)1, (IOMethod)&RootDomainUserClient::setPreventative, kIOUCScalarIScalarO, 2, 0
+ 1, (IOMethod) &RootDomainUserClient::setPreventative, kIOUCScalarIScalarO, 2, 0
},
};
+
+ if(index >= kNumPMMethods)
+ return NULL;
+ else {
+ if (sMethods[index].object)
+ *targetP = this;
+ else
+ *targetP = fOwner;
- if (selector > (sizeof(sMethods) / sizeof(sMethods[0])))
- return (kIOReturnBadArgument);
-
- if ((1 << selector) & ((1 << 0) | (1 << 7))
- target = this;
- else
- target = fOwner;
-
- return (super::externalMethod(selector, args, &sMethods[selector], target, 0));
+ return &sMethods[index];
+ }
}
-#endif
void
RootDomainUserClient::setPreventative(UInt32 on_off, UInt32 types_of_sleep)