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 | |
35 | static 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 | |
47 | OSDefineMetaClassAndAbstractStructors(IOWatchDogTimer, IOService); |
48 | |
49 | OSMetaClassDefineReservedUnused(IOWatchDogTimer, 0); |
50 | OSMetaClassDefineReservedUnused(IOWatchDogTimer, 1); |
51 | OSMetaClassDefineReservedUnused(IOWatchDogTimer, 2); |
52 | OSMetaClassDefineReservedUnused(IOWatchDogTimer, 3); |
53 | |
54 | bool |
55 | IOWatchDogTimer::start(IOService *provider) |
56 | { |
57 | if (!super::start(provider)) { |
58 | return false; |
59 | } |
60 | |
61 | notifier = registerSleepWakeInterest(IOWatchDogTimerSleepHandler, this); |
62 | if (notifier == NULL) { |
63 | return false; |
64 | } |
65 | |
66 | setProperty(kWatchDogEnabledProperty, anObject: kOSBooleanFalse); |
67 | setWatchDogTimer(0); |
68 | |
69 | registerService(); |
70 | |
71 | return true; |
72 | } |
73 | |
74 | void |
75 | IOWatchDogTimer::stop(IOService *provider) |
76 | { |
77 | setWatchDogTimer(0); |
78 | notifier->remove(); |
79 | } |
80 | |
81 | IOReturn |
82 | IOWatchDogTimer::setProperties(OSObject *properties) |
83 | { |
84 | OSNumber *theNumber; |
85 | UInt32 theValue; |
86 | IOReturn result; |
87 | |
88 | result = IOUserClient::clientHasPrivilege(securityToken: current_task(), |
89 | kIOClientPrivilegeAdministrator); |
90 | if (result != kIOReturnSuccess) { |
91 | return kIOReturnNotPrivileged; |
92 | } |
93 | |
94 | theNumber = OSDynamicCast(OSNumber, properties); |
95 | if (theNumber == NULL) { |
96 | return kIOReturnBadArgument; |
97 | } |
98 | |
99 | theValue = theNumber->unsigned32BitValue(); |
100 | if (theValue == 0) { |
101 | setProperty(kWatchDogEnabledProperty, anObject: kOSBooleanFalse); |
102 | } else { |
103 | setProperty(kWatchDogEnabledProperty, anObject: kOSBooleanTrue); |
104 | } |
105 | |
106 | setWatchDogTimer(theValue); |
107 | |
108 | return kIOReturnSuccess; |
109 | } |
110 | |
111 | static IOReturn |
112 | IOWatchDogTimerSleepHandler(void *target, void */*refCon*/, |
113 | UInt32 messageType, |
114 | IOService */*provider*/, |
115 | void *messageArgument, |
116 | vm_size_t /*argSize*/) |
117 | { |
118 | IOWatchDogTimer *watchDogTimer = (IOWatchDogTimer *)target; |
119 | sleepWakeNote *swNote = (sleepWakeNote *)messageArgument; |
120 | |
121 | if (messageType != kIOMessageSystemWillSleep) { |
122 | return kIOReturnUnsupported; |
123 | } |
124 | |
125 | watchDogTimer->setProperty(kWatchDogEnabledProperty, anObject: kOSBooleanFalse); |
126 | watchDogTimer->setWatchDogTimer(0); |
127 | |
128 | swNote->returnValue = 0; |
129 | acknowledgeSleepWakeNotification(swNote->powerRef); |
130 | |
131 | return kIOReturnSuccess; |
132 | } |
133 | |