1/* 2 * Copyright (c) 2000 Apple Computer, Inc. All rights reserved. 3 * 4 * @APPLE_OSREFERENCE_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. The rights granted to you under the License 10 * may not be used to create, or enable the creation or redistribution of, 11 * unlawful or unlicensed copies of an Apple operating system, or to 12 * circumvent, violate, or enable the circumvention or violation of, any 13 * terms of an Apple operating system software license agreement. 14 * 15 * Please obtain a copy of the License at 16 * http://www.opensource.apple.com/apsl/ and read it before using this file. 17 * 18 * The Original Code and all software distributed under the License are 19 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER 20 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES, 21 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, 22 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. 23 * Please see the License for the specific language governing rights and 24 * limitations under the License. 25 * 26 * @APPLE_OSREFERENCE_LICENSE_HEADER_END@ 27 */ 28 29#include <IOKit/IOUserClient.h> 30#include <IOKit/IOMessage.h> 31#include <IOKit/system_management/IOWatchDogTimer.h> 32#include <IOKit/pwr_mgt/RootDomain.h> 33 34 35static IOReturn IOWatchDogTimerSleepHandler(void *target, void *refCon, 36 UInt32 messageType, 37 IOService *provider, 38 void *messageArgument, 39 vm_size_t argSize); 40 41 42#define kWatchDogEnabledProperty "IOWatchDogEnabled" 43 44 45#define super IOService 46 47OSDefineMetaClassAndAbstractStructors(IOWatchDogTimer, IOService); 48 49OSMetaClassDefineReservedUnused(IOWatchDogTimer, 0); 50OSMetaClassDefineReservedUnused(IOWatchDogTimer, 1); 51OSMetaClassDefineReservedUnused(IOWatchDogTimer, 2); 52OSMetaClassDefineReservedUnused(IOWatchDogTimer, 3); 53 54bool IOWatchDogTimer::start(IOService *provider) 55{ 56 if (!super::start(provider)) return false; 57 58 notifier = registerSleepWakeInterest(IOWatchDogTimerSleepHandler, this); 59 if (notifier == 0) return false; 60 61 setProperty(kWatchDogEnabledProperty, kOSBooleanFalse); 62 setWatchDogTimer(0); 63 64 registerService(); 65 66 return true; 67} 68 69void IOWatchDogTimer::stop(IOService *provider) 70{ 71 setWatchDogTimer(0); 72 notifier->remove(); 73} 74 75IOReturn IOWatchDogTimer::setProperties(OSObject *properties) 76{ 77 OSNumber *theNumber; 78 UInt32 theValue; 79 IOReturn result; 80 81 result = IOUserClient::clientHasPrivilege(current_task(), 82 kIOClientPrivilegeAdministrator); 83 if (result != kIOReturnSuccess) return kIOReturnNotPrivileged; 84 85 theNumber = OSDynamicCast(OSNumber, properties); 86 if (theNumber == 0) return kIOReturnBadArgument; 87 88 theValue = theNumber->unsigned32BitValue(); 89 if (theValue == 0) { 90 setProperty(kWatchDogEnabledProperty, kOSBooleanFalse); 91 } else { 92 setProperty(kWatchDogEnabledProperty, kOSBooleanTrue); 93 } 94 95 setWatchDogTimer(theValue); 96 97 return kIOReturnSuccess; 98} 99 100static IOReturn IOWatchDogTimerSleepHandler(void *target, void */*refCon*/, 101 UInt32 messageType, 102 IOService */*provider*/, 103 void *messageArgument, 104 vm_size_t /*argSize*/) 105{ 106 IOWatchDogTimer *watchDogTimer = (IOWatchDogTimer *)target; 107 sleepWakeNote *swNote = (sleepWakeNote *)messageArgument; 108 109 if (messageType != kIOMessageSystemWillSleep) return kIOReturnUnsupported; 110 111 watchDogTimer->setProperty(kWatchDogEnabledProperty, kOSBooleanFalse); 112 watchDogTimer->setWatchDogTimer(0); 113 114 swNote->returnValue = 0; 115 acknowledgeSleepWakeNotification(swNote->powerRef); 116 117 return kIOReturnSuccess; 118} 119