iokit/Kernel/RootDomainUserClient.cpp

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