iokit/Kernel/RootDomainUserClient.cpp

   1 /*
   2  * Copyright (c) 1998-2000 Apple Computer, Inc. All rights reserved.
   3  *
   4  * @APPLE_LICENSE_HEADER_START@
   5  *
   6  * The contents of this file constitute Original Code as defined in and
   7  * are subject to the Apple Public Source License Version 1.1 (the
   8  * "License").  You may not use this file except in compliance with the
   9  * License.  Please obtain a copy of the License at
  10  * http://www.apple.com/publicsource and read it before using this file.
  11  *
  12  * This Original Code and all software distributed under the License are
  13  * distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  14  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  15  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  16  * FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT.  Please see the
  17  * License for the specific language governing rights and limitations
  18  * under the License.
  19  *
  20  * @APPLE_LICENSE_HEADER_END@
  21  */
  22 /*
  23  * Copyright (c) 1999 Apple Computer, Inc.  All rights reserved.
  24  *
  25  */
  26
  27 #include <IOKit/assert.h>
  28 #include <IOKit/IOLib.h>
  29 #include <IOKit/IOKitKeys.h>
  30 #include <IOKit/IOBufferMemoryDescriptor.h>
  31 #include "RootDomainUserClient.h"
  32 #include <IOKit/pwr_mgt/IOPMLibDefs.h>
  33
  34 #define super IOUserClient
  35
  36 /* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
  37
  38 OSDefineMetaClassAndStructors(RootDomainUserClient, IOUserClient)
  39
  40 /* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
  41
  42 bool RootDomainUserClient::initWithTask(task_t owningTask, void *security_id,
  43                                         UInt32 type, OSDictionary * properties)
  44 {
  45     if (properties)
  46         properties->setObject(kIOUserClientCrossEndianCompatibleKey, kOSBooleanTrue);
  47
  48     if (!super::initWithTask(owningTask, security_id, type, properties))
  49         return false;
  50
  51     fOwningTask = owningTask;
  52     task_reference (fOwningTask);
  53     return true;
  54 }
  55
  56
  57 bool RootDomainUserClient::start( IOService * provider )
  58 {
  59     assert(OSDynamicCast(IOPMrootDomain, provider));
  60     if(!super::start(provider))
  61         return false;
  62     fOwner = (IOPMrootDomain *)provider;
  63
  64
  65     return true;
  66 }
  67
  68 IOReturn RootDomainUserClient::secureSleepSystem( int *return_code )
  69 {
  70     int             local_priv = 0;
  71     int             admin_priv = 0;
  72     IOReturn        ret = kIOReturnNotPrivileged;
  73
  74     ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeLocalUser);
  75     local_priv = (kIOReturnSuccess == ret);
  76
  77     ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeAdministrator);
  78     admin_priv = (kIOReturnSuccess == ret);
  79
  80     if((local_priv || admin_priv) && fOwner) {
  81         *return_code = fOwner->sleepSystem();
  82         return kIOReturnSuccess;
  83     } else {
  84         *return_code = kIOReturnNotPrivileged;
  85         return kIOReturnSuccess;
  86     }
  87
  88 }
  89
  90 IOReturn RootDomainUserClient::secureSetAggressiveness(
  91     unsigned long   type,
  92     unsigned long   newLevel,
  93     int             *return_code )
  94 {
  95     int             local_priv = 0;
  96     int             admin_priv = 0;
  97     IOReturn        ret = kIOReturnNotPrivileged;
  98
  99     ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeLocalUser);
 100     local_priv = (kIOReturnSuccess == ret);
 101
 102     ret = clientHasPrivilege(fOwningTask, kIOClientPrivilegeAdministrator);
 103     admin_priv = (kIOReturnSuccess == ret);
 104
 105     if((local_priv || admin_priv) && fOwner) {
 106         *return_code = fOwner->setAggressiveness(type, newLevel);
 107         return kIOReturnSuccess;
 108     } else {
 109         *return_code = kIOReturnNotPrivileged;
 110         return kIOReturnSuccess;
 111     }
 112
 113 }
 114
 115
 116 IOReturn RootDomainUserClient::clientClose( void )
 117 {
 118     detach(fOwner);
 119
 120     if(fOwningTask) {
 121         task_deallocate(fOwningTask);
 122         fOwningTask = 0;
 123     }
 124
 125     return kIOReturnSuccess;
 126 }
 127
 128 IOExternalMethod *
 129 RootDomainUserClient::getTargetAndMethodForIndex( IOService ** targetP, UInt32 index )
 130 {
 131     static IOExternalMethod sMethods[] = {
 132         { // kPMSetAggressiveness, 0
 133             (IOService *)1, (IOMethod)&RootDomainUserClient::secureSetAggressiveness, kIOUCScalarIScalarO, 2, 1
 134         },
 135         { // kPMGetAggressiveness, 1
 136             0, (IOMethod)&IOPMrootDomain::getAggressiveness, kIOUCScalarIScalarO, 1, 1
 137         },
 138         { // kPMSleepSystem, 2
 139             (IOService *)1, (IOMethod)&RootDomainUserClient::secureSleepSystem, kIOUCScalarIScalarO, 0, 1
 140         },
 141         { // kPMAllowPowerChange, 3
 142             0, (IOMethod)&IOPMrootDomain::allowPowerChange, kIOUCScalarIScalarO, 1, 0
 143         },
 144         { // kPMCancelPowerChange, 4
 145             0, (IOMethod)&IOPMrootDomain::cancelPowerChange, kIOUCScalarIScalarO, 1, 0
 146         },
 147         { // kPMShutdownSystem, 5
 148             0, (IOMethod)&IOPMrootDomain::shutdownSystem, kIOUCScalarIScalarO, 0, 0
 149         },
 150         { // kPMRestartSystem, 6
 151             0, (IOMethod)&IOPMrootDomain::restartSystem, kIOUCScalarIScalarO, 0, 0
 152         },
 153         { // kPMSetPreventative, 7
 154             (IOService *)1, (IOMethod)&RootDomainUserClient::setPreventative, kIOUCScalarIScalarO, 2, 0
 155         },
 156     };
 157
 158     if(index >= kNumPMMethods)
 159         return NULL;
 160     else {
 161         if (sMethods[index].object)
 162             *targetP = this;
 163         else
 164             *targetP = fOwner;
 165
 166         return &sMethods[index];
 167     }
 168 }
 169
 170 void
 171 RootDomainUserClient::setPreventative(UInt32 on_off, UInt32 types_of_sleep)
 172 {
 173     return;
 174 }
 175