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