Loading...
--- xnu/xnu-12377.101.15/iokit/Kernel/RootDomainUserClient.cpp
+++ xnu/xnu-1504.7.4/iokit/Kernel/RootDomainUserClient.cpp
@@ -1,8 +1,8 @@
/*
- * Copyright (c) 1998-2020 Apple Computer, Inc. All rights reserved.
+ * Copyright (c) 1998-2000 Apple Computer, Inc. All rights reserved.
*
* @APPLE_OSREFERENCE_LICENSE_HEADER_START@
- *
+ *
* This file contains Original Code and/or Modifications of Original Code
* as defined in and that are subject to the Apple Public Source License
* Version 2.0 (the 'License'). You may not use this file except in
@@ -11,10 +11,10 @@
* unlawful or unlicensed copies of an Apple operating system, or to
* circumvent, violate, or enable the circumvention or violation of, any
* terms of an Apple operating system software license agreement.
- *
+ *
* Please obtain a copy of the License at
* http://www.opensource.apple.com/apsl/ and read it before using this file.
- *
+ *
* The Original Code and all software distributed under the License are
* distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
* EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
@@ -22,7 +22,7 @@
* FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
* Please see the License for the specific language governing rights and
* limitations under the License.
- *
+ *
* @APPLE_OSREFERENCE_LICENSE_HEADER_END@
*/
/*
@@ -34,672 +34,271 @@
#include <IOKit/IOLib.h>
#include <IOKit/IOKitKeys.h>
#include <IOKit/IOBufferMemoryDescriptor.h>
-#include <IOKit/IOSubMemoryDescriptor.h>
-#include <AssertMacros.h>
#include "RootDomainUserClient.h"
#include <IOKit/pwr_mgt/IOPMLibDefs.h>
-#include <IOKit/pwr_mgt/IOPMPrivate.h>
-#include <sys/proc.h>
-
-#define super IOUserClient2022
+
+#define super IOUserClient
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-OSDefineMetaClassAndStructors(RootDomainUserClient, IOUserClient2022)
+OSDefineMetaClassAndStructors(RootDomainUserClient, IOUserClient)
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-bool
-RootDomainUserClient::initWithTask(task_t owningTask, void *security_id,
- UInt32 type, OSDictionary * properties)
-{
- if (properties) {
- properties->setObject(kIOUserClientCrossEndianCompatibleKey, kOSBooleanTrue);
- }
-
- if (!super::initWithTask(owningTask, security_id, type, properties)) {
- return false;
- }
-
- fOwningTask = owningTask;
- task_reference(fOwningTask);
- return true;
-}
-
-
-bool
-RootDomainUserClient::start( IOService * provider )
-{
- assert(OSDynamicCast(IOPMrootDomain, provider));
- if (!super::start(provider)) {
- return false;
- }
- fOwner = (IOPMrootDomain *)provider;
-
- setProperty(kIOUserClientDefaultLockingKey, kOSBooleanTrue);
- setProperty(kIOUserClientDefaultLockingSetPropertiesKey, kOSBooleanTrue);
- setProperty(kIOUserClientDefaultLockingSingleThreadExternalMethodKey, kOSBooleanFalse);
-
- setProperty(kIOUserClientEntitlementsKey, kOSBooleanFalse);
-
- return true;
-}
-
-IOReturn
-RootDomainUserClient::secureSleepSystem( uint32_t *return_code )
-{
- return secureSleepSystemOptions(NULL, 0, return_code);
-}
-
-IOReturn
-RootDomainUserClient::secureSleepSystemOptions(
- const void *inOptions,
- IOByteCount inOptionsSize,
- uint32_t *returnCode)
-{
- int local_priv = 0;
- int admin_priv = 0;
- IOReturn ret = kIOReturnNotPrivileged;
-
- ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeLocalUser);
- local_priv = (kIOReturnSuccess == ret);
-
- ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeAdministrator);
- admin_priv = (kIOReturnSuccess == ret);
-
- if ((local_priv || admin_priv) && fOwner) {
- OSString *unserializeErrorString = NULL;
- OSObject *unserializedObject = NULL;
- OSDictionary *sleepOptionsDict = NULL; // do not release
-
- proc_t p;
- p = (proc_t)get_bsdtask_info(fOwningTask);
- if (p) {
- fOwner->setProperty("SleepRequestedByPID", proc_pid(p), 32);
- }
-
- if (inOptions) {
- unserializedObject = OSUnserializeXML((const char *)inOptions, inOptionsSize, &unserializeErrorString);
- sleepOptionsDict = OSDynamicCast( OSDictionary, unserializedObject);
- if (!sleepOptionsDict) {
- IOLog("IOPMRootDomain SleepSystem unserialization failure: %s\n",
- unserializeErrorString ? unserializeErrorString->getCStringNoCopy() : "Unknown");
- }
- }
-
- if (sleepOptionsDict) {
- // Publish Sleep Options in registry under root_domain
- fOwner->setProperty( kRootDomainSleepOptionsKey, sleepOptionsDict);
- } else {
- // No options
- // Clear any pre-existing options
- fOwner->removeProperty( kRootDomainSleepOptionsKey );
- }
-
- *returnCode = fOwner->sleepSystemOptions( sleepOptionsDict );
- OSSafeReleaseNULL(unserializedObject);
- OSSafeReleaseNULL(unserializeErrorString);
- } else {
- *returnCode = kIOReturnNotPrivileged;
- }
-
- return kIOReturnSuccess;
-}
-
-IOReturn
-RootDomainUserClient::secureSetAggressiveness(
- unsigned long type,
- unsigned long newLevel,
- int *return_code )
-{
- int local_priv = 0;
- int admin_priv = 0;
- IOReturn ret = kIOReturnNotPrivileged;
-
- ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeLocalUser);
- local_priv = (kIOReturnSuccess == ret);
-
- ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeAdministrator);
- admin_priv = (kIOReturnSuccess == ret);
-
- if ((local_priv || admin_priv) && fOwner) {
- *return_code = fOwner->setAggressiveness(type, newLevel);
- } else {
- *return_code = kIOReturnNotPrivileged;
- }
- return kIOReturnSuccess;
-}
-
-IOReturn
-RootDomainUserClient::secureSetMaintenanceWakeCalendar(
- IOPMCalendarStruct *inCalendar,
- uint32_t *returnCode)
-{
- int admin_priv = 0;
- IOReturn ret = kIOReturnNotPrivileged;
-
- ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeAdministrator);
- admin_priv = (kIOReturnSuccess == ret);
-
- if (admin_priv && fOwner) {
- *returnCode = fOwner->setMaintenanceWakeCalendar(inCalendar);
- } else {
- *returnCode = kIOReturnNotPrivileged;
- }
- return kIOReturnSuccess;
-}
-
-IOReturn
-RootDomainUserClient::secureSetUserAssertionLevels(
- uint32_t assertionBitfield)
-{
- int admin_priv = 0;
- IOReturn ret = kIOReturnNotPrivileged;
-
- ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeAdministrator);
- admin_priv = (kIOReturnSuccess == ret);
-
- if (admin_priv && fOwner) {
- ret = fOwner->setPMAssertionUserLevels(assertionBitfield);
- } else {
- ret = kIOReturnNotPrivileged;
- }
- return kIOReturnSuccess;
-}
-
-IOReturn
-RootDomainUserClient::secureGetSystemSleepType(
- uint32_t *outSleepType, uint32_t *sleepTimer)
-{
- int admin_priv = 0;
- IOReturn ret;
-
- ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeAdministrator);
- admin_priv = (kIOReturnSuccess == ret);
-
- if (admin_priv && fOwner) {
- ret = fOwner->getSystemSleepType(outSleepType, sleepTimer);
- } else {
- ret = kIOReturnNotPrivileged;
- }
- return ret;
-}
-
-IOReturn
-RootDomainUserClient::secureAttemptIdleSleepAbort(
- uint32_t *outReverted)
-{
- int admin_priv = 0;
- IOReturn ret;
-
- ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeAdministrator);
- admin_priv = (kIOReturnSuccess == ret);
-
- if (admin_priv && fOwner) {
- *outReverted = (uint32_t) fOwner->attemptIdleSleepAbort();
- } else {
- ret = kIOReturnNotPrivileged;
- }
- return ret;
-}
-
-IOReturn
-RootDomainUserClient::secureSetLockdownModeHibernation(
- uint32_t status)
-{
-#if HIBERNATION
- int admin_priv = 0;
- IOReturn ret;
-
- ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeAdministrator);
- admin_priv = (kIOReturnSuccess == ret);
-
- if (admin_priv && fOwner) {
- fOwner->setLockdownModeHibernation(status);
- } else {
- ret = kIOReturnNotPrivileged;
- }
- return kIOReturnSuccess;
+bool RootDomainUserClient::initWithTask(task_t owningTask, void *security_id,
+ UInt32 type, OSDictionary * properties)
+{
+ if (properties)
+ properties->setObject(kIOUserClientCrossEndianCompatibleKey, kOSBooleanTrue);
+
+ if (!super::initWithTask(owningTask, security_id, type, properties))
+ return false;
+
+ fOwningTask = owningTask;
+ task_reference (fOwningTask);
+ return true;
+}
+
+
+bool RootDomainUserClient::start( IOService * provider )
+{
+ assert(OSDynamicCast(IOPMrootDomain, provider));
+ if(!super::start(provider))
+ return false;
+ fOwner = (IOPMrootDomain *)provider;
+
+
+ return true;
+}
+
+IOReturn RootDomainUserClient::secureSleepSystem( uint32_t *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 = (uintptr_t)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 )
+ {
+ 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(
+ unsigned long type,
+ unsigned long newLevel,
+ int *return_code )
+{
+ int local_priv = 0;
+ int admin_priv = 0;
+ IOReturn ret = kIOReturnNotPrivileged;
+
+ ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeLocalUser);
+ local_priv = (kIOReturnSuccess == ret);
+
+ ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeAdministrator);
+ admin_priv = (kIOReturnSuccess == ret);
+
+ if((local_priv || admin_priv) && fOwner) {
+ *return_code = fOwner->setAggressiveness(type, newLevel);
+ return kIOReturnSuccess;
+ } else {
+ *return_code = kIOReturnNotPrivileged;
+ return kIOReturnSuccess;
+ }
+}
+
+IOReturn RootDomainUserClient::secureSetMaintenanceWakeCalendar(
+ void * p1, void * p2, void * p3,
+ void * p4, void * p5, void * p6 )
+{
+#if ROOT_DOMAIN_RUN_STATES
+ IOPMCalendarStruct * inCalendar = (IOPMCalendarStruct *) p1;
+ uint32_t * returnCode = (uint32_t *) p2;
+ IOByteCount * returnCodeSize = (IOByteCount *) p4;
+ int admin_priv = 0;
+ IOReturn ret = kIOReturnNotPrivileged;
+
+ ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeAdministrator);
+ admin_priv = (kIOReturnSuccess == ret);
+
+ *returnCodeSize = sizeof(uint32_t);
+
+ if (admin_priv && fOwner) {
+ *returnCode = fOwner->setMaintenanceWakeCalendar(inCalendar);
+ return kIOReturnSuccess;
+ } else {
+ *returnCode = kIOReturnNotPrivileged;
+ return kIOReturnSuccess;
+ }
#else
- return kIOReturnError;
+ return kIOReturnUnsupported;
#endif
}
-IOReturn
-RootDomainUserClient::secureGetAssertionLog(
- IOPMAssertionLogData *outLog)
-{
- if (!fOwner) {
- return kIOReturnError;
- }
-
- return fOwner->getAssertionLog(outLog);
-}
-
-IOReturn
-RootDomainUserClient::secureSetAssertionLogNotificationThreshold(
- uint64_t threshold)
-{
- if (!fOwner) {
- return kIOReturnError;
- }
-
- return fOwner->setAssertionLogNotificationThreshold(threshold);
-}
-
-IOReturn
-RootDomainUserClient::secureSetAssertionLogNotificationPort(
- mach_port_t port)
-{
- if (!fOwner) {
- return kIOReturnError;
- }
-
- IOReturn ret = fOwner->setAssertionLogNotificationPort(port);
-
- if (ret == kIOReturnSuccess) {
- fAssertionLogNotificationPortRegistered = (port != MACH_PORT_NULL);
- }
-
- return ret;
-}
-
-IOReturn
-RootDomainUserClient::clientClose( void )
-{
- terminate();
-
- if (fAssertionLogNotificationPortRegistered) {
- secureSetAssertionLogNotificationPort(MACH_PORT_NULL);
- }
-
- return kIOReturnSuccess;
-}
-
-void
-RootDomainUserClient::stop( IOService *provider)
-{
- if (fOwningTask) {
- task_deallocate(fOwningTask);
- fOwningTask = NULL;
- }
-
- super::stop(provider);
-}
-
-IOReturn
-RootDomainUserClient::registerNotificationPort(mach_port_t port, UInt32 type, UInt32 refCon)
-{
- switch ((IOPMUserClientNotificationType)type) {
- case IOPMUserClientNotificationType_AssertionLog:
- return secureSetAssertionLogNotificationPort(port);
- }
-
- return kIOReturnSuccess;
-}
-
-IOReturn
-RootDomainUserClient::externalMethod(uint32_t selector, IOExternalMethodArgumentsOpaque * args )
-{
- static const IOExternalMethodDispatch2022 dispatchArray[] = {
- [kPMSetAggressiveness] = {
- .function = &RootDomainUserClient::externalMethodDispatched,
- .checkScalarInputCount = 2,
- .checkStructureInputSize = 0,
- .checkScalarOutputCount = 1,
- .checkStructureOutputSize = 0,
- .allowAsync = false,
- .checkEntitlement = NULL,
- },
- [kPMGetAggressiveness] = {
- .function = &RootDomainUserClient::externalMethodDispatched,
- .checkScalarInputCount = 1,
- .checkStructureInputSize = 0,
- .checkScalarOutputCount = 1,
- .checkStructureOutputSize = 0,
- .allowAsync = false,
- .checkEntitlement = NULL,
- },
- [kPMSleepSystem] = {
- .function = &RootDomainUserClient::externalMethodDispatched,
- .checkScalarInputCount = 0,
- .checkStructureInputSize = 0,
- .checkScalarOutputCount = 1,
- .checkStructureOutputSize = 0,
- .allowAsync = false,
- .checkEntitlement = NULL,
- },
- [kPMAllowPowerChange] = {
- .function = &RootDomainUserClient::externalMethodDispatched,
- .checkScalarInputCount = 1,
- .checkStructureInputSize = 0,
- .checkScalarOutputCount = 0,
- .checkStructureOutputSize = 0,
- .allowAsync = false,
- .checkEntitlement = NULL,
- },
- [kPMCancelPowerChange] = {
- .function = &RootDomainUserClient::externalMethodDispatched,
- .checkScalarInputCount = 1,
- .checkStructureInputSize = 0,
- .checkScalarOutputCount = 0,
- .checkStructureOutputSize = 0,
- .allowAsync = false,
- .checkEntitlement = NULL,
- },
- [kPMShutdownSystem] = {
- .function = &RootDomainUserClient::externalMethodDispatched,
- .checkScalarInputCount = 0,
- .checkStructureInputSize = 0,
- .checkScalarOutputCount = 0,
- .checkStructureOutputSize = 0,
- .allowAsync = false,
- .checkEntitlement = NULL,
- },
- [kPMRestartSystem] = {
- .function = &RootDomainUserClient::externalMethodDispatched,
- .checkScalarInputCount = 0,
- .checkStructureInputSize = 0,
- .checkScalarOutputCount = 0,
- .checkStructureOutputSize = 0,
- .allowAsync = false,
- .checkEntitlement = NULL,
- },
- [kPMSleepSystemOptions] = {
- .function = &RootDomainUserClient::externalMethodDispatched,
- .checkScalarInputCount = 0,
- .checkStructureInputSize = kIOUCVariableStructureSize,
- .checkScalarOutputCount = 0,
- .checkStructureOutputSize = sizeof(uint32_t),
- .allowAsync = false,
- .checkEntitlement = NULL,
- },
- [kPMSetMaintenanceWakeCalendar] = {
- .function = &RootDomainUserClient::externalMethodDispatched,
- .checkScalarInputCount = 0,
- .checkStructureInputSize = sizeof(IOPMCalendarStruct),
- .checkScalarOutputCount = 0,
- .checkStructureOutputSize = sizeof(uint32_t),
- .allowAsync = false,
- .checkEntitlement = NULL,
- },
- [kPMSetUserAssertionLevels] = {
- .function = &RootDomainUserClient::externalMethodDispatched,
- .checkScalarInputCount = 1,
- .checkStructureInputSize = 0,
- .checkScalarOutputCount = 0,
- .checkStructureOutputSize = 0,
- .allowAsync = false,
- .checkEntitlement = NULL,
- },
- [kPMActivityTickle] = {
- .function = &RootDomainUserClient::externalMethodDispatched,
- .checkScalarInputCount = 0,
- .checkStructureInputSize = 0,
- .checkScalarOutputCount = 0,
- .checkStructureOutputSize = 0,
- .allowAsync = false,
- .checkEntitlement = NULL,
- },
- [kPMSetClamshellSleepState] = {
- .function = &RootDomainUserClient::externalMethodDispatched,
- .checkScalarInputCount = 1,
- .checkStructureInputSize = 0,
- .checkScalarOutputCount = 0,
- .checkStructureOutputSize = 0,
- .allowAsync = false,
- .checkEntitlement = NULL,
- },
- [kPMGetSystemSleepType] = {
- .function = &RootDomainUserClient::externalMethodDispatched,
- .checkScalarInputCount = 0,
- .checkStructureInputSize = 0,
- .checkScalarOutputCount = 2,
- .checkStructureOutputSize = 0,
- .allowAsync = false,
- .checkEntitlement = NULL,
- },
- [kPMSleepWakeWatchdogEnable] = {
- .function = &RootDomainUserClient::externalMethodDispatched,
- .checkScalarInputCount = 0,
- .checkStructureInputSize = 0,
- .checkScalarOutputCount = 0,
- .checkStructureOutputSize = 0,
- .allowAsync = false,
- .checkEntitlement = NULL,
- },
- [kPMSleepWakeDebugTrig] = {
- .function = &RootDomainUserClient::externalMethodDispatched,
- .checkScalarInputCount = 0,
- .checkStructureInputSize = 0,
- .checkScalarOutputCount = 0,
- .checkStructureOutputSize = 0,
- .allowAsync = false,
- .checkEntitlement = NULL,
- },
- [kPMSetDisplayPowerOn] = {
- .function = &RootDomainUserClient::externalMethodDispatched,
- .checkScalarInputCount = 1,
- .checkStructureInputSize = 0,
- .checkScalarOutputCount = 0,
- .checkStructureOutputSize = 0,
- .allowAsync = false,
- .checkEntitlement = NULL,
- },
- [kPMRequestIdleSleepRevert] = {
- .function = &RootDomainUserClient::externalMethodDispatched,
- .checkScalarInputCount = 0,
- .checkStructureInputSize = 0,
- .checkScalarOutputCount = 1,
- .checkStructureOutputSize = 0,
- .allowAsync = false,
- .checkEntitlement = NULL,
- },
- [kPMSetLDMHibernationDisable] = {
- .function = &RootDomainUserClient::externalMethodDispatched,
- .checkScalarInputCount = 1,
- .checkStructureInputSize = 0,
- .checkScalarOutputCount = 0,
- .checkStructureOutputSize = 0,
- .allowAsync = false,
- .checkEntitlement = NULL,
- },
- [kPMGetAssertionLog] = {
- .function = &RootDomainUserClient::externalMethodDispatched,
- .checkScalarInputCount = 0,
- .checkStructureInputSize = 0,
- .checkScalarOutputCount = 0,
- .checkStructureOutputSize = sizeof(IOPMAssertionLogData),
- .allowAsync = false,
- .checkEntitlement = NULL,
- },
- [kPMSetAssertionLogThreshold] = {
- .function = &RootDomainUserClient::externalMethodDispatched,
- .checkScalarInputCount = 1,
- .checkStructureInputSize = 0,
- .checkScalarOutputCount = 0,
- .checkStructureOutputSize = 0,
- .allowAsync = false,
- .checkEntitlement = NULL,
- },
- };
-
- return dispatchExternalMethod(selector, args, dispatchArray, sizeof(dispatchArray) / sizeof(dispatchArray[0]), this, NULL);
-}
-IOReturn
-RootDomainUserClient::externalMethodDispatched(OSObject * target, void * reference, IOExternalMethodArguments * arguments)
-{
- IOReturn ret = kIOReturnBadArgument;
- RootDomainUserClient *me = (typeof(me))target;
- IOMemoryDescriptor *outMD = nullptr;
- IOMemoryMap *outMap = nullptr;
-
- if (arguments->structureOutputSize > 0 || arguments->structureOutputDescriptorSize > 0) {
- if (arguments->structureOutputDescriptor) {
- outMD = arguments->structureOutputDescriptor;
- outMD->retain();
- } else {
- outMD = IOMemoryDescriptor::withAddressRange((mach_vm_address_t)arguments->structureOutput, arguments->structureOutputSize, kIODirectionIn, kernel_task);
- }
-
- require_action(outMD != nullptr, out, ret = kIOReturnError);
- outMap = outMD->map();
- require_action(outMap != nullptr, out, ret = kIOReturnError);
- }
-
- switch (arguments->selector) {
- case kPMSetAggressiveness:
- ret = me->secureSetAggressiveness(
- (unsigned long)arguments->scalarInput[0],
- (unsigned long)arguments->scalarInput[1],
- (int *)&arguments->scalarOutput[0]);
- break;
-
- case kPMGetAggressiveness:
- ret = me->fOwner->getAggressiveness(
- (unsigned long)arguments->scalarInput[0],
- (unsigned long *)&arguments->scalarOutput[0]);
- break;
-
- case kPMSleepSystem:
- ret = me->secureSleepSystem(
- (uint32_t *)&arguments->scalarOutput[0]);
- break;
-
- case kPMAllowPowerChange:
- ret = me->fOwner->allowPowerChange(
- arguments->scalarInput[0]);
- break;
-
- case kPMCancelPowerChange:
- ret = me->fOwner->cancelPowerChange(
- arguments->scalarInput[0]);
- break;
-
- case kPMShutdownSystem:
- // deprecated interface
- ret = kIOReturnUnsupported;
- break;
-
- case kPMRestartSystem:
- // deprecated interface
- ret = kIOReturnUnsupported;
- break;
-
- case kPMSleepSystemOptions:
- ret = me->secureSleepSystemOptions(
- arguments->structureInput,
- arguments->structureInputSize,
- (uint32_t *)&arguments->structureOutput);
- break;
- case kPMSetMaintenanceWakeCalendar:
- ret = me->secureSetMaintenanceWakeCalendar(
- (IOPMCalendarStruct *)arguments->structureInput,
- (uint32_t *)&arguments->structureOutput);
- arguments->structureOutputSize = sizeof(uint32_t);
- break;
-
- case kPMSetUserAssertionLevels:
- ret = me->secureSetUserAssertionLevels(
- (uint32_t)arguments->scalarInput[0]);
- break;
-
- case kPMActivityTickle:
- if (me->fOwner->checkSystemCanSustainFullWake()) {
- me->fOwner->reportUserInput();
- me->fOwner->setProperty(kIOPMRootDomainWakeTypeKey, "UserActivity Assertion");
- }
- ret = kIOReturnSuccess;
- break;
-
- case kPMSetClamshellSleepState:
- me->fOwner->setClamShellSleepDisable(arguments->scalarInput[0] ? true : false,
- IOPMrootDomain::kClamshellSleepDisablePowerd);
- ret = kIOReturnSuccess;
- break;
-
- case kPMGetSystemSleepType:
- ret = me->secureGetSystemSleepType(
- (uint32_t *) &arguments->scalarOutput[0],
- (uint32_t *) &arguments->scalarOutput[1]);
- break;
-
-#if defined(__i386__) || defined(__x86_64__)
- case kPMSleepWakeWatchdogEnable:
- ret = clientHasPrivilege(me->fOwningTask, kIOClientPrivilegeAdministrator);
- if (ret == kIOReturnSuccess) {
- me->fOwner->sleepWakeDebugEnableWdog();
- }
- break;
-
- case kPMSleepWakeDebugTrig:
- ret = clientHasPrivilege(me->fOwningTask, kIOClientPrivilegeAdministrator);
- if (ret == kIOReturnSuccess) {
- me->fOwner->sleepWakeDebugTrig(false);
- }
- break;
+IOReturn RootDomainUserClient::clientClose( void )
+{
+ detach(fOwner);
+
+ if(fOwningTask) {
+ task_deallocate(fOwningTask);
+ fOwningTask = 0;
+ }
+
+ return kIOReturnSuccess;
+}
+
+IOExternalMethod *
+RootDomainUserClient::getTargetAndMethodForIndex( IOService ** targetP, UInt32 index )
+{
+ static const IOExternalMethod 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
+ },
+ { // 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)
+ },
+ { // kPMSetMaintenanceWakeCalendar, 8
+ (IOService *)1, (IOMethod)&RootDomainUserClient::secureSetMaintenanceWakeCalendar,
+ kIOUCStructIStructO, sizeof(IOPMCalendarStruct), 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
+ },
+ { // 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
+ },
+ { // kPMSetPreventative, 7
+ (IOService *)1, (IOMethod)&RootDomainUserClient::setPreventative, kIOUCScalarIScalarO, 2, 0
+ },
+ };
+
+ 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));
+}
#endif
- case kPMSetDisplayPowerOn:
- ret = clientHasPrivilege(me->fOwningTask, kIOClientPrivilegeAdministrator);
- if (ret == kIOReturnSuccess) {
- me->fOwner->setDisplayPowerOn((uint32_t)arguments->scalarInput[0]);
- }
- break;
-
- case kPMRequestIdleSleepRevert:
- ret = me->secureAttemptIdleSleepAbort(
- (uint32_t *) &arguments->scalarOutput[0]);
- break;
-
- case kPMSetLDMHibernationDisable:
- ret = me->secureSetLockdownModeHibernation((uint32_t)arguments->scalarInput[0]);
- break;
-
- case kPMGetAssertionLog:
- require_action(outMap != nullptr, out, ret = kIOReturnBadArgument);
- ret = me->secureGetAssertionLog((IOPMAssertionLogData *)outMap->getAddress());
- break;
-
- case kPMSetAssertionLogThreshold:
- ret = me->secureSetAssertionLogNotificationThreshold(arguments->scalarInput[0]);
- break;
-
- default:
- // bad selector
- ret = kIOReturnBadArgument;
- break;
- }
-
-out:
- if (outMap) {
- outMap->unmap();
- }
-
- OSSafeReleaseNULL(outMap);
- OSSafeReleaseNULL(outMD);
-
- return ret;
-}
-
-/* getTargetAndMethodForIndex
- * Not used. We prefer to use externalMethod() for user client invocations.
- * We maintain getTargetAndExternalMethod since it's an exported symbol,
- * and only for that reason.
- */
-IOExternalMethod *
-RootDomainUserClient::getTargetAndMethodForIndex(
- IOService ** targetP, UInt32 index )
-{
- // DO NOT EDIT
- return super::getTargetAndMethodForIndex(targetP, index);
-}
-
-/* setPreventative
- * Does nothing. Exists only for exported symbol compatibility.
- */
-void
+void
RootDomainUserClient::setPreventative(UInt32 on_off, UInt32 types_of_sleep)
{
- return;
-} // DO NOT EDIT
+ return;
+}
+