Loading...
iokit/Kernel/IOPMrootDomain.cpp xnu-12377.121.6 xnu-2050.9.2
--- xnu/xnu-12377.121.6/iokit/Kernel/IOPMrootDomain.cpp
+++ xnu/xnu-2050.9.2/iokit/Kernel/IOPMrootDomain.cpp
@@ -1,8 +1,8 @@
 /*
- * Copyright (c) 1998-2021 Apple Inc. All rights reserved.
+ * Copyright (c) 1998-2008 Apple Inc. All rights reserved.
  *
  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
- *
+ * 
  * This file contains Original Code and/or Modifications of Original Code
  * as defined in and that are subject to the Apple Public Source License
  * Version 2.0 (the 'License'). You may not use this file except in
@@ -11,10 +11,10 @@
  * unlawful or unlicensed copies of an Apple operating system, or to
  * circumvent, violate, or enable the circumvention or violation of, any
  * terms of an Apple operating system software license agreement.
- *
+ * 
  * Please obtain a copy of the License at
  * http://www.opensource.apple.com/apsl/ and read it before using this file.
- *
+ * 
  * The Original Code and all software distributed under the License are
  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
@@ -22,23 +22,16 @@
  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  * Please see the License for the specific language governing rights and
  * limitations under the License.
- *
+ * 
  * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
  */
-
-#define IOKIT_ENABLE_SHARED_PTR
-
-#include <libkern/c++/OSAllocation.h>
 #include <libkern/c++/OSKext.h>
 #include <libkern/c++/OSMetaClass.h>
 #include <libkern/OSAtomic.h>
 #include <libkern/OSDebug.h>
 #include <IOKit/IOWorkLoop.h>
 #include <IOKit/IOCommandGate.h>
-#include <IOKit/IOTimerEventSource.h>
 #include <IOKit/IOPlatformExpert.h>
-#include <IOKit/IOCPU.h>
-#include <IOKit/IOPlatformActions.h>
 #include <IOKit/IOKitDebug.h>
 #include <IOKit/IOTimeStamp.h>
 #include <IOKit/pwr_mgt/IOPMlog.h>
@@ -47,104 +40,55 @@
 #include <IOKit/IODeviceTreeSupport.h>
 #include <IOKit/IOMessage.h>
 #include <IOKit/IOReturn.h>
-#include <IOKit/IONVRAM.h>
 #include "RootDomainUserClient.h"
 #include "IOKit/pwr_mgt/IOPowerConnection.h"
 #include "IOPMPowerStateQueue.h"
 #include <IOKit/IOCatalogue.h>
-#include <IOKit/IOReportMacros.h>
-#include <IOKit/IOLib.h>
-#include <IOKit/IOKitKeysPrivate.h>
-#include <IOKit/IOUserServer.h>
-#include <IOKit/IOBSD.h>
-#include "IOKitKernelInternal.h"
 #if HIBERNATION
 #include <IOKit/IOHibernatePrivate.h>
-#endif /* HIBERNATION */
-#include <machine/machine_routines.h>
+#endif
 #include <console/video_console.h>
 #include <sys/syslog.h>
 #include <sys/sysctl.h>
-#include <sys/vnode.h>
-#include <sys/vnode_internal.h>
-#include <sys/fcntl.h>
-#include <os/log.h>
-#include <os/log_private.h>
-#include <pexpert/device_tree.h>
-#include <pexpert/protos.h>
-#include <AssertMacros.h>
-
 #include <sys/time.h>
-#include "IOServicePrivate.h"   // _IOServiceInterestNotifier
+#include "IOServicePrivate.h"	// _IOServiceInterestNotifier
 #include "IOServicePMPrivate.h"
-
-#include <libkern/zlib.h>
-#include <os/cpp_util.h>
-#include <os/atomic_private.h>
-#include <libkern/c++/OSBoundedArrayRef.h>
-
-#if DEVELOPMENT || DEBUG
-#include <os/system_event_log.h>
-#endif /* DEVELOPMENT || DEBUG */
 
 __BEGIN_DECLS
 #include <mach/shared_region.h>
-#include <kern/clock.h>
-#include <vm/vm_pageout_xnu.h>
 __END_DECLS
 
 #if defined(__i386__) || defined(__x86_64__)
 __BEGIN_DECLS
 #include "IOPMrootDomainInternal.h"
-const char *processor_to_datastring(const char *prefix, processor_t target_processor);
 __END_DECLS
 #endif
 
-#define ARRAY_LEN(x) (sizeof (x) / sizeof (x[0]))
-
 #define kIOPMrootDomainClass    "IOPMrootDomain"
 #define LOG_PREFIX              "PMRD: "
 
-
 #define MSG(x...) \
     do { kprintf(LOG_PREFIX x); IOLog(x); } while (false)
 
 #define LOG(x...)    \
     do { kprintf(LOG_PREFIX x); } while (false)
 
-#if DEVELOPMENT || DEBUG
-#define DEBUG_LOG(x...) do { \
-    if (kIOLogPMRootDomain & gIOKitDebug) \
-    kprintf(LOG_PREFIX x); \
-    os_log_debug(OS_LOG_DEFAULT, LOG_PREFIX x); \
-} while (false)
-#else
-#define DEBUG_LOG(x...)
-#endif
-
 #define DLOG(x...)  do { \
-    if (kIOLogPMRootDomain & gIOKitDebug) \
-	IOLog(LOG_PREFIX x); \
-    else \
-	os_log(OS_LOG_DEFAULT, LOG_PREFIX x); \
-} while (false)
-
-#define DMSG(x...)  do { \
-    if (kIOLogPMRootDomain & gIOKitDebug) { \
-	kprintf(LOG_PREFIX x); \
-    } \
-} while (false)
-
+	if (kIOLogPMRootDomain & gIOKitDebug) \
+        kprintf(LOG_PREFIX x); } while (false)
 
 #define _LOG(x...)
+
+#define DARK_WAKE_DEBUG                     1
+#define SUSPEND_PM_NOTIFICATIONS_DEBUG      1
 
 #define CHECK_THREAD_CONTEXT
 #ifdef  CHECK_THREAD_CONTEXT
-static IOWorkLoop * gIOPMWorkLoop = NULL;
+static IOWorkLoop * gIOPMWorkLoop = 0;
 #define ASSERT_GATED()                                      \
 do {                                                        \
     if (gIOPMWorkLoop && gIOPMWorkLoop->inGate() != true) { \
-	panic("RootDomain: not inside PM gate");            \
+        panic("RootDomain: not inside PM gate");            \
     }                                                       \
 } while(false)
 #else
@@ -152,259 +96,121 @@
 #endif /* CHECK_THREAD_CONTEXT */
 
 #define CAP_LOSS(c)  \
-	(((_pendingCapability & (c)) == 0) && \
-	 ((_currentCapability & (c)) != 0))
+        (((_pendingCapability & (c)) == 0) && \
+         ((_currentCapability & (c)) != 0))
 
 #define CAP_GAIN(c)  \
-	(((_currentCapability & (c)) == 0) && \
-	 ((_pendingCapability & (c)) != 0))
+        (((_currentCapability & (c)) == 0) && \
+         ((_pendingCapability & (c)) != 0))
 
 #define CAP_CHANGE(c)    \
-	(((_currentCapability ^ _pendingCapability) & (c)) != 0)
+        (((_currentCapability ^ _pendingCapability) & (c)) != 0)
 
 #define CAP_CURRENT(c)  \
-	((_currentCapability & (c)) != 0)
+        ((_currentCapability & (c)) != 0)
 
 #define CAP_HIGHEST(c)  \
-	((_highestCapability & (c)) != 0)
-
-#define CAP_PENDING(c)  \
-	((_pendingCapability & (c)) != 0)
-
-// rdar://problem/9157444
-#if defined(__i386__) || defined(__x86_64__)
-#define DARK_TO_FULL_EVALUATE_CLAMSHELL_DELAY   20
-#endif
+        ((_highestCapability & (c)) != 0)
+
+#define DARK_TO_FULL_EVALUATE_CLAMSHELL     0
 
 // Event types for IOPMPowerStateQueue::submitPowerEvent()
 enum {
-	kPowerEventFeatureChanged = 1,             // 1
-	kPowerEventReceivedPowerNotification,      // 2
-	kPowerEventSystemBootCompleted,            // 3
-	kPowerEventSystemShutdown,                 // 4
-	kPowerEventUserDisabledSleep,              // 5
-	kPowerEventRegisterSystemCapabilityClient, // 6
-	kPowerEventRegisterKernelCapabilityClient, // 7
-	kPowerEventPolicyStimulus,                 // 8
-	kPowerEventAssertionCreate,                // 9
-	kPowerEventAssertionRelease,               // 10
-	kPowerEventAssertionSetLevel,              // 11
-	kPowerEventQueueSleepWakeUUID,             // 12
-	kPowerEventPublishSleepWakeUUID,           // 13
-	kPowerEventSetDisplayPowerOn,              // 14
-	kPowerEventPublishWakeType,                // 15
-	kPowerEventAOTEvaluate,                    // 16
-	kPowerEventRunModeRequest                  // 17
+    kPowerEventFeatureChanged = 1,              // 1
+    kPowerEventReceivedPowerNotification,       // 2
+    kPowerEventSystemBootCompleted,             // 3
+    kPowerEventSystemShutdown,                  // 4
+    kPowerEventUserDisabledSleep,               // 5
+    kPowerEventRegisterSystemCapabilityClient,  // 6
+    kPowerEventRegisterKernelCapabilityClient,  // 7
+    kPowerEventPolicyStimulus,                  // 8
+    kPowerEventAssertionCreate,                 // 9
+    kPowerEventAssertionRelease,                // 10
+    kPowerEventAssertionSetLevel,               // 11
+    kPowerEventQueueSleepWakeUUID,              // 12
+    kPowerEventPublishSleepWakeUUID,            // 13
+    kPowerEventSuspendClient                    // 14
 };
 
 // For evaluatePolicy()
 // List of stimuli that affects the root domain policy.
 enum {
-	kStimulusDisplayWranglerSleep,      // 0
-	kStimulusDisplayWranglerWake,       // 1
-	kStimulusAggressivenessChanged,     // 2
-	kStimulusDemandSystemSleep,         // 3
-	kStimulusAllowSystemSleepChanged,   // 4
-	kStimulusDarkWakeActivityTickle,    // 5
-	kStimulusDarkWakeEntry,             // 6
-	kStimulusDarkWakeReentry,           // 7
-	kStimulusDarkWakeEvaluate,          // 8
-	kStimulusNoIdleSleepPreventers,     // 9
-	kStimulusEnterUserActiveState,      // 10
-	kStimulusLeaveUserActiveState       // 11
-};
-
-// Internal power state change reasons
-// Must be less than kIOPMSleepReasonClamshell=101
-enum {
-	kCPSReasonNone = 0,                 // 0
-	kCPSReasonInit,                     // 1
-	kCPSReasonWake,                     // 2
-	kCPSReasonIdleSleepPrevent,         // 3
-	kCPSReasonIdleSleepAllow,           // 4
-	kCPSReasonPowerOverride,            // 5
-	kCPSReasonPowerDownCancel,          // 6
-	kCPSReasonAOTExit,                  // 7
-	kCPSReasonAdjustPowerState,         // 8
-	kCPSReasonDarkWakeCannotSleep,      // 9
-	kCPSReasonIdleSleepEnabled,         // 10
-	kCPSReasonEvaluatePolicy,           // 11
-	kCPSReasonSustainFullWake,          // 12
-	kCPSReasonPMInternals = (kIOPMSleepReasonClamshell - 1)
+    kStimulusDisplayWranglerSleep,      // 0
+    kStimulusDisplayWranglerWake,       // 1
+    kStimulusAggressivenessChanged,     // 2
+    kStimulusDemandSystemSleep,         // 3
+    kStimulusAllowSystemSleepChanged,   // 4
+    kStimulusDarkWakeActivityTickle,    // 5
+    kStimulusDarkWakeEntry,             // 6
+    kStimulusDarkWakeReentry,           // 7
+    kStimulusDarkWakeEvaluate,          // 8
+    kStimulusNoIdleSleepPreventers      // 9
 };
 
 extern "C" {
 IOReturn OSKextSystemSleepOrWake( UInt32 );
 }
-extern "C" ppnum_t      pmap_find_phys(pmap_t pmap, addr64_t va);
-extern "C" addr64_t     kvtophys(vm_offset_t va);
-extern "C" boolean_t    kdp_has_polled_corefile();
 
 static void idleSleepTimerExpired( thread_call_param_t, thread_call_param_t );
-static void notifySystemShutdown( IOService * root, uint32_t messageType );
+static void notifySystemShutdown( IOService * root, unsigned long event );
 static void handleAggressivesFunction( thread_call_param_t, thread_call_param_t );
 static void pmEventTimeStamp(uint64_t *recordTS);
-static void powerButtonUpCallout( thread_call_param_t, thread_call_param_t );
-static void powerButtonDownCallout( thread_call_param_t, thread_call_param_t );
-static OSPtr<const OSSymbol> copyKextIdentifierWithAddress(vm_address_t address);
-
-static int  IOPMConvertSecondsToCalendar(clock_sec_t secs, IOPMCalendarStruct * dt);
-static clock_sec_t IOPMConvertCalendarToSeconds(const IOPMCalendarStruct * dt);
-#define YMDTF       "%04d/%02d/%d %02d:%02d:%02d"
-#define YMDT(cal)   ((int)(cal)->year), (cal)->month, (cal)->day, (cal)->hour, (cal)->minute, (cal)->second
 
 // "IOPMSetSleepSupported"  callPlatformFunction name
-static OSSharedPtr<const OSSymbol>         sleepSupportedPEFunction;
-static OSSharedPtr<const OSSymbol>         sleepMessagePEFunction;
-static OSSharedPtr<const OSSymbol>         gIOPMWakeTypeUserKey;
-
-static OSSharedPtr<const OSSymbol>         gIOPMPSExternalConnectedKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSExternalChargeCapableKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSBatteryInstalledKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSIsChargingKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSAtWarnLevelKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSAtCriticalLevelKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSCurrentCapacityKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSMaxCapacityKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSDesignCapacityKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSTimeRemainingKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSAmperageKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSVoltageKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSCycleCountKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSMaxErrKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSAdapterInfoKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSLocationKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSErrorConditionKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSManufacturerKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSManufactureDateKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSModelKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSSerialKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSLegacyBatteryInfoKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSBatteryHealthKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSHealthConfidenceKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSCapacityEstimatedKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSBatteryChargeStatusKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSBatteryTemperatureKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSAdapterDetailsKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSChargerConfigurationKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSAdapterDetailsIDKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSAdapterDetailsWattsKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSAdapterDetailsRevisionKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSAdapterDetailsSerialNumberKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSAdapterDetailsFamilyKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSAdapterDetailsAmperageKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSAdapterDetailsDescriptionKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSAdapterDetailsPMUConfigurationKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSAdapterDetailsSourceIDKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSAdapterDetailsErrorFlagsKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSAdapterDetailsSharedSourceKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSAdapterDetailsCloakedKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSInvalidWakeSecondsKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSPostChargeWaitSecondsKey;
-static OSSharedPtr<const OSSymbol>         gIOPMPSPostDishargeWaitSecondsKey;
+static const OSSymbol *sleepSupportedPEFunction = NULL;
+static const OSSymbol *sleepMessagePEFunction   = NULL;
 
 #define kIOSleepSupportedKey        "IOSleepSupported"
 #define kIOPMSystemCapabilitiesKey  "System Capabilities"
-#define kIOPMSystemDefaultOverrideKey   "SystemPowerProfileOverrideDict"
-
-#define kIORequestWranglerIdleKey   "IORequestIdle"
-#define kDefaultWranglerIdlePeriod  1000 // in milliseconds
-
-#define kIOSleepWakeFailureString   "SleepWakeFailureString"
-#define kIOEFIBootRomFailureKey     "wake-failure"
-#define kIOSleepWakeFailurePanic    "SleepWakeFailurePanic"
 
 #define kRD_AllPowerSources (kIOPMSupportedOnAC \
-	                   | kIOPMSupportedOnBatt \
-	                   | kIOPMSupportedOnUPS)
-
-#define kLocalEvalClamshellCommand  (1 << 15)
-#define kIdleSleepRetryInterval     (3 * 60 * 1000)
-
-// Minimum time in milliseconds after AP wake that we allow idle timer to expire.
-// We impose this minimum to avoid race conditions in the AP wake path where
-// userspace clients are not able to acquire power assertions before the idle timer expires.
-#if XNU_TARGET_OS_IOS
-#define kMinimumTimeBeforeIdleSleep     3000
-#else
-#define kMinimumTimeBeforeIdleSleep     1000
-#endif
-
-#define DISPLAY_WRANGLER_PRESENT    (!NO_KERNEL_HID)
+                           | kIOPMSupportedOnBatt \
+                           | kIOPMSupportedOnUPS)
+
+enum 
+{
+    // not idle around autowake time, secs
+    kAutoWakePreWindow  = 45,
+    kAutoWakePostWindow = 15
+};
+
+#define kLocalEvalClamshellCommand        (1 << 15)
 
 enum {
-	kWranglerPowerStateMin   = 0,
-	kWranglerPowerStateSleep = 2,
-	kWranglerPowerStateDim   = 3,
-	kWranglerPowerStateMax   = 4
+    OFF_STATE           = 0,
+    RESTART_STATE       = 1,
+    SLEEP_STATE         = 2,
+    ON_STATE            = 3,
+    NUM_POWER_STATES
 };
-
-const char *
-getPowerStateString( uint32_t state )
-{
-#define POWER_STATE(x) {(uint32_t) x, #x}
-
-	static const IONamedValue powerStates[] = {
-		POWER_STATE( OFF_STATE ),
-		POWER_STATE( RESTART_STATE ),
-		POWER_STATE( SLEEP_STATE ),
-		POWER_STATE( AOT_STATE ),
-		POWER_STATE( ON_STATE ),
-		{ 0, NULL }
-	};
-	return IOFindNameForValue(state, powerStates);
-}
 
 #define ON_POWER        kIOPMPowerOn
 #define RESTART_POWER   kIOPMRestart
 #define SLEEP_POWER     kIOPMAuxPowerOn
 
-static IOPMPowerState
-    ourPowerStates[NUM_POWER_STATES] =
-{
-	{   .version                = 1,
-	    .capabilityFlags        = 0,
-	    .outputPowerCharacter   = 0,
-	    .inputPowerRequirement  = 0 },
-	{   .version                = 1,
-	    .capabilityFlags        = kIOPMRestartCapability,
-	    .outputPowerCharacter   = kIOPMRestart,
-	    .inputPowerRequirement  = RESTART_POWER },
-	{   .version                = 1,
-	    .capabilityFlags        = kIOPMSleepCapability,
-	    .outputPowerCharacter   = kIOPMSleep,
-	    .inputPowerRequirement  = SLEEP_POWER },
-	{   .version                = 1,
-	    .capabilityFlags        = kIOPMAOTCapability,
-	    .outputPowerCharacter   = kIOPMAOTPower,
-	    .inputPowerRequirement  = ON_POWER },
-	{   .version                = 1,
-	    .capabilityFlags        = kIOPMPowerOn,
-	    .outputPowerCharacter   = kIOPMPowerOn,
-	    .inputPowerRequirement  = ON_POWER },
+static IOPMPowerState ourPowerStates[NUM_POWER_STATES] =
+{
+    {1, 0,                      0,              0,             0,0,0,0,0,0,0,0},
+    {1, kIOPMRestartCapability,	kIOPMRestart,	RESTART_POWER, 0,0,0,0,0,0,0,0},	
+    {1, kIOPMSleepCapability,   kIOPMSleep,     SLEEP_POWER,   0,0,0,0,0,0,0,0},
+    {1, kIOPMPowerOn,           kIOPMPowerOn,   ON_POWER,      0,0,0,0,0,0,0,0}
 };
 
-#define kIOPMRootDomainWakeTypeSleepService     "SleepService"
-#define kIOPMRootDomainWakeTypeMaintenance      "Maintenance"
-#define kIOPMRootDomainWakeTypeSleepTimer       "SleepTimer"
-#define kIOPMrootDomainWakeTypeLowBattery       "LowBattery"
-#define kIOPMRootDomainWakeTypeUser             "User"
-#define kIOPMRootDomainWakeTypeAlarm            "Alarm"
-#define kIOPMRootDomainWakeTypeNetwork          "Network"
-#define kIOPMRootDomainWakeTypeHIDActivity      "HID Activity"
-#define kIOPMRootDomainWakeTypeNotification     "Notification"
-#define kIOPMRootDomainWakeTypeHibernateError   "HibernateError"
+#define kIOPMRootDomainWakeTypeSleepService "SleepService"
+#define kIOPMRootDomainWakeTypeMaintenance  "Maintenance"
+#define kIOPMRootDomainWakeTypeSleepTimer   "SleepTimer"
+#define kIOPMrootDomainWakeTypeLowBattery   "LowBattery"
+#define kIOPMRootDomainWakeTypeUser         "User"
+#define kIOPMRootDomainWakeTypeAlarm        "Alarm"
+#define kIOPMRootDomainWakeTypeNetwork      "Network"
+#define kIOPMRootDomainWakeTypeHIDActivity  "HID Activity"
 
 // Special interest that entitles the interested client from receiving
 // all system messages. Only used by powerd.
 //
 #define kIOPMSystemCapabilityInterest       "IOPMSystemCapabilityInterest"
 
-// Entitlement required for root domain clients
-#define kRootDomainEntitlementSetProperty   "com.apple.private.iokit.rootdomain-set-property"
-
-#define WAKEEVENT_LOCK()        IOLockLock(wakeEventLock)
-#define WAKEEVENT_UNLOCK()      IOLockUnlock(wakeEventLock)
+#define kPMSuspendedNotificationClients      "PMSuspendedNotificationClients"
 
 /*
  * Aggressiveness
@@ -414,254 +220,80 @@
 
 #define kAggressivesMinValue    1
 
-const char *
-getAggressivenessTypeString( uint32_t type )
-{
-#define AGGRESSIVENESS_TYPE(x) {(uint32_t) x, #x}
-
-	static const IONamedValue aggressivenessTypes[] = {
-		AGGRESSIVENESS_TYPE( kPMGeneralAggressiveness ),
-		AGGRESSIVENESS_TYPE( kPMMinutesToDim ),
-		AGGRESSIVENESS_TYPE( kPMMinutesToSpinDown ),
-		AGGRESSIVENESS_TYPE( kPMMinutesToSleep ),
-		AGGRESSIVENESS_TYPE( kPMEthernetWakeOnLANSettings ),
-		AGGRESSIVENESS_TYPE( kPMSetProcessorSpeed ),
-		AGGRESSIVENESS_TYPE( kPMPowerSource),
-		AGGRESSIVENESS_TYPE( kPMMotionSensor ),
-		AGGRESSIVENESS_TYPE( kPMLastAggressivenessType ),
-		{ 0, NULL }
-	};
-	return IOFindNameForValue(type, aggressivenessTypes);
-}
-
 enum {
-	kAggressivesStateBusy           = 0x01,
-	kAggressivesStateQuickSpindown  = 0x02
+    kAggressivesStateBusy           = 0x01,
+    kAggressivesStateQuickSpindown  = 0x02
 };
 
 struct AggressivesRecord {
-	uint32_t    flags;
-	uint32_t    type;
-	uint32_t    value;
+    uint32_t    flags;
+    uint32_t    type;
+    uint32_t    value;
 };
 
 struct AggressivesRequest {
-	queue_chain_t           chain;
-	uint32_t                options;
-	uint32_t                dataType;
-	union {
-		OSSharedPtr<IOService> service;
-		AggressivesRecord      record;
-	} data;
+    queue_chain_t           chain;
+    uint32_t                options;
+    uint32_t                dataType;
+    union {
+        IOService *         service;
+        AggressivesRecord   record;
+    } data;
 };
 
 enum {
-	kAggressivesRequestTypeService  = 1,
-	kAggressivesRequestTypeRecord
+    kAggressivesRequestTypeService  = 1,
+    kAggressivesRequestTypeRecord
 };
 
 enum {
-	kAggressivesOptionSynchronous          = 0x00000001,
-	kAggressivesOptionQuickSpindownEnable  = 0x00000100,
-	kAggressivesOptionQuickSpindownDisable = 0x00000200,
-	kAggressivesOptionQuickSpindownMask    = 0x00000300
+    kAggressivesOptionSynchronous          = 0x00000001,
+    kAggressivesOptionQuickSpindownEnable  = 0x00000100,
+    kAggressivesOptionQuickSpindownDisable = 0x00000200,
+    kAggressivesOptionQuickSpindownMask    = 0x00000300
 };
 
 enum {
-	kAggressivesRecordFlagModified         = 0x00000001,
-	kAggressivesRecordFlagMinValue         = 0x00000002
+    kAggressivesRecordFlagModified         = 0x00000001,
+    kAggressivesRecordFlagMinValue         = 0x00000002
 };
-
-// System Sleep Preventers
-
-enum {
-	kPMUserDisabledAllSleep = 1,
-	kPMSystemRestartBootingInProgress,
-	kPMConfigPreventSystemSleep,
-	kPMChildPreventSystemSleep,
-	kPMCPUAssertion,
-	kPMPCIUnsupported,
-	kPMDKNotReady,
-};
-
-const char *
-getSystemSleepPreventerString( uint32_t preventer )
-{
-#define SYSTEM_SLEEP_PREVENTER(x) {(int) x, #x}
-	static const IONamedValue systemSleepPreventers[] = {
-		SYSTEM_SLEEP_PREVENTER( kPMUserDisabledAllSleep ),
-		SYSTEM_SLEEP_PREVENTER( kPMSystemRestartBootingInProgress ),
-		SYSTEM_SLEEP_PREVENTER( kPMConfigPreventSystemSleep ),
-		SYSTEM_SLEEP_PREVENTER( kPMChildPreventSystemSleep ),
-		SYSTEM_SLEEP_PREVENTER( kPMCPUAssertion ),
-		SYSTEM_SLEEP_PREVENTER( kPMPCIUnsupported ),
-		SYSTEM_SLEEP_PREVENTER( kPMDKNotReady ),
-		{ 0, NULL }
-	};
-	return IOFindNameForValue(preventer, systemSleepPreventers);
-}
 
 // gDarkWakeFlags
 enum {
-	kDarkWakeFlagPromotionNone       = 0x0000,
-	kDarkWakeFlagPromotionEarly      = 0x0001, // promote before gfx clamp
-	kDarkWakeFlagPromotionLate       = 0x0002, // promote after gfx clamp
-	kDarkWakeFlagPromotionMask       = 0x0003,
-	kDarkWakeFlagAlarmIsDark         = 0x0100,
-	kDarkWakeFlagAudioNotSuppressed  = 0x0200,
-	kDarkWakeFlagUserWakeWorkaround  = 0x1000
+    kDarkWakeFlagHIDTickleEarly      = 0x01, // hid tickle before gfx suppression
+    kDarkWakeFlagHIDTickleLate       = 0x02, // hid tickle after gfx suppression
+    kDarkWakeFlagHIDTickleNone       = 0x03, // hid tickle is not posted
+    kDarkWakeFlagHIDTickleMask       = 0x03,
+    kDarkWakeFlagIgnoreDiskIOInDark  = 0x04, // ignore disk idle in DW
+    kDarkWakeFlagIgnoreDiskIOAlways  = 0x08, // always ignore disk idle
+    kDarkWakeFlagIgnoreDiskIOMask    = 0x0C,
+    kDarkWakeFlagAlarmIsDark         = 0x0100
 };
 
-// gClamshellFlags
-// The workaround for 9157444 is enabled at compile time using the
-// DARK_TO_FULL_EVALUATE_CLAMSHELL_DELAY macro and is not represented below.
-enum {
-	kClamshell_WAR_38378787 = 0x00000001,
-	kClamshell_WAR_47715679 = 0x00000002,
-	kClamshell_WAR_58009435 = 0x00000004
-};
-
-// acceptSystemWakeEvents()
-enum {
-	kAcceptSystemWakeEvents_Disable = 0,
-	kAcceptSystemWakeEvents_Enable,
-	kAcceptSystemWakeEvents_Reenable
-};
-
 static IOPMrootDomain * gRootDomain;
-static IORootParent *   gPatriarch;
-static IONotifier *     gSysPowerDownNotifier = NULL;
+static IONotifier *     gSysPowerDownNotifier = 0;
 static UInt32           gSleepOrShutdownPending = 0;
 static UInt32           gWillShutdown = 0;
 static UInt32           gPagingOff = 0;
 static UInt32           gSleepWakeUUIDIsSet = false;
 static uint32_t         gAggressivesState = 0;
-uint32_t                gHaltTimeMaxLog;
-uint32_t                gHaltTimeMaxPanic;
-IOLock *                gHaltLogLock;
-static char *           gHaltLog;
-enum                  { kHaltLogSize = 2048 };
-static size_t           gHaltLogPos;
-static uint64_t         gHaltStartTime;
-static char             gKextNameBuf[64];
-static size_t           gKextNamePos;
-static bool             gKextNameEnd;
-
-uuid_string_t bootsessionuuid_string;
-
-#if defined(XNU_TARGET_OS_OSX)
-#if DISPLAY_WRANGLER_PRESENT
-static uint32_t         gDarkWakeFlags = kDarkWakeFlagPromotionNone;
-#elif defined(__arm64__)
-// Enable temporary full wake promotion workarounds
-static uint32_t         gDarkWakeFlags = kDarkWakeFlagUserWakeWorkaround;
-#else
-// Enable full wake promotion workarounds
-static uint32_t         gDarkWakeFlags = kDarkWakeFlagUserWakeWorkaround;
-#endif
-#else  /* !defined(XNU_TARGET_OS_OSX) */
-static uint32_t         gDarkWakeFlags = kDarkWakeFlagPromotionEarly;
-#endif /* !defined(XNU_TARGET_OS_OSX) */
-
-static uint32_t         gNoIdleFlag = 0;
-static uint32_t         gSleepDisabledFlag = 0;
-static uint32_t         gSwdPanic = 1;
-static uint32_t         gSwdSleepTimeout = 0;
-static uint32_t         gSwdWakeTimeout = 0;
-static uint32_t         gSwdSleepWakeTimeout = 0;
+static uint32_t         gDarkWakeFlags = kDarkWakeFlagHIDTickleNone | kDarkWakeFlagIgnoreDiskIOAlways;
 static PMStatsStruct    gPMStats;
-#if DEVELOPMENT || DEBUG
-static uint32_t swd_panic_phase;
-#endif
-
-static uint32_t         gClamshellFlags = 0
-#if defined(__i386__) || defined(__x86_64__)
-    | kClamshell_WAR_58009435
-#endif
-;
-
-#if HIBERNATION
-
-#if defined(__arm64__)
-static IOReturn
-defaultSleepPolicyHandler(void *ctx, const IOPMSystemSleepPolicyVariables *vars, IOPMSystemSleepParameters *params)
-{
-	uint32_t sleepType = kIOPMSleepTypeDeepIdle;
-
-	assert(vars->signature == kIOPMSystemSleepPolicySignature);
-	assert(vars->version == kIOPMSystemSleepPolicyVersion);
-
-	// Hibernation enabled and either user forced hibernate or low battery sleep
-	if ((vars->hibernateMode & kIOHibernateModeOn) &&
-	    (((vars->hibernateMode & kIOHibernateModeSleep) == 0) ||
-	    (vars->sleepFactors & kIOPMSleepFactorBatteryLow))) {
-		sleepType = kIOPMSleepTypeHibernate;
-	}
-	params->version = kIOPMSystemSleepParametersVersion;
-	params->sleepType = sleepType;
-	return kIOReturnSuccess;
-}
-static IOPMSystemSleepPolicyHandler     gSleepPolicyHandler = &defaultSleepPolicyHandler;
-#else /* defined(__arm64__) */
-static IOPMSystemSleepPolicyHandler     gSleepPolicyHandler = NULL;
-#endif /* defined(__arm64__) */
-
-static IOPMSystemSleepPolicyVariables * gSleepPolicyVars = NULL;
-static void *                           gSleepPolicyTarget;
-#endif
 
 struct timeval gIOLastSleepTime;
 struct timeval gIOLastWakeTime;
-AbsoluteTime gIOLastWakeAbsTime;
-AbsoluteTime gIOLastSleepAbsTime;
-
-struct timeval gIOLastUserSleepTime;
-
-static char gWakeReasonString[128];
-static char gBootReasonString[80];
-static char gShutdownReasonString[80];
-static uint64_t gShutdownTime;
-static bool gWakeReasonSysctlRegistered = false;
-static bool gBootReasonSysctlRegistered = false;
-static bool gShutdownReasonSysctlRegistered = false;
-static bool gWillShutdownSysctlRegistered = false;
-static AbsoluteTime gUserActiveAbsTime;
-static AbsoluteTime gUserInactiveAbsTime;
-
-#if defined(__i386__) || defined(__x86_64__) || (defined(__arm64__) && HIBERNATION)
-static bool gSpinDumpBufferFull = false;
-#endif
-
-z_stream          swd_zs;
-vm_offset_t swd_zs_zmem;
-//size_t swd_zs_zsize;
-size_t swd_zs_zoffset;
-#if defined(__i386__) || defined(__x86_64__)
-IOCPU *currentShutdownTarget = NULL;
-#endif
-
-static unsigned int     gPMHaltBusyCount;
-static unsigned int     gPMHaltIdleCount;
-static int              gPMHaltDepth;
-static uint32_t         gPMHaltMessageType;
-static IOLock *         gPMHaltLock  = NULL;
-static OSSharedPtr<OSArray>        gPMHaltArray;
-static OSSharedPtr<const OSSymbol> gPMHaltClientAcknowledgeKey;
-static bool             gPMQuiesced;
 
 // Constants used as arguments to IOPMrootDomain::informCPUStateChange
 #define kCPUUnknownIndex    9999999
 enum {
-	kInformAC = 0,
-	kInformLid = 1,
-	kInformableCount = 2
+    kInformAC = 0,
+    kInformLid = 1,
+    kInformableCount = 2
 };
 
-OSSharedPtr<const OSSymbol> gIOPMStatsResponseTimedOut;
-OSSharedPtr<const OSSymbol> gIOPMStatsResponseCancel;
-OSSharedPtr<const OSSymbol> gIOPMStatsResponseSlow;
-OSSharedPtr<const OSSymbol> gIOPMStatsResponsePrompt;
-OSSharedPtr<const OSSymbol> gIOPMStatsDriverPSChangeSlow;
+const OSSymbol *gIOPMStatsApplicationResponseTimedOut;
+const OSSymbol *gIOPMStatsApplicationResponseCancel;
+const OSSymbol *gIOPMStatsApplicationResponseSlow;
 
 #define kBadPMFeatureID     0
 
@@ -671,12 +303,12 @@
  */
 class PMSettingHandle : public OSObject
 {
-	OSDeclareFinalStructors( PMSettingHandle );
-	friend class PMSettingObject;
+    OSDeclareFinalStructors( PMSettingHandle )
+    friend class PMSettingObject;
 
 private:
-	PMSettingObject *pmso;
-	void free(void) APPLE_KEXT_OVERRIDE;
+    PMSettingObject *pmso;
+    void free(void);
 };
 
 /*
@@ -685,46 +317,114 @@
  */
 class PMSettingObject : public OSObject
 {
-	OSDeclareFinalStructors( PMSettingObject );
-	friend class IOPMrootDomain;
+    OSDeclareFinalStructors( PMSettingObject )
+    friend class IOPMrootDomain;
 
 private:
-	queue_head_t                    calloutQueue;
-	thread_t                        waitThread;
-	IOPMrootDomain                  *parent;
-	PMSettingHandle                 *pmsh;
-	IOPMSettingControllerCallback   func;
-	OSObject                        *target;
-	uintptr_t                       refcon;
-	OSDataAllocation<uint32_t>      publishedFeatureID;
-	uint32_t                        settingCount;
-	bool                            disabled;
-
-	void free(void) APPLE_KEXT_OVERRIDE;
+    queue_head_t                    calloutQueue;
+    thread_t                        waitThread;
+    IOPMrootDomain                  *parent;
+    PMSettingHandle                 *pmsh;
+    IOPMSettingControllerCallback   func;
+    OSObject                        *target;
+    uintptr_t                       refcon;
+    uint32_t                        *publishedFeatureID;
+    uint32_t                        settingCount;
+    bool                            disabled;
+
+    void free(void);
 
 public:
-	static PMSettingObject *pmSettingObject(
-		IOPMrootDomain                  *parent_arg,
-		IOPMSettingControllerCallback   handler_arg,
-		OSObject                        *target_arg,
-		uintptr_t                       refcon_arg,
-		uint32_t                        supportedPowerSources,
-		const OSSymbol                  *settings[],
-		OSObject                        **handle_obj);
-
-	IOReturn dispatchPMSetting(const OSSymbol *type, OSObject *object);
-	void clientHandleFreed(void);
+    static PMSettingObject *pmSettingObject(
+                IOPMrootDomain                  *parent_arg,
+                IOPMSettingControllerCallback   handler_arg,
+                OSObject                        *target_arg,
+                uintptr_t                       refcon_arg,
+                uint32_t                        supportedPowerSources,
+                const OSSymbol                  *settings[],
+                OSObject                        **handle_obj);
+
+    void dispatchPMSetting(const OSSymbol *type, OSObject *object);
+    void clientHandleFreed(void);
 };
 
 struct PMSettingCallEntry {
-	queue_chain_t   link;
-	thread_t        thread;
+    queue_chain_t   link;
+    thread_t        thread;
 };
 
 #define PMSETTING_LOCK()    IOLockLock(settingsCtrlLock)
 #define PMSETTING_UNLOCK()  IOLockUnlock(settingsCtrlLock)
 #define PMSETTING_WAIT(p)   IOLockSleep(settingsCtrlLock, p, THREAD_UNINT)
 #define PMSETTING_WAKEUP(p) IOLockWakeup(settingsCtrlLock, p, true)
+
+//*********************************************************************************
+//*********************************************************************************
+//*********************************************************************************
+
+/* @class IOPMTimeline
+ * @astract Tracks & records PM activity.
+ * @discussion Intended for use only as a helper-class to IOPMrootDomain.
+ *      Do not subclass or directly invoke iOPMTimeline
+ */
+class IOPMTimeline : public OSObject 
+{
+    OSDeclareDefaultStructors( IOPMTimeline );
+
+public:  
+    static IOPMTimeline* timeline(IOPMrootDomain *root_domain);
+  
+    bool            setProperties(OSDictionary *d);
+    OSDictionary    *copyInfoDictionary(void);
+    
+    IOReturn    recordSystemPowerEvent( PMEventDetails *details );
+                                
+    IOReturn    recordDetailedPowerEvent( PMEventDetails *details );
+
+    IOMemoryDescriptor      *getPMTraceMemoryDescriptor();
+    
+    uint32_t getNumEventsLoggedThisPeriod();    
+    void     setNumEventsLoggedThisPeriod(uint32_t newCount);
+    bool     isSleepCycleInProgress();
+    void     setSleepCycleInProgressFlag(bool flag);
+private:
+    bool        init(void);
+    void        free(void);
+
+    void        setEventsTrackedCount(uint32_t newTracked);
+    void        setEventsRecordingLevel(uint32_t eventsTrackedBits);
+    static uint32_t _atomicIndexIncrement(uint32_t *index, uint32_t limit);
+    
+    enum {
+        kPMTimelineRecordTardyDrivers   = 1 << 0,
+        kPMTmielineRecordSystemEvents   = 1 << 1,
+        kPMTimelineRecordAllDrivers     = 1 << 2,
+        kPMTimelineRecordOff            = 0,
+        kPMTimelineRecordDefault        = 3,
+        kPMTimelineRecordDebug          = 7    
+    };
+
+    // eventsRecordingLevel is a bitfield defining which PM driver events will get logged
+    // into the PM buffer. 
+    uint32_t                    eventsRecordingLevel;
+    
+    // pmTraceMemoryDescriptor represents the memory block that IOPMTimeLine records PM trace points into.
+    IOBufferMemoryDescriptor    *pmTraceMemoryDescriptor;
+
+    // Pointer to starting address in pmTraceMemoryDescriptor
+    IOPMSystemEventRecord       *traceBuffer;
+    IOPMTraceBufferHeader       *hdr;
+
+    uint16_t                    systemState;
+    
+    IOLock                      *logLock;
+    IOPMrootDomain              *owner;
+
+    uint32_t                    numEventsLoggedThisPeriod;
+    bool                        sleepCycleInProgress;
+};
+
+OSDefineMetaClassAndStructors( IOPMTimeline, OSObject )
 
 /*
  * PMTraceWorker
@@ -734,124 +434,37 @@
  */
 
 typedef void (*IOPMTracePointHandler)(
-	void * target, uint32_t code, uint32_t data );
+        void * target, uint32_t code, uint32_t data );
 
 class PMTraceWorker : public OSObject
 {
-	OSDeclareDefaultStructors(PMTraceWorker);
+    OSDeclareDefaultStructors(PMTraceWorker)
 public:
-	typedef enum { kPowerChangeStart, kPowerChangeCompleted } change_t;
-
-	static OSPtr<PMTraceWorker> tracer( IOPMrootDomain * );
-	void                        tracePCIPowerChange(change_t, IOService *, uint32_t, uint32_t);
-	void                        tracePoint(uint8_t phase);
-	void                        traceDetail(uint32_t detail);
-	void                        traceComponentWakeProgress(uint32_t component, uint32_t data);
-	int                         recordTopLevelPCIDevice(IOService *);
-	void                        RTC_TRACE(void);
-	virtual bool                serialize(OSSerialize *s) const APPLE_KEXT_OVERRIDE;
-
-	IOPMTracePointHandler       tracePointHandler;
-	void *                      tracePointTarget;
-	uint64_t                    getPMStatusCode();
-	uint8_t                     getTracePhase();
-	uint32_t                    getTraceData();
+    typedef enum { kPowerChangeStart, kPowerChangeCompleted } change_t;
+
+    static PMTraceWorker        *tracer( IOPMrootDomain * );
+    void                        tracePCIPowerChange(change_t, IOService *, uint32_t, uint32_t);
+    void                        tracePoint(uint8_t phase);
+    void                        tracePoint(uint8_t phase, uint8_t data8);
+    void                        traceDetail(uint32_t detail);
+    void                        traceLoginWindowPhase(uint8_t phase);
+    int                         recordTopLevelPCIDevice(IOService *);
+    void                        RTC_TRACE(void);
+    virtual bool				serialize(OSSerialize *s) const;
+
+    IOPMTracePointHandler       tracePointHandler;
+    void *                      tracePointTarget;
 private:
-	IOPMrootDomain              *owner;
-	IOLock                      *pmTraceWorkerLock;
-	OSSharedPtr<OSArray>         pciDeviceBitMappings;
-
-	uint8_t                     addedToRegistry;
-	uint8_t                     tracePhase;
-	uint32_t                    traceData32;
-	uint8_t                     loginWindowData;
-	uint8_t                     coreDisplayData;
-	uint8_t                     coreGraphicsData;
+    IOPMrootDomain              *owner;
+    IOLock                      *pciMappingLock;
+    OSArray                     *pciDeviceBitMappings;
+
+    uint8_t                     addedToRegistry;
+    uint8_t                     tracePhase;
+    uint8_t                     loginWindowPhase;
+    uint8_t                     traceData8;
+    uint32_t                    traceData32;
 };
-
-struct IOPMAssertionLog {
-	IOPMAssertionLogData data;
-
-	mach_port_t  notificationPort;
-	uint64_t     notificationThreshold;
-
-public:
-	IOReturn
-	setNotificationThreshold(uint64_t threshold)
-	{
-		if (threshold > ARRAY_LEN(data.intervals)) {
-			return kIOReturnBadArgument;
-		}
-		notificationThreshold = threshold;
-		return kIOReturnSuccess;
-	}
-
-	IOReturn
-	setNotificationPort(mach_port_t port)
-	{
-		if (port != MACH_PORT_NULL && notificationPort != MACH_PORT_NULL) {
-			return kIOReturnExclusiveAccess;
-		}
-
-		notificationPort = port;
-		return kIOReturnSuccess;
-	}
-
-	void
-	notify()
-	{
-		if (notificationPort == MACH_PORT_NULL) {
-			return;
-		}
-
-		mach_msg_header_t msg;
-		msg.msgh_bits         = MACH_MSGH_BITS_SET(MACH_MSG_TYPE_COPY_SEND, 0, 0, 0);
-		msg.msgh_id           = 0;
-		msg.msgh_size         = sizeof(mach_msg_header_t);
-		msg.msgh_local_port   = MACH_PORT_NULL;
-		msg.msgh_remote_port  = notificationPort;
-		(void)mach_msg_send_from_kernel_with_options(&msg, msg.msgh_size, MACH64_SEND_TIMEOUT, MACH_MSG_TIMEOUT_NONE);
-	}
-
-	void
-	addInterval(IOPMDriverAssertionID id, uint64_t create_timestamp, uint64_t delete_timestamp)
-	{
-		IOPMAssertionLogData::Interval& interval = data.intervals[data.intervals_pos++ % ARRAY_LEN(data.intervals)];
-		interval.id = id;
-		interval.create_timestamp = create_timestamp;
-		interval.delete_timestamp = delete_timestamp;
-
-		if (notificationPort && notificationThreshold == (data.intervals_pos % ARRAY_LEN(data.intervals))) {
-			notify();
-		}
-	}
-
-	void
-	addName(IOPMDriverAssertionID id, const char *name)
-	{
-		IOPMAssertionLogData::Properties& prop = data.props[data.props_pos++ % ARRAY_LEN(data.props)];
-		prop.id = id;
-		strlcpy(prop.name, name, sizeof(prop.name));
-	}
-};
-
-/*
- * this should be treated as POD, as it's byte-copied around
- * and we cannot rely on d'tor firing at the right time
- */
-struct PMAssertStruct {
-	IOPMDriverAssertionID       id;
-	IOPMDriverAssertionType     assertionBits;
-	uint64_t                    createdTime;
-	uint64_t                    modifiedTime;
-	const OSSymbol              *ownerString;
-	IOService                   *ownerService;
-	uint64_t                    registryEntryID;
-	IOPMDriverAssertionLevel    level;
-	uint64_t                    assertCPUStartTime;
-	uint64_t                    assertCPUDuration;
-};
-OSDefineValueObjectForDependentType(PMAssertStruct)
 
 /*
  * PMAssertionsTracker
@@ -859,52 +472,53 @@
  */
 class PMAssertionsTracker : public OSObject
 {
-	OSDeclareFinalStructors(PMAssertionsTracker);
+    OSDeclareFinalStructors(PMAssertionsTracker)
 public:
-	static PMAssertionsTracker  *pmAssertionsTracker( IOPMrootDomain * );
-
-	IOReturn                    createAssertion(IOPMDriverAssertionType, IOPMDriverAssertionLevel, IOService *, const char *, IOPMDriverAssertionID *);
-	IOReturn                    releaseAssertion(IOPMDriverAssertionID);
-	IOReturn                    setAssertionLevel(IOPMDriverAssertionID, IOPMDriverAssertionLevel);
-	IOReturn                    setUserAssertionLevels(IOPMDriverAssertionType);
-
-	OSSharedPtr<OSArray>        copyAssertionsArray(void);
-	IOPMDriverAssertionType     getActivatedAssertions(void);
-	IOPMDriverAssertionLevel    getAssertionLevel(IOPMDriverAssertionType);
-
-	IOReturn                    handleCreateAssertion(OSValueObject<PMAssertStruct> *);
-	IOReturn                    handleReleaseAssertion(IOPMDriverAssertionID);
-	IOReturn                    handleSetAssertionLevel(IOPMDriverAssertionID, IOPMDriverAssertionLevel);
-	IOReturn                    handleSetUserAssertionLevels(void * arg0);
-	void                        publishProperties(void);
-	void                        reportCPUBitAccounting(void);
-	PMAssertStruct              *detailsForID(IOPMDriverAssertionID, int *);
+    static PMAssertionsTracker  *pmAssertionsTracker( IOPMrootDomain * );
+    
+    IOReturn                    createAssertion(IOPMDriverAssertionType, IOPMDriverAssertionLevel, IOService *, const char *, IOPMDriverAssertionID *);
+    IOReturn                    releaseAssertion(IOPMDriverAssertionID);
+    IOReturn                    setAssertionLevel(IOPMDriverAssertionID, IOPMDriverAssertionLevel);
+    IOReturn                    setUserAssertionLevels(IOPMDriverAssertionType);
+
+    OSArray                     *copyAssertionsArray(void);
+    IOPMDriverAssertionType     getActivatedAssertions(void);
+    IOPMDriverAssertionLevel    getAssertionLevel(IOPMDriverAssertionType);
+
+    IOReturn                    handleCreateAssertion(OSData *);
+    IOReturn                    handleReleaseAssertion(IOPMDriverAssertionID);
+    IOReturn                    handleSetAssertionLevel(IOPMDriverAssertionID, IOPMDriverAssertionLevel);
+    IOReturn                    handleSetUserAssertionLevels(void * arg0);
+    void                        publishProperties(void);
 
 private:
-	uint32_t                    tabulateProducerCount;
-	uint32_t                    tabulateConsumerCount;
-
-	uint64_t                    maxAssertCPUDuration;
-	uint64_t                    maxAssertCPUEntryId;
-
-	void                        tabulate(void);
-	void                        updateCPUBitAccounting(PMAssertStruct * assertStruct);
-
-	IOPMrootDomain              *owner;
-	OSSharedPtr<OSArray>        assertionsArray;
-	IOLock                      *assertionsArrayLock;
-	IOPMDriverAssertionID       issuingUniqueID __attribute__((aligned(8)));/* aligned for atomic access */
-	IOPMDriverAssertionType     assertionsKernel;
-	IOPMDriverAssertionType     assertionsUser;
-	IOPMDriverAssertionType     assertionsCombined;
-
-	IOPMAssertionLog            assertionsLog;
-
-	friend class IOPMrootDomain;
+    typedef struct {
+        IOPMDriverAssertionID       id;
+        IOPMDriverAssertionType     assertionBits;
+        uint64_t                    createdTime;
+        uint64_t                    modifiedTime;
+        const OSSymbol              *ownerString;
+        IOService                   *ownerService;
+        IOPMDriverAssertionLevel    level;
+    } PMAssertStruct;
+
+    uint32_t                    tabulateProducerCount;
+    uint32_t                    tabulateConsumerCount;
+
+    PMAssertStruct              *detailsForID(IOPMDriverAssertionID, int *);
+    void                        tabulate(void);
+ 
+    IOPMrootDomain              *owner;
+    OSArray                     *assertionsArray;
+    IOLock                      *assertionsArrayLock;
+    IOPMDriverAssertionID       issuingUniqueID __attribute__((aligned(8))); /* aligned for atomic access */
+    IOPMDriverAssertionType     assertionsKernel;
+    IOPMDriverAssertionType     assertionsUser;
+    IOPMDriverAssertionType     assertionsCombined;
 };
-
+ 
 OSDefineMetaClassAndFinalStructors(PMAssertionsTracker, OSObject);
-
+ 
 /*
  * PMHaltWorker
  * Internal helper object for Shutdown/Restart notifications.
@@ -914,21 +528,21 @@
 
 class PMHaltWorker : public OSObject
 {
-	OSDeclareFinalStructors( PMHaltWorker );
+    OSDeclareFinalStructors( PMHaltWorker )
 
 public:
-	IOService *  service;// service being worked on
-	AbsoluteTime startTime; // time when work started
-	int          depth;  // work on nubs at this PM-tree depth
-	int          visits; // number of nodes visited (debug)
-	IOLock *     lock;
-	bool         timeout;// service took too long
-
-	static  PMHaltWorker * worker( void );
-	static  void main( void * arg, wait_result_t waitResult );
-	static  void work( PMHaltWorker * me );
-	static  void checkTimeout( PMHaltWorker * me, AbsoluteTime * now );
-	virtual void free( void ) APPLE_KEXT_OVERRIDE;
+    IOService *  service;    // service being worked on
+    AbsoluteTime startTime;  // time when work started
+    int          depth;      // work on nubs at this PM-tree depth
+    int          visits;     // number of nodes visited (debug)
+    IOLock *     lock;
+    bool         timeout;    // service took too long
+
+    static  PMHaltWorker * worker( void );
+    static  void main( void * arg, wait_result_t waitResult );
+    static  void work( PMHaltWorker * me );
+    static  void checkTimeout( PMHaltWorker * me, AbsoluteTime * now );
+    virtual void free( void );
 };
 
 OSDefineMetaClassAndFinalStructors( PMHaltWorker, OSObject )
@@ -937,323 +551,163 @@
 #define super IOService
 OSDefineMetaClassAndFinalStructors(IOPMrootDomain, IOService)
 
-boolean_t
-IOPMRootDomainGetWillShutdown(void)
-{
-	return gWillShutdown != 0;
-}
-
-static void
-IOPMRootDomainWillShutdown(void)
-{
-	if (OSCompareAndSwap(0, 1, &gWillShutdown)) {
-		IOService::willShutdown();
-		for (int i = 0; i < 100; i++) {
-			if (OSCompareAndSwap(0, 1, &gSleepOrShutdownPending)) {
-				break;
-			}
-			IOSleep( 100 );
+static void IOPMRootDomainWillShutdown(void)
+{
+    if (OSCompareAndSwap(0, 1, &gWillShutdown))
+    {
+	OSKext::willShutdown();
+	for (int i = 0; i < 100; i++)
+	{
+	    if (OSCompareAndSwap(0, 1, &gSleepOrShutdownPending)) break;
+	    IOSleep( 100 );
+	}
+    }
+}
+
+extern "C"
+{
+    IONotifier * registerSleepWakeInterest(IOServiceInterestHandler handler, void * self, void * ref)
+    {
+        return gRootDomain->registerInterest( gIOGeneralInterest, handler, self, ref );
+    }
+
+    IONotifier * registerPrioritySleepWakeInterest(IOServiceInterestHandler handler, void * self, void * ref)
+    {
+        return gRootDomain->registerInterest( gIOPriorityPowerStateInterest, handler, self, ref );
+    }
+
+    IOReturn acknowledgeSleepWakeNotification(void * PMrefcon)
+    {
+        return gRootDomain->allowPowerChange ( (unsigned long)PMrefcon );
+    }
+
+    IOReturn vetoSleepWakeNotification(void * PMrefcon)
+    {
+        return gRootDomain->cancelPowerChange ( (unsigned long)PMrefcon );
+    }
+    
+    IOReturn rootDomainRestart ( void )
+    {
+        return gRootDomain->restartSystem();
+    }
+    
+    IOReturn rootDomainShutdown ( void )
+    {
+        return gRootDomain->shutdownSystem();
+    }
+
+    void IOSystemShutdownNotification(void)
+    {
+    	IOPMRootDomainWillShutdown();
+		if (OSCompareAndSwap(0, 1, &gPagingOff))
+		{
+#if !CONFIG_EMBEDDED
+			gRootDomain->handlePlatformHaltRestart(kPEPagingOff);
+#endif
 		}
-	}
-}
-
-extern "C" IONotifier *
-registerSleepWakeInterest(IOServiceInterestHandler handler, void * self, void * ref)
-{
-	return gRootDomain->registerInterest( gIOGeneralInterest, handler, self, ref ).detach();
-}
-
-extern "C" IONotifier *
-registerPrioritySleepWakeInterest(IOServiceInterestHandler handler, void * self, void * ref)
-{
-	return gRootDomain->registerInterest( gIOPriorityPowerStateInterest, handler, self, ref ).detach();
-}
-
-extern "C" IOReturn
-acknowledgeSleepWakeNotification(void * PMrefcon)
-{
-	return gRootDomain->allowPowerChange((unsigned long)PMrefcon );
-}
-
-extern "C" IOReturn
-vetoSleepWakeNotification(void * PMrefcon)
-{
-	return gRootDomain->cancelPowerChange((unsigned long)PMrefcon );
-}
-
-extern "C" IOReturn
-rootDomainRestart( void )
-{
-	return gRootDomain->restartSystem();
-}
-
-extern "C" IOReturn
-rootDomainShutdown( void )
-{
-	return gRootDomain->shutdownSystem();
-}
-
-static void
-halt_log_putc(char c)
-{
-	if (gHaltLogPos >= (kHaltLogSize - 2)) {
-		return;
-	}
-	gHaltLog[gHaltLogPos++] = c;
-}
-
-extern "C" void
-_doprnt_log(const char     *fmt,
-    va_list                 *argp,
-    void                    (*putc)(char),
-    int                     radix);
-
-static int
-halt_log(const char *fmt, ...)
-{
-	va_list listp;
-
-	va_start(listp, fmt);
-	_doprnt_log(fmt, &listp, &halt_log_putc, 16);
-	va_end(listp);
-
-	return 0;
-}
-
-extern "C" void
-halt_log_enter(const char * what, const void * pc, uint64_t time)
-{
-	uint64_t nano, millis;
-
-	if (!gHaltLog) {
-		return;
-	}
-	absolutetime_to_nanoseconds(time, &nano);
-	millis = nano / NSEC_PER_MSEC;
-	if (millis < 100) {
-		return;
-	}
-
-	IOLockLock(gHaltLogLock);
-	if (pc) {
-		halt_log("%s: %qd ms @ 0x%lx, ", what, millis, VM_KERNEL_UNSLIDE(pc));
-		OSKext::printKextsInBacktrace((vm_offset_t *) &pc, 1, &halt_log,
-		    OSKext::kPrintKextsLock | OSKext::kPrintKextsUnslide | OSKext::kPrintKextsTerse);
-	} else {
-		halt_log("%s: %qd ms\n", what, millis);
-	}
-
-	gHaltLog[gHaltLogPos] = 0;
-	IOLockUnlock(gHaltLogLock);
-}
-
-extern  uint32_t                           gFSState;
-
-extern "C" void
-IOSystemShutdownNotification(int howto, int stage)
-{
-	uint64_t startTime;
-
-	if (kIOSystemShutdownNotificationStageRootUnmount == stage) {
-#if defined(XNU_TARGET_OS_OSX)
-		uint64_t nano, millis;
-		startTime = mach_absolute_time();
-		IOService::getPlatform()->waitQuiet(30 * NSEC_PER_SEC);
-		absolutetime_to_nanoseconds(mach_absolute_time() - startTime, &nano);
-		millis = nano / NSEC_PER_MSEC;
-		if (gHaltTimeMaxLog && (millis >= gHaltTimeMaxLog)) {
-			printf("waitQuiet() for unmount %qd ms\n", millis);
-		}
-#endif /* defined(XNU_TARGET_OS_OSX) */
-		return;
-	}
-
-	if (kIOSystemShutdownNotificationTerminateDEXTs == stage) {
-		uint64_t nano, millis;
-		startTime = mach_absolute_time();
-		IOServicePH::systemHalt(howto);
-		absolutetime_to_nanoseconds(mach_absolute_time() - startTime, &nano);
-		millis = nano / NSEC_PER_MSEC;
-		if (true || (gHaltTimeMaxLog && (millis >= gHaltTimeMaxLog))) {
-			printf("IOServicePH::systemHalt took %qd ms\n", millis);
-		}
-		return;
-	}
-
-	assert(kIOSystemShutdownNotificationStageProcessExit == stage);
-
-	IOLockLock(gHaltLogLock);
-	if (!gHaltLog) {
-		gHaltLog = IONewData(char, (vm_size_t)kHaltLogSize);
-		gHaltStartTime = mach_absolute_time();
-		if (gHaltLog) {
-			halt_log_putc('\n');
-		}
-	}
-	IOLockUnlock(gHaltLogLock);
-
-	startTime = mach_absolute_time();
-	IOPMRootDomainWillShutdown();
-	halt_log_enter("IOPMRootDomainWillShutdown", NULL, mach_absolute_time() - startTime);
-#if HIBERNATION
-	startTime = mach_absolute_time();
-	IOHibernateSystemPostWake(true);
-	halt_log_enter("IOHibernateSystemPostWake", NULL, mach_absolute_time() - startTime);
+    }
+
+    int sync_internal(void);    
+}
+
+/*
+A device is always in the highest power state which satisfies its driver,
+its policy-maker, and any power children it has, but within the constraint
+of the power state provided by its parent.  The driver expresses its desire by
+calling changePowerStateTo(), the policy-maker expresses its desire by calling
+changePowerStateToPriv(), and the children express their desires by calling
+requestPowerDomainState().
+
+The Root Power Domain owns the policy for idle and demand sleep for the system.
+It is a power-managed IOService just like the others in the system.
+It implements several power states which map to what we see as Sleep and On.
+
+The sleep policy is as follows:
+1. Sleep is prevented if the case is open so that nobody will think the machine
+   is off and plug/unplug cards.
+2. Sleep is prevented if the sleep timeout slider in the prefs panel is zero.
+3. System cannot Sleep if some object in the tree is in a power state marked
+   kIOPMPreventSystemSleep.
+
+These three conditions are enforced using the "driver clamp" by calling
+changePowerStateTo(). For example, if the case is opened,
+changePowerStateTo(ON_STATE) is called to hold the system on regardless
+of the desires of the children of the root or the state of the other clamp.
+
+Demand Sleep is initiated by pressing the front panel power button, closing
+the clamshell, or selecting the menu item. In this case the root's parent
+actually initiates the power state change so that the root domain has no
+choice and does not give applications the opportunity to veto the change.
+
+Idle Sleep occurs if no objects in the tree are in a state marked
+kIOPMPreventIdleSleep.  When this is true, the root's children are not holding
+the root on, so it sets the "policy-maker clamp" by calling
+changePowerStateToPriv(ON_STATE) to hold itself on until the sleep timer expires.
+This timer is set for the difference between the sleep timeout slider and the
+display dim timeout slider. When the timer expires, it releases its clamp and
+now nothing is holding it awake, so it falls asleep.
+
+Demand sleep is prevented when the system is booting.  When preferences are
+transmitted by the loginwindow at the end of boot, a flag is cleared,
+and this allows subsequent Demand Sleep.
+*/
+
+//******************************************************************************
+
+IOPMrootDomain * IOPMrootDomain::construct( void )
+{
+    IOPMrootDomain  *root;
+
+    root = new IOPMrootDomain;
+    if( root)
+        root->init();
+
+    return( root );
+}
+
+//******************************************************************************
+
+static void disk_sync_callout( thread_call_param_t p0, thread_call_param_t p1 )
+{
+    IOService * rootDomain = (IOService *) p0;
+    uint32_t    notifyRef  = (uint32_t)(uintptr_t) p1;
+    uint32_t    powerState = rootDomain->getPowerState();
+
+    DLOG("disk_sync_callout ps=%u\n", powerState);
+
+    if (ON_STATE == powerState)
+    {
+#if	HIBERNATION
+        IOHibernateSystemSleep();
 #endif
-	if (OSCompareAndSwap(0, 1, &gPagingOff)) {
-		gRootDomain->handlePlatformHaltRestart(kPEPagingOff);
-	}
-}
-
-extern "C" int sync_internal(void);
-
-/*
- *  A device is always in the highest power state which satisfies its driver,
- *  its policy-maker, and any power children it has, but within the constraint
- *  of the power state provided by its parent.  The driver expresses its desire by
- *  calling changePowerStateTo(), the policy-maker expresses its desire by calling
- *  changePowerStateToPriv(), and the children express their desires by calling
- *  requestPowerDomainState().
- *
- *  The Root Power Domain owns the policy for idle and demand sleep for the system.
- *  It is a power-managed IOService just like the others in the system.
- *  It implements several power states which map to what we see as Sleep and On.
- *
- *  The sleep policy is as follows:
- *  1. Sleep is prevented if the case is open so that nobody will think the machine
- *  is off and plug/unplug cards.
- *  2. Sleep is prevented if the sleep timeout slider in the prefs panel is zero.
- *  3. System cannot Sleep if some object in the tree is in a power state marked
- *  kIOPMPreventSystemSleep.
- *
- *  These three conditions are enforced using the "driver clamp" by calling
- *  changePowerStateTo(). For example, if the case is opened,
- *  changePowerStateTo(ON_STATE) is called to hold the system on regardless
- *  of the desires of the children of the root or the state of the other clamp.
- *
- *  Demand Sleep is initiated by pressing the front panel power button, closing
- *  the clamshell, or selecting the menu item. In this case the root's parent
- *  actually initiates the power state change so that the root domain has no
- *  choice and does not give applications the opportunity to veto the change.
- *
- *  Idle Sleep occurs if no objects in the tree are in a state marked
- *  kIOPMPreventIdleSleep.  When this is true, the root's children are not holding
- *  the root on, so it sets the "policy-maker clamp" by calling
- *  changePowerStateToPriv(ON_STATE) to hold itself on until the sleep timer expires.
- *  This timer is set for the difference between the sleep timeout slider and the
- *  display dim timeout slider. When the timer expires, it releases its clamp and
- *  now nothing is holding it awake, so it falls asleep.
- *
- *  Demand sleep is prevented when the system is booting.  When preferences are
- *  transmitted by the loginwindow at the end of boot, a flag is cleared,
- *  and this allows subsequent Demand Sleep.
- */
-
-//******************************************************************************
-
-IOPMrootDomain *
-IOPMrootDomain::construct( void )
-{
-	IOPMrootDomain  *root;
-
-	root = new IOPMrootDomain;
-	if (root) {
-		root->init();
-	}
-
-	return root;
-}
-
-//******************************************************************************
-// updateConsoleUsersCallout
-//
-//******************************************************************************
-
-static void
-updateConsoleUsersCallout(thread_call_param_t p0, thread_call_param_t p1)
-{
-	IOPMrootDomain * rootDomain = (IOPMrootDomain *) p0;
-	rootDomain->updateConsoleUsers();
-}
-
-void
-IOPMrootDomain::updateConsoleUsers(void)
-{
-	IOService::updateConsoleUsers(NULL, kIOMessageSystemHasPoweredOn);
-	updateTasksSuspend(kTasksSuspendUnsuspended, kTasksSuspendNoChange);
-}
-
-bool
-IOPMrootDomain::updateTasksSuspend(int newTasksSuspended, int newAOTTasksSuspended)
-{
-	bool newSuspend;
-
-	WAKEEVENT_LOCK();
-	if (newTasksSuspended != kTasksSuspendNoChange) {
-		tasksSuspended = (newTasksSuspended != kTasksSuspendUnsuspended);
-	}
-	if (newAOTTasksSuspended != kTasksSuspendNoChange) {
-		_aotTasksSuspended = (newAOTTasksSuspended != kTasksSuspendUnsuspended);
-	}
-	newSuspend = (tasksSuspended || _aotTasksSuspended);
-	if (newSuspend == tasksSuspendState) {
-		WAKEEVENT_UNLOCK();
-		return false;
-	}
-	tasksSuspendState = newSuspend;
-	WAKEEVENT_UNLOCK();
-	tasks_system_suspend(newSuspend);
-	return true;
-}
-
-//******************************************************************************
-
-static void
-disk_sync_callout( thread_call_param_t p0, thread_call_param_t p1 )
-{
-	IOPMrootDomain * rootDomain = (IOPMrootDomain *) p0;
-	uint32_t    notifyRef  = (uint32_t)(uintptr_t) p1;
-	uint32_t    powerState = rootDomain->getPowerState();
-
-	DLOG("disk_sync_callout ps=%u\n", powerState);
-
-	if (ON_STATE == powerState) {
-		sync_internal();
-
-#if HIBERNATION
-		// Block sleep until trim issued on previous wake path is completed.
-		IOHibernateSystemPostWake(true);
+        sync_internal();
+    }
+#if	HIBERNATION
+    else
+    {
+        IOHibernateSystemPostWake();
+    }
 #endif
-	}
-#if HIBERNATION
-	else {
-		IOHibernateSystemPostWake(false);
-
-		rootDomain->sleepWakeDebugSaveSpinDumpFile();
-	}
-#endif
-
-	rootDomain->allowPowerChange(notifyRef);
-	DLOG("disk_sync_callout finish\n");
-}
-
-//******************************************************************************
-static UInt32
-computeDeltaTimeMS( const AbsoluteTime * startTime, AbsoluteTime * elapsedTime )
-{
-	AbsoluteTime    endTime;
-	UInt64          nano = 0;
+
+    rootDomain->allowPowerChange(notifyRef);
+    DLOG("disk_sync_callout finish\n");
+}
+
+//******************************************************************************
+
+static UInt32 computeDeltaTimeMS( const AbsoluteTime * startTime )
+{
+	AbsoluteTime	endTime;
+	UInt64			nano = 0;
 
 	clock_get_uptime(&endTime);
-	if (CMP_ABSOLUTETIME(&endTime, startTime) <= 0) {
-		*elapsedTime = 0;
-	} else {
+	if (CMP_ABSOLUTETIME(&endTime, startTime) > 0)
+	{
 		SUB_ABSOLUTETIME(&endTime, startTime);
 		absolutetime_to_nanoseconds(endTime, &nano);
-		*elapsedTime = endTime;
 	}
 
-	return (UInt32)(nano / NSEC_PER_MSEC);
+	return (UInt32)(nano / 1000000ULL);
 }
 
 //******************************************************************************
@@ -1261,735 +715,443 @@
 static int
 sysctl_sleepwaketime SYSCTL_HANDLER_ARGS
 {
-	struct timeval *swt = (struct timeval *)arg1;
-	struct proc *p = req->p;
-
-	if (p == kernproc) {
-		return sysctl_io_opaque(req, swt, sizeof(*swt), NULL);
-	} else if (proc_is64bit(p)) {
-		struct user64_timeval t = {};
-		t.tv_sec = swt->tv_sec;
-		t.tv_usec = swt->tv_usec;
-		return sysctl_io_opaque(req, &t, sizeof(t), NULL);
-	} else {
-		struct user32_timeval t = {};
-		t.tv_sec = (typeof(t.tv_sec))swt->tv_sec;
-		t.tv_usec = swt->tv_usec;
-		return sysctl_io_opaque(req, &t, sizeof(t), NULL);
-	}
+  struct timeval *swt = (struct timeval *)arg1;
+  struct proc *p = req->p;
+
+  if (p == kernproc) {
+    return sysctl_io_opaque(req, swt, sizeof(*swt), NULL);    
+  } else if(proc_is64bit(p)) {
+    struct user64_timeval t;
+    t.tv_sec = swt->tv_sec;
+    t.tv_usec = swt->tv_usec;
+    return sysctl_io_opaque(req, &t, sizeof(t), NULL);
+  } else {
+    struct user32_timeval t;
+    t.tv_sec = swt->tv_sec;
+    t.tv_usec = swt->tv_usec;
+    return sysctl_io_opaque(req, &t, sizeof(t), NULL);
+  }
 }
 
 static SYSCTL_PROC(_kern, OID_AUTO, sleeptime,
-    CTLTYPE_STRUCT | CTLFLAG_RD | CTLFLAG_KERN | CTLFLAG_LOCKED,
-    &gIOLastUserSleepTime, 0, sysctl_sleepwaketime, "S,timeval", "");
+	    CTLTYPE_STRUCT | CTLFLAG_RD | CTLFLAG_NOAUTO | CTLFLAG_KERN | CTLFLAG_LOCKED,
+	    &gIOLastSleepTime, 0, sysctl_sleepwaketime, "S,timeval", "");
 
 static SYSCTL_PROC(_kern, OID_AUTO, waketime,
-    CTLTYPE_STRUCT | CTLFLAG_RD | CTLFLAG_KERN | CTLFLAG_LOCKED,
-    &gIOLastWakeTime, 0, sysctl_sleepwaketime, "S,timeval", "");
-
-SYSCTL_QUAD(_kern, OID_AUTO, wake_abs_time, CTLFLAG_RD | CTLFLAG_LOCKED, &gIOLastWakeAbsTime, "");
-SYSCTL_QUAD(_kern, OID_AUTO, sleep_abs_time, CTLFLAG_RD | CTLFLAG_LOCKED, &gIOLastSleepAbsTime, "");
-SYSCTL_QUAD(_kern, OID_AUTO, useractive_abs_time, CTLFLAG_RD | CTLFLAG_LOCKED, &gUserActiveAbsTime, "");
-SYSCTL_QUAD(_kern, OID_AUTO, userinactive_abs_time, CTLFLAG_RD | CTLFLAG_LOCKED, &gUserInactiveAbsTime, "");
+	    CTLTYPE_STRUCT | CTLFLAG_RD | CTLFLAG_NOAUTO | CTLFLAG_KERN | CTLFLAG_LOCKED,
+	    &gIOLastWakeTime, 0, sysctl_sleepwaketime, "S,timeval", "");
+
 
 static int
-sysctl_willshutdown SYSCTL_HANDLER_ARGS
-{
-	int new_value, changed, error;
-
-	if (!gWillShutdownSysctlRegistered) {
-		return ENOENT;
-	}
-
-	error = sysctl_io_number(req, gWillShutdown, sizeof(int), &new_value, &changed);
-	if (changed) {
-		if (!gWillShutdown && (new_value == 1)) {
-			IOPMRootDomainWillShutdown();
-		} else {
-			error = EINVAL;
-		}
-	}
-	return error;
+sysctl_willshutdown
+(__unused struct sysctl_oid *oidp, __unused void *arg1, __unused int arg2, struct sysctl_req *req)
+{
+    int new_value, changed;
+    int error = sysctl_io_number(req, gWillShutdown, sizeof(int), &new_value, &changed);
+    if (changed) {
+	if (!gWillShutdown && (new_value == 1)) {
+	    IOPMRootDomainWillShutdown();
+	} else
+	    error = EINVAL;
+    }
+    return(error);
 }
 
 static SYSCTL_PROC(_kern, OID_AUTO, willshutdown,
-    CTLTYPE_INT | CTLFLAG_RW | CTLFLAG_KERN | CTLFLAG_LOCKED,
-    NULL, 0, sysctl_willshutdown, "I", "");
-
-#if defined(XNU_TARGET_OS_OSX)
+	    CTLTYPE_INT | CTLFLAG_RW | CTLFLAG_NOAUTO | CTLFLAG_KERN | CTLFLAG_LOCKED,
+	    0, 0, sysctl_willshutdown, "I", "");
+
+#if !CONFIG_EMBEDDED
 
 static int
 sysctl_progressmeterenable
 (__unused struct sysctl_oid *oidp, __unused void *arg1, __unused int arg2, struct sysctl_req *req)
 {
-	int error;
-	int new_value, changed;
-
-	error = sysctl_io_number(req, vc_progressmeter_enable, sizeof(int), &new_value, &changed);
-
-	if (changed) {
-		vc_enable_progressmeter(new_value);
-	}
-
-	return error;
+    int error;
+    int new_value, changed;
+
+    error = sysctl_io_number(req, vc_progress_meter_enable, sizeof(int), &new_value, &changed);
+
+    if (changed)
+	vc_enable_progressmeter(new_value);
+
+    return (error);
 }
 
 static int
 sysctl_progressmeter
 (__unused struct sysctl_oid *oidp, __unused void *arg1, __unused int arg2, struct sysctl_req *req)
 {
-	int error;
-	int new_value, changed;
-
-	error = sysctl_io_number(req, vc_progressmeter_value, sizeof(int), &new_value, &changed);
-
-	if (changed) {
-		vc_set_progressmeter(new_value);
-	}
-
-	return error;
+    int error;
+    int new_value, changed;
+
+    error = sysctl_io_number(req, vc_progress_meter_value, sizeof(int), &new_value, &changed);
+
+    if (changed)
+	vc_set_progressmeter(new_value);
+
+    return (error);
 }
 
 static SYSCTL_PROC(_kern, OID_AUTO, progressmeterenable,
-    CTLTYPE_INT | CTLFLAG_RW | CTLFLAG_KERN | CTLFLAG_LOCKED,
-    NULL, 0, sysctl_progressmeterenable, "I", "");
+	    CTLTYPE_INT | CTLFLAG_RW | CTLFLAG_NOAUTO | CTLFLAG_KERN | CTLFLAG_LOCKED,
+	    0, 0, sysctl_progressmeterenable, "I", "");
 
 static SYSCTL_PROC(_kern, OID_AUTO, progressmeter,
-    CTLTYPE_INT | CTLFLAG_RW | CTLFLAG_KERN | CTLFLAG_LOCKED,
-    NULL, 0, sysctl_progressmeter, "I", "");
-
-#endif /* defined(XNU_TARGET_OS_OSX) */
-
-
-
-static int
-sysctl_consoleoptions
-(__unused struct sysctl_oid *oidp, __unused void *arg1, __unused int arg2, struct sysctl_req *req)
-{
-	int error, changed;
-	uint32_t new_value;
-
-	error = sysctl_io_number(req, vc_user_options.options, sizeof(uint32_t), &new_value, &changed);
-
-	if (changed) {
-		vc_user_options.options = new_value;
-	}
-
-	return error;
-}
-
-static SYSCTL_PROC(_kern, OID_AUTO, consoleoptions,
-    CTLTYPE_INT | CTLFLAG_RW | CTLFLAG_KERN | CTLFLAG_LOCKED,
-    NULL, 0, sysctl_consoleoptions, "I", "");
-
-
-static int
-sysctl_progressoptions SYSCTL_HANDLER_ARGS
-{
-	return sysctl_io_opaque(req, &vc_user_options, sizeof(vc_user_options), NULL);
-}
-
-static SYSCTL_PROC(_kern, OID_AUTO, progressoptions,
-    CTLTYPE_STRUCT | CTLFLAG_RW | CTLFLAG_KERN | CTLFLAG_LOCKED | CTLFLAG_ANYBODY,
-    NULL, 0, sysctl_progressoptions, "S,vc_progress_user_options", "");
-
-
-static int
-sysctl_wakereason SYSCTL_HANDLER_ARGS
-{
-	char wr[sizeof(gWakeReasonString)];
-
-	wr[0] = '\0';
-	if (gRootDomain && gWakeReasonSysctlRegistered) {
-		gRootDomain->copyWakeReasonString(wr, sizeof(wr));
-	} else {
-		return ENOENT;
-	}
-
-	return sysctl_io_string(req, wr, 0, 0, NULL);
-}
-
-SYSCTL_PROC(_kern, OID_AUTO, wakereason,
-    CTLTYPE_STRING | CTLFLAG_RD | CTLFLAG_KERN | CTLFLAG_LOCKED,
-    NULL, 0, sysctl_wakereason, "A", "wakereason");
-
-static int
-sysctl_bootreason SYSCTL_HANDLER_ARGS
-{
-	if (!os_atomic_load(&gBootReasonSysctlRegistered, acquire)) {
-		return ENOENT;
-	}
-
-	return sysctl_io_string(req, gBootReasonString, 0, 0, NULL);
-}
-
-SYSCTL_PROC(_kern, OID_AUTO, bootreason,
-    CTLTYPE_STRING | CTLFLAG_RD | CTLFLAG_KERN | CTLFLAG_LOCKED,
-    NULL, 0, sysctl_bootreason, "A", "");
-
-static int
-sysctl_shutdownreason SYSCTL_HANDLER_ARGS
-{
-	char sr[sizeof(gShutdownReasonString)];
-
-	sr[0] = '\0';
-	if (gRootDomain && gShutdownReasonSysctlRegistered) {
-		gRootDomain->copyShutdownReasonString(sr, sizeof(sr));
-	} else {
-		return ENOENT;
-	}
-
-	return sysctl_io_string(req, sr, 0, 0, NULL);
-}
-
-SYSCTL_PROC(_kern, OID_AUTO, shutdownreason,
-    CTLTYPE_STRING | CTLFLAG_RD | CTLFLAG_KERN | CTLFLAG_LOCKED,
-    NULL, 0, sysctl_shutdownreason, "A", "shutdownreason");
-
-// This value is meant to represent the last time the device shut down
-// in a unit of the PMU driver's choosing see rdar://138590268 for details
-static int
-sysctl_shutdowntime SYSCTL_HANDLER_ARGS
-{
-	uint64_t shutdownTime = 0;
-
-	if (gRootDomain && gShutdownReasonSysctlRegistered) {
-		gRootDomain->copyShutdownTime(&shutdownTime);
-	} else {
-		return ENOENT;
-	}
-
-	return SYSCTL_OUT(req, &shutdownTime, sizeof(shutdownTime));
-}
-
-SYSCTL_PROC(_kern, OID_AUTO, shutdowntime,
-    CTLTYPE_QUAD | CTLFLAG_RD | CTLFLAG_KERN | CTLFLAG_LOCKED,
-    NULL, 0, sysctl_shutdowntime, "Q", "shutdowntime");
-
-static int
-sysctl_targettype SYSCTL_HANDLER_ARGS
-{
-	IOService * root;
-	OSSharedPtr<OSObject>  obj;
-	OSData *    data;
-	char        tt[32];
-
-	tt[0] = '\0';
-	root = IOService::getServiceRoot();
-	if (root && (obj = root->copyProperty(gIODTTargetTypeKey))) {
-		if ((data = OSDynamicCast(OSData, obj.get()))) {
-			strlcpy(tt, (const char *) data->getBytesNoCopy(), sizeof(tt));
-		}
-	}
-	return sysctl_io_string(req, tt, 0, 0, NULL);
-}
-
-SYSCTL_PROC(_hw, OID_AUTO, targettype,
-    CTLTYPE_STRING | CTLFLAG_RD | CTLFLAG_KERN | CTLFLAG_LOCKED,
-    NULL, 0, sysctl_targettype, "A", "targettype");
-
-static SECURITY_READ_ONLY_LATE(char*) jetsam_properties_product_type_string = NULL;
-static SECURITY_READ_ONLY_LATE(size_t) jetsam_properties_product_type_string_len = 0;
-
-/*
- * SecureDTLookupEntry() is only guaranteed to work before PE_init_iokit(),
- * so we load the jetsam_properties_product_type string (if available) in a startup handler.
- */
-__startup_func
-static void
-sysctl_load_jetsam_properties_product_type(void)
-{
-	DTEntry node;
-	void const *value = NULL;
-	unsigned int size = 0;
-
-	if (kSuccess != SecureDTLookupEntry(nullptr, "/product", &node)) {
-		return;
-	}
-
-	if (kSuccess != SecureDTGetProperty(node, "jetsam-properties-product-type", (void const **) &value, &size)) {
-		return;
-	}
-
-	if (size == 0) {
-		return;
-	}
-
-	jetsam_properties_product_type_string = (char *) zalloc_permanent(size, ZALIGN_NONE);
-	if (jetsam_properties_product_type_string == NULL) {
-		return;
-	}
-
-	memcpy(jetsam_properties_product_type_string, value, size);
-	jetsam_properties_product_type_string_len = size;
-}
-STARTUP(SYSCTL, STARTUP_RANK_MIDDLE, sysctl_load_jetsam_properties_product_type);
-
-static int
-sysctl_jetsam_properties_product_type SYSCTL_HANDLER_ARGS
-{
-	if (jetsam_properties_product_type_string != NULL) {
-		return SYSCTL_OUT(req, jetsam_properties_product_type_string, jetsam_properties_product_type_string_len);
-	}
-
-	IOService * root;
-	OSSharedPtr<OSObject>  obj;
-	OSData *    data;
-	char        tt[32];
-
-	tt[0] = '\0';
-	root = IOService::getServiceRoot();
-	if (root && (obj = root->copyProperty(gIODTTargetTypeKey))) {
-		if ((data = OSDynamicCast(OSData, obj.get()))) {
-			strlcpy(tt, (const char *) data->getBytesNoCopy(), sizeof(tt));
-		}
-	}
-	return sysctl_io_string(req, tt, 0, 0, NULL);
-}
-
-SYSCTL_PROC(_hw, OID_AUTO, jetsam_properties_product_type,
-    CTLTYPE_STRING | CTLFLAG_RD | CTLFLAG_KERN | CTLFLAG_LOCKED,
-    NULL, 0, sysctl_jetsam_properties_product_type, "A", "jetsam_properties_product_type");
-
-static SYSCTL_INT(_debug, OID_AUTO, noidle, CTLFLAG_RW, &gNoIdleFlag, 0, "");
-static SYSCTL_INT(_debug, OID_AUTO, swd_sleep_timeout, CTLFLAG_RW, &gSwdSleepTimeout, 0, "");
-static SYSCTL_INT(_debug, OID_AUTO, swd_wake_timeout, CTLFLAG_RW, &gSwdWakeTimeout, 0, "");
-static SYSCTL_INT(_debug, OID_AUTO, swd_timeout, CTLFLAG_RW, &gSwdSleepWakeTimeout, 0, "");
-static SYSCTL_INT(_debug, OID_AUTO, swd_panic, CTLFLAG_RW, &gSwdPanic, 0, "");
-#if DEVELOPMENT || DEBUG
-static SYSCTL_INT(_debug, OID_AUTO, swd_panic_phase, CTLFLAG_RW, &swd_panic_phase, 0, "");
-#if defined(XNU_TARGET_OS_OSX)
-static SYSCTL_INT(_debug, OID_AUTO, clamshell, CTLFLAG_RW, &gClamshellFlags, 0, "");
+	    CTLTYPE_INT | CTLFLAG_RW | CTLFLAG_NOAUTO | CTLFLAG_KERN | CTLFLAG_LOCKED,
+	    0, 0, sysctl_progressmeter, "I", "");
+
+#endif
+
 static SYSCTL_INT(_debug, OID_AUTO, darkwake, CTLFLAG_RW, &gDarkWakeFlags, 0, "");
-#endif /* defined(XNU_TARGET_OS_OSX) */
-#endif /* DEVELOPMENT || DEBUG */
-
-//******************************************************************************
-// AOT
-
-static int
-sysctl_aotmetrics SYSCTL_HANDLER_ARGS
-{
-	if (NULL == gRootDomain) {
-		return ENOENT;
-	}
-	if (NULL == gRootDomain->_aotMetrics) {
-		IOPMAOTMetrics nullMetrics = {};
-		return sysctl_io_opaque(req, &nullMetrics, sizeof(IOPMAOTMetrics), NULL);
-	}
-	return sysctl_io_opaque(req, gRootDomain->_aotMetrics, sizeof(IOPMAOTMetrics), NULL);
-}
-
-TUNABLE_DT_WRITEABLE(uint32_t, gAOTMode, "/product/iopm",
-    "aot-mode", "aot_mode", 0, TUNABLE_DT_NONE);
-static SYSCTL_PROC(_kern, OID_AUTO, aotmetrics,
-    CTLTYPE_STRUCT | CTLFLAG_RD | CTLFLAG_KERN | CTLFLAG_LOCKED | CTLFLAG_ANYBODY,
-    NULL, 0, sysctl_aotmetrics, "S,IOPMAOTMetrics", "");
-
-
-static int
-update_aotmode(uint32_t mode)
-{
-	int result;
-
-	if (!gIOPMWorkLoop) {
-		return ENOENT;
-	}
-	result = gIOPMWorkLoop->runActionBlock(^IOReturn (void) {
-		unsigned int oldCount;
-
-		if (mode && !gRootDomain->_aotMetrics) {
-		        gRootDomain->_aotMetrics = IOMallocType(IOPMAOTMetrics);
-		}
-
-		oldCount = gRootDomain->idleSleepPreventersCount();
-		gRootDomain->_aotMode = (mode & kIOPMAOTModeMask);
-		gRootDomain->updatePreventIdleSleepListInternal(NULL, false, oldCount);
-		return 0;
-	});
-	return result;
-}
-
-static int
-sysctl_aotmodebits
-(__unused struct sysctl_oid *oidp, __unused void *arg1, __unused int arg2, struct sysctl_req *req)
-{
-	int error, changed;
-	uint32_t new_value;
-
-	if (NULL == gRootDomain) {
-		return ENOENT;
-	}
-	error = sysctl_io_number(req, gRootDomain->_aotMode, sizeof(uint32_t), &new_value, &changed);
-	if (changed && gIOPMWorkLoop) {
-		error = update_aotmode(new_value);
-	}
-
-	return error;
-}
-
-static SYSCTL_PROC(_kern, OID_AUTO, aotmodebits,
-    CTLTYPE_INT | CTLFLAG_RW | CTLFLAG_KERN | CTLFLAG_LOCKED,
-    NULL, 0, sysctl_aotmodebits, "I", "");
-
-static int
-sysctl_aotmode
-(__unused struct sysctl_oid *oidp, __unused void *arg1, __unused int arg2, struct sysctl_req *req)
-{
-	int error, changed;
-	uint32_t new_value;
-
-	if (NULL == gRootDomain) {
-		return ENOENT;
-	}
-	error = sysctl_io_number(req, gRootDomain->_aotMode, sizeof(uint32_t), &new_value, &changed);
-	if (changed && gIOPMWorkLoop) {
-		if (new_value) {
-			new_value = kIOPMAOTModeDefault; // & ~kIOPMAOTModeRespectTimers;
-		}
-		error = update_aotmode(new_value);
-	}
-
-	return error;
-}
-
-static SYSCTL_PROC(_kern, OID_AUTO, aotmode,
-    CTLTYPE_INT | CTLFLAG_RW | CTLFLAG_KERN | CTLFLAG_LOCKED | CTLFLAG_ANYBODY,
-    NULL, 0, sysctl_aotmode, "I", "");
-
-TUNABLE_DT(uint32_t, gAOTLingerTimeMS, "/product/iopm",
-    "aot-linger-time-ms", "aot_linger_time_ms", 800, TUNABLE_DT_NONE);
-
-// Low Power Wake tunables
-TUNABLE_DT_WRITEABLE(uint64_t, gLPWFlags, "/product/iopm",
-    "low-power-wake", "low_power_wake", false, TUNABLE_DT_NONE);
-static SYSCTL_QUAD(_kern, OID_AUTO, lowpowerwake, CTLFLAG_RW | CTLFLAG_LOCKED,
-    &gLPWFlags, "Low Power Wake");
-
-//******************************************************************************
-
-static OSSharedPtr<const OSSymbol> gIOPMSettingAutoWakeCalendarKey;
-static OSSharedPtr<const OSSymbol> gIOPMSettingAutoWakeSecondsKey;
-static OSSharedPtr<const OSSymbol> gIOPMSettingAutoPowerCalendarKey;
-static OSSharedPtr<const OSSymbol> gIOPMSettingAutoPowerSecondsKey;
-static OSSharedPtr<const OSSymbol> gIOPMSettingDebugWakeRelativeKey;
-static OSSharedPtr<const OSSymbol> gIOPMSettingDebugPowerRelativeKey;
-static OSSharedPtr<const OSSymbol> gIOPMSettingMaintenanceWakeCalendarKey;
-static OSSharedPtr<const OSSymbol> gIOPMSettingSleepServiceWakeCalendarKey;
-static OSSharedPtr<const OSSymbol> gIOPMSettingSilentRunningKey;
-static OSSharedPtr<const OSSymbol> gIOPMUserTriggeredFullWakeKey;
-static OSSharedPtr<const OSSymbol> gIOPMUserIsActiveKey;
-static OSSharedPtr<const OSSymbol> gIOPMSettingLowLatencyAudioModeKey;
+
+static const OSSymbol * gIOPMSettingAutoWakeSecondsKey;
+static const OSSymbol * gIOPMSettingDebugWakeRelativeKey;
+static const OSSymbol * gIOPMSettingMaintenanceWakeCalendarKey;
+static const OSSymbol * gIOPMSettingSleepServiceWakeCalendarKey;
+static const OSSymbol * gIOPMSettingSilentRunningKey;
 
 //******************************************************************************
 // start
 //
 //******************************************************************************
 
-#define kRootDomainSettingsCount           21
-#define kRootDomainNoPublishSettingsCount  4
-
-bool
-IOPMrootDomain::start( IOService * nub )
-{
-	OSSharedPtr<OSIterator>      psIterator;
-	OSSharedPtr<OSDictionary>    tmpDict;
-
-	super::start(nub);
-
-	gRootDomain = this;
-	gIOPMSettingAutoWakeCalendarKey = OSSymbol::withCString(kIOPMSettingAutoWakeCalendarKey);
-	gIOPMSettingAutoWakeSecondsKey = OSSymbol::withCString(kIOPMSettingAutoWakeSecondsKey);
-	gIOPMSettingAutoPowerCalendarKey = OSSymbol::withCString(kIOPMSettingAutoPowerCalendarKey);
-	gIOPMSettingAutoPowerSecondsKey = OSSymbol::withCString(kIOPMSettingAutoPowerSecondsKey);
-	gIOPMSettingDebugWakeRelativeKey = OSSymbol::withCString(kIOPMSettingDebugWakeRelativeKey);
-	gIOPMSettingDebugPowerRelativeKey = OSSymbol::withCString(kIOPMSettingDebugPowerRelativeKey);
-	gIOPMSettingMaintenanceWakeCalendarKey = OSSymbol::withCString(kIOPMSettingMaintenanceWakeCalendarKey);
-	gIOPMSettingSleepServiceWakeCalendarKey = OSSymbol::withCString(kIOPMSettingSleepServiceWakeCalendarKey);
-	gIOPMSettingSilentRunningKey = OSSymbol::withCStringNoCopy(kIOPMSettingSilentRunningKey);
-	gIOPMUserTriggeredFullWakeKey = OSSymbol::withCStringNoCopy(kIOPMUserTriggeredFullWakeKey);
-	gIOPMUserIsActiveKey = OSSymbol::withCStringNoCopy(kIOPMUserIsActiveKey);
-	gIOPMSettingLowLatencyAudioModeKey = OSSymbol::withCStringNoCopy(kIOPMSettingLowLatencyAudioModeKey);
-
-	gIOPMStatsResponseTimedOut = OSSymbol::withCString(kIOPMStatsResponseTimedOut);
-	gIOPMStatsResponseCancel = OSSymbol::withCString(kIOPMStatsResponseCancel);
-	gIOPMStatsResponseSlow = OSSymbol::withCString(kIOPMStatsResponseSlow);
-	gIOPMStatsResponsePrompt = OSSymbol::withCString(kIOPMStatsResponsePrompt);
-	gIOPMStatsDriverPSChangeSlow = OSSymbol::withCString(kIOPMStatsDriverPSChangeSlow);
-
-	sleepSupportedPEFunction = OSSymbol::withCString("IOPMSetSleepSupported");
-	sleepMessagePEFunction = OSSymbol::withCString("IOPMSystemSleepMessage");
-	gIOPMWakeTypeUserKey = OSSymbol::withCStringNoCopy(kIOPMRootDomainWakeTypeUser);
-
-	OSSharedPtr<const OSSymbol> settingsArr[kRootDomainSettingsCount] =
-	{
-		OSSymbol::withCString(kIOPMSettingSleepOnPowerButtonKey),
-		gIOPMSettingAutoWakeSecondsKey,
-		gIOPMSettingAutoPowerSecondsKey,
-		gIOPMSettingAutoWakeCalendarKey,
-		gIOPMSettingAutoPowerCalendarKey,
-		gIOPMSettingDebugWakeRelativeKey,
-		gIOPMSettingDebugPowerRelativeKey,
-		OSSymbol::withCString(kIOPMSettingWakeOnRingKey),
-		OSSymbol::withCString(kIOPMSettingRestartOnPowerLossKey),
-		OSSymbol::withCString(kIOPMSettingRestartOnPowerConnectKey),
-		OSSymbol::withCString(kIOPMSettingWakeOnClamshellKey),
-		OSSymbol::withCString(kIOPMSettingWakeOnACChangeKey),
-		OSSymbol::withCString(kIOPMSettingTimeZoneOffsetKey),
-		OSSymbol::withCString(kIOPMSettingDisplaySleepUsesDimKey),
-		OSSymbol::withCString(kIOPMSettingMobileMotionModuleKey),
-		OSSymbol::withCString(kIOPMSettingGraphicsSwitchKey),
-		OSSymbol::withCString(kIOPMStateConsoleShutdown),
-		OSSymbol::withCString(kIOPMSettingProModeControl),
-		OSSymbol::withCString(kIOPMSettingProModeDefer),
-		gIOPMSettingSilentRunningKey,
-		gIOPMSettingLowLatencyAudioModeKey,
-	};
-
-	OSSharedPtr<const OSSymbol> noPublishSettingsArr[kRootDomainNoPublishSettingsCount] =
-	{
-		OSSymbol::withCString(kIOPMSettingProModeControl),
-		OSSymbol::withCString(kIOPMSettingProModeDefer),
-		gIOPMSettingSilentRunningKey,
-		gIOPMSettingLowLatencyAudioModeKey,
-	};
-
-#if DEVELOPMENT || DEBUG
-#if defined(XNU_TARGET_OS_OSX)
-	PE_parse_boot_argn("darkwake", &gDarkWakeFlags, sizeof(gDarkWakeFlags));
-	PE_parse_boot_argn("clamshell", &gClamshellFlags, sizeof(gClamshellFlags));
-#endif /* defined(XNU_TARGET_OS_OSX) */
-#endif /* DEVELOPMENT || DEBUG */
-
-	PE_parse_boot_argn("noidle", &gNoIdleFlag, sizeof(gNoIdleFlag));
-	PE_parse_boot_argn("swd_sleeptimeout", &gSwdSleepTimeout, sizeof(gSwdSleepTimeout));
-	PE_parse_boot_argn("swd_waketimeout", &gSwdWakeTimeout, sizeof(gSwdWakeTimeout));
-	PE_parse_boot_argn("swd_timeout", &gSwdSleepWakeTimeout, sizeof(gSwdSleepWakeTimeout));
-	PE_parse_boot_argn("haltmspanic", &gHaltTimeMaxPanic, sizeof(gHaltTimeMaxPanic));
-	PE_parse_boot_argn("haltmslog", &gHaltTimeMaxLog, sizeof(gHaltTimeMaxLog));
-
-	_aotMode = gAOTMode;
-	_aotLingerTime = gAOTLingerTimeMS;
-	_aotMetrics = _aotMode ? IOMallocType(IOPMAOTMetrics) : NULL;
-
-	// read noidle setting from Device Tree
-	if (PE_get_default("no-idle", &gNoIdleFlag, sizeof(gNoIdleFlag))) {
-		DLOG("Setting gNoIdleFlag to %u from device tree\n", gNoIdleFlag);
-	}
-
-	queue_init(&aggressivesQueue);
-	aggressivesThreadCall = thread_call_allocate(handleAggressivesFunction, this);
-	aggressivesData = OSData::withCapacity(
-		sizeof(AggressivesRecord) * (kPMLastAggressivenessType + 4));
-
-	featuresDictLock = IOLockAlloc();
-	settingsCtrlLock = IOLockAlloc();
-	wakeEventLock = IOLockAlloc();
-	gHaltLogLock = IOLockAlloc();
-	setPMRootDomain(this);
-
-	extraSleepTimer = thread_call_allocate(
-		idleSleepTimerExpired,
-		(thread_call_param_t) this);
-
-	powerButtonDown = thread_call_allocate(
-		powerButtonDownCallout,
-		(thread_call_param_t) this);
-
-	powerButtonUp = thread_call_allocate(
-		powerButtonUpCallout,
-		(thread_call_param_t) this);
-
-	diskSyncCalloutEntry = thread_call_allocate(
-		&disk_sync_callout,
-		(thread_call_param_t) this);
-	updateConsoleUsersEntry = thread_call_allocate(
-		&updateConsoleUsersCallout,
-		(thread_call_param_t) this);
-
-#if DARK_TO_FULL_EVALUATE_CLAMSHELL_DELAY
-	fullWakeThreadCall = thread_call_allocate_with_options(
-		OSMemberFunctionCast(thread_call_func_t, this,
-		&IOPMrootDomain::fullWakeDelayedWork),
-		(thread_call_param_t) this, THREAD_CALL_PRIORITY_KERNEL,
-		THREAD_CALL_OPTIONS_ONCE);
+#define kRootDomainSettingsCount        17
+
+bool IOPMrootDomain::start( IOService * nub )
+{
+    OSIterator      *psIterator;
+    OSDictionary    *tmpDict;
+    IORootParent *   patriarch;
+
+    super::start(nub);
+
+    gRootDomain = this;
+    gIOPMSettingAutoWakeSecondsKey = OSSymbol::withCString(kIOPMSettingAutoWakeSecondsKey);
+    gIOPMSettingDebugWakeRelativeKey = OSSymbol::withCString(kIOPMSettingDebugWakeRelativeKey);
+    gIOPMSettingMaintenanceWakeCalendarKey = OSSymbol::withCString(kIOPMSettingMaintenanceWakeCalendarKey);
+    gIOPMSettingSleepServiceWakeCalendarKey = OSSymbol::withCString(kIOPMSettingSleepServiceWakeCalendarKey);
+    gIOPMSettingSilentRunningKey = OSSymbol::withCStringNoCopy(kIOPMSettingSilentRunningKey);
+
+    gIOPMStatsApplicationResponseTimedOut = OSSymbol::withCString(kIOPMStatsResponseTimedOut);
+    gIOPMStatsApplicationResponseCancel = OSSymbol::withCString(kIOPMStatsResponseCancel);
+    gIOPMStatsApplicationResponseSlow = OSSymbol::withCString(kIOPMStatsResponseSlow);
+
+    sleepSupportedPEFunction = OSSymbol::withCString("IOPMSetSleepSupported");
+    sleepMessagePEFunction = OSSymbol::withCString("IOPMSystemSleepMessage");
+
+    const OSSymbol  *settingsArr[kRootDomainSettingsCount] = 
+        {
+            OSSymbol::withCString(kIOPMSettingSleepOnPowerButtonKey),
+            gIOPMSettingAutoWakeSecondsKey,
+            OSSymbol::withCString(kIOPMSettingAutoPowerSecondsKey),
+            OSSymbol::withCString(kIOPMSettingAutoWakeCalendarKey),
+            OSSymbol::withCString(kIOPMSettingAutoPowerCalendarKey),
+            gIOPMSettingDebugWakeRelativeKey,
+            OSSymbol::withCString(kIOPMSettingDebugPowerRelativeKey),
+            OSSymbol::withCString(kIOPMSettingWakeOnRingKey),
+            OSSymbol::withCString(kIOPMSettingRestartOnPowerLossKey),
+            OSSymbol::withCString(kIOPMSettingWakeOnClamshellKey),
+            OSSymbol::withCString(kIOPMSettingWakeOnACChangeKey),
+            OSSymbol::withCString(kIOPMSettingTimeZoneOffsetKey),
+            OSSymbol::withCString(kIOPMSettingDisplaySleepUsesDimKey),
+            OSSymbol::withCString(kIOPMSettingMobileMotionModuleKey),
+            OSSymbol::withCString(kIOPMSettingGraphicsSwitchKey),
+            OSSymbol::withCString(kIOPMStateConsoleShutdown),
+            gIOPMSettingSilentRunningKey
+        };
+
+    PE_parse_boot_argn("darkwake", &gDarkWakeFlags, sizeof(gDarkWakeFlags));
+    
+    queue_init(&aggressivesQueue);
+    aggressivesThreadCall = thread_call_allocate(handleAggressivesFunction, this);
+    aggressivesData = OSData::withCapacity(
+                        sizeof(AggressivesRecord) * (kPMLastAggressivenessType + 4));
+
+    featuresDictLock = IOLockAlloc();
+    settingsCtrlLock = IOLockAlloc();
+    setPMRootDomain(this);
+    
+    extraSleepTimer = thread_call_allocate(
+                        idleSleepTimerExpired,
+                        (thread_call_param_t) this);
+
+    diskSyncCalloutEntry = thread_call_allocate(
+                        &disk_sync_callout,
+                        (thread_call_param_t) this);
+    
+    setProperty(kIOSleepSupportedKey, true);
+
+    bzero(&gPMStats, sizeof(gPMStats));
+
+    pmTracer = PMTraceWorker::tracer(this);
+
+    pmAssertions = PMAssertionsTracker::pmAssertionsTracker(this);
+
+    userDisabledAllSleep = false;
+    systemBooting = true;
+    sleepSlider = 0;
+    idleSleepTimerPending = false;
+    wrangler = NULL;
+    clamshellClosed    = false;
+    clamshellExists    = false;
+    clamshellDisabled  = true;
+    acAdaptorConnected = true;
+
+    // Set the default system capabilities at boot.
+    _currentCapability = kIOPMSystemCapabilityCPU      |
+                         kIOPMSystemCapabilityGraphics |
+                         kIOPMSystemCapabilityAudio    |
+                         kIOPMSystemCapabilityNetwork;
+
+    _pendingCapability = _currentCapability;
+    _desiredCapability = _currentCapability;
+    _highestCapability = _currentCapability;
+    setProperty(kIOPMSystemCapabilitiesKey, _currentCapability, 64);
+
+    queuedSleepWakeUUIDString = NULL;
+    pmStatsAppResponses     = OSArray::withCapacity(5);
+    _statsNameKey           = OSSymbol::withCString(kIOPMStatsNameKey);
+    _statsPIDKey            = OSSymbol::withCString(kIOPMStatsPIDKey);
+    _statsTimeMSKey         = OSSymbol::withCString(kIOPMStatsTimeMSKey);
+    _statsResponseTypeKey   = OSSymbol::withCString(kIOPMStatsApplicationResponseTypeKey);
+    _statsMessageTypeKey    = OSSymbol::withCString(kIOPMStatsMessageTypeKey);
+
+    idxPMCPUClamshell = kCPUUnknownIndex;
+    idxPMCPULimitedPower = kCPUUnknownIndex;
+        
+    tmpDict = OSDictionary::withCapacity(1);
+    setProperty(kRootDomainSupportedFeatures, tmpDict);
+    tmpDict->release();
+    
+    settingsCallbacks = OSDictionary::withCapacity(1);
+
+    // Create a list of the valid PM settings that we'll relay to
+    // interested clients in setProperties() => setPMSetting()
+    allowedPMSettings = OSArray::withObjects(
+                    (const OSObject **)settingsArr,
+                    kRootDomainSettingsCount,
+                    0);
+
+    // List of PM settings that should not automatically publish itself
+    // as a feature when registered by a listener.
+    noPublishPMSettings = OSArray::withObjects(
+                    (const OSObject **) &gIOPMSettingSilentRunningKey, 1, 0);
+
+    fPMSettingsDict = OSDictionary::withCapacity(5);
+    preventIdleSleepList = OSSet::withCapacity(8);
+    preventSystemSleepList = OSSet::withCapacity(2);
+
+    PMinit();   // creates gIOPMWorkLoop
+
+    // Create IOPMPowerStateQueue used to queue external power
+    // events, and to handle those events on the PM work loop.
+    pmPowerStateQueue = IOPMPowerStateQueue::PMPowerStateQueue(
+        this, OSMemberFunctionCast(IOEventSource::Action, this,
+                &IOPMrootDomain::dispatchPowerEvent));
+    getPMworkloop()->addEventSource(pmPowerStateQueue);
+#ifdef CHECK_THREAD_CONTEXT
+    gIOPMWorkLoop = getPMworkloop();
 #endif
 
-	setProperty(kIOSleepSupportedKey, true);
-
-	bzero(&gPMStats, sizeof(gPMStats));
-
-	pmTracer = PMTraceWorker::tracer(this);
-
-	pmAssertions = PMAssertionsTracker::pmAssertionsTracker(this);
-
-	userDisabledAllSleep = false;
-	systemBooting = true;
-	idleSleepEnabled = false;
-	idleSleepRevertible = true;
-	sleepSlider = 0;
-	idleSleepTimerPending = false;
-	wrangler = NULL;
-	clamshellClosed = false;
-	clamshellExists = false;
-#if DISPLAY_WRANGLER_PRESENT
-	clamshellDisabled = true;
-#else
-	clamshellDisabled = false;
+    // create our power parent
+    patriarch = new IORootParent;
+    patriarch->init();
+    patriarch->attach(this);
+    patriarch->start(this);
+    patriarch->addPowerChild(this);
+
+    registerPowerDriver(this, ourPowerStates, NUM_POWER_STATES);
+    changePowerStateToPriv(ON_STATE);
+
+    if (gIOKitDebug & (kIOLogDriverPower1 | kIOLogDriverPower2))
+    {
+        // Setup our PM logging & recording code
+        timeline = IOPMTimeline::timeline(this);    
+        if (timeline) {
+            OSDictionary *tlInfo = timeline->copyInfoDictionary();
+            
+            if (tlInfo) 
+            {
+                setProperty(kIOPMTimelineDictionaryKey, tlInfo);
+                tlInfo->release();
+            }
+        }
+    }
+
+    // install power change handler
+    gSysPowerDownNotifier = registerPrioritySleepWakeInterest( &sysPowerDownHandler, this, 0);
+
+#if !NO_KERNEL_HID
+    // Register for a notification when IODisplayWrangler is published
+    if ((tmpDict = serviceMatching("IODisplayWrangler")))
+    {
+        _displayWranglerNotifier = addMatchingNotification( 
+                gIOPublishNotification, tmpDict, 
+                (IOServiceMatchingNotificationHandler) &displayWranglerMatchPublished,
+                this, 0);
+        tmpDict->release();
+    }
 #endif
-	clamshellIgnoreClose = false;
-	acAdaptorConnected = true;
-	clamshellSleepDisableMask = 0;
-	gWakeReasonString[0] = '\0';
-
-	// Initialize to user active.
-	// Will never transition to user inactive w/o wrangler.
-	fullWakeReason = kFullWakeReasonLocalUser;
-	userIsActive = userWasActive = true;
-	clock_get_uptime(&gUserActiveAbsTime);
-	setProperty(gIOPMUserIsActiveKey.get(), kOSBooleanTrue);
-
-	// Set the default system capabilities at boot.
-	_currentCapability = kIOPMSystemCapabilityCPU      |
-	    kIOPMSystemCapabilityGraphics |
-	    kIOPMSystemCapabilityAudio    |
-	    kIOPMSystemCapabilityNetwork;
-
-	_pendingCapability = _currentCapability;
-	_desiredCapability = _currentCapability;
-	_highestCapability = _currentCapability;
-	setProperty(kIOPMSystemCapabilitiesKey, _currentCapability, 64);
-
-	queuedSleepWakeUUIDString = NULL;
-	initializeBootSessionUUID();
-	pmStatsAppResponses     = OSArray::withCapacity(5);
-	_statsNameKey           = OSSymbol::withCString(kIOPMStatsNameKey);
-	_statsPIDKey            = OSSymbol::withCString(kIOPMStatsPIDKey);
-	_statsTimeMSKey         = OSSymbol::withCString(kIOPMStatsTimeMSKey);
-	_statsResponseTypeKey   = OSSymbol::withCString(kIOPMStatsApplicationResponseTypeKey);
-	_statsMessageTypeKey    = OSSymbol::withCString(kIOPMStatsMessageTypeKey);
-	_statsPowerCapsKey      = OSSymbol::withCString(kIOPMStatsPowerCapabilityKey);
-	assertOnWakeSecs        = -1;// Invalid value to prevent updates
-
-	pmStatsLock = IOLockAlloc();
-	idxPMCPUClamshell = kCPUUnknownIndex;
-	idxPMCPULimitedPower = kCPUUnknownIndex;
-
-	tmpDict = OSDictionary::withCapacity(1);
-	setProperty(kRootDomainSupportedFeatures, tmpDict.get());
-
-	// Set a default "SystemPowerProfileOverrideDict" for platform
-	// drivers without any overrides.
-	if (!propertyExists(kIOPMSystemDefaultOverrideKey)) {
-		tmpDict = OSDictionary::withCapacity(1);
-		setProperty(kIOPMSystemDefaultOverrideKey, tmpDict.get());
-	}
-
-	settingsCallbacks = OSDictionary::withCapacity(1);
-
-	// Create a list of the valid PM settings that we'll relay to
-	// interested clients in setProperties() => setPMSetting()
-	allowedPMSettings = OSArray::withObjects(
-		(const OSObject **)settingsArr,
-		kRootDomainSettingsCount,
-		0);
-
-	// List of PM settings that should not automatically publish itself
-	// as a feature when registered by a listener.
-	noPublishPMSettings = OSArray::withObjects(
-		(const OSObject **)noPublishSettingsArr,
-		kRootDomainNoPublishSettingsCount,
-		0);
-
-	fPMSettingsDict = OSDictionary::withCapacity(5);
-	preventIdleSleepList = OSSet::withCapacity(8);
-	preventSystemSleepList = OSSet::withCapacity(2);
-
-	PMinit(); // creates gIOPMWorkLoop
-	gIOPMWorkLoop = getIOPMWorkloop();
-
-	commandGate = IOCommandGate::commandGate(gIOPMWorkLoop);
-	gIOPMWorkLoop->addEventSource(commandGate.get());
-
-	// Create IOPMPowerStateQueue used to queue external power
-	// events, and to handle those events on the PM work loop.
-	pmPowerStateQueue = IOPMPowerStateQueue::PMPowerStateQueue(
-		this, OSMemberFunctionCast(IOEventSource::Action, this,
-		&IOPMrootDomain::dispatchPowerEvent));
-	gIOPMWorkLoop->addEventSource(pmPowerStateQueue);
-
-	_aotTimerES = IOTimerEventSource::timerEventSource(this,
-	    OSMemberFunctionCast(IOTimerEventSource::Action,
-	    this, &IOPMrootDomain::aotEvaluate));
-	gIOPMWorkLoop->addEventSource(_aotTimerES.get());
-
-	// Avoid publishing service early so gIOPMWorkLoop is
-	// guaranteed to be initialized by rootDomain.
-	publishPMRootDomain();
-
-	// create our power parent
-	gPatriarch = new IORootParent;
-	gPatriarch->init();
-	gPatriarch->attach(this);
-	gPatriarch->start(this);
-	gPatriarch->addPowerChild(this);
-
-	registerPowerDriver(this, ourPowerStates, NUM_POWER_STATES);
-	changePowerStateWithTagToPriv(ON_STATE, kCPSReasonInit);
-
-	// install power change handler
-	gSysPowerDownNotifier = registerPrioritySleepWakeInterest( &sysPowerDownHandler, this, NULL);
-
-#if DISPLAY_WRANGLER_PRESENT
-	wranglerIdleSettings = OSDictionary::withCapacity(1);
-	OSSharedPtr<OSNumber> wranglerIdlePeriod = OSNumber::withNumber(kDefaultWranglerIdlePeriod, 32);
-
-	if (wranglerIdleSettings && wranglerIdlePeriod) {
-		wranglerIdleSettings->setObject(kIORequestWranglerIdleKey,
-		    wranglerIdlePeriod.get());
-	}
-
-#endif /* DISPLAY_WRANGLER_PRESENT */
-
-	lowLatencyAudioNotifierDict       = OSDictionary::withCapacity(2);
-	lowLatencyAudioNotifyStateSym     = OSSymbol::withCString("LowLatencyAudioNotifyState");
-	lowLatencyAudioNotifyTimestampSym = OSSymbol::withCString("LowLatencyAudioNotifyTimestamp");
-	lowLatencyAudioNotifyStateVal     = OSNumber::withNumber(0ull, 32);
-	lowLatencyAudioNotifyTimestampVal = OSNumber::withNumber(0ull, 64);
-
-	if (lowLatencyAudioNotifierDict && lowLatencyAudioNotifyStateSym && lowLatencyAudioNotifyTimestampSym &&
-	    lowLatencyAudioNotifyStateVal && lowLatencyAudioNotifyTimestampVal) {
-		lowLatencyAudioNotifierDict->setObject(lowLatencyAudioNotifyStateSym.get(), lowLatencyAudioNotifyStateVal.get());
-		lowLatencyAudioNotifierDict->setObject(lowLatencyAudioNotifyTimestampSym.get(), lowLatencyAudioNotifyTimestampVal.get());
-	}
-
-	OSSharedPtr<const OSSymbol> ucClassName = OSSymbol::withCStringNoCopy("RootDomainUserClient");
-	setProperty(gIOUserClientClassKey, const_cast<OSObject *>(static_cast<const OSObject *>(ucClassName.get())));
-
-	// IOBacklightDisplay can take a long time to load at boot, or it may
-	// not load at all if you're booting with clamshell closed. We publish
-	// 'DisplayDims' here redundantly to get it published early and at all.
-	OSSharedPtr<OSDictionary> matching;
-	matching = serviceMatching("IOPMPowerSource");
-	psIterator = getMatchingServices(matching.get());
-
-	if (psIterator && psIterator->getNextObject()) {
-		// There's at least one battery on the system, so we publish
-		// 'DisplayDims' support for the LCD.
-		publishFeature("DisplayDims");
-	}
-
-	// read swd_panic boot-arg
-	PE_parse_boot_argn("swd_panic", &gSwdPanic, sizeof(gSwdPanic));
-	gWillShutdownSysctlRegistered = true;
-
-#if HIBERNATION
-	IOHibernateSystemInit(this);
+
+    const OSSymbol *ucClassName = OSSymbol::withCStringNoCopy("RootDomainUserClient");
+    setProperty(gIOUserClientClassKey, (OSObject *) ucClassName);
+    ucClassName->release();
+
+    // IOBacklightDisplay can take a long time to load at boot, or it may
+    // not load at all if you're booting with clamshell closed. We publish
+    // 'DisplayDims' here redundantly to get it published early and at all.
+    psIterator = getMatchingServices( serviceMatching("IOPMPowerSource") );
+    if( psIterator && psIterator->getNextObject() )
+    {
+        // There's at least one battery on the system, so we publish
+        // 'DisplayDims' support for the LCD.
+        publishFeature("DisplayDims");
+    }
+    if(psIterator) {
+        psIterator->release();        
+    }
+    
+    
+    pmSuspendedCapacity = pmSuspendedSize = 0;
+    pmSuspendedPIDS = NULL;
+    
+
+    sysctl_register_oid(&sysctl__kern_sleeptime);
+    sysctl_register_oid(&sysctl__kern_waketime);
+    sysctl_register_oid(&sysctl__kern_willshutdown);
+#if !CONFIG_EMBEDDED
+    sysctl_register_oid(&sysctl__kern_progressmeterenable);
+    sysctl_register_oid(&sysctl__kern_progressmeter);
+#endif /* !CONFIG_EMBEDDED */
+
+#if	HIBERNATION
+    IOHibernateSystemInit(this);
 #endif
 
-	registerService();                  // let clients find us
-
-	return true;
+    registerService();						// let clients find us
+
+    return true;
+}
+
+
+
+
+void IOPMrootDomain::handleSuspendPMNotificationClient(uint32_t pid, bool doSuspend)
+{
+    ASSERT_GATED();
+    
+    int index = -1;
+    unsigned int i;
+    
+    if (!pmSuspendedPIDS) {
+        pmSuspendedCapacity = 8;
+        pmSuspendedSize = pmSuspendedCapacity * sizeof(PMNotifySuspendedStruct);
+        pmSuspendedPIDS = (PMNotifySuspendedStruct *)IOMalloc(pmSuspendedSize);
+        bzero(pmSuspendedPIDS, pmSuspendedSize);
+    }
+    
+    /* Find the existing pid in the existing array */
+
+    for (i=0; i < pmSuspendedCapacity; i++) {
+        if (pmSuspendedPIDS[i].pid == pid) {
+            index = i;
+            break;
+        }
+    }
+    
+    if (-1 == index)
+    {
+        /* Find an unused slot in the suspended pids table. */
+
+        for (i=0; i < pmSuspendedCapacity; i++) {
+            if (pmSuspendedPIDS[i].refcount == 0) {
+                break;
+            }
+        }
+    
+        if (pmSuspendedCapacity == i) 
+        {
+            /* GROW if necessary */
+
+            PMNotifySuspendedStruct *newSuspended = NULL;
+            pmSuspendedCapacity     *= 2;
+            pmSuspendedSize         = pmSuspendedCapacity * sizeof(PMNotifySuspendedStruct);
+            newSuspended            = (PMNotifySuspendedStruct *)IOMalloc(pmSuspendedSize);
+
+            bzero(newSuspended, pmSuspendedSize);
+            bcopy(pmSuspendedPIDS,  newSuspended, pmSuspendedSize/2);
+            IOFree(pmSuspendedPIDS, pmSuspendedSize/2);
+        
+            pmSuspendedPIDS = newSuspended;
+        }
+
+        index = i;
+        pmSuspendedPIDS[index].pid = pid;
+    }
+
+    if (doSuspend) {
+        pmSuspendedPIDS[index].refcount++;
+    } else {
+        pmSuspendedPIDS[index].refcount--;
+    }
+        
+    /*
+     * Publish array of suspended pids in IOPMrootDomain
+     */
+    OSArray     *publish = OSArray::withCapacity(pmSuspendedCapacity);
+
+    for (i=0; i<pmSuspendedCapacity; i++)
+    {
+        if (pmSuspendedPIDS[i].refcount > 0) {
+            OSDictionary    *suspended = OSDictionary::withCapacity(2);
+            OSNumber        *n = NULL;
+            
+            n = OSNumber::withNumber(pmSuspendedPIDS[i].pid, 32);
+            suspended->setObject("pid", n);
+            n->release();
+            
+            n = OSNumber::withNumber(pmSuspendedPIDS[i].refcount, 32);
+            suspended->setObject("refcount", n);
+            n->release();
+            
+            publish->setObject(suspended);
+            suspended->release();
+            
+        }
+    }
+    
+    if (0 != publish->getCount()) {
+        setProperty(kPMSuspendedNotificationClients, publish);
+    } else {
+        removeProperty(kPMSuspendedNotificationClients);
+    }
+    
+    publish->release();
+    
+    return;
+}
+
+bool IOPMrootDomain::pmNotificationIsSuspended(uint32_t pid)
+{
+    unsigned int index;
+    
+    for (index=0; index < pmSuspendedCapacity; index++) {
+        if (pmSuspendedPIDS[index].pid == pid) {
+            return pmSuspendedPIDS[index].refcount > 0;
+        }
+    }
+    
+    return false;
+}
+
+
+void IOPMrootDomain::suspendPMNotificationsForPID(uint32_t pid, bool doSuspend)
+{
+    if(pmPowerStateQueue) {
+        pmPowerStateQueue->submitPowerEvent(kPowerEventSuspendClient, (void *)pid, (uint64_t)doSuspend );
+    }
+    return;
 }
 
 //******************************************************************************
@@ -1999,294 +1161,217 @@
 // The "System Boot" property means the system is completely booted.
 //******************************************************************************
 
-IOReturn
-IOPMrootDomain::setProperties( OSObject * props_obj )
-{
-	IOReturn        return_value = kIOReturnSuccess;
-	OSDictionary    *dict = OSDynamicCast(OSDictionary, props_obj);
-	OSBoolean       *b = NULL;
-	OSNumber        *n = NULL;
-	const OSSymbol  *key = NULL;
-	OSObject        *obj = NULL;
-	OSSharedPtr<OSCollectionIterator> iter;
-
-	if (!dict) {
-		return kIOReturnBadArgument;
-	}
-
-	bool clientEntitled = false;
-	{
-		OSSharedPtr<OSObject> obj = IOUserClient::copyClientEntitlement(current_task(), kRootDomainEntitlementSetProperty);
-		clientEntitled = (obj == kOSBooleanTrue);
-	}
-
-	if (!clientEntitled) {
-		const char * errorSuffix = NULL;
-
-		// IOPMSchedulePowerEvent() clients may not be entitled, but must be root.
-		// That API can set 6 possible keys that are checked below.
-		if ((dict->getCount() == 1) &&
-		    (dict->getObject(gIOPMSettingAutoWakeSecondsKey.get()) ||
-		    dict->getObject(gIOPMSettingAutoPowerSecondsKey.get()) ||
-		    dict->getObject(gIOPMSettingAutoWakeCalendarKey.get()) ||
-		    dict->getObject(gIOPMSettingAutoPowerCalendarKey.get()) ||
-		    dict->getObject(gIOPMSettingDebugWakeRelativeKey.get()) ||
-		    dict->getObject(gIOPMSettingDebugPowerRelativeKey.get()))) {
-			return_value = IOUserClient::clientHasPrivilege(current_task(), kIOClientPrivilegeAdministrator);
-			if (return_value != kIOReturnSuccess) {
-				errorSuffix = "privileged";
-			}
-		} else {
-			return_value = kIOReturnNotPermitted;
-			errorSuffix = "entitled";
-		}
-
-		if (return_value != kIOReturnSuccess) {
-			OSSharedPtr<OSString> procName(IOCopyLogNameForPID(proc_selfpid()), OSNoRetain);
-			DLOG("%s failed, process %s is not %s\n", __func__,
-			    procName ? procName->getCStringNoCopy() : "", errorSuffix);
-			return return_value;
-		}
-	}
-
-	OSSharedPtr<const OSSymbol> publish_simulated_battery_string    = OSSymbol::withCString("SoftwareSimulatedBatteries");
-	OSSharedPtr<const OSSymbol> boot_complete_string                = OSSymbol::withCString("System Boot Complete");
-	OSSharedPtr<const OSSymbol> sys_shutdown_string                 = OSSymbol::withCString("System Shutdown");
-	OSSharedPtr<const OSSymbol> stall_halt_string                   = OSSymbol::withCString("StallSystemAtHalt");
-	OSSharedPtr<const OSSymbol> battery_warning_disabled_string     = OSSymbol::withCString("BatteryWarningsDisabled");
-	OSSharedPtr<const OSSymbol> idle_seconds_string                 = OSSymbol::withCString("System Idle Seconds");
-	OSSharedPtr<const OSSymbol> idle_milliseconds_string            = OSSymbol::withCString("System Idle Milliseconds");
-	OSSharedPtr<const OSSymbol> sleepdisabled_string                = OSSymbol::withCString("SleepDisabled");
-	OSSharedPtr<const OSSymbol> ondeck_sleepwake_uuid_string        = OSSymbol::withCString(kIOPMSleepWakeUUIDKey);
-	OSSharedPtr<const OSSymbol> loginwindow_progress_string         = OSSymbol::withCString(kIOPMLoginWindowProgressKey);
-	OSSharedPtr<const OSSymbol> coredisplay_progress_string         = OSSymbol::withCString(kIOPMCoreDisplayProgressKey);
-	OSSharedPtr<const OSSymbol> coregraphics_progress_string        = OSSymbol::withCString(kIOPMCoreGraphicsProgressKey);
-#if DEBUG || DEVELOPMENT
-	OSSharedPtr<const OSSymbol> clamshell_close_string              = OSSymbol::withCString("IOPMTestClamshellClose");
-	OSSharedPtr<const OSSymbol> clamshell_open_string               = OSSymbol::withCString("IOPMTestClamshellOpen");
-	OSSharedPtr<const OSSymbol> ac_detach_string                    = OSSymbol::withCString("IOPMTestACDetach");
-	OSSharedPtr<const OSSymbol> ac_attach_string                    = OSSymbol::withCString("IOPMTestACAttach");
-	OSSharedPtr<const OSSymbol> desktopmode_set_string              = OSSymbol::withCString("IOPMTestDesktopModeSet");
-	OSSharedPtr<const OSSymbol> desktopmode_remove_string           = OSSymbol::withCString("IOPMTestDesktopModeRemove");
+IOReturn IOPMrootDomain::setProperties( OSObject * props_obj )
+{
+    IOReturn        return_value = kIOReturnSuccess;
+    OSDictionary    *dict = OSDynamicCast(OSDictionary, props_obj);
+    OSBoolean       *b;
+    OSNumber        *n;
+    OSDictionary    *d;
+    OSSymbol        *type;
+    OSObject        *obj;
+    unsigned int    i;
+
+    const OSSymbol *publish_simulated_battery_string    = OSSymbol::withCString("SoftwareSimulatedBatteries");
+    const OSSymbol *boot_complete_string                = OSSymbol::withCString("System Boot Complete");
+    const OSSymbol *sys_shutdown_string                 = OSSymbol::withCString("System Shutdown");
+    const OSSymbol *stall_halt_string                   = OSSymbol::withCString("StallSystemAtHalt");
+    const OSSymbol *battery_warning_disabled_string     = OSSymbol::withCString("BatteryWarningsDisabled");
+    const OSSymbol *idle_seconds_string                 = OSSymbol::withCString("System Idle Seconds");
+    const OSSymbol *sleepdisabled_string                = OSSymbol::withCString("SleepDisabled");
+    const OSSymbol *ondeck_sleepwake_uuid_string        = OSSymbol::withCString(kIOPMSleepWakeUUIDKey);
+    const OSSymbol *loginwindow_tracepoint_string       = OSSymbol::withCString(kIOPMLoginWindowSecurityDebugKey);
+    const OSSymbol *pmTimelineLogging_string            = OSSymbol::withCString(kIOPMTimelineDictionaryKey);
+#if	HIBERNATION
+    const OSSymbol *hibernatemode_string                = OSSymbol::withCString(kIOHibernateModeKey);
+    const OSSymbol *hibernatefile_string                = OSSymbol::withCString(kIOHibernateFileKey);
+    const OSSymbol *hibernatefreeratio_string           = OSSymbol::withCString(kIOHibernateFreeRatioKey);
+    const OSSymbol *hibernatefreetime_string            = OSSymbol::withCString(kIOHibernateFreeTimeKey);
 #endif
-
-#if HIBERNATION
-	OSSharedPtr<const OSSymbol> hibernatemode_string                = OSSymbol::withCString(kIOHibernateModeKey);
-	OSSharedPtr<const OSSymbol> hibernatefile_string                = OSSymbol::withCString(kIOHibernateFileKey);
-	OSSharedPtr<const OSSymbol> hibernatefilemin_string             = OSSymbol::withCString(kIOHibernateFileMinSizeKey);
-	OSSharedPtr<const OSSymbol> hibernatefilemax_string             = OSSymbol::withCString(kIOHibernateFileMaxSizeKey);
-	OSSharedPtr<const OSSymbol> hibernatefreeratio_string           = OSSymbol::withCString(kIOHibernateFreeRatioKey);
-	OSSharedPtr<const OSSymbol> hibernatefreetime_string            = OSSymbol::withCString(kIOHibernateFreeTimeKey);
+#if SUSPEND_PM_NOTIFICATIONS_DEBUG
+    const OSSymbol *suspendPMClient_string              = OSSymbol::withCString(kPMSuspendedNotificationClients);
 #endif
-
-	iter = OSCollectionIterator::withCollection(dict);
-	if (!iter) {
-		return_value = kIOReturnNoMemory;
-		goto exit;
-	}
-
-	while ((key = (const OSSymbol *) iter->getNextObject()) &&
-	    (obj = dict->getObject(key))) {
-		if (key->isEqualTo(publish_simulated_battery_string.get())) {
-			if (OSDynamicCast(OSBoolean, obj)) {
-				publishResource(key, kOSBooleanTrue);
-			}
-		} else if (key->isEqualTo(idle_seconds_string.get())) {
-			if ((n = OSDynamicCast(OSNumber, obj))) {
-				setProperty(key, n);
-				idleMilliSeconds = n->unsigned32BitValue() * 1000;
-			}
-		} else if (key->isEqualTo(idle_milliseconds_string.get())) {
-			if ((n = OSDynamicCast(OSNumber, obj))) {
-				setProperty(key, n);
-				idleMilliSeconds = n->unsigned32BitValue();
-			}
-		} else if (key->isEqualTo(boot_complete_string.get())) {
-			pmPowerStateQueue->submitPowerEvent(kPowerEventSystemBootCompleted);
-		} else if (key->isEqualTo(sys_shutdown_string.get())) {
-			if ((b = OSDynamicCast(OSBoolean, obj))) {
-				pmPowerStateQueue->submitPowerEvent(kPowerEventSystemShutdown, (void *) b);
-			}
-		} else if (key->isEqualTo(battery_warning_disabled_string.get())) {
-			setProperty(key, obj);
-		}
-#if HIBERNATION
-		else if (key->isEqualTo(hibernatemode_string.get()) ||
-		    key->isEqualTo(hibernatefilemin_string.get()) ||
-		    key->isEqualTo(hibernatefilemax_string.get()) ||
-		    key->isEqualTo(hibernatefreeratio_string.get()) ||
-		    key->isEqualTo(hibernatefreetime_string.get())) {
-			if ((n = OSDynamicCast(OSNumber, obj))) {
-				setProperty(key, n);
-			}
-		} else if (key->isEqualTo(hibernatefile_string.get())) {
-			OSString * str = OSDynamicCast(OSString, obj);
-			if (str) {
-				setProperty(key, str);
-			}
-		}
+    
+    if (!dict) 
+    {
+        return_value = kIOReturnBadArgument;
+        goto exit;
+    }
+    
+    if ((b = OSDynamicCast(OSBoolean, dict->getObject(publish_simulated_battery_string))))
+    {
+        publishResource(publish_simulated_battery_string, kOSBooleanTrue);
+    }
+
+    if ((n = OSDynamicCast(OSNumber, dict->getObject(idle_seconds_string))))
+    {
+        setProperty(idle_seconds_string, n);
+        idleSeconds = n->unsigned32BitValue();
+    }
+
+    if (boot_complete_string && dict->getObject(boot_complete_string)) 
+    {
+        pmPowerStateQueue->submitPowerEvent( kPowerEventSystemBootCompleted );
+    }
+    
+    if( battery_warning_disabled_string && dict->getObject(battery_warning_disabled_string))
+    {
+        setProperty( battery_warning_disabled_string, dict->getObject(battery_warning_disabled_string));
+    }
+    
+    if (pmTimelineLogging_string && (d = OSDynamicCast(OSDictionary, dict->getObject(pmTimelineLogging_string))))
+    {
+        if (timeline && timeline->setProperties(d)) 
+        {
+            OSDictionary *tlInfo = timeline->copyInfoDictionary();            
+            if (tlInfo) {
+                setProperty(kIOPMTimelineDictionaryKey, tlInfo);
+                tlInfo->release();
+            }
+        }
+    }
+
+    if( sys_shutdown_string && (b = OSDynamicCast(OSBoolean, dict->getObject(sys_shutdown_string)))) 
+    {
+        pmPowerStateQueue->submitPowerEvent(kPowerEventSystemShutdown, (void *) b);
+    }
+    
+    if( stall_halt_string && (b = OSDynamicCast(OSBoolean, dict->getObject(stall_halt_string))) ) 
+    {
+        setProperty(stall_halt_string, b);
+    }
+
+#if	HIBERNATION
+    if ( hibernatemode_string
+        && (n = OSDynamicCast(OSNumber, dict->getObject(hibernatemode_string))))
+    {
+    	setProperty(hibernatemode_string, n);
+    }
+    if ( hibernatefreeratio_string
+        && (n = OSDynamicCast(OSNumber, dict->getObject(hibernatefreeratio_string))))
+    {
+        setProperty(hibernatefreeratio_string, n);
+    }
+    if ( hibernatefreetime_string
+        && (n = OSDynamicCast(OSNumber, dict->getObject(hibernatefreetime_string))))
+    {
+        setProperty(hibernatefreetime_string, n);
+    }    
+    OSString *str;
+    if ( hibernatefile_string
+        && (str = OSDynamicCast(OSString, dict->getObject(hibernatefile_string))))
+    {
+        setProperty(hibernatefile_string, str);
+    }
 #endif
-		else if (key->isEqualTo(sleepdisabled_string.get())) {
-			if ((b = OSDynamicCast(OSBoolean, obj))) {
-				setProperty(key, b);
-				pmPowerStateQueue->submitPowerEvent(kPowerEventUserDisabledSleep, (void *) b);
-			}
-		} else if (key->isEqualTo(ondeck_sleepwake_uuid_string.get())) {
-			obj->retain();
-			pmPowerStateQueue->submitPowerEvent(kPowerEventQueueSleepWakeUUID, (void *)obj);
-		} else if (key->isEqualTo(loginwindow_progress_string.get())) {
-			if (pmTracer && (n = OSDynamicCast(OSNumber, obj))) {
-				uint32_t data = n->unsigned32BitValue();
-				pmTracer->traceComponentWakeProgress(kIOPMLoginWindowProgress, data);
-				kdebugTrace(kPMLogComponentWakeProgress, 0, kIOPMLoginWindowProgress, data);
-			}
-		} else if (key->isEqualTo(coredisplay_progress_string.get())) {
-			if (pmTracer && (n = OSDynamicCast(OSNumber, obj))) {
-				uint32_t data = n->unsigned32BitValue();
-				pmTracer->traceComponentWakeProgress(kIOPMCoreDisplayProgress, data);
-				kdebugTrace(kPMLogComponentWakeProgress, 0, kIOPMCoreDisplayProgress, data);
-			}
-		} else if (key->isEqualTo(coregraphics_progress_string.get())) {
-			if (pmTracer && (n = OSDynamicCast(OSNumber, obj))) {
-				uint32_t data = n->unsigned32BitValue();
-				pmTracer->traceComponentWakeProgress(kIOPMCoreGraphicsProgress, data);
-				kdebugTrace(kPMLogComponentWakeProgress, 0, kIOPMCoreGraphicsProgress, data);
-			}
-		} else if (key->isEqualTo(kIOPMDeepSleepEnabledKey) ||
-		    key->isEqualTo(kIOPMDestroyFVKeyOnStandbyKey) ||
-		    key->isEqualTo(kIOPMAutoPowerOffEnabledKey) ||
-		    key->isEqualTo(stall_halt_string.get())) {
-			if ((b = OSDynamicCast(OSBoolean, obj))) {
-				setProperty(key, b);
-			}
-		} else if (key->isEqualTo(kIOPMDeepSleepDelayKey) ||
-		    key->isEqualTo(kIOPMDeepSleepTimerKey) ||
-		    key->isEqualTo(kIOPMAutoPowerOffDelayKey) ||
-		    key->isEqualTo(kIOPMAutoPowerOffTimerKey)) {
-			if ((n = OSDynamicCast(OSNumber, obj))) {
-				setProperty(key, n);
-			}
-		} else if (key->isEqualTo(kIOPMUserWakeAlarmScheduledKey)) {
-			if (kOSBooleanTrue == obj) {
-				OSBitOrAtomic(kIOPMAlarmBitCalendarWake, &_userScheduledAlarmMask);
-			} else {
-				OSBitAndAtomic(~kIOPMAlarmBitCalendarWake, &_userScheduledAlarmMask);
-			}
-			DLOG("_userScheduledAlarmMask 0x%x\n", (uint32_t) _userScheduledAlarmMask);
-		}
-#if DEBUG || DEVELOPMENT
-		else if (key->isEqualTo(clamshell_close_string.get())) {
-			DLOG("SetProperties: setting clamshell close\n");
-			UInt32 msg = kIOPMClamshellClosed;
-			pmPowerStateQueue->submitPowerEvent(kPowerEventReceivedPowerNotification, (void *)(uintptr_t)msg);
-		} else if (key->isEqualTo(clamshell_open_string.get())) {
-			DLOG("SetProperties: setting clamshell open\n");
-			UInt32 msg = kIOPMClamshellOpened;
-			pmPowerStateQueue->submitPowerEvent(kPowerEventReceivedPowerNotification, (void *)(uintptr_t)msg);
-		} else if (key->isEqualTo(ac_detach_string.get())) {
-			DLOG("SetProperties: setting ac detach\n");
-			UInt32 msg = kIOPMSetACAdaptorConnected;
-			pmPowerStateQueue->submitPowerEvent(kPowerEventReceivedPowerNotification, (void *)(uintptr_t)msg);
-		} else if (key->isEqualTo(ac_attach_string.get())) {
-			DLOG("SetProperties: setting ac attach\n");
-			UInt32 msg = kIOPMSetACAdaptorConnected | kIOPMSetValue;
-			pmPowerStateQueue->submitPowerEvent(kPowerEventReceivedPowerNotification, (void *)(uintptr_t)msg);
-		} else if (key->isEqualTo(desktopmode_set_string.get())) {
-			DLOG("SetProperties: setting desktopmode");
-			UInt32 msg = kIOPMSetDesktopMode | kIOPMSetValue;
-			pmPowerStateQueue->submitPowerEvent(kPowerEventReceivedPowerNotification, (void *)(uintptr_t)msg);
-		} else if (key->isEqualTo(desktopmode_remove_string.get())) {
-			DLOG("SetProperties: removing desktopmode\n");
-			UInt32 msg = kIOPMSetDesktopMode;
-			pmPowerStateQueue->submitPowerEvent(kPowerEventReceivedPowerNotification, (void *)(uintptr_t)msg);
-		}
+    
+    if( sleepdisabled_string
+        && (b = OSDynamicCast(OSBoolean, dict->getObject(sleepdisabled_string))) )
+    {
+        setProperty(sleepdisabled_string, b);
+        pmPowerStateQueue->submitPowerEvent(kPowerEventUserDisabledSleep, (void *) b);
+    }
+    if (ondeck_sleepwake_uuid_string
+        && (obj = dict->getObject(ondeck_sleepwake_uuid_string)))
+    {
+        if(pmPowerStateQueue) {
+            obj->retain();
+            pmPowerStateQueue->submitPowerEvent(kPowerEventQueueSleepWakeUUID, (void *)obj);
+        }
+
+    }
+    
+    if (loginwindow_tracepoint_string
+        && (n = OSDynamicCast(OSNumber, dict->getObject(loginwindow_tracepoint_string)))
+        && pmTracer)
+    {
+        pmTracer->traceLoginWindowPhase( n->unsigned8BitValue() );
+    }
+
+    if ((b = OSDynamicCast(OSBoolean, dict->getObject(kIOPMDeepSleepEnabledKey))))
+    {
+        setProperty(kIOPMDeepSleepEnabledKey, b);
+    }
+    if ((n = OSDynamicCast(OSNumber, dict->getObject(kIOPMDeepSleepDelayKey))))
+    {
+        setProperty(kIOPMDeepSleepDelayKey, n);
+    }
+
+#if SUSPEND_PM_NOTIFICATIONS_DEBUG
+    if ((n = OSDynamicCast(OSNumber, dict->getObject(suspendPMClient_string))))
+    {
+        // Toggle the suspended status for pid n.
+        uint32_t pid_int = n->unsigned32BitValue();        
+        suspendPMNotificationsForPID(pid_int, !pmNotificationIsSuspended(pid_int));
+    }
 #endif
-		// Relay our allowed PM settings onto our registered PM clients
-		else if ((allowedPMSettings->getNextIndexOfObject(key, 0) != (unsigned int) -1)) {
-			return_value = setPMSetting(key, obj);
-			if (kIOReturnSuccess != return_value) {
-				break;
-			}
-		} else {
-			DLOG("setProperties(%s) not handled\n", key->getCStringNoCopy());
-		}
-	}
+    
+    if ((b = OSDynamicCast(OSBoolean, dict->getObject(kIOPMDestroyFVKeyOnStandbyKey))))
+    {
+        setProperty(kIOPMDestroyFVKeyOnStandbyKey, b);
+    }
+
+    // Relay our allowed PM settings onto our registered PM clients
+    for(i = 0; i < allowedPMSettings->getCount(); i++) {
+
+        type = (OSSymbol *)allowedPMSettings->getObject(i);
+        if(!type) continue;
+
+        obj = dict->getObject(type);
+        if(!obj) continue;
+
+        if ((gIOPMSettingAutoWakeSecondsKey == type) && ((n = OSDynamicCast(OSNumber, obj))))
+        {
+            UInt32 rsecs = n->unsigned32BitValue();
+            if (!rsecs)
+            autoWakeStart = autoWakeEnd = 0;
+            else
+            {
+            AbsoluteTime deadline;
+            clock_interval_to_deadline(rsecs + kAutoWakePostWindow, kSecondScale, &deadline);
+            autoWakeEnd = AbsoluteTime_to_scalar(&deadline);
+            if (rsecs > kAutoWakePreWindow)
+                rsecs -= kAutoWakePreWindow;
+            else
+                rsecs = 0;
+            clock_interval_to_deadline(rsecs, kSecondScale, &deadline);
+            autoWakeStart = AbsoluteTime_to_scalar(&deadline);
+            }
+        }
+        if (gIOPMSettingDebugWakeRelativeKey == type)
+        {
+            if ((n = OSDynamicCast(OSNumber, obj)))
+                _debugWakeSeconds = n->unsigned32BitValue();
+            else
+                _debugWakeSeconds = 0;
+        }
+        
+        return_value = setPMSetting(type, obj);
+        
+        if(kIOReturnSuccess != return_value) goto exit;
+    }
 
 exit:
-	return return_value;
-}
-
-#if HIBERNATION
-// MARK: -
-// MARK: setLockdownModeHibernation
-// ***************************************************************************
-void
-IOPMrootDomain::setLockdownModeHibernation(uint32_t status)
-{
-	if (!gIOPMWorkLoop->inGate()) {
-		gIOPMWorkLoop->runAction(
-			OSMemberFunctionCast(IOWorkLoop::Action, this,
-			&IOPMrootDomain::setLockdownModeHibernation),
-			this, (void *)(uintptr_t) status);
-		return;
-	}
-
-	ldmHibernateDisable = status;
-	DLOG("ldmHibernateDisable %d\n", status);
-	setProperty("IOPMLDMHibernationDisable", status);
-}
+    if(publish_simulated_battery_string) publish_simulated_battery_string->release();
+    if(boot_complete_string) boot_complete_string->release();
+    if(sys_shutdown_string) sys_shutdown_string->release();
+    if(stall_halt_string) stall_halt_string->release();
+    if(battery_warning_disabled_string) battery_warning_disabled_string->release();
+    if(idle_seconds_string) idle_seconds_string->release();
+    if(sleepdisabled_string) sleepdisabled_string->release();
+    if(ondeck_sleepwake_uuid_string) ondeck_sleepwake_uuid_string->release();
+    if(loginwindow_tracepoint_string) loginwindow_tracepoint_string->release();
+    if(pmTimelineLogging_string) pmTimelineLogging_string->release();
+#if	HIBERNATION
+    if(hibernatemode_string) hibernatemode_string->release();
+    if(hibernatefile_string) hibernatefile_string->release();
+    if(hibernatefreeratio_string) hibernatefreeratio_string->release();
+    if(hibernatefreetime_string) hibernatefreetime_string->release();
 #endif
-
-IOReturn
-IOPMrootDomain::getAssertionLog(IOPMAssertionLogData *outLog)
-{
-	if (!gIOPMWorkLoop->inGate()) {
-		return gIOPMWorkLoop->runAction(
-			OSMemberFunctionCast(IOWorkLoop::Action, this,
-			&IOPMrootDomain::getAssertionLog),
-			this, (void *)(uintptr_t) outLog);
-	}
-
-	if (!pmAssertions) {
-		return kIOReturnNotFound;
-	}
-
-	*outLog = pmAssertions->assertionsLog.data;
-	return kIOReturnSuccess;
-}
-
-IOReturn
-IOPMrootDomain::setAssertionLogNotificationPort(mach_port_t port)
-{
-	if (!gIOPMWorkLoop->inGate()) {
-		return gIOPMWorkLoop->runAction(
-			OSMemberFunctionCast(IOWorkLoop::Action, this,
-			&IOPMrootDomain::setAssertionLogNotificationPort),
-			this, (void *)(uintptr_t) port);
-	}
-
-	if (!pmAssertions) {
-		return kIOReturnNotFound;
-	}
-
-	return pmAssertions->assertionsLog.setNotificationPort(port);
-}
-
-IOReturn
-IOPMrootDomain::setAssertionLogNotificationThreshold(uint64_t threshold)
-{
-	if (!gIOPMWorkLoop->inGate()) {
-		return gIOPMWorkLoop->runAction(
-			OSMemberFunctionCast(IOWorkLoop::Action, this,
-			&IOPMrootDomain::setAssertionLogNotificationThreshold),
-			this, (void *)(uintptr_t) threshold);
-	}
-
-	if (!pmAssertions) {
-		return kIOReturnNotFound;
-	}
-
-	return pmAssertions->assertionsLog.setNotificationThreshold(threshold);
+#if SUSPEND_PM_NOTIFICATIONS_DEBUG
+    if(suspendPMClient_string) suspendPMClient_string->release();
+#endif
+    return return_value;
 }
 
 // MARK: -
@@ -2298,87 +1383,81 @@
 // Override IOService::setAggressiveness()
 //******************************************************************************
 
-IOReturn
-IOPMrootDomain::setAggressiveness(
-	unsigned long   type,
-	unsigned long   value )
-{
-	return setAggressiveness( type, value, 0 );
+IOReturn IOPMrootDomain::setAggressiveness(
+    unsigned long   type,
+    unsigned long   value )
+{
+    return setAggressiveness( type, value, 0 );
 }
 
 /*
  * Private setAggressiveness() with an internal options argument.
  */
-IOReturn
-IOPMrootDomain::setAggressiveness(
-	unsigned long   type,
-	unsigned long   value,
-	IOOptionBits    options )
-{
-	AggressivesRequest *    entry;
-	AggressivesRequest *    request;
-	bool                    found = false;
-
-	if ((type > UINT_MAX) || (value > UINT_MAX)) {
-		return kIOReturnBadArgument;
-	}
-
-	if (type == kPMMinutesToDim || type == kPMMinutesToSleep) {
-		DLOG("setAggressiveness(%x) %s = %u\n",
-		    (uint32_t) options, getAggressivenessTypeString((uint32_t) type), (uint32_t) value);
-	} else {
-		DEBUG_LOG("setAggressiveness(%x) %s = %u\n",
-		    (uint32_t) options, getAggressivenessTypeString((uint32_t) type), (uint32_t) value);
-	}
-
-	request = IOMallocType(AggressivesRequest);
-	request->options  = options;
-	request->dataType = kAggressivesRequestTypeRecord;
-	request->data.record.type  = (uint32_t) type;
-	request->data.record.value = (uint32_t) value;
-
-	AGGRESSIVES_LOCK();
-
-	// Update disk quick spindown flag used by getAggressiveness().
-	// Never merge requests with quick spindown flags set.
-
-	if (options & kAggressivesOptionQuickSpindownEnable) {
-		gAggressivesState |= kAggressivesStateQuickSpindown;
-	} else if (options & kAggressivesOptionQuickSpindownDisable) {
-		gAggressivesState &= ~kAggressivesStateQuickSpindown;
-	} else {
-		// Coalesce requests with identical aggressives types.
-		// Deal with callers that calls us too "aggressively".
-
-		queue_iterate(&aggressivesQueue, entry, AggressivesRequest *, chain)
-		{
-			if ((entry->dataType == kAggressivesRequestTypeRecord) &&
-			    (entry->data.record.type == type) &&
-			    ((entry->options & kAggressivesOptionQuickSpindownMask) == 0)) {
-				entry->data.record.value = (uint32_t) value;
-				found = true;
-				break;
-			}
-		}
-	}
-
-	if (!found) {
-		queue_enter(&aggressivesQueue, request, AggressivesRequest *, chain);
-	}
-
-	AGGRESSIVES_UNLOCK();
-
-	if (found) {
-		IOFreeType(request, AggressivesRequest);
-	}
-
-	if (options & kAggressivesOptionSynchronous) {
-		handleAggressivesRequests(); // not truly synchronous
-	} else {
-		thread_call_enter(aggressivesThreadCall);
-	}
-
-	return kIOReturnSuccess;
+IOReturn IOPMrootDomain::setAggressiveness(
+    unsigned long   type,
+    unsigned long   value,
+    IOOptionBits    options )
+{
+    AggressivesRequest *    entry;
+    AggressivesRequest *    request;
+    bool                    found = false;
+
+    DLOG("setAggressiveness(%x) 0x%x = %u\n",
+        (uint32_t) options, (uint32_t) type, (uint32_t) value);
+
+    request = IONew(AggressivesRequest, 1);
+    if (!request)
+        return kIOReturnNoMemory;
+
+    memset(request, 0, sizeof(*request));
+    request->options  = options;
+    request->dataType = kAggressivesRequestTypeRecord;
+    request->data.record.type  = (uint32_t) type;
+    request->data.record.value = (uint32_t) value;
+
+    AGGRESSIVES_LOCK();
+
+    // Update disk quick spindown flag used by getAggressiveness().
+    // Never merge requests with quick spindown flags set.
+
+    if (options & kAggressivesOptionQuickSpindownEnable)
+        gAggressivesState |= kAggressivesStateQuickSpindown;
+    else if (options & kAggressivesOptionQuickSpindownDisable)
+        gAggressivesState &= ~kAggressivesStateQuickSpindown;
+    else
+    {
+        // Coalesce requests with identical aggressives types.
+        // Deal with callers that calls us too "aggressively".
+
+        queue_iterate(&aggressivesQueue, entry, AggressivesRequest *, chain)
+        {
+            if ((entry->dataType == kAggressivesRequestTypeRecord) &&
+                (entry->data.record.type == type) &&
+                ((entry->options & kAggressivesOptionQuickSpindownMask) == 0))
+            {
+                entry->data.record.value = value;
+                found = true;
+                break;
+            }
+        }
+    }
+
+    if (!found)
+    {
+        queue_enter(&aggressivesQueue, request, AggressivesRequest *, chain);
+    }
+
+    AGGRESSIVES_UNLOCK();
+
+    if (found)
+        IODelete(request, AggressivesRequest, 1);
+
+    if (options & kAggressivesOptionSynchronous)
+        handleAggressivesRequests();   // not truly synchronous
+    else
+        thread_call_enter(aggressivesThreadCall);
+
+    return kIOReturnSuccess;
 }
 
 //******************************************************************************
@@ -2388,73 +1467,82 @@
 // Fetch the aggressiveness factor with the given type.
 //******************************************************************************
 
-IOReturn
-IOPMrootDomain::getAggressiveness(
-	unsigned long   type,
-	unsigned long * outLevel )
-{
-	uint32_t    value  = 0;
-	int         source = 0;
-
-	if (!outLevel || (type > UINT_MAX)) {
-		return kIOReturnBadArgument;
-	}
-
-	AGGRESSIVES_LOCK();
-
-	// Disk quick spindown in effect, report value = 1
-
-	if ((gAggressivesState & kAggressivesStateQuickSpindown) &&
-	    (type == kPMMinutesToSpinDown)) {
-		value  = kAggressivesMinValue;
-		source = 1;
-	}
-
-	// Consult the pending request queue.
-
-	if (!source) {
-		AggressivesRequest * entry;
-
-		queue_iterate(&aggressivesQueue, entry, AggressivesRequest *, chain)
-		{
-			if ((entry->dataType == kAggressivesRequestTypeRecord) &&
-			    (entry->data.record.type == type) &&
-			    ((entry->options & kAggressivesOptionQuickSpindownMask) == 0)) {
-				value  = entry->data.record.value;
-				source = 2;
-				break;
-			}
-		}
-	}
-
-	// Consult the backend records.
-
-	if (!source && aggressivesData) {
-		AggressivesRecord * record;
-		int                 i, count;
-
-		count  = aggressivesData->getLength() / sizeof(AggressivesRecord);
-		record = (AggressivesRecord *) aggressivesData->getBytesNoCopy();
-
-		for (i = 0; i < count; i++, record++) {
-			if (record->type == type) {
-				value  = record->value;
-				source = 3;
-				break;
-			}
-		}
-	}
-
-	AGGRESSIVES_UNLOCK();
-
-	if (source) {
-		*outLevel = (unsigned long) value;
-		return kIOReturnSuccess;
-	} else {
-		DLOG("getAggressiveness type 0x%x not found\n", (uint32_t) type);
-		*outLevel = 0; // default return = 0, driver may not check for error
-		return kIOReturnInvalid;
-	}
+IOReturn IOPMrootDomain::getAggressiveness (
+    unsigned long   type,
+    unsigned long * outLevel )
+{
+    uint32_t    value  = 0;
+    int         source = 0;
+
+    if (!outLevel)
+        return kIOReturnBadArgument;
+
+    AGGRESSIVES_LOCK();
+
+    // Disk quick spindown in effect, report value = 1
+
+    if ((gAggressivesState & kAggressivesStateQuickSpindown) &&
+        (type == kPMMinutesToSpinDown))
+    {
+        value  = kAggressivesMinValue;
+        source = 1;
+    }
+
+    // Consult the pending request queue.
+
+    if (!source)
+    {
+        AggressivesRequest * entry;
+
+        queue_iterate(&aggressivesQueue, entry, AggressivesRequest *, chain)
+        {
+            if ((entry->dataType == kAggressivesRequestTypeRecord) &&
+                (entry->data.record.type == type) &&
+                ((entry->options & kAggressivesOptionQuickSpindownMask) == 0))
+            {
+                value  = entry->data.record.value;
+                source = 2;
+                break;
+            }
+        }
+    }
+
+    // Consult the backend records.
+
+    if (!source && aggressivesData)
+    {
+        AggressivesRecord * record;
+        int                 i, count;
+
+        count  = aggressivesData->getLength() / sizeof(AggressivesRecord);
+        record = (AggressivesRecord *) aggressivesData->getBytesNoCopy();
+
+        for (i = 0; i < count; i++, record++)
+        {
+            if (record->type == type)
+            {
+                value  = record->value;
+                source = 3;
+                break;
+            }
+        }
+    }
+
+    AGGRESSIVES_UNLOCK();
+
+    if (source)
+    {
+        DLOG("getAggressiveness(%d) 0x%x = %u\n",
+            source, (uint32_t) type, value);
+        *outLevel = (unsigned long) value;
+        return kIOReturnSuccess;
+    }
+    else
+    {
+        DLOG("getAggressiveness type 0x%x not found\n", (uint32_t) type);
+        *outLevel = 0;  // default return = 0, driver may not check for error
+        return kIOReturnInvalid;
+    }
 }
 
 //******************************************************************************
@@ -2463,29 +1551,33 @@
 // Request from IOService to join future aggressiveness broadcasts.
 //******************************************************************************
 
-IOReturn
-IOPMrootDomain::joinAggressiveness(
-	IOService * service )
-{
-	AggressivesRequest *    request;
-
-	if (!service || (service == this)) {
-		return kIOReturnBadArgument;
-	}
-
-	DEBUG_LOG("joinAggressiveness %s %p\n", service->getName(), OBFUSCATE(service));
-
-	request = IOMallocType(AggressivesRequest);
-	request->dataType = kAggressivesRequestTypeService;
-	request->data.service.reset(service, OSRetain); // released by synchronizeAggressives()
-
-	AGGRESSIVES_LOCK();
-	queue_enter(&aggressivesQueue, request, AggressivesRequest *, chain);
-	AGGRESSIVES_UNLOCK();
-
-	thread_call_enter(aggressivesThreadCall);
-
-	return kIOReturnSuccess;
+IOReturn IOPMrootDomain::joinAggressiveness(
+    IOService * service )
+{
+    AggressivesRequest *    request;
+
+    if (!service || (service == this))
+        return kIOReturnBadArgument;
+
+    DLOG("joinAggressiveness %s %p\n", service->getName(), service);
+
+    request = IONew(AggressivesRequest, 1);
+    if (!request)
+        return kIOReturnNoMemory;
+
+    service->retain();  // released by synchronizeAggressives()
+
+    memset(request, 0, sizeof(*request));
+    request->dataType = kAggressivesRequestTypeService;
+    request->data.service = service;
+
+    AGGRESSIVES_LOCK();
+    queue_enter(&aggressivesQueue, request, AggressivesRequest *, chain);
+    AGGRESSIVES_UNLOCK();
+
+    thread_call_enter(aggressivesThreadCall);
+
+    return kIOReturnSuccess;
 }
 
 //******************************************************************************
@@ -2496,155 +1588,169 @@
 
 static void
 handleAggressivesFunction(
-	thread_call_param_t param1,
-	thread_call_param_t param2 )
-{
-	if (param1) {
-		((IOPMrootDomain *) param1)->handleAggressivesRequests();
-	}
-}
-
-void
-IOPMrootDomain::handleAggressivesRequests( void )
-{
-	AggressivesRecord *     start;
-	AggressivesRecord *     record;
-	AggressivesRequest *    request;
-	queue_head_t            joinedQueue;
-	int                     i, count;
-	bool                    broadcast;
-	bool                    found;
-	bool                    pingSelf = false;
-
-	AGGRESSIVES_LOCK();
-
-	if ((gAggressivesState & kAggressivesStateBusy) || !aggressivesData ||
-	    queue_empty(&aggressivesQueue)) {
-		goto unlock_done;
-	}
-
-	gAggressivesState |= kAggressivesStateBusy;
-	count = aggressivesData->getLength() / sizeof(AggressivesRecord);
-	start = (AggressivesRecord *) aggressivesData->getBytesNoCopy();
-
-	do{
-		broadcast = false;
-		queue_init(&joinedQueue);
-
-		do{
-			// Remove request from the incoming queue in FIFO order.
-			queue_remove_first(&aggressivesQueue, request, AggressivesRequest *, chain);
-			switch (request->dataType) {
-			case kAggressivesRequestTypeRecord:
-				// Update existing record if found.
-				found = false;
-				for (i = 0, record = start; i < count; i++, record++) {
-					if (record->type == request->data.record.type) {
-						found = true;
-
-						if (request->options & kAggressivesOptionQuickSpindownEnable) {
-							if ((record->flags & kAggressivesRecordFlagMinValue) == 0) {
-								broadcast = true;
-								record->flags |= (kAggressivesRecordFlagMinValue |
-								    kAggressivesRecordFlagModified);
-								DLOG("disk spindown accelerated, was %u min\n",
-								    record->value);
-							}
-						} else if (request->options & kAggressivesOptionQuickSpindownDisable) {
-							if (record->flags & kAggressivesRecordFlagMinValue) {
-								broadcast = true;
-								record->flags |= kAggressivesRecordFlagModified;
-								record->flags &= ~kAggressivesRecordFlagMinValue;
-								DLOG("disk spindown restored to %u min\n",
-								    record->value);
-							}
-						} else if (record->value != request->data.record.value) {
-							record->value = request->data.record.value;
-							if ((record->flags & kAggressivesRecordFlagMinValue) == 0) {
-								broadcast = true;
-								record->flags |= kAggressivesRecordFlagModified;
-							}
-						}
-						break;
-					}
-				}
-
-				// No matching record, append a new record.
-				if (!found &&
-				    ((request->options & kAggressivesOptionQuickSpindownDisable) == 0)) {
-					AggressivesRecord   newRecord;
-
-					newRecord.flags = kAggressivesRecordFlagModified;
-					newRecord.type  = request->data.record.type;
-					newRecord.value = request->data.record.value;
-					if (request->options & kAggressivesOptionQuickSpindownEnable) {
-						newRecord.flags |= kAggressivesRecordFlagMinValue;
-						DLOG("disk spindown accelerated\n");
-					}
-
-					aggressivesData->appendValue(newRecord);
-
-					// OSData may have switched to another (larger) buffer.
-					count = aggressivesData->getLength() / sizeof(AggressivesRecord);
-					start = (AggressivesRecord *) aggressivesData->getBytesNoCopy();
-					broadcast = true;
-				}
-
-				// Finished processing the request, release it.
-				IOFreeType(request, AggressivesRequest);
-				break;
-
-			case kAggressivesRequestTypeService:
-				// synchronizeAggressives() will free request.
-				queue_enter(&joinedQueue, request, AggressivesRequest *, chain);
-				break;
-
-			default:
-				panic("bad aggressives request type %x", request->dataType);
-				break;
-			}
-		} while (!queue_empty(&aggressivesQueue));
-
-		// Release the lock to perform work, with busy flag set.
-		if (!queue_empty(&joinedQueue) || broadcast) {
-			AGGRESSIVES_UNLOCK();
-			if (!queue_empty(&joinedQueue)) {
-				synchronizeAggressives(&joinedQueue, start, count);
-			}
-			if (broadcast) {
-				broadcastAggressives(start, count);
-			}
-			AGGRESSIVES_LOCK();
-		}
-
-		// Remove the modified flag from all records.
-		for (i = 0, record = start; i < count; i++, record++) {
-			if ((record->flags & kAggressivesRecordFlagModified) &&
-			    ((record->type == kPMMinutesToDim) ||
-			    (record->type == kPMMinutesToSleep))) {
-				pingSelf = true;
-			}
-
-			record->flags &= ~kAggressivesRecordFlagModified;
-		}
-
-		// Check the incoming queue again since new entries may have been
-		// added while lock was released above.
-	} while (!queue_empty(&aggressivesQueue));
-
-	gAggressivesState &= ~kAggressivesStateBusy;
+    thread_call_param_t param1,
+    thread_call_param_t param2 )
+{
+    if (param1)
+    {
+        ((IOPMrootDomain *) param1)->handleAggressivesRequests();
+    }
+}
+
+void IOPMrootDomain::handleAggressivesRequests( void )
+{
+    AggressivesRecord *     start;
+    AggressivesRecord *     record;
+    AggressivesRequest *    request;
+    queue_head_t            joinedQueue;
+    int                     i, count;
+    bool                    broadcast;
+    bool                    found;
+    bool                    pingSelf = false;
+
+    AGGRESSIVES_LOCK();
+
+    if ((gAggressivesState & kAggressivesStateBusy) || !aggressivesData ||
+        queue_empty(&aggressivesQueue))
+        goto unlock_done;
+
+    gAggressivesState |= kAggressivesStateBusy;
+    count = aggressivesData->getLength() / sizeof(AggressivesRecord);
+    start = (AggressivesRecord *) aggressivesData->getBytesNoCopy();
+
+    do
+    {
+        broadcast = false;
+        queue_init(&joinedQueue);
+
+        do
+        {
+            // Remove request from the incoming queue in FIFO order.
+            queue_remove_first(&aggressivesQueue, request, AggressivesRequest *, chain);
+            switch (request->dataType)
+            {
+                case kAggressivesRequestTypeRecord:
+                    // Update existing record if found.
+                    found = false;
+                    for (i = 0, record = start; i < count; i++, record++)
+                    {
+                        if (record->type == request->data.record.type)
+                        {
+                            found = true;
+
+                            if (request->options & kAggressivesOptionQuickSpindownEnable)
+                            {
+                                if ((record->flags & kAggressivesRecordFlagMinValue) == 0)
+                                {
+                                    broadcast = true;
+                                    record->flags |= (kAggressivesRecordFlagMinValue |
+                                                      kAggressivesRecordFlagModified);
+                                    DLOG("disk spindown accelerated, was %u min\n",
+                                        record->value);
+                                }
+                            }
+                            else if (request->options & kAggressivesOptionQuickSpindownDisable)
+                            {
+                                if (record->flags & kAggressivesRecordFlagMinValue)
+                                {
+                                    broadcast = true;
+                                    record->flags |= kAggressivesRecordFlagModified;
+                                    record->flags &= ~kAggressivesRecordFlagMinValue;
+                                    DLOG("disk spindown restored to %u min\n",
+                                        record->value);
+                                }
+                            }
+                            else if (record->value != request->data.record.value)
+                            {
+                                record->value = request->data.record.value;
+                                if ((record->flags & kAggressivesRecordFlagMinValue) == 0)
+                                {
+                                    broadcast = true;
+                                    record->flags |= kAggressivesRecordFlagModified;
+                                }
+                            }
+                            break;
+                        }
+                    }
+
+                    // No matching record, append a new record.
+                    if (!found &&
+                        ((request->options & kAggressivesOptionQuickSpindownDisable) == 0))
+                    {
+                        AggressivesRecord   newRecord;
+
+                        newRecord.flags = kAggressivesRecordFlagModified;
+                        newRecord.type  = request->data.record.type;
+                        newRecord.value = request->data.record.value;
+                        if (request->options & kAggressivesOptionQuickSpindownEnable)
+                        {
+                            newRecord.flags |= kAggressivesRecordFlagMinValue;
+                            DLOG("disk spindown accelerated\n");
+                        }
+
+                        aggressivesData->appendBytes(&newRecord, sizeof(newRecord));
+
+                        // OSData may have switched to another (larger) buffer.
+                        count = aggressivesData->getLength() / sizeof(AggressivesRecord);
+                        start = (AggressivesRecord *) aggressivesData->getBytesNoCopy();
+                        broadcast = true;
+                    }
+
+                    // Finished processing the request, release it.
+                    IODelete(request, AggressivesRequest, 1);
+                    break;
+
+                case kAggressivesRequestTypeService:
+                    // synchronizeAggressives() will free request.
+                    queue_enter(&joinedQueue, request, AggressivesRequest *, chain);
+                    break;
+
+                default:
+                    panic("bad aggressives request type %x\n", request->dataType);
+                    break;
+            }
+        } while (!queue_empty(&aggressivesQueue));
+
+        // Release the lock to perform work, with busy flag set.
+        if (!queue_empty(&joinedQueue) || broadcast)
+        {
+            AGGRESSIVES_UNLOCK();
+            if (!queue_empty(&joinedQueue))
+                synchronizeAggressives(&joinedQueue, start, count);
+            if (broadcast)
+                broadcastAggressives(start, count);
+            AGGRESSIVES_LOCK();
+        }
+
+        // Remove the modified flag from all records.
+        for (i = 0, record = start; i < count; i++, record++)
+        {
+            if ((record->flags & kAggressivesRecordFlagModified) &&
+                ((record->type == kPMMinutesToDim) ||
+                 (record->type == kPMMinutesToSleep)))
+                pingSelf = true;
+
+            record->flags &= ~kAggressivesRecordFlagModified;
+        }
+
+        // Check the incoming queue again since new entries may have been
+        // added while lock was released above.
+
+    } while (!queue_empty(&aggressivesQueue));
+
+    gAggressivesState &= ~kAggressivesStateBusy;
 
 unlock_done:
-	AGGRESSIVES_UNLOCK();
-
-	// Root domain is interested in system and display sleep slider changes.
-	// Submit a power event to handle those changes on the PM work loop.
-
-	if (pingSelf && pmPowerStateQueue) {
-		pmPowerStateQueue->submitPowerEvent(
-			kPowerEventPolicyStimulus,
-			(void *) kStimulusAggressivenessChanged );
-	}
+    AGGRESSIVES_UNLOCK();
+
+    // Root domain is interested in system and display sleep slider changes.
+    // Submit a power event to handle those changes on the PM work loop.
+
+    if (pingSelf && pmPowerStateQueue) {
+        pmPowerStateQueue->submitPowerEvent(
+            kPowerEventPolicyStimulus,
+            (void *) kStimulusAggressivenessChanged );
+    }
 }
 
 //******************************************************************************
@@ -2653,47 +1759,48 @@
 // Push all known aggressiveness records to one or more IOService.
 //******************************************************************************
 
-void
-IOPMrootDomain::synchronizeAggressives(
-	queue_head_t *              joinedQueue,
-	const AggressivesRecord *   array,
-	int                         count )
-{
-	OSSharedPtr<IOService>      service;
-	AggressivesRequest *        request;
-	const AggressivesRecord *   record;
-	IOPMDriverCallEntry         callEntry;
-	uint32_t                    value;
-	int                         i;
-
-	while (!queue_empty(joinedQueue)) {
-		queue_remove_first(joinedQueue, request, AggressivesRequest *, chain);
-		if (request->dataType == kAggressivesRequestTypeService) {
-			// retained by joinAggressiveness(), so take ownership
-			service = os::move(request->data.service);
-		} else {
-			service.reset();
-		}
-
-		IOFreeType(request, AggressivesRequest);
-		request = NULL;
-
-		if (service) {
-			if (service->assertPMDriverCall(&callEntry, kIOPMDriverCallMethodSetAggressive)) {
-				for (i = 0, record = array; i < count; i++, record++) {
-					value = record->value;
-					if (record->flags & kAggressivesRecordFlagMinValue) {
-						value = kAggressivesMinValue;
-					}
-
-					_LOG("synchronizeAggressives 0x%x = %u to %s\n",
-					    record->type, value, service->getName());
-					service->setAggressiveness(record->type, value);
-				}
-				service->deassertPMDriverCall(&callEntry);
-			}
-		}
-	}
+void IOPMrootDomain::synchronizeAggressives(
+    queue_head_t *              joinedQueue,
+    const AggressivesRecord *   array,
+    int                         count )
+{
+    IOService *                 service;
+    AggressivesRequest *        request;
+    const AggressivesRecord *   record;
+    IOPMDriverCallEntry         callEntry;
+    uint32_t                    value;
+    int                         i;
+
+    while (!queue_empty(joinedQueue))
+    {
+        queue_remove_first(joinedQueue, request, AggressivesRequest *, chain);
+        if (request->dataType == kAggressivesRequestTypeService)
+            service = request->data.service;
+        else
+            service = 0;
+
+        IODelete(request, AggressivesRequest, 1);
+        request = 0;
+
+        if (service)
+        {
+            if (service->assertPMDriverCall(&callEntry))
+            {
+                for (i = 0, record = array; i < count; i++, record++)
+                {
+                    value = record->value;
+                    if (record->flags & kAggressivesRecordFlagMinValue)
+                        value = kAggressivesMinValue;
+
+                    _LOG("synchronizeAggressives 0x%x = %u to %s\n",
+                        record->type, value, service->getName());
+                    service->setAggressiveness(record->type, value);
+                }
+                service->deassertPMDriverCall(&callEntry);
+            }
+            service->release();     // retained by joinAggressiveness()
+        }
+    }
 }
 
 //******************************************************************************
@@ -2702,81 +1809,58 @@
 // Traverse PM tree and call setAggressiveness() for records that have changed.
 //******************************************************************************
 
-void
-IOPMrootDomain::broadcastAggressives(
-	const AggressivesRecord *   array,
-	int                         count )
-{
-	OSSharedPtr<IORegistryIterator> iter;
-	IORegistryEntry                *entry;
-	OSSharedPtr<IORegistryEntry>    child;
-	IOPowerConnection              *connect;
-	IOService                      *service;
-	const AggressivesRecord        *record;
-	IOPMDriverCallEntry             callEntry;
-	uint32_t                        value;
-	int                             i;
-
-	iter = IORegistryIterator::iterateOver(
-		this, gIOPowerPlane, kIORegistryIterateRecursively);
-	if (iter) {
-		do{
-			// !! reset the iterator
-			iter->reset();
-			while ((entry = iter->getNextObject())) {
-				connect = OSDynamicCast(IOPowerConnection, entry);
-				if (!connect || !connect->getReadyFlag()) {
-					continue;
-				}
-
-				child = connect->copyChildEntry(gIOPowerPlane);
-				if (child) {
-					if ((service = OSDynamicCast(IOService, child.get()))) {
-						if (service->assertPMDriverCall(&callEntry, kIOPMDriverCallMethodSetAggressive)) {
-							for (i = 0, record = array; i < count; i++, record++) {
-								if (record->flags & kAggressivesRecordFlagModified) {
-									value = record->value;
-									if (record->flags & kAggressivesRecordFlagMinValue) {
-										value = kAggressivesMinValue;
-									}
-									_LOG("broadcastAggressives %x = %u to %s\n",
-									    record->type, value, service->getName());
-									service->setAggressiveness(record->type, value);
-								}
-							}
-							service->deassertPMDriverCall(&callEntry);
-						}
-					}
-				}
-			}
-		}while (!entry && !iter->isValid());
-	}
-}
-
-//*****************************************
-// stackshot on power button press
-// ***************************************
-static void
-powerButtonDownCallout(thread_call_param_t us, thread_call_param_t )
-{
-	/* Power button pressed during wake
-	 * Take a stackshot
-	 */
-	DEBUG_LOG("Powerbutton: down. Taking stackshot\n");
-	((IOPMrootDomain *)us)->takeStackshot(false);
-}
-
-static void
-powerButtonUpCallout(thread_call_param_t us, thread_call_param_t)
-{
-	/* Power button released.
-	 * Delete any stackshot data
-	 */
-	DEBUG_LOG("PowerButton: up callout. Delete stackshot\n");
-	((IOPMrootDomain *)us)->deleteStackshot();
-}
-//*************************************************************************
-//
+void IOPMrootDomain::broadcastAggressives(
+    const AggressivesRecord *   array,
+    int                         count )
+{
+    IORegistryIterator *        iter;
+    IORegistryEntry *           entry;
+    IOPowerConnection *         connect;
+    IOService *                 service;
+    const AggressivesRecord *   record;
+    IOPMDriverCallEntry         callEntry;
+    uint32_t                    value;
+    int                         i;
+
+    iter = IORegistryIterator::iterateOver(
+            this, gIOPowerPlane, kIORegistryIterateRecursively);
+    if (iter)
+    {
+        do
+        {
+            iter->reset();
+            while ((entry = iter->getNextObject()))
+            {
+                connect = OSDynamicCast(IOPowerConnection, entry);
+                if (!connect || !connect->getReadyFlag())
+                    continue;
+
+                if ((service = (IOService *) connect->copyChildEntry(gIOPowerPlane)))
+                {
+                    if (service->assertPMDriverCall(&callEntry))
+                    {
+                        for (i = 0, record = array; i < count; i++, record++)
+                        {
+                            if (record->flags & kAggressivesRecordFlagModified)
+                            {
+                                value = record->value;
+                                if (record->flags & kAggressivesRecordFlagMinValue)
+                                    value = kAggressivesMinValue;
+                                _LOG("broadcastAggressives %x = %u to %s\n",
+                                    record->type, value, service->getName());
+                                service->setAggressiveness(record->type, value);
+                            }
+                        }
+                        service->deassertPMDriverCall(&callEntry);
+                    }
+                    service->release();
+                }
+            }
+        }
+        while (!entry && !iter->isValid());
+        iter->release();
+    }
+}
 
 // MARK: -
 // MARK: System Sleep
@@ -2786,47 +1870,22 @@
 //
 //******************************************************************************
 
-void
-IOPMrootDomain::startIdleSleepTimer( uint32_t inMilliSeconds )
-{
-	AbsoluteTime deadline;
-
-	ASSERT_GATED();
-	if (gNoIdleFlag) {
-		DLOG("idle timer not set (noidle=%d)\n", gNoIdleFlag);
-		return;
-	}
-	if (inMilliSeconds) {
-		if (inMilliSeconds < kMinimumTimeBeforeIdleSleep) {
-			AbsoluteTime    now;
-			uint64_t        nsec_since_wake;
-			uint64_t                msec_since_wake;
-
-			// Adjust idle timer so it will not expire until atleast kMinimumTimeBeforeIdleSleep milliseconds
-			// after the most recent AP wake.
-			clock_get_uptime(&now);
-			SUB_ABSOLUTETIME(&now, &gIOLastWakeAbsTime);
-			absolutetime_to_nanoseconds(now, &nsec_since_wake);
-			msec_since_wake = nsec_since_wake / NSEC_PER_MSEC;
-
-			if (msec_since_wake < kMinimumTimeBeforeIdleSleep) {
-				uint32_t newIdleTimer = kMinimumTimeBeforeIdleSleep - (uint32_t)msec_since_wake;
-
-				// Ensure that our new idle timer is not less than inMilliSeconds,
-				// as we should only be increasing the timer duration, not decreasing it
-				if (newIdleTimer > inMilliSeconds) {
-					DLOG("startIdleSleepTimer increasing timeout from %u to %u\n", inMilliSeconds, newIdleTimer);
-					inMilliSeconds = newIdleTimer;
-				}
-			}
-		}
-		clock_interval_to_deadline(inMilliSeconds, kMillisecondScale, &deadline);
-		thread_call_enter_delayed(extraSleepTimer, deadline);
-		idleSleepTimerPending = true;
-	} else {
-		thread_call_enter(extraSleepTimer);
-	}
-	DLOG("idle timer set for %u milliseconds\n", inMilliSeconds);
+void IOPMrootDomain::startIdleSleepTimer( uint32_t inSeconds )
+{
+    AbsoluteTime deadline;
+
+    ASSERT_GATED();
+    if (inSeconds)
+    {
+        clock_interval_to_deadline(inSeconds, kSecondScale, &deadline);	
+        thread_call_enter_delayed(extraSleepTimer, deadline);
+        idleSleepTimerPending = true;
+    }
+    else
+    {
+        thread_call_enter(extraSleepTimer);
+    }
+    DLOG("idle timer set for %u seconds\n", inSeconds);
 }
 
 //******************************************************************************
@@ -2834,27 +1893,15 @@
 //
 //******************************************************************************
 
-void
-IOPMrootDomain::cancelIdleSleepTimer( void )
-{
-	ASSERT_GATED();
-	if (idleSleepTimerPending) {
-		DLOG("idle timer cancelled\n");
-		thread_call_cancel(extraSleepTimer);
-		idleSleepTimerPending = false;
-
-		if (!assertOnWakeSecs && gIOLastWakeAbsTime) {
-			AbsoluteTime    now;
-			clock_usec_t    microsecs;
-			clock_get_uptime(&now);
-			SUB_ABSOLUTETIME(&now, &gIOLastWakeAbsTime);
-			absolutetime_to_microtime(now, &assertOnWakeSecs, &microsecs);
-			if (assertOnWakeReport) {
-				HISTREPORT_TALLYVALUE(assertOnWakeReport, (int64_t)assertOnWakeSecs);
-				DLOG("Updated assertOnWake %lu\n", (unsigned long)assertOnWakeSecs);
-			}
-		}
-	}
+void IOPMrootDomain::cancelIdleSleepTimer( void )
+{
+    ASSERT_GATED();
+    if (idleSleepTimerPending)
+    {
+        DLOG("idle timer cancelled\n");
+        thread_call_cancel(extraSleepTimer);
+        idleSleepTimerPending = false;
+    }
 }
 
 //******************************************************************************
@@ -2862,11 +1909,10 @@
 //
 //******************************************************************************
 
-static void
-idleSleepTimerExpired(
-	thread_call_param_t us, thread_call_param_t )
-{
-	((IOPMrootDomain *)us)->handleSleepTimerExpiration();
+static void idleSleepTimerExpired(
+    thread_call_param_t us, thread_call_param_t )
+{
+    ((IOPMrootDomain *)us)->handleSleepTimerExpiration();
 }
 
 //******************************************************************************
@@ -2876,78 +1922,34 @@
 // It's time to sleep. Start that by removing the clamp that's holding us awake.
 //******************************************************************************
 
-void
-IOPMrootDomain::handleSleepTimerExpiration( void )
-{
-	if (!gIOPMWorkLoop->inGate()) {
-		gIOPMWorkLoop->runAction(
-			OSMemberFunctionCast(IOWorkLoop::Action, this,
-			&IOPMrootDomain::handleSleepTimerExpiration),
-			this);
-		return;
-	}
-
-	DLOG("sleep timer expired\n");
-	ASSERT_GATED();
-
-	idleSleepTimerPending = false;
-	setQuickSpinDownTimeout();
-	adjustPowerState(true);
-}
-
-//******************************************************************************
-// getTimeToIdleSleep
-//
-// Returns number of milliseconds left before going into idle sleep.
-// Caller has to make sure that idle sleep is allowed at the time of calling
-// this function
-//******************************************************************************
-
-uint32_t
-IOPMrootDomain::getTimeToIdleSleep( void )
-{
-	AbsoluteTime    now, lastActivityTime;
-	uint64_t        nanos;
-	uint32_t        minutesSinceUserInactive = 0;
-	uint32_t        sleepDelay = 0;
-
-	if (!idleSleepEnabled) {
-		return 0xffffffff;
-	}
-
-	if (userActivityTime) {
-		lastActivityTime = userActivityTime;
-	} else {
-		lastActivityTime = userBecameInactiveTime;
-	}
-
-	// Ignore any lastActivityTime that predates the last system wake.
-	// The goal is to avoid a sudden idle sleep right after a dark wake
-	// due to sleepDelay=0 computed below. The alternative 60s minimum
-	// timeout should be large enough to allow dark wake to complete,
-	// at which point the idle timer will be promptly cancelled.
-	clock_get_uptime(&now);
-	if ((CMP_ABSOLUTETIME(&lastActivityTime, &gIOLastWakeAbsTime) >= 0) &&
-	    (CMP_ABSOLUTETIME(&now, &lastActivityTime) > 0)) {
-		SUB_ABSOLUTETIME(&now, &lastActivityTime);
-		absolutetime_to_nanoseconds(now, &nanos);
-		minutesSinceUserInactive = nanos / (60000000000ULL);
-
-		if (minutesSinceUserInactive >= sleepSlider) {
-			sleepDelay = 0;
-		} else {
-			sleepDelay = sleepSlider - minutesSinceUserInactive;
-		}
-	} else {
-		DLOG("ignoring lastActivityTime 0x%qx, now 0x%qx, wake 0x%qx\n",
-		    lastActivityTime, now, gIOLastWakeAbsTime);
-		sleepDelay = sleepSlider;
-	}
-
-	DLOG("user inactive %u min, time to idle sleep %u min\n",
-	    minutesSinceUserInactive, sleepDelay);
-
-	return sleepDelay * 60 * 1000;
+void IOPMrootDomain::handleSleepTimerExpiration( void )
+{
+    if (!getPMworkloop()->inGate())
+    {
+        getPMworkloop()->runAction(
+            OSMemberFunctionCast(IOWorkLoop::Action, this,
+                &IOPMrootDomain::handleSleepTimerExpiration),
+            this);
+        return;
+    }
+
+    AbsoluteTime time;
+
+    DLOG("sleep timer expired\n");
+    ASSERT_GATED();
+
+    idleSleepTimerPending = false;
+
+    clock_get_uptime(&time);
+    if ((AbsoluteTime_to_scalar(&time) > autoWakeStart) &&
+        (AbsoluteTime_to_scalar(&time) < autoWakeEnd))
+    {
+        thread_call_enter_delayed(extraSleepTimer, *((AbsoluteTime *) &autoWakeEnd));
+        return;
+    }
+
+    setQuickSpinDownTimeout();
+    adjustPowerState(true);
 }
 
 //******************************************************************************
@@ -2955,12 +1957,11 @@
 //
 //******************************************************************************
 
-void
-IOPMrootDomain::setQuickSpinDownTimeout( void )
-{
-	ASSERT_GATED();
-	setAggressiveness(
-		kPMMinutesToSpinDown, 0, kAggressivesOptionQuickSpindownEnable );
+void IOPMrootDomain::setQuickSpinDownTimeout( void )
+{
+    ASSERT_GATED();
+    setAggressiveness(
+        kPMMinutesToSpinDown, 0, kAggressivesOptionQuickSpindownEnable );
 }
 
 //******************************************************************************
@@ -2968,12 +1969,11 @@
 //
 //******************************************************************************
 
-void
-IOPMrootDomain::restoreUserSpinDownTimeout( void )
-{
-	ASSERT_GATED();
-	setAggressiveness(
-		kPMMinutesToSpinDown, 0, kAggressivesOptionQuickSpindownDisable );
+void IOPMrootDomain::restoreUserSpinDownTimeout( void )
+{
+    ASSERT_GATED();
+    setAggressiveness(
+        kPMMinutesToSpinDown, 0, kAggressivesOptionQuickSpindownDisable );
 }
 
 //******************************************************************************
@@ -2982,60 +1982,137 @@
 //******************************************************************************
 
 /* public */
-IOReturn
-IOPMrootDomain::sleepSystem( void )
-{
-	return sleepSystemOptions(NULL);
+IOReturn IOPMrootDomain::sleepSystem( void )
+{
+    return sleepSystemOptions(NULL);
 }
 
 /* private */
-IOReturn
-IOPMrootDomain::sleepSystemOptions( OSDictionary *options )
-{
-	OSObject *obj = NULL;
-	OSString *reason = NULL;
+IOReturn IOPMrootDomain::sleepSystemOptions( OSDictionary *options )
+{
 	/* sleepSystem is a public function, and may be called by any kernel driver.
-	 * And that's bad - drivers should sleep the system by calling
-	 * receivePowerNotification() instead. Drivers should not use sleepSystem.
-	 *
-	 * Note that user space app calls to IOPMSleepSystem() will also travel
-	 * this code path and thus be correctly identified as software sleeps.
-	 */
-
-	if (options && options->getObject("OSSwitch")) {
-		// Log specific sleep cause for OS Switch hibernation
-		return privateSleepSystem( kIOPMSleepReasonOSSwitchHibernate);
-	}
-
-	if (options && (obj = options->getObject("Sleep Reason"))) {
-		reason = OSDynamicCast(OSString, obj);
-		if (reason && reason->isEqualTo(kIOPMDarkWakeThermalEmergencyKey)) {
-			return privateSleepSystem(kIOPMSleepReasonDarkWakeThermalEmergency);
-		}
-		if (reason && reason->isEqualTo(kIOPMNotificationWakeExitKey)) {
-			return privateSleepSystem(kIOPMSleepReasonNotificationWakeExit);
-		}
-	}
-
-	return privateSleepSystem( kIOPMSleepReasonSoftware);
+     * And that's bad - drivers should sleep the system by calling 
+     * receivePowerNotification() instead. Drivers should not use sleepSystem.
+     *
+     * Note that user space app calls to IOPMSleepSystem() will also travel
+     * this code path and thus be correctly identified as software sleeps.
+     */
+          
+    if (options && options->getObject("OSSwitch")) 
+    {
+        // Log specific sleep cause for OS Switch hibernation
+        return privateSleepSystem( kIOPMSleepReasonOSSwitchHibernate);
+    } else {
+        return privateSleepSystem( kIOPMSleepReasonSoftware);
+    }
 }
 
 /* private */
-IOReturn
-IOPMrootDomain::privateSleepSystem( uint32_t sleepReason )
-{
-	/* Called from both gated and non-gated context */
-
-	if (!checkSystemSleepEnabled() || !pmPowerStateQueue) {
-		return kIOReturnNotPermitted;
-	}
-
-	pmPowerStateQueue->submitPowerEvent(
-		kPowerEventPolicyStimulus,
-		(void *) kStimulusDemandSystemSleep,
-		sleepReason);
-
-	return kIOReturnSuccess;
+IOReturn IOPMrootDomain::privateSleepSystem( uint32_t sleepReason )
+{
+    static const char * IOPMSleepReasons[] = {
+        "",
+        kIOPMClamshellSleepKey,
+        kIOPMPowerButtonSleepKey,
+        kIOPMSoftwareSleepKey,
+        kIOPMOSSwitchHibernationKey,
+        kIOPMIdleSleepKey,
+        kIOPMLowPowerSleepKey,
+        kIOPMThermalEmergencySleepKey,
+        kIOPMMaintenanceSleepKey,
+        kIOPMSleepServiceExitKey,
+        kIOPMDarkWakeThermalEmergencyKey
+    };
+
+    PMEventDetails *details;
+
+    if (!checkSystemCanSleep())
+    {
+        // Record why the system couldn't sleep	
+        details = PMEventDetails::eventDetails(kIOPMEventTypeSleep, NULL,
+                                        sleepReason, kIOReturnNotPermitted);
+		
+		recordAndReleasePMEvent( details );
+        return kIOReturnNotPermitted;
+    }
+
+    if (kIOPMSleepReasonDarkWakeThermalEmergency == sleepReason)
+        messageClients(kIOPMMessageDarkWakeThermalEmergency);
+
+    if (timeline)
+        timeline->setSleepCycleInProgressFlag(true);
+  
+    // Time to publish a UUID for the Sleep --> Wake cycle  
+    if(pmPowerStateQueue) {
+        pmPowerStateQueue->submitPowerEvent(kPowerEventPublishSleepWakeUUID, (void *)true);
+    }
+  
+    // Log the beginning of system sleep.
+	details = PMEventDetails::eventDetails(kIOPMEventTypeSleep, NULL,
+                                            sleepReason, kIOReturnSuccess);
+	
+	recordAndReleasePMEvent( details );
+	
+    // Record sleep cause in IORegistry
+    lastSleepReason = sleepReason;
+    sleepReason -= (kIOPMSleepReasonClamshell - 1);
+    if (sleepReason && (sleepReason < sizeof(IOPMSleepReasons)/sizeof(IOPMSleepReasons[0]))) {
+        setProperty(kRootDomainSleepReasonKey, IOPMSleepReasons[sleepReason]);
+    }
+
+    if (pmPowerStateQueue)
+        pmPowerStateQueue->submitPowerEvent(
+                            kPowerEventPolicyStimulus,
+                            (void *) kStimulusDemandSystemSleep );
+
+    return kIOReturnSuccess;
+}
+
+IOReturn IOPMrootDomain::recordPMEventGated(PMEventDetails *record)
+{  
+  // If we don't have a place to log to, we can't actually
+  // log anything. Chances are, the person who is asking us to do    
+  // the PM logging has forgotten to set the right bootflags
+  if(!timeline)
+    return kIOReturnSuccess;
+
+  if(gIOPMWorkLoop->inGate() == false) {
+    
+    IOReturn ret = gIOPMWorkLoop->runAction(
+                     OSMemberFunctionCast(IOWorkLoop::Action, this, &IOPMrootDomain::recordPMEventGated),
+                     (OSObject *)this,
+                     (void *)record);
+    
+    return ret;
+  }
+  else {
+    // Now that we're guaranteed to be running in gate ...
+
+    // Check the validity of the argument we are given
+    if(!record) 
+      return kIOReturnBadArgument;
+    
+	  // Record a driver event, or a system event
+	  if(record->eventClassifier == kIOPMEventClassDriverEvent
+	     || record->eventClassifier == kIOPMEventClassSystemEvent)
+		  return this->recordPMEvent(record);
+	
+	  else
+		  return kIOReturnBadArgument;
+  }
+}
+
+IOReturn IOPMrootDomain::recordAndReleasePMEventGated(PMEventDetails *record)
+{
+    IOReturn ret = kIOReturnBadArgument;
+
+    if (record)
+    {
+        ret = recordPMEventGated(record);
+        record->release();
+    }
+    
+    return ret;
 }
 
 //******************************************************************************
@@ -3043,422 +2120,236 @@
 //
 // This overrides powerChangeDone in IOService.
 //******************************************************************************
-void
-IOPMrootDomain::powerChangeDone( unsigned long previousPowerState )
-{
-#if !__i386__ && !__x86_64__
-	uint64_t    timeSinceReset = 0;
+
+void IOPMrootDomain::powerChangeDone( unsigned long previousPowerState )
+{
+    PMEventDetails *details;
+
+    ASSERT_GATED();
+    DLOG("PowerChangeDone: %u->%u\n",
+        (uint32_t) previousPowerState, (uint32_t) getPowerState());
+	
+    switch ( getPowerState() )
+    {
+        case SLEEP_STATE: {
+            if (previousPowerState != ON_STATE)
+                break;
+			
+            details = PMEventDetails::eventDetails(
+                            kIOPMEventTypeSleepDone,
+                            NULL, 
+                            NULL, 
+                            kIOReturnSuccess);
+			
+            recordAndReleasePMEvent( details );
+
+            // re-enable this timer for next sleep
+            cancelIdleSleepTimer();
+
+            clock_sec_t		secs;
+            clock_usec_t	microsecs;
+            clock_get_calendar_microtime(&secs, &microsecs);
+            logtime(secs);
+            gIOLastSleepTime.tv_sec  = secs;
+            gIOLastSleepTime.tv_usec = microsecs;
+            gIOLastWakeTime.tv_sec = 0;
+            gIOLastWakeTime.tv_usec = 0;
+
+#if	HIBERNATION
+            LOG("System %sSleep\n", gIOHibernateState ? "Safe" : "");
+
+            IOHibernateSystemHasSlept();
+
+            evaluateSystemSleepPolicyFinal();
+#else
+            LOG("System Sleep\n");
 #endif
-	uint64_t           now;
-	unsigned long      newState;
-	clock_sec_t        secs;
-	clock_usec_t       microsecs;
-	uint32_t           lastDebugWakeSeconds;
-	clock_sec_t        adjWakeTime;
-	IOPMCalendarStruct nowCalendar;
-
-	ASSERT_GATED();
-	newState = getPowerState();
-	DLOG("PowerChangeDone: %s->%s\n",
-	    getPowerStateString((uint32_t) previousPowerState), getPowerStateString((uint32_t) getPowerState()));
-
-	if (previousPowerState == newState) {
-		return;
-	}
-
-	notifierThread = current_thread();
-	switch (getPowerState()) {
-	case SLEEP_STATE: {
-#if defined(__arm64__) && HIBERNATION
-		if (kIOHibernateStateInactive == gIOHibernateState)
-#endif /* defined(__arm64__) && HIBERNATION */
-		{
-			if (kIOPMDriverAssertionLevelOn == getPMAssertionLevel(kIOPMDriverAssertionForceWakeupBit)) {
-				IOLog("accelerate wake for assertion\n");
-				setWakeTime(mach_continuous_time());
-			}
-			if (kIOPMDriverAssertionLevelOn == getPMAssertionLevel(kIOPMDriverAssertionForceFullWakeupBit)) {
-				// Note: The scheduled RTC wakeup will trigger a full wake.
-				scheduleImmediateDebugWake();
-			}
-		}
-		if (kPMCalendarTypeInvalid != _aotWakeTimeCalendar.selector) {
-			secs = 0;
-			microsecs = 0;
-			PEGetUTCTimeOfDay(&secs, &microsecs);
-
-			adjWakeTime = 0;
-			if ((kIOPMAOTModeRespectTimers & _aotMode) && (_calendarWakeAlarmUTC < _aotWakeTimeUTC)) {
-				IOLog("use _calendarWakeAlarmUTC\n");
-				adjWakeTime = _calendarWakeAlarmUTC;
-			} else if (kIOPMWakeEventAOTExitFlags & _aotPendingFlags) {
-				IOLog("accelerate _aotWakeTime for exit\n");
-				adjWakeTime = secs;
-			} else if (kIOPMDriverAssertionLevelOn == getPMAssertionLevel(kIOPMDriverAssertionCPUBit)) {
-				IOLog("accelerate _aotWakeTime for assertion\n");
-				adjWakeTime = secs;
-			}
-			if (adjWakeTime) {
-				IOPMConvertSecondsToCalendar(adjWakeTime, &_aotWakeTimeCalendar);
-			}
-
-			IOPMConvertSecondsToCalendar(secs, &nowCalendar);
-			IOLog("aotSleep at " YMDTF " sched: " YMDTF "\n", YMDT(&nowCalendar), YMDT(&_aotWakeTimeCalendar));
-
-			IOReturn __unused ret = setMaintenanceWakeCalendar(&_aotWakeTimeCalendar);
-			assert(kIOReturnSuccess == ret);
-		}
-		if (_aotLastWakeTime) {
-			_aotMetrics->totalTime += mach_absolute_time() - _aotLastWakeTime;
-			if (_aotMetrics->sleepCount && (_aotMetrics->sleepCount <= kIOPMAOTMetricsKernelWakeCountMax)) {
-				strlcpy(&_aotMetrics->kernelWakeReason[_aotMetrics->sleepCount - 1][0],
-				    gWakeReasonString,
-				    sizeof(_aotMetrics->kernelWakeReason[_aotMetrics->sleepCount]));
-			}
-		}
-		_aotPendingFlags &= ~kIOPMWakeEventAOTPerCycleFlags;
-		if (_aotTimerScheduled) {
-			_aotTimerES->cancelTimeout();
-			_aotTimerScheduled = false;
-		}
-		acceptSystemWakeEvents(kAcceptSystemWakeEvents_Enable);
-
-		// re-enable this timer for next sleep
-		cancelIdleSleepTimer();
-
-		if (clamshellExists) {
-#if DARK_TO_FULL_EVALUATE_CLAMSHELL_DELAY
-			if (gClamshellFlags & kClamshell_WAR_58009435) {
-				// Disable clamshell sleep until system has completed full wake.
-				// This prevents a system sleep request (due to a clamshell close)
-				// from being queued until the end of system full wake - even if
-				// other clamshell disable bits outside of our control is wrong.
-				setClamShellSleepDisable(true, kClamshellSleepDisableInternal);
-			}
+
+            getPlatform()->sleepKernel();
+
+            // The CPU(s) are off at this point,
+            // Code will resume execution here upon wake.
+
+            clock_get_uptime(&systemWakeTime);
+
+#if	HIBERNATION
+            IOHibernateSystemWake();
 #endif
 
-			// Log the last known clamshell state before system sleep
-			DLOG("clamshell closed %d, disabled %d/%x, desktopMode %d, ac %d\n",
-			    clamshellClosed, clamshellDisabled, clamshellSleepDisableMask,
-			    desktopMode, acAdaptorConnected);
-		}
-
-		clock_get_calendar_absolute_and_microtime(&secs, &microsecs, &now);
-		logtime(secs);
-		gIOLastSleepTime.tv_sec  = secs;
-		gIOLastSleepTime.tv_usec = microsecs;
-		if (!_aotLastWakeTime) {
-			gIOLastUserSleepTime = gIOLastSleepTime;
-		}
-		gIOLastWakeTime.tv_sec = 0;
-		gIOLastWakeTime.tv_usec = 0;
-		gIOLastSleepAbsTime = now;
-
-		if (wake2DarkwakeDelay && sleepDelaysReport) {
-			clock_sec_t     wake2DarkwakeSecs, darkwake2SleepSecs;
-			// Update 'wake2DarkwakeDelay' histogram if this is a fullwake->sleep transition
-
-			SUB_ABSOLUTETIME(&now, &ts_sleepStart);
-			absolutetime_to_microtime(now, &darkwake2SleepSecs, &microsecs);
-			absolutetime_to_microtime(wake2DarkwakeDelay, &wake2DarkwakeSecs, &microsecs);
-			HISTREPORT_TALLYVALUE(sleepDelaysReport,
-			    (int64_t)(wake2DarkwakeSecs + darkwake2SleepSecs));
-
-			DLOG("Updated sleepDelaysReport %lu %lu\n", (unsigned long)wake2DarkwakeSecs, (unsigned long)darkwake2SleepSecs);
-			wake2DarkwakeDelay = 0;
-		}
-#if HIBERNATION
-		LOG("System %sSleep\n", gIOHibernateState ? "Safe" : "");
-#if (DEVELOPMENT || DEBUG)
-		record_system_event(SYSTEM_EVENT_TYPE_INFO,
-		    SYSTEM_EVENT_SUBSYSTEM_PMRD,
-		    "System State",
-		    gIOHibernateState ? "Enter Hibernate" : "Enter Sleep"
-		    );
-#endif /* DEVELOPMENT || DEBUG */
-		IOHibernateSystemHasSlept();
-
-		evaluateSystemSleepPolicyFinal();
-#else
-		LOG("System Sleep\n");
-#if (DEVELOPMENT || DEBUG)
-		record_system_event(SYSTEM_EVENT_TYPE_INFO,
-		    SYSTEM_EVENT_SUBSYSTEM_PMRD,
-		    "System State", "Enter Sleep");
-#endif /* DEVELOPMENT || DEBUG */
+            // sleep transition complete
+            gSleepOrShutdownPending = 0;
+
+            // trip the reset of the calendar clock
+            clock_wakeup_calendar();
+
+#if	HIBERNATION
+            LOG("System %sWake\n", gIOHibernateState ? "SafeSleep " : "");
 #endif
-		if (thermalWarningState) {
-			OSSharedPtr<const OSSymbol> event = OSSymbol::withCString(kIOPMThermalLevelWarningKey);
-			if (event) {
-				systemPowerEventOccurred(event.get(), kIOPMThermalLevelUnknown);
-			}
-		}
-		assertOnWakeSecs = 0;
-		lowBatteryCondition = false;
-		thermalEmergencyState = false;
-
-#if DEVELOPMENT || DEBUG
-		extern int g_should_log_clock_adjustments;
-		if (g_should_log_clock_adjustments) {
-			clock_sec_t  secs = 0;
-			clock_usec_t microsecs = 0;
-			uint64_t now_b = mach_absolute_time();
-
-			secs = 0;
-			microsecs = 0;
-			PEGetUTCTimeOfDay(&secs, &microsecs);
-
-			uint64_t now_a = mach_absolute_time();
-			os_log(OS_LOG_DEFAULT, "%s PMU before going to sleep %lu s %d u %llu abs_b_PEG %llu abs_a_PEG \n",
-			    __func__, (unsigned long)secs, microsecs, now_b, now_a);
-		}
+
+            // log system wake
+            getPlatform()->PMLog(kIOPMrootDomainClass, kPMLogSystemWake, 0, 0);
+            lowBatteryCondition = false;
+            lastSleepReason = 0;
+            
+            _lastDebugWakeSeconds = _debugWakeSeconds;
+            _debugWakeSeconds = 0;
+
+            // And start logging the wake event here
+            // TODO: Publish the wakeReason string as an integer
+            details = PMEventDetails::eventDetails(
+                            kIOPMEventTypeWake,
+                            NULL, 
+                            0, 
+                            kIOReturnSuccess);
+			
+            recordAndReleasePMEvent( details );
+			
+#ifndef __LP64__
+            systemWake();
 #endif
 
-		getPlatform()->sleepKernel();
-
-		// The CPU(s) are off at this point,
-		// Code will resume execution here upon wake.
-
-		clock_get_uptime(&gIOLastWakeAbsTime);
-		IOLog("gIOLastWakeAbsTime: %lld\n", gIOLastWakeAbsTime);
-#if DEVELOPMENT || DEBUG
-		record_system_event(SYSTEM_EVENT_TYPE_INFO,
-		    SYSTEM_EVENT_SUBSYSTEM_PMRD,
-		    "System State", "Waking Up"
-		    );
-#endif /* DEVELOPMENT || DEBUG */
-		_highestCapability = 0;
-
-#if HIBERNATION
-		IOHibernateSystemWake();
+#if defined(__i386__) || defined(__x86_64__)
+            wranglerTickled    = false;
+            graphicsSuppressed = false;
+            darkWakePostTickle = false;
+            logGraphicsClamp   = true;
+            logWranglerTickle  = true;
+            sleepTimerMaintenance = false;
+            wranglerTickleLatched = false;
+            darkWakeThermalAlarm  = false;
+            darkWakeThermalEmergency = false;
+
+            OSString * wakeType = OSDynamicCast(
+                OSString, getProperty(kIOPMRootDomainWakeTypeKey));
+            OSString * wakeReason = OSDynamicCast(
+                OSString, getProperty(kIOPMRootDomainWakeReasonKey));
+
+            if (wakeType && wakeType->isEqualTo(kIOPMrootDomainWakeTypeLowBattery))
+            {
+                lowBatteryCondition = true;
+                darkWakeMaintenance = true;
+                darkWakeToSleepASAP = true;
+            }
+            else if ((gDarkWakeFlags & kDarkWakeFlagHIDTickleMask) != 0)
+            {
+                OSNumber * hibOptions = OSDynamicCast(
+                    OSNumber, getProperty(kIOHibernateOptionsKey));
+
+                if (hibernateAborted || ((hibOptions &&
+                    !(hibOptions->unsigned32BitValue() & kIOHibernateOptionDarkWake))))
+                {
+                    // Hibernate aborted, or EFI brought up graphics
+                    wranglerTickled = true;
+                }
+                else
+                if (wakeType && (
+                    wakeType->isEqualTo(kIOPMRootDomainWakeTypeUser) ||
+                    wakeType->isEqualTo(kIOPMRootDomainWakeTypeAlarm)))
+                {
+                    // User wake or RTC alarm
+                    wranglerTickled = true;
+                }
+                else
+                if (wakeType &&
+                    wakeType->isEqualTo(kIOPMRootDomainWakeTypeSleepTimer))
+                {
+                    // SMC standby timer trumps SleepX
+                    darkWakeMaintenance = true;
+                    darkWakeToSleepASAP = true;
+                    sleepTimerMaintenance = true;
+                }
+                else
+                if ((_lastDebugWakeSeconds != 0) &&
+                    ((gDarkWakeFlags & kDarkWakeFlagAlarmIsDark) == 0))
+                {
+                    // SleepX before maintenance
+                    wranglerTickled = true;
+                }
+                else
+                if (wakeType &&
+                    wakeType->isEqualTo(kIOPMRootDomainWakeTypeMaintenance))
+                {
+                    darkWakeMaintenance = true;
+                    darkWakeToSleepASAP = true;
+                }
+                else
+                if (wakeType &&
+                    wakeType->isEqualTo(kIOPMRootDomainWakeTypeSleepService))
+                {
+                    darkWakeToSleepASAP = true;
+//                    darkWakeMaintenance = true; // ????
+                    darkWakeSleepService = true;
+                }
+                else
+                {
+                    // Unidentified wake source, resume to full wake if debug
+                    // alarm is pending.
+
+                    if (_lastDebugWakeSeconds &&
+                        (!wakeReason || wakeReason->isEqualTo("")))
+                        wranglerTickled = true;
+                    else
+                        darkWakeToSleepASAP = true;
+                }
+            }
+            else
+            {
+                if (wakeType &&
+                    wakeType->isEqualTo(kIOPMRootDomainWakeTypeSleepTimer))
+                {
+                    darkWakeMaintenance = true;
+                    darkWakeToSleepASAP = true;
+                    sleepTimerMaintenance = true;
+                }
+                else if (hibernateAborted || !wakeType ||
+                    !wakeType->isEqualTo(kIOPMRootDomainWakeTypeMaintenance) ||
+                    !wakeReason || !wakeReason->isEqualTo("RTC"))
+                {
+                    // Post a HID tickle immediately - except for RTC maintenance wake.
+                    wranglerTickled = true;
+                }
+                else
+                {
+                    darkWakeMaintenance = true;
+                    darkWakeToSleepASAP = true;
+                }
+            }
+
+            if (wranglerTickled)
+                reportUserInput();
+            else if (!darkWakeMaintenance)
+            {
+                // Early/late tickle for non-maintenance wake.
+                if (((gDarkWakeFlags & kDarkWakeFlagHIDTickleMask) == 
+                     kDarkWakeFlagHIDTickleEarly) ||
+                    ((gDarkWakeFlags & kDarkWakeFlagHIDTickleMask) == 
+                     kDarkWakeFlagHIDTickleLate))
+                {
+                    darkWakePostTickle = true;
+                }
+            }
+#else   /* !__i386__ && !__x86_64__ */
+            // stay awake for at least 30 seconds
+            wranglerTickled = true;
+            startIdleSleepTimer(30);
 #endif
 
-		// sleep transition complete
-		gSleepOrShutdownPending = 0;
-
-		// trip the reset of the calendar clock
-		clock_wakeup_calendar();
-		clock_get_calendar_microtime(&secs, &microsecs);
-		gIOLastWakeTime.tv_sec  = secs;
-		gIOLastWakeTime.tv_usec = microsecs;
-
-		// aot
-		if (_aotWakeTimeCalendar.selector != kPMCalendarTypeInvalid) {
-			_aotWakeTimeCalendar.selector = kPMCalendarTypeInvalid;
-			secs = 0;
-			microsecs = 0;
-			PEGetUTCTimeOfDay(&secs, &microsecs);
-			IOPMConvertSecondsToCalendar(secs, &nowCalendar);
-			IOLog("aotWake at " YMDTF " sched: " YMDTF "\n", YMDT(&nowCalendar), YMDT(&_aotWakeTimeCalendar));
-			_aotMetrics->sleepCount++;
-			_aotLastWakeTime = gIOLastWakeAbsTime;
-			if (_aotMetrics->sleepCount <= kIOPMAOTMetricsKernelWakeCountMax) {
-				_aotMetrics->kernelSleepTime[_aotMetrics->sleepCount - 1]
-				        = (((uint64_t) gIOLastSleepTime.tv_sec) << 10) + (gIOLastSleepTime.tv_usec / 1000);
-				_aotMetrics->kernelWakeTime[_aotMetrics->sleepCount - 1]
-				        = (((uint64_t) gIOLastWakeTime.tv_sec) << 10) + (gIOLastWakeTime.tv_usec / 1000);
-			}
-
-			if (_aotTestTime) {
-				if (_aotWakeTimeUTC <= secs) {
-					_aotTestTime = mach_continuous_time() + _aotTestInterval;
-				}
-				if (_aotTestTime < _aotEndTime) {
-					_setWakeTime(_aotTestTime);
-				}
-			}
-		}
-
-#if HIBERNATION
-		LOG("System %sWake\n", gIOHibernateState ? "SafeSleep " : "");
-#endif
-
-		lastSleepReason = 0;
-
-		lastDebugWakeSeconds    = _debugWakeSeconds;
-		_debugWakeSeconds       = 0;
-		_scheduledAlarmMask     = 0;
-		_nextScheduledAlarmType = NULL;
-
-		darkWakeExit            = false;
-		darkWakePowerClamped    = false;
-		darkWakePostTickle      = false;
-		darkWakeHibernateError  = false;
-		darkWakeToSleepASAP     = true;
-		darkWakeLogClamp        = true;
-		sleepTimerMaintenance   = false;
-		sleepToStandby          = false;
-		wranglerTickled         = false;
-		userWasActive           = false;
-		isRTCAlarmWake          = false;
-		clamshellIgnoreClose    = false;
-		fullWakeReason = kFullWakeReasonNone;
-		idleSleepRevertible     = true;
-
-#if defined(__i386__) || defined(__x86_64__)
-		kdebugTrace(kPMLogSystemWake, 0, 0, 0);
-
-		OSSharedPtr<OSObject> wakeTypeProp   = copyProperty(kIOPMRootDomainWakeTypeKey);
-		OSSharedPtr<OSObject> wakeReasonProp = copyProperty(kIOPMRootDomainWakeReasonKey);
-		OSString * wakeType = OSDynamicCast(OSString, wakeTypeProp.get());
-		OSString * wakeReason = OSDynamicCast(OSString, wakeReasonProp.get());
-
-		if (wakeReason && (wakeReason->getLength() >= 2) &&
-		    gWakeReasonString[0] == '\0') {
-			WAKEEVENT_LOCK();
-			// Until the platform driver can claim its wake reasons
-			strlcat(gWakeReasonString, wakeReason->getCStringNoCopy(),
-			    sizeof(gWakeReasonString));
-			if (!gWakeReasonSysctlRegistered) {
-				gWakeReasonSysctlRegistered = true;
-			}
-			WAKEEVENT_UNLOCK();
-		}
-
-		if (wakeType && wakeType->isEqualTo(kIOPMrootDomainWakeTypeLowBattery)) {
-			lowBatteryCondition = true;
-			darkWakeMaintenance = true;
-		} else {
-#if HIBERNATION
-			OSSharedPtr<OSObject> hibOptionsProp = copyProperty(kIOHibernateOptionsKey);
-			OSNumber * hibOptions = OSDynamicCast(  OSNumber, hibOptionsProp.get());
-			if (hibernateAborted || ((hibOptions &&
-			    !(hibOptions->unsigned32BitValue() & kIOHibernateOptionDarkWake)))) {
-				// Hibernate aborted, or EFI brought up graphics
-				darkWakeExit = true;
-				if (hibernateAborted) {
-					DLOG("Hibernation aborted\n");
-				} else {
-					DLOG("EFI brought up graphics. Going to full wake. HibOptions: 0x%x\n", hibOptions->unsigned32BitValue());
-				}
-			} else
-#endif
-			if (wakeType && (
-				    wakeType->isEqualTo(kIOPMRootDomainWakeTypeUser) ||
-				    wakeType->isEqualTo(kIOPMRootDomainWakeTypeAlarm))) {
-				// User wake or RTC alarm
-				darkWakeExit = true;
-				if (wakeType->isEqualTo(kIOPMRootDomainWakeTypeAlarm)) {
-					isRTCAlarmWake = true;
-				}
-			} else if (wakeType &&
-			    wakeType->isEqualTo(kIOPMRootDomainWakeTypeSleepTimer)) {
-				// SMC standby timer trumps SleepX
-				darkWakeMaintenance = true;
-				sleepTimerMaintenance = true;
-			} else if ((lastDebugWakeSeconds != 0) &&
-			    ((gDarkWakeFlags & kDarkWakeFlagAlarmIsDark) == 0)) {
-				// SleepX before maintenance
-				darkWakeExit = true;
-			} else if (wakeType &&
-			    wakeType->isEqualTo(kIOPMRootDomainWakeTypeMaintenance)) {
-				darkWakeMaintenance = true;
-			} else if (wakeType &&
-			    wakeType->isEqualTo(kIOPMRootDomainWakeTypeSleepService)) {
-				darkWakeMaintenance = true;
-				darkWakeSleepService = true;
-#if HIBERNATION
-				if (kIOHibernateStateWakingFromHibernate == gIOHibernateState) {
-					sleepToStandby = true;
-				}
-#endif
-			} else if (wakeType &&
-			    wakeType->isEqualTo(kIOPMRootDomainWakeTypeHibernateError)) {
-				darkWakeMaintenance = true;
-				darkWakeHibernateError = true;
-			} else {
-				// Unidentified wake source, resume to full wake if debug
-				// alarm is pending.
-
-				if (lastDebugWakeSeconds &&
-				    (!wakeReason || wakeReason->isEqualTo(""))) {
-					darkWakeExit = true;
-				}
-			}
-		}
-
-		if (darkWakeExit) {
-			darkWakeToSleepASAP = false;
-			fullWakeReason = kFullWakeReasonLocalUser;
-			reportUserInput();
-		} else if (displayPowerOnRequested && checkSystemCanSustainFullWake()) {
-			handleSetDisplayPowerOn(true);
-		} else if (!darkWakeMaintenance) {
-			// Early/late tickle for non-maintenance wake.
-			if ((gDarkWakeFlags & kDarkWakeFlagPromotionMask) != kDarkWakeFlagPromotionNone) {
-				darkWakePostTickle = true;
-			}
-		}
-#else   /* !__i386__ && !__x86_64__ */
-		timeSinceReset = ml_get_time_since_reset();
-		kdebugTrace(kPMLogSystemWake, 0, (uintptr_t)(timeSinceReset >> 32), (uintptr_t) timeSinceReset);
-
-		if ((gDarkWakeFlags & kDarkWakeFlagPromotionMask) == kDarkWakeFlagPromotionEarly) {
-			wranglerTickled = true;
-			fullWakeReason = kFullWakeReasonLocalUser;
-			requestUserActive(this, "Full wake on dark wake promotion boot-arg");
-		} else if ((lastDebugWakeSeconds != 0) && !(gDarkWakeFlags & kDarkWakeFlagAlarmIsDark)) {
-			isRTCAlarmWake = true;
-			fullWakeReason = kFullWakeReasonLocalUser;
-			requestUserActive(this, "RTC debug alarm");
-		} else {
-#if HIBERNATION
-			OSSharedPtr<OSObject> hibOptionsProp = copyProperty(kIOHibernateOptionsKey);
-			OSNumber * hibOptions = OSDynamicCast(OSNumber, hibOptionsProp.get());
-			if (hibOptions && !(hibOptions->unsigned32BitValue() & kIOHibernateOptionDarkWake)) {
-				fullWakeReason = kFullWakeReasonLocalUser;
-				requestUserActive(this, "hibernate user wake");
-			}
-#endif
-		}
-
-		// stay awake for at least 30 seconds
-		startIdleSleepTimer(30 * 1000);
-#endif
-		sleepCnt++;
-
-		thread_call_enter(updateConsoleUsersEntry);
-
-		// Skip AOT_STATE if we are waking up from an RTC timer.
-		// This check needs to be done after the epoch change is processed
-		// and before the changePowerStateWithTagToPriv() call below.
-		WAKEEVENT_LOCK();
-		aotShouldExit(false);
-		unsigned long newState = getRUN_STATE();
-		if (AOT_STATE == newState) {
-			if (gLPWFlags) {
-				_aotRunMode = gLPWFlags | _aotWakeEventRunMode;
-			}
-			IOLog("_aotRunMode = 0x%llx|0x%llx\n", gLPWFlags, _aotWakeEventRunMode);
-		}
-		WAKEEVENT_UNLOCK();
-
-		changePowerStateWithTagToPriv(newState, kCPSReasonWake);
-		break;
-	}
-#if !__i386__ && !__x86_64__
-	case ON_STATE:
-	case AOT_STATE:
-	{
-		DLOG("Force re-evaluating aggressiveness\n");
-		/* Force re-evaluate the aggressiveness values to set appropriate idle sleep timer */
-		pmPowerStateQueue->submitPowerEvent(
-			kPowerEventPolicyStimulus,
-			(void *) kStimulusNoIdleSleepPreventers );
-
-		// After changing to ON_STATE, invalidate any previously queued
-		// request to change to a state less than ON_STATE. This isn't
-		// necessary for AOT_STATE or if the device has only one running
-		// state since the changePowerStateToPriv() issued at the tail
-		// end of SLEEP_STATE case should take care of that.
-		if (getPowerState() == ON_STATE) {
-			changePowerStateWithTagToPriv(ON_STATE, kCPSReasonWake);
-		}
-		break;
-	}
-#endif /* !__i386__ && !__x86_64__ */
-	}
-	notifierThread = NULL;
+            changePowerStateToPriv(ON_STATE);
+        }   break;
+
+        case ON_STATE: {
+            if (previousPowerState != ON_STATE)
+            {
+                details = PMEventDetails::eventDetails(
+                                kIOPMEventTypeWakeDone,
+                                NULL, 
+                                0, 
+                                kIOReturnSuccess);
+                
+                recordAndReleasePMEvent( details );
+            }
+        }   break;
+    }
 }
 
 //******************************************************************************
@@ -3467,132 +2358,70 @@
 // Extend implementation in IOService. Running on PM work loop thread.
 //******************************************************************************
 
-IOReturn
-IOPMrootDomain::requestPowerDomainState(
-	IOPMPowerFlags      childDesire,
-	IOPowerConnection * childConnection,
-	unsigned long       specification )
-{
-	// Idle and system sleep prevention flags affects driver desire.
-	// Children desire are irrelevant so they are cleared.
-
-	return super::requestPowerDomainState(0, childConnection, specification);
-}
-
-
-static void
-makeSleepPreventersListLog(const OSSharedPtr<OSSet> &preventers, char *buf, size_t buf_size)
-{
-	if (!preventers->getCount()) {
-		return;
-	}
-
-	char *buf_iter = buf + strlen(buf);
-	char *buf_end = buf + buf_size;
-
-	OSSharedPtr<OSCollectionIterator> iterator = OSCollectionIterator::withCollection(preventers.get());
-	OSObject *obj = NULL;
-
-	while ((obj = iterator->getNextObject())) {
-		IOService *srv = OSDynamicCast(IOService, obj);
-		if (buf_iter < buf_end) {
-			buf_iter += snprintf(buf_iter, buf_end - buf_iter, " %s", srv->getName());
-		} else {
-			DLOG("Print buffer exhausted for sleep preventers list\n");
-			break;
-		}
-	}
+IOReturn IOPMrootDomain::requestPowerDomainState (
+    IOPMPowerFlags      childDesire,
+    IOPowerConnection * childConnection,
+    unsigned long       specification )
+{
+    // Idle and system sleep prevention flags affects driver desire.
+    // Children desire are irrelevant so they are cleared.
+
+    return super::requestPowerDomainState(0, childConnection, specification);
 }
 
 //******************************************************************************
 // updatePreventIdleSleepList
 //
 // Called by IOService on PM work loop.
-// Returns true if PM policy recognized the driver's desire to prevent idle
-// sleep and updated the list of idle sleep preventers. Returns false otherwise
-//******************************************************************************
-
-bool
-IOPMrootDomain::updatePreventIdleSleepList(
-	IOService * service, bool addNotRemove)
-{
-	unsigned int oldCount;
-
-	oldCount = idleSleepPreventersCount();
-	return updatePreventIdleSleepListInternal(service, addNotRemove, oldCount);
-}
-
-bool
-IOPMrootDomain::updatePreventIdleSleepListInternal(
-	IOService * service, bool addNotRemove, unsigned int oldCount)
-{
-	unsigned int newCount;
-
-	ASSERT_GATED();
-
-#if defined(XNU_TARGET_OS_OSX)
-	// Only the display wrangler and no-idle-sleep kernel assertions
-	// can prevent idle sleep. The kIOPMPreventIdleSleep capability flag
-	// reported by drivers in their power state table is ignored.
-	if (service && (service != wrangler) && (service != this)) {
-		return false;
-	}
-#endif
-
-	if (service) {
-		if (addNotRemove) {
-			preventIdleSleepList->setObject(service);
-			DLOG("Added %s to idle sleep preventers list (Total %u)\n",
-			    service->getName(), preventIdleSleepList->getCount());
-		} else if (preventIdleSleepList->member(service)) {
-			preventIdleSleepList->removeObject(service);
-			DLOG("Removed %s from idle sleep preventers list (Total %u)\n",
-			    service->getName(), preventIdleSleepList->getCount());
-		}
-
-		if (preventIdleSleepList->getCount()) {
-			char buf[256] = "Idle Sleep Preventers:";
-			makeSleepPreventersListLog(preventIdleSleepList, buf, sizeof(buf));
-			DLOG("%s\n", buf);
-		}
-	}
-
-	newCount = idleSleepPreventersCount();
-
-	if ((oldCount == 0) && (newCount != 0)) {
-		// Driver added to empty prevent list.
-		// Update the driver desire to prevent idle sleep.
-		// Driver desire does not prevent demand sleep.
-
-		changePowerStateWithTagTo(getRUN_STATE(), kCPSReasonIdleSleepPrevent);
-	} else if ((oldCount != 0) && (newCount == 0)) {
-		// Last driver removed from prevent list.
-		// Drop the driver clamp to allow idle sleep.
-
-		changePowerStateWithTagTo(SLEEP_STATE, kCPSReasonIdleSleepAllow);
-		evaluatePolicy( kStimulusNoIdleSleepPreventers );
-	}
-	messageClient(kIOPMMessageIdleSleepPreventers, systemCapabilityNotifier.get(),
-	    &newCount, sizeof(newCount));
-
-#if defined(XNU_TARGET_OS_OSX)
-	if (addNotRemove && (service == wrangler) && !checkSystemCanSustainFullWake()) {
-		DLOG("Cannot cancel idle sleep\n");
-		return false; // do not idle-cancel
-	}
-#endif
-
-	return true;
-}
-
-//******************************************************************************
-// startSpinDump
-//******************************************************************************
-
-void
-IOPMrootDomain::startSpinDump(uint32_t spindumpKind)
-{
-	messageClients(kIOPMMessageLaunchBootSpinDump, (void *)(uintptr_t)spindumpKind);
+//******************************************************************************
+
+void IOPMrootDomain::updatePreventIdleSleepList(
+        IOService * service, bool addNotRemove )
+{
+    unsigned int oldCount, newCount;
+
+    ASSERT_GATED();
+
+    // Disregard disk I/O (anything besides the display wrangler)
+    // as a factor preventing idle sleep,except in the case of legacy disk I/O
+
+    if ((gDarkWakeFlags & kDarkWakeFlagIgnoreDiskIOAlways) &&
+        addNotRemove && (service != wrangler) && (service != this))
+    {
+        return;
+    }
+
+    oldCount = preventIdleSleepList->getCount();
+    if (addNotRemove)
+    {
+        preventIdleSleepList->setObject(service);
+        DLOG("prevent idle sleep list: %s+ (%u)\n",
+            service->getName(), preventIdleSleepList->getCount());
+    }
+    else if (preventIdleSleepList->member(service))
+    {
+        preventIdleSleepList->removeObject(service);
+        DLOG("prevent idle sleep list: %s- (%u)\n",
+            service->getName(), preventIdleSleepList->getCount());
+    }
+    newCount = preventIdleSleepList->getCount();
+    
+    if ((oldCount == 0) && (newCount != 0))
+    {
+        // Driver added to empty prevent list.
+        // Update the driver desire to prevent idle sleep.
+        // Driver desire does not prevent demand sleep.
+        
+        changePowerStateTo(ON_STATE);
+    }
+    else if ((oldCount != 0) && (newCount == 0))
+    {
+        // Last driver removed from prevent list.
+        // Drop the driver clamp to allow idle sleep.
+
+        changePowerStateTo(SLEEP_STATE);
+        evaluatePolicy( kStimulusNoIdleSleepPreventers );
+    }
 }
 
 //******************************************************************************
@@ -3601,166 +2430,35 @@
 // Called by IOService on PM work loop.
 //******************************************************************************
 
-void
-IOPMrootDomain::updatePreventSystemSleepList(
-	IOService * service, bool addNotRemove )
-{
-	unsigned int oldCount, newCount;
-
-	ASSERT_GATED();
-	if (this == service) {
-		return;
-	}
-
-	oldCount = preventSystemSleepList->getCount();
-	if (addNotRemove) {
-		preventSystemSleepList->setObject(service);
-		DLOG("Added %s to system sleep preventers list (Total %u)\n",
-		    service->getName(), preventSystemSleepList->getCount());
-		if (!assertOnWakeSecs && gIOLastWakeAbsTime) {
-			AbsoluteTime    now;
-			clock_usec_t    microsecs;
-			clock_get_uptime(&now);
-			SUB_ABSOLUTETIME(&now, &gIOLastWakeAbsTime);
-			absolutetime_to_microtime(now, &assertOnWakeSecs, &microsecs);
-			if (assertOnWakeReport) {
-				HISTREPORT_TALLYVALUE(assertOnWakeReport, (int64_t)assertOnWakeSecs);
-				DLOG("Updated assertOnWake %lu\n", (unsigned long)assertOnWakeSecs);
-			}
-		}
-	} else if (preventSystemSleepList->member(service)) {
-		preventSystemSleepList->removeObject(service);
-		DLOG("Removed %s from system sleep preventers list (Total %u)\n",
-		    service->getName(), preventSystemSleepList->getCount());
-
-		if ((oldCount != 0) && (preventSystemSleepList->getCount() == 0)) {
-			// Lost all system sleep preventers.
-			// Send stimulus if system sleep was blocked, and is in dark wake.
-			evaluatePolicy( kStimulusDarkWakeEvaluate );
-		}
-	}
-
-	newCount = preventSystemSleepList->getCount();
-	if (newCount) {
-		char buf[256] = "System Sleep Preventers:";
-		makeSleepPreventersListLog(preventSystemSleepList, buf, sizeof(buf));
-		DLOG("%s\n", buf);
-	}
-
-	messageClient(kIOPMMessageSystemSleepPreventers, systemCapabilityNotifier.get(),
-	    &newCount, sizeof(newCount));
-}
-
-void
-IOPMrootDomain::copySleepPreventersList(OSArray **idleSleepList, OSArray **systemSleepList)
-{
-	OSSharedPtr<OSCollectionIterator> iterator;
-	OSObject    *object = NULL;
-	OSSharedPtr<OSArray>     array;
-
-	if (!gIOPMWorkLoop->inGate()) {
-		gIOPMWorkLoop->runAction(
-			OSMemberFunctionCast(IOWorkLoop::Action, this,
-			&IOPMrootDomain::IOPMrootDomain::copySleepPreventersList),
-			this, (void *)idleSleepList, (void *)systemSleepList);
-		return;
-	}
-
-	if (idleSleepList && preventIdleSleepList && (preventIdleSleepList->getCount() != 0)) {
-		iterator = OSCollectionIterator::withCollection(preventIdleSleepList.get());
-		array = OSArray::withCapacity(5);
-
-		if (iterator && array) {
-			while ((object = iterator->getNextObject())) {
-				IOService *service = OSDynamicCast(IOService, object);
-				if (service) {
-					OSSharedPtr<const OSSymbol> name = service->copyName();
-					if (name) {
-						array->setObject(name.get());
-					}
-				}
-			}
-		}
-		*idleSleepList = array.detach();
-	}
-
-	if (systemSleepList && preventSystemSleepList && (preventSystemSleepList->getCount() != 0)) {
-		iterator = OSCollectionIterator::withCollection(preventSystemSleepList.get());
-		array = OSArray::withCapacity(5);
-
-		if (iterator && array) {
-			while ((object = iterator->getNextObject())) {
-				IOService *service = OSDynamicCast(IOService, object);
-				if (service) {
-					OSSharedPtr<const OSSymbol> name = service->copyName();
-					if (name) {
-						array->setObject(name.get());
-					}
-				}
-			}
-		}
-		*systemSleepList = array.detach();
-	}
-}
-
-void
-IOPMrootDomain::copySleepPreventersListWithID(OSArray **idleSleepList, OSArray **systemSleepList)
-{
-	OSSharedPtr<OSCollectionIterator> iterator;
-	OSObject    *object = NULL;
-	OSSharedPtr<OSArray>     array;
-
-	if (!gIOPMWorkLoop->inGate()) {
-		gIOPMWorkLoop->runAction(
-			OSMemberFunctionCast(IOWorkLoop::Action, this,
-			&IOPMrootDomain::IOPMrootDomain::copySleepPreventersListWithID),
-			this, (void *)idleSleepList, (void *)systemSleepList);
-		return;
-	}
-
-	if (idleSleepList && preventIdleSleepList && (preventIdleSleepList->getCount() != 0)) {
-		iterator = OSCollectionIterator::withCollection(preventIdleSleepList.get());
-		array = OSArray::withCapacity(5);
-
-		if (iterator && array) {
-			while ((object = iterator->getNextObject())) {
-				IOService *service = OSDynamicCast(IOService, object);
-				if (service) {
-					OSSharedPtr<OSDictionary> dict = OSDictionary::withCapacity(2);
-					OSSharedPtr<const OSSymbol> name = service->copyName();
-					OSSharedPtr<OSNumber> id = OSNumber::withNumber(service->getRegistryEntryID(), 64);
-					if (dict && name && id) {
-						dict->setObject(kIOPMDriverAssertionRegistryEntryIDKey, id.get());
-						dict->setObject(kIOPMDriverAssertionOwnerStringKey, name.get());
-						array->setObject(dict.get());
-					}
-				}
-			}
-		}
-		*idleSleepList = array.detach();
-	}
-
-	if (systemSleepList && preventSystemSleepList && (preventSystemSleepList->getCount() != 0)) {
-		iterator = OSCollectionIterator::withCollection(preventSystemSleepList.get());
-		array = OSArray::withCapacity(5);
-
-		if (iterator && array) {
-			while ((object = iterator->getNextObject())) {
-				IOService *service = OSDynamicCast(IOService, object);
-				if (service) {
-					OSSharedPtr<OSDictionary> dict = OSDictionary::withCapacity(2);
-					OSSharedPtr<const OSSymbol> name = service->copyName();
-					OSSharedPtr<OSNumber> id = OSNumber::withNumber(service->getRegistryEntryID(), 64);
-					if (dict && name && id) {
-						dict->setObject(kIOPMDriverAssertionRegistryEntryIDKey, id.get());
-						dict->setObject(kIOPMDriverAssertionOwnerStringKey, name.get());
-						array->setObject(dict.get());
-					}
-				}
-			}
-		}
-		*systemSleepList = array.detach();
-	}
+void IOPMrootDomain::updatePreventSystemSleepList(
+        IOService * service, bool addNotRemove )
+{
+    unsigned int oldCount;
+
+    ASSERT_GATED();
+    if (this == service)
+        return;
+
+    oldCount = preventSystemSleepList->getCount();
+    if (addNotRemove)
+    {
+        preventSystemSleepList->setObject(service);
+        DLOG("prevent system sleep list: %s+ (%u)\n",
+            service->getName(), preventSystemSleepList->getCount());
+    }
+    else if (preventSystemSleepList->member(service))
+    {
+        preventSystemSleepList->removeObject(service);
+        DLOG("prevent system sleep list: %s- (%u)\n",
+            service->getName(), preventSystemSleepList->getCount());
+
+        if ((oldCount != 0) && (preventSystemSleepList->getCount() == 0))
+        {
+            // Lost all system sleep preventers.
+            // Send stimulus if system sleep was blocked, and is in dark wake.
+            evaluatePolicy( kStimulusDarkWakeEvaluate );
+        }
+    }
 }
 
 //******************************************************************************
@@ -3769,41 +2467,43 @@
 // Override the superclass implementation to send a different message type.
 //******************************************************************************
 
-bool
-IOPMrootDomain::tellChangeDown( unsigned long stateNum )
-{
-	DLOG("tellChangeDown %s->%s\n",
-	    getPowerStateString((uint32_t) getPowerState()), getPowerStateString((uint32_t) stateNum));
-
-	if (SLEEP_STATE == stateNum) {
-		// Legacy apps were already told in the full->dark transition
-		if (!ignoreTellChangeDown) {
-			tracePoint( kIOPMTracePointSleepApplications );
-		} else {
-			tracePoint( kIOPMTracePointSleepPriorityClients );
-		}
-	}
-
-	if (!ignoreTellChangeDown) {
-		userActivityAtSleep = userActivityCount;
-		DLOG("tellChangeDown::userActivityAtSleep %d\n", userActivityAtSleep);
-
-		if (SLEEP_STATE == stateNum) {
-			hibernateAborted = false;
-
-			// Direct callout into OSKext so it can disable kext unloads
-			// during sleep/wake to prevent deadlocks.
-			OSKextSystemSleepOrWake( kIOMessageSystemWillSleep );
-
-			IOService::updateConsoleUsers(NULL, kIOMessageSystemWillSleep);
-
-			// Two change downs are sent by IOServicePM. Ignore the 2nd.
-			// But tellClientsWithResponse() must be called for both.
-			ignoreTellChangeDown = true;
-		}
-	}
-
-	return super::tellClientsWithResponse( kIOMessageSystemWillSleep );
+bool IOPMrootDomain::tellChangeDown( unsigned long stateNum )
+{
+    DLOG("tellChangeDown %u->%u\n",
+        (uint32_t) getPowerState(), (uint32_t) stateNum);
+
+    if (SLEEP_STATE == stateNum)
+    {
+        if (!ignoreTellChangeDown)
+            tracePoint( kIOPMTracePointSleepApplications );
+        else
+            tracePoint( kIOPMTracePointSleepPriorityClients );   
+    }
+
+    if ((SLEEP_STATE == stateNum) && !ignoreTellChangeDown)
+    {
+        userActivityAtSleep = userActivityCount;
+        hibernateAborted = false;
+        DLOG("tellChangeDown::userActivityAtSleep %d\n", userActivityAtSleep);
+
+        // Direct callout into OSKext so it can disable kext unloads
+        // during sleep/wake to prevent deadlocks.
+        OSKextSystemSleepOrWake( kIOMessageSystemWillSleep );
+
+        IOService::updateConsoleUsers(NULL, kIOMessageSystemWillSleep);
+
+        // Notify platform that sleep has begun
+        getPlatform()->callPlatformFunction(
+                        sleepMessagePEFunction, false,
+                        (void *)(uintptr_t) kIOMessageSystemWillSleep,
+                        NULL, NULL, NULL);
+
+        // Two change downs are sent by IOServicePM. Ignore the 2nd.
+        // But tellClientsWithResponse() must be called for both.
+        ignoreTellChangeDown = true;
+    }
+
+    return super::tellClientsWithResponse( kIOMessageSystemWillSleep );
 }
 
 //******************************************************************************
@@ -3813,128 +2513,48 @@
 // This must be idle sleep since we don't ask during any other power change.
 //******************************************************************************
 
-bool
-IOPMrootDomain::askChangeDown( unsigned long stateNum )
-{
-	DLOG("askChangeDown %s->%s\n",
-	    getPowerStateString((uint32_t) getPowerState()), getPowerStateString((uint32_t) stateNum));
-
-	// Don't log for dark wake entry
-	if (kSystemTransitionSleep == _systemTransitionType) {
-		tracePoint( kIOPMTracePointSleepApplications );
-	}
-
-	return super::tellClientsWithResponse( kIOMessageCanSystemSleep );
+bool IOPMrootDomain::askChangeDown( unsigned long stateNum )
+{
+    DLOG("askChangeDown %u->%u\n",
+        (uint32_t) getPowerState(), (uint32_t) stateNum);
+
+    // Don't log for dark wake entry
+    if (kSystemTransitionSleep == _systemTransitionType)
+        tracePoint( kIOPMTracePointSleepApplications );
+
+    return super::tellClientsWithResponse( kIOMessageCanSystemSleep );
 }
 
 //******************************************************************************
 // askChangeDownDone
 //
-// An opportunity for root domain to cancel the power transition,
-// possibily due to an assertion created by powerd in response to
-// kIOMessageCanSystemSleep.
-//
-// Idle sleep:
-//   full -> dark wake transition
-//     1. Notify apps and powerd with kIOMessageCanSystemSleep
-//     2. askChangeDownDone()
-//   dark -> sleep transition
-//     1. Notify powerd with kIOMessageCanSystemSleep
-//     2. askChangeDownDone()
-//
-// Demand sleep:
-//   full -> dark wake transition
-//     1. Notify powerd with kIOMessageCanSystemSleep
-//     2. askChangeDownDone()
-//   dark -> sleep transition
-//     1. Notify powerd with kIOMessageCanSystemSleep
-//     2. askChangeDownDone()
-//******************************************************************************
-
-void
-IOPMrootDomain::askChangeDownDone(
-	IOPMPowerChangeFlags * inOutChangeFlags, bool * cancel )
-{
-	DLOG("askChangeDownDone(0x%x, %u) type %x, cap %x->%x\n",
-	    *inOutChangeFlags, *cancel,
-	    _systemTransitionType,
-	    _currentCapability, _pendingCapability);
-
-	if ((false == *cancel) && (kSystemTransitionSleep == _systemTransitionType)) {
-		// Dark->Sleep transition.
-		// Check if there are any deny sleep assertions.
-		// lastSleepReason already set by handleOurPowerChangeStart()
-
-		if (!checkSystemCanSleep(lastSleepReason)) {
-			// Cancel dark wake to sleep transition.
-			// Must re-scan assertions upon entering dark wake.
-
-			*cancel = true;
-			DLOG("cancel dark->sleep\n");
-		}
-		if (_aotMode && (kPMCalendarTypeInvalid != _aotWakeTimeCalendar.selector)) {
-			uint64_t now = mach_continuous_time();
-			if (((now + _aotWakePreWindow) >= _aotWakeTimeContinuous)
-			    && (now < (_aotWakeTimeContinuous + _aotWakePostWindow))) {
-				*cancel = true;
-				IOLog("AOT wake window cancel: %qd, %qd\n", now, _aotWakeTimeContinuous);
-			}
-		}
-	}
-}
-
-//******************************************************************************
-// systemDidNotSleep
-//
-// Work common to both canceled or aborted sleep.
-//******************************************************************************
-
-void
-IOPMrootDomain::systemDidNotSleep( void )
-{
-	// reset console lock state
-	thread_call_enter(updateConsoleUsersEntry);
-
-	if (idleSleepEnabled) {
-		if (!wrangler) {
-#if defined(XNU_TARGET_OS_OSX) && !DISPLAY_WRANGLER_PRESENT
-			startIdleSleepTimer(kIdleSleepRetryInterval);
-#else
-			startIdleSleepTimer(idleMilliSeconds);
-#endif
-		} else if (!userIsActive) {
-			// Manually start the idle sleep timer besides waiting for
-			// the user to become inactive.
-			startIdleSleepTimer(kIdleSleepRetryInterval);
-		}
-	}
-
-	preventTransitionToUserActive(false);
-	IOService::setAdvisoryTickleEnable( true );
-	idleSleepRevertible = true;
-
-	// After idle revert and cancel, send a did-change message to powerd
-	// to balance the previous will-change message. Kernel clients do not
-	// need this since sleep cannot be canceled once they are notified.
-
-	if (toldPowerdCapWillChange && systemCapabilityNotifier &&
-	    (_pendingCapability != _currentCapability) &&
-	    ((_systemMessageClientMask & kSystemMessageClientPowerd) != 0)) {
-		// Differs from a real capability gain change where notifyRef != 0,
-		// but it is zero here since no response is expected.
-
-		IOPMSystemCapabilityChangeParameters params;
-
-		bzero(&params, sizeof(params));
-		params.fromCapabilities = _pendingCapability;
-		params.toCapabilities = _currentCapability;
-		params.changeFlags = kIOPMSystemCapabilityDidChange;
-
-		DLOG("MESG cap %x->%x did change\n",
-		    params.fromCapabilities, params.toCapabilities);
-		messageClient(kIOMessageSystemCapabilityChange, systemCapabilityNotifier.get(),
-		    &params, sizeof(params));
-	}
+// Called by PM after all apps have responded to kIOMessageCanSystemSleep.
+// pmconfigd may create a deny sleep assertion before ack'ing.
+//******************************************************************************
+
+void IOPMrootDomain::askChangeDownDone(
+        IOPMPowerChangeFlags * inOutChangeFlags, bool * cancel )
+{
+    DLOG("askChangeDownDone(0x%x, %u) type %x, cap %x->%x\n",
+        *inOutChangeFlags, *cancel,
+        _systemTransitionType,
+        _currentCapability, _pendingCapability);
+
+    if ((false == *cancel) && (kSystemTransitionSleep == _systemTransitionType))
+    {
+        // Dark->Sleep transition.
+        // Check if there are any deny sleep assertions.
+        // Full->Dark transition is never cancelled.
+
+        if (!checkSystemCanSleep(true))
+        {
+            // Cancel dark wake to sleep transition.
+            // Must re-scan assertions upon entering dark wake.
+
+            *cancel = true;
+            DLOG("cancel dark->sleep\n");
+        }
+    }
 }
 
 //******************************************************************************
@@ -3949,17 +2569,20 @@
 // This must be a vetoed idle sleep, since no other power change can be vetoed.
 //******************************************************************************
 
-void
-IOPMrootDomain::tellNoChangeDown( unsigned long stateNum )
-{
-	DLOG("tellNoChangeDown %s->%s\n",
-	    getPowerStateString((uint32_t) getPowerState()), getPowerStateString((uint32_t) stateNum));
+void IOPMrootDomain::tellNoChangeDown( unsigned long stateNum )
+{
+    DLOG("tellNoChangeDown %u->%u\n",
+        (uint32_t) getPowerState(), (uint32_t) stateNum);
 
 	// Sleep canceled, clear the sleep trace point.
-	tracePoint(kIOPMTracePointSystemUp);
-
-	systemDidNotSleep();
-	return tellClients( kIOMessageSystemWillNotSleep );
+    tracePoint(kIOPMTracePointSystemUp);
+
+    if (idleSeconds && !wrangler)
+    {
+        // stay awake for at least idleSeconds
+        startIdleSleepTimer(idleSeconds);
+    }
+    return tellClients( kIOMessageSystemWillNotSleep );
 }
 
 //******************************************************************************
@@ -3971,74 +2594,49 @@
 // type to the client or application being notified.
 //******************************************************************************
 
-void
-IOPMrootDomain::tellChangeUp( unsigned long stateNum )
-{
-	DLOG("tellChangeUp %s->%s\n",
-	    getPowerStateString((uint32_t) getPowerState()), getPowerStateString((uint32_t) stateNum));
-
-	ignoreTellChangeDown = false;
-
-	if (stateNum == ON_STATE) {
-		// Direct callout into OSKext so it can disable kext unloads
-		// during sleep/wake to prevent deadlocks.
-		OSKextSystemSleepOrWake( kIOMessageSystemHasPoweredOn );
-
-		// Notify platform that sleep was cancelled or resumed.
-		getPlatform()->callPlatformFunction(
-			sleepMessagePEFunction.get(), false,
-			(void *)(uintptr_t) kIOMessageSystemHasPoweredOn,
-			NULL, NULL, NULL);
-
-		if (getPowerState() == ON_STATE) {
-			// Sleep was cancelled by idle cancel or revert
-			if (!CAP_CURRENT(kIOPMSystemCapabilityGraphics)) {
-				// rdar://problem/50363791
-				// If system is in dark wake and sleep is cancelled, do not
-				// send SystemWillPowerOn/HasPoweredOn messages to kernel
-				// priority clients. They haven't yet seen a SystemWillSleep
-				// message before the cancellation. So make sure the kernel
-				// client bit is cleared in _systemMessageClientMask before
-				// invoking the tellClients() below. This bit may have been
-				// set by handleOurPowerChangeStart() anticipating a successful
-				// sleep and setting the filter mask ahead of time allows the
-				// SystemWillSleep message to go through.
-				_systemMessageClientMask &= ~kSystemMessageClientKernel;
-			}
-
-			systemDidNotSleep();
-			tellClients( kIOMessageSystemWillPowerOn );
-		}
-
-		tracePoint( kIOPMTracePointWakeApplications );
-		tellClients( kIOMessageSystemHasPoweredOn );
-	} else if (stateNum == AOT_STATE) {
-		if (getPowerState() == AOT_STATE) {
-			// Sleep was cancelled by idle cancel or revert
-			startIdleSleepTimer(idleMilliSeconds);
-		}
-	}
-}
-
-#define CAP_WILL_CHANGE_TO_OFF(params, flag) \
-    (((params)->changeFlags & kIOPMSystemCapabilityWillChange) && \
-     ((params)->fromCapabilities & (flag)) && \
-     (((params)->toCapabilities & (flag)) == 0))
-
-#define CAP_DID_CHANGE_TO_ON(params, flag) \
-    (((params)->changeFlags & kIOPMSystemCapabilityDidChange) && \
-     ((params)->toCapabilities & (flag)) && \
-     (((params)->fromCapabilities & (flag)) == 0))
-
-#define CAP_DID_CHANGE_TO_OFF(params, flag) \
-    (((params)->changeFlags & kIOPMSystemCapabilityDidChange) && \
-     ((params)->fromCapabilities & (flag)) && \
-     (((params)->toCapabilities & (flag)) == 0))
-
-#define CAP_WILL_CHANGE_TO_ON(params, flag) \
-    (((params)->changeFlags & kIOPMSystemCapabilityWillChange) && \
-     ((params)->toCapabilities & (flag)) && \
-     (((params)->fromCapabilities & (flag)) == 0))
+void IOPMrootDomain::tellChangeUp( unsigned long stateNum )
+{
+
+    DLOG("tellChangeUp %u->%u\n",
+        (uint32_t) getPowerState(), (uint32_t) stateNum);
+
+    ignoreTellChangeDown = false;
+
+    if ( stateNum == ON_STATE )
+    {
+        // Direct callout into OSKext so it can disable kext unloads
+        // during sleep/wake to prevent deadlocks.
+        OSKextSystemSleepOrWake( kIOMessageSystemHasPoweredOn );
+
+        // Notify platform that sleep was cancelled or resumed.
+        getPlatform()->callPlatformFunction(
+                        sleepMessagePEFunction, false,
+                        (void *)(uintptr_t) kIOMessageSystemHasPoweredOn,
+                        NULL, NULL, NULL);
+
+        if (getPowerState() == ON_STATE)
+        {
+            // this is a quick wake from aborted sleep
+            if (idleSeconds && !wrangler)
+            {
+                // stay awake for at least idleSeconds
+                startIdleSleepTimer(idleSeconds);
+            }
+            tellClients( kIOMessageSystemWillPowerOn );
+        }
+
+        tracePoint( kIOPMTracePointWakeApplications );
+
+        if (pmStatsAppResponses) 
+        {
+            setProperty(kIOPMSleepStatisticsAppsKey, pmStatsAppResponses);
+            pmStatsAppResponses->release();
+            pmStatsAppResponses = OSArray::withCapacity(5);
+        }
+
+        tellClients( kIOMessageSystemHasPoweredOn );
+    }
+}
 
 //******************************************************************************
 // sysPowerDownHandler
@@ -4046,106 +2644,86 @@
 // Perform a vfs sync before system sleep.
 //******************************************************************************
 
-IOReturn
-IOPMrootDomain::sysPowerDownHandler(
-	void * target, void * refCon,
-	UInt32 messageType, IOService * service,
-	void * messageArgs, vm_size_t argSize )
-{
-	static UInt32 lastSystemMessageType = 0;
-	IOReturn    ret = 0;
-
-	DLOG("sysPowerDownHandler message %s\n", getIOMessageString(messageType));
-
-	// rdar://problem/50363791
-	// Sanity check to make sure the SystemWill/Has message types are
-	// received in the expected order for all kernel priority clients.
-	if (messageType == kIOMessageSystemWillSleep ||
-	    messageType == kIOMessageSystemWillPowerOn ||
-	    messageType == kIOMessageSystemHasPoweredOn) {
-		switch (messageType) {
-		case kIOMessageSystemWillPowerOn:
-			assert(lastSystemMessageType == kIOMessageSystemWillSleep);
-			break;
-		case kIOMessageSystemHasPoweredOn:
-			assert(lastSystemMessageType == kIOMessageSystemWillPowerOn);
-			break;
-		}
-
-		lastSystemMessageType = messageType;
-	}
-
-	if (!gRootDomain) {
-		return kIOReturnUnsupported;
-	}
-
-	if (messageType == kIOMessageSystemCapabilityChange) {
-		IOPMSystemCapabilityChangeParameters * params =
-		    (IOPMSystemCapabilityChangeParameters *) messageArgs;
-
-		// Interested applications have been notified of an impending power
-		// change and have acked (when applicable).
-		// This is our chance to save whatever state we can before powering
-		// down.
-		// We call sync_internal defined in xnu/bsd/vfs/vfs_syscalls.c,
-		// via callout
-
-		DLOG("sysPowerDownHandler cap %x -> %x (flags %x)\n",
-		    params->fromCapabilities, params->toCapabilities,
-		    params->changeFlags);
-
-		if (CAP_WILL_CHANGE_TO_OFF(params, kIOPMSystemCapabilityCPU)) {
-			// We will ack within 20 seconds
-			params->maxWaitForReply = 20 * 1000 * 1000;
-
-#if HIBERNATION
-			gRootDomain->evaluateSystemSleepPolicyEarly();
-
-			// add in time we could spend freeing pages
-			if (gRootDomain->hibernateMode && !gRootDomain->hibernateDisabled) {
-				params->maxWaitForReply = kCapabilityClientMaxWait;
-			}
-			DLOG("sysPowerDownHandler max wait %d s\n",
-			    (int) (params->maxWaitForReply / 1000 / 1000));
+IOReturn IOPMrootDomain::sysPowerDownHandler(
+    void * target, void * refCon,
+    UInt32 messageType, IOService * service,
+    void * messageArgs, vm_size_t argSize )
+{
+    IOReturn    ret;
+
+    DLOG("sysPowerDownHandler message %s\n", getIOMessageString(messageType));
+
+    if (!gRootDomain)
+        return kIOReturnUnsupported;
+
+    if (messageType == kIOMessageSystemCapabilityChange)
+    {
+        IOPMSystemCapabilityChangeParameters * params =
+            (IOPMSystemCapabilityChangeParameters *) messageArgs;
+
+        // Interested applications have been notified of an impending power
+        // change and have acked (when applicable).
+        // This is our chance to save whatever state we can before powering
+        // down.
+        // We call sync_internal defined in xnu/bsd/vfs/vfs_syscalls.c,
+        // via callout
+
+        DLOG("sysPowerDownHandler cap %x -> %x (flags %x)\n",
+            params->fromCapabilities, params->toCapabilities,
+            params->changeFlags);
+
+        if ((params->changeFlags & kIOPMSystemCapabilityWillChange) &&
+            (params->fromCapabilities & kIOPMSystemCapabilityCPU) &&
+            (params->toCapabilities & kIOPMSystemCapabilityCPU) == 0)
+        {
+            // We will ack within 20 seconds
+            params->maxWaitForReply = 20 * 1000 * 1000;
+#if	HIBERNATION
+            gRootDomain->evaluateSystemSleepPolicyEarly();
+
+            // add in time we could spend freeing pages
+            if (gRootDomain->hibernateMode && !gRootDomain->hibernateDisabled)
+            {
+                params->maxWaitForReply = kCapabilityClientMaxWait;
+            }
+            DLOG("sysPowerDownHandler timeout %d s\n", (int) (params->maxWaitForReply / 1000 / 1000));
 #endif
 
-			// Notify platform that sleep has begun, after the early
-			// sleep policy evaluation.
-			getPlatform()->callPlatformFunction(
-				sleepMessagePEFunction.get(), false,
-				(void *)(uintptr_t) kIOMessageSystemWillSleep,
-				NULL, NULL, NULL);
-
-			if (!OSCompareAndSwap( 0, 1, &gSleepOrShutdownPending )) {
-				// Purposely delay the ack and hope that shutdown occurs quickly.
-				// Another option is not to schedule the thread and wait for
-				// ack timeout...
-				AbsoluteTime deadline;
-				clock_interval_to_deadline( 30, kSecondScale, &deadline );
-				thread_call_enter1_delayed(
-					gRootDomain->diskSyncCalloutEntry,
-					(thread_call_param_t)(uintptr_t) params->notifyRef,
-					deadline );
-			} else {
-				thread_call_enter1(
-					gRootDomain->diskSyncCalloutEntry,
-					(thread_call_param_t)(uintptr_t) params->notifyRef);
-			}
-		}
-#if HIBERNATION
-		else if (CAP_DID_CHANGE_TO_ON(params, kIOPMSystemCapabilityCPU)) {
-			// We will ack within 110 seconds
-			params->maxWaitForReply = 110 * 1000 * 1000;
-
-			thread_call_enter1(
-				gRootDomain->diskSyncCalloutEntry,
-				(thread_call_param_t)(uintptr_t) params->notifyRef);
-		}
+            if ( !OSCompareAndSwap( 0, 1, &gSleepOrShutdownPending ) )
+            {
+                // Purposely delay the ack and hope that shutdown occurs quickly.
+                // Another option is not to schedule the thread and wait for
+                // ack timeout...
+                AbsoluteTime deadline;
+                clock_interval_to_deadline( 30, kSecondScale, &deadline );
+                thread_call_enter1_delayed(
+                    gRootDomain->diskSyncCalloutEntry, 
+                    (thread_call_param_t) params->notifyRef,
+                    deadline );
+            }
+            else
+                thread_call_enter1(
+                    gRootDomain->diskSyncCalloutEntry,
+                    (thread_call_param_t) params->notifyRef);
+        }
+#if	HIBERNATION
+        else
+        if ((params->changeFlags & kIOPMSystemCapabilityDidChange) &&
+            (params->toCapabilities & kIOPMSystemCapabilityCPU) &&
+            (params->fromCapabilities & kIOPMSystemCapabilityCPU) == 0)
+        {
+            // We will ack within 110 seconds
+            params->maxWaitForReply = 110 * 1000 * 1000;
+
+            thread_call_enter1(
+                gRootDomain->diskSyncCalloutEntry,
+                (thread_call_param_t) params->notifyRef);
+        }
 #endif
-		ret = kIOReturnSuccess;
-	}
-
-	return ret;
+        ret = kIOReturnSuccess;
+    }
+
+    return ret;
 }
 
 //******************************************************************************
@@ -4160,21 +2738,32 @@
 // @param   obj has a retain on it. We're responsible for releasing that retain.
 //******************************************************************************
 
-void
-IOPMrootDomain::handleQueueSleepWakeUUID(OSObject *obj)
-{
-	OSSharedPtr<OSString>    str;
-
-	if (kOSBooleanFalse == obj) {
-		handlePublishSleepWakeUUID(false);
-	} else {
-		str.reset(OSDynamicCast(OSString, obj), OSNoRetain);
-		if (str) {
-			// This branch caches the UUID for an upcoming sleep/wake
-			queuedSleepWakeUUIDString = str;
-			DLOG("SleepWake UUID queued: %s\n", queuedSleepWakeUUIDString->getCStringNoCopy());
-		}
-	}
+void IOPMrootDomain::handleQueueSleepWakeUUID(OSObject *obj)
+{        
+    OSString    *str = NULL;
+
+    if (kOSBooleanFalse == obj) 
+    {
+        handlePublishSleepWakeUUID(NULL);
+    }
+    else if ((str = OSDynamicCast(OSString, obj))) 
+    {
+        // This branch caches the UUID for an upcoming sleep/wake        
+        if (queuedSleepWakeUUIDString) {
+            queuedSleepWakeUUIDString->release();
+            queuedSleepWakeUUIDString = NULL;
+        }
+        queuedSleepWakeUUIDString = str;
+        queuedSleepWakeUUIDString->retain();
+
+        DLOG("SleepWake UUID queued: %s\n", queuedSleepWakeUUIDString->getCStringNoCopy());
+    }
+
+    if (obj) {
+        obj->release();
+    }
+    return;
+
 }
 //******************************************************************************
 // handlePublishSleepWakeUUID
@@ -4186,196 +2775,95 @@
 // sleep/wake.
 //******************************************************************************
 
-void
-IOPMrootDomain::handlePublishSleepWakeUUID( bool shouldPublish )
-{
-	ASSERT_GATED();
-
-	/*
-	 * Clear the current UUID
-	 */
-	if (gSleepWakeUUIDIsSet) {
-		DLOG("SleepWake UUID cleared\n");
-
-		gSleepWakeUUIDIsSet = false;
-
-		removeProperty(kIOPMSleepWakeUUIDKey);
-		messageClients(kIOPMMessageSleepWakeUUIDChange, kIOPMMessageSleepWakeUUIDCleared);
-	}
-
-	/*
-	 * Optionally, publish a new UUID
-	 */
-	if (queuedSleepWakeUUIDString && shouldPublish) {
-		OSSharedPtr<OSString> publishThisUUID;
-
-		publishThisUUID = queuedSleepWakeUUIDString;
-
-		if (publishThisUUID) {
-			setProperty(kIOPMSleepWakeUUIDKey, publishThisUUID.get());
-		}
-
-		gSleepWakeUUIDIsSet = true;
-		messageClients(kIOPMMessageSleepWakeUUIDChange, kIOPMMessageSleepWakeUUIDSet);
-
-		queuedSleepWakeUUIDString.reset();
-	}
-}
-
-//******************************************************************************
-// IOPMGetSleepWakeUUIDKey
+void IOPMrootDomain::handlePublishSleepWakeUUID( bool shouldPublish )
+{
+   ASSERT_GATED();
+
+   /* 
+    * Clear the current UUID
+    */
+   if (gSleepWakeUUIDIsSet)
+   {
+        DLOG("SleepWake UUID cleared\n");
+
+        OSString *UUIDstring = NULL;
+        
+        if (timeline && 
+            (UUIDstring = OSDynamicCast(OSString, getProperty(kIOPMSleepWakeUUIDKey)))) 
+        {
+            PMEventDetails *details = PMEventDetails::eventDetails(kIOPMEventTypeUUIDClear, 
+                            UUIDstring->getCStringNoCopy(), NULL, 0);
+            if (details) {
+                timeline->recordSystemPowerEvent( details );
+                details->release();
+            }
+            timeline->setNumEventsLoggedThisPeriod(0); 
+        }
+
+        gSleepWakeUUIDIsSet = false;
+
+        removeProperty(kIOPMSleepWakeUUIDKey);
+        messageClients(kIOPMMessageSleepWakeUUIDChange, kIOPMMessageSleepWakeUUIDCleared);
+    }
+
+    /*
+     * Optionally, publish a new UUID
+     */
+    if (queuedSleepWakeUUIDString && shouldPublish) {
+
+        OSString  *publishThisUUID = NULL;
+
+        publishThisUUID = queuedSleepWakeUUIDString;
+        publishThisUUID->retain();
+
+        if (timeline) {
+            PMEventDetails  *details;
+            details = PMEventDetails::eventDetails(kIOPMEventTypeUUIDSet,
+                              publishThisUUID->getCStringNoCopy(), NULL, 0);
+            if (details) {
+                timeline->recordSystemPowerEvent( details );
+                details->release();
+            }
+        }
+        
+        if (publishThisUUID)
+        {
+            setProperty(kIOPMSleepWakeUUIDKey, publishThisUUID);
+            publishThisUUID->release();
+        }
+        
+        gSleepWakeUUIDIsSet = true;
+        messageClients(kIOPMMessageSleepWakeUUIDChange, kIOPMMessageSleepWakeUUIDSet);
+
+        queuedSleepWakeUUIDString->release();
+        queuedSleepWakeUUIDString = NULL;
+    }
+}
+
+//******************************************************************************
+// changePowerStateTo & changePowerStateToPriv
 //
-// Return the truth value of gSleepWakeUUIDIsSet and optionally copy the key.
-// To get the full key -- a C string -- the buffer must large enough for
-// the end-of-string character.
-// The key is expected to be an UUID string
-//******************************************************************************
-
-extern "C" bool
-IOPMCopySleepWakeUUIDKey(char *buffer, size_t buf_len)
-{
-	if (!gSleepWakeUUIDIsSet) {
-		return false;
-	}
-
-	if (buffer != NULL) {
-		OSSharedPtr<OSString> string =
-		    OSDynamicPtrCast<OSString>(gRootDomain->copyProperty(kIOPMSleepWakeUUIDKey));
-
-		if (!string) {
-			*buffer = '\0';
-		} else {
-			strlcpy(buffer, string->getCStringNoCopy(), buf_len);
-		}
-	}
-
-	return true;
-}
-
-//******************************************************************************
-// lowLatencyAudioNotify
-//
-// Used to send an update about low latency audio activity to interested
-// clients. To keep the overhead minimal the OSDictionary used here
-// is initialized at boot.
-//******************************************************************************
-
-void
-IOPMrootDomain::lowLatencyAudioNotify(uint64_t time, boolean_t state)
-{
-	if (lowLatencyAudioNotifierDict && lowLatencyAudioNotifyStateSym && lowLatencyAudioNotifyTimestampSym &&
-	    lowLatencyAudioNotifyStateVal && lowLatencyAudioNotifyTimestampVal) {
-		lowLatencyAudioNotifyTimestampVal->setValue(time);
-		lowLatencyAudioNotifyStateVal->setValue(state);
-		setPMSetting(gIOPMSettingLowLatencyAudioModeKey.get(), lowLatencyAudioNotifierDict.get());
-	} else {
-		DLOG("LowLatencyAudioNotify error\n");
-	}
-	return;
-}
-
-//******************************************************************************
-// IOPMrootDomainRTNotifier
-//
-// Used by performance controller to update the timestamp and state associated
-// with low latency audio activity in the system.
-//******************************************************************************
-
-extern "C" void
-IOPMrootDomainRTNotifier(uint64_t time, boolean_t state)
-{
-	gRootDomain->lowLatencyAudioNotify(time, state);
-	return;
-}
-
-//******************************************************************************
-// initializeBootSessionUUID
-//
-// Initialize the boot session uuid at boot up and sets it into registry.
-//******************************************************************************
-
-void
-IOPMrootDomain::initializeBootSessionUUID(void)
-{
-	uuid_t          new_uuid;
-	uuid_string_t   new_uuid_string;
-
-	uuid_generate(new_uuid);
-	uuid_unparse_upper(new_uuid, new_uuid_string);
-	memcpy(bootsessionuuid_string, new_uuid_string, sizeof(uuid_string_t));
-
-	setProperty(kIOPMBootSessionUUIDKey, new_uuid_string);
-}
-
-//******************************************************************************
-// Root domain uses the private and tagged changePowerState methods for
-// tracking and logging purposes.
-//******************************************************************************
-
-#define REQUEST_TAG_TO_REASON(x)        ((uint16_t)x)
-
-static uint32_t
-nextRequestTag( IOPMRequestTag tag )
-{
-	static SInt16 msb16 = 1;
-	uint16_t id = OSAddAtomic16(1, &msb16);
-	return ((uint32_t)id << 16) | REQUEST_TAG_TO_REASON(tag);
-}
-
-// TODO: remove this shim function and exported symbol
-IOReturn
-IOPMrootDomain::changePowerStateTo( unsigned long ordinal )
-{
-	return changePowerStateWithTagTo(ordinal, kCPSReasonNone);
-}
-
-// TODO: remove this shim function and exported symbol
-IOReturn
-IOPMrootDomain::changePowerStateToPriv( unsigned long ordinal )
-{
-	return changePowerStateWithTagToPriv(ordinal, kCPSReasonNone);
-}
-
-IOReturn
-IOPMrootDomain::changePowerStateWithOverrideTo(
-	IOPMPowerStateIndex ordinal, IOPMRequestTag reason )
-{
-	uint32_t tag = nextRequestTag(reason);
-	DLOG("%s(%s, %x)\n", __FUNCTION__, getPowerStateString((uint32_t) ordinal), tag);
-
-	if ((ordinal != ON_STATE) && (ordinal != AOT_STATE) && (ordinal != SLEEP_STATE)) {
-		return kIOReturnUnsupported;
-	}
-
-	return super::changePowerStateWithOverrideTo(ordinal, tag);
-}
-
-IOReturn
-IOPMrootDomain::changePowerStateWithTagTo(
-	IOPMPowerStateIndex ordinal, IOPMRequestTag reason )
-{
-	uint32_t tag = nextRequestTag(reason);
-	DLOG("%s(%s, %x)\n", __FUNCTION__, getPowerStateString((uint32_t) ordinal), tag);
-
-	if ((ordinal != ON_STATE) && (ordinal != AOT_STATE) && (ordinal != SLEEP_STATE)) {
-		return kIOReturnUnsupported;
-	}
-
-	return super::changePowerStateWithTagTo(ordinal, tag);
-}
-
-IOReturn
-IOPMrootDomain::changePowerStateWithTagToPriv(
-	IOPMPowerStateIndex ordinal, IOPMRequestTag reason )
-{
-	uint32_t tag = nextRequestTag(reason);
-	DLOG("%s(%s, %x)\n", __FUNCTION__, getPowerStateString((uint32_t) ordinal), tag);
-
-	if ((ordinal != ON_STATE) && (ordinal != AOT_STATE) && (ordinal != SLEEP_STATE)) {
-		return kIOReturnUnsupported;
-	}
-
-	return super::changePowerStateWithTagToPriv(ordinal, tag);
+// Override of these methods for logging purposes.
+//******************************************************************************
+
+IOReturn IOPMrootDomain::changePowerStateTo( unsigned long ordinal )
+{
+    DLOG("changePowerStateTo(%lu)\n", ordinal);
+
+    if ((ordinal != ON_STATE) && (ordinal != SLEEP_STATE))
+        return kIOReturnUnsupported;
+
+    return super::changePowerStateTo(ordinal);
+}
+
+IOReturn IOPMrootDomain::changePowerStateToPriv( unsigned long ordinal )
+{
+    DLOG("changePowerStateToPriv(%lu)\n", ordinal);
+
+    if ((ordinal != ON_STATE) && (ordinal != SLEEP_STATE))
+        return kIOReturnUnsupported;
+
+    return super::changePowerStateToPriv(ordinal);
 }
 
 //******************************************************************************
@@ -4383,204 +2871,30 @@
 //
 //******************************************************************************
 
-bool
-IOPMrootDomain::activitySinceSleep(void)
-{
-	return userActivityCount != userActivityAtSleep;
-}
-
-bool
-IOPMrootDomain::abortHibernation(void)
-{
-#if __arm64__
-	// don't allow hibernation to be aborted on ARM due to user activity
-	// since once ApplePMGR decides we're hibernating, we can't turn back
-	// see: <rdar://problem/63848862> Tonga ApplePMGR diff quiesce path support
-	return false;
-#else
-	bool ret = activitySinceSleep();
-
-	if (ret && !hibernateAborted && checkSystemCanSustainFullWake()) {
-		DLOG("activitySinceSleep ABORT [%d, %d]\n", userActivityCount, userActivityAtSleep);
-		hibernateAborted = true;
-	}
-	return ret;
-#endif
+bool IOPMrootDomain::activitySinceSleep(void)
+{
+    return (userActivityCount != userActivityAtSleep);
+}
+
+bool IOPMrootDomain::abortHibernation(void)
+{
+    bool ret = activitySinceSleep();
+
+    if (ret && !hibernateAborted && checkSystemCanSustainFullWake())
+    {
+        DLOG("activitySinceSleep ABORT [%d, %d]\n", userActivityCount, userActivityAtSleep);
+        hibernateAborted = true;
+    }
+    return (ret);
 }
 
 extern "C" int
 hibernate_should_abort(void)
 {
-	if (gRootDomain) {
-		return gRootDomain->abortHibernation();
-	} else {
-		return 0;
-	}
-}
-
-//******************************************************************************
-// scheduleImmediateDebugWake
-//
-// Schedule a wake with RTC to wake us back up immediately after we sleep.
-// Useful when a cancel request comes in past the revert point on the sleep path
-//******************************************************************************
-
-void
-IOPMrootDomain::scheduleImmediateDebugWake( void )
-{
-	OSSharedPtr<OSDictionary> dict = OSDictionary::withCapacity(1);
-	OSSharedPtr<OSNumber> secs = OSNumber::withNumber(1, 32);
-
-	if (dict && secs) {
-		dict->setObject(gIOPMSettingDebugWakeRelativeKey.get(), secs.get());
-		gRootDomain->setProperties(dict.get());
-		MSG("Reverting sleep with relative wake\n");
-	}
-}
-
-//******************************************************************************
-// willNotifyInterest
-//
-// Called after all priority clients have all acknowledged the power change,
-// but before any interested drivers and any power children are informed.
-// Dispatched though a thread call, so it is safe to perform work that might block on a
-// sleeping disk. PM state machine (not thread) will block w/o timeout until this function returns.
-//******************************************************************************
-
-void
-IOPMrootDomain::willNotifyInterested( IOPMPowerStateIndex newPowerState )
-{
-	if (SLEEP_STATE == newPowerState) {
-		_aotReadyToFullWake = false;
-#if 0
-		if (_aotLingerTime) {
-			uint64_t interval, deadline;
-			IOLog("aot linger no return\n");
-			nanoseconds_to_absolutetime(_aotLingerTime * NSEC_PER_MSEC, &interval);
-			clock_absolutetime_interval_to_deadline(interval, &deadline);
-			clock_delay_until(deadline);
-		}
-#endif
-		if (!_aotMode) {
-			_aotTestTime = 0;
-			_aotWakeTimeCalendar.selector = kPMCalendarTypeInvalid;
-			_aotLastWakeTime = 0;
-			if (_aotMetrics) {
-				bzero(_aotMetrics, sizeof(IOPMAOTMetrics));
-			}
-		} else if (!_aotNow && !_debugWakeSeconds) {
-			_aotNow            = true;
-			_aotPendingFlags   = 0;
-			_aotTasksSuspended = true;
-			_aotLastWakeTime   = 0;
-			bzero(_aotMetrics, sizeof(IOPMAOTMetrics));
-			if (kIOPMAOTModeCycle & _aotMode) {
-				clock_interval_to_absolutetime_interval(10, kSecondScale, &_aotTestInterval);
-				_aotTestTime = mach_continuous_time() + _aotTestInterval;
-				AbsoluteTime endInterval;
-				clock_interval_to_absolutetime_interval(60, kSecondScale, &endInterval);
-				_aotEndTime = mach_continuous_time() + endInterval;
-				_setWakeTime(_aotTestTime);
-			}
-			clock_interval_to_absolutetime_interval(2000, kMillisecondScale, &_aotWakePreWindow);
-			clock_interval_to_absolutetime_interval(1100, kMillisecondScale, &_aotWakePostWindow);
-		}
-
-		if (updateTasksSuspend(kTasksSuspendSuspended, kTasksSuspendNoChange)) {
-			IOLog("PMRD: tasks suspend\n");
-			AbsoluteTime deadline;
-
-			clock_interval_to_deadline(10, kSecondScale, &deadline);
-#if defined(XNU_TARGET_OS_OSX)
-			vm_pageout_wait(AbsoluteTime_to_scalar(&deadline));
-#endif /* defined(XNU_TARGET_OS_OSX) */
-		}
-	}
-}
-
-//******************************************************************************
-// willNotifyPowerChildren
-//
-// Called after all interested drivers have all acknowledged the power change,
-// but before any power children are informed.
-// Dispatched though a thread call, so it is safe to perform work that might block on a
-// sleeping disk. PM state machine (not thread) will block w/o timeout until this function returns.
-//******************************************************************************
-
-void
-IOPMrootDomain::willNotifyPowerChildren( IOPMPowerStateIndex newPowerState )
-{
-	if (SLEEP_STATE == newPowerState) {
-		notifierThread = current_thread();
-
-#if HIBERNATION
-		// Adjust watchdog for IOHibernateSystemSleep
-		int defaultTimeout = getWatchdogTimeout();
-		int timeout = defaultTimeout > WATCHDOG_HIBERNATION_TIMEOUT ?
-		    defaultTimeout : WATCHDOG_HIBERNATION_TIMEOUT;
-		reset_watchdog_timer(timeout);
-
-		IOHibernateSystemSleep();
-		IOHibernateIOKitSleep();
-#endif
-#if defined(__arm64__) && HIBERNATION
-		if (gIOHibernateState == kIOHibernateStateInactive) {
-			setProperty(kIOPMSystemSleepTypeKey, kIOPMSleepTypeDeepIdle, 32);
-		}
-		// On AS, hibernation cannot be aborted. Resetting RTC to 1s during hibernation upon detecting
-		// user activity is pointless (we are likely to spend >1s hibernating). It also clears existing
-		// alarms, which can mess with cycler tools.
-		if (gRootDomain->activitySinceSleep() && gIOHibernateState == kIOHibernateStateInactive) {
-#else /* defined(__arm64__) && HIBERNATION */
-		// On non-AS, hibernation can be aborted if user activity is detected. So continue to reset the
-		// RTC alarm (even during hibernation) so we can immediately wake from regular S2R if needed.
-		if (gRootDomain->activitySinceSleep()) {
-#endif /* defined(__arm64__) && HIBERNATION */
-			scheduleImmediateDebugWake();
-		}
-
-		notifierThread = NULL;
-	}
-}
-
-//******************************************************************************
-// willTellSystemCapabilityDidChange
-//
-// IOServicePM calls this from OurChangeTellCapabilityDidChange() when root
-// domain is raising its power state, immediately after notifying interested
-// drivers and power children.
-//******************************************************************************
-
-void
-IOPMrootDomain::willTellSystemCapabilityDidChange( void )
-{
-	if ((_systemTransitionType == kSystemTransitionWake) &&
-	    !CAP_GAIN(kIOPMSystemCapabilityGraphics)) {
-		// After powering up drivers, dark->full promotion on the current wake
-		// transition is no longer possible. That is because the next machine
-		// state will issue the system capability change messages.
-		// The darkWakePowerClamped flag may already be set if the system has
-		// at least one driver that was power clamped due to dark wake.
-		// This function sets the darkWakePowerClamped flag in case there
-		// is no power-clamped driver in the system.
-		//
-		// Last opportunity to exit dark wake using:
-		// requestFullWake( kFullWakeReasonLocalUser );
-
-		if (!darkWakePowerClamped) {
-			if (darkWakeLogClamp) {
-				AbsoluteTime    now;
-				uint64_t        nsec;
-
-				clock_get_uptime(&now);
-				SUB_ABSOLUTETIME(&now, &gIOLastWakeAbsTime);
-				absolutetime_to_nanoseconds(now, &nsec);
-				DLOG("dark wake promotion disabled at %u ms\n",
-				    ((int)((nsec) / NSEC_PER_MSEC)));
-			}
-			darkWakePowerClamped = true;
-		}
-	}
+    if (gRootDomain)
+        return (gRootDomain->abortHibernation());
+    else
+        return (0);
 }
 
 //******************************************************************************
@@ -4590,54 +2904,35 @@
 // is closed.
 //******************************************************************************
 
-bool
-IOPMrootDomain::shouldSleepOnClamshellClosed( void )
-{
-	if (!clamshellExists) {
-		return false;
-	}
-
-	DLOG("clamshell closed %d, disabled %d/%x, desktopMode %d, ac %d\n",
-	    clamshellClosed, clamshellDisabled, clamshellSleepDisableMask, desktopMode, acAdaptorConnected);
-
-	return !clamshellDisabled && !(desktopMode && acAdaptorConnected) && !clamshellSleepDisableMask;
-}
-
-bool
-IOPMrootDomain::shouldSleepOnRTCAlarmWake( void )
-{
-	// Called once every RTC/Alarm wake. Device should go to sleep if on clamshell
-	// closed && battery
-	if (!clamshellExists) {
-		return false;
-	}
-
-	DLOG("shouldSleepOnRTCAlarmWake: clamshell closed %d, disabled %d/%x, desktopMode %d, ac %d\n",
-	    clamshellClosed, clamshellDisabled, clamshellSleepDisableMask, desktopMode, acAdaptorConnected);
-
-	return !acAdaptorConnected && !clamshellSleepDisableMask;
-}
-
-void
-IOPMrootDomain::sendClientClamshellNotification( void )
-{
-	/* Only broadcast clamshell alert if clamshell exists. */
-	if (!clamshellExists) {
-		return;
-	}
-
-	setProperty(kAppleClamshellStateKey,
-	    clamshellClosed ? kOSBooleanTrue : kOSBooleanFalse);
-
-	setProperty(kAppleClamshellCausesSleepKey,
-	    shouldSleepOnClamshellClosed() ? kOSBooleanTrue : kOSBooleanFalse);
-
-	/* Argument to message is a bitfiel of
-	 *      ( kClamshellStateBit | kClamshellSleepBit )
-	 */
-	messageClients(kIOPMMessageClamshellStateChange,
-	    (void *)(uintptr_t) ((clamshellClosed ? kClamshellStateBit : 0)
-	    | (shouldSleepOnClamshellClosed() ? kClamshellSleepBit : 0)));
+bool IOPMrootDomain::shouldSleepOnClamshellClosed( void )
+{
+    if (!clamshellExists)
+        return false;
+
+    DLOG("clamshell closed %d, disabled %d, desktopMode %d, ac %d\n",
+        clamshellClosed, clamshellDisabled, desktopMode, acAdaptorConnected);
+
+    return ( !clamshellDisabled && !(desktopMode && acAdaptorConnected) );
+}
+
+void IOPMrootDomain::sendClientClamshellNotification( void )
+{
+    /* Only broadcast clamshell alert if clamshell exists. */
+    if (!clamshellExists)
+        return;
+
+    setProperty(kAppleClamshellStateKey, 
+        clamshellClosed ? kOSBooleanTrue : kOSBooleanFalse);
+
+    setProperty(kAppleClamshellCausesSleepKey, 
+        shouldSleepOnClamshellClosed() ? kOSBooleanTrue : kOSBooleanFalse);
+
+    /* Argument to message is a bitfiel of 
+     *      ( kClamshellStateBit | kClamshellSleepBit )
+     */
+    messageClients(kIOPMMessageClamshellStateChange,
+        (void *) ( (clamshellClosed ? kClamshellStateBit : 0)
+             | ( shouldSleepOnClamshellClosed() ? kClamshellSleepBit : 0)) );
 }
 
 //******************************************************************************
@@ -4646,10 +2941,9 @@
 // Deprecated
 //******************************************************************************
 
-IOOptionBits
-IOPMrootDomain::getSleepSupported( void )
-{
-	return platformSleepSupport;
+IOOptionBits IOPMrootDomain::getSleepSupported( void )
+{
+    return( platformSleepSupport );
 }
 
 //******************************************************************************
@@ -4658,46 +2952,10 @@
 // Deprecated
 //******************************************************************************
 
-void
-IOPMrootDomain::setSleepSupported( IOOptionBits flags )
-{
-	DLOG("setSleepSupported(%x)\n", (uint32_t) flags);
-	OSBitOrAtomic(flags, &platformSleepSupport);
-}
-
-//******************************************************************************
-// setClamShellSleepDisable
-//
-//******************************************************************************
-
-void
-IOPMrootDomain::setClamShellSleepDisable( bool disable, uint32_t bitmask )
-{
-	uint32_t oldMask;
-
-	// User client calls this in non-gated context
-	if (gIOPMWorkLoop->inGate() == false) {
-		gIOPMWorkLoop->runAction(
-			OSMemberFunctionCast(IOWorkLoop::Action, this,
-			&IOPMrootDomain::setClamShellSleepDisable),
-			(OSObject *) this,
-			(void *) disable, (void *)(uintptr_t) bitmask);
-		return;
-	}
-
-	oldMask = clamshellSleepDisableMask;
-	if (disable) {
-		clamshellSleepDisableMask |= bitmask;
-	} else {
-		clamshellSleepDisableMask &= ~bitmask;
-	}
-	DLOG("setClamShellSleepDisable(%x->%x)\n", oldMask, clamshellSleepDisableMask);
-
-	if (clamshellExists && clamshellClosed &&
-	    (clamshellSleepDisableMask != oldMask) &&
-	    (clamshellSleepDisableMask == 0)) {
-		handlePowerNotification(kLocalEvalClamshellCommand);
-	}
+void IOPMrootDomain::setSleepSupported( IOOptionBits flags )
+{
+    DLOG("setSleepSupported(%x)\n", (uint32_t) flags);
+    OSBitOrAtomic(flags, &platformSleepSupport);
 }
 
 //******************************************************************************
@@ -4706,93 +2964,9 @@
 // Deprecated.
 //******************************************************************************
 
-void
-IOPMrootDomain::wakeFromDoze( void )
-{
-	// Preserve symbol for familes (IOUSBFamily and IOGraphics)
-}
-
-//******************************************************************************
-// recordRTCAlarm
-//
-// Record the earliest scheduled RTC alarm to determine whether a RTC wake
-// should be a dark wake or a full wake. Both Maintenance and SleepService
-// alarms are dark wake, while AutoWake (WakeByCalendarDate) and DebugWake
-// (WakeRelativeToSleep) should trigger a full wake. Scheduled power-on
-// PMSettings are ignored.
-//
-// Caller serialized using settingsCtrlLock.
-//******************************************************************************
-
-void
-IOPMrootDomain::recordRTCAlarm(
-	const OSSymbol  *type,
-	OSObject        *object )
-{
-	uint32_t previousAlarmMask = _scheduledAlarmMask;
-
-	if (type == gIOPMSettingDebugWakeRelativeKey) {
-		OSNumber * n = OSDynamicCast(OSNumber, object);
-		if (n) {
-			// Debug wake has highest scheduling priority so it overrides any
-			// pre-existing alarm.
-			uint32_t debugSecs = n->unsigned32BitValue();
-			_nextScheduledAlarmType.reset(type, OSRetain);
-			_nextScheduledAlarmUTC = debugSecs;
-
-			_debugWakeSeconds = debugSecs;
-			OSBitOrAtomic(kIOPMAlarmBitDebugWake, &_scheduledAlarmMask);
-			DLOG("next alarm (%s) in %u secs\n",
-			    type->getCStringNoCopy(), debugSecs);
-		}
-	} else if ((type == gIOPMSettingAutoWakeCalendarKey.get()) ||
-	    (type == gIOPMSettingMaintenanceWakeCalendarKey.get()) ||
-	    (type == gIOPMSettingSleepServiceWakeCalendarKey.get())) {
-		OSData * data = OSDynamicCast(OSData, object);
-		if (data && (data->getLength() == sizeof(IOPMCalendarStruct))) {
-			const IOPMCalendarStruct * cs;
-			bool replaceNextAlarm = false;
-			clock_sec_t secs;
-
-			cs = (const IOPMCalendarStruct *) data->getBytesNoCopy();
-			secs = IOPMConvertCalendarToSeconds(cs);
-			DLOG("%s " YMDTF "\n", type->getCStringNoCopy(), YMDT(cs));
-
-			// Update the next scheduled alarm type
-			if ((_nextScheduledAlarmType == NULL) ||
-			    ((_nextScheduledAlarmType != gIOPMSettingDebugWakeRelativeKey) &&
-			    (secs < _nextScheduledAlarmUTC))) {
-				replaceNextAlarm = true;
-			}
-
-			if (type == gIOPMSettingAutoWakeCalendarKey.get()) {
-				if (cs->year) {
-					_calendarWakeAlarmUTC = IOPMConvertCalendarToSeconds(cs);
-					OSBitOrAtomic(kIOPMAlarmBitCalendarWake, &_scheduledAlarmMask);
-				} else {
-					// TODO: can this else-block be removed?
-					_calendarWakeAlarmUTC = 0;
-					OSBitAndAtomic(~kIOPMAlarmBitCalendarWake, &_scheduledAlarmMask);
-				}
-			}
-			if (type == gIOPMSettingMaintenanceWakeCalendarKey.get()) {
-				OSBitOrAtomic(kIOPMAlarmBitMaintenanceWake, &_scheduledAlarmMask);
-			}
-			if (type == gIOPMSettingSleepServiceWakeCalendarKey.get()) {
-				OSBitOrAtomic(kIOPMAlarmBitSleepServiceWake, &_scheduledAlarmMask);
-			}
-
-			if (replaceNextAlarm) {
-				_nextScheduledAlarmType.reset(type, OSRetain);
-				_nextScheduledAlarmUTC = secs;
-				DLOG("next alarm (%s) " YMDTF "\n", type->getCStringNoCopy(), YMDT(cs));
-			}
-		}
-	}
-
-	if (_scheduledAlarmMask != previousAlarmMask) {
-		DLOG("scheduled alarm mask 0x%x\n", (uint32_t) _scheduledAlarmMask);
-	}
+void IOPMrootDomain::wakeFromDoze( void )
+{
+    // Preserve symbol for familes (IOUSBFamily and IOGraphics)
 }
 
 // MARK: -
@@ -4804,10 +2978,9 @@
 // Adds a new feature to the supported features dictionary
 //******************************************************************************
 
-void
-IOPMrootDomain::publishFeature( const char * feature )
-{
-	publishFeature(feature, kRD_AllPowerSources, NULL);
+void IOPMrootDomain::publishFeature( const char * feature )
+{
+    publishFeature(feature, kRD_AllPowerSources, NULL);
 }
 
 //******************************************************************************
@@ -4816,99 +2989,102 @@
 // Adds a new feature to the supported features dictionary
 //******************************************************************************
 
-void
-IOPMrootDomain::publishFeature(
-	const char *feature,
-	uint32_t supportedWhere,
-	uint32_t *uniqueFeatureID)
-{
-	static uint16_t       next_feature_id = 500;
-
-	OSSharedPtr<OSNumber> new_feature_data;
-	OSNumber             *existing_feature = NULL;
-	OSArray              *existing_feature_arr_raw = NULL;
-	OSSharedPtr<OSArray>  existing_feature_arr;
-	OSObject             *osObj = NULL;
-	uint32_t              feature_value = 0;
-
-	supportedWhere &= kRD_AllPowerSources; // mask off any craziness!
-
-	if (!supportedWhere) {
-		// Feature isn't supported anywhere!
-		return;
-	}
-
-	if (next_feature_id > 5000) {
-		// Far, far too many features!
-		return;
-	}
-
-	if (featuresDictLock) {
-		IOLockLock(featuresDictLock);
-	}
-
-	OSSharedPtr<OSObject> origFeaturesProp = copyProperty(kRootDomainSupportedFeatures);
-	OSDictionary *origFeatures = OSDynamicCast(OSDictionary, origFeaturesProp.get());
-	OSSharedPtr<OSDictionary> features;
-
-	// Create new features dict if necessary
-	if (origFeatures) {
-		features = OSDictionary::withDictionary(origFeatures);
-	} else {
-		features = OSDictionary::withCapacity(1);
-	}
-
-	// Create OSNumber to track new feature
-
-	next_feature_id += 1;
-	if (uniqueFeatureID) {
-		// We don't really mind if the calling kext didn't give us a place
-		// to stash their unique id. Many kexts don't plan to unload, and thus
-		// have no need to remove themselves later.
-		*uniqueFeatureID = next_feature_id;
-	}
-
-	feature_value = (uint32_t)next_feature_id;
-	feature_value <<= 16;
-	feature_value += supportedWhere;
-
-	new_feature_data = OSNumber::withNumber(
-		(unsigned long long)feature_value, 32);
-
-	// Does features object already exist?
-	if ((osObj = features->getObject(feature))) {
-		if ((existing_feature = OSDynamicCast(OSNumber, osObj))) {
-			// We need to create an OSArray to hold the now 2 elements.
-			existing_feature_arr = OSArray::withObjects(
-				(const OSObject **)&existing_feature, 1, 2);
-		} else if ((existing_feature_arr_raw = OSDynamicCast(OSArray, osObj))) {
-			// Add object to existing array
-			existing_feature_arr = OSArray::withArray(
-				existing_feature_arr_raw,
-				existing_feature_arr_raw->getCount() + 1);
-		}
-
-		if (existing_feature_arr) {
-			existing_feature_arr->setObject(new_feature_data.get());
-			features->setObject(feature, existing_feature_arr.get());
-		}
-	} else {
-		// The easy case: no previously existing features listed. We simply
-		// set the OSNumber at key 'feature' and we're on our way.
-		features->setObject(feature, new_feature_data.get());
-	}
-
-	setProperty(kRootDomainSupportedFeatures, features.get());
-
-	if (featuresDictLock) {
-		IOLockUnlock(featuresDictLock);
-	}
-
-	// Notify EnergySaver and all those in user space so they might
-	// re-populate their feature specific UI
-	if (pmPowerStateQueue) {
-		pmPowerStateQueue->submitPowerEvent( kPowerEventFeatureChanged );
-	}
+void IOPMrootDomain::publishFeature(
+    const char *feature, 
+    uint32_t supportedWhere,
+    uint32_t *uniqueFeatureID)
+{
+    static uint16_t     next_feature_id = 500;
+
+    OSNumber            *new_feature_data = NULL;
+    OSNumber            *existing_feature = NULL;
+    OSArray             *existing_feature_arr = NULL;
+    OSObject            *osObj = NULL;
+    uint32_t            feature_value = 0;
+
+    supportedWhere &= kRD_AllPowerSources; // mask off any craziness!
+
+    if(!supportedWhere) {
+        // Feature isn't supported anywhere!
+        return;
+    }
+    
+    if(next_feature_id > 5000) {
+        // Far, far too many features!
+        return;
+    }
+
+    if(featuresDictLock) IOLockLock(featuresDictLock);
+
+    OSDictionary *features =
+        (OSDictionary *) getProperty(kRootDomainSupportedFeatures);
+    
+    // Create new features dict if necessary
+    if ( features && OSDynamicCast(OSDictionary, features)) {
+        features = OSDictionary::withDictionary(features);
+    } else {
+        features = OSDictionary::withCapacity(1);
+    }
+    
+    // Create OSNumber to track new feature
+    
+    next_feature_id += 1;
+    if( uniqueFeatureID ) {
+        // We don't really mind if the calling kext didn't give us a place
+        // to stash their unique id. Many kexts don't plan to unload, and thus
+        // have no need to remove themselves later.
+        *uniqueFeatureID = next_feature_id;
+    }
+
+    feature_value = (uint32_t)next_feature_id;
+    feature_value <<= 16;
+    feature_value += supportedWhere;
+
+    new_feature_data = OSNumber::withNumber(
+                                (unsigned long long)feature_value, 32);
+
+    // Does features object already exist?
+    if( (osObj = features->getObject(feature)) )
+    {
+        if(( existing_feature = OSDynamicCast(OSNumber, osObj) ))
+        {
+            // We need to create an OSArray to hold the now 2 elements.
+            existing_feature_arr = OSArray::withObjects(
+                            (const OSObject **)&existing_feature, 1, 2);
+        } else if(( existing_feature_arr = OSDynamicCast(OSArray, osObj) ))
+        {
+            // Add object to existing array        
+            existing_feature_arr = OSArray::withArray(
+                            existing_feature_arr,
+                            existing_feature_arr->getCount() + 1);
+        }
+
+        if (existing_feature_arr)
+        {
+            existing_feature_arr->setObject(new_feature_data);
+            features->setObject(feature, existing_feature_arr);
+            existing_feature_arr->release();
+            existing_feature_arr = 0;
+        }
+    } else {
+        // The easy case: no previously existing features listed. We simply
+        // set the OSNumber at key 'feature' and we're on our way.
+        features->setObject(feature, new_feature_data);
+    }
+    
+    new_feature_data->release();
+
+    setProperty(kRootDomainSupportedFeatures, features);
+
+    features->release();
+
+    if(featuresDictLock) IOLockUnlock(featuresDictLock);    
+
+    // Notify EnergySaver and all those in user space so they might
+    // re-populate their feature specific UI    
+    if(pmPowerStateQueue) {
+        pmPowerStateQueue->submitPowerEvent( kPowerEventFeatureChanged );
+    }
 }
 
 //******************************************************************************
@@ -4917,123 +3093,129 @@
 // Removes previously published feature
 //******************************************************************************
 
-IOReturn
-IOPMrootDomain::removePublishedFeature( uint32_t removeFeatureID )
-{
-	IOReturn                ret = kIOReturnError;
-	uint32_t                feature_value = 0;
-	uint16_t                feature_id = 0;
-	bool                    madeAChange = false;
-
-	OSSymbol                *dictKey = NULL;
-	OSSharedPtr<OSCollectionIterator>    dictIterator;
-	OSArray                 *arrayMember  = NULL;
-	OSNumber                *numberMember = NULL;
-	OSObject                *osObj        = NULL;
-	OSNumber                *osNum        = NULL;
-	OSSharedPtr<OSArray>    arrayMemberCopy;
-
-	if (kBadPMFeatureID == removeFeatureID) {
-		return kIOReturnNotFound;
-	}
-
-	if (featuresDictLock) {
-		IOLockLock(featuresDictLock);
-	}
-
-	OSSharedPtr<OSObject> origFeaturesProp = copyProperty(kRootDomainSupportedFeatures);
-	OSDictionary *origFeatures = OSDynamicCast(OSDictionary, origFeaturesProp.get());
-	OSSharedPtr<OSDictionary> features;
-
-	if (origFeatures) {
-		// Any modifications to the dictionary are made to the copy to prevent
-		// races & crashes with userland clients. Dictionary updated
-		// automically later.
-		features = OSDictionary::withDictionary(origFeatures);
-	} else {
-		features = NULL;
-		ret = kIOReturnNotFound;
-		goto exit;
-	}
-
-	// We iterate 'features' dictionary looking for an entry tagged
-	// with 'removeFeatureID'. If found, we remove it from our tracking
-	// structures and notify the OS via a general interest message.
-
-	dictIterator = OSCollectionIterator::withCollection(features.get());
-	if (!dictIterator) {
-		goto exit;
-	}
-
-	while ((dictKey = OSDynamicCast(OSSymbol, dictIterator->getNextObject()))) {
-		osObj = features->getObject(dictKey);
-
-		// Each Feature is either tracked by an OSNumber
-		if (osObj && (numberMember = OSDynamicCast(OSNumber, osObj))) {
-			feature_value = numberMember->unsigned32BitValue();
-			feature_id = (uint16_t)(feature_value >> 16);
-
-			if (feature_id == (uint16_t)removeFeatureID) {
-				// Remove this node
-				features->removeObject(dictKey);
-				madeAChange = true;
-				break;
-			}
-
-			// Or tracked by an OSArray of OSNumbers
-		} else if (osObj && (arrayMember = OSDynamicCast(OSArray, osObj))) {
-			unsigned int arrayCount = arrayMember->getCount();
-
-			for (unsigned int i = 0; i < arrayCount; i++) {
-				osNum = OSDynamicCast(OSNumber, arrayMember->getObject(i));
-				if (!osNum) {
-					continue;
-				}
-
-				feature_value = osNum->unsigned32BitValue();
-				feature_id = (uint16_t)(feature_value >> 16);
-
-				if (feature_id == (uint16_t)removeFeatureID) {
-					// Remove this node
-					if (1 == arrayCount) {
-						// If the array only contains one element, remove
-						// the whole thing.
-						features->removeObject(dictKey);
-					} else {
-						// Otherwise remove the element from a copy of the array.
-						arrayMemberCopy = OSArray::withArray(arrayMember);
-						if (arrayMemberCopy) {
-							arrayMemberCopy->removeObject(i);
-							features->setObject(dictKey, arrayMemberCopy.get());
-						}
-					}
-
-					madeAChange = true;
-					break;
-				}
-			}
-		}
-	}
-
-	if (madeAChange) {
-		ret = kIOReturnSuccess;
-
-		setProperty(kRootDomainSupportedFeatures, features.get());
-
-		// Notify EnergySaver and all those in user space so they might
-		// re-populate their feature specific UI
-		if (pmPowerStateQueue) {
-			pmPowerStateQueue->submitPowerEvent( kPowerEventFeatureChanged );
-		}
-	} else {
-		ret = kIOReturnNotFound;
-	}
-
+IOReturn IOPMrootDomain::removePublishedFeature( uint32_t removeFeatureID )
+{
+    IOReturn                ret = kIOReturnError;
+    uint32_t                feature_value = 0;
+    uint16_t                feature_id = 0;
+    bool                    madeAChange = false;
+    
+    OSSymbol                *dictKey = NULL;
+    OSCollectionIterator    *dictIterator = NULL;
+    OSArray                 *arrayMember  = NULL;
+    OSNumber                *numberMember = NULL;
+    OSObject                *osObj        = NULL;
+    OSNumber                *osNum        = NULL;
+    OSArray                 *arrayMemberCopy;
+
+    if (kBadPMFeatureID == removeFeatureID)
+        return kIOReturnNotFound;
+
+    if(featuresDictLock) IOLockLock(featuresDictLock);
+
+    OSDictionary *features =
+        (OSDictionary *) getProperty(kRootDomainSupportedFeatures);
+    
+    if ( features && OSDynamicCast(OSDictionary, features) )
+    {
+        // Any modifications to the dictionary are made to the copy to prevent
+        // races & crashes with userland clients. Dictionary updated
+        // automically later.
+        features = OSDictionary::withDictionary(features);
+    } else {
+        features = NULL;
+        ret = kIOReturnNotFound;
+        goto exit;
+    }
+    
+    // We iterate 'features' dictionary looking for an entry tagged
+    // with 'removeFeatureID'. If found, we remove it from our tracking
+    // structures and notify the OS via a general interest message.
+    
+    dictIterator = OSCollectionIterator::withCollection(features);
+    if(!dictIterator) {
+        goto exit;
+    }
+    
+    while( (dictKey = OSDynamicCast(OSSymbol, dictIterator->getNextObject())) )
+    {
+        osObj = features->getObject(dictKey);
+        
+        // Each Feature is either tracked by an OSNumber
+        if( osObj && (numberMember = OSDynamicCast(OSNumber, osObj)) )
+        {
+            feature_value = numberMember->unsigned32BitValue();
+            feature_id = (uint16_t)(feature_value >> 16);
+
+            if( feature_id == (uint16_t)removeFeatureID )
+            {
+                // Remove this node
+                features->removeObject(dictKey);
+                madeAChange = true;
+                break;
+            }
+        
+        // Or tracked by an OSArray of OSNumbers
+        } else if( osObj && (arrayMember = OSDynamicCast(OSArray, osObj)) )
+        {
+            unsigned int arrayCount = arrayMember->getCount();
+            
+            for(unsigned int i=0; i<arrayCount; i++)
+            {
+                osNum = OSDynamicCast(OSNumber, arrayMember->getObject(i));
+                if(!osNum) {
+                    continue;
+                }
+                
+                feature_value = osNum->unsigned32BitValue();
+                feature_id = (uint16_t)(feature_value >> 16);
+
+                if( feature_id == (uint16_t)removeFeatureID )
+                {
+                    // Remove this node
+                    if( 1 == arrayCount ) {
+                        // If the array only contains one element, remove
+                        // the whole thing.
+                        features->removeObject(dictKey);
+                    } else {
+                        // Otherwise remove the element from a copy of the array.
+                        arrayMemberCopy = OSArray::withArray(arrayMember);
+                        if (arrayMemberCopy)
+                        {
+                            arrayMemberCopy->removeObject(i);
+                            features->setObject(dictKey, arrayMemberCopy);
+                            arrayMemberCopy->release();
+                        }
+                    }
+
+                    madeAChange = true;
+                    break;
+                }
+            }
+        }    
+    }
+    
+    dictIterator->release();
+    
+    if( madeAChange )
+    {
+        ret = kIOReturnSuccess;    
+
+        setProperty(kRootDomainSupportedFeatures, features);
+    
+        // Notify EnergySaver and all those in user space so they might
+        // re-populate their feature specific UI    
+        if(pmPowerStateQueue) {
+            pmPowerStateQueue->submitPowerEvent( kPowerEventFeatureChanged );
+        }
+    } else {
+        ret = kIOReturnNotFound;
+    }
+    
 exit:
-	if (featuresDictLock) {
-		IOLockUnlock(featuresDictLock);
-	}
-	return ret;
+    if(features)    features->release();
+    if(featuresDictLock) IOLockUnlock(featuresDictLock);    
+    return ret;
 }
 
 //******************************************************************************
@@ -5043,19 +3225,19 @@
 // supported feature.
 //******************************************************************************
 
-void
-IOPMrootDomain::publishPMSetting(
-	const OSSymbol * feature, uint32_t where, uint32_t * featureID )
-{
-	if (noPublishPMSettings &&
-	    (noPublishPMSettings->getNextIndexOfObject(feature, 0) != (unsigned int)-1)) {
-		// Setting found in noPublishPMSettings array
-		*featureID = kBadPMFeatureID;
-		return;
-	}
-
-	publishFeature(
-		feature->getCStringNoCopy(), where, featureID);
+void IOPMrootDomain::publishPMSetting(
+    const OSSymbol * feature, uint32_t where, uint32_t * featureID )
+{
+    if (noPublishPMSettings &&
+        (noPublishPMSettings->getNextIndexOfObject(feature, 0) != (unsigned int)-1))
+    {
+        // Setting found in noPublishPMSettings array
+        *featureID = kBadPMFeatureID;
+        return;
+    }
+
+    publishFeature(
+        feature->getCStringNoCopy(), where, featureID);
 }
 
 //******************************************************************************
@@ -5065,104 +3247,82 @@
 // drivers. Should be called only by IOPMrootDomain::setProperties.
 //******************************************************************************
 
-IOReturn
-IOPMrootDomain::setPMSetting(
-	const OSSymbol  *type,
-	OSObject        *object )
-{
-	PMSettingCallEntry  *entries = NULL;
-	OSSharedPtr<OSArray>    chosen;
-	const OSArray       *array;
-	PMSettingObject     *pmso;
-	thread_t            thisThread;
-	int                 i, j, count, capacity;
-	bool                ok = false;
-	IOReturn            ret;
-
-	if (NULL == type) {
-		return kIOReturnBadArgument;
-	}
-
-	PMSETTING_LOCK();
-
-	// Update settings dict so changes are visible from copyPMSetting().
-	fPMSettingsDict->setObject(type, object);
-
-	// Prep all PMSetting objects with the given 'type' for callout.
-	array = OSDynamicCast(OSArray, settingsCallbacks->getObject(type));
-	if (!array || ((capacity = array->getCount()) == 0)) {
-		goto unlock_exit;
-	}
-
-	// Array to retain PMSetting objects targeted for callout.
-	chosen = OSArray::withCapacity(capacity);
-	if (!chosen) {
-		goto unlock_exit; // error
-	}
-	entries = IONew(PMSettingCallEntry, capacity);
-	if (!entries) {
-		goto unlock_exit; // error
-	}
-	memset(entries, 0, sizeof(PMSettingCallEntry) * capacity);
-
-	thisThread = current_thread();
-
-	for (i = 0, j = 0; i < capacity; i++) {
-		pmso = (PMSettingObject *) array->getObject(i);
-		if (pmso->disabled) {
-			continue;
-		}
-		entries[j].thread = thisThread;
-		queue_enter(&pmso->calloutQueue, &entries[j], PMSettingCallEntry *, link);
-		chosen->setObject(pmso);
-		j++;
-	}
-	count = j;
-	if (!count) {
-		goto unlock_exit;
-	}
-
-	PMSETTING_UNLOCK();
-
-	// Call each pmso in the chosen array.
-	for (i = 0; i < count; i++) {
-		pmso = (PMSettingObject *) chosen->getObject(i);
-		ret = pmso->dispatchPMSetting(type, object);
-		if (ret == kIOReturnSuccess) {
-			// At least one setting handler was successful
-			ok = true;
-#if DEVELOPMENT || DEBUG
-		} else {
-			// Log the handler and kext that failed
-			OSSharedPtr<const OSSymbol> kextName = copyKextIdentifierWithAddress((vm_address_t) pmso->func);
-			if (kextName) {
-				DLOG("PMSetting(%s) error 0x%x from %s\n",
-				    type->getCStringNoCopy(), ret, kextName->getCStringNoCopy());
-			}
-#endif
-		}
-	}
-
-	PMSETTING_LOCK();
-	for (i = 0; i < count; i++) {
-		pmso = (PMSettingObject *) chosen->getObject(i);
-		queue_remove(&pmso->calloutQueue, &entries[i], PMSettingCallEntry *, link);
-		if (pmso->waitThread) {
-			PMSETTING_WAKEUP(pmso);
-		}
-	}
-
-	if (ok) {
-		recordRTCAlarm(type, object);
-	}
+IOReturn IOPMrootDomain::setPMSetting(
+    const OSSymbol  *type,
+    OSObject        *object )
+{
+    PMSettingCallEntry  *entries = 0;
+    OSArray             *chosen  = 0;
+    const OSArray       *array;
+    PMSettingObject     *pmso;
+    thread_t            thisThread;
+    int                 i, j, count, capacity;
+
+    if (NULL == type)
+        return kIOReturnBadArgument;
+
+    PMSETTING_LOCK();
+
+    // Update settings dict so changes are visible from copyPMSetting().    
+    fPMSettingsDict->setObject(type, object);
+
+    // Prep all PMSetting objects with the given 'type' for callout.
+    array = (const OSArray *) settingsCallbacks->getObject(type);
+    if (!array || ((capacity = array->getCount()) == 0))
+        goto unlock_exit;
+
+    // Array to retain PMSetting objects targeted for callout.
+    chosen = OSArray::withCapacity(capacity);
+    if (!chosen)
+        goto unlock_exit;   // error
+
+    entries = IONew(PMSettingCallEntry, capacity);
+    if (!entries)
+        goto unlock_exit;   // error
+    memset(entries, 0, sizeof(PMSettingCallEntry) * capacity);
+
+    thisThread = current_thread();
+
+    for (i = 0, j = 0; i<capacity; i++)
+    {
+        pmso = (PMSettingObject *) array->getObject(i);
+        if (pmso->disabled)
+            continue;
+        entries[j].thread = thisThread;        
+        queue_enter(&pmso->calloutQueue, &entries[j], PMSettingCallEntry *, link);
+        chosen->setObject(pmso);
+        j++;
+    }
+    count = j;
+    if (!count)
+        goto unlock_exit; 
+
+    PMSETTING_UNLOCK();
+
+    // Call each pmso in the chosen array.
+    for (i=0; i<count; i++)
+    {
+        pmso = (PMSettingObject *) chosen->getObject(i);
+        pmso->dispatchPMSetting(type, object);
+    }
+
+    PMSETTING_LOCK();
+    for (i=0; i<count; i++)
+    {
+        pmso = (PMSettingObject *) chosen->getObject(i);
+        queue_remove(&pmso->calloutQueue, &entries[i], PMSettingCallEntry *, link);
+        if (pmso->waitThread)
+        {
+            PMSETTING_WAKEUP(pmso);
+        }
+    }
 unlock_exit:
-	PMSETTING_UNLOCK();
-
-	if (entries) {
-		IODelete(entries, PMSettingCallEntry, capacity);
-	}
-
-	return kIOReturnSuccess;
+    PMSETTING_UNLOCK();
+
+    if (chosen)  chosen->release();
+    if (entries) IODelete(entries, PMSettingCallEntry, capacity);
+
+    return kIOReturnSuccess;
 }
 
 //******************************************************************************
@@ -5172,21 +3332,21 @@
 // notifications.
 //******************************************************************************
 
-OSSharedPtr<OSObject>
-IOPMrootDomain::copyPMSetting(
-	OSSymbol *whichSetting)
-{
-	OSSharedPtr<OSObject> obj;
-
-	if (!whichSetting) {
-		return NULL;
-	}
-
-	PMSETTING_LOCK();
-	obj.reset(fPMSettingsDict->getObject(whichSetting), OSRetain);
-	PMSETTING_UNLOCK();
-
-	return obj;
+OSObject * IOPMrootDomain::copyPMSetting(
+    OSSymbol *whichSetting)
+{
+    OSObject *obj = NULL;
+
+    if(!whichSetting) return NULL;
+
+    PMSETTING_LOCK();
+    obj = fPMSettingsDict->getObject(whichSetting);
+    if(obj) {
+        obj->retain();
+    }
+    PMSETTING_UNLOCK();
+    
+    return obj;
 }
 
 //******************************************************************************
@@ -5195,18 +3355,17 @@
 // direct wrapper to registerPMSettingController with uint32_t power source arg
 //******************************************************************************
 
-IOReturn
-IOPMrootDomain::registerPMSettingController(
-	const OSSymbol *                settings[],
-	IOPMSettingControllerCallback   func,
-	OSObject                        *target,
-	uintptr_t                       refcon,
-	OSObject                        **handle)
-{
-	return registerPMSettingController(
-		settings,
-		(kIOPMSupportedOnAC | kIOPMSupportedOnBatt | kIOPMSupportedOnUPS),
-		func, target, refcon, handle);
+IOReturn IOPMrootDomain::registerPMSettingController(
+    const OSSymbol *                settings[],
+    IOPMSettingControllerCallback   func,
+    OSObject                        *target,
+    uintptr_t                       refcon,
+    OSObject                        **handle)
+{
+    return registerPMSettingController( 
+            settings,
+            (kIOPMSupportedOnAC | kIOPMSupportedOnBatt | kIOPMSupportedOnUPS),
+            func, target, refcon, handle);
 }
 
 //******************************************************************************
@@ -5218,9 +3377,9 @@
 //  * settings - An OSArray containing OSSymbols. Caller should populate this
 //          array with a list of settings caller wants notifications from.
 //  * func - A C function callback of the type IOPMSettingControllerCallback
-//  * target - caller may provide an OSObject *, which PM will pass as an
+//  * target - caller may provide an OSObject *, which PM will pass as an 
 //          target to calls to "func"
-//  * refcon - caller may provide an void *, which PM will pass as an
+//  * refcon - caller may provide an void *, which PM will pass as an 
 //          argument to calls to "func"
 //  * handle - This is a return argument. We will populate this pointer upon
 //          call success. Hold onto this and pass this argument to
@@ -5229,54 +3388,55 @@
 //      kIOReturnSuccess on success
 //******************************************************************************
 
-IOReturn
-IOPMrootDomain::registerPMSettingController(
-	const OSSymbol *                settings[],
-	uint32_t                        supportedPowerSources,
-	IOPMSettingControllerCallback   func,
-	OSObject                        *target,
-	uintptr_t                       refcon,
-	OSObject                        **handle)
-{
-	PMSettingObject *pmso = NULL;
-	OSObject        *pmsh = NULL;
-	int             i;
-
-	if (NULL == settings ||
-	    NULL == func ||
-	    NULL == handle) {
-		return kIOReturnBadArgument;
-	}
-
-	pmso = PMSettingObject::pmSettingObject(
-		(IOPMrootDomain *) this, func, target,
-		refcon, supportedPowerSources, settings, &pmsh);
-
-	if (!pmso) {
-		*handle = NULL;
-		return kIOReturnInternalError;
-	}
-
-	PMSETTING_LOCK();
-	for (i = 0; settings[i]; i++) {
-		OSSharedPtr<OSArray> newList;
-		OSArray *list = OSDynamicCast(OSArray, settingsCallbacks->getObject(settings[i]));
-		if (!list) {
-			// New array of callbacks for this setting
-			newList = OSArray::withCapacity(1);
-			settingsCallbacks->setObject(settings[i], newList.get());
-			list = newList.get();
-		}
-
-		// Add caller to the callback list
-		list->setObject(pmso);
-	}
-	PMSETTING_UNLOCK();
-
-	// Return handle to the caller, the setting object is private.
-	*handle = pmsh;
-
-	return kIOReturnSuccess;
+IOReturn IOPMrootDomain::registerPMSettingController(
+    const OSSymbol *                settings[],
+    uint32_t                        supportedPowerSources,
+    IOPMSettingControllerCallback   func,
+    OSObject                        *target,
+    uintptr_t                       refcon,
+    OSObject                        **handle)
+{
+    PMSettingObject *pmso = NULL;
+    OSObject        *pmsh = NULL;
+    OSArray         *list = NULL;
+    int             i;
+
+    if (NULL == settings ||
+        NULL == func     ||
+        NULL == handle)
+    {
+        return kIOReturnBadArgument;
+    }
+
+    pmso = PMSettingObject::pmSettingObject(
+                (IOPMrootDomain *) this, func, target, 
+                refcon, supportedPowerSources, settings, &pmsh);
+
+    if (!pmso) {
+        *handle = NULL;
+        return kIOReturnInternalError;
+    }
+
+    PMSETTING_LOCK();
+    for (i=0; settings[i]; i++)
+    {
+        list = (OSArray *) settingsCallbacks->getObject(settings[i]);
+        if (!list) {
+            // New array of callbacks for this setting
+            list = OSArray::withCapacity(1);
+            settingsCallbacks->setObject(settings[i], list);
+            list->release();
+        }
+
+        // Add caller to the callback list
+        list->setObject(pmso);
+    }
+    PMSETTING_UNLOCK();
+
+    // Return handle to the caller, the setting object is private.
+    *handle = pmsh;
+
+    return kIOReturnSuccess;
 }
 
 //******************************************************************************
@@ -5285,54 +3445,58 @@
 // Only called from PMSettingObject.
 //******************************************************************************
 
-void
-IOPMrootDomain::deregisterPMSettingObject( PMSettingObject * pmso )
-{
-	thread_t                thisThread = current_thread();
-	PMSettingCallEntry      *callEntry;
-	OSSharedPtr<OSCollectionIterator>    iter;
-	OSSymbol                *sym;
-	OSArray                 *array;
-	int                     index;
-	bool                    wait;
-
-	PMSETTING_LOCK();
-
-	pmso->disabled = true;
-
-	// Wait for all callout threads to finish.
-	do {
-		wait = false;
-		queue_iterate(&pmso->calloutQueue, callEntry, PMSettingCallEntry *, link)
-		{
-			if (callEntry->thread != thisThread) {
-				wait = true;
-				break;
-			}
-		}
-		if (wait) {
-			assert(NULL == pmso->waitThread);
-			pmso->waitThread = thisThread;
-			PMSETTING_WAIT(pmso);
-			pmso->waitThread = NULL;
-		}
-	} while (wait);
-
-	// Search each PM settings array in the kernel.
-	iter = OSCollectionIterator::withCollection(settingsCallbacks.get());
-	if (iter) {
-		while ((sym = OSDynamicCast(OSSymbol, iter->getNextObject()))) {
-			array = OSDynamicCast(OSArray, settingsCallbacks->getObject(sym));
-			index = array->getNextIndexOfObject(pmso, 0);
-			if (-1 != index) {
-				array->removeObject(index);
-			}
-		}
-	}
-
-	PMSETTING_UNLOCK();
-
-	pmso->release();
+void IOPMrootDomain::deregisterPMSettingObject( PMSettingObject * pmso )
+{
+    thread_t                thisThread = current_thread();
+    PMSettingCallEntry      *callEntry;
+    OSCollectionIterator    *iter;
+    OSSymbol                *sym;
+    OSArray                 *array;
+    int                     index;
+    bool                    wait;
+
+    PMSETTING_LOCK();
+
+    pmso->disabled = true;
+
+    // Wait for all callout threads to finish.
+    do {
+        wait = false;
+        queue_iterate(&pmso->calloutQueue, callEntry, PMSettingCallEntry *, link)
+        {
+            if (callEntry->thread != thisThread)
+            {
+                wait = true;
+                break;
+            }
+        }
+        if (wait)
+        {
+            assert(0 == pmso->waitThread);
+            pmso->waitThread = thisThread;
+            PMSETTING_WAIT(pmso);
+            pmso->waitThread = 0;
+        }
+    } while (wait);
+
+    // Search each PM settings array in the kernel.
+    iter = OSCollectionIterator::withCollection(settingsCallbacks);
+    if (iter) 
+    {
+        while ((sym = OSDynamicCast(OSSymbol, iter->getNextObject())))
+        {
+            array = (OSArray *) settingsCallbacks->getObject(sym);
+            index = array->getNextIndexOfObject(pmso, 0);
+            if (-1 != index) {
+                array->removeObject(index);
+            }
+        }
+        iter->release();
+    }
+
+    PMSETTING_UNLOCK();
+
+    pmso->release();
 }
 
 //******************************************************************************
@@ -5345,857 +3509,483 @@
 // only x86 has explicit support in the IntelCPUPowerManagement kext
 //******************************************************************************
 
-void
-IOPMrootDomain::informCPUStateChange(
-	uint32_t type,
-	uint32_t value )
+void IOPMrootDomain::informCPUStateChange(
+    uint32_t type, 
+    uint32_t value )
 {
 #if defined(__i386__) || defined(__x86_64__)
 
-	pmioctlVariableInfo_t varInfoStruct;
-	int                 pmCPUret = 0;
-	const char          *varNameStr = NULL;
-	int32_t             *varIndex   = NULL;
-
-	if (kInformAC == type) {
-		varNameStr = kIOPMRootDomainBatPowerCString;
-		varIndex = &idxPMCPULimitedPower;
-	} else if (kInformLid == type) {
-		varNameStr = kIOPMRootDomainLidCloseCString;
-		varIndex = &idxPMCPUClamshell;
-	} else {
-		return;
-	}
-
-	// Set the new value!
-	// pmCPUControl will assign us a new ID if one doesn't exist yet
-	bzero(&varInfoStruct, sizeof(pmioctlVariableInfo_t));
-	varInfoStruct.varID         = *varIndex;
-	varInfoStruct.varType       = vBool;
-	varInfoStruct.varInitValue  = value;
-	varInfoStruct.varCurValue   = value;
-	strlcpy((char *)varInfoStruct.varName,
-	    (const char *)varNameStr,
-	    sizeof(varInfoStruct.varName));
-
-	// Set!
-	pmCPUret = pmCPUControl( PMIOCSETVARINFO, (void *)&varInfoStruct );
-
-	// pmCPU only assigns numerical id's when a new varName is specified
-	if ((0 == pmCPUret)
-	    && (*varIndex == kCPUUnknownIndex)) {
-		// pmCPUControl has assigned us a new variable ID.
-		// Let's re-read the structure we just SET to learn that ID.
-		pmCPUret = pmCPUControl( PMIOCGETVARNAMEINFO, (void *)&varInfoStruct );
-
-		if (0 == pmCPUret) {
-			// Store it in idxPMCPUClamshell or idxPMCPULimitedPower
-			*varIndex = varInfoStruct.varID;
+    pmioctlVariableInfo_t varInfoStruct;                            
+    int                 pmCPUret = 0;
+    const char          *varNameStr = NULL;
+    int32_t             *varIndex   = NULL;
+
+    if (kInformAC == type) {
+        varNameStr = kIOPMRootDomainBatPowerCString;
+        varIndex = &idxPMCPULimitedPower;
+    } else if (kInformLid == type) {
+        varNameStr = kIOPMRootDomainLidCloseCString;
+        varIndex = &idxPMCPUClamshell;
+    } else {
+        return;
+    }
+    
+    // Set the new value!
+    // pmCPUControl will assign us a new ID if one doesn't exist yet
+    bzero(&varInfoStruct, sizeof(pmioctlVariableInfo_t));
+    varInfoStruct.varID         = *varIndex;
+    varInfoStruct.varType       = vBool;
+    varInfoStruct.varInitValue  = value;
+    varInfoStruct.varCurValue   = value;
+    strncpy( (char *)varInfoStruct.varName,
+             (const char *)varNameStr,
+             strlen(varNameStr) + 1 );                 
+    
+    // Set!
+    pmCPUret = pmCPUControl( PMIOCSETVARINFO, (void *)&varInfoStruct );
+
+    // pmCPU only assigns numerical id's when a new varName is specified
+    if ((0 == pmCPUret)
+        && (*varIndex == kCPUUnknownIndex))
+    {
+        // pmCPUControl has assigned us a new variable ID. 
+        // Let's re-read the structure we just SET to learn that ID.
+        pmCPUret = pmCPUControl( PMIOCGETVARNAMEINFO, (void *)&varInfoStruct );
+
+        if (0 == pmCPUret) 
+        {        
+            // Store it in idxPMCPUClamshell or idxPMCPULimitedPower
+            *varIndex = varInfoStruct.varID;
+        }
+    } 
+    
+    return;
+    
+#endif /* __i386__ || __x86_64__ */
+}
+
+// MARK: -
+// MARK: Deep Sleep Policy
+
+#if HIBERNATION
+
+//******************************************************************************
+// evaluateSystemSleepPolicy
+//******************************************************************************
+
+struct IOPMSystemSleepPolicyEntry
+{
+    uint32_t    factorMask;
+    uint32_t    factorBits;
+    uint32_t    sleepFlags;
+    uint32_t    wakeEvents;
+};
+
+struct IOPMSystemSleepPolicyTable
+{
+    uint8_t     signature[4];
+    uint16_t    version;
+    uint16_t    entryCount;
+    IOPMSystemSleepPolicyEntry  entries[];
+};
+
+enum {
+    kIOPMSleepFactorSleepTimerWake          = 0x00000001,
+    kIOPMSleepFactorLidOpen                 = 0x00000002,
+    kIOPMSleepFactorACPower                 = 0x00000004,
+    kIOPMSleepFactorLowBattery              = 0x00000008,
+    kIOPMSleepFactorDeepSleepNoDelay        = 0x00000010,
+    kIOPMSleepFactorDeepSleepDemand         = 0x00000020,
+    kIOPMSleepFactorDeepSleepDisable        = 0x00000040,
+    kIOPMSleepFactorUSBExternalDevice       = 0x00000080,
+    kIOPMSleepFactorBluetoothHIDDevice      = 0x00000100,
+    kIOPMSleepFactorExternalMediaMounted    = 0x00000200,
+    kIOPMSleepFactorDriverAssertBit5        = 0x00000400,   /* Reserved for ThunderBolt */
+    kIOPMSleepFactorDriverAssertBit6        = 0x00000800,
+    kIOPMSleepFactorDriverAssertBit7        = 0x00001000    /* Reserved for legacy I/O */
+};
+
+enum {
+    kSleepPhaseEarly, kSleepPhaseFinal
+};
+
+bool IOPMrootDomain::evaluateSystemSleepPolicy( IOPMSystemSleepParameters * p, int sleepPhase )
+{
+    const IOPMSystemSleepPolicyTable * pt;
+    OSObject *  prop = 0;
+    OSData *    policyData;
+    uint32_t    currentFactors;
+    uint32_t    deepSleepDelay = 0;
+    bool        success = false;
+
+    if (getProperty(kIOPMDeepSleepEnabledKey) != kOSBooleanTrue)
+        return false;
+
+    getSleepOption(kIOPMDeepSleepDelayKey, &deepSleepDelay);
+
+    prop = getServiceRoot()->copyProperty(kIOPlatformSystemSleepPolicyKey);
+    if (!prop)
+        return false;
+
+    policyData = OSDynamicCast(OSData, prop);
+    if (!policyData ||
+        (policyData->getLength() < sizeof(IOPMSystemSleepPolicyTable)))
+    {
+        goto done;
+    }
+
+    pt = (const IOPMSystemSleepPolicyTable *) policyData->getBytesNoCopy();
+    if ((pt->signature[0] != 'S') ||
+        (pt->signature[1] != 'L') ||
+        (pt->signature[2] != 'P') ||
+        (pt->signature[3] != 'T') ||
+        (pt->version      != 1)   ||
+        (pt->entryCount   == 0))
+    {
+        goto done;
+    }
+
+    if ((policyData->getLength() - sizeof(IOPMSystemSleepPolicyTable)) !=
+        (sizeof(IOPMSystemSleepPolicyEntry) * pt->entryCount))
+    {
+        goto done;
+    }
+
+    currentFactors = 0;
+    if (getPMAssertionLevel(kIOPMDriverAssertionUSBExternalDeviceBit) !=
+        kIOPMDriverAssertionLevelOff)
+        currentFactors |= kIOPMSleepFactorUSBExternalDevice;
+    if (getPMAssertionLevel(kIOPMDriverAssertionBluetoothHIDDevicePairedBit) !=
+        kIOPMDriverAssertionLevelOff)
+        currentFactors |= kIOPMSleepFactorBluetoothHIDDevice;
+    if (getPMAssertionLevel(kIOPMDriverAssertionExternalMediaMountedBit) !=
+        kIOPMDriverAssertionLevelOff)
+        currentFactors |= kIOPMSleepFactorExternalMediaMounted;
+    if (getPMAssertionLevel(kIOPMDriverAssertionReservedBit5) !=    /* AssertionBit5 = Thunderbolt */
+        kIOPMDriverAssertionLevelOff)
+        currentFactors |= kIOPMSleepFactorDriverAssertBit5;
+    if (getPMAssertionLevel(kIOPMDriverAssertionReservedBit7) !=
+        kIOPMDriverAssertionLevelOff)
+        currentFactors |= kIOPMSleepFactorDriverAssertBit7;
+    if (0 == deepSleepDelay)
+        currentFactors |= kIOPMSleepFactorDeepSleepNoDelay;
+    if (!clamshellClosed)
+        currentFactors |= kIOPMSleepFactorLidOpen;
+    if (acAdaptorConnected)
+        currentFactors |= kIOPMSleepFactorACPower;
+    if (lowBatteryCondition)
+        currentFactors |= kIOPMSleepFactorLowBattery;
+    if (sleepTimerMaintenance)
+        currentFactors |= kIOPMSleepFactorSleepTimerWake;
+
+    // pmset overrides
+    if ((hibernateMode & kIOHibernateModeOn) == 0)
+        currentFactors |= kIOPMSleepFactorDeepSleepDisable;
+    else if ((hibernateMode & kIOHibernateModeSleep) == 0)
+        currentFactors |= kIOPMSleepFactorDeepSleepDemand;
+    
+    DLOG("Sleep policy %u entries, current factors 0x%x\n",
+        pt->entryCount, currentFactors);
+
+    for (uint32_t i = 0; i < pt->entryCount; i++)
+    {
+        const IOPMSystemSleepPolicyEntry * policyEntry = &pt->entries[i];
+
+        DLOG("factor mask 0x%08x, bits 0x%08x, flags 0x%08x, wake 0x%08x\n",
+            policyEntry->factorMask, policyEntry->factorBits,
+            policyEntry->sleepFlags, policyEntry->wakeEvents);
+
+        if ((currentFactors ^ policyEntry->factorBits) & policyEntry->factorMask)
+            continue;   // mismatch, try next
+
+        if (p)
+        {
+            p->version    = 1;
+            p->sleepFlags = policyEntry->sleepFlags;
+            p->sleepTimer = 0;
+            p->wakeEvents = policyEntry->wakeEvents;
+            if (p->sleepFlags & kIOPMSleepFlagSleepTimerEnable)
+            {
+                if (kSleepPhaseFinal == sleepPhase)
+                {
+                    clock_sec_t now_secs = gIOLastSleepTime.tv_sec;
+
+                    if (!_standbyTimerResetSeconds ||
+                        (now_secs <= _standbyTimerResetSeconds))
+                    {
+                        // Reset standby timer adjustment
+                        _standbyTimerResetSeconds = now_secs;
+                        DLOG("standby delay %u, reset %u\n",
+                            deepSleepDelay, (uint32_t) _standbyTimerResetSeconds);
+                    }
+                    else if (deepSleepDelay)
+                    {
+                        // Shorten the standby delay timer
+                        clock_sec_t elapsed = now_secs - _standbyTimerResetSeconds;
+                        if (deepSleepDelay > elapsed)
+                            deepSleepDelay -= elapsed;
+                        else
+                            deepSleepDelay = 1; // must be > 0
+
+                        DLOG("standby delay %u, elapsed %u\n",
+                            deepSleepDelay, (uint32_t) elapsed);
+                    }
+                }
+                p->sleepTimer = deepSleepDelay;
+            }
+            else if (kSleepPhaseFinal == sleepPhase)
+            {
+                // A sleep that does not enable the sleep timer will reset
+                // the standby delay adjustment.
+                _standbyTimerResetSeconds = 0;
+            }            
+        }
+
+        DLOG("matched policy entry %u\n", i);
+        success = true;
+        break;
+    }
+
+done:
+    if (prop)
+        prop->release();
+
+    return success;
+}
+
+void IOPMrootDomain::evaluateSystemSleepPolicyEarly( void )
+{
+    IOPMSystemSleepParameters   params;
+
+    // Evaluate sleep policy before driver sleep phase.
+
+    DLOG("%s\n", __FUNCTION__);
+    removeProperty(kIOPMSystemSleepParametersKey);
+
+    // Full wake resets the standby timer delay adjustment
+    if (_highestCapability & kIOPMSystemCapabilityGraphics)
+        _standbyTimerResetSeconds = 0;
+
+    hibernateDisabled = false;
+    hibernateMode = 0;
+    getSleepOption(kIOHibernateModeKey, &hibernateMode);
+
+    if (!hibernateNoDefeat &&
+        evaluateSystemSleepPolicy(&params, kSleepPhaseEarly) &&
+        ((params.sleepFlags & kIOPMSleepFlagHibernate) == 0))
+    {
+        hibernateDisabled = true;
+    }
+}
+
+void IOPMrootDomain::evaluateSystemSleepPolicyFinal( void )
+{
+    IOPMSystemSleepParameters   params;
+    OSData *                    paramsData;
+
+    // Evaluate sleep policy after drivers but before platform sleep.
+
+    DLOG("%s\n", __FUNCTION__);
+
+    if (evaluateSystemSleepPolicy(&params, kSleepPhaseFinal))
+    {
+        if ((hibernateDisabled || hibernateAborted) &&
+            (params.sleepFlags & kIOPMSleepFlagHibernate))
+        {
+            // Should hibernate but unable to or aborted.
+            // Arm timer for a short sleep and retry or wake fully.
+
+            params.sleepFlags &= ~kIOPMSleepFlagHibernate;
+            params.sleepFlags |= kIOPMSleepFlagSleepTimerEnable;
+            params.sleepTimer = 1;
+            hibernateNoDefeat = true;
+            DLOG("wake in %u secs for hibernateDisabled %d, hibernateAborted %d\n",
+                        params.sleepTimer, hibernateDisabled, hibernateAborted);
+        }
+        else
+            hibernateNoDefeat = false;
+
+        paramsData = OSData::withBytes(&params, sizeof(params));
+        if (paramsData)
+        {
+            setProperty(kIOPMSystemSleepParametersKey, paramsData);
+            paramsData->release();
+        }
+
+        if (params.sleepFlags & kIOPMSleepFlagHibernate)
+        {
+            // Force hibernate
+            gIOHibernateMode &= ~kIOHibernateModeSleep;
+        }
+    }
+}
+
+bool IOPMrootDomain::getHibernateSettings(
+    uint32_t *  hibernateMode,
+    uint32_t *  hibernateFreeRatio,
+    uint32_t *  hibernateFreeTime )
+{
+    bool ok = getSleepOption(kIOHibernateModeKey, hibernateMode);
+    getSleepOption(kIOHibernateFreeRatioKey, hibernateFreeRatio);
+    getSleepOption(kIOHibernateFreeTimeKey, hibernateFreeTime);
+    if (hibernateDisabled)
+        *hibernateMode = 0;
+    DLOG("hibernateMode 0x%x\n", *hibernateMode);
+    return ok;
+}
+
+bool IOPMrootDomain::getSleepOption( const char * key, uint32_t * option )
+{
+    OSObject *      optionsProp;
+    OSDictionary *  optionsDict;
+    OSObject *      obj = 0;
+    OSNumber *      num;
+    bool            ok = false;
+
+    optionsProp = copyProperty(kRootDomainSleepOptionsKey);
+    optionsDict = OSDynamicCast(OSDictionary, optionsProp);
+    
+    if (optionsDict)
+    {
+        obj = optionsDict->getObject(key);
+        if (obj) obj->retain();
+    }
+    if (!obj)
+    {
+        obj = copyProperty(key);
+    }
+    if (obj && (num = OSDynamicCast(OSNumber, obj)))
+    {
+        *option = num->unsigned32BitValue();
+        ok = true;
+    }
+
+    if (obj)
+        obj->release();
+    if (optionsProp)
+        optionsProp->release();
+
+    return true;
+}
+#endif /* HIBERNATION */
+
+// MARK: -
+// MARK: Shutdown and Restart
+
+//******************************************************************************
+// handlePlatformHaltRestart
+//
+//******************************************************************************
+
+struct HaltRestartApplierContext {
+	IOPMrootDomain *	RootDomain;
+	unsigned long		PowerState;
+	IOPMPowerFlags		PowerFlags;
+	UInt32				MessageType;
+	UInt32				Counter;
+};
+
+static void
+platformHaltRestartApplier( OSObject * object, void * context )
+{
+	IOPowerStateChangeNotification	notify;
+	HaltRestartApplierContext *		ctx;
+	AbsoluteTime					startTime;
+	UInt32							deltaTime;
+
+	ctx = (HaltRestartApplierContext *) context;
+	
+	memset(&notify, 0, sizeof(notify));
+    notify.powerRef    = (void *)ctx->Counter;
+    notify.returnValue = 0;
+    notify.stateNumber = ctx->PowerState;
+    notify.stateFlags  = ctx->PowerFlags;
+
+	clock_get_uptime(&startTime);
+    ctx->RootDomain->messageClient( ctx->MessageType, object, (void *)&notify );
+	deltaTime = computeDeltaTimeMS(&startTime);
+
+	if ((deltaTime > kPMHaltTimeoutMS) ||
+        (gIOKitDebug & kIOLogPMRootDomain))
+	{
+		_IOServiceInterestNotifier * notifier;
+		notifier = OSDynamicCast(_IOServiceInterestNotifier, object);
+
+		// IOService children of IOPMrootDomain are not instrumented.
+		// Only IORootParent currently falls under that group.
+
+		if (notifier)
+		{
+			LOG("%s handler %p took %u ms\n",
+				(ctx->MessageType == kIOMessageSystemWillPowerOff) ? "PowerOff" :
+					 (ctx->MessageType == kIOMessageSystemPagingOff) ? "PagingOff" : "Restart",
+				notifier->handler, (uint32_t) deltaTime );
 		}
 	}
 
-	return;
-
-#endif /* __i386__ || __x86_64__ */
-}
-
-// MARK: -
-// MARK: Deep Sleep Policy
-
-#if HIBERNATION
-
-//******************************************************************************
-// evaluateSystemSleepPolicy
-//******************************************************************************
-
-#define kIOPlatformSystemSleepPolicyKey     "IOPlatformSystemSleepPolicy"
-
-// Sleep flags
-enum {
-	kIOPMSleepFlagHibernate         = 0x00000001,
-	kIOPMSleepFlagSleepTimerEnable  = 0x00000002
-};
-
-struct IOPMSystemSleepPolicyEntry {
-	uint32_t    factorMask;
-	uint32_t    factorBits;
-	uint32_t    sleepFlags;
-	uint32_t    wakeEvents;
-} __attribute__((packed));
-
-struct IOPMSystemSleepPolicyTable {
-	uint32_t    signature;
-	uint16_t    version;
-	uint16_t    entryCount;
-	IOPMSystemSleepPolicyEntry  entries[];
-} __attribute__((packed));
-
-enum {
-	kIOPMSleepAttributeHibernateSetup   = 0x00000001,
-	kIOPMSleepAttributeHibernateSleep   = 0x00000002
-};
-
-static uint32_t
-getSleepTypeAttributes( uint32_t sleepType )
-{
-	static const uint32_t sleepTypeAttributes[kIOPMSleepTypeLast] =
+	ctx->Counter++;
+}
+
+void IOPMrootDomain::handlePlatformHaltRestart( UInt32 pe_type )
+{
+	HaltRestartApplierContext	ctx;
+	AbsoluteTime				startTime;
+	UInt32						deltaTime;
+
+	memset(&ctx, 0, sizeof(ctx));
+	ctx.RootDomain = this;
+
+	clock_get_uptime(&startTime);
+	switch (pe_type)
 	{
-		/* invalid   */ 0,
-		/* abort     */ 0,
-		/* normal    */ 0,
-		/* safesleep */ kIOPMSleepAttributeHibernateSetup,
-		/* hibernate */ kIOPMSleepAttributeHibernateSetup | kIOPMSleepAttributeHibernateSleep,
-		/* standby   */ kIOPMSleepAttributeHibernateSetup | kIOPMSleepAttributeHibernateSleep,
-		/* poweroff  */ kIOPMSleepAttributeHibernateSetup | kIOPMSleepAttributeHibernateSleep,
-		/* deepidle  */ 0
-	};
-
-	if (sleepType >= kIOPMSleepTypeLast) {
-		return 0;
+		case kPEHaltCPU:
+        case kPEUPSDelayHaltCPU:
+			ctx.PowerState  = OFF_STATE;
+			ctx.MessageType = kIOMessageSystemWillPowerOff;
+			break;
+
+		case kPERestartCPU:
+			ctx.PowerState  = RESTART_STATE;
+			ctx.MessageType = kIOMessageSystemWillRestart;
+			break;
+
+		case kPEPagingOff:
+			ctx.PowerState  = ON_STATE;
+			ctx.MessageType = kIOMessageSystemPagingOff;
+			IOService::updateConsoleUsers(NULL, kIOMessageSystemPagingOff);
+			break;
+
+		default:
+			return;
 	}
 
-	return sleepTypeAttributes[sleepType];
-}
-
-bool
-IOPMrootDomain::evaluateSystemSleepPolicy(
-	IOPMSystemSleepParameters * params, int sleepPhase, uint32_t * hibMode )
-{
-#define SLEEP_FACTOR(x) {(uint32_t) kIOPMSleepFactor ## x, #x}
-
-	static const IONamedValue factorValues[] = {
-		SLEEP_FACTOR( SleepTimerWake ),
-		SLEEP_FACTOR( LidOpen ),
-		SLEEP_FACTOR( ACPower ),
-		SLEEP_FACTOR( BatteryLow ),
-		SLEEP_FACTOR( StandbyNoDelay ),
-		SLEEP_FACTOR( StandbyForced ),
-		SLEEP_FACTOR( StandbyDisabled ),
-		SLEEP_FACTOR( USBExternalDevice ),
-		SLEEP_FACTOR( BluetoothHIDDevice ),
-		SLEEP_FACTOR( ExternalMediaMounted ),
-		SLEEP_FACTOR( ThunderboltDevice ),
-		SLEEP_FACTOR( RTCAlarmScheduled ),
-		SLEEP_FACTOR( MagicPacketWakeEnabled ),
-		SLEEP_FACTOR( HibernateForced ),
-		SLEEP_FACTOR( AutoPowerOffDisabled ),
-		SLEEP_FACTOR( AutoPowerOffForced ),
-		SLEEP_FACTOR( ExternalDisplay ),
-		SLEEP_FACTOR( NetworkKeepAliveActive ),
-		SLEEP_FACTOR( LocalUserActivity ),
-		SLEEP_FACTOR( HibernateFailed ),
-		SLEEP_FACTOR( ThermalWarning ),
-		SLEEP_FACTOR( DisplayCaptured ),
-		{ 0, NULL }
-	};
-
-	const IOPMSystemSleepPolicyTable * pt;
-	OSSharedPtr<OSObject>  prop;
-	OSData *    policyData;
-	uint64_t    currentFactors = 0;
-	char        currentFactorsBuf[512];
-	uint32_t    standbyDelay   = 0;
-	uint32_t    powerOffDelay  = 0;
-	uint32_t    powerOffTimer  = 0;
-	uint32_t    standbyTimer  = 0;
-	uint32_t    mismatch;
-	bool        standbyEnabled;
-	bool        powerOffEnabled;
-	bool        found = false;
-
-	// Get platform's sleep policy table
-	if (!gSleepPolicyHandler) {
-		prop = getServiceRoot()->copyProperty(kIOPlatformSystemSleepPolicyKey);
-		if (!prop) {
-			goto done;
-		}
+	// Notify legacy clients
+	applyToInterested(gIOPriorityPowerStateInterest, platformHaltRestartApplier, &ctx);
+
+    // For normal shutdown, turn off File Server Mode.
+    if (kPEHaltCPU == pe_type)
+    {
+        const OSSymbol * setting = OSSymbol::withCString(kIOPMSettingRestartOnPowerLossKey);
+        OSNumber * num = OSNumber::withNumber((unsigned long long) 0, 32);
+        if (setting && num)
+        {
+            setPMSetting(setting, num);
+            setting->release();
+            num->release();
+        }
+    }
+
+	if (kPEPagingOff != pe_type)
+	{
+		// Notify in power tree order
+		notifySystemShutdown(this, ctx.MessageType);
 	}
 
-	// Fetch additional settings
-	standbyEnabled = (getSleepOption(kIOPMDeepSleepDelayKey, &standbyDelay)
-	    && propertyHasValue(kIOPMDeepSleepEnabledKey, kOSBooleanTrue));
-	powerOffEnabled = (getSleepOption(kIOPMAutoPowerOffDelayKey, &powerOffDelay)
-	    && propertyHasValue(kIOPMAutoPowerOffEnabledKey, kOSBooleanTrue));
-	if (!getSleepOption(kIOPMAutoPowerOffTimerKey, &powerOffTimer)) {
-		powerOffTimer = powerOffDelay;
-	}
-	if (!getSleepOption(kIOPMDeepSleepTimerKey, &standbyTimer)) {
-		standbyTimer = standbyDelay;
-	}
-
-	DLOG("phase %d, standby %d delay %u timer %u, poweroff %d delay %u timer %u, hibernate 0x%x\n",
-	    sleepPhase, standbyEnabled, standbyDelay, standbyTimer,
-	    powerOffEnabled, powerOffDelay, powerOffTimer, *hibMode);
-
-	currentFactorsBuf[0] = 0;
-	// pmset level overrides
-	if ((*hibMode & kIOHibernateModeOn) == 0) {
-		if (!gSleepPolicyHandler) {
-			standbyEnabled  = false;
-			powerOffEnabled = false;
-		}
-	} else if (!(*hibMode & kIOHibernateModeSleep)) {
-		// Force hibernate (i.e. mode 25)
-		// If standby is enabled, force standy.
-		// If poweroff is enabled, force poweroff.
-		if (standbyEnabled) {
-			currentFactors |= kIOPMSleepFactorStandbyForced;
-		} else if (powerOffEnabled) {
-			currentFactors |= kIOPMSleepFactorAutoPowerOffForced;
-		} else {
-			currentFactors |= kIOPMSleepFactorHibernateForced;
-		}
-	}
-
-	// Current factors based on environment and assertions
-	if (sleepTimerMaintenance) {
-		currentFactors |= kIOPMSleepFactorSleepTimerWake;
-	}
-	if (standbyEnabled && sleepToStandby && !gSleepPolicyHandler) {
-		currentFactors |= kIOPMSleepFactorSleepTimerWake;
-	}
-	if (!clamshellClosed) {
-		currentFactors |= kIOPMSleepFactorLidOpen;
-	}
-	if (acAdaptorConnected) {
-		currentFactors |= kIOPMSleepFactorACPower;
-	}
-	if (lowBatteryCondition) {
-		hibernateMode = 0;
-		getSleepOption(kIOHibernateModeKey, &hibernateMode);
-		if ((hibernateMode & kIOHibernateModeOn) == 0) {
-			DLOG("HibernateMode is 0. Not sending LowBattery factor to IOPPF\n");
-		} else {
-			currentFactors |= kIOPMSleepFactorBatteryLow;
-		}
-	}
-	if (!standbyDelay || !standbyTimer) {
-		currentFactors |= kIOPMSleepFactorStandbyNoDelay;
-	}
-	if (standbyNixed || !standbyEnabled) {
-		currentFactors |= kIOPMSleepFactorStandbyDisabled;
-	}
-	if (resetTimers) {
-		currentFactors |= kIOPMSleepFactorLocalUserActivity;
-		currentFactors &= ~kIOPMSleepFactorSleepTimerWake;
-	}
-	if (getPMAssertionLevel(kIOPMDriverAssertionUSBExternalDeviceBit) !=
-	    kIOPMDriverAssertionLevelOff) {
-		currentFactors |= kIOPMSleepFactorUSBExternalDevice;
-	}
-	if (getPMAssertionLevel(kIOPMDriverAssertionBluetoothHIDDevicePairedBit) !=
-	    kIOPMDriverAssertionLevelOff) {
-		currentFactors |= kIOPMSleepFactorBluetoothHIDDevice;
-	}
-	if (getPMAssertionLevel(kIOPMDriverAssertionExternalMediaMountedBit) !=
-	    kIOPMDriverAssertionLevelOff) {
-		currentFactors |= kIOPMSleepFactorExternalMediaMounted;
-	}
-	if (getPMAssertionLevel(kIOPMDriverAssertionReservedBit5) !=
-	    kIOPMDriverAssertionLevelOff) {
-		currentFactors |= kIOPMSleepFactorThunderboltDevice;
-	}
-	if (_scheduledAlarmMask != 0) {
-		currentFactors |= kIOPMSleepFactorRTCAlarmScheduled;
-	}
-	if (getPMAssertionLevel(kIOPMDriverAssertionMagicPacketWakeEnabledBit) !=
-	    kIOPMDriverAssertionLevelOff) {
-		currentFactors |= kIOPMSleepFactorMagicPacketWakeEnabled;
-	}
-#define TCPKEEPALIVE 1
-#if TCPKEEPALIVE
-	if (getPMAssertionLevel(kIOPMDriverAssertionNetworkKeepAliveActiveBit) !=
-	    kIOPMDriverAssertionLevelOff) {
-		currentFactors |= kIOPMSleepFactorNetworkKeepAliveActive;
-	}
-#endif
-	if (!powerOffEnabled) {
-		currentFactors |= kIOPMSleepFactorAutoPowerOffDisabled;
-	}
-	if (desktopMode) {
-		currentFactors |= kIOPMSleepFactorExternalDisplay;
-	}
-	if (userWasActive) {
-		currentFactors |= kIOPMSleepFactorLocalUserActivity;
-	}
-	if (darkWakeHibernateError && !CAP_HIGHEST(kIOPMSystemCapabilityGraphics)) {
-		currentFactors |= kIOPMSleepFactorHibernateFailed;
-	}
-	if (thermalWarningState) {
-		currentFactors |= kIOPMSleepFactorThermalWarning;
-	}
-
-	for (int factorBit = 0; factorBit < (8 * sizeof(uint32_t)); factorBit++) {
-		uint32_t factor = 1 << factorBit;
-		if (factor & currentFactors) {
-			strlcat(currentFactorsBuf, ", ", sizeof(currentFactorsBuf));
-			strlcat(currentFactorsBuf, IOFindNameForValue(factor, factorValues), sizeof(currentFactorsBuf));
-		}
-	}
-	DLOG("sleep factors 0x%llx%s\n", currentFactors, currentFactorsBuf);
-
-	if (gSleepPolicyHandler) {
-		uint32_t    savedHibernateMode;
-		IOReturn    result;
-
-		if (!gSleepPolicyVars) {
-			gSleepPolicyVars = IOMallocType(IOPMSystemSleepPolicyVariables);
-		}
-		gSleepPolicyVars->signature = kIOPMSystemSleepPolicySignature;
-		gSleepPolicyVars->version   = kIOPMSystemSleepPolicyVersion;
-		gSleepPolicyVars->currentCapability = _currentCapability;
-		gSleepPolicyVars->highestCapability = _highestCapability;
-		gSleepPolicyVars->sleepFactors      = currentFactors;
-		gSleepPolicyVars->sleepReason       = lastSleepReason;
-		gSleepPolicyVars->sleepPhase        = sleepPhase;
-		gSleepPolicyVars->standbyDelay      = standbyDelay;
-		gSleepPolicyVars->standbyTimer      = standbyTimer;
-		gSleepPolicyVars->poweroffDelay     = powerOffDelay;
-		gSleepPolicyVars->scheduledAlarms   = _scheduledAlarmMask | _userScheduledAlarmMask;
-		gSleepPolicyVars->poweroffTimer     = powerOffTimer;
-
-		if (kIOPMSleepPhase0 == sleepPhase) {
-			// preserve hibernateMode
-			savedHibernateMode = gSleepPolicyVars->hibernateMode;
-			gSleepPolicyVars->hibernateMode = *hibMode;
-		} else if (kIOPMSleepPhase1 == sleepPhase) {
-			// use original hibernateMode for phase2
-			gSleepPolicyVars->hibernateMode = *hibMode;
-		}
-
-		result = gSleepPolicyHandler(gSleepPolicyTarget, gSleepPolicyVars, params);
-
-		if (kIOPMSleepPhase0 == sleepPhase) {
-			// restore hibernateMode
-			gSleepPolicyVars->hibernateMode = savedHibernateMode;
-		}
-
-		if ((result != kIOReturnSuccess) ||
-		    (kIOPMSleepTypeInvalid == params->sleepType) ||
-		    (params->sleepType >= kIOPMSleepTypeLast) ||
-		    (kIOPMSystemSleepParametersVersion != params->version)) {
-			MSG("sleep policy handler error\n");
-			goto done;
-		}
-
-		if ((getSleepTypeAttributes(params->sleepType) &
-		    kIOPMSleepAttributeHibernateSetup) &&
-		    ((*hibMode & kIOHibernateModeOn) == 0)) {
-			*hibMode |= (kIOHibernateModeOn | kIOHibernateModeSleep);
-		}
-
-		DLOG("sleep params v%u, type %u, flags 0x%x, wake 0x%x, timer %u, poweroff %u\n",
-		    params->version, params->sleepType, params->sleepFlags,
-		    params->ecWakeEvents, params->ecWakeTimer, params->ecPoweroffTimer);
-		found = true;
-		goto done;
-	}
-
-	// Policy table is meaningless without standby enabled
-	if (!standbyEnabled) {
-		goto done;
-	}
-
-	// Validate the sleep policy table
-	policyData = OSDynamicCast(OSData, prop.get());
-	if (!policyData || (policyData->getLength() <= sizeof(IOPMSystemSleepPolicyTable))) {
-		goto done;
-	}
-
-	pt = (const IOPMSystemSleepPolicyTable *) policyData->getBytesNoCopy();
-	if ((pt->signature != kIOPMSystemSleepPolicySignature) ||
-	    (pt->version != 1) || (0 == pt->entryCount)) {
-		goto done;
-	}
-
-	if (((policyData->getLength() - sizeof(IOPMSystemSleepPolicyTable)) !=
-	    (sizeof(IOPMSystemSleepPolicyEntry) * pt->entryCount))) {
-		goto done;
-	}
-
-	for (uint32_t i = 0; i < pt->entryCount; i++) {
-		const IOPMSystemSleepPolicyEntry * entry = &pt->entries[i];
-		mismatch = (((uint32_t)currentFactors ^ entry->factorBits) & entry->factorMask);
-
-		DLOG("mask 0x%08x, bits 0x%08x, flags 0x%08x, wake 0x%08x, mismatch 0x%08x\n",
-		    entry->factorMask, entry->factorBits,
-		    entry->sleepFlags, entry->wakeEvents, mismatch);
-		if (mismatch) {
-			continue;
-		}
-
-		DLOG("^ found match\n");
-		found = true;
-
-		params->version = kIOPMSystemSleepParametersVersion;
-		params->reserved1 = 1;
-		if (entry->sleepFlags & kIOPMSleepFlagHibernate) {
-			params->sleepType = kIOPMSleepTypeStandby;
-		} else {
-			params->sleepType = kIOPMSleepTypeNormalSleep;
-		}
-
-		params->ecWakeEvents = entry->wakeEvents;
-		if (entry->sleepFlags & kIOPMSleepFlagSleepTimerEnable) {
-			if (kIOPMSleepPhase2 == sleepPhase) {
-				clock_sec_t now_secs = gIOLastSleepTime.tv_sec;
-
-				if (!_standbyTimerResetSeconds ||
-				    (now_secs <= _standbyTimerResetSeconds)) {
-					// Reset standby timer adjustment
-					_standbyTimerResetSeconds = now_secs;
-					DLOG("standby delay %u, reset %u\n",
-					    standbyDelay, (uint32_t) _standbyTimerResetSeconds);
-				} else if (standbyDelay) {
-					// Shorten the standby delay timer
-					clock_sec_t elapsed = now_secs - _standbyTimerResetSeconds;
-					if (standbyDelay > elapsed) {
-						standbyDelay -= elapsed;
-					} else {
-						standbyDelay = 1; // must be > 0
-					}
-					DLOG("standby delay %u, elapsed %u\n",
-					    standbyDelay, (uint32_t) elapsed);
-				}
-			}
-			params->ecWakeTimer = standbyDelay;
-		} else if (kIOPMSleepPhase2 == sleepPhase) {
-			// A sleep that does not enable the sleep timer will reset
-			// the standby delay adjustment.
-			_standbyTimerResetSeconds = 0;
-		}
-		break;
-	}
-
-done:
-	return found;
-}
-
-static IOPMSystemSleepParameters gEarlySystemSleepParams;
-
-void
-IOPMrootDomain::evaluateSystemSleepPolicyEarly( void )
-{
-	// Evaluate early (priority interest phase), before drivers sleep.
-
-	DLOG("%s\n", __FUNCTION__);
-	removeProperty(kIOPMSystemSleepParametersKey);
-
-	// Full wake resets the standby timer delay adjustment
-	if (_highestCapability & kIOPMSystemCapabilityGraphics) {
-		_standbyTimerResetSeconds = 0;
-	}
-
-	hibernateDisabled = false;
-	hibernateMode = 0;
-	getSleepOption(kIOHibernateModeKey, &hibernateMode);
-
-	// Save for late evaluation if sleep is aborted
-	bzero(&gEarlySystemSleepParams, sizeof(gEarlySystemSleepParams));
-
-	if (evaluateSystemSleepPolicy(&gEarlySystemSleepParams, kIOPMSleepPhase1,
-	    &hibernateMode)) {
-		if (!hibernateRetry &&
-		    ((getSleepTypeAttributes(gEarlySystemSleepParams.sleepType) &
-		    kIOPMSleepAttributeHibernateSetup) == 0)) {
-			// skip hibernate setup
-			hibernateDisabled = true;
-		}
-	}
-
-	// Publish IOPMSystemSleepType
-	uint32_t sleepType = gEarlySystemSleepParams.sleepType;
-	if (sleepType == kIOPMSleepTypeInvalid) {
-		// no sleep policy
-		sleepType = kIOPMSleepTypeNormalSleep;
-		if (hibernateMode & kIOHibernateModeOn) {
-			sleepType = (hibernateMode & kIOHibernateModeSleep) ?
-			    kIOPMSleepTypeSafeSleep : kIOPMSleepTypeHibernate;
-		}
-	} else if ((sleepType == kIOPMSleepTypeStandby) &&
-	    (gEarlySystemSleepParams.ecPoweroffTimer)) {
-		// report the lowest possible sleep state
-		sleepType = kIOPMSleepTypePowerOff;
-	}
-
-	setProperty(kIOPMSystemSleepTypeKey, sleepType, 32);
-}
-
-void
-IOPMrootDomain::evaluateSystemSleepPolicyFinal( void )
-{
-	IOPMSystemSleepParameters   params;
-	OSSharedPtr<OSData>         paramsData;
-	bool                        wakeNow;
-	// Evaluate sleep policy after sleeping drivers but before platform sleep.
-
-	DLOG("%s\n", __FUNCTION__);
-
-	bzero(&params, sizeof(params));
-	wakeNow = false;
-	if (evaluateSystemSleepPolicy(&params, kIOPMSleepPhase2, &hibernateMode)) {
-		if ((kIOPMSleepTypeStandby == params.sleepType)
-		    && gIOHibernateStandbyDisabled && gSleepPolicyVars
-		    && (!((kIOPMSleepFactorStandbyForced | kIOPMSleepFactorAutoPowerOffForced | kIOPMSleepFactorHibernateForced)
-		    & gSleepPolicyVars->sleepFactors))) {
-			standbyNixed = true;
-			wakeNow = true;
-		}
-		if (wakeNow
-		    || ((hibernateDisabled || hibernateAborted) &&
-		    (getSleepTypeAttributes(params.sleepType) &
-		    kIOPMSleepAttributeHibernateSetup))) {
-			// Final evaluation picked a state requiring hibernation,
-			// but hibernate isn't going to proceed. Arm a short sleep using
-			// the early non-hibernate sleep parameters.
-			bcopy(&gEarlySystemSleepParams, &params, sizeof(params));
-			params.sleepType = kIOPMSleepTypeAbortedSleep;
-			params.ecWakeTimer = 1;
-			if (standbyNixed) {
-				resetTimers = true;
-			} else {
-				// Set hibernateRetry flag to force hibernate setup on the
-				// next sleep.
-				hibernateRetry = true;
-			}
-			DLOG("wake in %u secs for hibernateDisabled %d, hibernateAborted %d, standbyNixed %d\n",
-			    params.ecWakeTimer, hibernateDisabled, hibernateAborted, standbyNixed);
-		} else {
-			hibernateRetry = false;
-		}
-
-		if (kIOPMSleepTypeAbortedSleep != params.sleepType) {
-			resetTimers = false;
-		}
-
-		paramsData = OSData::withValue(params);
-		if (paramsData) {
-			setProperty(kIOPMSystemSleepParametersKey, paramsData.get());
-		}
-
-		if (getSleepTypeAttributes(params.sleepType) &
-		    kIOPMSleepAttributeHibernateSleep) {
-			// Disable sleep to force hibernation
-			gIOHibernateMode &= ~kIOHibernateModeSleep;
-		}
-	}
-}
-
-bool
-IOPMrootDomain::getHibernateSettings(
-	uint32_t *  hibernateModePtr,
-	uint32_t *  hibernateFreeRatio,
-	uint32_t *  hibernateFreeTime )
-{
-	// Called by IOHibernateSystemSleep() after evaluateSystemSleepPolicyEarly()
-	// has updated the hibernateDisabled flag.
-
-	bool ok = getSleepOption(kIOHibernateModeKey, hibernateModePtr);
-	getSleepOption(kIOHibernateFreeRatioKey, hibernateFreeRatio);
-	getSleepOption(kIOHibernateFreeTimeKey, hibernateFreeTime);
-	if (hibernateDisabled) {
-		*hibernateModePtr = 0;
-	} else if (gSleepPolicyHandler) {
-		*hibernateModePtr = hibernateMode;
-	}
-	DLOG("hibernateMode 0x%x\n", *hibernateModePtr);
-	return ok;
-}
-
-bool
-IOPMrootDomain::getSleepOption( const char * key, uint32_t * option )
-{
-	OSSharedPtr<OSObject>       optionsProp;
-	OSDictionary *              optionsDict;
-	OSSharedPtr<OSObject>       obj;
-	OSNumber *                  num;
-	bool                        ok = false;
-
-	optionsProp = copyProperty(kRootDomainSleepOptionsKey);
-	optionsDict = OSDynamicCast(OSDictionary, optionsProp.get());
-
-	if (optionsDict) {
-		obj.reset(optionsDict->getObject(key), OSRetain);
-	}
-	if (!obj) {
-		obj = copyProperty(key);
-	}
-	if (obj) {
-		if ((num = OSDynamicCast(OSNumber, obj.get()))) {
-			*option = num->unsigned32BitValue();
-			ok = true;
-		} else if (OSDynamicCast(OSBoolean, obj.get())) {
-			*option = (obj == kOSBooleanTrue) ? 1 : 0;
-			ok = true;
-		}
-	}
-
-	return ok;
-}
-#endif /* HIBERNATION */
-
-IOReturn
-IOPMrootDomain::getSystemSleepType( uint32_t * sleepType, uint32_t * standbyTimer )
-{
-#if HIBERNATION
-	IOPMSystemSleepParameters   params;
-	uint32_t                    hibMode = 0;
-	bool                        ok;
-
-	if (gIOPMWorkLoop->inGate() == false) {
-		IOReturn ret = gIOPMWorkLoop->runAction(
-			OSMemberFunctionCast(IOWorkLoop::Action, this,
-			&IOPMrootDomain::getSystemSleepType),
-			(OSObject *) this,
-			(void *) sleepType, (void *) standbyTimer);
-		return ret;
-	}
-
-	getSleepOption(kIOHibernateModeKey, &hibMode);
-	bzero(&params, sizeof(params));
-
-	ok = evaluateSystemSleepPolicy(&params, kIOPMSleepPhase0, &hibMode);
-	if (ok) {
-		*sleepType = params.sleepType;
-		if (!getSleepOption(kIOPMDeepSleepTimerKey, standbyTimer) &&
-		    !getSleepOption(kIOPMDeepSleepDelayKey, standbyTimer)) {
-			DLOG("Standby delay is not set\n");
-			*standbyTimer = 0;
-		}
-		return kIOReturnSuccess;
-	}
-#endif
-
-	return kIOReturnUnsupported;
-}
-
-// MARK: -
-// MARK: Shutdown and Restart
-
-//******************************************************************************
-// handlePlatformHaltRestart
-//
-//******************************************************************************
-
-// Phases while performing shutdown/restart
-typedef enum {
-	kNotifyDone                 = 0x00,
-	kNotifyPriorityClients      = 0x10,
-	kNotifyPowerPlaneDrivers    = 0x20,
-	kNotifyHaltRestartAction    = 0x30,
-	kQuiescePM                  = 0x40,
-} shutdownPhase_t;
-
-
-struct HaltRestartApplierContext {
-	IOPMrootDomain *    RootDomain;
-	unsigned long       PowerState;
-	IOPMPowerFlags      PowerFlags;
-	UInt32              MessageType;
-	UInt32              Counter;
-	const char *        LogString;
-	shutdownPhase_t     phase;
-
-	IOServiceInterestHandler    handler;
-} gHaltRestartCtx;
-
-const char *
-shutdownPhase2String(shutdownPhase_t phase)
-{
-	switch (phase) {
-	case kNotifyDone:
-		return "Notifications completed";
-	case kNotifyPriorityClients:
-		return "Notifying priority clients";
-	case kNotifyPowerPlaneDrivers:
-		return "Notifying power plane drivers";
-	case kNotifyHaltRestartAction:
-		return "Notifying HaltRestart action handlers";
-	case kQuiescePM:
-		return "Quiescing PM";
-	default:
-		return "Unknown";
-	}
-}
-
-static void
-platformHaltRestartApplier( OSObject * object, void * context )
-{
-	IOPowerStateChangeNotification  notify;
-	HaltRestartApplierContext *     ctx;
-	AbsoluteTime                    startTime, elapsedTime;
-	uint32_t                        deltaTime;
-
-	ctx = (HaltRestartApplierContext *) context;
-
-	_IOServiceInterestNotifier * notifier;
-	notifier = OSDynamicCast(_IOServiceInterestNotifier, object);
-	memset(&notify, 0, sizeof(notify));
-	notify.powerRef    = (void *)(uintptr_t)ctx->Counter;
-	notify.returnValue = 0;
-	notify.stateNumber = ctx->PowerState;
-	notify.stateFlags  = ctx->PowerFlags;
-
-	if (notifier) {
-		ctx->handler = notifier->handler;
-	}
-
-	clock_get_uptime(&startTime);
-	ctx->RootDomain->messageClient( ctx->MessageType, object, (void *)&notify );
-	deltaTime = computeDeltaTimeMS(&startTime, &elapsedTime);
-
-	if ((deltaTime > kPMHaltTimeoutMS) && notifier) {
-		LOG("%s handler %p took %u ms\n",
-		    ctx->LogString, OBFUSCATE(notifier->handler), deltaTime);
-		halt_log_enter("PowerOff/Restart message to priority client", (const void *) notifier->handler, elapsedTime);
-	}
-
-	ctx->handler = NULL;
-	ctx->Counter++;
-}
-
-static void
-quiescePowerTreeCallback( void * target, void * param )
-{
-	IOLockLock(gPMHaltLock);
-	gPMQuiesced = true;
-	thread_wakeup(param);
-	IOLockUnlock(gPMHaltLock);
-}
-
-void
-IOPMrootDomain::handlePlatformHaltRestart( UInt32 pe_type )
-{
-	AbsoluteTime                startTime, elapsedTime;
-	uint32_t                    deltaTime;
-	bool                        nvramSync = false;
-
-	memset(&gHaltRestartCtx, 0, sizeof(gHaltRestartCtx));
-	gHaltRestartCtx.RootDomain = this;
-
-	clock_get_uptime(&startTime);
-	switch (pe_type) {
-	case kPEHaltCPU:
-	case kPEUPSDelayHaltCPU:
-		gHaltRestartCtx.PowerState  = OFF_STATE;
-		gHaltRestartCtx.MessageType = kIOMessageSystemWillPowerOff;
-		gHaltRestartCtx.LogString   = "PowerOff";
-		nvramSync = true;
-		break;
-
-	case kPERestartCPU:
-		gHaltRestartCtx.PowerState  = RESTART_STATE;
-		gHaltRestartCtx.MessageType = kIOMessageSystemWillRestart;
-		gHaltRestartCtx.LogString   = "Restart";
-		nvramSync = true;
-		break;
-
-	case kPEPagingOff:
-		gHaltRestartCtx.PowerState  = ON_STATE;
-		gHaltRestartCtx.MessageType = kIOMessageSystemPagingOff;
-		gHaltRestartCtx.LogString   = "PagingOff";
-		IOService::updateConsoleUsers(NULL, kIOMessageSystemPagingOff);
-#if HIBERNATION
-		IOHibernateSystemRestart();
-#endif
-		break;
-
-	default:
-		return;
-	}
-
-	if (nvramSync) {
-		PESyncNVRAM();
-	}
-
-	gHaltRestartCtx.phase = kNotifyPriorityClients;
-	// Notify legacy clients
-	applyToInterested(gIOPriorityPowerStateInterest, platformHaltRestartApplier, &gHaltRestartCtx);
-
-	// For normal shutdown, turn off File Server Mode.
-	if (kPEHaltCPU == pe_type) {
-		OSSharedPtr<const OSSymbol> setting = OSSymbol::withCString(kIOPMSettingRestartOnPowerLossKey);
-		OSSharedPtr<OSNumber> num = OSNumber::withNumber((unsigned long long) 0, 32);
-		if (setting && num) {
-			setPMSetting(setting.get(), num.get());
-		}
-	}
-
-	if (kPEPagingOff != pe_type) {
-		gHaltRestartCtx.phase = kNotifyPowerPlaneDrivers;
-		// Notify in power tree order
-		notifySystemShutdown(this, gHaltRestartCtx.MessageType);
-	}
-
-	gHaltRestartCtx.phase = kNotifyHaltRestartAction;
-#if defined(XNU_TARGET_OS_OSX)
-	IOCPURunPlatformHaltRestartActions(pe_type);
-#else /* !defined(XNU_TARGET_OS_OSX) */
-	if (kPEPagingOff != pe_type) {
-		IOCPURunPlatformHaltRestartActions(pe_type);
-	}
-#endif /* !defined(XNU_TARGET_OS_OSX) */
-
-	// Wait for PM to quiesce
-	if ((kPEPagingOff != pe_type) && gPMHaltLock) {
-		gHaltRestartCtx.phase = kQuiescePM;
-		AbsoluteTime quiesceTime = mach_absolute_time();
-
-		IOLockLock(gPMHaltLock);
-		gPMQuiesced = false;
-		if (quiescePowerTree(this, &quiescePowerTreeCallback, &gPMQuiesced) ==
-		    kIOReturnSuccess) {
-			while (!gPMQuiesced) {
-				IOLockSleep(gPMHaltLock, &gPMQuiesced, THREAD_UNINT);
-			}
-		}
-		IOLockUnlock(gPMHaltLock);
-		deltaTime = computeDeltaTimeMS(&quiesceTime, &elapsedTime);
-		DLOG("PM quiesce took %u ms\n", deltaTime);
-		halt_log_enter("Quiesce", NULL, elapsedTime);
-	}
-	gHaltRestartCtx.phase = kNotifyDone;
-
-	deltaTime = computeDeltaTimeMS(&startTime, &elapsedTime);
-	LOG("%s all drivers took %u ms\n", gHaltRestartCtx.LogString, deltaTime);
-
-	halt_log_enter(gHaltRestartCtx.LogString, NULL, elapsedTime);
-
-	deltaTime = computeDeltaTimeMS(&gHaltStartTime, &elapsedTime);
-	LOG("%s total %u ms\n", gHaltRestartCtx.LogString, deltaTime);
-
-	if (gHaltLog && gHaltTimeMaxLog && (deltaTime >= gHaltTimeMaxLog)) {
-		printf("%s total %d ms:%s\n", gHaltRestartCtx.LogString, deltaTime, gHaltLog);
-	}
-
-	checkShutdownTimeout();
-}
-
-bool
-IOPMrootDomain::checkShutdownTimeout()
-{
-	AbsoluteTime   elapsedTime;
-	uint32_t deltaTime = computeDeltaTimeMS(&gHaltStartTime, &elapsedTime);
-
-	if (gHaltTimeMaxPanic && (deltaTime >= gHaltTimeMaxPanic)) {
-		return true;
-	}
-	return false;
-}
-
-void
-IOPMrootDomain::panicWithShutdownLog(uint32_t timeoutInMs)
-{
-	if (gHaltLog) {
-		if ((gHaltRestartCtx.phase == kNotifyPriorityClients) && gHaltRestartCtx.handler) {
-			halt_log_enter("Blocked on priority client", (void *)gHaltRestartCtx.handler, mach_absolute_time() - gHaltStartTime);
-		}
-		panic("%s timed out in phase '%s'. Total %d ms:%s",
-		    gHaltRestartCtx.LogString, shutdownPhase2String(gHaltRestartCtx.phase), timeoutInMs, gHaltLog);
-	} else {
-		panic("%s timed out in phase \'%s\'. Total %d ms",
-		    gHaltRestartCtx.LogString, shutdownPhase2String(gHaltRestartCtx.phase), timeoutInMs);
-	}
+	deltaTime = computeDeltaTimeMS(&startTime);
+	LOG("%s all drivers took %u ms\n",
+		(ctx.MessageType == kIOMessageSystemWillPowerOff) ? "PowerOff" :
+			 (ctx.MessageType == kIOMessageSystemPagingOff) ? "PagingOff" : "Restart",
+		(uint32_t) deltaTime );
 }
 
 //******************************************************************************
@@ -6203,10 +3993,9 @@
 //
 //******************************************************************************
 
-IOReturn
-IOPMrootDomain::shutdownSystem( void )
-{
-	return kIOReturnUnsupported;
+IOReturn IOPMrootDomain::shutdownSystem( void )
+{
+    return kIOReturnUnsupported;
 }
 
 //******************************************************************************
@@ -6214,10 +4003,9 @@
 //
 //******************************************************************************
 
-IOReturn
-IOPMrootDomain::restartSystem( void )
-{
-	return kIOReturnUnsupported;
+IOReturn IOPMrootDomain::restartSystem( void )
+{
+    return kIOReturnUnsupported;
 }
 
 // MARK: -
@@ -6229,1187 +4017,804 @@
 // Running on PM work loop thread.
 //******************************************************************************
 
-void
-IOPMrootDomain::tagPowerPlaneService(
-	IOService *         service,
-	IOPMActions *       actions,
-	IOPMPowerStateIndex maxPowerState )
-{
-	uint32_t    flags = 0;
-
-	memset(actions, 0, sizeof(*actions));
-	actions->target = this;
-
-	if (service == this) {
-		actions->actionPowerChangeStart =
-		    OSMemberFunctionCast(
-			IOPMActionPowerChangeStart, this,
-			&IOPMrootDomain::handleOurPowerChangeStart);
-
-		actions->actionPowerChangeDone =
-		    OSMemberFunctionCast(
-			IOPMActionPowerChangeDone, this,
-			&IOPMrootDomain::handleOurPowerChangeDone);
-
-		actions->actionPowerChangeOverride =
-		    OSMemberFunctionCast(
-			IOPMActionPowerChangeOverride, this,
-			&IOPMrootDomain::overrideOurPowerChange);
-		return;
-	}
-
-#if DISPLAY_WRANGLER_PRESENT
-	if (NULL != service->metaCast("IODisplayWrangler")) {
-		// XXX should this really retain?
-		wrangler.reset(service, OSRetain);
-		wrangler->registerInterest(gIOGeneralInterest,
-		    &displayWranglerNotification, this, NULL);
-
-		// found the display wrangler, check for any display assertions already created
-		if (pmAssertions->getActivatedAssertions() & kIOPMDriverAssertionPreventDisplaySleepBit) {
-			DLOG("wrangler setIgnoreIdleTimer\(1) due to pre-existing assertion\n");
-			wrangler->setIgnoreIdleTimer( true );
-		}
-		flags |= kPMActionsFlagIsDisplayWrangler;
-	}
-#endif /* DISPLAY_WRANGLER_PRESENT */
-
-	if (service->propertyExists("IOPMStrictTreeOrder")) {
-		flags |= kPMActionsFlagIsGraphicsDriver;
-	}
-	if (service->propertyExists("IOPMUnattendedWakePowerState")) {
-		flags |= kPMActionsFlagIsAudioDriver;
-	}
-
-	// Find the power connection object that is a child of the PCI host
-	// bridge, and has a graphics/audio device attached below. Mark the
-	// power branch for delayed child notifications.
-
-	if (flags) {
-		IORegistryEntry * child  = service;
-		IORegistryEntry * parent = child->getParentEntry(gIOPowerPlane);
-
-		while (child != this) {
-			if (child->propertyHasValue("IOPCITunnelled", kOSBooleanTrue)) {
-				// Skip delaying notifications and clamping power on external graphics and audio devices.
-				DLOG("Avoiding delayChildNotification on object 0x%llx. flags: 0x%x\n", service->getRegistryEntryID(), flags);
-				flags = 0;
-				break;
-			}
-			if ((parent == pciHostBridgeDriver) ||
-			    (parent == this)) {
-				if (OSDynamicCast(IOPowerConnection, child)) {
-					IOPowerConnection * conn = (IOPowerConnection *) child;
-					conn->delayChildNotification = true;
-					DLOG("delayChildNotification for 0x%llx\n", conn->getRegistryEntryID());
-				}
-				break;
-			}
-			child = parent;
-			parent = child->getParentEntry(gIOPowerPlane);
-		}
-	}
-
-	OSSharedPtr<OSObject> prop = service->copyProperty(kIOPMDarkWakeMaxPowerStateKey);
-	if (prop) {
-		OSNumber * num = OSDynamicCast(OSNumber, prop.get());
-		if (num) {
-			actions->darkWakePowerState = num->unsigned32BitValue();
-			if (actions->darkWakePowerState < maxPowerState) {
-				flags |= kPMActionsFlagHasDarkWakePowerState;
-			}
-		}
-	}
-
-
-	if (flags) {
-		DLOG("%s tag flags %x\n", service->getName(), flags);
-		actions->flags |= flags;
-		actions->actionPowerChangeOverride =
-		    OSMemberFunctionCast(
-			IOPMActionPowerChangeOverride, this,
-			&IOPMrootDomain::overridePowerChangeForService);
-
-		if (flags & kPMActionsFlagIsDisplayWrangler) {
-			actions->actionActivityTickle =
-			    OSMemberFunctionCast(
-				IOPMActionActivityTickle, this,
-				&IOPMrootDomain::handleActivityTickleForDisplayWrangler);
-
-			actions->actionUpdatePowerClient =
-			    OSMemberFunctionCast(
-				IOPMActionUpdatePowerClient, this,
-				&IOPMrootDomain::handleUpdatePowerClientForDisplayWrangler);
-		}
-		return;
-	}
-
-	// Locate the first PCI host bridge for PMTrace.
-	if (!pciHostBridgeDevice && service->metaCast("IOPCIBridge")) {
-		IOService * provider = service->getProvider();
-		if (OSDynamicCast(IOPlatformDevice, provider) &&
-		    provider->inPlane(gIODTPlane)) {
-			pciHostBridgeDevice.reset(provider, OSNoRetain);
-			pciHostBridgeDriver.reset(service, OSNoRetain);
-			DLOG("PMTrace found PCI host bridge %s->%s\n",
-			    provider->getName(), service->getName());
-		}
-	}
-
-	// Tag top-level PCI devices. The order of PMinit() call does not
+void IOPMrootDomain::tagPowerPlaneService(
+        IOService *     service,
+        IOPMActions *   actions )
+{
+    uint32_t    flags = 0;
+    bool        isDisplayWrangler;
+
+    memset(actions, 0, sizeof(*actions));
+    actions->target = this;
+
+    if (service == this)
+    {
+        actions->actionPowerChangeStart =
+            OSMemberFunctionCast(
+                IOPMActionPowerChangeStart, this,
+                &IOPMrootDomain::handleOurPowerChangeStart);
+
+        actions->actionPowerChangeDone =
+            OSMemberFunctionCast(
+                IOPMActionPowerChangeDone, this,
+                &IOPMrootDomain::handleOurPowerChangeDone);
+
+        actions->actionPowerChangeOverride =
+            OSMemberFunctionCast(
+                IOPMActionPowerChangeOverride, this,
+                &IOPMrootDomain::overrideOurPowerChange);
+        return;
+    }
+
+#if !NO_KERNEL_HID
+    isDisplayWrangler = (0 != service->metaCast("IODisplayWrangler"));
+    if (isDisplayWrangler)
+    {
+        wrangler = service;
+    }
+#else
+    isDisplayWrangler = false;
+#endif
+
+#if defined(__i386__) || defined(__x86_64__)
+    if (isDisplayWrangler)
+        flags |= kPMActionsFlagIsDisplayWrangler;
+    if (service->getProperty("IOPMStrictTreeOrder"))
+        flags |= kPMActionsFlagIsGraphicsDevice;
+    if (service->getProperty("IOPMUnattendedWakePowerState"))
+        flags |= kPMActionsFlagIsAudioDevice;
+#endif
+
+    // Find the power connection object that is a child of the PCI host
+    // bridge, and has a graphics/audio device attached below. Mark the
+    // power branch for delayed child notifications.
+
+    if (flags)
+    {
+        IORegistryEntry * child  = service;
+        IORegistryEntry * parent = child->getParentEntry(gIOPowerPlane);
+
+        while (child != this)
+        {
+            if ((parent == pciHostBridgeDriver) ||
+                (parent == this))
+            {
+                if (OSDynamicCast(IOPowerConnection, child))
+                {
+                    IOPowerConnection * conn = (IOPowerConnection *) child;
+                    conn->delayChildNotification = true;
+                }
+                break;
+            }
+            child = parent;
+            parent = child->getParentEntry(gIOPowerPlane);
+        }
+    }
+
+    if (flags)
+    {
+        DLOG("%s tag flags %x\n", service->getName(), flags);
+        actions->parameter |= flags;
+        actions->actionPowerChangeOverride =
+            OSMemberFunctionCast(
+                IOPMActionPowerChangeOverride, this,
+                &IOPMrootDomain::overridePowerChangeForUIService);
+
+        if (flags & kPMActionsFlagIsDisplayWrangler)
+        {
+            actions->actionActivityTickle =
+                OSMemberFunctionCast(
+                    IOPMActionActivityTickle, this,
+                    &IOPMrootDomain::handleActivityTickleForDisplayWrangler);
+        }
+        return;
+    }
+
+    // Locate the first PCI host bridge for PMTrace.
+    if (!pciHostBridgeDevice && service->metaCast("IOPCIBridge"))
+    {
+        IOService * provider = service->getProvider();
+        if (OSDynamicCast(IOPlatformDevice, provider) &&
+            provider->inPlane(gIODTPlane))
+        {
+            pciHostBridgeDevice = provider;
+            pciHostBridgeDriver = service;
+            DLOG("PMTrace found PCI host bridge %s->%s\n",
+                provider->getName(), service->getName());
+        }
+    }
+
+    // Tag top-level PCI devices. The order of PMinit() call does not
 	// change across boots and is used as the PCI bit number.
-	if (pciHostBridgeDevice && service->metaCast("IOPCIDevice")) {
-		// Would prefer to check built-in property, but tagPowerPlaneService()
-		// is called before pciDevice->registerService().
-		IORegistryEntry * parent = service->getParentEntry(gIODTPlane);
-		if ((parent == pciHostBridgeDevice) && service->propertyExists("acpi-device")) {
-			int bit = pmTracer->recordTopLevelPCIDevice( service );
-			if (bit >= 0) {
+    if (pciHostBridgeDevice && service->metaCast("IOPCIDevice"))
+    {
+        // Would prefer to check built-in property, but tagPowerPlaneService()
+        // is called before pciDevice->registerService().
+        IORegistryEntry * parent = service->getParentEntry(gIODTPlane);
+        if ((parent == pciHostBridgeDevice) && service->getProperty("acpi-device"))
+        {
+            int bit = pmTracer->recordTopLevelPCIDevice( service );
+            if (bit >= 0)
+            {
 				// Save the assigned bit for fast lookup.
-				actions->flags |= (bit & kPMActionsPCIBitNumberMask);
-
-				actions->actionPowerChangeStart =
-				    OSMemberFunctionCast(
-					IOPMActionPowerChangeStart, this,
-					&IOPMrootDomain::handlePowerChangeStartForPCIDevice);
-
-				actions->actionPowerChangeDone =
-				    OSMemberFunctionCast(
-					IOPMActionPowerChangeDone, this,
-					&IOPMrootDomain::handlePowerChangeDoneForPCIDevice);
-			}
-		}
-	}
+                actions->parameter |= (bit & kPMActionsPCIBitNumberMask);
+
+                actions->actionPowerChangeStart =
+                    OSMemberFunctionCast(
+                        IOPMActionPowerChangeStart, this,
+                        &IOPMrootDomain::handlePowerChangeStartForPCIDevice);
+
+                actions->actionPowerChangeDone =
+                    OSMemberFunctionCast(
+                        IOPMActionPowerChangeDone, this,
+                        &IOPMrootDomain::handlePowerChangeDoneForPCIDevice);
+            }
+        }
+    }
 }
 
 //******************************************************************************
 // PM actions for root domain
 //******************************************************************************
 
-void
-IOPMrootDomain::overrideOurPowerChange(
-	IOService *             service,
-	IOPMActions *           actions,
-	const IOPMRequest *     request,
-	IOPMPowerStateIndex *   inOutPowerState,
-	IOPMPowerChangeFlags *  inOutChangeFlags )
-{
-	uint32_t changeFlags = *inOutChangeFlags;
-	uint32_t desiredPowerState = (uint32_t) *inOutPowerState;
-	uint32_t currentPowerState = (uint32_t) getPowerState();
-
-	if (request->getTag() == 0) {
-		// Set a tag for any request that originates from IOServicePM
-		(const_cast<IOPMRequest *>(request))->fTag = nextRequestTag(kCPSReasonPMInternals);
-	}
-
-	DLOG("PowerChangeOverride (%s->%s, %x, 0x%x) tag 0x%x\n",
-	    getPowerStateString(currentPowerState),
-	    getPowerStateString(desiredPowerState),
-	    _currentCapability, changeFlags,
-	    request->getTag());
-
-
-#if defined(XNU_TARGET_OS_OSX) && !DISPLAY_WRANGLER_PRESENT
-	/*
-	 * ASBM send lowBattery notifications every 1 second until the device
-	 * enters hibernation. This queues up multiple sleep requests.
-	 * After the device wakes from hibernation, none of these previously
-	 * queued sleep requests are valid.
-	 * lowBattteryCondition variable is set when ASBM notifies rootDomain
-	 * and is cleared at the very last point in sleep.
-	 * Any attempt to sleep with reason kIOPMSleepReasonLowPower without
-	 * lowBatteryCondition is invalid
-	 */
-	if (REQUEST_TAG_TO_REASON(request->getTag()) == kIOPMSleepReasonLowPower) {
-		if (!lowBatteryCondition) {
-			DLOG("Duplicate lowBattery sleep");
-			*inOutChangeFlags |= kIOPMNotDone;
-			return;
-		}
-	}
+void IOPMrootDomain::overrideOurPowerChange(
+    IOService *     service,
+    IOPMActions *   actions,
+    unsigned long * inOutPowerState,
+    uint32_t *      inOutChangeFlags )
+{
+    uint32_t    powerState  = (uint32_t) *inOutPowerState;
+    uint32_t    changeFlags = *inOutChangeFlags;
+    uint32_t    currentPowerState = (uint32_t) getPowerState();
+
+    if ((currentPowerState == powerState) ||
+        (changeFlags & kIOPMParentInitiated))
+    {
+        // FIXME: cancel any parent change (unexpected)
+        // Root parent is permanently pegged at max power,
+        // kIOPMParentInitiated is unexpected.
+        return;
+    }
+
+    if (powerState < currentPowerState)
+    {
+        if ((changeFlags & kIOPMSkipAskPowerDown) == 0)
+        {
+            /* Convenient place to run any code at idle sleep time
+             * IOPMrootDomain initiates an idle sleep here
+             *
+             * Set last sleep cause accordingly.
+             */
+            pmPowerStateQueue->submitPowerEvent(kPowerEventPublishSleepWakeUUID, (void *)true);
+
+            lastSleepReason = kIOPMSleepReasonIdle;
+            setProperty(kRootDomainSleepReasonKey, kIOPMIdleSleepKey);
+        }
+        if (CAP_CURRENT(kIOPMSystemCapabilityGraphics))
+        {
+            // Root domain is dropping power state ON->SLEEP.
+            // If system is in full wake, first drop to dark wake.
+
+            darkWakeToSleepASAP = true;
+
+            // Drop graphics capability.
+            // No transition if system is already in dark wake.
+
+            _desiredCapability &= ~(
+                kIOPMSystemCapabilityGraphics |
+                kIOPMSystemCapabilityAudio    );
+
+            *inOutPowerState = ON_STATE;
+            *inOutChangeFlags |= kIOPMSynchronize;
+
+            // Revert device desire from SLEEP->ON.
+            changePowerStateToPriv(ON_STATE);
+        }
+    }
+}
+
+void IOPMrootDomain::handleOurPowerChangeStart(
+    IOService *     service,
+    IOPMActions *   actions,
+    uint32_t        powerState,
+    uint32_t *      inOutChangeFlags )
+{
+    uint32_t changeFlags = *inOutChangeFlags;
+    uint32_t currentPowerState = (uint32_t) getPowerState();
+
+    _systemTransitionType    = kSystemTransitionNone;
+    _systemMessageClientMask = 0;
+    capabilityLoss           = false;
+
+    // 1. Explicit capability change.
+
+    if (changeFlags & kIOPMSynchronize)
+    {
+        if (powerState == ON_STATE)
+        {
+            if (changeFlags & kIOPMSyncNoChildNotify)
+                _systemTransitionType = kSystemTransitionNewCapClient;
+            else
+                _systemTransitionType = kSystemTransitionCapability;
+        }
+    }
+
+    // 2. Going to sleep (cancellation still possible).
+
+    else if (powerState < currentPowerState)
+        _systemTransitionType = kSystemTransitionSleep;
+
+    // 3. Woke from (idle or demand) sleep.
+
+    else if (!systemBooting &&
+             (changeFlags & kIOPMSelfInitiated) &&
+             (powerState > currentPowerState))
+    {
+        _systemTransitionType = kSystemTransitionWake;
+        _desiredCapability = kIOPMSystemCapabilityCPU |
+                             kIOPMSystemCapabilityNetwork;
+
+        // Check for early HID events (e.g. LID open)
+        if (wranglerTickled)
+        {
+            _desiredCapability |= (
+                kIOPMSystemCapabilityGraphics |
+                kIOPMSystemCapabilityAudio );
+        }
+    }
+
+    // Update pending wake capability at the beginning of every
+    // state transition (including synchronize). This will become
+    // the current capability at the end of the transition.
+
+    if (kSystemTransitionSleep == _systemTransitionType)
+    {
+        _pendingCapability = 0;
+        capabilityLoss = true;
+    }
+    else if (kSystemTransitionNewCapClient != _systemTransitionType)
+    {
+        _pendingCapability = _desiredCapability |
+                             kIOPMSystemCapabilityCPU |
+                             kIOPMSystemCapabilityNetwork;
+
+        if (_pendingCapability & kIOPMSystemCapabilityGraphics)
+            _pendingCapability |= kIOPMSystemCapabilityAudio;
+
+        if ((kSystemTransitionCapability == _systemTransitionType) &&
+            (_pendingCapability == _currentCapability))
+        {
+            // Cancel the PM state change.
+            _systemTransitionType = kSystemTransitionNone;
+            *inOutChangeFlags |= kIOPMNotDone;
+        }
+        if (__builtin_popcount(_pendingCapability) <
+            __builtin_popcount(_currentCapability))
+            capabilityLoss = true;
+        if (CAP_LOSS(kIOPMSystemCapabilityGraphics))
+            rejectWranglerTickle = true;
+    }
+
+    // 1. Capability change.
+
+    if (kSystemTransitionCapability == _systemTransitionType)
+    {
+        // Dark to Full transition.
+        if (CAP_GAIN(kIOPMSystemCapabilityGraphics))
+        {
+            tracePoint( kIOPMTracePointDarkWakeExit );
+            wranglerSleepIgnored = false;
+            sleepTimerMaintenance = false;
+            hibernateNoDefeat = false;
+            _systemMessageClientMask = kSystemMessageClientUser;
+            if ((_highestCapability & kIOPMSystemCapabilityGraphics) == 0)
+                _systemMessageClientMask |= kSystemMessageClientKernel;
+
+            tellClients(kIOMessageSystemWillPowerOn);
+        }
+
+        // Full to Dark transition.
+        if (CAP_LOSS(kIOPMSystemCapabilityGraphics))
+        {
+            tracePoint( kIOPMTracePointDarkWakeEntry );
+            *inOutChangeFlags |= kIOPMSyncTellPowerDown;
+            _systemMessageClientMask = kSystemMessageClientUser;
+        }
+    }
+
+    // 2. System sleep.
+
+    else if (kSystemTransitionSleep == _systemTransitionType)
+    {
+        // Beginning of a system sleep transition.
+        // Cancellation is still possible.
+        tracePoint( kIOPMTracePointSleepStarted, lastSleepReason );
+
+        _systemMessageClientMask = kSystemMessageClientAll;
+        if ((_currentCapability & kIOPMSystemCapabilityGraphics) == 0)
+            _systemMessageClientMask &= ~kSystemMessageClientApp;
+        if ((_highestCapability & kIOPMSystemCapabilityGraphics) == 0)
+            _systemMessageClientMask &= ~kSystemMessageClientKernel;
+
+        // Optimization to ignore wrangler power down thus skipping
+        // the disk spindown and arming the idle timer for demand sleep.
+
+        if (changeFlags & kIOPMIgnoreChildren)
+        {
+            wranglerSleepIgnored = true;
+        }
+
+        logWranglerTickle = false;
+    }
+
+    // 3. System wake.
+
+    else if (kSystemTransitionWake == _systemTransitionType)
+    {
+        wranglerSleepIgnored = false;
+
+        if (_pendingCapability & kIOPMSystemCapabilityGraphics)
+        {
+            _systemMessageClientMask = kSystemMessageClientAll;
+        }
+        else
+        {
+            _systemMessageClientMask = kSystemMessageClientConfigd;
+        }
+
+        tracePoint( kIOPMTracePointWakeWillPowerOnClients );
+        tellClients(kIOMessageSystemWillPowerOn);
+    }
+
+    if ((kSystemTransitionNone != _systemTransitionType) &&
+        (kSystemTransitionNewCapClient != _systemTransitionType))
+    {
+        _systemStateGeneration++;
+        systemDarkWake = false;
+
+        DLOG("=== START (%u->%u, 0x%x) type %u, gen %u, msg %x, "
+             "dcp %x:%x:%x\n",
+            currentPowerState, powerState, *inOutChangeFlags,
+            _systemTransitionType, _systemStateGeneration,
+            _systemMessageClientMask,
+            _desiredCapability, _currentCapability, _pendingCapability);
+    }
+}
+
+void IOPMrootDomain::handleOurPowerChangeDone(
+    IOService *     service,
+    IOPMActions *   actions,
+    uint32_t        powerState,
+    uint32_t        changeFlags )
+{
+    if (kSystemTransitionNewCapClient == _systemTransitionType)
+    {
+        _systemTransitionType = kSystemTransitionNone;
+        return;
+    }
+
+    if (_systemTransitionType != kSystemTransitionNone)
+    {
+        uint32_t currentPowerState = (uint32_t) getPowerState();
+
+        if (changeFlags & kIOPMNotDone)
+        {
+            // Power down was cancelled or vetoed.
+            _pendingCapability = _currentCapability;
+            lastSleepReason = 0;
+
+            if (((_currentCapability & kIOPMSystemCapabilityGraphics) == 0) &&
+                (_currentCapability & kIOPMSystemCapabilityCPU))
+            {
+                pmPowerStateQueue->submitPowerEvent(
+                    kPowerEventPolicyStimulus,
+                    (void *) kStimulusDarkWakeReentry,
+                    _systemStateGeneration );                
+            }
+            
+            // Revert device desire to max.
+            changePowerStateToPriv(ON_STATE);
+        }
+        else
+        {
+            // Send message on dark wake to full wake promotion.
+            // tellChangeUp() handles the normal SLEEP->ON case.
+
+            if (kSystemTransitionCapability == _systemTransitionType)
+            {
+                if (CAP_GAIN(kIOPMSystemCapabilityGraphics))
+                {
+                    tellClients(kIOMessageSystemHasPoweredOn);
+#if DARK_TO_FULL_EVALUATE_CLAMSHELL
+                    // Re-evaluate clamshell state ourselves when graphics
+                    // will not get kIOMessageSystemHasPoweredOn.
+
+                    if (clamshellClosed &&
+                        ((_systemMessageClientMask & kSystemMessageClientKernel) == 0))
+                    {
+                        receivePowerNotification( kLocalEvalClamshellCommand );
+                    }
 #endif
-
-	if ((AOT_STATE == desiredPowerState) && (ON_STATE == currentPowerState)) {
-		// Assertion may have been taken in AOT leading to changePowerStateTo(AOT)
-		*inOutChangeFlags |= kIOPMNotDone;
-		return;
-	}
-
-	if (changeFlags & kIOPMParentInitiated) {
-		// Root parent is permanently pegged at max power,
-		// a parent initiated power change is unexpected.
-		*inOutChangeFlags |= kIOPMNotDone;
-		return;
-	}
-
-#if HIBERNATION && defined(__arm64__)
-	if (lowBatteryCondition && (desiredPowerState < currentPowerState)) {
-		if (!ml_is_secure_hib_supported() || ldmHibernateDisable) {
-			// If hibernation is unsupported, reject sleep requests to avoid
-			// racing with system shutdown.
-			*inOutChangeFlags |= kIOPMNotDone;
-			return;
-		}
-	}
-#endif /* HIBERNATION && defined(__arm64__) */
-
-	if (desiredPowerState < currentPowerState) {
-		if (CAP_CURRENT(kIOPMSystemCapabilityGraphics)) {
-			// Root domain is dropping power state from ON->SLEEP.
-			// If system is in full wake, first enter dark wake by
-			// converting the power drop to a capability change.
-			// Once in dark wake, transition to sleep state ASAP.
-
-			darkWakeToSleepASAP = true;
-
-			// Drop graphics and audio capability
-			_desiredCapability &= ~(
-				kIOPMSystemCapabilityGraphics |
-				kIOPMSystemCapabilityAudio);
-
-			// Convert to capability change (ON->ON)
-			*inOutPowerState = getRUN_STATE();
-			*inOutChangeFlags |= kIOPMSynchronize;
-
-			// Revert device desire from SLEEP to ON
-			changePowerStateWithTagToPriv(getRUN_STATE(), kCPSReasonPowerOverride);
-		} else {
-			// System is already in dark wake, ok to drop power state.
-			// Broadcast root power down to entire tree.
-			*inOutChangeFlags |= kIOPMRootChangeDown;
-		}
-	} else if (desiredPowerState > currentPowerState) {
-		if ((_currentCapability & kIOPMSystemCapabilityCPU) == 0) {
-			// Broadcast power up when waking from sleep, but not for the
-			// initial power change at boot by checking for cpu capability.
-			*inOutChangeFlags |= kIOPMRootChangeUp;
-		}
-	}
-}
-
-void
-IOPMrootDomain::handleOurPowerChangeStart(
-	IOService *             service,
-	IOPMActions *           actions,
-	const IOPMRequest *     request,
-	IOPMPowerStateIndex     newPowerState,
-	IOPMPowerChangeFlags *  inOutChangeFlags )
-{
-	IOPMRequestTag requestTag = request->getTag();
-	IOPMRequestTag sleepReason;
-
-	uint32_t changeFlags        = *inOutChangeFlags;
-	uint32_t currentPowerState  = (uint32_t) getPowerState();
-	bool     publishSleepReason = false;
-
-	// Check if request has a valid sleep reason
-	sleepReason = REQUEST_TAG_TO_REASON(requestTag);
-	if (sleepReason < kIOPMSleepReasonClamshell) {
-		sleepReason = kIOPMSleepReasonIdle;
-	}
-
-	_systemTransitionType    = kSystemTransitionNone;
-	_systemMessageClientMask = 0;
-	capabilityLoss           = false;
-	toldPowerdCapWillChange  = false;
-
-	// Emergency notifications may arrive after the initial sleep request
-	// has been queued. Override the sleep reason so powerd and others can
-	// treat this as an emergency sleep.
-	if (lowBatteryCondition) {
-		sleepReason = kIOPMSleepReasonLowPower;
-	} else if (thermalEmergencyState) {
-		sleepReason = kIOPMSleepReasonThermalEmergency;
-	}
-
-	// 1. Explicit capability change.
-	if (changeFlags & kIOPMSynchronize) {
-		if (newPowerState == ON_STATE) {
-			if (changeFlags & kIOPMSyncNoChildNotify) {
-				setSystemTransitionTypeGated(kSystemTransitionNewCapClient);
-			} else {
-				setSystemTransitionTypeGated(kSystemTransitionCapability);
-			}
-		}
-	}
-	// 2. Going to sleep (cancellation still possible).
-	else if (newPowerState < currentPowerState) {
-		setSystemTransitionTypeGated(kSystemTransitionSleep);
-	}
-	// 3. Woke from (idle or demand) sleep.
-	else if (!systemBooting &&
-	    (changeFlags & kIOPMSelfInitiated) &&
-	    (newPowerState > currentPowerState)) {
-		setSystemTransitionTypeGated(kSystemTransitionWake);
-
-		_desiredCapability = kIOPMSystemCapabilityCPU | kIOPMSystemCapabilityNetwork;
-
-		// Early exit from dark wake to full (e.g. LID open)
-		if (kFullWakeReasonNone != fullWakeReason) {
-			_desiredCapability |= (
-				kIOPMSystemCapabilityGraphics |
-				kIOPMSystemCapabilityAudio);
-
-#if defined(XNU_TARGET_OS_OSX) && !DISPLAY_WRANGLER_PRESENT
-			if (fullWakeReason == kFullWakeReasonLocalUser) {
-				darkWakeExit = true;
-				darkWakeToSleepASAP = false;
-				setProperty(kIOPMRootDomainWakeTypeKey, isRTCAlarmWake ?
-				    kIOPMRootDomainWakeTypeAlarm : kIOPMRootDomainWakeTypeUser);
-			}
-#endif
-		}
-#if HIBERNATION
-		IOHibernateSetWakeCapabilities(_desiredCapability);
-#endif
-	}
-
-	// Update pending wake capability at the beginning of every
-	// state transition (including synchronize). This will become
-	// the current capability at the end of the transition.
-
-	if (kSystemTransitionSleep == _systemTransitionType) {
-		_pendingCapability = 0;
-		capabilityLoss = true;
-	} else if (kSystemTransitionNewCapClient != _systemTransitionType) {
-		_pendingCapability = _desiredCapability |
-		    kIOPMSystemCapabilityCPU |
-		    kIOPMSystemCapabilityNetwork;
-
-		if (_pendingCapability & kIOPMSystemCapabilityGraphics) {
-			_pendingCapability |= kIOPMSystemCapabilityAudio;
-		}
-
-		if ((kSystemTransitionCapability == _systemTransitionType) &&
-		    (_pendingCapability == _currentCapability)) {
-			// Cancel the PM state change.
-			setSystemTransitionTypeGated(kSystemTransitionNone);
-			*inOutChangeFlags |= kIOPMNotDone;
-		}
-		if (__builtin_popcount(_pendingCapability) <
-		    __builtin_popcount(_currentCapability)) {
-			capabilityLoss = true;
-		}
-	}
-
-	// 1. Capability change.
-	if (kSystemTransitionCapability == _systemTransitionType) {
-		// Dark to Full transition.
-		if (CAP_GAIN(kIOPMSystemCapabilityGraphics)) {
-			tracePoint( kIOPMTracePointDarkWakeExit );
-
-#if defined(XNU_TARGET_OS_OSX)
-			// rdar://problem/65627936
-			// When a dark->full wake promotion is scheduled before an ON->SLEEP
-			// power state drop, invalidate any request to drop power state already
-			// in the queue, including the override variant, unless full wake cannot
-			// be sustained. Any power state drop queued after this SustainFullWake
-			// request will not be affected.
-			if (checkSystemCanSustainFullWake()) {
-				changePowerStateWithOverrideTo(getRUN_STATE(), kCPSReasonSustainFullWake);
-			}
-#endif
-
-			willEnterFullWake();
-		}
-
-		// Full to Dark transition.
-		if (CAP_LOSS(kIOPMSystemCapabilityGraphics)) {
-			// Clear previous stats
-			IOLockLock(pmStatsLock);
-			if (pmStatsAppResponses) {
-				pmStatsAppResponses = OSArray::withCapacity(5);
-			}
-			IOLockUnlock(pmStatsLock);
-
-			tracePoint( kIOPMTracePointDarkWakeEntry );
-			*inOutChangeFlags |= kIOPMSyncTellPowerDown;
-			_systemMessageClientMask = kSystemMessageClientPowerd |
-			    kSystemMessageClientLegacyApp;
-
-			// rdar://15971327
-			// Prevent user active transitions before notifying clients
-			// that system will sleep.
-			preventTransitionToUserActive(true);
-
-			IOService::setAdvisoryTickleEnable( false );
-
-			// Publish the sleep reason for full to dark wake
-			publishSleepReason = true;
-			lastSleepReason = fullToDarkReason = sleepReason;
-
-			// Publish a UUID for the Sleep --> Wake cycle
-			handlePublishSleepWakeUUID(true);
-			if (sleepDelaysReport) {
-				clock_get_uptime(&ts_sleepStart);
-				DLOG("sleepDelaysReport f->9 start at 0x%llx\n", ts_sleepStart);
-			}
-
-			darkWakeExit = false;
-		}
-	}
-	// 2. System sleep.
-	else if (kSystemTransitionSleep == _systemTransitionType) {
-		// Beginning of a system sleep transition.
-		// Cancellation is still possible.
-		tracePoint( kIOPMTracePointSleepStarted );
-
-		_systemMessageClientMask = kSystemMessageClientAll;
-		if ((_currentCapability & kIOPMSystemCapabilityGraphics) == 0) {
-			_systemMessageClientMask &= ~kSystemMessageClientLegacyApp;
-		}
-		if ((_highestCapability & kIOPMSystemCapabilityGraphics) == 0) {
-			// Kernel priority clients are only notified on the initial
-			// transition to full wake, so don't notify them unless system
-			// has gained graphics capability since the last system wake.
-			_systemMessageClientMask &= ~kSystemMessageClientKernel;
-		} else {
-			// System was in full wake, but the downwards power transition is driven
-			// by a request that originates from IOServicePM, so it isn't tagged with
-			// a valid system sleep reason.
-			if (REQUEST_TAG_TO_REASON(requestTag) == kCPSReasonPMInternals) {
-				// Publish the same reason for full to dark
-				sleepReason = fullToDarkReason;
-			}
-		}
-#if HIBERNATION
-		gIOHibernateState = 0;
-#endif
-
-		// Record the reason for dark wake back to sleep
-		// System may not have ever achieved full wake
-
-		publishSleepReason = true;
-		lastSleepReason = sleepReason;
-		if (sleepDelaysReport) {
-			clock_get_uptime(&ts_sleepStart);
-			DLOG("sleepDelaysReport 9->0 start at 0x%llx\n", ts_sleepStart);
-		}
-	}
-	// 3. System wake.
-	else if (kSystemTransitionWake == _systemTransitionType) {
-		tracePoint( kIOPMTracePointWakeWillPowerOnClients );
-		// Clear stats about sleep
-
-		if (AOT_STATE == newPowerState) {
-			// Temporarily increase the capacity of the log subsystem to buffer logs in AOT.
-			os_log_adjust_buffering_capacity(LOG_BUFFERING_CAPACITY_MAX);
-			_pendingCapability = kIOPMSystemCapabilityAOT;
-		} else {
-			// Restore the maximum capacity of the log subsystem once waking to full wake.
-			os_log_adjust_buffering_capacity(LOG_BUFFERING_CAPACITY_DEFAULT);
-		}
-
-		if (AOT_STATE == currentPowerState) {
-			// Wake events are no longer accepted after waking to AOT_STATE.
-			// Re-enable wake event acceptance to append wake events claimed
-			// during the AOT to ON_STATE transition.
-			acceptSystemWakeEvents(kAcceptSystemWakeEvents_Reenable);
-		}
-
-		if (_pendingCapability & kIOPMSystemCapabilityGraphics) {
-			willEnterFullWake();
-		}
-	}
-
-	// The only location where the sleep reason is published. At this point
-	// sleep can still be cancelled, but sleep reason should be published
-	// early for logging purposes.
-
-	if (publishSleepReason) {
-		static const char * IOPMSleepReasons[] =
-		{
-			kIOPMClamshellSleepKey,
-			kIOPMPowerButtonSleepKey,
-			kIOPMSoftwareSleepKey,
-			kIOPMOSSwitchHibernationKey,
-			kIOPMIdleSleepKey,
-			kIOPMLowPowerSleepKey,
-			kIOPMThermalEmergencySleepKey,
-			kIOPMMaintenanceSleepKey,
-			kIOPMSleepServiceExitKey,
-			kIOPMDarkWakeThermalEmergencyKey,
-			kIOPMNotificationWakeExitKey
-		};
-
-		// Record sleep cause in IORegistry
-		uint32_t reasonIndex = sleepReason - kIOPMSleepReasonClamshell;
-		if (reasonIndex < sizeof(IOPMSleepReasons) / sizeof(IOPMSleepReasons[0])) {
-			DLOG("sleep reason %s\n", IOPMSleepReasons[reasonIndex]);
-#if DEVELOPMENT || DEBUG
-			record_system_event(SYSTEM_EVENT_TYPE_INFO,
-			    SYSTEM_EVENT_SUBSYSTEM_PMRD,
-			    "Sleep Reason", "%s\n", IOPMSleepReasons[reasonIndex]
-			    );
-#endif /* DEVELOPMENT || DEBUG */
-			setProperty(kRootDomainSleepReasonKey, IOPMSleepReasons[reasonIndex]);
-		}
-	}
-
-	if ((kSystemTransitionNone != _systemTransitionType) &&
-	    (kSystemTransitionNewCapClient != _systemTransitionType)) {
-		_systemStateGeneration++;
-		systemDarkWake = false;
-
-		DLOG("=== START (%s->%s, %x->%x, 0x%x) gen %u, msg %x, tag %x\n",
-		    getPowerStateString(currentPowerState),
-		    getPowerStateString((uint32_t) newPowerState),
-		    _currentCapability, _pendingCapability,
-		    *inOutChangeFlags, _systemStateGeneration, _systemMessageClientMask,
-		    requestTag);
-#if DEVELOPMENT || DEBUG
-		if (currentPowerState != (uint32_t) newPowerState) {
-			record_system_event(SYSTEM_EVENT_TYPE_INFO,
-			    SYSTEM_EVENT_SUBSYSTEM_PMRD,
-			    "Start Power State Trans.",
-			    "(%s->%s, %x->%x, 0x%x) gen %u, msg %x, tag %x\n",
-			    getPowerStateString(currentPowerState),
-			    getPowerStateString((uint32_t) newPowerState),
-			    _currentCapability,
-			    _pendingCapability,
-			    *inOutChangeFlags,
-			    _systemStateGeneration,
-			    _systemMessageClientMask,
-			    requestTag
-			    );
-		}
-#endif /* DEVELOPMENT || DEBUG */
-	}
-
-	if ((AOT_STATE == newPowerState) && (SLEEP_STATE != currentPowerState)) {
-		panic("illegal AOT entry from %s", getPowerStateString(currentPowerState));
-	}
-	if (_aotNow && (ON_STATE == newPowerState)) {
-		WAKEEVENT_LOCK();
-		aotShouldExit(true);
-		WAKEEVENT_UNLOCK();
-		aotExit(false);
-	}
-}
-
-void
-IOPMrootDomain::setSystemTransitionTypeGated(SystemTransitionType type)
-{
-	assert(gIOPMWorkLoop->inGate());
-	_systemTransitionType = type;
-	commandGate->commandWakeup(&_systemTransitionType);
-}
-
-void
-IOPMrootDomain::waitForSystemTransitionToMinPowerState(IOPMRootDomainPowerState state)
-{
-	while (true) {
-		IOReturn ret = gIOPMWorkLoop->runActionBlock(^{
-			// Block until all in progress transitions have completed.
-			while (_systemTransitionType != kSystemTransitionNone) {
-			        commandGate->commandSleep(&_systemTransitionType);
-			}
-
-			// Check the current power state.
-			if (getPowerState() >= state) {
-			        return kIOReturnSuccess;
-			}
-
-			return kIOReturnError;
-		});
-
-		if (ret == kIOReturnSuccess) {
-			break;
-		}
-	}
-}
-
-void
-IOPMrootDomain::handleOurPowerChangeDone(
-	IOService *             service,
-	IOPMActions *           actions,
-	const IOPMRequest *     request,
-	IOPMPowerStateIndex     oldPowerState,
-	IOPMPowerChangeFlags    changeFlags )
-{
-	if (kSystemTransitionNewCapClient == _systemTransitionType) {
-		setSystemTransitionTypeGated(kSystemTransitionNone);
-		return;
-	}
-
-	if (_systemTransitionType != kSystemTransitionNone) {
-		uint32_t currentPowerState = (uint32_t) getPowerState();
-
-		if (changeFlags & kIOPMNotDone) {
-			// Power down was cancelled or vetoed.
-			_pendingCapability = _currentCapability;
-			lastSleepReason = 0;
-
-			// When sleep is cancelled or reverted, don't report
-			// the target (lower) power state as the previous state.
-			oldPowerState = currentPowerState;
-
-			if (!CAP_CURRENT(kIOPMSystemCapabilityGraphics) &&
-			    CAP_CURRENT(kIOPMSystemCapabilityCPU)) {
-#if defined(XNU_TARGET_OS_OSX)
-				pmPowerStateQueue->submitPowerEvent(
-					kPowerEventPolicyStimulus,
-					(void *) kStimulusDarkWakeReentry,
-					_systemStateGeneration );
-#else /* !defined(XNU_TARGET_OS_OSX) */
-				// On embedded, there are no factors that can prolong a
-				// "darkWake" when a power down is vetoed. We need to
-				// promote to "fullWake" at least once so that factors
-				// that prevent idle sleep can assert themselves if required
-				pmPowerStateQueue->submitPowerEvent(
-					kPowerEventPolicyStimulus,
-					(void *) kStimulusDarkWakeActivityTickle);
-#endif /* !defined(XNU_TARGET_OS_OSX) */
-			}
-
-			// Revert device desire to max.
-			changePowerStateWithTagToPriv(getRUN_STATE(), kCPSReasonPowerDownCancel);
-		} else {
-			// Send message on dark wake to full wake promotion.
-			// tellChangeUp() handles the normal SLEEP->ON case.
-
-			if (kSystemTransitionCapability == _systemTransitionType) {
-				if (CAP_GAIN(kIOPMSystemCapabilityGraphics)) {
-					lastSleepReason = 0; // stop logging wrangler tickles
-					tellClients(kIOMessageSystemHasPoweredOn);
-				}
-				if (CAP_LOSS(kIOPMSystemCapabilityGraphics)) {
-					// Going dark, reset full wake state
-					// userIsActive will be cleared by wrangler powering down
-					fullWakeReason = kFullWakeReasonNone;
-
-					if (ts_sleepStart) {
-						clock_get_uptime(&wake2DarkwakeDelay);
-						SUB_ABSOLUTETIME(&wake2DarkwakeDelay, &ts_sleepStart);
-						DLOG("sleepDelaysReport f->9 end 0x%llx\n", wake2DarkwakeDelay);
-						ts_sleepStart = 0;
-					}
-				}
-			}
-
-			// Reset state after exiting from dark wake.
-
-			if (CAP_GAIN(kIOPMSystemCapabilityGraphics) ||
-			    CAP_LOSS(kIOPMSystemCapabilityCPU)) {
-				darkWakeMaintenance = false;
-				darkWakeToSleepASAP = false;
-				pciCantSleepValid   = false;
-				darkWakeSleepService = false;
-
-				if (CAP_LOSS(kIOPMSystemCapabilityCPU)) {
-					// Remove the influence of display power assertion
-					// before next system wake.
-					if (wrangler) {
-						wrangler->changePowerStateForRootDomain(
-							kWranglerPowerStateMin );
-					}
-					removeProperty(gIOPMUserTriggeredFullWakeKey.get());
-				}
-			}
-
-			// Entered dark mode.
-
-			if (((_pendingCapability & kIOPMSystemCapabilityGraphics) == 0) &&
-			    (_pendingCapability & kIOPMSystemCapabilityCPU)) {
-				// Queue an evaluation of whether to remain in dark wake,
-				// and for how long. This serves the purpose of draining
-				// any assertions from the queue.
-
-				pmPowerStateQueue->submitPowerEvent(
-					kPowerEventPolicyStimulus,
-					(void *) kStimulusDarkWakeEntry,
-					_systemStateGeneration );
-			}
-		}
-
-#if DEVELOPMENT || DEBUG
-		if (currentPowerState != (uint32_t) oldPowerState) {
-			record_system_event(SYSTEM_EVENT_TYPE_INFO,
-			    SYSTEM_EVENT_SUBSYSTEM_PMRD,
-			    "Finish Power State Trans.",
-			    "(%s->%s, %x->%x, 0x%x) gen %u, msg %x, tag %x\n",
-			    getPowerStateString((uint32_t)oldPowerState),
-			    getPowerStateString(currentPowerState),
-			    _currentCapability,
-			    _pendingCapability,
-			    changeFlags,
-			    _systemStateGeneration,
-			    _systemMessageClientMask,
-			    request->getTag()
-			    );
-		}
-#endif /* DEVELOPMENT || DEBUG */
-
-		DLOG("=== FINISH (%s->%s, %x->%x, 0x%x) gen %u, msg %x, tag %x\n",
-		    getPowerStateString((uint32_t) oldPowerState), getPowerStateString(currentPowerState),
-		    _currentCapability, _pendingCapability,
-		    changeFlags, _systemStateGeneration, _systemMessageClientMask,
-		    request->getTag());
-
-		if ((currentPowerState == ON_STATE) && pmAssertions) {
-			pmAssertions->reportCPUBitAccounting();
-		}
-
-		if (_pendingCapability & kIOPMSystemCapabilityGraphics) {
-			displayWakeCnt++;
-#if DARK_TO_FULL_EVALUATE_CLAMSHELL_DELAY
-			if (clamshellExists && fullWakeThreadCall) {
-				AbsoluteTime deadline;
-				clock_interval_to_deadline(DARK_TO_FULL_EVALUATE_CLAMSHELL_DELAY, kSecondScale, &deadline);
-				thread_call_enter_delayed(fullWakeThreadCall, deadline);
-			}
-#endif
-		} else if (CAP_GAIN(kIOPMSystemCapabilityCPU)) {
-			darkWakeCnt++;
-		}
-
-		// Update current system capability.
-		if (_currentCapability != _pendingCapability) {
-			_currentCapability = _pendingCapability;
-		}
-
-		// Update highest system capability.
-
-		_highestCapability |= _currentCapability;
-
-		if (darkWakePostTickle &&
-		    (kSystemTransitionWake == _systemTransitionType) &&
-		    (gDarkWakeFlags & kDarkWakeFlagPromotionMask) ==
-		    kDarkWakeFlagPromotionLate) {
-			darkWakePostTickle = false;
-			reportUserInput();
-		} else if (darkWakeExit) {
-			requestFullWake( kFullWakeReasonLocalUser );
-		}
-
-		// Reset tracepoint at completion of capability change,
-		// completion of wake transition, and aborted sleep transition.
-
-		if ((_systemTransitionType == kSystemTransitionCapability) ||
-		    (_systemTransitionType == kSystemTransitionWake) ||
-		    ((_systemTransitionType == kSystemTransitionSleep) &&
-		    (changeFlags & kIOPMNotDone))) {
-			setProperty(kIOPMSystemCapabilitiesKey, _currentCapability, 64);
-			tracePoint( kIOPMTracePointSystemUp );
-		}
-
-		setSystemTransitionTypeGated(kSystemTransitionNone);
-
-		_systemMessageClientMask = 0;
-		toldPowerdCapWillChange  = false;
-
-		darkWakeLogClamp = false;
-
-		if (lowBatteryCondition) {
-			privateSleepSystem(kIOPMSleepReasonLowPower);
-		} else if (thermalEmergencyState) {
-			privateSleepSystem(kIOPMSleepReasonThermalEmergency);
-		} else if ((fullWakeReason == kFullWakeReasonDisplayOn) && !displayPowerOnRequested) {
-			// Request for full wake is removed while system is waking up to full wake
-			DLOG("DisplayOn fullwake request is removed\n");
-			handleSetDisplayPowerOn(false);
-		}
-
-		if ((gClamshellFlags & kClamshell_WAR_47715679) && isRTCAlarmWake) {
-			pmPowerStateQueue->submitPowerEvent(
-				kPowerEventReceivedPowerNotification, (void *)(uintptr_t) kLocalEvalClamshellCommand );
-		}
-	}
+                }
+                if (CAP_LOSS(kIOPMSystemCapabilityGraphics))
+                    wranglerTickled = false;
+            }
+
+            // Reset state after exiting from dark wake.
+
+            if (CAP_GAIN(kIOPMSystemCapabilityGraphics) ||
+                CAP_LOSS(kIOPMSystemCapabilityCPU))
+            {
+                darkWakeMaintenance = false;
+                darkWakeToSleepASAP = false;
+                pciCantSleepValid   = false;
+                rejectWranglerTickle = false;
+                darkWakeSleepService = false;
+            }
+
+            // Entered dark mode.
+
+            if (((_pendingCapability & kIOPMSystemCapabilityGraphics) == 0) &&
+                 (_pendingCapability & kIOPMSystemCapabilityCPU))
+            {
+                if (((gDarkWakeFlags & kDarkWakeFlagIgnoreDiskIOInDark) == 0) &&
+                    (kSystemTransitionWake == _systemTransitionType) &&
+                    (_lastDebugWakeSeconds == 0))
+                {
+                    OSObject * prop = copyProperty(kIOPMRootDomainWakeTypeKey);
+                    if (prop)
+                    {
+                        OSString * wakeType = OSDynamicCast(OSString, prop);
+                        if (wakeType &&
+                            wakeType->isEqualTo(kIOPMRootDomainWakeTypeNetwork))
+                        {
+                            // Woke from network and entered dark wake.                    
+                            if (darkWakeToSleepASAP)
+                            {
+                                DLOG("cleared darkWakeToSleepASAP\n");
+                                darkWakeToSleepASAP = false;
+                            }
+                        }
+                        prop->release();
+                    }
+                }
+
+                // Queue an evaluation of whether to remain in dark wake,
+                // and for how long. This serves the purpose of draining
+                // any assertions from the queue.
+
+                pmPowerStateQueue->submitPowerEvent(
+                    kPowerEventPolicyStimulus,
+                    (void *) kStimulusDarkWakeEntry,
+                    _systemStateGeneration );
+            }
+        }
+
+        DLOG("=== FINISH (%u->%u, 0x%x) type %u, gen %u, msg %x, "
+             "dcp %x:%x:%x, dbgtimer %u\n",
+            currentPowerState, powerState, changeFlags,
+            _systemTransitionType, _systemStateGeneration,
+            _systemMessageClientMask,
+            _desiredCapability, _currentCapability, _pendingCapability,
+            _lastDebugWakeSeconds);
+
+        // Update current system capability.
+
+        if (_currentCapability != _pendingCapability)
+            _currentCapability = _pendingCapability;
+
+        // Update highest system capability.
+
+        if (!CAP_CURRENT(kIOPMSystemCapabilityCPU))
+            _highestCapability = 0;     // reset at sleep state
+        else
+            _highestCapability |= _currentCapability;
+
+        if (darkWakePostTickle &&
+            (kSystemTransitionWake == _systemTransitionType) &&
+            (gDarkWakeFlags & kDarkWakeFlagHIDTickleMask) ==
+             kDarkWakeFlagHIDTickleLate)
+        {
+            darkWakePostTickle = false;
+            reportUserInput();
+        }
+
+        // Reset tracepoint at completion of capability change,
+        // completion of wake transition, and aborted sleep transition.
+
+        if ((_systemTransitionType == kSystemTransitionCapability) ||
+            (_systemTransitionType == kSystemTransitionWake) ||
+            ((_systemTransitionType == kSystemTransitionSleep) &&
+             (changeFlags & kIOPMNotDone)))
+        {
+            setProperty(kIOPMSystemCapabilitiesKey, _currentCapability, 64);
+            tracePoint( kIOPMTracePointSystemUp, 0 );
+
+            // kIOPMDWOverTemp notification handling was postponed
+            if (darkWakeThermalAlarm)
+            {
+                if (!wranglerTickled && !darkWakeThermalEmergency &&
+                    CAP_CURRENT(kIOPMSystemCapabilityCPU) &&
+                    !CAP_CURRENT(kIOPMSystemCapabilityGraphics))
+                {
+                    darkWakeThermalEmergency = true;
+                    privateSleepSystem(kIOPMSleepReasonDarkWakeThermalEmergency);
+                    MSG("DarkWake thermal limits breached. Going to sleep!\n");
+                }
+                darkWakeThermalAlarm = false;
+            }
+        }
+
+        _systemTransitionType = kSystemTransitionNone;
+        _systemMessageClientMask = 0;
+
+        logGraphicsClamp = false;
+    }
 }
 
 //******************************************************************************
 // PM actions for graphics and audio.
 //******************************************************************************
 
-void
-IOPMrootDomain::overridePowerChangeForService(
-	IOService *             service,
-	IOPMActions *           actions,
-	const IOPMRequest *     request,
-	IOPMPowerStateIndex *   inOutPowerState,
-	IOPMPowerChangeFlags *  inOutChangeFlags )
-{
-	uint32_t powerState  = (uint32_t) *inOutPowerState;
-	uint32_t changeFlags = (uint32_t) *inOutChangeFlags;
-	const uint32_t actionFlags = actions->flags;
-
-	if (kSystemTransitionNone == _systemTransitionType) {
-		// Not in midst of a system transition.
-		// Do not set kPMActionsStatePowerClamped.
-	} else if ((actions->state & kPMActionsStatePowerClamped) == 0) {
-		bool enableClamp = false;
-
-		// For most drivers, enable the clamp during ON->Dark transition
-		// which has the kIOPMSynchronize flag set in changeFlags.
-		if ((actionFlags & kPMActionsFlagIsDisplayWrangler) &&
-		    ((_pendingCapability & kIOPMSystemCapabilityGraphics) == 0) &&
-		    (changeFlags & kIOPMSynchronize)) {
-			enableClamp = true;
-		} else if ((actionFlags & kPMActionsFlagIsAudioDriver) &&
-		    ((gDarkWakeFlags & kDarkWakeFlagAudioNotSuppressed) == 0) &&
-		    ((_pendingCapability & kIOPMSystemCapabilityAudio) == 0) &&
-		    (changeFlags & kIOPMSynchronize)) {
-			enableClamp = true;
-		} else if ((actionFlags & kPMActionsFlagHasDarkWakePowerState) &&
-		    ((_pendingCapability & kIOPMSystemCapabilityGraphics) == 0) &&
-		    (changeFlags & kIOPMSynchronize)) {
-			enableClamp = true;
-		} else if ((actionFlags & kPMActionsFlagIsGraphicsDriver) &&
-		    (_systemTransitionType == kSystemTransitionSleep)) {
-			// For graphics drivers, clamp power when entering
-			// system sleep. Not when dropping to dark wake.
-			enableClamp = true;
-		}
-
-		if (enableClamp) {
-			actions->state |= kPMActionsStatePowerClamped;
-			DLOG("power clamp enabled %s %qx, pendingCap 0x%x, ps %d, cflags 0x%x\n",
-			    service->getName(), service->getRegistryEntryID(),
-			    _pendingCapability, powerState, changeFlags);
-		}
-	} else if ((actions->state & kPMActionsStatePowerClamped) != 0) {
-		bool disableClamp = false;
-
-		if ((actionFlags & (
-			    kPMActionsFlagIsDisplayWrangler |
-			    kPMActionsFlagIsGraphicsDriver)) &&
-		    (_pendingCapability & kIOPMSystemCapabilityGraphics)) {
-			disableClamp = true;
-		} else if ((actionFlags & kPMActionsFlagIsAudioDriver) &&
-		    (_pendingCapability & kIOPMSystemCapabilityAudio)) {
-			disableClamp = true;
-		} else if ((actionFlags & kPMActionsFlagHasDarkWakePowerState) &&
-		    (_pendingCapability & kIOPMSystemCapabilityGraphics)) {
-			disableClamp = true;
-		}
-
-		if (disableClamp) {
-			actions->state &= ~kPMActionsStatePowerClamped;
-			DLOG("power clamp removed %s %qx, pendingCap 0x%x, ps %d, cflags 0x%x\n",
-			    service->getName(), service->getRegistryEntryID(),
-			    _pendingCapability, powerState, changeFlags);
-		}
-	}
-
-	if (actions->state & kPMActionsStatePowerClamped) {
-		uint32_t maxPowerState = 0;
-
-		// Determine the max power state allowed when clamp is enabled
-		if (changeFlags & (kIOPMDomainDidChange | kIOPMDomainWillChange)) {
-			// Parent intiated power state changes
-			if ((service->getPowerState() > maxPowerState) &&
-			    (actionFlags & kPMActionsFlagIsDisplayWrangler)) {
-				maxPowerState++;
-
-				// Remove lingering effects of any tickle before entering
-				// dark wake. It will take a new tickle to return to full
-				// wake, so the existing tickle state is useless.
-
-				if (changeFlags & kIOPMDomainDidChange) {
-					*inOutChangeFlags |= kIOPMExpireIdleTimer;
-				}
-			} else if (actionFlags & kPMActionsFlagIsGraphicsDriver) {
-				maxPowerState++;
-			} else if (actionFlags & kPMActionsFlagHasDarkWakePowerState) {
-				maxPowerState = actions->darkWakePowerState;
-			}
-		} else {
-			// Deny all self-initiated changes when power is limited.
-			// Wrangler tickle should never defeat the limiter.
-			maxPowerState = service->getPowerState();
-		}
-
-		if (powerState > maxPowerState) {
-			DLOG("power clamped %s %qx, ps %u->%u, cflags 0x%x)\n",
-			    service->getName(), service->getRegistryEntryID(),
-			    powerState, maxPowerState, changeFlags);
-			*inOutPowerState = maxPowerState;
-
-			if (darkWakePostTickle &&
-			    (actionFlags & kPMActionsFlagIsDisplayWrangler) &&
-			    (changeFlags & kIOPMDomainWillChange) &&
-			    ((gDarkWakeFlags & kDarkWakeFlagPromotionMask) ==
-			    kDarkWakeFlagPromotionEarly)) {
-				darkWakePostTickle = false;
-				reportUserInput();
-			}
-		}
-
-		if (!darkWakePowerClamped && (changeFlags & kIOPMDomainDidChange)) {
-			if (darkWakeLogClamp) {
-				AbsoluteTime    now;
-				uint64_t        nsec;
-
-				clock_get_uptime(&now);
-				SUB_ABSOLUTETIME(&now, &gIOLastWakeAbsTime);
-				absolutetime_to_nanoseconds(now, &nsec);
-				DLOG("dark wake power clamped after %u ms\n",
-				    ((int)((nsec) / NSEC_PER_MSEC)));
-			}
-			darkWakePowerClamped = true;
-		}
-	}
-}
-
-void
-IOPMrootDomain::handleActivityTickleForDisplayWrangler(
-	IOService *     service,
-	IOPMActions *   actions )
-{
-#if DISPLAY_WRANGLER_PRESENT
-	// Warning: Not running in PM work loop context - don't modify state !!!
-	// Trap tickle directed to IODisplayWrangler while running with graphics
-	// capability suppressed.
-
-	assert(service == wrangler);
-
-	clock_get_uptime(&userActivityTime);
-	bool aborting = ((lastSleepReason == kIOPMSleepReasonIdle)
-	    || (lastSleepReason == kIOPMSleepReasonMaintenance)
-	    || (lastSleepReason == kIOPMSleepReasonSoftware));
-	if (aborting) {
-		userActivityCount++;
-		DLOG("display wrangler tickled1 %d lastSleepReason %d\n",
-		    userActivityCount, lastSleepReason);
-	}
-
-	if (!darkWakeExit && ((_pendingCapability & kIOPMSystemCapabilityGraphics) == 0)) {
-		DLOG("display wrangler tickled\n");
-		if (kIOLogPMRootDomain & gIOKitDebug) {
-			OSReportWithBacktrace("Dark wake display tickle");
-		}
-		if (pmPowerStateQueue) {
-			pmPowerStateQueue->submitPowerEvent(
-				kPowerEventPolicyStimulus,
-				(void *) kStimulusDarkWakeActivityTickle,
-				true /* set wake type */ );
-		}
-	}
-#endif /* DISPLAY_WRANGLER_PRESENT */
-}
-
-void
-IOPMrootDomain::handleUpdatePowerClientForDisplayWrangler(
-	IOService *             service,
-	IOPMActions *           actions,
-	const OSSymbol *        powerClient,
-	IOPMPowerStateIndex     oldPowerState,
-	IOPMPowerStateIndex     newPowerState )
-{
-#if DISPLAY_WRANGLER_PRESENT
-	assert(service == wrangler);
-
-	// This function implements half of the user active detection
-	// by monitoring changes to the display wrangler's device desire.
-	//
-	// User becomes active when either:
-	// 1. Wrangler's DeviceDesire increases to max, but wrangler is already
-	//    in max power state. This desire change in absence of a power state
-	//    change is detected within. This handles the case when user becomes
-	//    active while the display is already lit by setDisplayPowerOn().
-	//
-	// 2. Power state change to max, and DeviceDesire is also at max.
-	//    Handled by displayWranglerNotification().
-	//
-	// User becomes inactive when DeviceDesire drops to sleep state or below.
-
-	DLOG("wrangler %s (ps %u, %u->%u)\n",
-	    powerClient->getCStringNoCopy(),
-	    (uint32_t) service->getPowerState(),
-	    (uint32_t) oldPowerState, (uint32_t) newPowerState);
-
-	if (powerClient == gIOPMPowerClientDevice) {
-		if ((newPowerState > oldPowerState) &&
-		    (newPowerState == kWranglerPowerStateMax) &&
-		    (service->getPowerState() == kWranglerPowerStateMax)) {
-			evaluatePolicy( kStimulusEnterUserActiveState );
-		} else if ((newPowerState < oldPowerState) &&
-		    (newPowerState <= kWranglerPowerStateSleep)) {
-			evaluatePolicy( kStimulusLeaveUserActiveState );
-		}
-	}
-
-	if (newPowerState <= kWranglerPowerStateSleep) {
-		evaluatePolicy( kStimulusDisplayWranglerSleep );
-	} else if (newPowerState == kWranglerPowerStateMax) {
-		evaluatePolicy( kStimulusDisplayWranglerWake );
-	}
-#endif /* DISPLAY_WRANGLER_PRESENT */
-}
-
-//******************************************************************************
-// User active state management
-//******************************************************************************
-
-void
-IOPMrootDomain::preventTransitionToUserActive( bool prevent )
-{
-#if DISPLAY_WRANGLER_PRESENT
-	_preventUserActive = prevent;
-	if (wrangler && !_preventUserActive) {
-		// Allowing transition to user active, but the wrangler may have
-		// already powered ON in case of sleep cancel/revert. Poll the
-		// same conditions checked for in displayWranglerNotification()
-		// to bring the user active state up to date.
-
-		if ((wrangler->getPowerState() == kWranglerPowerStateMax) &&
-		    (wrangler->getPowerStateForClient(gIOPMPowerClientDevice) ==
-		    kWranglerPowerStateMax)) {
-			evaluatePolicy( kStimulusEnterUserActiveState );
-		}
-	}
-#endif /* DISPLAY_WRANGLER_PRESENT */
+void IOPMrootDomain::overridePowerChangeForUIService(
+    IOService *     service,
+    IOPMActions *   actions,
+    unsigned long * inOutPowerState,
+    uint32_t *      inOutChangeFlags )
+{
+    uint32_t powerState  = (uint32_t) *inOutPowerState;
+    uint32_t changeFlags = (uint32_t) *inOutChangeFlags;
+
+    if (kSystemTransitionNone == _systemTransitionType)
+    {
+        // Not in midst of a system transition.
+        // Do not modify power limit enable state.
+    }
+    else if ((actions->parameter & kPMActionsFlagLimitPower) == 0)
+    {
+        // Activate power limiter.
+
+        if ((actions->parameter & kPMActionsFlagIsDisplayWrangler) &&
+            ((_pendingCapability & kIOPMSystemCapabilityGraphics) == 0) &&
+            (changeFlags & kIOPMSynchronize))
+        {
+            actions->parameter |= kPMActionsFlagLimitPower;
+        }
+        else if ((actions->parameter & kPMActionsFlagIsAudioDevice) &&
+                 ((_pendingCapability & kIOPMSystemCapabilityAudio) == 0) &&
+                 (changeFlags & kIOPMSynchronize))
+        {
+            actions->parameter |= kPMActionsFlagLimitPower;
+        }
+        else if ((actions->parameter & kPMActionsFlagIsGraphicsDevice) &&
+                 (_systemTransitionType == kSystemTransitionSleep))
+        {
+            // For graphics devices, arm the limiter when entering
+            // system sleep. Not when dropping to dark wake.
+            actions->parameter |= kPMActionsFlagLimitPower; 
+        }
+
+        if (actions->parameter & kPMActionsFlagLimitPower)
+        {
+            DLOG("+ plimit %s %p\n",
+                service->getName(), service);
+        }
+    }
+    else
+    {
+        // Remove power limit.
+
+        if ((actions->parameter & (
+            kPMActionsFlagIsDisplayWrangler |
+            kPMActionsFlagIsGraphicsDevice )) &&
+            (_pendingCapability & kIOPMSystemCapabilityGraphics))
+        {
+            actions->parameter &= ~kPMActionsFlagLimitPower;
+        }
+        else if ((actions->parameter & kPMActionsFlagIsAudioDevice) &&
+                 (_pendingCapability & kIOPMSystemCapabilityAudio))
+        {
+            actions->parameter &= ~kPMActionsFlagLimitPower;
+        }
+
+        if ((actions->parameter & kPMActionsFlagLimitPower) == 0)
+        {
+            DLOG("- plimit %s %p\n",
+                service->getName(), service);
+        }
+    }
+
+    if (actions->parameter & kPMActionsFlagLimitPower)
+    {
+        uint32_t maxPowerState = (uint32_t)(-1);
+
+        if (changeFlags & (kIOPMDomainDidChange | kIOPMDomainWillChange))
+        {
+            // Enforce limit for system power/cap transitions.
+
+            maxPowerState = 0;
+            if ((actions->parameter & kPMActionsFlagIsDisplayWrangler) &&
+                (service->getPowerState() > 0))
+            {
+                // Forces a 3->1 transition sequence
+                if (changeFlags & kIOPMDomainWillChange)
+                    maxPowerState = 3;
+                else
+                    maxPowerState = 1;
+            }
+        }
+        else
+        {
+            // Deny all self-initiated changes when power is limited.
+            // Wrangler tickle should never defeat the limiter.
+
+            maxPowerState = service->getPowerState();
+        }
+
+        if (powerState > maxPowerState)
+        {
+            DLOG("> plimit %s %p (%u->%u, 0x%x)\n",
+                service->getName(), service, powerState, maxPowerState,
+                changeFlags);
+            *inOutPowerState = maxPowerState;
+
+            if (darkWakePostTickle &&
+                (actions->parameter & kPMActionsFlagIsDisplayWrangler) &&
+                (changeFlags & kIOPMDomainWillChange) &&
+                ((gDarkWakeFlags & kDarkWakeFlagHIDTickleMask) ==
+                 kDarkWakeFlagHIDTickleEarly))
+            {
+                darkWakePostTickle = false;
+                reportUserInput();
+            }
+        }
+
+        if (!graphicsSuppressed && (changeFlags & kIOPMDomainDidChange))
+        {
+            if (logGraphicsClamp)
+            {
+                AbsoluteTime    now;
+                uint64_t        nsec;
+
+                clock_get_uptime(&now);
+                SUB_ABSOLUTETIME(&now, &systemWakeTime);
+                absolutetime_to_nanoseconds(now, &nsec);
+                MSG("Graphics suppressed %u ms\n",
+                    ((int)((nsec) / 1000000ULL)));
+            }
+            graphicsSuppressed = true;
+        }
+    }
+}
+
+void IOPMrootDomain::handleActivityTickleForDisplayWrangler(
+    IOService *     service,
+    IOPMActions *   actions )
+{
+    // Warning: Not running in PM work loop context - don't modify state !!!
+    // Trap tickle directed to IODisplayWrangler while running with graphics
+    // capability suppressed.
+
+    assert(service == wrangler);
+
+    if (service == wrangler)
+    {
+        bool aborting = ((lastSleepReason == kIOPMSleepReasonIdle) 
+                       || (lastSleepReason == kIOPMSleepReasonMaintenance));
+        if (aborting) {
+            userActivityCount++;
+            DLOG("display wrangler tickled1 %d lastSleepReason %d\n", userActivityCount, lastSleepReason);
+        }
+    }
+
+    if (!wranglerTickled &&
+        ((_pendingCapability & kIOPMSystemCapabilityGraphics) == 0))
+    {
+        setProperty(kIOPMRootDomainWakeTypeKey, kIOPMRootDomainWakeTypeHIDActivity);
+        DLOG("display wrangler tickled\n");
+        if (kIOLogPMRootDomain & gIOKitDebug)
+            OSReportWithBacktrace("Dark wake display tickle");
+        if (pmPowerStateQueue)
+        {
+            pmPowerStateQueue->submitPowerEvent(
+                kPowerEventPolicyStimulus,
+                (void *) kStimulusDarkWakeActivityTickle );
+        }
+    }
 }
 
 //******************************************************************************
 // Approve usage of delayed child notification by PM.
 //******************************************************************************
 
-bool
-IOPMrootDomain::shouldDelayChildNotification(
-	IOService * service )
-{
-	if ((kFullWakeReasonNone == fullWakeReason) &&
-	    (kSystemTransitionWake == _systemTransitionType)) {
-		DLOG("%s: delay child notify\n", service->getName());
-		return true;
-	}
-	return false;
+bool IOPMrootDomain::shouldDelayChildNotification(
+    IOService * service )
+{
+    if (((gDarkWakeFlags & kDarkWakeFlagHIDTickleMask) != 0) &&
+        !wranglerTickled &&
+        (kSystemTransitionWake == _systemTransitionType))
+    {
+        DLOG("%s: delay child notify\n", service->getName());
+        return true;
+    }
+    return false;
 }
 
 //******************************************************************************
 // PM actions for PCI device.
 //******************************************************************************
 
-void
-IOPMrootDomain::handlePowerChangeStartForPCIDevice(
-	IOService *             service,
-	IOPMActions *           actions,
-	const IOPMRequest *     request,
-	IOPMPowerStateIndex     powerState,
-	IOPMPowerChangeFlags *  inOutChangeFlags )
-{
-	pmTracer->tracePCIPowerChange(
-		PMTraceWorker::kPowerChangeStart,
-		service, *inOutChangeFlags,
-		(actions->flags & kPMActionsPCIBitNumberMask));
-}
-
-void
-IOPMrootDomain::handlePowerChangeDoneForPCIDevice(
-	IOService *             service,
-	IOPMActions *           actions,
-	const IOPMRequest *     request,
-	IOPMPowerStateIndex     powerState,
-	IOPMPowerChangeFlags    changeFlags )
-{
-	pmTracer->tracePCIPowerChange(
-		PMTraceWorker::kPowerChangeCompleted,
-		service, changeFlags,
-		(actions->flags & kPMActionsPCIBitNumberMask));
+void IOPMrootDomain::handlePowerChangeStartForPCIDevice(
+    IOService *     service,
+    IOPMActions *   actions, 
+    uint32_t        powerState,
+    uint32_t *      inOutChangeFlags )
+{
+    pmTracer->tracePCIPowerChange(
+        PMTraceWorker::kPowerChangeStart,
+        service, *inOutChangeFlags,
+        (actions->parameter & kPMActionsPCIBitNumberMask));
+}
+
+void IOPMrootDomain::handlePowerChangeDoneForPCIDevice(
+    IOService *     service,
+    IOPMActions *   actions, 
+    uint32_t        powerState,
+    uint32_t        changeFlags )
+{
+    pmTracer->tracePCIPowerChange(
+        PMTraceWorker::kPowerChangeCompleted,
+        service, changeFlags,
+        (actions->parameter & kPMActionsPCIBitNumberMask));
 }
 
 //******************************************************************************
 // registerInterest
 //
-// Override IOService::registerInterest() for root domain clients.
-//******************************************************************************
-
-class IOPMServiceInterestNotifier : public _IOServiceInterestNotifier
-{
-	friend class IOPMrootDomain;
-	OSDeclareDefaultStructors(IOPMServiceInterestNotifier);
-
-protected:
-	uint32_t        ackTimeoutCnt;
-	uint32_t        msgType;    // Last type seen by the message filter
-	uint32_t        lastSleepWakeMsgType;
-	uint32_t        msgIndex;
-	uint32_t        maxMsgDelayMS;
-	uint32_t        maxAckDelayMS;
-	uint64_t        msgAbsTime;
-	uint64_t        uuid0;
-	uint64_t        uuid1;
-	OSSharedPtr<const OSSymbol> identifier;
-	OSSharedPtr<const OSSymbol> clientName;
-};
-
-OSDefineMetaClassAndStructors(IOPMServiceInterestNotifier, _IOServiceInterestNotifier)
-
-OSSharedPtr<IONotifier>
-IOPMrootDomain::registerInterest(
-	const OSSymbol * typeOfInterest,
-	IOServiceInterestHandler handler,
-	void * target, void * ref )
-{
-	IOPMServiceInterestNotifier* notifier;
-	bool            isSystemCapabilityClient;
-	bool            isKernelCapabilityClient;
-	IOReturn        rc = kIOReturnError;
-
-	isSystemCapabilityClient = typeOfInterest &&
-	    typeOfInterest->isEqualTo(kIOPMSystemCapabilityInterest);
-
-	isKernelCapabilityClient = typeOfInterest &&
-	    typeOfInterest->isEqualTo(gIOPriorityPowerStateInterest);
-
-	if (isSystemCapabilityClient) {
-		typeOfInterest = gIOAppPowerStateInterest;
-	}
-
-	notifier = new IOPMServiceInterestNotifier;
-	if (!notifier) {
-		return NULL;
-	}
-
-	if (notifier->init()) {
-		rc  = super::registerInterestForNotifier(notifier, typeOfInterest, handler, target, ref);
-	}
-	if (rc != kIOReturnSuccess) {
-		OSSafeReleaseNULL(notifier);
-		return NULL;
-	}
-
-	notifier->ackTimeoutCnt = 0;
-
-	if (pmPowerStateQueue) {
-		if (isSystemCapabilityClient) {
-			notifier->retain();
-			if (pmPowerStateQueue->submitPowerEvent(
-				    kPowerEventRegisterSystemCapabilityClient, notifier) == false) {
-				notifier->release();
-			}
-		}
-
-		if (isKernelCapabilityClient) {
-			notifier->retain();
-			if (pmPowerStateQueue->submitPowerEvent(
-				    kPowerEventRegisterKernelCapabilityClient, notifier) == false) {
-				notifier->release();
-			}
-		}
-	}
-
-	OSSharedPtr<OSData> data;
-	uint8_t *uuid = NULL;
-	OSSharedPtr<OSKext> kext = OSKext::lookupKextWithAddress((vm_address_t)handler);
-	if (kext) {
-		data = kext->copyUUID();
-	}
-	if (data && (data->getLength() == sizeof(uuid_t))) {
-		uuid = (uint8_t *)(data->getBytesNoCopy());
-
-		notifier->uuid0 = ((uint64_t)(uuid[0]) << 56) | ((uint64_t)(uuid[1]) << 48) | ((uint64_t)(uuid[2]) << 40) |
-		    ((uint64_t)(uuid[3]) << 32) | ((uint64_t)(uuid[4]) << 24) | ((uint64_t)(uuid[5]) << 16) |
-		    ((uint64_t)(uuid[6]) << 8) | (uuid[7]);
-		notifier->uuid1 = ((uint64_t)(uuid[8]) << 56) | ((uint64_t)(uuid[9]) << 48) | ((uint64_t)(uuid[10]) << 40) |
-		    ((uint64_t)(uuid[11]) << 32) | ((uint64_t)(uuid[12]) << 24) | ((uint64_t)(uuid[13]) << 16) |
-		    ((uint64_t)(uuid[14]) << 8) | (uuid[15]);
-
-		notifier->identifier = copyKextIdentifierWithAddress((vm_address_t) handler);
-	}
-	return OSSharedPtr<IOPMServiceInterestNotifier>(notifier, OSNoRetain);
+// Override IOService::registerInterest() to intercept special clients.
+//******************************************************************************
+
+IONotifier * IOPMrootDomain::registerInterest(
+                const OSSymbol * typeOfInterest,
+                IOServiceInterestHandler handler,
+                void * target, void * ref )
+{
+    IONotifier *    notifier;
+    bool            isSystemCapabilityClient;
+    bool            isKernelCapabilityClient;
+
+    isSystemCapabilityClient =
+        typeOfInterest &&
+        typeOfInterest->isEqualTo(kIOPMSystemCapabilityInterest);
+
+    isKernelCapabilityClient =
+        typeOfInterest &&
+        typeOfInterest->isEqualTo(gIOPriorityPowerStateInterest);
+
+    if (isSystemCapabilityClient)
+        typeOfInterest = gIOAppPowerStateInterest;
+
+    notifier = super::registerInterest(typeOfInterest, handler, target, ref);
+    if (notifier && pmPowerStateQueue)
+    {
+        if (isSystemCapabilityClient)
+        {
+            notifier->retain();
+            if (pmPowerStateQueue->submitPowerEvent(
+                kPowerEventRegisterSystemCapabilityClient, notifier) == false)
+                notifier->release();
+        }
+
+        if (isKernelCapabilityClient)
+        {
+            notifier->retain();
+            if (pmPowerStateQueue->submitPowerEvent(
+                kPowerEventRegisterKernelCapabilityClient, notifier) == false)
+                notifier->release();
+        }
+    }
+
+    return notifier;
 }
 
 //******************************************************************************
@@ -7417,167 +4822,126 @@
 //
 //******************************************************************************
 
-bool
-IOPMrootDomain::systemMessageFilter(
-	void * object, void * arg1, void * arg2, void * arg3 )
-{
-	const IOPMInterestContext * context = (const IOPMInterestContext *) arg1;
-	bool  isCapMsg = (context->messageType == kIOMessageSystemCapabilityChange);
-	bool  isCapPowerd = (object == (void *) systemCapabilityNotifier.get());
-	bool  isCapClient = false;
-	bool  allow = false;
-	OSBoolean **waitForReply = (typeof(waitForReply))arg3;
-	IOPMServiceInterestNotifier *notifier;
-
-	notifier = OSDynamicCast(IOPMServiceInterestNotifier, (OSObject *)object);
-
-	do {
-		// When powerd and kernel priority clients register capability interest,
-		// the power tree is sync'ed to inform those clients about the current
-		// system capability. Only allow capability change messages during sync.
-		if ((kSystemTransitionNewCapClient == _systemTransitionType) &&
-		    (!isCapMsg || !_joinedCapabilityClients ||
-		    !_joinedCapabilityClients->containsObject((OSObject *) object))) {
-			break;
-		}
-
-		// Capability change message for powerd and kernel clients
-		if (isCapMsg) {
-			// Kernel priority clients
-			if ((context->notifyType == kNotifyPriority) ||
-			    (context->notifyType == kNotifyCapabilityChangePriority)) {
-				isCapClient = true;
-			}
-
-			// powerd will maintain two client registrations with root domain.
-			// isCapPowerd will be TRUE for any message targeting the powerd
-			// exclusive (capability change) interest registration.
-			if (isCapPowerd && (context->notifyType == kNotifyCapabilityChangeApps)) {
-				isCapClient = true;
-			}
-		}
-
-		if (isCapClient) {
-			IOPMSystemCapabilityChangeParameters * capArgs =
-			    (IOPMSystemCapabilityChangeParameters *) arg2;
-
-			if (kSystemTransitionNewCapClient == _systemTransitionType) {
-				capArgs->fromCapabilities = 0;
-				capArgs->toCapabilities = _currentCapability;
-				capArgs->changeFlags = 0;
-			} else {
-				capArgs->fromCapabilities = _currentCapability;
-				capArgs->toCapabilities = _pendingCapability;
-
-				if (context->isPreChange) {
-					capArgs->changeFlags = kIOPMSystemCapabilityWillChange;
-				} else {
-					capArgs->changeFlags = kIOPMSystemCapabilityDidChange;
-				}
-
-				if (isCapPowerd && context->isPreChange) {
-					toldPowerdCapWillChange = true;
-				}
-			}
-
-			// App level capability change messages must only go to powerd.
-			// Wait for response post-change if capabilitiy is increasing.
-			// Wait for response pre-change if capability is decreasing.
-
-			if ((context->notifyType == kNotifyCapabilityChangeApps) && waitForReply &&
-			    ((capabilityLoss && context->isPreChange) ||
-			    (!capabilityLoss && !context->isPreChange))) {
-				*waitForReply = kOSBooleanTrue;
-			}
-
-			allow = true;
-			break;
-		}
-
-		// powerd will always receive CanSystemSleep, even for a demand sleep.
-		// It will also have a final chance to veto sleep after all clients
-		// have responded to SystemWillSleep
-
-		if ((kIOMessageCanSystemSleep == context->messageType) ||
-		    (kIOMessageSystemWillNotSleep == context->messageType)) {
-			if (isCapPowerd) {
-				allow = true;
-				break;
-			}
-
-			// Demand sleep, don't ask apps for permission
-			if (context->changeFlags & kIOPMSkipAskPowerDown) {
-				break;
-			}
-		}
-
-		if (kIOPMMessageLastCallBeforeSleep == context->messageType) {
-			if (isCapPowerd && CAP_HIGHEST(kIOPMSystemCapabilityGraphics) &&
-			    (fullToDarkReason == kIOPMSleepReasonIdle)) {
-				allow = true;
-			}
-			break;
-		}
-
-		// Drop capability change messages for legacy clients.
-		// Drop legacy system sleep messages for powerd capability interest.
-		if (isCapMsg || isCapPowerd) {
-			break;
-		}
-
-		// Not a capability change message.
-		// Perform message filtering based on _systemMessageClientMask.
-
-		if ((context->notifyType == kNotifyApps) &&
-		    (_systemMessageClientMask & kSystemMessageClientLegacyApp)) {
-			if (!notifier) {
-				break;
-			}
-
-			if ((notifier->lastSleepWakeMsgType == context->messageType) &&
-			    (notifier->lastSleepWakeMsgType == kIOMessageSystemWillPowerOn)) {
-				break; // drop any duplicate WillPowerOn for AOT devices
-			}
-
-			allow = true;
-
-			if (waitForReply) {
-				if (notifier->ackTimeoutCnt >= 3) {
-					*waitForReply = kOSBooleanFalse;
-				} else {
-					*waitForReply = kOSBooleanTrue;
-				}
-			}
-		} else if ((context->notifyType == kNotifyPriority) &&
-		    (_systemMessageClientMask & kSystemMessageClientKernel)) {
-			allow = true;
-		}
-
-		// Check sleep/wake message ordering
-		if (allow) {
-			if (context->messageType == kIOMessageSystemWillSleep ||
-			    context->messageType == kIOMessageSystemWillPowerOn ||
-			    context->messageType == kIOMessageSystemHasPoweredOn) {
-				notifier->lastSleepWakeMsgType = context->messageType;
-			}
-		}
-	} while (false);
-
-	if (allow && isCapMsg && _joinedCapabilityClients) {
-		_joinedCapabilityClients->removeObject((OSObject *) object);
-		if (_joinedCapabilityClients->getCount() == 0) {
-			DMSG("destroyed capability client set %p\n",
-			    OBFUSCATE(_joinedCapabilityClients.get()));
-			_joinedCapabilityClients.reset();
-		}
-	}
-	if (notifier) {
-		// Record the last seen message type even if the message is dropped
-		// for traceFilteredNotification().
-		notifier->msgType = context->messageType;
-	}
-
-	return allow;
+bool IOPMrootDomain::systemMessageFilter(
+    void * object, void * arg1, void * arg2, void * arg3 )
+{
+    const IOPMInterestContext * context = (const IOPMInterestContext *) arg1;
+    bool  isCapMsg = (context->messageType == kIOMessageSystemCapabilityChange);
+    bool  isCapClient = false;
+    bool  allow = false;
+
+    do {
+        if ((kSystemTransitionNewCapClient == _systemTransitionType) &&
+            (!isCapMsg || !_joinedCapabilityClients ||
+             !_joinedCapabilityClients->containsObject((OSObject *) object)))
+            break;
+
+        // Capability change message for app and kernel clients.
+
+        if (isCapMsg)
+        {
+            if ((context->notifyType == kNotifyPriority) ||
+                (context->notifyType == kNotifyCapabilityChangePriority))
+                isCapClient = true;
+
+            if ((context->notifyType == kNotifyCapabilityChangeApps) &&
+                (object == (void *) systemCapabilityNotifier))
+                isCapClient = true;
+        }
+
+        if (isCapClient)
+        {
+            IOPMSystemCapabilityChangeParameters * capArgs =
+                (IOPMSystemCapabilityChangeParameters *) arg2;
+
+            if (kSystemTransitionNewCapClient == _systemTransitionType)
+            {
+                capArgs->fromCapabilities = 0;
+                capArgs->toCapabilities = _currentCapability;
+                capArgs->changeFlags = 0;
+            }
+            else
+            {
+                capArgs->fromCapabilities = _currentCapability;
+                capArgs->toCapabilities = _pendingCapability;
+
+                if (context->isPreChange)
+                    capArgs->changeFlags = kIOPMSystemCapabilityWillChange;
+                else
+                    capArgs->changeFlags = kIOPMSystemCapabilityDidChange;
+            }
+
+            // Capability change messages only go to the PM configd plugin. 
+            // Wait for response post-change if capabilitiy is increasing.
+            // Wait for response pre-change if capability is decreasing.
+
+            if ((context->notifyType == kNotifyCapabilityChangeApps) && arg3 &&
+                ( (capabilityLoss && context->isPreChange) ||
+                  (!capabilityLoss && !context->isPreChange) ) )
+            {
+                // app has not replied yet, wait for it
+                *((OSObject **) arg3) = kOSBooleanFalse;
+            }
+
+            allow = true;
+            break;
+        }
+
+        // Capability client will always see kIOMessageCanSystemSleep,
+        // even for demand sleep.
+
+        if ((kIOMessageCanSystemSleep == context->messageType) ||
+            (kIOMessageSystemWillNotSleep == context->messageType))
+        {
+            if (object == (OSObject *) systemCapabilityNotifier)
+            {
+                allow = true;
+                break;
+            }
+            
+            // Not idle sleep, don't ask apps.
+            if (context->changeFlags & kIOPMSkipAskPowerDown)
+            {
+                break;
+            }
+        }
+
+        // Reject capability change messages for legacy clients.
+        // Reject legacy system sleep messages for capability client.
+
+        if (isCapMsg || (object == (OSObject *) systemCapabilityNotifier))
+        {
+            break;
+        }
+
+        // Filter system sleep messages.
+
+        if ((context->notifyType == kNotifyApps) &&
+            (_systemMessageClientMask & kSystemMessageClientApp))
+        {
+            allow = true;
+        }
+        else if ((context->notifyType == kNotifyPriority) &&
+                 (_systemMessageClientMask & kSystemMessageClientKernel))
+        {
+            allow = true;
+        }
+    }
+    while (false);
+
+    if (allow && isCapMsg && _joinedCapabilityClients)
+    {
+        _joinedCapabilityClients->removeObject((OSObject *) object);
+        if (_joinedCapabilityClients->getCount() == 0)
+        {
+            DLOG("destroyed capability client set %p\n",
+                _joinedCapabilityClients);
+            _joinedCapabilityClients->release();
+            _joinedCapabilityClients = 0;
+        }
+    }
+
+    return allow;
 }
 
 //******************************************************************************
@@ -7585,29 +4949,30 @@
 //
 //******************************************************************************
 
-IOReturn
-IOPMrootDomain::setMaintenanceWakeCalendar(
-	const IOPMCalendarStruct * calendar )
-{
-	OSSharedPtr<OSData> data;
-	IOReturn ret = 0;
-
-	if (!calendar) {
-		return kIOReturnBadArgument;
-	}
-
-	data = OSData::withValue(*calendar);
-	if (!data) {
-		return kIOReturnNoMemory;
-	}
-
-	if (kPMCalendarTypeMaintenance == calendar->selector) {
-		ret = setPMSetting(gIOPMSettingMaintenanceWakeCalendarKey.get(), data.get());
-	} else if (kPMCalendarTypeSleepService == calendar->selector) {
-		ret = setPMSetting(gIOPMSettingSleepServiceWakeCalendarKey.get(), data.get());
-	}
-
-	return ret;
+IOReturn IOPMrootDomain::setMaintenanceWakeCalendar(
+    const IOPMCalendarStruct * calendar )
+{
+    OSData * data;
+    IOReturn ret;
+
+    if (!calendar)
+        return kIOReturnBadArgument;
+    
+    data = OSData::withBytesNoCopy((void *) calendar, sizeof(*calendar));
+    if (!data)
+        return kIOReturnNoMemory;
+
+    if (kPMCalendarTypeMaintenance == calendar->selector) {
+        ret = setPMSetting(gIOPMSettingMaintenanceWakeCalendarKey, data);
+    } else 
+    if (kPMCalendarTypeSleepService == calendar->selector)
+    {
+        ret = setPMSetting(gIOPMSettingSleepServiceWakeCalendarKey, data);
+    }
+    
+
+    data->release();
+    return ret;
 }
 
 // MARK: -
@@ -7619,64 +4984,83 @@
 // Handle the notification when the IODisplayWrangler changes power state.
 //******************************************************************************
 
-IOReturn
-IOPMrootDomain::displayWranglerNotification(
-	void * target, void * refCon,
-	UInt32 messageType, IOService * service,
-	void * messageArgument, vm_size_t argSize )
-{
-#if DISPLAY_WRANGLER_PRESENT
-	IOPMPowerStateIndex                 displayPowerState;
-	IOPowerStateChangeNotification *    params =
-	    (IOPowerStateChangeNotification *) messageArgument;
-
-	if ((messageType != kIOMessageDeviceWillPowerOff) &&
-	    (messageType != kIOMessageDeviceHasPoweredOn)) {
-		return kIOReturnUnsupported;
-	}
-
-	ASSERT_GATED();
-	if (!gRootDomain) {
-		return kIOReturnUnsupported;
-	}
-
-	displayPowerState = params->stateNumber;
-	DLOG("wrangler %s ps %d\n",
-	    getIOMessageString(messageType), (uint32_t) displayPowerState);
-
-	switch (messageType) {
-	case kIOMessageDeviceWillPowerOff:
-		// Display wrangler has dropped power due to display idle
-		// or force system sleep.
-		//
-		// 4 Display ON             kWranglerPowerStateMax
-		// 3 Display Dim            kWranglerPowerStateDim
-		// 2 Display Sleep          kWranglerPowerStateSleep
-		// 1 Not visible to user
-		// 0 Not visible to user    kWranglerPowerStateMin
-
-		if (displayPowerState <= kWranglerPowerStateSleep) {
-			gRootDomain->evaluatePolicy( kStimulusDisplayWranglerSleep );
-		}
-		break;
-
-	case kIOMessageDeviceHasPoweredOn:
-		// Display wrangler has powered on due to user activity
-		// or wake from sleep.
-
-		if (kWranglerPowerStateMax == displayPowerState) {
-			gRootDomain->evaluatePolicy( kStimulusDisplayWranglerWake );
-
-			// See comment in handleUpdatePowerClientForDisplayWrangler
-			if (service->getPowerStateForClient(gIOPMPowerClientDevice) ==
-			    kWranglerPowerStateMax) {
-				gRootDomain->evaluatePolicy( kStimulusEnterUserActiveState );
-			}
-		}
-		break;
-	}
-#endif /* DISPLAY_WRANGLER_PRESENT */
-	return kIOReturnUnsupported;
+IOReturn IOPMrootDomain::displayWranglerNotification(
+    void * target, void * refCon,
+    UInt32 messageType, IOService * service,
+    void * messageArgument, vm_size_t argSize )
+{
+#if !NO_KERNEL_HID
+    int                                 displayPowerState;
+    IOPowerStateChangeNotification *    params =
+            (IOPowerStateChangeNotification *) messageArgument;
+
+    if ((messageType != kIOMessageDeviceWillPowerOff) &&
+        (messageType != kIOMessageDeviceHasPoweredOn))
+        return kIOReturnUnsupported;
+
+    ASSERT_GATED();
+    if (!gRootDomain)
+        return kIOReturnUnsupported;
+
+    displayPowerState = params->stateNumber;
+    DLOG("DisplayWrangler message 0x%x, power state %d\n",
+              (uint32_t) messageType, displayPowerState);
+
+    switch (messageType) {
+       case kIOMessageDeviceWillPowerOff:
+
+            // Display wrangler has dropped power due to display idle
+            // or force system sleep.
+            //
+            // 4 Display ON
+            // 3 Display Dim
+            // 2 Display Sleep
+            // 1 Not visible to user
+            // 0 Not visible to user
+
+            if (displayPowerState > 2)
+                break;
+
+            gRootDomain->evaluatePolicy( kStimulusDisplayWranglerSleep );
+            break;
+
+        case kIOMessageDeviceHasPoweredOn:
+
+            // Display wrangler has powered on due to user activity 
+            // or wake from sleep.
+
+            if ( 4 != displayPowerState )
+                break;
+
+            gRootDomain->evaluatePolicy( kStimulusDisplayWranglerWake );
+            break;
+    }
+#endif
+    return kIOReturnUnsupported;
+}
+
+//******************************************************************************
+// displayWranglerMatchPublished
+//
+// Receives a notification when the IODisplayWrangler is published.
+// When it's published we install a power state change handler.
+//******************************************************************************
+
+bool IOPMrootDomain::displayWranglerMatchPublished( 
+    void * target, 
+    void * refCon,
+    IOService * newService,
+    IONotifier * notifier __unused)
+{
+#if !NO_KERNEL_HID
+    // found the display wrangler, now install a handler
+    if( !newService->registerInterest( gIOGeneralInterest, 
+                            &displayWranglerNotification, target, 0) ) 
+    {
+        return false;
+    }
+#endif
+    return true;
 }
 
 //******************************************************************************
@@ -7684,698 +5068,205 @@
 //
 //******************************************************************************
 
-void
-IOPMrootDomain::updateUserActivity( void )
-{
-#if defined(XNU_TARGET_OS_OSX) && !DISPLAY_WRANGLER_PRESENT
-	clock_get_uptime(&userActivityTime);
-	bool aborting =  ((lastSleepReason == kIOPMSleepReasonSoftware)
-	    || (lastSleepReason == kIOPMSleepReasonIdle)
-	    || (lastSleepReason == kIOPMSleepReasonMaintenance));
-	if (aborting) {
-		userActivityCount++;
-		DLOG("user activity reported %d lastSleepReason %d\n", userActivityCount, lastSleepReason);
-	}
+void IOPMrootDomain::reportUserInput( void )
+{
+#if !NO_KERNEL_HID
+    OSIterator * iter;
+
+    if(!wrangler) 
+    {
+        iter = getMatchingServices(serviceMatching("IODisplayWrangler"));
+        if(iter) 
+        {
+            wrangler = (IOService *) iter->getNextObject();
+            iter->release();
+        }
+    }
+
+    if(wrangler)
+        wrangler->activityTickle(0,0);
 #endif
 }
-void
-IOPMrootDomain::reportUserInput( void )
-{
-	if (wrangler) {
-		wrangler->activityTickle(0, 0);
-	}
-#if defined(XNU_TARGET_OS_OSX) && !DISPLAY_WRANGLER_PRESENT
-	// Update user activity
-	updateUserActivity();
-
-	if (!darkWakeExit && ((_pendingCapability & kIOPMSystemCapabilityGraphics) == 0)) {
-		// update user active abs time
-		clock_get_uptime(&gUserActiveAbsTime);
-		pmPowerStateQueue->submitPowerEvent(
-			kPowerEventPolicyStimulus,
-			(void *) kStimulusDarkWakeActivityTickle,
-			true /* set wake type */ );
-	}
+
+//******************************************************************************
+// blockDisplayWranglerTickle
+//******************************************************************************
+
+bool IOPMrootDomain::latchDisplayWranglerTickle( bool latch )
+{
+#if !NO_KERNEL_HID
+    if (latch)
+    {
+        // Not too late to prevent the display from lighting up
+        if (!(_currentCapability & kIOPMSystemCapabilityGraphics) &&
+            !(_pendingCapability & kIOPMSystemCapabilityGraphics) &&
+            !checkSystemCanSustainFullWake())
+        {
+            wranglerTickleLatched = true;
+        }
+        else
+        {
+            wranglerTickleLatched = false;
+        }
+    }
+    else if (wranglerTickleLatched && checkSystemCanSustainFullWake())
+    {
+        wranglerTickleLatched = false;
+
+        pmPowerStateQueue->submitPowerEvent(
+            kPowerEventPolicyStimulus,
+            (void *) kStimulusDarkWakeActivityTickle );
+    }
+
+    return wranglerTickleLatched;
+#else
+    return false;
 #endif
 }
 
-void
-IOPMrootDomain::requestUserActive(IOService *device, const char *reason)
-{
-#if DISPLAY_WRANGLER_PRESENT
-	if (wrangler) {
-		wrangler->activityTickle(0, 0);
-	}
-#else
-	if (!device) {
-		DLOG("requestUserActive: device is null\n");
-		return;
-	}
-	OSSharedPtr<const OSSymbol> deviceName = device->copyName();
-	uint64_t registryID = device->getRegistryEntryID();
-
-	if (!deviceName || !registryID) {
-		DLOG("requestUserActive: no device name or registry entry\n");
-		return;
-	}
-	const char *name = deviceName->getCStringNoCopy();
-	char payload[128];
-	snprintf(payload, sizeof(payload), "%s:%s", name, reason);
-	DLOG("requestUserActive from %s (0x%llx) for %s\n", name, registryID, reason);
-	messageClient(kIOPMMessageRequestUserActive, systemCapabilityNotifier.get(), (void *)payload, sizeof(payload));
-#endif
-}
-
-//******************************************************************************
-// latchDisplayWranglerTickle
-//******************************************************************************
-
-bool
-IOPMrootDomain::latchDisplayWranglerTickle( bool latch )
-{
-#if DISPLAY_WRANGLER_PRESENT
-	if (latch) {
-		if (!(_currentCapability & kIOPMSystemCapabilityGraphics) &&
-		    !(_pendingCapability & kIOPMSystemCapabilityGraphics) &&
-		    !checkSystemCanSustainFullWake()) {
-			// Currently in dark wake, and not transitioning to full wake.
-			// Full wake is unsustainable, so latch the tickle to prevent
-			// the display from lighting up momentarily.
-			wranglerTickled = true;
-		} else {
-			wranglerTickled = false;
-		}
-	} else if (wranglerTickled && checkSystemCanSustainFullWake()) {
-		wranglerTickled = false;
-
-		pmPowerStateQueue->submitPowerEvent(
-			kPowerEventPolicyStimulus,
-			(void *) kStimulusDarkWakeActivityTickle );
-	}
-
-	return wranglerTickled;
-#else  /* ! DISPLAY_WRANGLER_PRESENT */
-	return false;
-#endif /* ! DISPLAY_WRANGLER_PRESENT */
-}
-
-//******************************************************************************
-// setDisplayPowerOn
+// MARK: -
+// MARK: Battery
+
+//******************************************************************************
+// batteryPublished
 //
-// For root domain user client
-//******************************************************************************
-
-void
-IOPMrootDomain::setDisplayPowerOn( uint32_t options )
-{
-	pmPowerStateQueue->submitPowerEvent( kPowerEventSetDisplayPowerOn,
-	    (void *) NULL, options );
+// Notification on battery class IOPowerSource appearance
+//******************************************************************************
+
+bool IOPMrootDomain::batteryPublished( 
+    void * target, 
+    void * root_domain,
+    IOService * resourceService,
+    IONotifier * notifier __unused )
+{    
+    // rdar://2936060&4435589    
+    // All laptops have dimmable LCD displays
+    // All laptops have batteries
+    // So if this machine has a battery, publish the fact that the backlight
+    // supports dimming.
+    ((IOPMrootDomain *)root_domain)->publishFeature("DisplayDims");
+
+    return (true);
 }
 
 // MARK: -
 // MARK: System PM Policy
 
 //******************************************************************************
-// checkSystemSleepAllowed
+// checkSystemCanSleep
 //
 //******************************************************************************
 
-bool
-IOPMrootDomain::checkSystemSleepAllowed( IOOptionBits options,
-    uint32_t     sleepReason )
-{
-	uint32_t err = 0;
-
-	// Conditions that prevent idle and demand system sleep.
-
-	do {
-		if (gSleepDisabledFlag) {
-			err = kPMConfigPreventSystemSleep;
-			break;
-		}
-
-		if (userDisabledAllSleep) {
-			err = kPMUserDisabledAllSleep; // 1. user-space sleep kill switch
-			break;
-		}
-
-		if (systemBooting || systemShutdown || gWillShutdown) {
-			err = kPMSystemRestartBootingInProgress; // 2. restart or shutdown in progress
-			break;
-		}
-
-		if (options == 0) {
-			break;
-		}
-
-		// Conditions above pegs the system at full wake.
-		// Conditions below prevent system sleep but does not prevent
-		// dark wake, and must be called from gated context.
+bool IOPMrootDomain::checkSystemCanSleep( IOOptionBits options )
+{
+    int err = 0;
+
+    // Conditions that prevent idle and demand system sleep.
+
+    do {
+        if (userDisabledAllSleep)
+        {
+            err = 1;        // 1. user-space sleep kill switch
+            break;
+        }
+
+        if (systemBooting || systemShutdown)
+        {
+            err = 2;        // 2. restart or shutdown in progress
+            break;
+        }
+
+        if (options == 0)
+            break;
+
+        // Conditions above pegs the system at full wake.
+        // Conditions below prevent system sleep but does not prevent
+        // dark wake, and must be called from gated context.
 
 #if !CONFIG_SLEEP
-		err = kPMConfigPreventSystemSleep;    // 3. config does not support sleep
-		break;
+        err = 3;            // 3. config does not support sleep
+        break;
 #endif
 
-		if (_driverKitMatchingAssertionCount != 0 || _driverKitSyncedAssertionCount != 0) {
-			err = kPMCPUAssertion;
-			break;
-		}
-
-		// Check for any dexts currently being added to the PM tree. Sleeping while
-		// this is in flight can cause IOServicePH to timeout.
-		if (!IOServicePH::checkPMReady()) {
-#if !defined(XNU_TARGET_OS_OSX)
-			if (!(lowBatteryCondition || thermalWarningState || thermalEmergencyState)) {
-				// 116893363: kPMDKNotReady sleep cancellations often leaves embedded devices
-				// in dark wake for long periods of time, which causes issues as apps were
-				// already informed of sleep during the f->9 transition. As a temporary
-				// measure, always full wake if we hit this specific condition.
-				pmPowerStateQueue->submitPowerEvent(
-					kPowerEventPolicyStimulus,
-					(void *) kStimulusDarkWakeActivityTickle);
-			}
+        if (lowBatteryCondition)
+        {
+            break;          // always sleep on low battery
+        }
+
+        if(darkWakeThermalEmergency)
+        {
+            break;          // always sleep on dark wake thermal emergencies
+        }
+
+        if (preventSystemSleepList->getCount() != 0)
+        {
+            err = 4;        // 4. child prevent system sleep clamp
+            break;
+        }
+
+        if (getPMAssertionLevel( kIOPMDriverAssertionCPUBit ) ==
+            kIOPMDriverAssertionLevelOn)
+        {
+            err = 5;        // 5. CPU assertion
+            break;
+        }
+
+        if (pciCantSleepValid)
+        {
+            if (pciCantSleepFlag)
+                err = 6;    // 6. PCI card does not support PM (cached)
+            break;
+        }
+        else if (sleepSupportedPEFunction &&
+                 CAP_HIGHEST(kIOPMSystemCapabilityGraphics))
+        {            
+            IOReturn ret;
+            OSBitAndAtomic(~kPCICantSleep, &platformSleepSupport);
+            ret = getPlatform()->callPlatformFunction(
+                                    sleepSupportedPEFunction, false,
+                                    NULL, NULL, NULL, NULL);
+            pciCantSleepValid = true;
+            pciCantSleepFlag  = false;
+            if ((platformSleepSupport & kPCICantSleep) ||
+                ((ret != kIOReturnSuccess) && (ret != kIOReturnUnsupported)))
+            {
+                err = 6;    // 6. PCI card does not support PM
+                pciCantSleepFlag = true;
+                break;
+            }
+        }
+    }
+    while (false);
+
+    if (err)
+    {
+        DLOG("System sleep prevented by %d\n", err);
+        return false;
+    }
+    return true;
+}
+
+//******************************************************************************
+// checkSystemCanSustainFullWake
+//******************************************************************************
+
+bool IOPMrootDomain::checkSystemCanSustainFullWake( void )
+{
+#if !NO_KERNEL_HID
+    if (lowBatteryCondition)
+    {
+        // Low battery wake, or received a low battery notification
+        // while system is awake.
+        return false;
+    }
+
+    if (clamshellExists && clamshellClosed && !acAdaptorConnected)
+    {
+        // Lid closed on battery power
+        return false;
+    }
 #endif
-			err = kPMDKNotReady;
-			break;
-		}
-
-		if (lowBatteryCondition || thermalWarningState || thermalEmergencyState) {
-			break; // always sleep on low battery or when in thermal warning/emergency state
-		}
-
-		if (sleepReason == kIOPMSleepReasonDarkWakeThermalEmergency) {
-			break; // always sleep on dark wake thermal emergencies
-		}
-
-		if (preventSystemSleepList->getCount() != 0) {
-			err = kPMChildPreventSystemSleep; // 4. child prevent system sleep clamp
-			break;
-		}
-
-
-		if (getPMAssertionLevel( kIOPMDriverAssertionCPUBit ) ==
-		    kIOPMDriverAssertionLevelOn) {
-			err = kPMCPUAssertion; // 5. CPU assertion
-			break;
-		}
-
-		if (pciCantSleepValid) {
-			if (pciCantSleepFlag) {
-				err = kPMPCIUnsupported; // 6. PCI card does not support PM (cached)
-			}
-			break;
-		} else if (sleepSupportedPEFunction &&
-		    CAP_HIGHEST(kIOPMSystemCapabilityGraphics)) {
-			IOReturn ret;
-			OSBitAndAtomic(~kPCICantSleep, &platformSleepSupport);
-			ret = getPlatform()->callPlatformFunction(
-				sleepSupportedPEFunction.get(), false,
-				NULL, NULL, NULL, NULL);
-			pciCantSleepValid = true;
-			pciCantSleepFlag  = false;
-			if ((platformSleepSupport & kPCICantSleep) ||
-			    ((ret != kIOReturnSuccess) && (ret != kIOReturnUnsupported))) {
-				err = 6; // 6. PCI card does not support PM
-				pciCantSleepFlag = true;
-				break;
-			}
-		}
-	}while (false);
-
-	if (err) {
-		DLOG("System sleep prevented by %s\n", getSystemSleepPreventerString(err));
-		return false;
-	}
-	return true;
-}
-
-bool
-IOPMrootDomain::checkSystemSleepEnabled( void )
-{
-	return checkSystemSleepAllowed(0, 0);
-}
-
-bool
-IOPMrootDomain::checkSystemCanSleep( uint32_t sleepReason )
-{
-	ASSERT_GATED();
-	return checkSystemSleepAllowed(1, sleepReason);
-}
-
-//******************************************************************************
-// checkSystemCanSustainFullWake
-//******************************************************************************
-
-bool
-IOPMrootDomain::checkSystemCanSustainFullWake( void )
-{
-	if (lowBatteryCondition || thermalWarningState || thermalEmergencyState) {
-		// Low battery wake, or received a low battery notification
-		// while system is awake. This condition will persist until
-		// the following wake.
-		return false;
-	}
-
-	if (clamshellExists && clamshellClosed && !clamshellSleepDisableMask) {
-		// Graphics state is unknown and external display might not be probed.
-		// Do not incorporate state that requires graphics to be in max power
-		// such as desktopMode or clamshellDisabled.
-
-		if (!acAdaptorConnected) {
-			DLOG("full wake check: no AC\n");
-			return false;
-		}
-	}
-	return true;
-}
-
-//******************************************************************************
-// checkSystemCanAbortIdleSleep
-//******************************************************************************
-
-bool
-IOPMrootDomain::checkSystemCanAbortIdleSleep( void )
-{
-	bool abortableSleepType =  ((lastSleepReason == kIOPMSleepReasonIdle)
-	    || (lastSleepReason == 0));
-	return idleSleepRevertible && abortableSleepType;
-}
-
-//******************************************************************************
-// considerRunMode
-// consider the driver for AOT power on via the runmode mask
-//******************************************************************************
-
-int32_t
-IOPMrootDomain::considerRunMode(IOService * service, uint64_t pmDriverClass)
-{
-	int32_t promote;
-
-	if ((0 == _aotRunMode) || (service == this)) {
-		// neutral
-		return 0;
-	}
-	if (pmDriverClass) {
-		IOLog("considerRunMode: %s 0x%llx 0x%llx\n", service->getName(), pmDriverClass, _aotRunMode);
-	}
-	promote = (0 != (_aotRunMode & pmDriverClass)) ? 1 : -1;
-	if (promote > 0) {
-		IOLog("IOPMRD: %s 0x%llx runmode to %s\n", service->getName(), pmDriverClass, (promote < 0) ? "OFF" : "ON");
-	}
-	return promote;
-}
-
-void
-IOPMrootDomain::handleRegisterPowerDriver(IOService * child)
-{
-	OSNumber * num;
-	IOService * userServer;
-	uint64_t driverClassFlags;
-	OSSharedPtr<OSObject> prop = child->copyProperty(kIOPMAOTAllowKey);
-
-	if (!prop || (NULL == (num = OSDynamicCast(OSNumber, prop.get())))) {
-		return;
-	}
-	driverClassFlags = num->unsigned64BitValue();
-
-	userServer = NULL;
-	if (child->reserved->uvars && (userServer = child->reserved->uvars->userServer)) {
-		WAKEEVENT_LOCK();
-		_aotWakeEventRunModeImpliesStorage |= driverClassFlags;
-		WAKEEVENT_UNLOCK();
-	}
-
-	IOLog("addPMDriverClass %s %llx\n", child->getName(), driverClassFlags);
-	if (driverClassFlags) {
-		child->addPMDriverClass(driverClassFlags);
-		if (userServer) {
-			userServer->addPMDriverClass(driverClassFlags);
-		}
-	}
-}
-
-//******************************************************************************
-// attemptIdleSleepAbort
-//******************************************************************************
-
-bool
-IOPMrootDomain::attemptIdleSleepAbort( void )
-{
-	if (!gIOPMWorkLoop->inGate()) {
-		bool ret = gIOPMWorkLoop->runAction(
-			OSMemberFunctionCast(IOWorkLoop::Action, this,
-			&IOPMrootDomain::attemptIdleSleepAbort),
-			this);
-		return ret;
-	}
-
-	bool canAbort = checkSystemCanAbortIdleSleep();
-	if (canAbort) {
-		cancelIdlePowerDownSync();
-	} else if (lastSleepReason == kIOPMSleepReasonIdle) {
-		scheduleImmediateDebugWake();
-	}
-
-	return canAbort;
-}
-
-//******************************************************************************
-// setIdleSleepRevertible
-//******************************************************************************
-
-void
-IOPMrootDomain::setIdleSleepRevertible( bool revertible )
-{
-	idleSleepRevertible = revertible;
-}
-
-//******************************************************************************
-// mustHibernate
-//******************************************************************************
-
-#if HIBERNATION
-
-bool
-IOPMrootDomain::mustHibernate( void )
-{
-	return lowBatteryCondition || thermalWarningState;
-}
-
-#endif /* HIBERNATION */
-
-//******************************************************************************
-// AOT
-//******************************************************************************
-
-// Tables for accumulated days in year by month, latter used for leap years
-
-static const unsigned int daysbymonth[] =
-{ 0, 31, 59, 90, 120, 151, 181, 212, 243, 273, 304, 334, 365 };
-
-static const unsigned int lydaysbymonth[] =
-{ 0, 31, 60, 91, 121, 152, 182, 213, 244, 274, 305, 335, 366 };
-
-static int __unused
-IOPMConvertSecondsToCalendar(clock_sec_t secs, IOPMCalendarStruct * dt)
-{
-	const unsigned int *    dbm = daysbymonth;
-	clock_sec_t             n, x, y, z;
-
-	// Calculate seconds, minutes and hours
-
-	n = secs % (24 * 3600);
-	dt->second = n % 60;
-	n /= 60;
-	dt->minute = n % 60;
-	dt->hour = (typeof(dt->hour))(n / 60);
-
-	// Calculate day of week
-
-	n = secs / (24 * 3600);
-//	dt->dayWeek = (n + 4) % 7;
-
-	// Calculate year
-	// Rebase from days since Unix epoch (1/1/1970) store in 'n',
-	// to days since 1/1/1968 to start on 4 year cycle, beginning
-	// on a leap year.
-
-	n += (366 + 365);
-
-	// Every 4 year cycle will be exactly (366 + 365 * 3) = 1461 days.
-	// Valid before 2100, since 2100 is not a leap year.
-
-	x = n / 1461;       // number of 4 year cycles
-	y = n % 1461;       // days into current 4 year cycle
-	z = 1968 + (4 * x);
-
-	// Add in years in the current 4 year cycle
-
-	if (y >= 366) {
-		y -= 366;   // days after the leap year
-		n = y % 365; // days into the current year
-		z += (1 + y / 365); // years after the past 4-yr cycle
-	} else {
-		n = y;
-		dbm = lydaysbymonth;
-	}
-	if (z > 2099) {
-		return 0;
-	}
-
-	dt->year = (typeof(dt->year))z;
-
-	// Adjust remaining days value to start at 1
-
-	n += 1;
-
-	// Calculate month
-
-	for (x = 1; (n > dbm[x]) && (x < 12); x++) {
-		continue;
-	}
-	dt->month = (typeof(dt->month))x;
-
-	// Calculate day of month
-
-	dt->day = (typeof(dt->day))(n - dbm[x - 1]);
-
-	return 1;
-}
-
-static clock_sec_t
-IOPMConvertCalendarToSeconds(const IOPMCalendarStruct * dt)
-{
-	const unsigned int *    dbm = daysbymonth;
-	long                    y, secs, days;
-
-	if (dt->year < 1970 || dt->month > 12) {
-		return 0;
-	}
-
-	// Seconds elapsed in the current day
-
-	secs = dt->second + 60 * dt->minute + 3600 * dt->hour;
-
-	// Number of days from 1/1/70 to beginning of current year
-	// Account for extra day every 4 years starting at 1973
-
-	y = dt->year - 1970;
-	days = (y * 365) + ((y + 1) / 4);
-
-	// Change table if current year is a leap year
-
-	if ((dt->year % 4) == 0) {
-		dbm = lydaysbymonth;
-	}
-
-	// Add in days elapsed in the current year
-
-	days += (dt->day - 1) + dbm[dt->month - 1];
-
-	// Add accumulated days to accumulated seconds
-
-	secs += 24 * 3600 * days;
-
-	return secs;
-}
-
-unsigned long
-IOPMrootDomain::getRUN_STATE(void)
-{
-	return (_aotNow && !(kIOPMWakeEventAOTExitFlags & _aotPendingFlags)) ? AOT_STATE : ON_STATE;
-}
-
-bool
-IOPMrootDomain::isAOTMode()
-{
-	return _aotNow;
-}
-
-bool
-IOPMrootDomain::isLPWMode()
-{
-	return gLPWFlags && currentOrPendingPowerState(AOT_STATE);
-}
-
-bool
-IOPMIsAOTMode(void)
-{
-	return gIOPMRootDomain && gIOPMRootDomain->isAOTMode();
-}
-bool
-IOPMIsLPWMode(void)
-{
-	return gIOPMRootDomain && gIOPMRootDomain->isLPWMode();
-}
-
-void
-IOPMNetworkStackFullWake(uint64_t flags, const char * reason)
-{
-	assert(kIOPMNetworkStackFullWakeFlag == flags);
-	assert(gIOPMRootDomain);
-	gIOPMRootDomain->claimSystemWakeEvent(gIOPMRootDomain, kIOPMWakeEventAOTExit, reason, NULL);
-}
-
-IOReturn
-IOPMrootDomain::setWakeTime(uint64_t wakeContinuousTime)
-{
-	if (kIOPMAOTModeCycle & _aotMode) {
-		return kIOReturnSuccess;
-	}
-	return _setWakeTime(wakeContinuousTime);
-}
-
-IOReturn
-IOPMrootDomain::_setWakeTime(uint64_t wakeContinuousTime)
-{
-	clock_sec_t     nowsecs, wakesecs;
-	clock_usec_t    nowmicrosecs, wakemicrosecs;
-	uint64_t        nowAbs, wakeAbs;
-
-	if (!_aotMode) {
-		return kIOReturnNotReady;
-	}
-
-	clock_gettimeofday_and_absolute_time(&nowsecs, &nowmicrosecs, &nowAbs);
-	wakeAbs = continuoustime_to_absolutetime(wakeContinuousTime);
-	if (wakeAbs < nowAbs) {
-		printf(LOG_PREFIX "wakeAbs %qd < nowAbs %qd\n", wakeAbs, nowAbs);
-		wakeAbs = nowAbs;
-	}
-	wakeAbs -= nowAbs;
-	absolutetime_to_microtime(wakeAbs, &wakesecs, &wakemicrosecs);
-
-	wakesecs += nowsecs;
-	wakemicrosecs += nowmicrosecs;
-	if (wakemicrosecs >= USEC_PER_SEC) {
-		wakesecs++;
-		wakemicrosecs -= USEC_PER_SEC;
-	}
-	if (wakemicrosecs >= (USEC_PER_SEC / 10)) {
-		wakesecs++;
-	}
-
-	IOPMConvertSecondsToCalendar(wakesecs, &_aotWakeTimeCalendar);
-
-	if (_aotWakeTimeContinuous != wakeContinuousTime) {
-		_aotWakeTimeContinuous = wakeContinuousTime;
-		IOLog(LOG_PREFIX "setWakeTime: " YMDTF "\n", YMDT(&_aotWakeTimeCalendar));
-	}
-	_aotWakeTimeCalendar.selector = kPMCalendarTypeMaintenance;
-	_aotWakeTimeUTC               = wakesecs;
-
-	return kIOReturnSuccess;
-}
-
-// assumes WAKEEVENT_LOCK
-bool
-IOPMrootDomain::aotShouldExit(bool software)
-{
-	bool exitNow = false;
-	const char * reason = "";
-
-	if (!_aotNow) {
-		return false;
-	}
-
-	if (software) {
-		exitNow = true;
-		_aotMetrics->softwareRequestCount++;
-		reason = "software request";
-	} else if (kIOPMWakeEventAOTExitFlags & _aotPendingFlags) {
-		exitNow = true;
-		reason = gWakeReasonString;
-	} else if ((kIOPMAOTModeRespectTimers & _aotMode) && _calendarWakeAlarmUTC) {
-		clock_sec_t     sec;
-		clock_usec_t    usec;
-		clock_get_calendar_microtime(&sec, &usec);
-		if (_calendarWakeAlarmUTC <= sec) {
-			exitNow = true;
-			_aotMetrics->rtcAlarmsCount++;
-			reason = "user alarm";
-		}
-	}
-	if (exitNow) {
-		_aotPendingFlags |= kIOPMWakeEventAOTExit;
-		IOLog(LOG_PREFIX "AOT exit for %s, sc %d po %d, cp %d, rj %d, ex %d, nt %d, rt %d\n",
-		    reason,
-		    _aotMetrics->sleepCount,
-		    _aotMetrics->possibleCount,
-		    _aotMetrics->confirmedPossibleCount,
-		    _aotMetrics->rejectedPossibleCount,
-		    _aotMetrics->expiredPossibleCount,
-		    _aotMetrics->noTimeSetCount,
-		    _aotMetrics->rtcAlarmsCount);
-	}
-	return exitNow;
-}
-
-void
-IOPMrootDomain::aotExit(bool cps)
-{
-	uint32_t savedMessageMask;
-
-	ASSERT_GATED();
-	_aotNow = false;
-	_aotRunMode = 0;
-	_aotReadyToFullWake = false;
-	if (_aotTimerScheduled) {
-		_aotTimerES->cancelTimeout();
-		_aotTimerScheduled = false;
-	}
-	updateTasksSuspend(kTasksSuspendNoChange, kTasksSuspendUnsuspended);
-
-	_aotMetrics->totalTime += mach_absolute_time() - _aotLastWakeTime;
-	_aotLastWakeTime = 0;
-	if (_aotMetrics->sleepCount && (_aotMetrics->sleepCount <= kIOPMAOTMetricsKernelWakeCountMax)) {
-		WAKEEVENT_LOCK();
-		strlcpy(&_aotMetrics->kernelWakeReason[_aotMetrics->sleepCount - 1][0],
-		    gWakeReasonString,
-		    sizeof(_aotMetrics->kernelWakeReason[_aotMetrics->sleepCount]));
-		WAKEEVENT_UNLOCK();
-	}
-
-	_aotWakeTimeCalendar.selector = kPMCalendarTypeInvalid;
-
-	// Preserve the message mask since a system wake transition
-	// may have already started and initialized the mask.
-	savedMessageMask = _systemMessageClientMask;
-	_systemMessageClientMask = kSystemMessageClientLegacyApp;
-	tellClients(kIOMessageSystemWillPowerOn);
-	_systemMessageClientMask = savedMessageMask | kSystemMessageClientLegacyApp;
-
-	if (cps) {
-		changePowerStateWithTagToPriv(getRUN_STATE(), kCPSReasonAOTExit);
-	}
-}
-
-void
-IOPMrootDomain::aotEvaluate(IOTimerEventSource * timer)
-{
-	bool exitNow;
-
-	IOLog("aotEvaluate(%d) 0x%x\n", (timer != NULL), _aotPendingFlags);
-
-	WAKEEVENT_LOCK();
-	exitNow = aotShouldExit(false);
-	if (timer != NULL) {
-		_aotTimerScheduled = false;
-	}
-	WAKEEVENT_UNLOCK();
-	if (exitNow) {
-		aotExit(true);
-	} else {
-#if 0
-		if (_aotLingerTime) {
-			uint64_t deadline;
-			IOLog("aot linger before sleep\n");
-			clock_absolutetime_interval_to_deadline(_aotLingerTime, &deadline);
-			clock_delay_until(deadline);
-		}
-#endif
-		privateSleepSystem(kIOPMSleepReasonSoftware);
-	}
+    return true;
 }
 
 //******************************************************************************
@@ -8388,88 +5279,22 @@
 // to SLEEP_STATE.
 //******************************************************************************
 
-void
-IOPMrootDomain::adjustPowerState( bool sleepASAP )
-{
-	DEBUG_LOG("adjustPowerState %s, asap %d, idleSleepEnabled %d, _aotNow %d\n",
-	    getPowerStateString((uint32_t) getPowerState()), sleepASAP, idleSleepEnabled, _aotNow);
-
-	ASSERT_GATED();
-
-	if (_aotNow) {
-		bool exitNow;
-
-		if (AOT_STATE != getPowerState()) {
-			return;
-		}
-		WAKEEVENT_LOCK();
-		exitNow = aotShouldExit(false);
-		if (!exitNow
-		    && !_aotTimerScheduled
-		    && (kIOPMWakeEventAOTPossibleExit == (kIOPMWakeEventAOTPossibleFlags & _aotPendingFlags))) {
-			_aotTimerScheduled = true;
-			_aotTimerES->setTimeout(_aotLingerTime, kMillisecondScale);
-		}
-		WAKEEVENT_UNLOCK();
-		if (exitNow) {
-			aotExit(true);
-		} else {
-			_aotReadyToFullWake = true;
-			if (!_aotTimerScheduled) {
-				if (kIOPMDriverAssertionLevelOn == getPMAssertionLevel(kIOPMDriverAssertionCPUBit)) {
-					// Don't try to force sleep during AOT while IOMobileFramebuffer is holding a power assertion.
-					// Doing so will result in the sleep being cancelled anyway,
-					// but this check avoids unnecessary thrashing in the power state engine.
-					return;
-				}
-				privateSleepSystem(kIOPMSleepReasonSoftware);
-			}
-		}
-		return;
-	}
-
-	if ((!idleSleepEnabled) || !checkSystemSleepEnabled()) {
-		changePowerStateWithTagToPriv(getRUN_STATE(), kCPSReasonAdjustPowerState);
-	} else if (sleepASAP) {
-		changePowerStateWithTagToPriv(SLEEP_STATE, kCPSReasonAdjustPowerState);
-	}
-}
-
-void
-IOPMrootDomain::handleSetDisplayPowerOn(bool powerOn)
-{
-	if (powerOn) {
-		if (!checkSystemCanSustainFullWake()) {
-			DLOG("System cannot sustain full wake\n");
-			return;
-		}
-
-		// Force wrangler to max power state. If system is in dark wake
-		// this alone won't raise the wrangler's power state.
-		if (wrangler) {
-			wrangler->changePowerStateForRootDomain(kWranglerPowerStateMax);
-		}
-
-		// System in dark wake, always requesting full wake should
-		// not have any bad side-effects, even if the request fails.
-
-		if (!CAP_CURRENT(kIOPMSystemCapabilityGraphics)) {
-			setProperty(kIOPMRootDomainWakeTypeKey, kIOPMRootDomainWakeTypeNotification);
-			requestFullWake( kFullWakeReasonDisplayOn );
-		}
-	} else {
-		// Relenquish desire to power up display.
-		// Must first transition to state 1 since wrangler doesn't
-		// power off the displays at state 0. At state 0 the root
-		// domain is removed from the wrangler's power client list.
-		if (wrangler) {
-			wrangler->changePowerStateForRootDomain(kWranglerPowerStateMin + 1);
-			wrangler->changePowerStateForRootDomain(kWranglerPowerStateMin);
-		}
-	}
-}
-
-TUNABLE(bool, test_sleep_in_vm, "test_sleep_in_vm", false);
+void IOPMrootDomain::adjustPowerState( bool sleepASAP )
+{
+    DLOG("adjustPowerState ps %u, asap %d, slider %ld\n",
+        (uint32_t) getPowerState(), sleepASAP, sleepSlider);
+
+    ASSERT_GATED();
+
+    if ((sleepSlider == 0) || !checkSystemCanSleep())
+    {
+        changePowerStateToPriv(ON_STATE);
+    }
+    else if ( sleepASAP )
+    {
+        changePowerStateToPriv(SLEEP_STATE);
+    }
+}
 
 //******************************************************************************
 // dispatchPowerEvent
@@ -8477,306 +5302,205 @@
 // IOPMPowerStateQueue callback function. Running on PM work loop thread.
 //******************************************************************************
 
-void
-IOPMrootDomain::dispatchPowerEvent(
-	uint32_t event, void * arg0, uint64_t arg1 )
-{
-	ASSERT_GATED();
-
-	switch (event) {
-	case kPowerEventFeatureChanged:
-		DMSG("power event %u args %p 0x%llx\n", event, OBFUSCATE(arg0), arg1);
-		messageClients(kIOPMMessageFeatureChange, this);
-		break;
-
-	case kPowerEventReceivedPowerNotification:
-		DMSG("power event %u args %p 0x%llx\n", event, OBFUSCATE(arg0), arg1);
-		handlePowerNotification((UInt32)(uintptr_t) arg0 );
-		break;
-
-	case kPowerEventSystemBootCompleted:
-		DLOG("power event %u args %p 0x%llx\n", event, OBFUSCATE(arg0), arg1);
-		if (systemBooting) {
-			systemBooting = false;
-
-			if (PE_get_default("sleep-disabled", &gSleepDisabledFlag, sizeof(gSleepDisabledFlag))) {
-				DLOG("Setting gSleepDisabledFlag to %u from device tree\n", gSleepDisabledFlag);
-				if (test_sleep_in_vm && gSleepDisabledFlag) {
-					DLOG("Clearing gSleepDisabledFlag due to test_sleep_in_vm boot-arg\n");
-					gSleepDisabledFlag = 0;
-				}
-			}
-
-			if (lowBatteryCondition || thermalEmergencyState) {
-				if (lowBatteryCondition) {
-					privateSleepSystem(kIOPMSleepReasonLowPower);
-				} else {
-					privateSleepSystem(kIOPMSleepReasonThermalEmergency);
-				}
-				// The rest is unnecessary since the system is expected
-				// to sleep immediately. The following wake will update
-				// everything.
-				break;
-			}
-
-			sleepWakeDebugMemAlloc();
-			saveFailureData2File();
-
-			// If lid is closed, re-send lid closed notification
-			// now that booting is complete.
-			if (clamshellClosed) {
-				handlePowerNotification(kLocalEvalClamshellCommand);
-			}
-			evaluatePolicy( kStimulusAllowSystemSleepChanged );
-		}
-		break;
-
-	case kPowerEventSystemShutdown:
-		DLOG("power event %u args %p 0x%llx\n", event, OBFUSCATE(arg0), arg1);
-		if (kOSBooleanTrue == (OSBoolean *) arg0) {
-			/* We set systemShutdown = true during shutdown
-			 *  to prevent sleep at unexpected times while loginwindow is trying
-			 *  to shutdown apps and while the OS is trying to transition to
-			 *  complete power of.
-			 *
-			 *  Set to true during shutdown, as soon as loginwindow shows
-			 *  the "shutdown countdown dialog", through individual app
-			 *  termination, and through black screen kernel shutdown.
-			 */
-			systemShutdown = true;
-		} else {
-			/*
-			 *  A shutdown was initiated, but then the shutdown
-			 *  was cancelled, clearing systemShutdown to false here.
-			 */
-			systemShutdown = false;
-		}
-		break;
-
-	case kPowerEventUserDisabledSleep:
-		DLOG("power event %u args %p 0x%llx\n", event, OBFUSCATE(arg0), arg1);
-		userDisabledAllSleep = (kOSBooleanTrue == (OSBoolean *) arg0);
-		break;
-
-	case kPowerEventRegisterSystemCapabilityClient:
-		DLOG("power event %u args %p 0x%llx\n", event, OBFUSCATE(arg0), arg1);
-
-		// reset() handles the arg0 == nullptr case for us
-		systemCapabilityNotifier.reset((IONotifier *) arg0, OSRetain);
-		/* intentional fall-through */
-		[[clang::fallthrough]];
-
-	case kPowerEventRegisterKernelCapabilityClient:
-		DLOG("power event %u args %p 0x%llx\n", event, OBFUSCATE(arg0), arg1);
-		if (!_joinedCapabilityClients) {
-			_joinedCapabilityClients = OSSet::withCapacity(8);
-		}
-		if (arg0) {
-			OSSharedPtr<IONotifier> notify((IONotifier *) arg0, OSNoRetain);
-			if (_joinedCapabilityClients) {
-				_joinedCapabilityClients->setObject(notify.get());
-				synchronizePowerTree( kIOPMSyncNoChildNotify );
-			}
-		}
-		break;
-
-	case kPowerEventPolicyStimulus:
-		DMSG("power event %u args %p 0x%llx\n", event, OBFUSCATE(arg0), arg1);
-		if (arg0) {
-			int stimulus = (int)(uintptr_t) arg0;
-			evaluatePolicy(stimulus, (uint32_t) arg1);
-		}
-		break;
-
-	case kPowerEventAssertionCreate:
-		DMSG("power event %u args %p 0x%llx\n", event, OBFUSCATE(arg0), arg1);
-		if (pmAssertions) {
-			pmAssertions->handleCreateAssertion((OSValueObject<PMAssertStruct> *)arg0);
-		}
-		break;
-
-
-	case kPowerEventAssertionRelease:
-		DMSG("power event %u args %p 0x%llx\n", event, OBFUSCATE(arg0), arg1);
-		if (pmAssertions) {
-			pmAssertions->handleReleaseAssertion(arg1);
-		}
-		break;
-
-	case kPowerEventAssertionSetLevel:
-		DMSG("power event %u args %p 0x%llx\n", event, OBFUSCATE(arg0), arg1);
-		if (pmAssertions) {
-			pmAssertions->handleSetAssertionLevel(arg1, (IOPMDriverAssertionLevel)(uintptr_t)arg0);
-		}
-		break;
-
-	case kPowerEventQueueSleepWakeUUID:
-		DLOG("power event %u args %p 0x%llx\n", event, OBFUSCATE(arg0), arg1);
-		handleQueueSleepWakeUUID((OSObject *)arg0);
-		break;
-	case kPowerEventPublishSleepWakeUUID:
-		DLOG("power event %u args %p 0x%llx\n", event, OBFUSCATE(arg0), arg1);
-		handlePublishSleepWakeUUID((bool)arg0);
-		break;
-
-	case kPowerEventSetDisplayPowerOn:
-		DLOG("power event %u args %p 0x%llx\n", event, OBFUSCATE(arg0), arg1);
-		if (arg1 != 0) {
-			displayPowerOnRequested = true;
-		} else {
-			displayPowerOnRequested = false;
-		}
-		handleSetDisplayPowerOn(displayPowerOnRequested);
-		break;
-
-	case kPowerEventPublishWakeType:
-		DLOG("power event %u args %p 0x%llx\n", event, OBFUSCATE(arg0), arg1);
-
-		// Don't replace wake type property if already set
-		if ((arg0 == gIOPMWakeTypeUserKey) ||
-		    !propertyExists(kIOPMRootDomainWakeTypeKey)) {
-			const char * wakeType = NULL;
-
-			if (arg0 == gIOPMWakeTypeUserKey) {
-				requestUserActive(this, "WakeTypeUser");
-				wakeType = kIOPMRootDomainWakeTypeUser;
-			} else if (arg0 == gIOPMSettingDebugWakeRelativeKey) {
-				if (!(gDarkWakeFlags & kDarkWakeFlagAlarmIsDark)) {
-					requestUserActive(this, "WakeTypeAlarm");
-				}
-				wakeType = kIOPMRootDomainWakeTypeAlarm;
-			} else if (arg0 == gIOPMSettingSleepServiceWakeCalendarKey) {
-				darkWakeSleepService = true;
-				wakeType = kIOPMRootDomainWakeTypeSleepService;
-			} else if (arg0 == gIOPMSettingMaintenanceWakeCalendarKey) {
-				wakeType = kIOPMRootDomainWakeTypeMaintenance;
-			}
-
-			if (wakeType) {
-				setProperty(kIOPMRootDomainWakeTypeKey, wakeType);
-			}
-		}
-		break;
-
-	case kPowerEventAOTEvaluate:
-		DLOG("power event %u args %p 0x%llx\n", event, OBFUSCATE(arg0), arg1);
-		if (_aotReadyToFullWake) {
-			aotEvaluate(NULL);
-		}
-		break;
-	case kPowerEventRunModeRequest:
-		DLOG("power event %u args %p 0x%llx\n", event, OBFUSCATE(arg0), arg1);
-		// arg1 == runModeMask
-		handleRequestRunMode(arg1);
-		break;
-	}
+void IOPMrootDomain::dispatchPowerEvent(
+    uint32_t event, void * arg0, uint64_t arg1 )
+{
+    DLOG("power event %u args %p 0x%llx\n", event, arg0, arg1);
+    ASSERT_GATED();
+
+    switch (event)
+    {
+        case kPowerEventFeatureChanged:
+            messageClients(kIOPMMessageFeatureChange, this);
+            break;
+
+        case kPowerEventReceivedPowerNotification:
+            handlePowerNotification( (UInt32)(uintptr_t) arg0 );
+            break;
+        
+        case kPowerEventSystemBootCompleted:
+            if (systemBooting)
+            {
+                systemBooting = false;
+
+                // If lid is closed, re-send lid closed notification
+                // now that booting is complete.
+                if ( clamshellClosed )
+                {
+                    handlePowerNotification(kLocalEvalClamshellCommand);
+                }
+                evaluatePolicy( kStimulusAllowSystemSleepChanged );
+            }
+            break;
+
+        case kPowerEventSystemShutdown:
+            if (kOSBooleanTrue == (OSBoolean *) arg0)
+            {
+                /* We set systemShutdown = true during shutdown
+                   to prevent sleep at unexpected times while loginwindow is trying
+                   to shutdown apps and while the OS is trying to transition to
+                   complete power of.
+                   
+                   Set to true during shutdown, as soon as loginwindow shows
+                   the "shutdown countdown dialog", through individual app
+                   termination, and through black screen kernel shutdown.
+                 */
+                systemShutdown = true;
+            } else {
+                /*
+                 A shutdown was initiated, but then the shutdown
+                 was cancelled, clearing systemShutdown to false here.
+                */
+                systemShutdown = false;            
+            }
+            break;
+
+        case kPowerEventUserDisabledSleep:
+            userDisabledAllSleep = (kOSBooleanTrue == (OSBoolean *) arg0);
+            break;
+
+        case kPowerEventRegisterSystemCapabilityClient:
+            if (systemCapabilityNotifier)
+            {
+                systemCapabilityNotifier->release();
+                systemCapabilityNotifier = 0;
+            }
+            if (arg0)
+            {
+                systemCapabilityNotifier = (IONotifier *) arg0;
+                systemCapabilityNotifier->retain();
+            }
+            /* intentional fall-through */
+
+        case kPowerEventRegisterKernelCapabilityClient:
+            if (!_joinedCapabilityClients)
+                _joinedCapabilityClients = OSSet::withCapacity(8);
+            if (arg0)
+            {
+                IONotifier * notify = (IONotifier *) arg0;
+                if (_joinedCapabilityClients)
+                {
+                    _joinedCapabilityClients->setObject(notify);
+                    synchronizePowerTree( kIOPMSyncNoChildNotify );
+                }
+                notify->release();
+            }
+            break;
+
+        case kPowerEventPolicyStimulus:
+            if (arg0)
+            {
+                int stimulus = (uintptr_t) arg0;
+                evaluatePolicy( stimulus, (uint32_t) arg1 );
+            }
+            break;
+
+        case kPowerEventAssertionCreate:
+            if (pmAssertions) {
+                pmAssertions->handleCreateAssertion((OSData *)arg0);
+            }
+            break;
+
+
+        case kPowerEventAssertionRelease:
+            if (pmAssertions) {
+                pmAssertions->handleReleaseAssertion(arg1);
+            }
+            break;
+
+        case kPowerEventAssertionSetLevel:
+            if (pmAssertions) {
+                pmAssertions->handleSetAssertionLevel(arg1, (IOPMDriverAssertionLevel)(uintptr_t)arg0);
+            }
+            break;
+            
+        case kPowerEventQueueSleepWakeUUID:
+            handleQueueSleepWakeUUID((OSObject *)arg0);
+            break;
+        case kPowerEventPublishSleepWakeUUID:
+            handlePublishSleepWakeUUID((bool)arg0);
+            break;
+        case kPowerEventSuspendClient:
+            handleSuspendPMNotificationClient((uintptr_t)arg0, (bool)arg1);
+            break;
+    }
 }
 
 //******************************************************************************
 // systemPowerEventOccurred
 //
 // The power controller is notifying us of a hardware-related power management
-// event that we must handle.
+// event that we must handle. 
 //
 // systemPowerEventOccurred covers the same functionality that
 // receivePowerNotification does; it simply provides a richer API for conveying
 // more information.
 //******************************************************************************
 
-IOReturn
-IOPMrootDomain::systemPowerEventOccurred(
-	const OSSymbol *event,
-	uint32_t intValue)
-{
-	IOReturn        attempt = kIOReturnSuccess;
-	OSSharedPtr<OSNumber>        newNumber;
-
-	if (!event) {
-		return kIOReturnBadArgument;
-	}
-
-	newNumber = OSNumber::withNumber(intValue, 8 * sizeof(intValue));
-	if (!newNumber) {
-		return kIOReturnInternalError;
-	}
-
-	attempt = systemPowerEventOccurred(event, static_cast<OSObject *>(newNumber.get()));
-
-	return attempt;
-}
-
-void
-IOPMrootDomain::setThermalState(OSObject *value)
-{
-	OSNumber * num;
-
-	if (gIOPMWorkLoop->inGate() == false) {
-		gIOPMWorkLoop->runAction(
-			OSMemberFunctionCast(IOWorkLoop::Action, this, &IOPMrootDomain::setThermalState),
-			(OSObject *)this,
-			(void *)value);
-
-		return;
-	}
-	if (value && (num = OSDynamicCast(OSNumber, value))) {
-		thermalWarningState = ((num->unsigned32BitValue() == kIOPMThermalLevelWarning) ||
-		    (num->unsigned32BitValue() == kIOPMThermalLevelTrap)) ? 1 : 0;
-	}
-}
-
-IOReturn
-IOPMrootDomain::systemPowerEventOccurred(
-	const OSSymbol *event,
-	OSObject *value)
-{
-	OSSharedPtr<OSDictionary> thermalsDict;
-	bool shouldUpdate = true;
-
-	if (!event || !value) {
-		return kIOReturnBadArgument;
-	}
-
-	// LOCK
-	// We reuse featuresDict Lock because it already exists and guards
-	// the very infrequently used publish/remove feature mechanism; so there's zero rsk
-	// of stepping on that lock.
-	if (featuresDictLock) {
-		IOLockLock(featuresDictLock);
-	}
-
-	OSSharedPtr<OSObject> origThermalsProp = copyProperty(kIOPMRootDomainPowerStatusKey);
-	OSDictionary * origThermalsDict = OSDynamicCast(OSDictionary, origThermalsProp.get());
-
-	if (origThermalsDict) {
-		thermalsDict = OSDictionary::withDictionary(origThermalsDict);
-	} else {
-		thermalsDict = OSDictionary::withCapacity(1);
-	}
-
-	if (!thermalsDict) {
-		shouldUpdate = false;
-		goto exit;
-	}
-
-	thermalsDict->setObject(event, value);
-
-	setProperty(kIOPMRootDomainPowerStatusKey, thermalsDict.get());
+IOReturn IOPMrootDomain::systemPowerEventOccurred(
+    const OSSymbol *event,
+    uint32_t intValue)
+{
+    IOReturn        attempt = kIOReturnSuccess;
+    OSNumber        *newNumber = NULL;
+
+    if (!event) 
+        return kIOReturnBadArgument;
+        
+    newNumber = OSNumber::withNumber(intValue, 8*sizeof(intValue));
+    if (!newNumber)
+        return kIOReturnInternalError;
+
+    attempt = systemPowerEventOccurred(event, (OSObject *)newNumber);
+
+    newNumber->release();
+
+    return attempt;
+}
+
+IOReturn IOPMrootDomain::systemPowerEventOccurred(
+    const OSSymbol *event,
+    OSObject *value)
+{
+    OSDictionary *thermalsDict = NULL;
+    bool shouldUpdate = true;
+    
+    if (!event || !value) 
+        return kIOReturnBadArgument;
+
+    // LOCK
+    // We reuse featuresDict Lock because it already exists and guards
+    // the very infrequently used publish/remove feature mechanism; so there's zero rsk
+    // of stepping on that lock.
+    if (featuresDictLock) IOLockLock(featuresDictLock);
+
+    thermalsDict = (OSDictionary *)getProperty(kIOPMRootDomainPowerStatusKey);
+                   
+    if (thermalsDict && OSDynamicCast(OSDictionary, thermalsDict)) {
+        thermalsDict = OSDictionary::withDictionary(thermalsDict);                        
+    } else {
+        thermalsDict = OSDictionary::withCapacity(1);
+    }
+
+    if (!thermalsDict) {
+        shouldUpdate = false;
+        goto exit;
+    }
+
+    thermalsDict->setObject (event, value);
+
+    setProperty (kIOPMRootDomainPowerStatusKey, thermalsDict);
+
+    thermalsDict->release();
 
 exit:
-	// UNLOCK
-	if (featuresDictLock) {
-		IOLockUnlock(featuresDictLock);
-	}
-
-	if (shouldUpdate) {
-		if (event &&
-		    event->isEqualTo(kIOPMThermalLevelWarningKey)) {
-			setThermalState(value);
-		}
-		messageClients(kIOPMMessageSystemPowerEventOccurred, (void *)NULL);
-	}
-
-	return kIOReturnSuccess;
+    // UNLOCK
+    if (featuresDictLock) IOLockUnlock(featuresDictLock);
+
+    if (shouldUpdate)
+        messageClients (kIOPMMessageSystemPowerEventOccurred, (void *)NULL);
+
+    return kIOReturnSuccess;
 }
 
 //******************************************************************************
@@ -8787,297 +5511,243 @@
 // from the power mgt micro.
 //******************************************************************************
 
-IOReturn
-IOPMrootDomain::receivePowerNotification( UInt32 msg )
-{
-	if (msg & kIOPMPowerButton) {
-		uint32_t currentPhase = pmTracer->getTracePhase();
-		if (currentPhase != kIOPMTracePointSystemUp && currentPhase > kIOPMTracePointSystemSleep) {
-			DEBUG_LOG("power button pressed during wake. phase = %u\n", currentPhase);
-			swd_flags |= SWD_PWR_BTN_STACKSHOT;
-			thread_call_enter(powerButtonDown);
-		} else {
-			DEBUG_LOG("power button pressed when system is up\n");
-		}
-	} else if (msg & kIOPMPowerButtonUp) {
-		if (swd_flags & SWD_PWR_BTN_STACKSHOT) {
-			swd_flags &= ~SWD_PWR_BTN_STACKSHOT;
-			thread_call_enter(powerButtonUp);
-		}
-	} else {
-		pmPowerStateQueue->submitPowerEvent(
-			kPowerEventReceivedPowerNotification, (void *)(uintptr_t) msg );
-	}
-	return kIOReturnSuccess;
-}
-
-void
-IOPMrootDomain::handlePowerNotification( UInt32 msg )
-{
-	bool        eval_clamshell = false;
-	bool        eval_clamshell_alarm = false;
-
-	ASSERT_GATED();
-
-	/*
-	 * Local (IOPMrootDomain only) eval clamshell command
-	 */
-	if (msg & kLocalEvalClamshellCommand) {
-		if ((gClamshellFlags & kClamshell_WAR_47715679) && isRTCAlarmWake) {
-			eval_clamshell_alarm = true;
-
-			// reset isRTCAlarmWake. This evaluation should happen only once
-			// on RTC/Alarm wake. Any clamshell events after wake should follow
-			// the regular evaluation
-			isRTCAlarmWake = false;
-		} else {
-			eval_clamshell = true;
-		}
-	}
-
-	/*
-	 * Overtemp
-	 */
-	if (msg & kIOPMOverTemp) {
-		DLOG("Thermal overtemp message received!\n");
-		thermalEmergencyState = true;
-		privateSleepSystem(kIOPMSleepReasonThermalEmergency);
-	}
-
-	/*
-	 * Forward DW thermal notification to client, if system is not going to sleep
-	 */
-	if ((msg & kIOPMDWOverTemp) && (_systemTransitionType != kSystemTransitionSleep)) {
-		DLOG("DarkWake thermal limits message received!\n");
-		messageClients(kIOPMMessageDarkWakeThermalEmergency);
-	}
-
-	/*
-	 * Sleep Now!
-	 */
-	if (msg & kIOPMSleepNow) {
-		privateSleepSystem(kIOPMSleepReasonSoftware);
-	}
-
-	/*
-	 * Power Emergency
-	 */
-	if (msg & kIOPMPowerEmergency) {
-		DLOG("Received kIOPMPowerEmergency");
-#if HIBERNATION && defined(__arm64__)
-		if (!ml_is_secure_hib_supported() || ldmHibernateDisable) {
-			// Wait for the next low battery notification if the system state is
-			// in transition.
-			if ((_systemTransitionType == kSystemTransitionNone) &&
-			    CAP_CURRENT(kIOPMSystemCapabilityCPU) &&
-			    !systemBooting && !systemShutdown && !gWillShutdown) {
-				// Setting lowBatteryCondition will prevent system sleep
-				lowBatteryCondition = true;
-
-				// Notify userspace to initiate system shutdown
-				DLOG("Initiating userspace shutdown ml_is_secure_hib_supported %d lockdownMode %d", ml_is_secure_hib_supported(), ldmHibernateDisable);
-				messageClients(kIOPMMessageRequestSystemShutdown);
-			}
-		} else {
-			lowBatteryCondition = true;
-			privateSleepSystem(kIOPMSleepReasonLowPower);
-		}
-#else  /* HIBERNATION && defined(__arm64__) */
-		lowBatteryCondition = true;
-		privateSleepSystem(kIOPMSleepReasonLowPower);
-#endif /* HIBERNATION && defined(__arm64__) */
-	}
-
-	/*
-	 * Clamshell OPEN
-	 */
-	if (msg & kIOPMClamshellOpened) {
-		DLOG("Clamshell opened\n");
-		// Received clamshel open message from clamshell controlling driver
-		// Update our internal state and tell general interest clients
-		clamshellClosed = false;
-		clamshellExists = true;
-
-		// Don't issue a hid tickle when lid is open and polled on wake
-		if (msg & kIOPMSetValue) {
-			setProperty(kIOPMRootDomainWakeTypeKey, "Lid Open");
-			reportUserInput();
-		}
-
-		// Tell PMCPU
-		informCPUStateChange(kInformLid, 0);
-
-		// Tell general interest clients
-		sendClientClamshellNotification();
-
-		bool aborting =  ((lastSleepReason == kIOPMSleepReasonClamshell)
-		    || (lastSleepReason == kIOPMSleepReasonIdle)
-		    || (lastSleepReason == kIOPMSleepReasonMaintenance));
-		if (aborting) {
-			userActivityCount++;
-		}
-		DLOG("clamshell tickled %d lastSleepReason %d\n", userActivityCount, lastSleepReason);
-	}
-
-	/*
-	 * Clamshell CLOSED
-	 * Send the clamshell interest notification since the lid is closing.
-	 */
-	if (msg & kIOPMClamshellClosed) {
-		if ((clamshellIgnoreClose || (gClamshellFlags & kClamshell_WAR_38378787)) &&
-		    clamshellClosed && clamshellExists) {
-			DLOG("Ignoring redundant Clamshell close event\n");
-		} else {
-			DLOG("Clamshell closed\n");
-			// Received clamshel open message from clamshell controlling driver
-			// Update our internal state and tell general interest clients
-			clamshellClosed = true;
-			clamshellExists = true;
-
-			// Ignore all following clamshell close events until the clamshell
-			// is opened or the system sleeps. When a clamshell close triggers
-			// a system wake, the lid driver may send us two clamshell close
-			// events, one for the clamshell close event itself, and a second
-			// close event when the driver polls the lid state on wake.
-			clamshellIgnoreClose = true;
-
-			// Tell PMCPU
-			informCPUStateChange(kInformLid, 1);
-
-			// Tell general interest clients
-			sendClientClamshellNotification();
-
-			// And set eval_clamshell = so we can attempt
-			eval_clamshell = true;
-		}
-	}
-
-	/*
-	 * Set Desktop mode (sent from graphics)
-	 *
-	 *  -> reevaluate lid state
-	 */
-	if (msg & kIOPMSetDesktopMode) {
-		desktopMode = (0 != (msg & kIOPMSetValue));
-		msg &= ~(kIOPMSetDesktopMode | kIOPMSetValue);
-		DLOG("Desktop mode %d\n", desktopMode);
-
-		sendClientClamshellNotification();
-
-		// Re-evaluate the lid state
-		eval_clamshell = true;
-	}
-
-	/*
-	 * AC Adaptor connected
-	 *
-	 *  -> reevaluate lid state
-	 */
-	if (msg & kIOPMSetACAdaptorConnected) {
-		acAdaptorConnected = (0 != (msg & kIOPMSetValue));
-		msg &= ~(kIOPMSetACAdaptorConnected | kIOPMSetValue);
-
-		// Tell CPU PM
-		informCPUStateChange(kInformAC, !acAdaptorConnected);
-
-		// Tell BSD if AC is connected
-		//      0 == external power source; 1 == on battery
-		post_sys_powersource(acAdaptorConnected ? 0:1);
-
-		sendClientClamshellNotification();
-
-		IOUserServer::powerSourceChanged(acAdaptorConnected);
-
-		// Re-evaluate the lid state
-		eval_clamshell = true;
-
-		// Lack of AC may have latched a display wrangler tickle.
-		// This mirrors the hardware's USB wake event latch, where a latched
-		// USB wake event followed by an AC attach will trigger a full wake.
-		latchDisplayWranglerTickle( false );
+IOReturn IOPMrootDomain::receivePowerNotification( UInt32 msg )
+{
+    pmPowerStateQueue->submitPowerEvent(
+        kPowerEventReceivedPowerNotification, (void *) msg );
+    return kIOReturnSuccess;
+}
+
+void IOPMrootDomain::handlePowerNotification( UInt32 msg )
+{
+    bool        eval_clamshell = false;
+
+    ASSERT_GATED();
+
+    /*
+     * Local (IOPMrootDomain only) eval clamshell command
+     */
+    if (msg & kLocalEvalClamshellCommand)
+    {
+        eval_clamshell = true;
+    }
+
+    /*
+     * Overtemp
+     */
+    if (msg & kIOPMOverTemp)
+    {
+        MSG("PowerManagement emergency overtemp signal. Going to sleep!");
+        privateSleepSystem (kIOPMSleepReasonThermalEmergency);
+    }
+
+    if (msg & kIOPMDWOverTemp)
+    {
+        if (!CAP_CURRENT(kIOPMSystemCapabilityCPU) ||
+            (_systemTransitionType == kSystemTransitionSleep) ||
+            (_systemTransitionType == kSystemTransitionWake)  ||
+            (_systemTransitionType == kSystemTransitionCapability))
+        {
+            // During early wake or when system capability is changing,
+            // set flag and take action at end of transition.
+            darkWakeThermalAlarm = true;
+        }
+        else if (!wranglerTickled && !darkWakeThermalEmergency &&
+                 !CAP_CURRENT(kIOPMSystemCapabilityGraphics))
+        {
+            // System in steady state and in dark wake
+            darkWakeThermalEmergency = true;
+            privateSleepSystem(kIOPMSleepReasonDarkWakeThermalEmergency);
+            MSG("DarkWake thermal limits breached. Going to sleep!\n");
+        }
+    }
+
+    /*
+     * Sleep Now!
+     */
+    if (msg & kIOPMSleepNow) 
+    {
+        privateSleepSystem (kIOPMSleepReasonSoftware);
+    }
+    
+    /*
+     * Power Emergency
+     */
+    if (msg & kIOPMPowerEmergency) 
+    {
+        lowBatteryCondition = true;
+        privateSleepSystem (kIOPMSleepReasonLowPower);
+    }
+
+    /*
+     * Clamshell OPEN
+     */
+    if (msg & kIOPMClamshellOpened) 
+    {
+        // Received clamshel open message from clamshell controlling driver
+        // Update our internal state and tell general interest clients
+        clamshellClosed = false;
+        clamshellExists = true;
+
+        // Don't issue a hid tickle when lid is open and polled on wake
+        if (msg & kIOPMSetValue)
+        {
+            setProperty(kIOPMRootDomainWakeTypeKey, "Lid Open");
+            reportUserInput();
+        }
+
+        // Tell PMCPU
+        informCPUStateChange(kInformLid, 0);
+
+        // Tell general interest clients        
+        sendClientClamshellNotification();
+
+        bool aborting =  ((lastSleepReason == kIOPMSleepReasonClamshell)
+                       || (lastSleepReason == kIOPMSleepReasonIdle) 
+                       || (lastSleepReason == kIOPMSleepReasonMaintenance));
+        if (aborting) userActivityCount++;
+        DLOG("clamshell tickled %d lastSleepReason %d\n", userActivityCount, lastSleepReason);
+    }
+
+    /* 
+     * Clamshell CLOSED
+     * Send the clamshell interest notification since the lid is closing. 
+     */
+    if (msg & kIOPMClamshellClosed)
+    {
+        // Received clamshel open message from clamshell controlling driver
+        // Update our internal state and tell general interest clients
+        clamshellClosed = true;
+        clamshellExists = true;
+
+        // Tell PMCPU
+        informCPUStateChange(kInformLid, 1);
+
+        // Tell general interest clients
+        sendClientClamshellNotification();
+        
+        // And set eval_clamshell = so we can attempt 
+        eval_clamshell = true;
+    }
+
+    /*
+     * Set Desktop mode (sent from graphics)
+     *
+     *  -> reevaluate lid state
+     */
+    if (msg & kIOPMSetDesktopMode) 
+    {
+        desktopMode = (0 != (msg & kIOPMSetValue));
+        msg &= ~(kIOPMSetDesktopMode | kIOPMSetValue);
+
+        sendClientClamshellNotification();
+
+        // Re-evaluate the lid state
+        if( clamshellClosed )
+        {
+            eval_clamshell = true;
+        }
+    }
+    
+    /*
+     * AC Adaptor connected
+     *
+     *  -> reevaluate lid state
+     */
+    if (msg & kIOPMSetACAdaptorConnected) 
+    {
+        acAdaptorConnected = (0 != (msg & kIOPMSetValue));
+        msg &= ~(kIOPMSetACAdaptorConnected | kIOPMSetValue);
+
+        // Tell CPU PM
+        informCPUStateChange(kInformAC, !acAdaptorConnected);
+
+        // Tell BSD if AC is connected
+        //      0 == external power source; 1 == on battery
+        post_sys_powersource(acAdaptorConnected ? 0:1);
+
+        sendClientClamshellNotification();
+
+        // Re-evaluate the lid state
+        if( clamshellClosed )
+        {
+            eval_clamshell = true;
+        }
+
+        // Lack of AC may have latched a display wrangler tickle.
+        // This mirrors the hardware's USB wake event latch, where a latched
+        // USB wake event followed by an AC attach will trigger a full wake.
+        latchDisplayWranglerTickle( false );
 
 #if HIBERNATION
-		// AC presence will reset the standy timer delay adjustment.
-		_standbyTimerResetSeconds = 0;
+        // AC presence will reset the standy timer delay adjustment.
+        _standbyTimerResetSeconds = 0;
 #endif
-		if (!userIsActive) {
-			// Reset userActivityTime when power supply is changed(rdr 13789330)
-			clock_get_uptime(&userActivityTime);
-		}
-	}
-
-	/*
-	 * Enable Clamshell (external display disappear)
-	 *
-	 *  -> reevaluate lid state
-	 */
-	if (msg & kIOPMEnableClamshell) {
-		DLOG("Clamshell enabled\n");
-
-		// Re-evaluate the lid state
-		// System should sleep on external display disappearance
-		// in lid closed operation.
-		if (true == clamshellDisabled) {
-			eval_clamshell = true;
-
-#if DARK_TO_FULL_EVALUATE_CLAMSHELL_DELAY
-			// Also clear kClamshellSleepDisableInternal when graphics enables
-			// the clamshell during a full wake. When graphics is behaving as
-			// expected, this will allow clamshell close to be honored earlier
-			// rather than waiting for the delayed evaluation.
-			if ((clamshellSleepDisableMask & kClamshellSleepDisableInternal) &&
-			    (CAP_PENDING(kIOPMSystemCapabilityGraphics) ||
-			    CAP_CURRENT(kIOPMSystemCapabilityGraphics))) {
-				setClamShellSleepDisable(false, kClamshellSleepDisableInternal);
-
-				// Cancel the TC to avoid an extra kLocalEvalClamshellCommand
-				// when timer expires which is harmless but useless.
-				thread_call_cancel(fullWakeThreadCall);
-			}
-#endif
-		}
-
-		clamshellDisabled = false;
-		sendClientClamshellNotification();
-	}
-
-	/*
-	 * Disable Clamshell (external display appeared)
-	 * We don't bother re-evaluating clamshell state. If the system is awake,
-	 * the lid is probably open.
-	 */
-	if (msg & kIOPMDisableClamshell) {
-		DLOG("Clamshell disabled\n");
-		clamshellDisabled = true;
-		sendClientClamshellNotification();
-	}
-
-	/*
-	 * Evaluate clamshell and SLEEP if appropriate
-	 */
-	if (eval_clamshell_alarm && clamshellClosed) {
-		if (shouldSleepOnRTCAlarmWake()) {
-			privateSleepSystem(kIOPMSleepReasonClamshell);
-		}
-	} else if (eval_clamshell && clamshellClosed) {
-		if (shouldSleepOnClamshellClosed()) {
-			privateSleepSystem(kIOPMSleepReasonClamshell);
-		} else {
-			evaluatePolicy( kStimulusDarkWakeEvaluate );
-		}
-	}
-
-	if (msg & kIOPMProModeEngaged) {
-		int newState = 1;
-		DLOG("ProModeEngaged\n");
-		messageClient(kIOPMMessageProModeStateChange, systemCapabilityNotifier.get(), &newState, sizeof(newState));
-	}
-
-	if (msg & kIOPMProModeDisengaged) {
-		int newState = 0;
-		DLOG("ProModeDisengaged\n");
-		messageClient(kIOPMMessageProModeStateChange, systemCapabilityNotifier.get(), &newState, sizeof(newState));
-	}
+    }
+    
+    /*
+     * Enable Clamshell (external display disappear)
+     *
+     *  -> reevaluate lid state
+     */
+    if (msg & kIOPMEnableClamshell) 
+    {
+        // Re-evaluate the lid state
+        // System should sleep on external display disappearance
+        // in lid closed operation.
+        if( clamshellClosed && (true == clamshellDisabled) )        
+        {
+            eval_clamshell = true;
+        }
+
+        clamshellDisabled = false;
+
+        sendClientClamshellNotification();
+    }
+    
+    /*
+     * Disable Clamshell (external display appeared)
+     * We don't bother re-evaluating clamshell state. If the system is awake,
+     * the lid is probably open. 
+     */
+    if (msg & kIOPMDisableClamshell) 
+    {
+        clamshellDisabled = true;
+
+        sendClientClamshellNotification();
+    }
+
+    /*
+     * Evaluate clamshell and SLEEP if appropiate
+     */
+    if ( eval_clamshell && shouldSleepOnClamshellClosed() ) 
+    {
+
+
+        privateSleepSystem (kIOPMSleepReasonClamshell);
+    }
+    else if ( eval_clamshell )
+    {
+        evaluatePolicy( kStimulusDarkWakeEvaluate );
+    }
+
+    /*
+     * Power Button
+     */
+    if (msg & kIOPMPowerButton) 
+    {
+        if (!wranglerAsleep)
+        {
+            OSString *pbs = OSString::withCString("DisablePowerButtonSleep");
+            // Check that power button sleep is enabled
+            if( pbs ) {
+                if( kOSBooleanTrue != getProperty(pbs))
+                    privateSleepSystem (kIOPMSleepReasonPowerButton);
+            }
+        }
+        else
+            reportUserInput();
+    }
 }
 
 //******************************************************************************
@@ -9086,606 +5756,438 @@
 // Evaluate root-domain policy in response to external changes.
 //******************************************************************************
 
-void
-IOPMrootDomain::evaluatePolicy( int stimulus, uint32_t arg )
-{
-	union {
-		struct {
-			int idleSleepEnabled    : 1;
-			int idleSleepDisabled   : 1;
-			int displaySleep        : 1;
-			int sleepDelayChanged   : 1;
-			int evaluateDarkWake    : 1;
-			int adjustPowerState    : 1;
-			int userBecameInactive  : 1;
-			int displaySleepEntry   : 1;
-		} bit;
-		uint32_t u32;
-	} flags;
-
-
-	ASSERT_GATED();
-	flags.u32 = 0;
-
-	switch (stimulus) {
-	case kStimulusDisplayWranglerSleep:
-		DLOG("evaluatePolicy( %d, 0x%x )\n", stimulus, arg);
-		if (!wranglerPowerOff) {
-			// wrangler is in sleep state or lower
-			flags.bit.displaySleep = true;
-		}
-		if (!wranglerAsleep) {
-			// transition from wrangler wake to wrangler sleep
-			flags.bit.displaySleepEntry = true;
-			wranglerAsleep = true;
-		}
-		break;
-
-	case kStimulusDisplayWranglerWake:
-		DLOG("evaluatePolicy( %d, 0x%x )\n", stimulus, arg);
-		displayIdleForDemandSleep = false;
-		wranglerPowerOff = false;
-		wranglerAsleep = false;
-		break;
-
-	case kStimulusEnterUserActiveState:
-		DLOG("evaluatePolicy( %d, 0x%x )\n", stimulus, arg);
-		if (_preventUserActive) {
-			DLOG("user active dropped\n");
-			break;
-		}
-		if (!userIsActive) {
-			userIsActive = true;
-			userWasActive = true;
-			clock_get_uptime(&gUserActiveAbsTime);
-
-			// Stay awake after dropping demand for display power on
-			if (kFullWakeReasonDisplayOn == fullWakeReason) {
-				fullWakeReason = fFullWakeReasonDisplayOnAndLocalUser;
-				DLOG("User activity while in notification wake\n");
-				changePowerStateWithOverrideTo( getRUN_STATE(), 0);
-			}
-
-			kdebugTrace(kPMLogUserActiveState, 0, 1, 0);
-			setProperty(gIOPMUserIsActiveKey.get(), kOSBooleanTrue);
-			messageClients(kIOPMMessageUserIsActiveChanged);
-		}
-		flags.bit.idleSleepDisabled = true;
-		break;
-
-	case kStimulusLeaveUserActiveState:
-		DLOG("evaluatePolicy( %d, 0x%x )\n", stimulus, arg);
-		if (userIsActive) {
-			clock_get_uptime(&gUserInactiveAbsTime);
-			userIsActive = false;
-			clock_get_uptime(&userBecameInactiveTime);
-			flags.bit.userBecameInactive = true;
-
-			kdebugTrace(kPMLogUserActiveState, 0, 0, 0);
-			setProperty(gIOPMUserIsActiveKey.get(), kOSBooleanFalse);
-			messageClients(kIOPMMessageUserIsActiveChanged);
-		}
-		break;
-
-	case kStimulusAggressivenessChanged:
-	{
-		DMSG("evaluatePolicy( %d, 0x%x )\n", stimulus, arg);
-		unsigned long   aggressiveValue;
-		uint32_t        minutesToIdleSleep  = 0;
-		uint32_t        minutesToDisplayDim = 0;
-		uint32_t        minutesDelta        = 0;
-
-		// Fetch latest display and system sleep slider values.
-		aggressiveValue = 0;
-		getAggressiveness(kPMMinutesToSleep, &aggressiveValue);
-		minutesToIdleSleep = (uint32_t) aggressiveValue;
-
-		aggressiveValue = 0;
-		getAggressiveness(kPMMinutesToDim, &aggressiveValue);
-		minutesToDisplayDim = (uint32_t) aggressiveValue;
-		DLOG("aggressiveness changed: system %u->%u, display %u\n",
-		    sleepSlider, minutesToIdleSleep, minutesToDisplayDim);
-
-		DLOG("idle time -> %d ms (ena %d)\n",
-		    idleMilliSeconds, (minutesToIdleSleep != 0));
-
-		// How long to wait before sleeping the system once
-		// the displays turns off is indicated by 'extraSleepDelay'.
-
-		if (minutesToIdleSleep > minutesToDisplayDim) {
-			minutesDelta = minutesToIdleSleep - minutesToDisplayDim;
-		} else if (minutesToIdleSleep == minutesToDisplayDim) {
-			minutesDelta = 1;
-		}
-
-		if ((!idleSleepEnabled) && (minutesToIdleSleep != 0)) {
-			idleSleepEnabled = flags.bit.idleSleepEnabled = true;
-		}
-
-		if ((idleSleepEnabled) && (minutesToIdleSleep == 0)) {
-			flags.bit.idleSleepDisabled = true;
-			idleSleepEnabled = false;
-		}
-#if !defined(XNU_TARGET_OS_OSX)
-		if (0x7fffffff == minutesToIdleSleep) {
-			minutesToIdleSleep = idleMilliSeconds / 1000;
-		}
-#endif /* !defined(XNU_TARGET_OS_OSX) */
-
-		if (((minutesDelta != extraSleepDelay) ||
-		    (userActivityTime != userActivityTime_prev)) &&
-		    !flags.bit.idleSleepEnabled && !flags.bit.idleSleepDisabled) {
-			flags.bit.sleepDelayChanged = true;
-		}
-
-		if (systemDarkWake && !darkWakeToSleepASAP &&
-		    (flags.bit.idleSleepEnabled || flags.bit.idleSleepDisabled)) {
-			// Reconsider decision to remain in dark wake
-			flags.bit.evaluateDarkWake = true;
-		}
-
-		sleepSlider = minutesToIdleSleep;
-		extraSleepDelay = minutesDelta;
-		userActivityTime_prev = userActivityTime;
-	}   break;
-
-	case kStimulusDemandSystemSleep:
-		DLOG("evaluatePolicy( %d, 0x%x )\n", stimulus, arg);
-		displayIdleForDemandSleep = true;
-		if (wrangler && wranglerIdleSettings) {
-			// Request wrangler idle only when demand sleep is triggered
-			// from full wake.
-			if (CAP_CURRENT(kIOPMSystemCapabilityGraphics)) {
-				wrangler->setProperties(wranglerIdleSettings.get());
-				DLOG("Requested wrangler idle\n");
-			}
-		}
-		// arg = sleepReason
-		changePowerStateWithOverrideTo( SLEEP_STATE, arg );
-		break;
-
-	case kStimulusAllowSystemSleepChanged:
-		DLOG("evaluatePolicy( %d, 0x%x )\n", stimulus, arg);
-		flags.bit.adjustPowerState = true;
-		break;
-
-	case kStimulusDarkWakeActivityTickle:
-		DLOG("evaluatePolicy( %d, 0x%x )\n", stimulus, arg);
-		// arg == true implies real and not self generated wrangler tickle.
-		// Update wake type on PM work loop instead of the tickle thread to
-		// eliminate the possibility of an early tickle clobbering the wake
-		// type set by the platform driver.
-		if (arg == true) {
-			setProperty(kIOPMRootDomainWakeTypeKey, kIOPMRootDomainWakeTypeHIDActivity);
-		}
-
-		if (!darkWakeExit) {
-			if (latchDisplayWranglerTickle(true)) {
-				DLOG("latched tickle\n");
-				break;
-			}
-
-			darkWakeExit = true;
-			DLOG("Requesting full wake due to dark wake activity tickle\n");
-			requestFullWake( kFullWakeReasonLocalUser );
-		}
-		break;
-
-	case kStimulusDarkWakeEntry:
-	case kStimulusDarkWakeReentry:
-		DLOG("evaluatePolicy( %d, 0x%x )\n", stimulus, arg);
-		// Any system transitions since the last dark wake transition
-		// will invalid the stimulus.
-
-		if (arg == _systemStateGeneration) {
-			DLOG("dark wake entry\n");
-			systemDarkWake = true;
-
-			// Keep wranglerPowerOff an invariant when wrangler is absent
-			if (wrangler) {
-				wranglerPowerOff = true;
-			}
-
-			if (kStimulusDarkWakeEntry == stimulus) {
-				clock_get_uptime(&userBecameInactiveTime);
-				flags.bit.evaluateDarkWake = true;
-				if (activitySinceSleep()) {
-					DLOG("User activity recorded while going to darkwake\n");
-					reportUserInput();
-				}
-			}
-
-			// Always accelerate disk spindown while in dark wake,
-			// even if system does not support/allow sleep.
-
-			cancelIdleSleepTimer();
-			setQuickSpinDownTimeout();
-		}
-		break;
-
-	case kStimulusDarkWakeEvaluate:
-		DMSG("evaluatePolicy( %d, 0x%x )\n", stimulus, arg);
-		if (systemDarkWake) {
-			flags.bit.evaluateDarkWake = true;
-		}
-		break;
-
-	case kStimulusNoIdleSleepPreventers:
-		DMSG("evaluatePolicy( %d, 0x%x )\n", stimulus, arg);
-		flags.bit.adjustPowerState = true;
-		break;
-	} /* switch(stimulus) */
-
-	if (flags.bit.evaluateDarkWake && (kFullWakeReasonNone == fullWakeReason)) {
-		DLOG("DarkWake: sleepASAP %d, clamshell closed %d, disabled %d/%x, desktopMode %d, ac %d\n",
-		    darkWakeToSleepASAP, clamshellClosed, clamshellDisabled, clamshellSleepDisableMask, desktopMode, acAdaptorConnected);
-		if (darkWakeToSleepASAP ||
-		    (clamshellClosed && !(desktopMode && acAdaptorConnected))) {
-			uint32_t newSleepReason;
-
-			if (CAP_HIGHEST(kIOPMSystemCapabilityGraphics)) {
-				// System was previously in full wake. Sleep reason from
-				// full to dark already recorded in fullToDarkReason.
-
-				if (lowBatteryCondition) {
-					newSleepReason = kIOPMSleepReasonLowPower;
-				} else if (thermalEmergencyState) {
-					newSleepReason = kIOPMSleepReasonThermalEmergency;
-				} else {
-					newSleepReason = fullToDarkReason;
-				}
-			} else {
-				// In dark wake from system sleep.
-
-				if (darkWakeSleepService) {
-					newSleepReason = kIOPMSleepReasonSleepServiceExit;
-				} else {
-					newSleepReason = kIOPMSleepReasonMaintenance;
-				}
-			}
-
-			if (checkSystemCanSleep(newSleepReason)) {
-				privateSleepSystem(newSleepReason);
-			}
-		} else { // non-maintenance (network) dark wake
-			if (checkSystemCanSleep(kIOPMSleepReasonIdle)) {
-				// Release power clamp, and wait for children idle.
-				adjustPowerState(true);
-			} else {
-				changePowerStateWithTagToPriv(getRUN_STATE(), kCPSReasonDarkWakeCannotSleep);
-			}
-		}
-	}
-
-	if (systemDarkWake) {
-		// The rest are irrelevant while system is in dark wake.
-		flags.u32 = 0;
-	}
-
-	if ((flags.bit.displaySleepEntry) &&
-	    (kFullWakeReasonDisplayOn == fullWakeReason)) {
-		// kIOPMSleepReasonNotificationWakeExit
-		DLOG("Display sleep while in notification wake\n");
-		changePowerStateWithOverrideTo(SLEEP_STATE, kIOPMSleepReasonNotificationWakeExit);
-	}
-
-	if (flags.bit.userBecameInactive || flags.bit.sleepDelayChanged) {
-		bool cancelQuickSpindown = false;
-
-		if (flags.bit.sleepDelayChanged) {
-			// Cancel existing idle sleep timer and quick disk spindown.
-			// New settings will be applied by the idleSleepEnabled flag
-			// handler below if idle sleep is enabled.
-
-			DLOG("extra sleep timer changed\n");
-			cancelIdleSleepTimer();
-			cancelQuickSpindown = true;
-		} else {
-			DLOG("user inactive\n");
-		}
-
-		if (!userIsActive && idleSleepEnabled) {
-			startIdleSleepTimer(getTimeToIdleSleep());
-		}
-
-		if (cancelQuickSpindown) {
-			restoreUserSpinDownTimeout();
-		}
-	}
-
-	if (flags.bit.idleSleepEnabled) {
-		DLOG("idle sleep timer enabled\n");
-		if (!wrangler) {
-#if defined(XNU_TARGET_OS_OSX) && !DISPLAY_WRANGLER_PRESENT
-			startIdleSleepTimer(getTimeToIdleSleep());
-#else
-			changePowerStateWithTagToPriv(getRUN_STATE(), kCPSReasonIdleSleepEnabled);
-			startIdleSleepTimer( idleMilliSeconds );
+void IOPMrootDomain::evaluatePolicy( int stimulus, uint32_t arg )
+{
+    union {
+        struct {
+            int idleSleepEnabled    : 1;
+            int idleSleepDisabled   : 1;
+            int displaySleep        : 1;
+            int sleepDelayChanged   : 1;
+            int evaluateDarkWake    : 1;
+            int adjustPowerState    : 1;
+        } bit;
+        uint32_t u32;
+    } flags;
+
+    DLOG("evaluatePolicy( %d, 0x%x )\n", stimulus, arg);
+
+    ASSERT_GATED();
+    flags.u32 = 0;
+
+    switch (stimulus)
+    {
+        case kStimulusDisplayWranglerSleep:
+            if (!wranglerAsleep)
+            {
+                wranglerAsleep = true;
+                clock_get_uptime(&wranglerSleepTime);
+                flags.bit.displaySleep = true;
+            }
+            break;
+
+        case kStimulusDisplayWranglerWake:
+            wranglerAsleep = false;
+            flags.bit.idleSleepDisabled = true;
+            break;
+
+        case kStimulusAggressivenessChanged:
+        {
+            unsigned long   minutesToIdleSleep  = 0;
+            unsigned long   minutesToDisplayDim = 0;
+            unsigned long   minutesDelta        = 0;
+
+            // Fetch latest display and system sleep slider values.
+            getAggressiveness(kPMMinutesToSleep, &minutesToIdleSleep);
+            getAggressiveness(kPMMinutesToDim,   &minutesToDisplayDim);
+            DLOG("aggressiveness changed: system %u->%u, display %u\n",
+                (uint32_t) sleepSlider,
+                (uint32_t) minutesToIdleSleep,
+                (uint32_t) minutesToDisplayDim);
+
+            DLOG("idle time -> %ld secs (ena %d)\n",
+                idleSeconds, (minutesToIdleSleep != 0));
+
+            if (0x7fffffff == minutesToIdleSleep)
+                minutesToIdleSleep = idleSeconds;
+
+            // How long to wait before sleeping the system once
+            // the displays turns off is indicated by 'extraSleepDelay'.
+
+            if ( minutesToIdleSleep > minutesToDisplayDim )
+                minutesDelta = minutesToIdleSleep - minutesToDisplayDim;
+            else if( minutesToIdleSleep == minutesToDisplayDim )
+                minutesDelta = 1;
+
+            if ((sleepSlider == 0) && (minutesToIdleSleep != 0))
+                flags.bit.idleSleepEnabled = true;
+
+            if ((sleepSlider != 0) && (minutesToIdleSleep == 0))
+                flags.bit.idleSleepDisabled = true;
+
+            if ((minutesDelta != extraSleepDelay) &&
+                !flags.bit.idleSleepEnabled && !flags.bit.idleSleepDisabled)
+                flags.bit.sleepDelayChanged = true;
+
+            if (systemDarkWake && !darkWakeToSleepASAP &&
+                (flags.bit.idleSleepEnabled || flags.bit.idleSleepDisabled))
+            {
+                // Reconsider decision to remain in dark wake
+                flags.bit.evaluateDarkWake = true;
+            }
+
+            sleepSlider = minutesToIdleSleep;
+            extraSleepDelay = minutesDelta;
+        }   break;
+
+        case kStimulusDemandSystemSleep:
+            changePowerStateWithOverrideTo( SLEEP_STATE );
+            break;
+
+        case kStimulusAllowSystemSleepChanged:
+            flags.bit.adjustPowerState = true;
+            break;
+
+        case kStimulusDarkWakeActivityTickle:
+            if (false == wranglerTickled)
+            {
+                uint32_t    options = 0;
+                IOService * pciRoot = 0;
+
+                if (rejectWranglerTickle)
+                {
+                    DLOG("rejected tickle, type %u capability %x:%x\n",
+                        _systemTransitionType,
+                        _currentCapability, _pendingCapability);
+                    break;
+                }
+
+                if (latchDisplayWranglerTickle(true))
+                {
+                    DLOG("latched tickle\n");
+                    break;
+                }
+
+                _desiredCapability |=
+                    (kIOPMSystemCapabilityGraphics |
+                     kIOPMSystemCapabilityAudio);
+                
+                if ((kSystemTransitionWake == _systemTransitionType) &&
+                    !(_pendingCapability & kIOPMSystemCapabilityGraphics) &&
+                    !graphicsSuppressed)
+                {
+                    DLOG("Promoting to full wake\n");
+
+                    // Elevate to full wake while waking up to dark wake.
+                    // PM will hold off notifying the graphics subsystem about
+                    // system wake as late as possible, so if a HID event does
+                    // arrive, we can turn on graphics on this wake cycle, and
+                    // not have to wait till the following cycle. That latency
+                    // can be huge on some systems. However, once any graphics
+                    // suppression has taken effect, it is too late. All other
+                    // graphics devices must be similarly suppressed. But the
+                    // delay till the following cycle should be very short.
+
+                    _pendingCapability |=
+                        (kIOPMSystemCapabilityGraphics |
+                         kIOPMSystemCapabilityAudio);
+
+                    // Immediately bring up audio and graphics.
+                    pciRoot = pciHostBridgeDriver;
+
+                    // Notify clients about full wake.
+                    _systemMessageClientMask = kSystemMessageClientAll;
+                    tellClients(kIOMessageSystemWillPowerOn);
+                }
+
+                // Unsafe to cancel once graphics was powered.
+                // If system woke from dark wake, the return to sleep can
+                // be cancelled. But "awake -> dark -> sleep" transition
+                // cannot be cancelled.
+                
+                if (!CAP_HIGHEST(kIOPMSystemCapabilityGraphics)) {
+                    options |= kIOPMSyncCancelPowerDown;                    
+                }
+
+                synchronizePowerTree( options, pciRoot );
+                wranglerTickled = true;
+                // IOGraphics doesn't lit the display even though graphics
+                // is enanbled in kIOMessageSystemCapabilityChange message(radar 9502104)
+                // So, do an explicit activity tickle
+                if(wrangler)
+                    wrangler->activityTickle(0,0);
+
+                if (logWranglerTickle)
+                {
+                    AbsoluteTime    now;
+                    uint64_t        nsec;
+
+                    clock_get_uptime(&now);
+                    SUB_ABSOLUTETIME(&now, &systemWakeTime);
+                    absolutetime_to_nanoseconds(now, &nsec);
+                    MSG("HID tickle %u ms\n",
+                        ((int)((nsec) / 1000000ULL)));
+                    logWranglerTickle = false;
+                }
+            }
+            break;
+
+        case kStimulusDarkWakeEntry:
+        case kStimulusDarkWakeReentry:
+            // Any system transitions since the last dark wake transition
+            // will invalid the stimulus.
+
+            if (arg == _systemStateGeneration)
+            {
+                DLOG("dark wake entry\n");
+                systemDarkWake = true;
+                wranglerAsleep = true;
+                clock_get_uptime(&wranglerSleepTime);
+
+                // Always accelerate disk spindown while in dark wake,
+                // even if system does not support/allow sleep.
+
+                cancelIdleSleepTimer();
+                setQuickSpinDownTimeout();
+                flags.bit.evaluateDarkWake = true;
+            }
+            break;
+
+        case kStimulusDarkWakeEvaluate:
+            if (systemDarkWake)
+            {
+                flags.bit.evaluateDarkWake = true;
+            }
+#if !DARK_TO_FULL_EVALUATE_CLAMSHELL
+            else
+            {
+                // Not through kLocalEvalClamshellCommand to avoid loop.
+                if (clamshellClosed && shouldSleepOnClamshellClosed() &&
+                    checkSystemCanSleep(true))
+                {
+                    privateSleepSystem( kIOPMSleepReasonClamshell );
+                }
+            }
 #endif
-		} else {
-			// Start idle timer if prefs now allow system sleep
-			// and user is already inactive. Disk spindown is
-			// accelerated upon timer expiration.
-
-			if (!userIsActive) {
-				startIdleSleepTimer(getTimeToIdleSleep());
-			}
-		}
-	}
-
-	if (flags.bit.idleSleepDisabled) {
-		DLOG("idle sleep timer disabled\n");
-		cancelIdleSleepTimer();
-		restoreUserSpinDownTimeout();
-		adjustPowerState();
-	}
-
-	if (flags.bit.adjustPowerState) {
-		bool sleepASAP = false;
-
-		if (!systemBooting && (0 == idleSleepPreventersCount())) {
-			if (!wrangler) {
-				if (kStimulusNoIdleSleepPreventers != stimulus) {
-					changePowerStateWithTagToPriv(getRUN_STATE(), kCPSReasonEvaluatePolicy);
-				}
-				if (idleSleepEnabled) {
-#if defined(XNU_TARGET_OS_OSX) && !DISPLAY_WRANGLER_PRESENT
-					if (!extraSleepDelay && !idleSleepTimerPending && !gNoIdleFlag) {
-						sleepASAP = true;
-					}
-#else
-					// stay awake for at least idleMilliSeconds
-					startIdleSleepTimer(idleMilliSeconds);
-#endif
-				}
-			} else if (!extraSleepDelay && !idleSleepTimerPending && !systemDarkWake && !gNoIdleFlag) {
-				sleepASAP = true;
-			}
-		}
-
-		adjustPowerState(sleepASAP);
-	}
-}
-
-//******************************************************************************
-
-unsigned int
-IOPMrootDomain::idleSleepPreventersCount()
-{
-	if (_aotMode) {
-		unsigned int count __block;
-		count = 0;
-		preventIdleSleepList->iterateObjects(^bool (OSObject * obj)
-		{
-			count += (NULL == obj->metaCast("AppleARMBacklight"));
-			return false;
-		});
-		return count;
-	}
-
-	return preventIdleSleepList->getCount();
-}
-
-
-//******************************************************************************
-// requestFullWake
-//
-// Request transition from dark wake to full wake
-//******************************************************************************
-
-void
-IOPMrootDomain::requestFullWake( FullWakeReason reason )
-{
-	uint32_t        options = 0;
-	IOService *     pciRoot = NULL;
-	bool            promotion = false;
-
-	// System must be in dark wake and a valid reason for entering full wake
-	if ((kFullWakeReasonNone == reason) ||
-	    (kFullWakeReasonNone != fullWakeReason) ||
-	    (CAP_CURRENT(kIOPMSystemCapabilityGraphics))) {
-		return;
-	}
-
-	// Will clear reason upon exit from full wake
-	fullWakeReason = reason;
-
-	_desiredCapability |= (kIOPMSystemCapabilityGraphics |
-	    kIOPMSystemCapabilityAudio);
-
-	if ((kSystemTransitionWake == _systemTransitionType) &&
-	    !(_pendingCapability & kIOPMSystemCapabilityGraphics) &&
-	    !darkWakePowerClamped) {
-		// Promote to full wake while waking up to dark wake due to tickle.
-		// PM will hold off notifying the graphics subsystem about system wake
-		// as late as possible, so if a HID tickle does arrive, graphics can
-		// power up from this same wake transition. Otherwise, the latency to
-		// power up graphics on the following transition can be huge on certain
-		// systems. However, once any power clamping has taken effect, it is
-		// too late to promote the current dark wake transition to a full wake.
-		_pendingCapability |= (kIOPMSystemCapabilityGraphics |
-		    kIOPMSystemCapabilityAudio);
-
-		// Tell the PCI parent of audio and graphics drivers to stop
-		// delaying the child notifications. Same for root domain.
-		pciRoot = pciHostBridgeDriver.get();
-		willEnterFullWake();
-		promotion = true;
-	}
-
-	// Unsafe to cancel once graphics was powered.
-	// If system woke from dark wake, the return to sleep can
-	// be cancelled. "awake -> dark -> sleep" transition
-	// can be cancelled also, during the "dark -> sleep" phase
-	// *prior* to driver power down.
-	if (!CAP_HIGHEST(kIOPMSystemCapabilityGraphics) ||
-	    _pendingCapability == 0) {
-		options |= kIOPMSyncCancelPowerDown;
-	}
-
-	synchronizePowerTree(options, pciRoot);
-
-	if (kFullWakeReasonLocalUser == fullWakeReason) {
-		// IOGraphics doesn't light the display even though graphics is
-		// enabled in kIOMessageSystemCapabilityChange message(radar 9502104)
-		// So, do an explicit activity tickle
-		if (wrangler) {
-			wrangler->activityTickle(0, 0);
-		}
-	}
-
-	// Log a timestamp for the initial full wake request.
-	// System may not always honor this full wake request.
-	if (!CAP_HIGHEST(kIOPMSystemCapabilityGraphics)) {
-		AbsoluteTime    now;
-		uint64_t        nsec;
-
-		clock_get_uptime(&now);
-		SUB_ABSOLUTETIME(&now, &gIOLastWakeAbsTime);
-		absolutetime_to_nanoseconds(now, &nsec);
-		MSG("full wake %s (reason %u) %u ms\n",
-		    promotion ? "promotion" : "request",
-		    fullWakeReason, ((int)((nsec) / NSEC_PER_MSEC)));
-	}
-}
-
-//******************************************************************************
-// willEnterFullWake
-//
-// System will enter full wake from sleep, from dark wake, or from dark
-// wake promotion. This function aggregate things that are in common to
-// all three full wake transitions.
-//
-// Assumptions: fullWakeReason was updated
-//******************************************************************************
-
-void
-IOPMrootDomain::willEnterFullWake( void )
-{
-	hibernateRetry = false;
-	sleepToStandby = false;
-	standbyNixed   = false;
-	resetTimers    = false;
-	sleepTimerMaintenance = false;
-
-	assert(!CAP_CURRENT(kIOPMSystemCapabilityGraphics));
-
-	_systemMessageClientMask = kSystemMessageClientPowerd |
-	    kSystemMessageClientLegacyApp;
-
-	if ((_highestCapability & kIOPMSystemCapabilityGraphics) == 0) {
-		// First time to attain full wake capability since the last wake
-		_systemMessageClientMask |= kSystemMessageClientKernel;
-
-		// Set kIOPMUserTriggeredFullWakeKey before full wake for IOGraphics
-		setProperty(gIOPMUserTriggeredFullWakeKey.get(),
-		    (kFullWakeReasonLocalUser == fullWakeReason) ?
-		    kOSBooleanTrue : kOSBooleanFalse);
-	}
-#if HIBERNATION
-	IOHibernateSetWakeCapabilities(_pendingCapability);
-#endif
-
-	IOService::setAdvisoryTickleEnable( true );
-	tellClients(kIOMessageSystemWillPowerOn);
-	preventTransitionToUserActive(false);
-}
-
-//******************************************************************************
-// fullWakeDelayedWork
-//
-// System has already entered full wake. Invoked by a delayed thread call.
-//******************************************************************************
-
-void
-IOPMrootDomain::fullWakeDelayedWork( void )
-{
-#if DARK_TO_FULL_EVALUATE_CLAMSHELL_DELAY
-	if (!gIOPMWorkLoop->inGate()) {
-		gIOPMWorkLoop->runAction(
-			OSMemberFunctionCast(IOWorkLoop::Action, this,
-			&IOPMrootDomain::fullWakeDelayedWork), this);
-		return;
-	}
-
-	DLOG("fullWakeDelayedWork cap cur %x pend %x high %x, clamshell disable %x/%x\n",
-	    _currentCapability, _pendingCapability, _highestCapability,
-	    clamshellDisabled, clamshellSleepDisableMask);
-
-	if (clamshellExists &&
-	    CAP_CURRENT(kIOPMSystemCapabilityGraphics) &&
-	    !CAP_CHANGE(kIOPMSystemCapabilityGraphics)) {
-		if (clamshellSleepDisableMask & kClamshellSleepDisableInternal) {
-			setClamShellSleepDisable(false, kClamshellSleepDisableInternal);
-		} else {
-			// Not the initial full wake after waking from sleep.
-			// Evaluate the clamshell for rdar://problem/9157444.
-			receivePowerNotification(kLocalEvalClamshellCommand);
-		}
-	}
-#endif
+            break;
+
+        case kStimulusNoIdleSleepPreventers:
+            flags.bit.adjustPowerState = true;
+            break;
+
+    } /* switch(stimulus) */
+
+    if (flags.bit.evaluateDarkWake && !wranglerTickled)
+    {
+        if (darkWakeToSleepASAP ||
+            (clamshellClosed && !(desktopMode && acAdaptorConnected)))
+        {
+            // System currently in dark wake, and no children and
+            // assertion prevent system sleep.
+
+            if (checkSystemCanSleep(true))
+            {
+                if (lowBatteryCondition)
+                {
+                    lastSleepReason = kIOPMSleepReasonLowPower;
+                    setProperty(kRootDomainSleepReasonKey, kIOPMLowPowerSleepKey);
+                }
+                else  if (darkWakeMaintenance)
+                {
+                    lastSleepReason = kIOPMSleepReasonMaintenance;
+                    setProperty(kRootDomainSleepReasonKey, kIOPMMaintenanceSleepKey);
+                } 
+                else if (darkWakeSleepService)
+                {
+                    lastSleepReason = kIOPMSleepReasonSleepServiceExit;
+                    setProperty(kRootDomainSleepReasonKey, kIOPMSleepServiceExitKey);
+                }
+                changePowerStateWithOverrideTo( SLEEP_STATE );
+            }
+            else
+            {
+                // Parked in dark wake, a tickle will return to full wake
+                rejectWranglerTickle = false;
+            }
+        }
+        else // non-maintenance (network) dark wake
+        {
+            if (checkSystemCanSleep(true))
+            {
+                // Release power clamp, and wait for children idle.
+                adjustPowerState(true);
+            }
+            else
+            {
+                changePowerStateToPriv(ON_STATE);
+            }
+            rejectWranglerTickle = false;
+        }
+    }
+
+    if (systemDarkWake)
+    {
+        // The rest are irrelevant while system is in dark wake.
+        flags.u32 = 0;
+    }
+
+    if (flags.bit.displaySleep || flags.bit.sleepDelayChanged)
+    {
+        bool cancelQuickSpindown = false;
+
+        if (flags.bit.sleepDelayChanged)
+        {
+            DLOG("extra sleep timer changed\n");
+            cancelIdleSleepTimer();
+            cancelQuickSpindown = true;
+        }
+        else
+        {
+            DLOG("display sleep\n");        
+        }
+
+        if (wranglerAsleep && !wranglerSleepIgnored)
+        {
+            if ( extraSleepDelay )
+            {
+                // Start a timer here if the System Sleep timer is greater
+                // than the Display Sleep timer.
+
+                startIdleSleepTimer(gRootDomain->extraSleepDelay * 60);            
+            }
+            else if ( sleepSlider )
+            {
+                // Accelerate disk spindown if system sleep and display sleep
+                // sliders are set to the same value (e.g. both set to 5 min),
+                // and display is about to go dark. Check the system sleep is
+                // not set to never sleep. Disk sleep setting is ignored.
+
+                setQuickSpinDownTimeout();
+                cancelQuickSpindown = false;
+            }
+        }
+        
+        if (cancelQuickSpindown)
+            restoreUserSpinDownTimeout();
+    }
+
+    if (flags.bit.idleSleepEnabled)
+    {
+        DLOG("idle sleep timer enabled\n");
+        if (!wrangler)
+        {
+            changePowerStateToPriv(ON_STATE);
+            if (idleSeconds)
+            {
+                startIdleSleepTimer( idleSeconds );
+            }
+        }
+        else
+        {
+            // Start idle sleep timer if wrangler went to sleep
+            // while system sleep was disabled. Disk spindown is
+            // accelerated upon timer expiration.
+
+            if (wranglerAsleep)
+            {
+                AbsoluteTime    now;
+                uint64_t        nanos;
+                uint32_t        minutesSinceDisplaySleep = 0;
+                uint32_t        sleepDelay = 0;
+
+                clock_get_uptime(&now);
+                if (CMP_ABSOLUTETIME(&now, &wranglerSleepTime) > 0)
+                {
+                    SUB_ABSOLUTETIME(&now, &wranglerSleepTime);
+                    absolutetime_to_nanoseconds(now, &nanos);
+                    minutesSinceDisplaySleep = nanos / (60000000000ULL);
+                }
+
+                if (extraSleepDelay > minutesSinceDisplaySleep)
+                {
+                    sleepDelay = extraSleepDelay - minutesSinceDisplaySleep;
+                }
+
+                startIdleSleepTimer(sleepDelay * 60);
+                DLOG("display slept %u min, set idle timer to %u min\n",
+                    minutesSinceDisplaySleep, sleepDelay);
+            }
+        }
+    }
+
+    if (flags.bit.idleSleepDisabled)
+    {
+        DLOG("idle sleep timer disabled\n");
+        cancelIdleSleepTimer();
+        restoreUserSpinDownTimeout();
+        adjustPowerState();
+    }
+
+    if (flags.bit.adjustPowerState)
+    {
+        bool sleepASAP = false;
+
+        if (!systemBooting && (preventIdleSleepList->getCount() == 0))
+        {
+            if (!wrangler)
+            {
+                changePowerStateToPriv(ON_STATE);
+                if (idleSeconds)
+                {
+                    // stay awake for at least idleSeconds
+                    startIdleSleepTimer(idleSeconds);
+                }
+            }
+            else if (!extraSleepDelay && !idleSleepTimerPending && !systemDarkWake)
+            {
+                sleepASAP = true;
+            }
+        }
+        if(sleepASAP)
+        {
+            lastSleepReason = kIOPMSleepReasonIdle;
+            setProperty(kRootDomainSleepReasonKey, kIOPMIdleSleepKey);
+        }
+
+        adjustPowerState(sleepASAP);
+    }
 }
 
 //******************************************************************************
 // evaluateAssertions
 //
 //******************************************************************************
-
-// Bitmask of all kernel assertions that prevent system idle sleep.
-// kIOPMDriverAssertionReservedBit7 is reserved for IOMediaBSDClient.
-#define NO_IDLE_SLEEP_ASSERTIONS_MASK \
-	(kIOPMDriverAssertionReservedBit7 | \
-	 kIOPMDriverAssertionPreventSystemIdleSleepBit)
-
-void
-IOPMrootDomain::evaluateAssertions(IOPMDriverAssertionType newAssertions, IOPMDriverAssertionType oldAssertions)
-{
-	IOPMDriverAssertionType changedBits = newAssertions ^ oldAssertions;
-
-	messageClients(kIOPMMessageDriverAssertionsChanged);
-
-	if (changedBits & kIOPMDriverAssertionPreventDisplaySleepBit) {
-		if (wrangler) {
-			bool value = (newAssertions & kIOPMDriverAssertionPreventDisplaySleepBit) ? true : false;
-
-			DLOG("wrangler->setIgnoreIdleTimer\(%d)\n", value);
-			wrangler->setIgnoreIdleTimer( value );
-		}
-	}
-
-	if (changedBits & kIOPMDriverAssertionCPUBit) {
-		if (_aotNow) {
-			IOLog("CPU assertions %d\n", (0 != (kIOPMDriverAssertionCPUBit & newAssertions)));
-		}
-		evaluatePolicy(_aotNow ? kStimulusNoIdleSleepPreventers : kStimulusDarkWakeEvaluate);
-		if (!assertOnWakeSecs && gIOLastWakeAbsTime) {
-			AbsoluteTime    now;
-			clock_usec_t    microsecs;
-			clock_get_uptime(&now);
-			SUB_ABSOLUTETIME(&now, &gIOLastWakeAbsTime);
-			absolutetime_to_microtime(now, &assertOnWakeSecs, &microsecs);
-			if (assertOnWakeReport) {
-				HISTREPORT_TALLYVALUE(assertOnWakeReport, (int64_t)assertOnWakeSecs);
-				DLOG("Updated assertOnWake %lu\n", (unsigned long)assertOnWakeSecs);
-			}
-		}
-	}
-
-	if (changedBits & NO_IDLE_SLEEP_ASSERTIONS_MASK) {
-		if ((newAssertions & NO_IDLE_SLEEP_ASSERTIONS_MASK) != 0) {
-			if ((oldAssertions & NO_IDLE_SLEEP_ASSERTIONS_MASK) == 0) {
-				DLOG("PreventIdleSleep driver assertion raised\n");
-				bool ok = updatePreventIdleSleepList(this, true);
-				if (ok && (changedBits & kIOPMDriverAssertionPreventSystemIdleSleepBit)) {
-					// Cancel idle sleep if there is one in progress
-					cancelIdlePowerDown(this);
-				}
-			}
-		} else {
-			DLOG("PreventIdleSleep driver assertion dropped\n");
-			updatePreventIdleSleepList(this, false);
-		}
-	}
+void IOPMrootDomain::evaluateAssertions(IOPMDriverAssertionType newAssertions, IOPMDriverAssertionType oldAssertions)
+{
+    IOPMDriverAssertionType changedBits = newAssertions ^ oldAssertions;
+
+    messageClients(kIOPMMessageDriverAssertionsChanged);        
+
+    if (changedBits & kIOPMDriverAssertionPreventDisplaySleepBit) {
+
+        if (wrangler) {
+            bool value = (newAssertions & kIOPMDriverAssertionPreventDisplaySleepBit) ? true : false;
+
+            DLOG("wrangler->setIgnoreIdleTimer\(%d)\n", value);
+            wrangler->setIgnoreIdleTimer( value );
+        }
+    }
+
+    if (changedBits & kIOPMDriverAssertionCPUBit)
+        evaluatePolicy(kStimulusDarkWakeEvaluate);
+
+    if (changedBits & kIOPMDriverAssertionReservedBit7) {
+        bool value = (newAssertions & kIOPMDriverAssertionReservedBit7) ? true : false;
+        if (value) {
+            DLOG("Driver assertion ReservedBit7 raised. Legacy IO preventing sleep\n");
+            updatePreventIdleSleepList(this, true);
+        }
+        else {
+            DLOG("Driver assertion ReservedBit7 dropped\n");
+            updatePreventIdleSleepList(this, false);
+        }
+
+    }
 }
 
 // MARK: -
@@ -9696,184 +6198,134 @@
 //
 //******************************************************************************
 
-void
-IOPMrootDomain::pmStatsRecordEvent(
-	int                 eventIndex,
-	AbsoluteTime        timestamp)
-{
-	bool        starting = eventIndex & kIOPMStatsEventStartFlag ? true:false;
-	bool        stopping = eventIndex & kIOPMStatsEventStopFlag ? true:false;
-	uint64_t    delta;
-	uint64_t    nsec;
-	OSSharedPtr<OSData> publishPMStats;
-
-	eventIndex &= ~(kIOPMStatsEventStartFlag | kIOPMStatsEventStopFlag);
-
-	absolutetime_to_nanoseconds(timestamp, &nsec);
-
-	switch (eventIndex) {
-	case kIOPMStatsHibernateImageWrite:
-		if (starting) {
-			gPMStats.hibWrite.start = nsec;
-		} else if (stopping) {
-			gPMStats.hibWrite.stop = nsec;
-		}
-
-		if (stopping) {
-			delta = gPMStats.hibWrite.stop - gPMStats.hibWrite.start;
-			IOLog("PMStats: Hibernate write took %qd ms\n", delta / NSEC_PER_MSEC);
-		}
-		break;
-	case kIOPMStatsHibernateImageRead:
-		if (starting) {
-			gPMStats.hibRead.start = nsec;
-		} else if (stopping) {
-			gPMStats.hibRead.stop = nsec;
-		}
-
-		if (stopping) {
-			delta = gPMStats.hibRead.stop - gPMStats.hibRead.start;
-			IOLog("PMStats: Hibernate read took %qd ms\n", delta / NSEC_PER_MSEC);
-
-			publishPMStats = OSData::withValue(gPMStats);
-			setProperty(kIOPMSleepStatisticsKey, publishPMStats.get());
-			bzero(&gPMStats, sizeof(gPMStats));
-		}
-		break;
-	}
+void IOPMrootDomain::pmStatsRecordEvent(
+    int                 eventIndex,
+    AbsoluteTime        timestamp)
+{
+    bool        starting = eventIndex & kIOPMStatsEventStartFlag ? true:false;
+    bool        stopping = eventIndex & kIOPMStatsEventStopFlag ? true:false;
+    uint64_t    delta;
+    uint64_t    nsec;
+    OSData *publishPMStats = NULL;
+
+    eventIndex &= ~(kIOPMStatsEventStartFlag | kIOPMStatsEventStopFlag);
+
+    absolutetime_to_nanoseconds(timestamp, &nsec);
+
+    switch (eventIndex) {
+        case kIOPMStatsHibernateImageWrite:
+            if (starting)
+                gPMStats.hibWrite.start = nsec;
+            else if (stopping)
+                gPMStats.hibWrite.stop = nsec;
+
+            if (stopping) {
+                delta = gPMStats.hibWrite.stop - gPMStats.hibWrite.start;
+                IOLog("PMStats: Hibernate write took %qd ms\n", delta/1000000ULL);
+            }
+            break;
+        case kIOPMStatsHibernateImageRead:
+            if (starting)
+                gPMStats.hibRead.start = nsec;
+            else if (stopping)
+                gPMStats.hibRead.stop = nsec;
+
+            if (stopping) {
+                delta = gPMStats.hibRead.stop - gPMStats.hibRead.start;
+                IOLog("PMStats: Hibernate read took %qd ms\n", delta/1000000ULL);
+
+                publishPMStats = OSData::withBytes(&gPMStats, sizeof(gPMStats));
+                setProperty(kIOPMSleepStatisticsKey, publishPMStats);
+                publishPMStats->release();
+                bzero(&gPMStats, sizeof(gPMStats));
+            }
+            break;
+    }
 }
 
 /*
  * Appends a record of the application response to
  * IOPMrootDomain::pmStatsAppResponses
  */
-void
-IOPMrootDomain::pmStatsRecordApplicationResponse(
-	const OSSymbol      *response,
+void IOPMrootDomain::pmStatsRecordApplicationResponse(
+	const OSSymbol		*response,
 	const char          *name,
 	int                 messageType,
-	uint32_t            delay_ms,
-	uint64_t            id,
-	OSObject            *object,
-	IOPMPowerStateIndex powerState,
-	bool                async)
-{
-	OSSharedPtr<OSDictionary>    responseDescription;
-	OSSharedPtr<OSNumber>        delayNum;
-	OSSharedPtr<OSNumber>        powerCaps;
-	OSSharedPtr<OSNumber>        pidNum;
-	OSSharedPtr<OSNumber>        msgNum;
-	OSSharedPtr<const OSSymbol>  appname;
-	OSSharedPtr<const OSSymbol>  sleep;
-	OSSharedPtr<const OSSymbol>  wake;
-	IOPMServiceInterestNotifier *notify = NULL;
-
-	if (object && (notify = OSDynamicCast(IOPMServiceInterestNotifier, object))) {
-		if (response->isEqualTo(gIOPMStatsResponseTimedOut.get())) {
-			notify->ackTimeoutCnt++;
-		} else {
-			notify->ackTimeoutCnt = 0;
-		}
-	}
-
-	if (response->isEqualTo(gIOPMStatsResponsePrompt.get()) ||
-	    (_systemTransitionType == kSystemTransitionNone) || (_systemTransitionType == kSystemTransitionNewCapClient)) {
-		return;
-	}
-
-
-	if (response->isEqualTo(gIOPMStatsDriverPSChangeSlow.get())) {
-		kdebugTrace(kPMLogDrvPSChangeDelay, id, messageType, delay_ms);
-	} else if (notify) {
-		// User space app or kernel capability client
-		if (id) {
-			kdebugTrace(kPMLogAppResponseDelay, id, notify->msgType, delay_ms);
-		} else {
-			kdebugTrace(kPMLogDrvResponseDelay, notify->uuid0, messageType, delay_ms);
-		}
-		notify->msgType = 0;
-	}
-
-	responseDescription = OSDictionary::withCapacity(5);
-	if (responseDescription) {
-		if (response) {
-			responseDescription->setObject(_statsResponseTypeKey.get(), response);
-		}
-
-		msgNum = OSNumber::withNumber(messageType, 32);
-		if (msgNum) {
-			responseDescription->setObject(_statsMessageTypeKey.get(), msgNum.get());
-		}
-
-		if (!name && notify && notify->identifier) {
-			name = notify->identifier->getCStringNoCopy();
-		}
-
-		if (name && (strlen(name) > 0)) {
-			appname = OSSymbol::withCString(name);
-			if (appname) {
-				responseDescription->setObject(_statsNameKey.get(), appname.get());
-			}
-		}
-
-		if (!id && notify) {
-			id = notify->uuid0;
-		}
-		pidNum = OSNumber::withNumber(id, 64);
-		if (pidNum) {
-			responseDescription->setObject(_statsPIDKey.get(), pidNum.get());
-		}
-
-		delayNum = OSNumber::withNumber(delay_ms, 32);
-		if (delayNum) {
-			responseDescription->setObject(_statsTimeMSKey.get(), delayNum.get());
-		}
-
-		if (response->isEqualTo(gIOPMStatsDriverPSChangeSlow.get())) {
-			powerCaps = OSNumber::withNumber(powerState, 32);
-
-#if !defined(__i386__) && !defined(__x86_64__) && (DEVELOPMENT || DEBUG)
-			static const char * driverCallTypes[] = {
-				[kDriverCallInformPreChange]  = "powerStateWillChangeTo",
-				[kDriverCallInformPostChange] = "powerStateDidChangeTo",
-				[kDriverCallSetPowerState]    = "setPowerState"
-			};
-
-			if (messageType < (sizeof(driverCallTypes) / sizeof(driverCallTypes[0]))) {
-				DLOG("%s[0x%qx]::%s(%u) %stook %d ms\n",
-				    name, id, driverCallTypes[messageType], (uint32_t) powerState,
-				    async ? "async " : "", delay_ms);
-			}
-#endif
-		} else {
-			powerCaps = OSNumber::withNumber(_pendingCapability, 32);
-		}
-		if (powerCaps) {
-			responseDescription->setObject(_statsPowerCapsKey.get(), powerCaps.get());
-		}
-
-		sleep = OSSymbol::withCString("Sleep");
-		wake = OSSymbol::withCString("Wake");
-		if (_systemTransitionType == kSystemTransitionSleep) {
-			responseDescription->setObject(kIOPMStatsSystemTransitionKey, sleep.get());
-		} else if (_systemTransitionType == kSystemTransitionWake) {
-			responseDescription->setObject(kIOPMStatsSystemTransitionKey, wake.get());
-		} else if (_systemTransitionType == kSystemTransitionCapability) {
-			if (CAP_LOSS(kIOPMSystemCapabilityGraphics)) {
-				responseDescription->setObject(kIOPMStatsSystemTransitionKey, sleep.get());
-			} else if (CAP_GAIN(kIOPMSystemCapabilityGraphics)) {
-				responseDescription->setObject(kIOPMStatsSystemTransitionKey, wake.get());
-			}
-		}
-
-		IOLockLock(pmStatsLock);
-		if (pmStatsAppResponses && pmStatsAppResponses->getCount() < 50) {
-			pmStatsAppResponses->setObject(responseDescription.get());
-		}
-		IOLockUnlock(pmStatsLock);
-	}
-
-	return;
+    uint32_t            delay_ms,
+    int                 app_pid)
+{
+    OSDictionary    *responseDescription    = NULL;
+    OSNumber        *delayNum               = NULL;
+    OSNumber        *pidNum                 = NULL;
+    OSNumber        *msgNum                 = NULL;
+    const OSSymbol  *appname;
+    const OSSymbol  *entryName;
+    OSObject        *entryType;
+    int             i;
+
+    if (!pmStatsAppResponses || pmStatsAppResponses->getCount() > 50)
+        return;
+
+    i = 0;
+    while ((responseDescription = (OSDictionary *) pmStatsAppResponses->getObject(i++)))
+    {
+        entryType = responseDescription->getObject(_statsResponseTypeKey);
+        entryName = (OSSymbol *) responseDescription->getObject(_statsNameKey);
+        if (entryName && (entryType == response) && entryName->isEqualTo(name))
+        {
+            OSNumber * entryValue;
+            entryValue = (OSNumber *) responseDescription->getObject(_statsTimeMSKey);
+            if (entryValue && (entryValue->unsigned32BitValue() < delay_ms))
+                entryValue->setValue(delay_ms);
+            return;
+        }
+    }
+
+    responseDescription = OSDictionary::withCapacity(5);
+    if (responseDescription) 
+    {
+        if (response) {
+            responseDescription->setObject(_statsResponseTypeKey, response);
+        }
+        
+        if (messageType != 0) {
+            msgNum = OSNumber::withNumber(messageType, 32);
+            if (msgNum) {
+                responseDescription->setObject(_statsMessageTypeKey, msgNum);
+                msgNum->release();
+            }
+        }
+
+        if (name && (strlen(name) > 0))
+        {
+            appname = OSSymbol::withCString(name);
+            if (appname) {
+                responseDescription->setObject(_statsNameKey, appname);
+                appname->release();
+            }
+        }
+
+        if (app_pid != -1) {
+            pidNum = OSNumber::withNumber(app_pid, 32);
+            if (pidNum) {
+                responseDescription->setObject(_statsPIDKey, pidNum);
+                pidNum->release();
+            }
+        }
+
+        delayNum = OSNumber::withNumber(delay_ms, 32);
+        if (delayNum) {
+            responseDescription->setObject(_statsTimeMSKey, delayNum);
+            delayNum->release();
+        }
+
+        if (pmStatsAppResponses) {
+            pmStatsAppResponses->setObject(responseDescription);
+        }
+
+        responseDescription->release();
+    }
+    return;
 }
 
 // MARK: -
@@ -9884,483 +6336,64 @@
 //
 //******************************************************************************
 
-#define kIOPMRegisterNVRAMTracePointHandlerKey  \
-	"IOPMRegisterNVRAMTracePointHandler"
-
-IOReturn
-IOPMrootDomain::callPlatformFunction(
-	const OSSymbol * functionName,
-	bool waitForFunction,
-	void * param1, void * param2,
-	void * param3, void * param4 )
-{
-	if (pmTracer && functionName &&
-	    functionName->isEqualTo(kIOPMRegisterNVRAMTracePointHandlerKey) &&
-	    !pmTracer->tracePointHandler && !pmTracer->tracePointTarget) {
-		uint32_t    tracePointPhases, tracePointPCI;
-		uint64_t    statusCode;
-
-		pmTracer->tracePointHandler = (IOPMTracePointHandler) param1;
-		pmTracer->tracePointTarget  = (void *) param2;
-		tracePointPCI               = (uint32_t)(uintptr_t) param3;
-		tracePointPhases            = (uint32_t)(uintptr_t) param4;
-		if ((tracePointPhases & 0xff) == kIOPMTracePointSystemSleep) {
-			OSSharedPtr<IORegistryEntry> node = IORegistryEntry::fromPath( "/chosen", gIODTPlane );
-			if (node) {
-				OSSharedPtr<OSObject> bootRomFailureProp;
-				bootRomFailureProp = node->copyProperty(kIOEFIBootRomFailureKey);
-				OSData *data = OSDynamicCast(OSData, bootRomFailureProp.get());
-				uint32_t bootFailureCode;
-				if (data && data->getLength() == sizeof(bootFailureCode)) {
-					// Failure code from EFI/BootRom is a four byte structure
-					memcpy(&bootFailureCode, data->getBytesNoCopy(), sizeof(bootFailureCode));
-					tracePointPCI = OSSwapBigToHostInt32(bootFailureCode);
-				}
-			}
-		}
-		statusCode = (((uint64_t)tracePointPCI) << 32) | tracePointPhases;
-		if ((tracePointPhases & 0xff) != kIOPMTracePointSystemUp) {
-			MSG("Sleep failure code 0x%08x 0x%08x\n",
-			    tracePointPCI, tracePointPhases);
-		}
+#define kIOPMRegisterNVRAMTracePointHandlerKey	\
+		"IOPMRegisterNVRAMTracePointHandler"
+
+IOReturn IOPMrootDomain::callPlatformFunction(
+    const OSSymbol * functionName,
+    bool waitForFunction,
+    void * param1, void * param2,
+    void * param3, void * param4 )
+{
+    if (pmTracer && functionName &&
+        functionName->isEqualTo(kIOPMRegisterNVRAMTracePointHandlerKey) &&
+        !pmTracer->tracePointHandler && !pmTracer->tracePointTarget)
+    {
+        uint32_t    tracePointPhases, tracePointPCI;
+		uint64_t	statusCode;
+
+        pmTracer->tracePointHandler = (IOPMTracePointHandler) param1;
+        pmTracer->tracePointTarget  = (void *) param2;
+        tracePointPCI				= (uint32_t)(uintptr_t) param3;
+        tracePointPhases			= (uint32_t)(uintptr_t) param4;
+        statusCode = (((uint64_t)tracePointPCI) << 32) | tracePointPhases;
+		if ((tracePointPhases >> 24) != kIOPMTracePointSystemUp)
+        {
+            MSG("Sleep failure code 0x%08x 0x%08x\n",
+                tracePointPCI, tracePointPhases);
+        }
 		setProperty(kIOPMSleepWakeFailureCodeKey, statusCode, 64);
-		pmTracer->tracePointHandler( pmTracer->tracePointTarget, 0, 0 );
-
-		return kIOReturnSuccess;
-	}
-#if HIBERNATION
-	else if (functionName &&
-	    functionName->isEqualTo(kIOPMInstallSystemSleepPolicyHandlerKey)) {
-		if (gSleepPolicyHandler) {
-			return kIOReturnExclusiveAccess;
-		}
-		if (!param1) {
-			return kIOReturnBadArgument;
-		}
-		gSleepPolicyHandler = (IOPMSystemSleepPolicyHandler) param1;
-		gSleepPolicyTarget  = (void *) param2;
-		setProperty("IOPMSystemSleepPolicyHandler", kOSBooleanTrue);
-		return kIOReturnSuccess;
-	}
+        pmTracer->tracePointHandler( pmTracer->tracePointTarget, 0, 0 );
+
+        return kIOReturnSuccess;
+    }
+
+    return super::callPlatformFunction(
+        functionName, waitForFunction, param1, param2, param3, param4);
+}
+
+void IOPMrootDomain::tracePoint( uint8_t point )
+{
+    if (systemBooting) return;
+
+    pmTracer->tracePoint(point);
+
+#if	HIBERNATION
+    if (kIOPMTracePointSleepPowerPlaneDrivers == point) IOHibernateIOKitSleep();
 #endif
-
-	return super::callPlatformFunction(
-		functionName, waitForFunction, param1, param2, param3, param4);
-}
-
-void
-IOPMrootDomain::kdebugTrace(uint32_t event, uint64_t id,
-    uintptr_t param1, uintptr_t param2, uintptr_t param3)
-{
-	uint32_t code   = IODBG_POWER(event);
-	uint64_t regId  = id;
-	if (regId == 0) {
-		regId  = getRegistryEntryID();
-	}
-	KERNEL_DEBUG_CONSTANT_IST(KDEBUG_TRACE, code, (uintptr_t) regId, param1, param2, param3, 0);
-}
-
-void
-IOPMrootDomain::tracePoint( uint8_t point )
-{
-	if (systemBooting) {
-		return;
-	}
-
-	if (kIOPMTracePointWakeCapabilityClients == point) {
-		acceptSystemWakeEvents(kAcceptSystemWakeEvents_Disable);
-	}
-
-	kdebugTrace(kPMLogSleepWakeTracePoint, 0, point, 0);
-	pmTracer->tracePoint(point);
-}
-
-static void
-kext_log_putc(char c)
-{
-	if (gKextNameEnd || gKextNamePos >= (sizeof(gKextNameBuf) - 1)) {
-		return;
-	}
-	if (c == '(' || c == '[' || c == ' ') {
-		c = 0;
-		gKextNameEnd = true;
-	}
-
-	gKextNameBuf[gKextNamePos++] = c;
-}
-
-static int
-kext_log(const char *fmt, ...)
-{
-	va_list listp;
-
-	va_start(listp, fmt);
-	_doprnt(fmt, &listp, &kext_log_putc, 16);
-	va_end(listp);
-
-	return 0;
-}
-
-static OSPtr<const OSSymbol>
-copyKextIdentifierWithAddress(vm_address_t address)
-{
-	OSSharedPtr<const OSSymbol> identifer;
-
-	IOLockLock(gHaltLogLock);
-
-	gKextNameEnd = false;
-	gKextNamePos = 0;
-	gKextNameBuf[0] = 0;
-
-	OSKext::printKextsInBacktrace(&address, 1, kext_log, OSKext::kPrintKextsLock | OSKext::kPrintKextsTerse);
-	gKextNameBuf[sizeof(gKextNameBuf) - 1] = 0;
-	identifer = OSSymbol::withCString((gKextNameBuf[0] != 0) ? gKextNameBuf : kOSKextKernelIdentifier);
-
-	IOLockUnlock(gHaltLogLock);
-
-	return identifer;
-}
-
-// Caller serialized using PM workloop
-const char *
-IOPMrootDomain::getNotificationClientName(OSObject *object)
-{
-	IOPMServiceInterestNotifier *notifier = (typeof(notifier))object;
-	const char *clientName = "UNKNOWN";
-
-	if (!notifier->clientName) {
-		// Check for user client
-		if (systemCapabilityNotifier && (((IOPMServiceInterestNotifier *) systemCapabilityNotifier.get())->handler == notifier->handler)) {
-			OSNumber *clientID = NULL;
-			messageClient(kIOMessageCopyClientID, object, &clientID);
-			if (clientID) {
-				OSSharedPtr<OSString> string(IOCopyLogNameForPID(clientID->unsigned32BitValue()), OSNoRetain);
-				if (string) {
-					notifier->clientName = OSSymbol::withString(string.get());
-				}
-				clientID->release();
-			}
-		} else if (notifier->identifier) {
-			notifier->clientName.reset(notifier->identifier.get(), OSRetain);
-		}
-	}
-
-	if (notifier->clientName) {
-		clientName = notifier->clientName->getCStringNoCopy();
-	}
-
-	return clientName;
-}
-
-void
-IOPMrootDomain::traceNotification(OSObject *object, bool start, uint64_t timestamp, uint32_t msgIndex)
-{
-	IOPMServiceInterestNotifier *notifier;
-
-	if (systemBooting) {
-		return;
-	}
-	notifier = OSDynamicCast(IOPMServiceInterestNotifier, object);
-	if (!notifier) {
-		return;
-	}
-
-	if (start) {
-		pmTracer->traceDetail(notifier->uuid0 >> 32);
-		kdebugTrace(kPMLogSleepWakeMessage, pmTracer->getTracePhase(),
-		    (uintptr_t) notifier->msgType, (uintptr_t) notifier->uuid0, (uintptr_t) notifier->uuid1);
-
-		// Update notifier state used for response/ack logging
-		notifier->msgIndex = msgIndex;
-		notifier->msgAbsTime = timestamp;
-
-		if (msgIndex != UINT_MAX) {
-			DLOG("%s[%u] to %s\n", getIOMessageString(notifier->msgType), msgIndex, getNotificationClientName(notifier));
-		} else {
-			DLOG("%s to %s\n", getIOMessageString(notifier->msgType), getNotificationClientName(notifier));
-		}
-
-		assert(notifierObject == NULL);
-		notifierThread = current_thread();
-		notifierObject.reset(notifier, OSRetain);
-	} else {
-		uint64_t nsec;
-		uint32_t delayMS;
-
-		SUB_ABSOLUTETIME(&timestamp, &notifier->msgAbsTime);
-		absolutetime_to_nanoseconds(timestamp, &nsec);
-		delayMS = (uint32_t)(nsec / 1000000ULL);
-		if (delayMS > notifier->maxMsgDelayMS) {
-			notifier->maxMsgDelayMS = delayMS;
-		}
-
-		assert(notifierObject == notifier);
-		notifierObject.reset();
-		notifierThread = NULL;
-	}
-}
-
-void
-IOPMrootDomain::traceNotificationAck(OSObject *object, uint32_t delay_ms)
-{
-	if (systemBooting) {
-		return;
-	}
-	IOPMServiceInterestNotifier *notifier = OSDynamicCast(IOPMServiceInterestNotifier, object);
-	if (!notifier) {
-		return;
-	}
-
-	kdebugTrace(kPMLogDrvResponseDelay, notifier->uuid0,
-	    (uintptr_t) notifier->uuid1, (uintptr_t) 0, (uintptr_t) delay_ms);
-
-	DLOG("%s[%u] ack from %s took %d ms\n",
-	    getIOMessageString(notifier->msgType), notifier->msgIndex, getNotificationClientName(notifier), delay_ms);
-	if (delay_ms > notifier->maxAckDelayMS) {
-		notifier->maxAckDelayMS = delay_ms;
-	}
-}
-
-void
-IOPMrootDomain::traceNotificationResponse(OSObject *object, uint32_t delay_ms, uint32_t ack_time_us)
-{
-	if (systemBooting) {
-		return;
-	}
-	IOPMServiceInterestNotifier *notifier = OSDynamicCast(IOPMServiceInterestNotifier, object);
-	if (!notifier) {
-		return;
-	}
-
-	kdebugTrace(kPMLogDrvResponseDelay, notifier->uuid0,
-	    (uintptr_t) notifier->uuid1, (uintptr_t)(ack_time_us / 1000), (uintptr_t) delay_ms);
-
-	if (ack_time_us == 0) {
-		// Client work is done and ack will not be forthcoming
-		DLOG("%s[%u] response from %s took %d ms\n",
-		    getIOMessageString(notifier->msgType), notifier->msgIndex, getNotificationClientName(notifier), delay_ms);
-	} else {
-		// Client needs more time and it must ack within ack_time_us
-		DLOG("%s[%u] response from %s took %d ms (ack in %d us)\n",
-		    getIOMessageString(notifier->msgType), notifier->msgIndex, getNotificationClientName(notifier), delay_ms, ack_time_us);
-	}
-}
-
-void
-IOPMrootDomain::traceFilteredNotification(OSObject *object)
-{
-	if ((kIOLogDebugPower & gIOKitDebug) == 0) {
-		return;
-	}
-	if (systemBooting) {
-		return;
-	}
-	IOPMServiceInterestNotifier *notifier = OSDynamicCast(IOPMServiceInterestNotifier, object);
-	if (!notifier) {
-		return;
-	}
-
-	DLOG("%s to %s dropped\n", getIOMessageString(notifier->msgType), getNotificationClientName(notifier));
-}
-
-void
-IOPMrootDomain::traceDetail(uint32_t msgType, uint32_t msgIndex, uint32_t delay)
-{
-	if (!systemBooting) {
-		uint32_t detail = ((msgType & 0xffff) << 16) | (delay & 0xffff);
-		pmTracer->traceDetail( detail );
-		kdebugTrace(kPMLogSleepWakeTracePoint, pmTracer->getTracePhase(), msgType, delay);
-		DLOG("trace point 0x%02x msgType 0x%x detail 0x%08x\n", pmTracer->getTracePhase(), msgType, delay);
-	}
-}
-
-void
-IOPMrootDomain::configureReportGated(uint64_t channel_id, uint64_t action, void *result)
-{
-	size_t      reportSize;
-	void        **report = NULL;
-	uint32_t    bktCnt;
-	uint32_t    bktSize;
-	uint32_t    *clientCnt;
-
-	ASSERT_GATED();
-
-	report = NULL;
-	if (channel_id == kAssertDelayChID) {
-		report = &assertOnWakeReport;
-		bktCnt = kAssertDelayBcktCnt;
-		bktSize = kAssertDelayBcktSize;
-		clientCnt = &assertOnWakeClientCnt;
-	} else if (channel_id == kSleepDelaysChID) {
-		report = &sleepDelaysReport;
-		bktCnt = kSleepDelaysBcktCnt;
-		bktSize = kSleepDelaysBcktSize;
-		clientCnt = &sleepDelaysClientCnt;
-	} else {
-		assert(false);
-		return;
-	}
-
-	switch (action) {
-	case kIOReportEnable:
-
-		if (*report) {
-			(*clientCnt)++;
-			break;
-		}
-
-		reportSize = HISTREPORT_BUFSIZE(bktCnt);
-		*report = IOMallocZeroData(reportSize);
-		if (*report == NULL) {
-			break;
-		}
-		HISTREPORT_INIT((uint16_t)bktCnt, bktSize, *report, reportSize,
-		    getRegistryEntryID(), channel_id, kIOReportCategoryPower);
-
-		if (channel_id == kAssertDelayChID) {
-			assertOnWakeSecs = 0;
-		}
-
-		break;
-
-	case kIOReportDisable:
-		if (*clientCnt == 0) {
-			break;
-		}
-		if (*clientCnt == 1) {
-			IOFreeData(*report, HISTREPORT_BUFSIZE(bktCnt));
-			*report = NULL;
-		}
-		(*clientCnt)--;
-
-		if (channel_id == kAssertDelayChID) {
-			assertOnWakeSecs = -1; // Invalid value to prevent updates
-		}
-		break;
-
-	case kIOReportGetDimensions:
-		if (*report) {
-			HISTREPORT_UPDATERES(*report, kIOReportGetDimensions, result);
-		}
-		break;
-	}
-
-	return;
-}
-
-IOReturn
-IOPMrootDomain::configureReport(IOReportChannelList    *channelList,
-    IOReportConfigureAction action,
-    void                   *result,
-    void                   *destination)
-{
-	unsigned cnt;
-	uint64_t configAction = (uint64_t)action;
-
-	for (cnt = 0; cnt < channelList->nchannels; cnt++) {
-		if ((channelList->channels[cnt].channel_id == kSleepCntChID) ||
-		    (channelList->channels[cnt].channel_id == kDarkWkCntChID) ||
-		    (channelList->channels[cnt].channel_id == kUserWkCntChID)) {
-			if (action != kIOReportGetDimensions) {
-				continue;
-			}
-			SIMPLEREPORT_UPDATERES(kIOReportGetDimensions, result);
-		} else if ((channelList->channels[cnt].channel_id == kAssertDelayChID) ||
-		    (channelList->channels[cnt].channel_id == kSleepDelaysChID)) {
-			gIOPMWorkLoop->runAction(
-				OSMemberFunctionCast(IOWorkLoop::Action, this, &IOPMrootDomain::configureReportGated),
-				(OSObject *)this, (void *)channelList->channels[cnt].channel_id,
-				(void *)configAction, (void *)result);
-		}
-	}
-
-	return super::configureReport(channelList, action, result, destination);
-}
-
-IOReturn
-IOPMrootDomain::updateReportGated(uint64_t ch_id, void *result, IOBufferMemoryDescriptor *dest)
-{
-	uint32_t    size2cpy;
-	void        *data2cpy;
-	void        **report;
-
-	ASSERT_GATED();
-
-	report = NULL;
-	if (ch_id == kAssertDelayChID) {
-		report = &assertOnWakeReport;
-	} else if (ch_id == kSleepDelaysChID) {
-		report = &sleepDelaysReport;
-	} else {
-		assert(false);
-		return kIOReturnBadArgument;
-	}
-
-	if (*report == NULL) {
-		return kIOReturnNotOpen;
-	}
-
-	HISTREPORT_UPDATEPREP(*report, data2cpy, size2cpy);
-	if (size2cpy > (dest->getCapacity() - dest->getLength())) {
-		return kIOReturnOverrun;
-	}
-
-	HISTREPORT_UPDATERES(*report, kIOReportCopyChannelData, result);
-	dest->appendBytes(data2cpy, size2cpy);
-
-	return kIOReturnSuccess;
-}
-
-IOReturn
-IOPMrootDomain::updateReport(IOReportChannelList      *channelList,
-    IOReportUpdateAction      action,
-    void                     *result,
-    void                     *destination)
-{
-	uint32_t size2cpy;
-	void *data2cpy;
-	uint8_t buf[SIMPLEREPORT_BUFSIZE];
-	IOBufferMemoryDescriptor *dest = OSDynamicCast(IOBufferMemoryDescriptor, (OSObject *)destination);
-	unsigned cnt;
-	uint64_t ch_id;
-
-	if (action != kIOReportCopyChannelData) {
-		goto exit;
-	}
-
-	for (cnt = 0; cnt < channelList->nchannels; cnt++) {
-		ch_id = channelList->channels[cnt].channel_id;
-
-		if ((ch_id == kAssertDelayChID) || (ch_id == kSleepDelaysChID)) {
-			gIOPMWorkLoop->runAction(
-				OSMemberFunctionCast(IOWorkLoop::Action, this, &IOPMrootDomain::updateReportGated),
-				(OSObject *)this, (void *)ch_id,
-				(void *)result, (void *)dest);
-			continue;
-		} else if ((ch_id == kSleepCntChID) ||
-		    (ch_id == kDarkWkCntChID) || (ch_id == kUserWkCntChID)) {
-			SIMPLEREPORT_INIT(buf, sizeof(buf), getRegistryEntryID(), ch_id, kIOReportCategoryPower);
-		} else {
-			continue;
-		}
-
-		if (ch_id == kSleepCntChID) {
-			SIMPLEREPORT_SETVALUE(buf, sleepCnt);
-		} else if (ch_id == kDarkWkCntChID) {
-			SIMPLEREPORT_SETVALUE(buf, darkWakeCnt);
-		} else if (ch_id == kUserWkCntChID) {
-			SIMPLEREPORT_SETVALUE(buf, displayWakeCnt);
-		}
-
-		SIMPLEREPORT_UPDATEPREP(buf, data2cpy, size2cpy);
-		SIMPLEREPORT_UPDATERES(kIOReportCopyChannelData, result);
-		dest->appendBytes(data2cpy, size2cpy);
-	}
-
-exit:
-	return super::updateReport(channelList, action, result, destination);
-}
-
+}
+
+void IOPMrootDomain::tracePoint( uint8_t point, uint8_t data )
+{
+    if (!systemBooting)
+        pmTracer->tracePoint(point, data);
+}
+
+void IOPMrootDomain::traceDetail( uint32_t detail )
+{
+    if (!systemBooting)
+        pmTracer->traceDetail( detail );
+}
 
 //******************************************************************************
 // PMTraceWorker Class
@@ -10374,207 +6407,175 @@
 #define kPMBestGuessPCIDevicesCount     25
 #define kPMMaxRTCBitfieldSize           32
 
-OSPtr<PMTraceWorker>
-PMTraceWorker::tracer(IOPMrootDomain * owner)
-{
-	OSSharedPtr<PMTraceWorker> me = OSMakeShared<PMTraceWorker>();
-	if (!me || !me->init()) {
-		return NULL;
-	}
-
-	DLOG("PMTraceWorker %p\n", OBFUSCATE(me.get()));
-
-	// Note that we cannot instantiate the PCI device -> bit mappings here, since
-	// the IODeviceTree has not yet been created by IOPlatformExpert. We create
-	// this dictionary lazily.
-	me->owner = owner;
-	me->pciDeviceBitMappings = NULL;
-	me->pmTraceWorkerLock = IOLockAlloc();
-	me->tracePhase = kIOPMTracePointSystemUp;
-	me->traceData32 = 0;
-	me->loginWindowData = 0;
-	me->coreDisplayData = 0;
-	me->coreGraphicsData = 0;
-	return me;
-}
-
-void
-PMTraceWorker::RTC_TRACE(void)
-{
-	if (tracePointHandler && tracePointTarget) {
+PMTraceWorker *PMTraceWorker::tracer(IOPMrootDomain *owner)
+{
+    PMTraceWorker           *me;
+    
+    me = OSTypeAlloc( PMTraceWorker );
+    if (!me || !me->init())
+    {
+        return NULL;
+    }
+
+    DLOG("PMTraceWorker %p\n", me);
+
+    // Note that we cannot instantiate the PCI device -> bit mappings here, since
+    // the IODeviceTree has not yet been created by IOPlatformExpert. We create
+    // this dictionary lazily.
+    me->owner = owner;
+    me->pciDeviceBitMappings = NULL;
+    me->pciMappingLock = IOLockAlloc();
+    me->tracePhase = kIOPMTracePointSystemUp;
+    me->loginWindowPhase = 0;
+    me->traceData32 = 0;
+    return me;
+}
+
+void PMTraceWorker::RTC_TRACE(void)
+{
+	if (tracePointHandler && tracePointTarget)
+	{
 		uint32_t    wordA;
 
-		IOLockLock(pmTraceWorkerLock);
-		wordA = (loginWindowData << 24) | (coreDisplayData << 16) |
-		    (coreGraphicsData << 8) | tracePhase;
-		IOLockUnlock(pmTraceWorkerLock);
-
-		tracePointHandler( tracePointTarget, traceData32, wordA );
+        wordA = (tracePhase << 24) | (loginWindowPhase << 16) |
+                (traceData8 << 8);
+
+        tracePointHandler( tracePointTarget, traceData32, wordA );
 		_LOG("RTC_TRACE wrote 0x%08x 0x%08x\n", traceData32, wordA);
 	}
-#if DEVELOPMENT || DEBUG
-	if ((swd_panic_phase != 0) && (swd_panic_phase == tracePhase)) {
-		DEBUG_LOG("Causing sleep wake failure in phase 0x%08x\n", tracePhase);
-		IOLock *l = IOLockAlloc();
-		IOLockLock(l);
-		IOLockLock(l);
-	}
-#endif /* DEVELOPMENT || DEBUG */
-}
-
-int
-PMTraceWorker::recordTopLevelPCIDevice(IOService * pciDevice)
-{
-	OSSharedPtr<const OSSymbol>    deviceName;
-	int                 index = -1;
-
-	IOLockLock(pmTraceWorkerLock);
-
-	if (!pciDeviceBitMappings) {
-		pciDeviceBitMappings = OSArray::withCapacity(kPMBestGuessPCIDevicesCount);
-		if (!pciDeviceBitMappings) {
-			goto exit;
-		}
-	}
-
-	// Check for bitmask overflow.
-	if (pciDeviceBitMappings->getCount() >= kPMMaxRTCBitfieldSize) {
-		goto exit;
-	}
-
-	if ((deviceName = pciDevice->copyName()) &&
-	    (pciDeviceBitMappings->getNextIndexOfObject(deviceName.get(), 0) == (unsigned int)-1) &&
-	    pciDeviceBitMappings->setObject(deviceName.get())) {
-		index = pciDeviceBitMappings->getCount() - 1;
-		_LOG("PMTrace PCI array: set object %s => %d\n",
-		    deviceName->getCStringNoCopy(), index);
-	}
-
-	if (!addedToRegistry && (index >= 0)) {
-		addedToRegistry = owner->setProperty("PCITopLevel", this);
-	}
+}
+
+int PMTraceWorker::recordTopLevelPCIDevice(IOService * pciDevice)
+{
+    const OSSymbol *    deviceName;
+    int                 index = -1;
+
+    IOLockLock(pciMappingLock);
+
+    if (!pciDeviceBitMappings)
+    {
+        pciDeviceBitMappings = OSArray::withCapacity(kPMBestGuessPCIDevicesCount);
+        if (!pciDeviceBitMappings)
+            goto exit;
+    }
+
+    // Check for bitmask overflow.
+    if (pciDeviceBitMappings->getCount() >= kPMMaxRTCBitfieldSize)
+        goto exit;
+
+    if ((deviceName = pciDevice->copyName()) &&
+        (pciDeviceBitMappings->getNextIndexOfObject(deviceName, 0) == (unsigned int)-1) &&
+        pciDeviceBitMappings->setObject(deviceName))
+    {
+        index = pciDeviceBitMappings->getCount() - 1;
+        _LOG("PMTrace PCI array: set object %s => %d\n",
+            deviceName->getCStringNoCopy(), index);
+    }
+    if (deviceName)
+        deviceName->release();
+    if (!addedToRegistry && (index >= 0))
+        addedToRegistry = owner->setProperty("PCITopLevel", this);
 
 exit:
-	IOLockUnlock(pmTraceWorkerLock);
-	return index;
-}
-
-bool
-PMTraceWorker::serialize(OSSerialize *s) const
-{
-	bool ok = false;
-	if (pciDeviceBitMappings) {
-		IOLockLock(pmTraceWorkerLock);
-		ok = pciDeviceBitMappings->serialize(s);
-		IOLockUnlock(pmTraceWorkerLock);
-	}
-	return ok;
-}
-
-void
-PMTraceWorker::tracePoint(uint8_t phase)
-{
-	// clear trace detail when phase begins
-	if (tracePhase != phase) {
-		traceData32 = 0;
-	}
-
-	tracePhase = phase;
-
-	DLOG("trace point 0x%02x\n", tracePhase);
-	RTC_TRACE();
-}
-
-void
-PMTraceWorker::traceDetail(uint32_t detail)
-{
-	if (detail == traceData32) {
-		return;
-	}
-	traceData32 = detail;
-	RTC_TRACE();
-}
-
-void
-PMTraceWorker::traceComponentWakeProgress(uint32_t component, uint32_t data)
-{
-	switch (component) {
-	case kIOPMLoginWindowProgress:
-		loginWindowData = data & kIOPMLoginWindowProgressMask;
-		break;
-	case kIOPMCoreDisplayProgress:
-		coreDisplayData = data & kIOPMCoreDisplayProgressMask;
-		break;
-	case kIOPMCoreGraphicsProgress:
-		coreGraphicsData = data & kIOPMCoreGraphicsProgressMask;
-		break;
-	default:
-		return;
-	}
-
-	DLOG("component trace point 0x%02x data 0x%08x\n", component, data);
-	RTC_TRACE();
-}
-
-void
-PMTraceWorker::tracePCIPowerChange(
+    IOLockUnlock(pciMappingLock);
+    return index;
+}
+
+bool PMTraceWorker::serialize(OSSerialize *s) const
+{
+    bool ok = false;
+    if (pciDeviceBitMappings)
+    {
+        IOLockLock(pciMappingLock);
+        ok = pciDeviceBitMappings->serialize(s);
+        IOLockUnlock(pciMappingLock);
+    }
+    return ok;
+}
+
+void PMTraceWorker::tracePoint(uint8_t phase)
+{
+    // clear trace detail when phase begins
+    if (tracePhase != phase)
+        traceData32 = 0;
+
+    tracePhase = phase;
+
+    DLOG("trace point 0x%02x\n", tracePhase);
+    RTC_TRACE();
+}
+
+void PMTraceWorker::tracePoint(uint8_t phase, uint8_t data8)
+{
+    // clear trace detail when phase begins
+    if (tracePhase != phase)
+        traceData32 = 0;
+
+    tracePhase = phase;
+    traceData8 = data8;
+
+    DLOG("trace point 0x%02x 0x%02x\n", tracePhase, traceData8);
+    RTC_TRACE();
+}
+
+void PMTraceWorker::traceDetail(uint32_t detail)
+{
+    if (kIOPMTracePointSleepPriorityClients != tracePhase)
+        return;
+
+    traceData32 = detail;
+    DLOG("trace point 0x%02x detail 0x%08x\n", tracePhase, traceData32);
+
+    RTC_TRACE();
+}
+
+void PMTraceWorker::traceLoginWindowPhase(uint8_t phase)
+{
+    loginWindowPhase = phase;
+
+    DLOG("loginwindow tracepoint 0x%02x\n", loginWindowPhase);
+    RTC_TRACE();
+}
+
+void PMTraceWorker::tracePCIPowerChange(
 	change_t type, IOService *service, uint32_t changeFlags, uint32_t bitNum)
 {
-	uint32_t    bitMask;
-	uint32_t    expectedFlag;
+    uint32_t	bitMask;
+	uint32_t	expectedFlag;
 
 	// Ignore PCI changes outside of system sleep/wake.
-	if ((kIOPMTracePointSleepPowerPlaneDrivers != tracePhase) &&
-	    (kIOPMTracePointWakePowerPlaneDrivers != tracePhase)) {
-		return;
-	}
+    if ((kIOPMTracePointSleepPowerPlaneDrivers != tracePhase) &&
+        (kIOPMTracePointWakePowerPlaneDrivers  != tracePhase))
+        return;
 
 	// Only record the WillChange transition when going to sleep,
 	// and the DidChange on the way up.
 	changeFlags &= (kIOPMDomainWillChange | kIOPMDomainDidChange);
 	expectedFlag = (kIOPMTracePointSleepPowerPlaneDrivers == tracePhase) ?
-	    kIOPMDomainWillChange : kIOPMDomainDidChange;
-	if (changeFlags != expectedFlag) {
+					kIOPMDomainWillChange : kIOPMDomainDidChange;
+	if (changeFlags != expectedFlag)
 		return;
-	}
-
-	// Mark this device off in our bitfield
-	if (bitNum < kPMMaxRTCBitfieldSize) {
-		bitMask = (1 << bitNum);
-
-		if (kPowerChangeStart == type) {
-			traceData32 |= bitMask;
-			_LOG("PMTrace: Device %s started  - bit %2d mask 0x%08x => 0x%08x\n",
-			    service->getName(), bitNum, bitMask, traceData32);
-			owner->kdebugTrace(kPMLogPCIDevChangeStart, service->getRegistryEntryID(), traceData32, 0);
-		} else {
-			traceData32 &= ~bitMask;
-			_LOG("PMTrace: Device %s finished - bit %2d mask 0x%08x => 0x%08x\n",
-			    service->getName(), bitNum, bitMask, traceData32);
-			owner->kdebugTrace(kPMLogPCIDevChangeDone, service->getRegistryEntryID(), traceData32, 0);
-		}
-
-		DLOG("trace point 0x%02x detail 0x%08x\n", tracePhase, traceData32);
-		RTC_TRACE();
-	}
-}
-
-uint64_t
-PMTraceWorker::getPMStatusCode()
-{
-	return ((uint64_t)traceData32 << 32) | ((uint64_t)tracePhase);
-}
-
-uint8_t
-PMTraceWorker::getTracePhase()
-{
-	return tracePhase;
-}
-
-uint32_t
-PMTraceWorker::getTraceData()
-{
-	return traceData32;
+
+    // Mark this device off in our bitfield
+    if (bitNum < kPMMaxRTCBitfieldSize)
+    {
+        bitMask = (1 << bitNum);
+
+        if (kPowerChangeStart == type)
+        {
+            traceData32 |= bitMask;
+            _LOG("PMTrace: Device %s started  - bit %2d mask 0x%08x => 0x%08x\n",
+                service->getName(), bitNum, bitMask, traceData32);
+        }
+        else
+        {
+            traceData32 &= ~bitMask;
+            _LOG("PMTrace: Device %s finished - bit %2d mask 0x%08x => 0x%08x\n",
+                service->getName(), bitNum, bitMask, traceData32);
+        }
+
+        RTC_TRACE();
+    }
 }
 
 // MARK: -
@@ -10585,52 +6586,56 @@
 //
 //******************************************************************************
 
-PMHaltWorker *
-PMHaltWorker::worker( void )
-{
-	PMHaltWorker *  me;
-	IOThread        thread;
+static unsigned int		gPMHaltBusyCount;
+static unsigned int		gPMHaltIdleCount;
+static int				gPMHaltDepth;
+static unsigned long    gPMHaltEvent;
+static IOLock *			gPMHaltLock  = 0;
+static OSArray *		gPMHaltArray = 0;
+static const OSSymbol * gPMHaltClientAcknowledgeKey = 0;
+
+PMHaltWorker * PMHaltWorker::worker( void )
+{
+	PMHaltWorker *	me;
+	IOThread		thread;
 
 	do {
 		me = OSTypeAlloc( PMHaltWorker );
-		if (!me || !me->init()) {
+		if (!me || !me->init())
 			break;
-		}
 
 		me->lock = IOLockAlloc();
-		if (!me->lock) {
+		if (!me->lock)
 			break;
-		}
-
-		DLOG("PMHaltWorker %p\n", OBFUSCATE(me));
-		me->retain(); // thread holds extra retain
-		if (KERN_SUCCESS != kernel_thread_start(&PMHaltWorker::main, (void *) me, &thread)) {
+
+		DLOG("PMHaltWorker %p\n", me);
+		me->retain();	// thread holds extra retain
+		if (KERN_SUCCESS != kernel_thread_start(&PMHaltWorker::main, (void *) me, &thread))
+		{
 			me->release();
 			break;
 		}
 		thread_deallocate(thread);
 		return me;
+
 	} while (false);
 
-	if (me) {
-		me->release();
-	}
-	return NULL;
-}
-
-void
-PMHaltWorker::free( void )
-{
-	DLOG("PMHaltWorker free %p\n", OBFUSCATE(this));
-	if (lock) {
+	if (me) me->release();
+	return 0;
+}
+
+void PMHaltWorker::free( void )
+{
+	DLOG("PMHaltWorker free %p\n", this);
+	if (lock)
+	{
 		IOLockFree(lock);
-		lock = NULL;
+		lock = 0;
 	}
 	return OSObject::free();
 }
 
-void
-PMHaltWorker::main( void * arg, wait_result_t waitResult )
+void PMHaltWorker::main( void * arg, wait_result_t waitResult )
 {
 	PMHaltWorker * me = (PMHaltWorker *) arg;
 
@@ -10639,18 +6644,22 @@
 	me->depth = gPMHaltDepth;
 	IOLockUnlock( gPMHaltLock );
 
-	while (me->depth >= 0) {
+	while (me->depth >= 0)
+	{
 		PMHaltWorker::work( me );
 
 		IOLockLock( gPMHaltLock );
-		if (++gPMHaltIdleCount >= gPMHaltBusyCount) {
+		if (++gPMHaltIdleCount >= gPMHaltBusyCount)
+		{
 			// This is the last thread to finish work on this level,
 			// inform everyone to start working on next lower level.
 			gPMHaltDepth--;
 			me->depth = gPMHaltDepth;
 			gPMHaltIdleCount = 0;
 			thread_wakeup((event_t) &gPMHaltIdleCount);
-		} else {
+		}
+		else
+		{
 			// One or more threads are still working on this level,
 			// this thread must wait.
 			me->depth = gPMHaltDepth - 1;
@@ -10662,132 +6671,140 @@
 	}
 
 	// No more work to do, terminate thread
-	DLOG("All done for worker: %p (visits = %u)\n", OBFUSCATE(me), me->visits);
+	DLOG("All done for worker: %p (visits = %u)\n", me, me->visits);
 	thread_wakeup( &gPMHaltDepth );
 	me->release();
 }
 
-void
-PMHaltWorker::work( PMHaltWorker * me )
-{
-	OSSharedPtr<IOService>     service;
-	OSSet *         inner;
-	AbsoluteTime    startTime, elapsedTime;
-	UInt32          deltaTime;
-	bool            timeout;
-
-	while (true) {
+void PMHaltWorker::work( PMHaltWorker * me )
+{
+	IOService *		service;
+	OSSet *			inner;
+	AbsoluteTime	startTime;
+	UInt32			deltaTime;
+	bool			timeout;
+
+	while (true)
+	{
+		service = 0;
 		timeout = false;
 
 		// Claim an unit of work from the shared pool
 		IOLockLock( gPMHaltLock );
 		inner = (OSSet *)gPMHaltArray->getObject(me->depth);
-		if (inner) {
-			service.reset(OSDynamicCast(IOService, inner->getAnyObject()), OSRetain);
-			if (service) {
-				inner->removeObject(service.get());
+		if (inner)
+		{
+			service = (IOService *)inner->getAnyObject();
+			if (service)
+			{
+				service->retain();
+				inner->removeObject(service);
 			}
 		}
-		IOLockUnlock( gPMHaltLock );
-		if (!service) {
-			break; // no more work at this depth
-		}
+		IOLockUnlock( gPMHaltLock );	
+		if (!service)
+			break;  // no more work at this depth
+
 		clock_get_uptime(&startTime);
 
 		if (!service->isInactive() &&
-		    service->setProperty(gPMHaltClientAcknowledgeKey.get(), me)) {
+			service->setProperty(gPMHaltClientAcknowledgeKey, me))
+		{
 			IOLockLock(me->lock);
 			me->startTime = startTime;
-			me->service   = service.get();
+			me->service   = service;
 			me->timeout   = false;
 			IOLockUnlock(me->lock);
 
-			service->systemWillShutdown( gPMHaltMessageType);
+			service->systemWillShutdown( gPMHaltEvent );
 
 			// Wait for driver acknowledgement
 			IOLockLock(me->lock);
-			while (service->propertyExists(gPMHaltClientAcknowledgeKey.get())) {
-				IOLockSleep(me->lock, me, THREAD_UNINT);
+			while (service->getProperty(gPMHaltClientAcknowledgeKey))
+			{
+				IOLockSleep(me->lock, me, THREAD_UNINT);			
 			}
-			me->service = NULL;
+			me->service = 0;
 			timeout = me->timeout;
 			IOLockUnlock(me->lock);
 		}
 
-		deltaTime = computeDeltaTimeMS(&startTime, &elapsedTime);
-		if ((deltaTime > kPMHaltTimeoutMS) || timeout) {
-			LOG("%s driver %s (0x%llx) took %u ms\n",
-			    (gPMHaltMessageType == kIOMessageSystemWillPowerOff) ?
-			    "PowerOff" : "Restart",
-			    service->getName(), service->getRegistryEntryID(),
-			    (uint32_t) deltaTime );
-			halt_log_enter("PowerOff/Restart handler completed",
-			    OSMemberFunctionCast(const void *, service.get(), &IOService::systemWillShutdown),
-			    elapsedTime);
+		deltaTime = computeDeltaTimeMS(&startTime);
+		if ((deltaTime > kPMHaltTimeoutMS) || timeout ||
+			(gIOKitDebug & kIOLogPMRootDomain))
+		{
+			LOG("%s driver %s (%p) took %u ms\n",
+				(gPMHaltEvent == kIOMessageSystemWillPowerOff) ?
+					"PowerOff" : "Restart",
+				service->getName(), service,
+				(uint32_t) deltaTime );
 		}
 
+		service->release();
 		me->visits++;
 	}
 }
 
-void
-PMHaltWorker::checkTimeout( PMHaltWorker * me, AbsoluteTime * now )
-{
-	UInt64          nano;
-	AbsoluteTime    startTime;
-	AbsoluteTime    endTime;
+void PMHaltWorker::checkTimeout( PMHaltWorker * me, AbsoluteTime * now )
+{
+	UInt64			nano;
+	AbsoluteTime	startTime;
+	AbsoluteTime	endTime;
 
 	endTime = *now;
 
 	IOLockLock(me->lock);
-	if (me->service && !me->timeout) {
+	if (me->service && !me->timeout)
+	{
 		startTime = me->startTime;
 		nano = 0;
-		if (CMP_ABSOLUTETIME(&endTime, &startTime) > 0) {
+	    if (CMP_ABSOLUTETIME(&endTime, &startTime) > 0)
+	    {
 			SUB_ABSOLUTETIME(&endTime, &startTime);
 			absolutetime_to_nanoseconds(endTime, &nano);
-		}
-		if (nano > 3000000000ULL) {
+	    }
+		if (nano > 3000000000ULL)
+		{
 			me->timeout = true;
-
-			halt_log_enter("PowerOff/Restart still waiting on handler",
-			    OSMemberFunctionCast(const void *, me->service, &IOService::systemWillShutdown),
-			    endTime);
 			MSG("%s still waiting on %s\n",
-			    (gPMHaltMessageType == kIOMessageSystemWillPowerOff) ?  "PowerOff" : "Restart",
-			    me->service->getName());
+				(gPMHaltEvent == kIOMessageSystemWillPowerOff) ?
+					"PowerOff" : "Restart",
+				me->service->getName());
 		}
 	}
 	IOLockUnlock(me->lock);
 }
 
+
 //******************************************************************************
 // acknowledgeSystemWillShutdown
 //
 // Acknowledgement from drivers that they have prepared for shutdown/restart.
 //******************************************************************************
 
-void
-IOPMrootDomain::acknowledgeSystemWillShutdown( IOService * from )
-{
-	PMHaltWorker            * worker;
-	OSSharedPtr<OSObject>     prop;
-
-	if (!from) {
+void IOPMrootDomain::acknowledgeSystemWillShutdown( IOService * from )
+{
+	PMHaltWorker *	worker;
+	OSObject *		prop;
+
+	if (!from)
 		return;
-	}
 
 	//DLOG("%s acknowledged\n", from->getName());
-	prop = from->copyProperty( gPMHaltClientAcknowledgeKey.get());
-	if (prop) {
-		worker = (PMHaltWorker *) prop.get();
+	prop = from->copyProperty( gPMHaltClientAcknowledgeKey );
+	if (prop)
+	{
+		worker = (PMHaltWorker *) prop;
 		IOLockLock(worker->lock);
-		from->removeProperty( gPMHaltClientAcknowledgeKey.get());
+		from->removeProperty( gPMHaltClientAcknowledgeKey );
 		thread_wakeup((event_t) worker);
 		IOLockUnlock(worker->lock);
-	} else {
+		worker->release();
+	}
+	else
+	{
 		DLOG("%s acknowledged without worker property\n",
-		    from->getName());
+			from->getName());
 	}
 }
 
@@ -10799,88 +6816,80 @@
 //******************************************************************************
 
 static void
-notifySystemShutdown( IOService * root, uint32_t messageType )
-{
-#define PLACEHOLDER ((OSSet *)gPMHaltArray.get())
-	OSSharedPtr<IORegistryIterator>  iter;
-	IORegistryEntry *                entry;
-	IOService *                      node;
-	OSSet *                          inner;
-	OSSharedPtr<OSSet>               newInner;
-	PMHaltWorker *                   workers[kPMHaltMaxWorkers];
-	AbsoluteTime                     deadline;
-	unsigned int                     totalNodes = 0;
-	unsigned int                     depth;
-	unsigned int                     rootDepth;
-	unsigned int                     numWorkers;
-	unsigned int                     count;
-	int                              waitResult;
-	void *                           baseFunc;
-	bool                             ok;
-
-	DLOG("%s msgType = 0x%x\n", __FUNCTION__, messageType);
+notifySystemShutdown( IOService * root, unsigned long event )
+{
+#define PLACEHOLDER ((OSSet *)gPMHaltArray)
+	IORegistryIterator *	iter;
+	IORegistryEntry *		entry;
+	IOService *				node;
+	OSSet *					inner;
+	PMHaltWorker *			workers[kPMHaltMaxWorkers];
+	AbsoluteTime			deadline;
+	unsigned int			totalNodes = 0;
+	unsigned int			depth;
+	unsigned int			rootDepth;
+	unsigned int			numWorkers;
+	unsigned int			count;
+	int						waitResult;
+	void *					baseFunc;
+	bool					ok;
+
+	DLOG("%s event = %lx\n", __FUNCTION__, event);
 
 	baseFunc = OSMemberFunctionCast(void *, root, &IOService::systemWillShutdown);
 
 	// Iterate the entire PM tree starting from root
 
 	rootDepth = root->getDepth( gIOPowerPlane );
-	if (!rootDepth) {
-		goto done;
+	if (!rootDepth) goto done;
+
+	// debug - for repeated test runs
+	while (PMHaltWorker::metaClass->getInstanceCount())
+		IOSleep(1);
+
+	if (!gPMHaltArray)
+	{
+		gPMHaltArray = OSArray::withCapacity(40);
+		if (!gPMHaltArray) goto done;
 	}
-
-	// debug - for repeated test runs
-	while (PMHaltWorker::metaClass->getInstanceCount()) {
-		IOSleep(1);
+	else // debug
+		gPMHaltArray->flushCollection();
+
+	if (!gPMHaltLock)
+	{
+		gPMHaltLock = IOLockAlloc();
+		if (!gPMHaltLock) goto done;
 	}
 
-	if (!gPMHaltArray) {
-		gPMHaltArray = OSArray::withCapacity(40);
-		if (!gPMHaltArray) {
-			goto done;
-		}
-	} else { // debug
-		gPMHaltArray->flushCollection();
+	if (!gPMHaltClientAcknowledgeKey)
+	{
+		gPMHaltClientAcknowledgeKey =
+			OSSymbol::withCStringNoCopy("PMShutdown");
+		if (!gPMHaltClientAcknowledgeKey) goto done;
 	}
 
-	if (!gPMHaltLock) {
-		gPMHaltLock = IOLockAlloc();
-		if (!gPMHaltLock) {
-			goto done;
-		}
-	}
-
-	if (!gPMHaltClientAcknowledgeKey) {
-		gPMHaltClientAcknowledgeKey =
-		    OSSymbol::withCStringNoCopy("PMShutdown");
-		if (!gPMHaltClientAcknowledgeKey) {
-			goto done;
-		}
-	}
-
-	gPMHaltMessageType = messageType;
+	gPMHaltEvent = event;
 
 	// Depth-first walk of PM plane
 
 	iter = IORegistryIterator::iterateOver(
 		root, gIOPowerPlane, kIORegistryIterateRecursively);
 
-	if (iter) {
-		while ((entry = iter->getNextObject())) {
+	if (iter)
+	{
+		while ((entry = iter->getNextObject()))
+		{
 			node = OSDynamicCast(IOService, entry);
-			if (!node) {
+			if (!node)
 				continue;
-			}
-
-			if (baseFunc ==
-			    OSMemberFunctionCast(void *, node, &IOService::systemWillShutdown)) {
+
+			if (baseFunc == 
+				OSMemberFunctionCast(void *, node, &IOService::systemWillShutdown))
 				continue;
-			}
 
 			depth = node->getDepth( gIOPowerPlane );
-			if (depth <= rootDepth) {
+			if (depth <= rootDepth)
 				continue;
-			}
 
 			ok = false;
 
@@ -10891,61 +6900,64 @@
 			// refers to nodes with the same depth.
 
 			count = gPMHaltArray->getCount();
-			while (depth >= count) {
+			while (depth >= count)
+			{
 				// expand array and insert placeholders
 				gPMHaltArray->setObject(PLACEHOLDER);
 				count++;
 			}
 			count = gPMHaltArray->getCount();
-			if (depth < count) {
+			if (depth < count)
+			{
 				inner = (OSSet *)gPMHaltArray->getObject(depth);
-				if (inner == PLACEHOLDER) {
-					newInner = OSSet::withCapacity(40);
-					if (newInner) {
-						gPMHaltArray->replaceObject(depth, newInner.get());
-						inner = newInner.get();
+				if (inner == PLACEHOLDER)
+				{
+					inner = OSSet::withCapacity(40);
+					if (inner)
+					{
+						gPMHaltArray->replaceObject(depth, inner);
+						inner->release();
 					}
 				}
 
 				// PM nodes that appear more than once in the tree will have
 				// the same depth, OSSet will refuse to add the node twice.
-				if (inner) {
+				if (inner)
 					ok = inner->setObject(node);
-				}
 			}
-			if (!ok) {
+			if (!ok)
 				DLOG("Skipped PM node %s\n", node->getName());
-			}
 		}
+		iter->release();
 	}
 
 	// debug only
-	for (int i = 0; (inner = (OSSet *)gPMHaltArray->getObject(i)); i++) {
+	for (int i = 0; (inner = (OSSet *)gPMHaltArray->getObject(i)); i++)
+	{
 		count = 0;
-		if (inner != PLACEHOLDER) {
+		if (inner != PLACEHOLDER)
 			count = inner->getCount();
-		}
 		DLOG("Nodes at depth %u = %u\n", i, count);
 	}
 
 	// strip placeholders (not all depths are populated)
 	numWorkers = 0;
-	for (int i = 0; (inner = (OSSet *)gPMHaltArray->getObject(i));) {
-		if (inner == PLACEHOLDER) {
+	for (int i = 0; (inner = (OSSet *)gPMHaltArray->getObject(i)); )
+	{
+		if (inner == PLACEHOLDER)
+		{
 			gPMHaltArray->removeObject(i);
 			continue;
 		}
 		count = inner->getCount();
-		if (count > numWorkers) {
+		if (count > numWorkers)
 			numWorkers = count;
-		}
 		totalNodes += count;
 		i++;
 	}
 
-	if (gPMHaltArray->getCount() == 0 || !numWorkers) {
+	if (gPMHaltArray->getCount() == 0 || !numWorkers)
 		goto done;
-	}
 
 	gPMHaltBusyCount = 0;
 	gPMHaltIdleCount = 0;
@@ -10953,34 +6965,34 @@
 
 	// Create multiple workers (and threads)
 
-	if (numWorkers > kPMHaltMaxWorkers) {
+	if (numWorkers > kPMHaltMaxWorkers)
 		numWorkers = kPMHaltMaxWorkers;
-	}
-
-	DLOG("PM nodes %u, maxDepth %u, workers %u\n",
-	    totalNodes, gPMHaltArray->getCount(), numWorkers);
-
-	for (unsigned int i = 0; i < numWorkers; i++) {
+
+	DLOG("PM nodes = %u, maxDepth = %u, workers = %u\n",
+		totalNodes, gPMHaltArray->getCount(), numWorkers);
+
+	for (unsigned int i = 0; i < numWorkers; i++)
 		workers[i] = PMHaltWorker::worker();
-	}
 
 	// Wait for workers to exhaust all available work
 
 	IOLockLock(gPMHaltLock);
-	while (gPMHaltDepth >= 0) {
+	while (gPMHaltDepth >= 0)
+	{
 		clock_interval_to_deadline(1000, kMillisecondScale, &deadline);
 
 		waitResult = IOLockSleepDeadline(
 			gPMHaltLock, &gPMHaltDepth, deadline, THREAD_UNINT);
-		if (THREAD_TIMED_OUT == waitResult) {
+		if (THREAD_TIMED_OUT == waitResult)
+		{
 			AbsoluteTime now;
 			clock_get_uptime(&now);
 
 			IOLockUnlock(gPMHaltLock);
-			for (unsigned int i = 0; i < numWorkers; i++) {
-				if (workers[i]) {
+			for (unsigned int i = 0 ; i < numWorkers; i++)
+			{
+				if (workers[i])
 					PMHaltWorker::checkTimeout(workers[i], &now);
-				}
 			}
 			IOLockLock(gPMHaltLock);
 		}
@@ -10989,10 +7001,10 @@
 
 	// Release all workers
 
-	for (unsigned int i = 0; i < numWorkers; i++) {
-		if (workers[i]) {
+	for (unsigned int i = 0; i < numWorkers; i++)
+	{
+		if (workers[i])
 			workers[i]->release();
-		}
 		// worker also retained by it's own thread
 	}
 
@@ -11001,715 +7013,147 @@
 	return;
 }
 
-// MARK: -
-// MARK: Kernel Assertion
+//*********************************************************************************
+// Sleep/Wake logging
+//
+//*********************************************************************************
+
+IOMemoryDescriptor *IOPMrootDomain::getPMTraceMemoryDescriptor(void)
+{
+    if (timeline)
+        return timeline->getPMTraceMemoryDescriptor();
+    else
+        return NULL;
+}
+
+// Forwards external reports of detailed events to IOPMTimeline
+IOReturn IOPMrootDomain::recordPMEvent(PMEventDetails *details)
+{
+    if (timeline && details) {
+        
+		  IOReturn rc;
+		
+		  // Record a detailed driver power change event, or... 
+		  if(details->eventClassifier == kIOPMEventClassDriverEvent) {
+			  rc = timeline->recordDetailedPowerEvent( details );
+		  }
+		
+		  // Record a system power management event
+		  else if(details->eventClassifier == kIOPMEventClassSystemEvent) {
+			  rc = timeline->recordSystemPowerEvent( details );
+		  }
+		  else {
+			  return kIOReturnBadArgument;
+		  }
+      
+      // If we get to record this message, then we've reached the 
+      // end of another successful Sleep --> Wake cycle
+      // At this point, we pat ourselves in the back and allow
+      // our Sleep --> Wake UUID to be published
+      if(details->eventType == kIOPMEventTypeWakeDone) {
+        timeline->setSleepCycleInProgressFlag(false);
+      }
+
+/*
+      // Check if its time to clear the timeline buffer
+      if(getProperty(kIOPMSleepWakeUUIDKey) 
+         && timeline->isSleepCycleInProgress() == false
+         && timeline->getNumEventsLoggedThisPeriod() > 500) {
+            
+        // Clear the old UUID
+        if(pmPowerStateQueue) {
+            pmPowerStateQueue->submitPowerEvent(kPowerEventPublishSleepWakeUUID, (void *)false );
+        }
+      }
+*/
+      return rc;
+    }
+    else
+        return kIOReturnNotReady;
+}
+
+IOReturn IOPMrootDomain::recordAndReleasePMEvent(PMEventDetails *details)
+{
+    IOReturn ret = kIOReturnBadArgument;
+
+    if (details)
+    {
+        ret = recordPMEvent(details);
+        details->release();
+    }
+
+    return ret;
+}
 
 /* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
 
-IOPMDriverAssertionID
-IOPMrootDomain::createPMAssertion(
-	IOPMDriverAssertionType whichAssertionBits,
-	IOPMDriverAssertionLevel assertionLevel,
-	IOService *ownerService,
-	const char *ownerDescription)
-{
-	IOReturn            ret;
-	IOPMDriverAssertionID     newAssertion;
-
-	if (!pmAssertions) {
-		return 0;
-	}
-
-	ret = pmAssertions->createAssertion(whichAssertionBits, assertionLevel, ownerService, ownerDescription, &newAssertion);
-
-	if (kIOReturnSuccess == ret) {
-#if (DEVELOPMENT || DEBUG)
-		if (_aotNow || (kIOLogPMRootDomain & gIOKitDebug)) {
-			const char *serviceName = (ownerService && ownerService->reserved) ? ownerService->getName() : NULL;
-			OSReportWithBacktrace("PMRD: createPMAssertion(0x%qx) %s (%s)", newAssertion,
-			    serviceName, ownerDescription);
-		}
-#endif /* (DEVELOPMENT || DEBUG) */
-
-		const bool waitForWakeup = (whichAssertionBits & kIOPMDriverAssertionForceWakeupBit);
-		if (waitForWakeup) {
-			waitForSystemTransitionToMinPowerState(AOT_STATE);
-		}
-
-		return newAssertion;
-	} else {
-		return 0;
-	}
-}
-
-IOReturn
-IOPMrootDomain::releasePMAssertion(IOPMDriverAssertionID releaseAssertion)
-{
-#if (DEVELOPMENT || DEBUG)
-	if (_aotNow || (kIOLogPMRootDomain & gIOKitDebug)) {
-		PMAssertStruct *details = pmAssertions->detailsForID(releaseAssertion, NULL);
-		if (details) {
-			const char *serviceName = (details->ownerService && details->ownerService->reserved) ?
-			    details->ownerService->getName() : NULL;
-			const char *ownerString = details->ownerString ? details->ownerString->getCStringNoCopy() : NULL;
-			OSReportWithBacktrace("PMRD: releasePMAssertion(0x%qx) %s (%s)", releaseAssertion, serviceName, ownerString);
-		} else {
-			OSReportWithBacktrace("PMRD: releasePMAssertion(0x%qx)", releaseAssertion);
-		}
-	}
-#endif /* (DEVELOPMENT || DEBUG) */
-	if (!pmAssertions) {
-		return kIOReturnInternalError;
-	}
-	return pmAssertions->releaseAssertion(releaseAssertion);
-}
-
-
-IOReturn
-IOPMrootDomain::setPMAssertionLevel(
-	IOPMDriverAssertionID assertionID,
-	IOPMDriverAssertionLevel assertionLevel)
-{
-	return pmAssertions->setAssertionLevel(assertionID, assertionLevel);
-}
-
-IOPMDriverAssertionLevel
-IOPMrootDomain::getPMAssertionLevel(IOPMDriverAssertionType whichAssertion)
-{
-	IOPMDriverAssertionType       sysLevels;
-
-	if (!pmAssertions || whichAssertion == 0) {
-		return kIOPMDriverAssertionLevelOff;
-	}
-
-	sysLevels = pmAssertions->getActivatedAssertions();
-
-	// Check that every bit set in argument 'whichAssertion' is asserted
-	// in the aggregate bits.
-	if ((sysLevels & whichAssertion) == whichAssertion) {
-		return kIOPMDriverAssertionLevelOn;
-	} else {
-		return kIOPMDriverAssertionLevelOff;
-	}
-}
-
-IOReturn
-IOPMrootDomain::setPMAssertionUserLevels(IOPMDriverAssertionType inLevels)
-{
-	if (!pmAssertions) {
-		return kIOReturnNotFound;
-	}
-
-	return pmAssertions->setUserAssertionLevels(inLevels);
-}
-
-IOReturn
-IOPMrootDomain::acquireDriverKitMatchingAssertion()
-{
-	return gIOPMWorkLoop->runActionBlock(^{
-		if (_driverKitMatchingAssertionCount != 0) {
-		        _driverKitMatchingAssertionCount++;
-		        return kIOReturnSuccess;
-		} else {
-		        if (kSystemTransitionSleep == _systemTransitionType && !idleSleepRevertible) {
-		                // system going to sleep
-		                return kIOReturnBusy;
-			} else {
-		                // createPMAssertion is asynchronous.
-		                // we must also set _driverKitMatchingAssertionCount under the PM workloop lock so that we can cancel sleep immediately
-		                // The assertion is used so that on release, we reevaluate all assertions
-		                _driverKitMatchingAssertion = createPMAssertion(kIOPMDriverAssertionCPUBit, kIOPMDriverAssertionLevelOn, this, "DK matching");
-		                if (_driverKitMatchingAssertion != kIOPMUndefinedDriverAssertionID) {
-		                        _driverKitMatchingAssertionCount = 1;
-		                        return kIOReturnSuccess;
-				} else {
-		                        return kIOReturnBusy;
-				}
-			}
-		}
-	});
-}
-
-void
-IOPMrootDomain::releaseDriverKitMatchingAssertion()
-{
-	gIOPMWorkLoop->runActionBlock(^{
-		if (_driverKitMatchingAssertionCount != 0) {
-		        _driverKitMatchingAssertionCount--;
-		        if (_driverKitMatchingAssertionCount == 0) {
-		                releasePMAssertion(_driverKitMatchingAssertion);
-		                _driverKitMatchingAssertion = kIOPMUndefinedDriverAssertionID;
-			}
-		} else {
-		        panic("Over-release of driverkit matching assertion");
-		}
-		return kIOReturnSuccess;
-	});
-}
-
-IOReturn
-IOPMrootDomain::acquireDriverKitSyncedAssertion(IOService * from, IOPMDriverAssertionID * assertionID)
-{
-	return gIOPMWorkLoop->runActionBlock(^{
-		if (kSystemTransitionSleep == _systemTransitionType && !idleSleepRevertible) {
-		        // system going to sleep
-		        return kIOReturnBusy;
-		}
-		// createPMAssertion is asynchronous.
-		// we must also set _driverKitSyncedAssertionCount under the PM workloop lock so that we can cancel sleep immediately
-		// only kIOPMDriverAssertionCPUBit is used for "synced" assertion
-		*assertionID = createPMAssertion(kIOPMDriverAssertionCPUBit, kIOPMDriverAssertionLevelOn, this, from->getName());
-		if (*assertionID != kIOPMUndefinedDriverAssertionID) {
-		        _driverKitSyncedAssertionCount++;
-		        return kIOReturnSuccess;
-		} else {
-		        return kIOReturnBusy;
-		}
-	});
-}
-
-void
-IOPMrootDomain::releaseDriverKitSyncedAssertion(IOPMDriverAssertionID assertionID)
-{
-	gIOPMWorkLoop->runActionBlock(^{
-		if (_driverKitSyncedAssertionCount != 0) {
-		        _driverKitSyncedAssertionCount--;
-		        releasePMAssertion(assertionID);
-		} else {
-		        panic("Over-release of driverkit synced assertion");
-		}
-		return kIOReturnSuccess;
-	});
-}
-
-
-IOReturn
-IOPMrootDomain::createPMAssertionSafe(
-	IOPMDriverAssertionID *assertionID,
-	IOPMDriverAssertionType whichAssertionBits,
-	IOPMDriverAssertionLevel assertionLevel,
-	IOService *ownerService,
-	const char *ownerDescription)
-{
-	IOReturn ret;
-	IOPMDriverAssertionID __block id;
-
-	if (!assertionID) {
-		return kIOReturnBadArgument;
-	}
-
-	// Grab workloop to check current transition
-	ret = gIOPMWorkLoop->runActionBlock(^{
-		if (_systemTransitionType == kSystemTransitionSleep) {
-		        return kIOReturnBusy;
-		}
-		id = createPMAssertion(whichAssertionBits, assertionLevel, ownerService, ownerDescription);
-		return id ? kIOReturnSuccess : kIOReturnError;
-	});
-
-	if (ret == kIOReturnSuccess) {
-		*assertionID = id;
-	} else if (ret == kIOReturnBusy && (kIOLogPMRootDomain & gIOKitDebug)) {
-		DLOG("assertion denied due to ongoing sleep transition (%s)\n", ownerDescription);
-	}
-
-	return ret;
-}
-
-bool
-IOPMrootDomain::serializeProperties( OSSerialize * s ) const
-{
-	if (pmAssertions) {
-		pmAssertions->publishProperties();
-	}
-	return IOService::serializeProperties(s);
-}
-
-OSSharedPtr<OSObject>
-IOPMrootDomain::copyProperty( const char * aKey) const
-{
-	OSSharedPtr<OSObject> obj;
-	obj = IOService::copyProperty(aKey);
-
-	if (obj) {
-		return obj;
-	}
-
-	if (!strncmp(aKey, kIOPMSleepWakeWdogRebootKey,
-	    sizeof(kIOPMSleepWakeWdogRebootKey))) {
-		if (swd_flags & SWD_BOOT_BY_SW_WDOG) {
-			return OSSharedPtr<OSBoolean>(kOSBooleanTrue, OSNoRetain);
-		} else {
-			return OSSharedPtr<OSBoolean>(kOSBooleanFalse, OSNoRetain);
-		}
-	}
-
-	if (!strncmp(aKey, kIOPMSleepWakeWdogLogsValidKey,
-	    sizeof(kIOPMSleepWakeWdogLogsValidKey))) {
-		if (swd_flags & SWD_VALID_LOGS) {
-			return OSSharedPtr<OSBoolean>(kOSBooleanTrue, OSNoRetain);
-		} else {
-			return OSSharedPtr<OSBoolean>(kOSBooleanFalse, OSNoRetain);
-		}
-	}
-
-	/*
-	 * XXX: We should get rid of "DesktopMode" property  when 'kAppleClamshellCausesSleepKey'
-	 * is set properly in darwake from sleep. For that, kIOPMEnableClamshell msg has to be
-	 * issued by DisplayWrangler on darkwake.
-	 */
-	if (!strcmp(aKey, "DesktopMode")) {
-		if (desktopMode) {
-			return OSSharedPtr<OSBoolean>(kOSBooleanTrue, OSNoRetain);
-		} else {
-			return OSSharedPtr<OSBoolean>(kOSBooleanFalse, OSNoRetain);
-		}
-	}
-	if (!strcmp(aKey, "DisplayIdleForDemandSleep")) {
-		if (displayIdleForDemandSleep) {
-			return OSSharedPtr<OSBoolean>(kOSBooleanTrue, OSNoRetain);
-		} else {
-			return OSSharedPtr<OSBoolean>(kOSBooleanFalse, OSNoRetain);
-		}
-	}
-
-	if (!strcmp(aKey, kIOPMDriverWakeEventsKey)) {
-		OSSharedPtr<OSArray> array;
-		WAKEEVENT_LOCK();
-		if (_systemWakeEventsArray && _systemWakeEventsArray->getCount()) {
-			OSSharedPtr<OSCollection> collection = _systemWakeEventsArray->copyCollection();
-			if (collection) {
-				array = OSDynamicPtrCast<OSArray>(collection);
-			}
-		}
-		WAKEEVENT_UNLOCK();
-		return os::move(array);
-	}
-
-	if (!strcmp(aKey, kIOPMSleepStatisticsAppsKey)) {
-		OSSharedPtr<OSArray> array;
-		IOLockLock(pmStatsLock);
-		if (pmStatsAppResponses && pmStatsAppResponses->getCount()) {
-			OSSharedPtr<OSCollection> collection = pmStatsAppResponses->copyCollection();
-			if (collection) {
-				array = OSDynamicPtrCast<OSArray>(collection);
-			}
-		}
-		IOLockUnlock(pmStatsLock);
-		return os::move(array);
-	}
-
-	if (!strcmp(aKey, kIOPMIdleSleepPreventersKey)) {
-		OSArray *idleSleepList = NULL;
-		gRootDomain->copySleepPreventersList(&idleSleepList, NULL);
-		return OSSharedPtr<OSArray>(idleSleepList, OSNoRetain);
-	}
-
-	if (!strcmp(aKey, kIOPMSystemSleepPreventersKey)) {
-		OSArray *systemSleepList = NULL;
-		gRootDomain->copySleepPreventersList(NULL, &systemSleepList);
-		return OSSharedPtr<OSArray>(systemSleepList, OSNoRetain);
-	}
-
-	if (!strcmp(aKey, kIOPMIdleSleepPreventersWithIDKey)) {
-		OSArray *idleSleepList = NULL;
-		gRootDomain->copySleepPreventersListWithID(&idleSleepList, NULL);
-		return OSSharedPtr<OSArray>(idleSleepList, OSNoRetain);
-	}
-
-	if (!strcmp(aKey, kIOPMSystemSleepPreventersWithIDKey)) {
-		OSArray *systemSleepList = NULL;
-		gRootDomain->copySleepPreventersListWithID(NULL, &systemSleepList);
-		return OSSharedPtr<OSArray>(systemSleepList, OSNoRetain);
-	}
-	return NULL;
-}
-
-// MARK: -
-// MARK: Wake Event Reporting
-
-void
-IOPMrootDomain::copyWakeReasonString( char * outBuf, size_t bufSize )
-{
-	WAKEEVENT_LOCK();
-	strlcpy(outBuf, gWakeReasonString, bufSize);
-	WAKEEVENT_UNLOCK();
-}
-
-void
-IOPMrootDomain::copyShutdownReasonString( char * outBuf, size_t bufSize )
-{
-	WAKEEVENT_LOCK();
-	strlcpy(outBuf, gShutdownReasonString, bufSize);
-	WAKEEVENT_UNLOCK();
-}
-
-void
-IOPMrootDomain::copyShutdownTime( uint64_t * time )
-{
-	WAKEEVENT_LOCK();
-	*time = gShutdownTime;
-	WAKEEVENT_UNLOCK();
-}
-
-//******************************************************************************
-// acceptSystemWakeEvents
-//
-// Private control for the acceptance of driver wake event claims.
-//******************************************************************************
-
-void
-IOPMrootDomain::acceptSystemWakeEvents( uint32_t control )
-{
-	bool logWakeReason = false;
-
-	WAKEEVENT_LOCK();
-	switch (control) {
-	case kAcceptSystemWakeEvents_Enable:
-		assert(_acceptSystemWakeEvents == false);
-		if (!_systemWakeEventsArray) {
-			_systemWakeEventsArray = OSArray::withCapacity(4);
-		}
-		_acceptSystemWakeEvents = (_systemWakeEventsArray != NULL);
-		if (!(_aotNow && (kIOPMWakeEventAOTExitFlags & _aotPendingFlags))) {
-			gWakeReasonString[0] = '\0';
-			if (_systemWakeEventsArray) {
-				_systemWakeEventsArray->flushCollection();
-			}
-		}
-		_aotWakeEventRunMode = 0;
-
-		// Remove stale WakeType property before system sleep
-		removeProperty(kIOPMRootDomainWakeTypeKey);
-		removeProperty(kIOPMRootDomainWakeReasonKey);
-		break;
-
-	case kAcceptSystemWakeEvents_Disable:
-		_acceptSystemWakeEvents = false;
-#if defined(XNU_TARGET_OS_OSX)
-		logWakeReason = (gWakeReasonString[0] != '\0');
-#else /* !defined(XNU_TARGET_OS_OSX) */
-		logWakeReason = gWakeReasonSysctlRegistered;
-#if DEVELOPMENT
-		static int panic_allowed = -1;
-
-		if ((panic_allowed == -1) &&
-		    (PE_parse_boot_argn("swd_wakereason_panic", &panic_allowed, sizeof(panic_allowed)) == false)) {
-			panic_allowed = 0;
-		}
-
-		if (panic_allowed) {
-			size_t i = 0;
-			// Panic if wake reason is null or empty
-			for (i = 0; (i < strlen(gWakeReasonString)); i++) {
-				if ((gWakeReasonString[i] != ' ') && (gWakeReasonString[i] != '\t')) {
-					break;
-				}
-			}
-			if (i >= strlen(gWakeReasonString)) {
-				panic("Wake reason is empty");
-			}
-		}
-#endif /* DEVELOPMENT */
-#endif /* !defined(XNU_TARGET_OS_OSX) */
-
-		// publish kIOPMRootDomainWakeReasonKey if not already set
-		if (!propertyExists(kIOPMRootDomainWakeReasonKey)) {
-			setProperty(kIOPMRootDomainWakeReasonKey, gWakeReasonString);
-		}
-		break;
-
-	case kAcceptSystemWakeEvents_Reenable:
-		assert(_acceptSystemWakeEvents == false);
-		_acceptSystemWakeEvents = (_systemWakeEventsArray != NULL);
-		removeProperty(kIOPMRootDomainWakeReasonKey);
-		break;
-	}
-	WAKEEVENT_UNLOCK();
-
-	if (logWakeReason) {
-		MSG("system wake events: %s\n", gWakeReasonString);
-	}
-}
-
-//******************************************************************************
-// claimSystemWakeEvent
-//
-// For a driver to claim a device is the source/conduit of a system wake event.
-//******************************************************************************
-
-void
-IOPMrootDomain::claimSystemWakeEvent(
-	IOService *     device,
-	IOOptionBits    flags,
-	const char *    reason,
-	OSObject *      details )
-{
-	OSSharedPtr<const OSSymbol>     deviceName;
-	OSSharedPtr<OSNumber>           deviceRegId;
-	OSSharedPtr<OSNumber>           claimTime;
-	OSSharedPtr<OSData>             flagsData;
-	OSSharedPtr<OSString>           reasonString;
-	OSSharedPtr<OSDictionary>       dict;
-	uint64_t                        timestamp;
-	bool                            addWakeReason;
-
-	if (!device || !reason) {
-		return;
-	}
-
-	pmEventTimeStamp(&timestamp);
-
-	uint64_t args[3] = {};
-	strlcpy((char *)args, reason, sizeof(args));
-	kdebugTrace(kPMLogClaimSystemWake, args[0], args[1], args[2], device->getRegistryEntryID());
-
-	IOOptionBits        aotFlags = 0;
-	bool                needAOTEvaluate = FALSE;
-
-	if ((kIOPMAOTModeAddEventFlags & _aotMode) && (!flags || (flags == kIOPMWakeEventSource))) {
-		flags |= kIOPMWakeEventAOTExit;
-
-		// Only allow lingering in AOT_STATE for the two wake reasons used for the wrist raise gesture.
-		if (!strcmp("AOP.OutboxNotEmpty", reason) || !strcmp("spu_gesture", reason)) {
-			flags &= ~kIOPMWakeEventAOTExit;
-		}
-	}
-
-#if defined(XNU_TARGET_OS_OSX) && !DISPLAY_WRANGLER_PRESENT
-	// Publishing the WakeType is serialized by the PM work loop
-	if (!strcmp("rtc", reason) && (_nextScheduledAlarmType != NULL)) {
-		pmPowerStateQueue->submitPowerEvent(kPowerEventPublishWakeType,
-		    (void *) _nextScheduledAlarmType.get());
-	}
-
-	// Workaround for the missing wake HID event
-	if (gDarkWakeFlags & kDarkWakeFlagUserWakeWorkaround) {
-		if (!strcmp("trackpadkeyboard", reason)) {
-			pmPowerStateQueue->submitPowerEvent(kPowerEventPublishWakeType,
-			    (void *) gIOPMWakeTypeUserKey.get());
-		}
-	}
-#endif
-
-	deviceName   = device->copyName(gIOServicePlane);
-	deviceRegId  = OSNumber::withNumber(device->getRegistryEntryID(), 64);
-	claimTime    = OSNumber::withNumber(timestamp, 64);
-	flagsData    = OSData::withValue(flags);
-	reasonString = OSString::withCString(reason);
-	dict = OSDictionary::withCapacity(5 + (details ? 1 : 0));
-	if (!dict || !deviceName || !deviceRegId || !claimTime || !flagsData || !reasonString) {
-		goto done;
-	}
-
-	dict->setObject(gIONameKey, deviceName.get());
-	dict->setObject(gIORegistryEntryIDKey, deviceRegId.get());
-	dict->setObject(kIOPMWakeEventTimeKey, claimTime.get());
-	dict->setObject(kIOPMWakeEventFlagsKey, flagsData.get());
-	dict->setObject(kIOPMWakeEventReasonKey, reasonString.get());
-	if (details) {
-		dict->setObject(kIOPMWakeEventDetailsKey, details);
-	}
-
-	WAKEEVENT_LOCK();
-	addWakeReason = _acceptSystemWakeEvents;
-	if (_aotMode) {
-		IOLog("claimSystemWakeEvent(%s, %s, 0x%x) 0x%x %d\n", reason, deviceName->getCStringNoCopy(), (int)flags, _aotPendingFlags, _aotReadyToFullWake);
-	}
-	aotFlags        = (kIOPMWakeEventAOTFlags & flags);
-	aotFlags        = (aotFlags & ~_aotPendingFlags);
-	needAOTEvaluate = false;
-	if (_aotNow && aotFlags) {
-		if (kIOPMWakeEventAOTPossibleExit & flags) {
-			_aotMetrics->possibleCount++;
-		}
-		if (kIOPMWakeEventAOTConfirmedPossibleExit & flags) {
-			_aotMetrics->confirmedPossibleCount++;
-		}
-		if (kIOPMWakeEventAOTRejectedPossibleExit & flags) {
-			_aotMetrics->rejectedPossibleCount++;
-		}
-		if (kIOPMWakeEventAOTExpiredPossibleExit & flags) {
-			_aotMetrics->expiredPossibleCount++;
-		}
-
-		_aotPendingFlags |= aotFlags;
-		addWakeReason     = _aotNow && _systemWakeEventsArray && ((kIOPMWakeEventAOTExitFlags & aotFlags));
-		needAOTEvaluate   = _aotReadyToFullWake;
-	}
-	DMSG("claimSystemWakeEvent(%s, 0x%x, %s, 0x%llx) aot %d phase 0x%x add %d\n",
-	    reason, (int)flags, deviceName->getCStringNoCopy(), device->getRegistryEntryID(),
-	    _aotNow, pmTracer->getTracePhase(), addWakeReason);
-
-#if DEVELOPMENT || DEBUG
-	if (addWakeReason) {
-		record_system_event(SYSTEM_EVENT_TYPE_INFO,
-		    SYSTEM_EVENT_SUBSYSTEM_PMRD,
-		    "Report System Wake Event",
-		    "Reason: %s Flags: 0x%x Device: %s, DeviceRegEntry: 0x%llx\n",
-		    reason,
-		    (int)flags,
-		    deviceName->getCStringNoCopy(),
-		    device->getRegistryEntryID()
-		    );
-	}
-#endif /* DEVELOPMENT || DEBUG */
-
-	if (!gWakeReasonSysctlRegistered) {
-		// Lazy registration until the platform driver stops registering
-		// the same name.
-		gWakeReasonSysctlRegistered = true;
-	}
-	if (addWakeReason) {
-		_systemWakeEventsArray->setObject(dict.get());
-		if (gWakeReasonString[0] != '\0') {
-			strlcat(gWakeReasonString, " ", sizeof(gWakeReasonString));
-		}
-		strlcat(gWakeReasonString, reason, sizeof(gWakeReasonString));
-	}
-
-	if (_aotNow && _acceptSystemWakeEvents) {
-		uint64_t runModeBits = (kIOPMAOTModeRunModeMask & flags) >> kIOPMAOTModeRunModeShift;
-		if (runModeBits) {
-			if (_aotWakeEventRunModeImpliesStorage & runModeBits) {
-				runModeBits |= kIOPMDriverClassStorage;
-			}
-			IOLog("AOT wake event %s -> mode 0x%llx\n", reasonString->getCStringNoCopy(), runModeBits);
-			_aotWakeEventRunMode |= runModeBits;
-		}
-	}
-
-	WAKEEVENT_UNLOCK();
-	if (needAOTEvaluate) {
-		// Call aotEvaluate() on PM work loop since it may call
-		// aotExit() which accesses PM state.
-		pmPowerStateQueue->submitPowerEvent(kPowerEventAOTEvaluate);
-	}
-
-done:
-	return;
-}
-
-//******************************************************************************
-// claimSystemBootEvent
-//
-// For a driver to claim a device is the source/conduit of a system boot event.
-//******************************************************************************
-
-void
-IOPMrootDomain::claimSystemBootEvent(
-	IOService *              device,
-	IOOptionBits             flags,
-	const char *             reason,
-	__unused OSObject *      details )
-{
-	if (!device || !reason) {
-		return;
-	}
-
-	DEBUG_LOG("claimSystemBootEvent(%s, %s, 0x%x)\n", reason, device->getName(), (uint32_t) flags);
-#if DEVELOPMENT || DEBUG
-	record_system_event(SYSTEM_EVENT_TYPE_INFO,
-	    SYSTEM_EVENT_SUBSYSTEM_PMRD,
-	    "Report System Boot Device",
-	    "Reason: %s Flags: 0x%x Device: %s",
-	    reason,
-	    (int)flags,
-	    device->getName()
-	    );
-#endif /* DEVELOPMENT || DEBUG */
-	WAKEEVENT_LOCK();
-	if (!gBootReasonSysctlRegistered) {
-		// Lazy sysctl registration after setting gBootReasonString
-		strlcat(gBootReasonString, reason, sizeof(gBootReasonString));
-		os_atomic_store(&gBootReasonSysctlRegistered, true, release);
-	}
-	WAKEEVENT_UNLOCK();
-}
-
-//******************************************************************************
-// claimSystemShutdownEvent
-//
-// For drivers to claim a system shutdown event on the ensuing boot.
-//******************************************************************************
-
-void
-IOPMrootDomain::claimSystemShutdownEvent(
-	IOService *              device,
-	IOOptionBits             flags,
-	const char *             reason,
-	OSObject *               details )
-{
-	if (!device || !reason) {
-		return;
-	}
-
-	DEBUG_LOG("claimSystemShutdownEvent(%s, %s, 0x%x)\n", reason, device->getName(), (uint32_t) flags);
-#if DEVELOPMENT || DEBUG
-	record_system_event(SYSTEM_EVENT_TYPE_INFO,
-	    SYSTEM_EVENT_SUBSYSTEM_PMRD,
-	    "Report System Shutdown Cause From Previous Boot",
-	    "Reason: %s Flags: 0x%x Device: %s",
-	    reason,
-	    (int)flags,
-	    device->getName()
-	    );
-#endif /* DEVELOPMENT || DEBUG */
-	WAKEEVENT_LOCK();
-	if (gShutdownReasonString[0] != '\0') {
-		strlcat(gShutdownReasonString, " ", sizeof(gShutdownReasonString));
-	}
-	strlcat(gShutdownReasonString, reason, sizeof(gShutdownReasonString));
-
-	if (details) {
-		OSDictionary *dict = OSDynamicCast(OSDictionary, details);
-		if (dict) {
-			OSSharedPtr<OSString> sharedKey = OSString::withCString(kIOPMRootDomainShutdownTime);
-			if (sharedKey) {
-				OSNumber *num = OSDynamicCast(OSNumber, dict->getObject(sharedKey.get()));
-				if (num) {
-					gShutdownTime = (uint64_t)(num->unsigned64BitValue());
-				}
-			}
-		}
-	}
-
-	gShutdownReasonSysctlRegistered = true;
-	WAKEEVENT_UNLOCK();
-}
-
-//******************************************************************************
-// requestRunMode
-//
-// For clients to request a LPW run mode. Only full wake is supported currently.
-//******************************************************************************
-
-IOReturn
-IOPMrootDomain::requestRunMode(uint64_t runModeMask)
-{
-	// We only support requesting full wake at the moment
-	if (runModeMask == kIOPMRunModeFullWake) {
-		pmPowerStateQueue->submitPowerEvent(kPowerEventRunModeRequest, NULL, runModeMask);
-		return kIOReturnSuccess;
-	}
-	return kIOReturnUnsupported;
-}
-
-IOReturn
-IOPMrootDomain::handleRequestRunMode(uint64_t runModeMask)
-{
-	// TODO: Replace with run mode logic when implemented
-	IOReturn ret = kIOReturnUnsupported;
-
-	// We only support requesting full wake at the moment
-	if (runModeMask == kIOPMRunModeFullWake) {
-		// A simple CPS should suffice for now
-		changePowerStateWithTagToPriv(ON_STATE, kCPSReasonEvaluatePolicy);
-		ret = kIOReturnSuccess;
-	}
-
-	DLOG("%s: mask %llx ret %x\n", __func__, runModeMask, ret);
-	return ret;
+IOPMDriverAssertionID IOPMrootDomain::createPMAssertion(
+    IOPMDriverAssertionType whichAssertionBits,
+    IOPMDriverAssertionLevel assertionLevel,
+    IOService *ownerService,
+    const char *ownerDescription)
+{
+    IOReturn            ret;
+    IOPMDriverAssertionID     newAssertion;
+ 
+    if (!pmAssertions)
+        return 0;
+
+    ret = pmAssertions->createAssertion(whichAssertionBits, assertionLevel, ownerService, ownerDescription, &newAssertion);
+
+    if (kIOReturnSuccess == ret)
+        return newAssertion;
+    else
+        return 0;
+}
+
+IOReturn IOPMrootDomain::releasePMAssertion(IOPMDriverAssertionID releaseAssertion)
+{
+    if (!pmAssertions)
+        return kIOReturnInternalError;
+    
+    return pmAssertions->releaseAssertion(releaseAssertion);
+}
+
+IOReturn IOPMrootDomain::setPMAssertionLevel(
+    IOPMDriverAssertionID assertionID, 
+    IOPMDriverAssertionLevel assertionLevel)
+{
+    return pmAssertions->setAssertionLevel(assertionID, assertionLevel);
+}
+
+IOPMDriverAssertionLevel IOPMrootDomain::getPMAssertionLevel(IOPMDriverAssertionType whichAssertion)
+{
+    IOPMDriverAssertionType       sysLevels;
+
+    if (!pmAssertions || whichAssertion == 0) 
+        return kIOPMDriverAssertionLevelOff;
+
+    sysLevels = pmAssertions->getActivatedAssertions();
+    
+    // Check that every bit set in argument 'whichAssertion' is asserted
+    // in the aggregate bits.
+    if ((sysLevels & whichAssertion) == whichAssertion)
+        return kIOPMDriverAssertionLevelOn;
+    else
+        return kIOPMDriverAssertionLevelOff;
+}
+
+IOReturn IOPMrootDomain::setPMAssertionUserLevels(IOPMDriverAssertionType inLevels)
+{
+    if (!pmAssertions)
+        return kIOReturnNotFound;
+
+    return pmAssertions->setUserAssertionLevels(inLevels);
+}
+
+bool IOPMrootDomain::serializeProperties( OSSerialize * s ) const
+{
+    if (pmAssertions)
+    {
+        pmAssertions->publishProperties();
+    }
+    return( IOService::serializeProperties(s) );
 }
 
 /* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
@@ -11719,16 +7163,16 @@
 
 OSDefineMetaClassAndStructors( PMSettingHandle, OSObject )
 
-void
-PMSettingHandle::free( void )
-{
-	if (pmso) {
-		pmso->clientHandleFreed();
-		pmso->release();
-		pmso = NULL;
-	}
-
-	OSObject::free();
+void PMSettingHandle::free( void )
+{
+    if (pmso)
+    {
+        pmso->clientHandleFreed();
+        pmso->release();
+        pmso = 0;
+    }
+
+    OSObject::free();
 }
 
 // MARK: -
@@ -11738,104 +7182,454 @@
 #define super OSObject
 OSDefineMetaClassAndFinalStructors( PMSettingObject, OSObject )
 
-/*
+/* 
  * Static constructor/initializer for PMSettingObject
  */
 PMSettingObject *PMSettingObject::pmSettingObject(
-	IOPMrootDomain                      * parent_arg,
-	IOPMSettingControllerCallback       handler_arg,
-	OSObject                            * target_arg,
-	uintptr_t                           refcon_arg,
-	uint32_t                            supportedPowerSources,
-	const OSSymbol *                    settings[],
-	OSObject                            * *handle_obj)
-{
-	uint32_t                            settingCount = 0;
-	PMSettingObject                     *pmso = NULL;
-	PMSettingHandle                     *pmsh = NULL;
-
-	if (!parent_arg || !handler_arg || !settings || !handle_obj) {
-		return NULL;
-	}
-
-	// count OSSymbol entries in NULL terminated settings array
-	while (settings[settingCount]) {
-		settingCount++;
-	}
-	if (0 == settingCount) {
-		return NULL;
-	}
-
-	pmso = new PMSettingObject;
-	if (!pmso || !pmso->init()) {
-		goto fail;
-	}
-
-	pmsh = new PMSettingHandle;
-	if (!pmsh || !pmsh->init()) {
-		goto fail;
-	}
-
-	queue_init(&pmso->calloutQueue);
-	pmso->parent       = parent_arg;
-	pmso->func         = handler_arg;
-	pmso->target       = target_arg;
-	pmso->refcon       = refcon_arg;
-	pmso->settingCount = settingCount;
-
-	pmso->retain(); // handle holds a retain on pmso
-	pmsh->pmso = pmso;
-	pmso->pmsh = pmsh;
-
-	pmso->publishedFeatureID = OSDataAllocation<uint32_t>(settingCount, OSAllocateMemory);
-	if (pmso->publishedFeatureID) {
-		for (unsigned int i = 0; i < settingCount; i++) {
-			// Since there is now at least one listener to this setting, publish
-			// PM root domain support for it.
-			parent_arg->publishPMSetting( settings[i],
-			    supportedPowerSources, &pmso->publishedFeatureID[i] );
-		}
-	}
-
-	*handle_obj = pmsh;
-	return pmso;
+    IOPMrootDomain                      *parent_arg,
+    IOPMSettingControllerCallback       handler_arg,
+    OSObject                            *target_arg,
+    uintptr_t                           refcon_arg,
+    uint32_t                            supportedPowerSources,
+    const OSSymbol *                    settings[],
+    OSObject                            **handle_obj)
+{
+    uint32_t                            settingCount = 0;
+    PMSettingObject                     *pmso = 0;
+    PMSettingHandle                     *pmsh = 0;
+
+    if ( !parent_arg || !handler_arg || !settings || !handle_obj )
+        return NULL;
+
+    // count OSSymbol entries in NULL terminated settings array
+    while (settings[settingCount]) {
+        settingCount++;
+    }
+    if (0 == settingCount)
+        return NULL;
+
+    pmso = new PMSettingObject;
+    if (!pmso || !pmso->init())
+        goto fail;
+
+    pmsh = new PMSettingHandle;
+    if (!pmsh || !pmsh->init())
+        goto fail;
+
+    queue_init(&pmso->calloutQueue);
+    pmso->parent       = parent_arg;
+    pmso->func         = handler_arg;
+    pmso->target       = target_arg;
+    pmso->refcon       = refcon_arg;
+    pmso->settingCount = settingCount;
+
+    pmso->retain();     // handle holds a retain on pmso
+    pmsh->pmso = pmso;
+    pmso->pmsh = pmsh;
+
+    pmso->publishedFeatureID = (uint32_t *)IOMalloc(sizeof(uint32_t)*settingCount);
+    if (pmso->publishedFeatureID) {
+        for (unsigned int i=0; i<settingCount; i++) {
+            // Since there is now at least one listener to this setting, publish
+            // PM root domain support for it.
+            parent_arg->publishPMSetting( settings[i],
+                    supportedPowerSources, &pmso->publishedFeatureID[i] );
+        }
+    }
+
+    *handle_obj = pmsh;
+    return pmso;
 
 fail:
-	if (pmso) {
-		pmso->release();
-	}
-	if (pmsh) {
-		pmsh->release();
-	}
-	return NULL;
-}
-
-void
-PMSettingObject::free( void )
-{
-	if (publishedFeatureID) {
-		for (const auto& featureID : publishedFeatureID) {
-			if (featureID) {
-				parent->removePublishedFeature( featureID );
-			}
-		}
-
-		publishedFeatureID = {};
-	}
-
-	super::free();
-}
-
-IOReturn
-PMSettingObject::dispatchPMSetting( const OSSymbol * type, OSObject * object )
-{
-	return (*func)(target, type, object, refcon);
-}
-
-void
-PMSettingObject::clientHandleFreed( void )
-{
-	parent->deregisterPMSettingObject(this);
+    if (pmso) pmso->release();
+    if (pmsh) pmsh->release();
+    return NULL;
+}
+
+void PMSettingObject::free( void )
+{
+    if (publishedFeatureID) {
+        for (uint32_t i=0; i<settingCount; i++) {
+            if (publishedFeatureID[i]) {
+                parent->removePublishedFeature( publishedFeatureID[i] );
+            }
+        }
+
+        IOFree(publishedFeatureID, sizeof(uint32_t) * settingCount);
+    }
+
+    super::free();
+}
+
+void PMSettingObject::dispatchPMSetting( const OSSymbol * type, OSObject * object )
+{
+    (*func)(target, type, object, refcon);
+}
+
+void PMSettingObject::clientHandleFreed( void )
+{
+    parent->deregisterPMSettingObject(this);
+}
+
+// MARK: -
+// MARK: IOPMTimeline
+
+#undef super
+#define super OSObject
+
+//*********************************************************************************
+//*********************************************************************************
+//*********************************************************************************
+
+IOPMTimeline *IOPMTimeline::timeline(IOPMrootDomain *root_domain)
+{
+    IOPMTimeline    *myself;
+    
+    if (!root_domain)
+        return NULL;
+    
+    myself = new IOPMTimeline;
+ 
+    if (myself) {
+        myself->owner = root_domain;
+        myself->init();
+    }
+ 
+    return myself;
+}
+
+bool IOPMTimeline::init(void)
+{
+    if (!super::init()) {
+        return false;
+    }
+
+    logLock = IOLockAlloc();
+    
+    // Fresh timeline, no events logged yet
+    this->numEventsLoggedThisPeriod = 0;
+    this->sleepCycleInProgress = false;
+
+    //this->setEventsRecordingLevel(1);   // TODO
+    this->setEventsTrackedCount(kIOPMDefaultSystemEventsTracked);
+
+    return true;
+}
+
+void IOPMTimeline::free(void)
+{
+    if (pmTraceMemoryDescriptor) {
+        pmTraceMemoryDescriptor->release();
+        pmTraceMemoryDescriptor = NULL;
+    }
+    
+    IOLockFree(logLock);
+
+    super::free();
+}
+
+IOMemoryDescriptor *IOPMTimeline::getPMTraceMemoryDescriptor()
+{
+    return pmTraceMemoryDescriptor;
+}
+
+//*********************************************************************************
+//*********************************************************************************
+//*********************************************************************************
+
+bool IOPMTimeline::setProperties(OSDictionary *d)
+{
+    OSNumber    *n = NULL;
+    OSBoolean   *b = NULL;
+    bool        changed = false;
+
+    /* Changes size of detailed events buffer */
+    n = (OSNumber *)d->getObject(kIOPMTimelineSystemNumberTrackedKey);
+    if (OSDynamicCast(OSNumber, n))
+    {
+        changed = true;
+        this->setEventsTrackedCount(n->unsigned32BitValue());        
+    }
+
+
+    /* enables or disables system events */
+    b = (OSBoolean *)d->getObject(kIOPMTimelineEnabledKey);
+    if (b)
+    {
+        changed = true;
+        this->setEventsRecordingLevel((int)(kOSBooleanTrue == b));        
+    }
+
+    return changed;
+}
+
+//*********************************************************************************
+//*********************************************************************************
+//*********************************************************************************
+
+OSDictionary *IOPMTimeline::copyInfoDictionary(void)
+{
+    OSDictionary *out = OSDictionary::withCapacity(3);
+    OSNumber    *n = NULL;
+
+    if (!out || !hdr)
+        return NULL;
+
+    n = OSNumber::withNumber(hdr->sizeEntries, 32);
+    out->setObject(kIOPMTimelineSystemNumberTrackedKey, n);
+    n->release();
+    
+    n = OSNumber::withNumber(hdr->sizeBytes, 32);
+    out->setObject(kIOPMTimelineSystemBufferSizeKey, n);
+    n->release();
+
+    // bool
+    out->setObject(kIOPMTimelineEnabledKey, eventsRecordingLevel ? kOSBooleanTrue : kOSBooleanFalse);
+
+    return out;
+}
+
+//*********************************************************************************
+//*********************************************************************************
+//*********************************************************************************
+
+/* IOPMTimeline::recordSystemPowerEvent()
+ *
+ * Expected "type" arguments are listed in IOPMPrivate.h under enum "SystemEventTypes"
+ * Type arguments include "system events", and "Intermediate events"
+ *
+ * - System Events have paired "start" and "stop" events.
+ * - A start event shall be followed by a stop event.
+ * - Any number of Intermediate Events may fall between the 
+ *   start and stop events.
+ * - Intermediate events are meaningless outside the bounds of a system event's
+ *   start & stoup routines.
+ * - It's invalid to record a Start event without a following Stop event; e.g. two
+ *   Start events without an intervenining Stop event is invalid.
+ *
+ * Buffer invariants
+ * - The first recorded system event shall be preceded by an entry with type == 0
+ * - IOPMTimeline may choose not to record intermediate events while there's not
+ *   a system event in process.
+ */
+IOReturn IOPMTimeline::recordSystemPowerEvent( PMEventDetails *details )
+{
+    static bool                 wakeDonePending = true;
+    IOPMSystemEventRecord       *record_to = NULL;
+    OSString                    *swUUIDKey = NULL;
+    uint32_t                    useIndex = 0;
+
+    if (!details)
+        return kIOReturnBadArgument;
+
+    if (!traceBuffer) 
+        return kIOReturnNotReady;
+    
+    if (details->eventType == kIOPMEventTypeWakeDone)
+    {
+      if(!wakeDonePending)  
+        return kIOReturnBadArgument;
+    }
+
+    IOLockLock(logLock);
+    
+    if (details->eventType == kIOPMEventTypeWake) {
+        wakeDonePending = true;
+    } else if (details->eventType == kIOPMEventTypeWakeDone) {
+        wakeDonePending = false;
+    }
+
+    systemState = details->eventType;
+   
+    useIndex = _atomicIndexIncrement(&hdr->index, hdr->sizeEntries);
+    
+    // The entry immediately after the latest entry (and thus
+    //  immediately before the first entry) shall have a type 0.
+    if (useIndex + 1 >= hdr->sizeEntries) {
+        traceBuffer[useIndex + 1].eventType = 0;
+    } else {
+        traceBuffer[0].eventType = 0;
+    }
+    
+    record_to = &traceBuffer[useIndex];
+    bzero(record_to, sizeof(IOPMSystemEventRecord));
+
+    /*****/
+    record_to->eventType    = details->eventType;
+    record_to->eventReason  = details->reason;
+    record_to->eventResult  = details->result;
+    pmEventTimeStamp(&record_to->timestamp);
+
+    // If caller doesn't provide a UUID, we'll use the UUID that's posted
+    // on IOPMrootDomain under key kIOPMSleepWakeUUIDKey
+    if (!details->uuid)  {
+        swUUIDKey = OSDynamicCast(OSString, owner->copyProperty(kIOPMSleepWakeUUIDKey));
+
+        if (swUUIDKey)
+            details->uuid = swUUIDKey->getCStringNoCopy();
+    }
+
+    if (details->uuid)
+        strncpy(record_to->uuid, details->uuid, kMaxPMStringLength);
+
+    if (swUUIDKey) 
+        swUUIDKey->release();
+
+    numEventsLoggedThisPeriod++;
+    /*****/
+
+    IOLockUnlock(logLock);
+    
+    return kIOReturnSuccess;
+
+}
+
+//*********************************************************************************
+//*********************************************************************************
+//*********************************************************************************
+
+IOReturn IOPMTimeline::recordDetailedPowerEvent( PMEventDetails *details )
+{
+    IOPMSystemEventRecord *record_to = NULL;
+    uint32_t                useIndex;
+
+    if (!details->eventType || !details->ownerName) 
+        return kIOReturnBadArgument;
+        
+    IOLockLock(logLock);
+
+    useIndex = _atomicIndexIncrement(&hdr->index, hdr->sizeEntries);
+    
+    record_to = (IOPMSystemEventRecord *)&traceBuffer[useIndex];
+    bzero(record_to, sizeof(IOPMSystemEventRecord));
+
+    /*****/
+    record_to->eventType = details->eventType;
+    if (details->ownerName && (strlen(details->ownerName) > 1)) {
+        strlcpy( record_to->ownerName, 
+                 details->ownerName, 
+                 sizeof(record_to->ownerName));
+    }
+    
+    record_to->ownerDisambiguateID = details->ownerUnique;
+    
+    if (details->interestName && (strlen(details->interestName) > 1)) {
+        strlcpy(record_to->interestName, 
+                details->interestName, 
+                sizeof(record_to->interestName));
+    }
+
+    record_to->oldState      = details->oldState;
+    record_to->newState      = details->newState;
+    record_to->eventResult   = details->result;
+    record_to->elapsedTimeUS = details->elapsedTimeUS;
+    pmEventTimeStamp(&record_to->timestamp);
+
+    numEventsLoggedThisPeriod++;
+    /*****/
+
+    IOLockUnlock(logLock);
+    return kIOReturnSuccess;
+}
+
+uint32_t IOPMTimeline::getNumEventsLoggedThisPeriod() {
+  return this->numEventsLoggedThisPeriod;
+}
+
+void IOPMTimeline::setNumEventsLoggedThisPeriod(uint32_t newCount) {
+  this->numEventsLoggedThisPeriod = newCount;
+}
+
+bool IOPMTimeline::isSleepCycleInProgress() {
+  return this->sleepCycleInProgress;
+}
+
+void IOPMTimeline::setSleepCycleInProgressFlag(bool flag) {
+  this->sleepCycleInProgress = flag;
+}
+//*********************************************************************************
+//*********************************************************************************
+//*********************************************************************************
+    
+void IOPMTimeline::setEventsTrackedCount(uint32_t newTracked)
+{
+    size_t      make_buf_size = 0;
+    
+    make_buf_size = sizeof(IOPMTraceBufferHeader) + (newTracked * sizeof(IOPMSystemEventRecord));
+
+    IOLockLock(logLock);
+
+    if (pmTraceMemoryDescriptor) {
+        pmTraceMemoryDescriptor->release();
+        pmTraceMemoryDescriptor = NULL;
+    }
+
+    hdr = NULL;
+    traceBuffer = NULL;
+
+    if (0 == newTracked)
+    {
+        IOLog("IOPMrootDomain -> erased buffer.\n");
+        goto exit;
+    }
+
+    pmTraceMemoryDescriptor = IOBufferMemoryDescriptor::withOptions(
+                    kIOMemoryKernelUserShared | kIODirectionIn, make_buf_size);
+
+    if (!pmTraceMemoryDescriptor)
+    {
+        IOLog("IOPMRootDomain -> IOBufferMemoryDescriptor(%d) returns NULL\n", (int)make_buf_size);
+        goto exit;
+    }    
+
+    pmTraceMemoryDescriptor->prepare(kIODirectionIn);
+    
+    // Header occupies the first sizeof(IOPMTraceBufferHeader) bytes
+    hdr = (IOPMTraceBufferHeader *)pmTraceMemoryDescriptor->getBytesNoCopy();
+
+    // Recorded events occupy the remaining bulk of the buffer
+    traceBuffer = (IOPMSystemEventRecord *)((uint8_t *)hdr + sizeof(IOPMTraceBufferHeader));
+
+    bzero(hdr, make_buf_size);
+
+    hdr->sizeBytes = make_buf_size;
+    hdr->sizeEntries = newTracked;
+
+    IOLog("IOPMRootDomain -> IOBufferMemoryDescriptor(%d) returns bufferMB with address 0x%08x\n", (int)make_buf_size, (unsigned int)(uintptr_t)traceBuffer);
+
+exit:
+    IOLockUnlock(logLock);
+}
+
+//*********************************************************************************
+//*********************************************************************************
+//*********************************************************************************
+
+void IOPMTimeline::setEventsRecordingLevel(uint32_t eventsTrackedBits)
+{
+
+    // TODO
+
+    return;
+
+}
+
+/* static helper to IOPMTimeline 
+ */
+uint32_t IOPMTimeline::_atomicIndexIncrement(uint32_t *index, uint32_t limit)
+{
+    uint32_t    was_index;
+    uint32_t    inc_index;
+    
+    if(!index)
+        return NULL;
+    
+    do {
+        was_index = *index;
+        inc_index = (was_index+1)%limit;
+    } while (!OSCompareAndSwap(was_index, inc_index, index));
+
+    return inc_index;
 }
 
 // MARK: -
@@ -11848,33 +7642,28 @@
 
 #define kAssertUniqueIDStart    500
 
-PMAssertionsTracker *
-PMAssertionsTracker::pmAssertionsTracker( IOPMrootDomain *rootDomain )
-{
-	PMAssertionsTracker    *me;
-
-	me = new PMAssertionsTracker;
-	if (!me || !me->init()) {
-		if (me) {
-			me->release();
-		}
-		return NULL;
-	}
-
-	me->owner = rootDomain;
-	me->issuingUniqueID = kAssertUniqueIDStart;
-	me->assertionsArray = OSArray::withCapacity(5);
-	me->assertionsKernel = 0;
-	me->assertionsUser = 0;
-	me->assertionsCombined = 0;
-	me->assertionsArrayLock = IOLockAlloc();
-	me->tabulateProducerCount = me->tabulateConsumerCount = 0;
-	bzero(&me->assertionsLog, sizeof(me->assertionsLog));
-
-	assert(me->assertionsArray);
-	assert(me->assertionsArrayLock);
-
-	return me;
+PMAssertionsTracker *PMAssertionsTracker::pmAssertionsTracker( IOPMrootDomain *rootDomain )
+{
+    PMAssertionsTracker    *myself;
+    
+    myself = new PMAssertionsTracker;
+ 
+    if (myself) {
+        myself->init();
+        myself->owner = rootDomain;
+        myself->issuingUniqueID = kAssertUniqueIDStart; 
+        myself->assertionsArray = OSArray::withCapacity(5);
+        myself->assertionsKernel = 0;
+        myself->assertionsUser = 0;
+        myself->assertionsCombined = 0;
+        myself->assertionsArrayLock = IOLockAlloc();
+        myself->tabulateProducerCount = myself->tabulateConsumerCount = 0;
+    
+        if (!myself->assertionsArray || !myself->assertionsArrayLock)
+            myself = NULL;
+    }
+
+    return myself;
 }
 
 /* tabulate
@@ -11882,470 +7671,367 @@
  * assertions in the kernel.
  * - Update assertionsCombined to reflect both kernel & user space.
  */
-void
-PMAssertionsTracker::tabulate(void)
-{
-	int i;
-	int count;
-	const PMAssertStruct *_a = nullptr;
-	OSValueObject<PMAssertStruct> *_d = nullptr;
-
-	IOPMDriverAssertionType oldKernel = assertionsKernel;
-	IOPMDriverAssertionType oldCombined = assertionsCombined;
-
-	ASSERT_GATED();
-
-	assertionsKernel = 0;
-	assertionsCombined = 0;
-
-	if (!assertionsArray) {
-		return;
-	}
-
-	if ((count = assertionsArray->getCount())) {
-		for (i = 0; i < count; i++) {
-			_d = OSDynamicCast(OSValueObject<PMAssertStruct>, assertionsArray->getObject(i));
-			if (_d) {
-				_a = _d->getBytesNoCopy();
-				if (_a && (kIOPMDriverAssertionLevelOn == _a->level)) {
-					assertionsKernel |= _a->assertionBits;
-				}
-			}
-		}
-	}
-
-	tabulateProducerCount++;
-	assertionsCombined = assertionsKernel | assertionsUser;
-
-	if ((assertionsKernel != oldKernel) ||
-	    (assertionsCombined != oldCombined)) {
-		owner->evaluateAssertions(assertionsCombined, oldCombined);
-	}
-}
-
-void
-PMAssertionsTracker::updateCPUBitAccounting( PMAssertStruct *assertStruct )
-{
-	AbsoluteTime now, elapsed;
-	uint64_t     nsec;
-
-	if (((assertStruct->assertionBits & kIOPMDriverAssertionCPUBit) == 0) ||
-	    (assertStruct->assertCPUStartTime == 0)) {
-		return;
-	}
-
-	now = mach_continuous_time();
-	assertionsLog.addInterval(assertStruct->id, assertStruct->assertCPUStartTime, now);
-
-	elapsed = now - assertStruct->assertCPUStartTime;
-	absolutetime_to_nanoseconds(elapsed, &nsec);
-	assertStruct->assertCPUDuration += nsec;
-	assertStruct->assertCPUStartTime = 0;
-
-	if (assertStruct->assertCPUDuration > maxAssertCPUDuration) {
-		maxAssertCPUDuration = assertStruct->assertCPUDuration;
-		maxAssertCPUEntryId = assertStruct->registryEntryID;
-	}
-}
-
-void
-PMAssertionsTracker::reportCPUBitAccounting( void )
-{
-	const PMAssertStruct *_a = nullptr;
-	OSValueObject<PMAssertStruct> *_d = nullptr;
-	int            i, count;
-	AbsoluteTime   now;
-	uint64_t       nsec;
-
-	ASSERT_GATED();
-
-	// Account for drivers that are still holding the CPU assertion
-	if (assertionsKernel & kIOPMDriverAssertionCPUBit) {
-		now = mach_absolute_time();
-		if ((count = assertionsArray->getCount())) {
-			for (i = 0; i < count; i++) {
-				_d = OSDynamicCast(OSValueObject<PMAssertStruct>, assertionsArray->getObject(i));
-				if (_d) {
-					_a = _d->getBytesNoCopy();
-					if ((_a->assertionBits & kIOPMDriverAssertionCPUBit) &&
-					    (_a->level == kIOPMDriverAssertionLevelOn) &&
-					    (_a->assertCPUStartTime != 0)) {
-						// Don't modify PMAssertStruct, leave that
-						// for updateCPUBitAccounting()
-						SUB_ABSOLUTETIME(&now, &_a->assertCPUStartTime);
-						absolutetime_to_nanoseconds(now, &nsec);
-						nsec += _a->assertCPUDuration;
-						if (nsec > maxAssertCPUDuration) {
-							maxAssertCPUDuration = nsec;
-							maxAssertCPUEntryId = _a->registryEntryID;
-						}
-					}
-				}
-			}
-		}
-	}
-
-	if (maxAssertCPUDuration) {
-		DLOG("cpu assertion held for %llu ms by 0x%llx\n",
-		    (maxAssertCPUDuration / NSEC_PER_MSEC), maxAssertCPUEntryId);
-	}
-
-	maxAssertCPUDuration = 0;
-	maxAssertCPUEntryId = 0;
-}
-
-void
-PMAssertionsTracker::publishProperties( void )
-{
-	OSSharedPtr<OSArray>             assertionsSummary;
-
-	if (tabulateConsumerCount != tabulateProducerCount) {
-		IOLockLock(assertionsArrayLock);
-
-		tabulateConsumerCount = tabulateProducerCount;
-
-		/* Publish the IOPMrootDomain property "DriverPMAssertionsDetailed"
-		 */
-		assertionsSummary = copyAssertionsArray();
-		if (assertionsSummary) {
-			owner->setProperty(kIOPMAssertionsDriverDetailedKey, assertionsSummary.get());
-		} else {
-			owner->removeProperty(kIOPMAssertionsDriverDetailedKey);
-		}
-
-		/* Publish the IOPMrootDomain property "DriverPMAssertions"
-		 */
-		owner->setProperty(kIOPMAssertionsDriverKey, assertionsKernel, 64);
-
-		IOLockUnlock(assertionsArrayLock);
-	}
-}
-
-PMAssertStruct *
-PMAssertionsTracker::detailsForID(IOPMDriverAssertionID _id, int *index)
-{
-	PMAssertStruct      *_a = NULL;
-	OSValueObject<PMAssertStruct> *_d = nullptr;
-	int                 found = -1;
-	int                 count = 0;
-	int                 i = 0;
-
-	if (assertionsArray
-	    && (count = assertionsArray->getCount())) {
-		for (i = 0; i < count; i++) {
-			_d = OSDynamicCast(OSValueObject<PMAssertStruct>, assertionsArray->getObject(i));
-			if (_d) {
-				_a = _d->getMutableBytesNoCopy();
-				if (_a && (_id == _a->id)) {
-					found = i;
-					break;
-				}
-			}
-		}
-	}
-
-	if (-1 == found) {
-		return NULL;
-	} else {
-		if (index) {
-			*index = found;
-		}
-		return _a;
-	}
+void PMAssertionsTracker::tabulate(void)
+{
+    int i;
+    int count;
+    PMAssertStruct      *_a = NULL;
+    OSData              *_d = NULL;
+
+    IOPMDriverAssertionType oldKernel = assertionsKernel;
+    IOPMDriverAssertionType oldCombined = assertionsCombined;
+
+    ASSERT_GATED();
+
+    assertionsKernel = 0;
+    assertionsCombined = 0;
+
+    if (!assertionsArray)
+        return;
+
+    if ((count = assertionsArray->getCount()))
+    {
+        for (i=0; i<count; i++)
+        {
+            _d = OSDynamicCast(OSData, assertionsArray->getObject(i));
+            if (_d)
+            {
+                _a = (PMAssertStruct *)_d->getBytesNoCopy();
+                if (_a && (kIOPMDriverAssertionLevelOn == _a->level))
+                    assertionsKernel |= _a->assertionBits;
+            }
+        }
+    }
+
+    tabulateProducerCount++;
+    assertionsCombined = assertionsKernel | assertionsUser;
+
+    if ((assertionsKernel != oldKernel) ||
+        (assertionsCombined != oldCombined))
+    {    
+        owner->evaluateAssertions(assertionsCombined, oldCombined);
+    }
+}
+
+void PMAssertionsTracker::publishProperties( void )
+{
+    OSArray             *assertionsSummary = NULL;
+
+    if (tabulateConsumerCount != tabulateProducerCount)
+    {
+        IOLockLock(assertionsArrayLock);
+
+        tabulateConsumerCount = tabulateProducerCount;
+
+        /* Publish the IOPMrootDomain property "DriverPMAssertionsDetailed"
+         */
+        assertionsSummary = copyAssertionsArray();
+        if (assertionsSummary)
+        {
+            owner->setProperty(kIOPMAssertionsDriverDetailedKey, assertionsSummary);
+            assertionsSummary->release();
+        }
+        else
+        {
+            owner->removeProperty(kIOPMAssertionsDriverDetailedKey);
+        }
+
+        /* Publish the IOPMrootDomain property "DriverPMAssertions"
+         */
+        owner->setProperty(kIOPMAssertionsDriverKey, assertionsKernel, 64);
+
+        IOLockUnlock(assertionsArrayLock);
+    }
+}
+
+PMAssertionsTracker::PMAssertStruct *PMAssertionsTracker::detailsForID(IOPMDriverAssertionID _id, int *index)
+{
+    PMAssertStruct      *_a = NULL;
+    OSData              *_d = NULL;
+    int                 found = -1;
+    int                 count = 0;
+    int                 i = 0;
+
+    if (assertionsArray
+        && (count = assertionsArray->getCount()))
+    {
+        for (i=0; i<count; i++)
+        {
+            _d = OSDynamicCast(OSData, assertionsArray->getObject(i));
+            if (_d)
+            {
+                _a = (PMAssertStruct *)_d->getBytesNoCopy();
+                if (_a && (_id == _a->id)) {
+                    found = i;
+                    break;
+                }
+            }
+        }
+    }
+
+    if (-1 == found) {
+        return NULL;
+    } else {
+        if (index)
+            *index = found;
+        return _a;
+    }
 }
 
 /* PMAssertionsTracker::handleCreateAssertion
  * Perform assertion work on the PM workloop. Do not call directly.
  */
-IOReturn
-PMAssertionsTracker::handleCreateAssertion(OSValueObject<PMAssertStruct> *newAssertion)
-{
-	PMAssertStruct *assertStruct = nullptr;
-
-	ASSERT_GATED();
-
-	if (newAssertion) {
-		assertStruct = newAssertion->getMutableBytesNoCopy();
-
-		if ((assertStruct->assertionBits & kIOPMDriverAssertionCPUBit) &&
-		    (assertStruct->level == kIOPMDriverAssertionLevelOn)) {
-			assertStruct->assertCPUStartTime = mach_continuous_time();
-		}
-
-		assertionsLog.addName(assertStruct->id, assertStruct->ownerString->getCStringNoCopy());
-
-		IOLockLock(assertionsArrayLock);
-		assertionsArray->setObject(newAssertion);
-		IOLockUnlock(assertionsArrayLock);
-		newAssertion->release();
-
-		tabulate();
-	}
-	return kIOReturnSuccess;
+IOReturn PMAssertionsTracker::handleCreateAssertion(OSData *newAssertion)
+{
+    ASSERT_GATED();
+
+    if (newAssertion)
+    {
+        IOLockLock(assertionsArrayLock);
+        assertionsArray->setObject(newAssertion);
+        IOLockUnlock(assertionsArrayLock);
+        newAssertion->release();
+
+        tabulate();
+    }
+    return kIOReturnSuccess;
 }
 
 /* PMAssertionsTracker::createAssertion
- * createAssertion allocates memory for a new PM assertion, and affects system behavior, if
+ * createAssertion allocates memory for a new PM assertion, and affects system behavior, if 
  * appropiate.
  */
-IOReturn
-PMAssertionsTracker::createAssertion(
-	IOPMDriverAssertionType which,
-	IOPMDriverAssertionLevel level,
-	IOService *serviceID,
-	const char *whoItIs,
-	IOPMDriverAssertionID *outID)
-{
-	OSSharedPtr<OSValueObject<PMAssertStruct> > dataStore;
-	PMAssertStruct  track;
-
-	// Warning: trillions and trillions of created assertions may overflow the unique ID.
-	track.id = OSIncrementAtomic64((SInt64*) &issuingUniqueID);
-	track.level = level;
-	track.assertionBits = which;
-
-	// NB: ownerString is explicitly managed by PMAssertStruct
-	// it will be released in `handleReleaseAssertion' below
-	track.ownerString = whoItIs ? OSSymbol::withCString(whoItIs).detach():nullptr;
-	track.ownerService = serviceID;
-	track.registryEntryID = serviceID ? serviceID->getRegistryEntryID():0;
-	track.modifiedTime = 0;
-	pmEventTimeStamp(&track.createdTime);
-	track.assertCPUStartTime = 0;
-	track.assertCPUDuration = 0;
-
-	dataStore = OSValueObjectWithValue(track);
-	if (!dataStore) {
-		if (track.ownerString) {
-			track.ownerString->release();
-			track.ownerString = NULL;
-		}
-		return kIOReturnNoMemory;
-	}
-
-	*outID = track.id;
-
-	if (owner && owner->pmPowerStateQueue) {
-		// queue action is responsible for releasing dataStore
-		owner->pmPowerStateQueue->submitPowerEvent(kPowerEventAssertionCreate, (void *)dataStore.detach());
-	}
-
-	return kIOReturnSuccess;
+IOReturn PMAssertionsTracker::createAssertion(
+    IOPMDriverAssertionType which,
+    IOPMDriverAssertionLevel level,
+    IOService *serviceID, 
+    const char *whoItIs, 
+    IOPMDriverAssertionID *outID)
+{
+    OSData          *dataStore = NULL;
+    PMAssertStruct  track;
+
+    // Warning: trillions and trillions of created assertions may overflow the unique ID.
+    track.id = OSIncrementAtomic64((SInt64*) &issuingUniqueID);
+    track.level = level;
+    track.assertionBits = which;
+    track.ownerString = whoItIs ? OSSymbol::withCString(whoItIs) : 0;
+    track.ownerService = serviceID;
+    track.modifiedTime = 0;
+    pmEventTimeStamp(&track.createdTime);
+    
+    dataStore = OSData::withBytes(&track, sizeof(PMAssertStruct));
+    if (!dataStore)
+    {
+        if (track.ownerString)
+            track.ownerString->release();
+        return kIOReturnNoMemory;
+    }
+
+    *outID = track.id;
+    
+    if (owner && owner->pmPowerStateQueue) {
+        owner->pmPowerStateQueue->submitPowerEvent(kPowerEventAssertionCreate, (void *)dataStore);
+    }
+    
+    return kIOReturnSuccess;
 }
 
 /* PMAssertionsTracker::handleReleaseAssertion
  * Runs in PM workloop. Do not call directly.
  */
-IOReturn
-PMAssertionsTracker::handleReleaseAssertion(
-	IOPMDriverAssertionID _id)
-{
-	ASSERT_GATED();
-
-	int             index;
-	PMAssertStruct  *assertStruct = detailsForID(_id, &index);
-
-	if (!assertStruct) {
-		return kIOReturnNotFound;
-	}
-
-	IOLockLock(assertionsArrayLock);
-
-	if ((assertStruct->assertionBits & kIOPMDriverAssertionCPUBit) &&
-	    (assertStruct->level == kIOPMDriverAssertionLevelOn)) {
-		updateCPUBitAccounting(assertStruct);
-	}
-
-	if (assertStruct->ownerString) {
-		assertStruct->ownerString->release();
-		assertStruct->ownerString = NULL;
-	}
-
-	assertionsArray->removeObject(index);
-	IOLockUnlock(assertionsArrayLock);
-
-	tabulate();
-	return kIOReturnSuccess;
+IOReturn PMAssertionsTracker::handleReleaseAssertion(
+    IOPMDriverAssertionID _id)
+{
+    ASSERT_GATED();
+
+    int             index;
+    PMAssertStruct  *assertStruct = detailsForID(_id, &index);
+    
+    if (!assertStruct)
+        return kIOReturnNotFound;
+
+    IOLockLock(assertionsArrayLock);
+    if (assertStruct->ownerString) 
+        assertStruct->ownerString->release();
+
+    assertionsArray->removeObject(index);
+    IOLockUnlock(assertionsArrayLock);
+    
+    tabulate();
+    return kIOReturnSuccess;
 }
 
 /* PMAssertionsTracker::releaseAssertion
  * Releases an assertion and affects system behavior if appropiate.
  * Actual work happens on PM workloop.
  */
-IOReturn
-PMAssertionsTracker::releaseAssertion(
-	IOPMDriverAssertionID _id)
-{
-	if (owner && owner->pmPowerStateQueue) {
-		owner->pmPowerStateQueue->submitPowerEvent(kPowerEventAssertionRelease, NULL, _id);
-	}
-	return kIOReturnSuccess;
+IOReturn PMAssertionsTracker::releaseAssertion(
+    IOPMDriverAssertionID _id)
+{
+    if (owner && owner->pmPowerStateQueue) {
+        owner->pmPowerStateQueue->submitPowerEvent(kPowerEventAssertionRelease, 0, _id);
+    }
+    return kIOReturnSuccess;
 }
 
 /* PMAssertionsTracker::handleSetAssertionLevel
  * Runs in PM workloop. Do not call directly.
  */
-IOReturn
-PMAssertionsTracker::handleSetAssertionLevel(
-	IOPMDriverAssertionID    _id,
-	IOPMDriverAssertionLevel _level)
-{
-	PMAssertStruct      *assertStruct = detailsForID(_id, NULL);
-
-	ASSERT_GATED();
-
-	if (!assertStruct) {
-		return kIOReturnNotFound;
-	}
-
-	IOLockLock(assertionsArrayLock);
-	pmEventTimeStamp(&assertStruct->modifiedTime);
-	if ((assertStruct->assertionBits & kIOPMDriverAssertionCPUBit) &&
-	    (assertStruct->level != _level)) {
-		if (_level == kIOPMDriverAssertionLevelOn) {
-			assertStruct->assertCPUStartTime = mach_continuous_time();
-		} else {
-			updateCPUBitAccounting(assertStruct);
-		}
-	}
-	assertStruct->level = _level;
-	IOLockUnlock(assertionsArrayLock);
-
-	tabulate();
-	return kIOReturnSuccess;
+IOReturn PMAssertionsTracker::handleSetAssertionLevel(
+    IOPMDriverAssertionID    _id, 
+    IOPMDriverAssertionLevel _level)
+{
+    PMAssertStruct      *assertStruct = detailsForID(_id, NULL);
+
+    ASSERT_GATED();
+
+    if (!assertStruct) {
+        return kIOReturnNotFound;
+    }
+
+    IOLockLock(assertionsArrayLock);
+    pmEventTimeStamp(&assertStruct->modifiedTime);
+    assertStruct->level = _level;
+    IOLockUnlock(assertionsArrayLock);
+
+    tabulate();
+    return kIOReturnSuccess;
 }
 
 /* PMAssertionsTracker::setAssertionLevel
  */
-IOReturn
-PMAssertionsTracker::setAssertionLevel(
-	IOPMDriverAssertionID    _id,
-	IOPMDriverAssertionLevel _level)
-{
-	if (owner && owner->pmPowerStateQueue) {
-		owner->pmPowerStateQueue->submitPowerEvent(kPowerEventAssertionSetLevel,
-		    (void *)(uintptr_t)_level, _id);
-	}
-
-	return kIOReturnSuccess;
-}
-
-IOReturn
-PMAssertionsTracker::handleSetUserAssertionLevels(void * arg0)
-{
-	IOPMDriverAssertionType new_user_levels = *(IOPMDriverAssertionType *) arg0;
-
-	ASSERT_GATED();
-
-	if (new_user_levels != assertionsUser) {
-		DLOG("assertionsUser 0x%llx->0x%llx\n", assertionsUser, new_user_levels);
-		assertionsUser = new_user_levels;
-	}
-
-	tabulate();
-	return kIOReturnSuccess;
-}
-
-IOReturn
-PMAssertionsTracker::setUserAssertionLevels(
-	IOPMDriverAssertionType new_user_levels)
-{
-	if (gIOPMWorkLoop) {
-		gIOPMWorkLoop->runAction(
-			OSMemberFunctionCast(
-				IOWorkLoop::Action,
-				this,
-				&PMAssertionsTracker::handleSetUserAssertionLevels),
-			this,
-			(void *) &new_user_levels, NULL, NULL, NULL);
-	}
-
-	return kIOReturnSuccess;
-}
-
-
-OSSharedPtr<OSArray>
-PMAssertionsTracker::copyAssertionsArray(void)
-{
-	int count;
-	int i;
-	OSSharedPtr<OSArray>     outArray = NULL;
-
-	if (!assertionsArray || (0 == (count = assertionsArray->getCount()))) {
-		goto exit;
-	}
-	outArray = OSArray::withCapacity(count);
-	if (!outArray) {
-		goto exit;
-	}
-
-	for (i = 0; i < count; i++) {
-		const PMAssertStruct *_a = nullptr;
-		OSValueObject<PMAssertStruct> *_d = nullptr;
-		OSSharedPtr<OSDictionary>    details;
-
-		_d = OSDynamicCast(OSValueObject<PMAssertStruct>, assertionsArray->getObject(i));
-		if (_d && (_a = _d->getBytesNoCopy())) {
-			OSSharedPtr<OSNumber>        _n;
-
-			details = OSDictionary::withCapacity(7);
-			if (!details) {
-				continue;
-			}
-
-			outArray->setObject(details.get());
-
-			_n = OSNumber::withNumber(_a->id, 64);
-			if (_n) {
-				details->setObject(kIOPMDriverAssertionIDKey, _n.get());
-			}
-			_n = OSNumber::withNumber(_a->createdTime, 64);
-			if (_n) {
-				details->setObject(kIOPMDriverAssertionCreatedTimeKey, _n.get());
-			}
-			_n = OSNumber::withNumber(_a->modifiedTime, 64);
-			if (_n) {
-				details->setObject(kIOPMDriverAssertionModifiedTimeKey, _n.get());
-			}
-			_n = OSNumber::withNumber((uintptr_t)_a->registryEntryID, 64);
-			if (_n) {
-				details->setObject(kIOPMDriverAssertionRegistryEntryIDKey, _n.get());
-			}
-			_n = OSNumber::withNumber(_a->level, 64);
-			if (_n) {
-				details->setObject(kIOPMDriverAssertionLevelKey, _n.get());
-			}
-			_n = OSNumber::withNumber(_a->assertionBits, 64);
-			if (_n) {
-				details->setObject(kIOPMDriverAssertionAssertedKey, _n.get());
-			}
-
-			if (_a->ownerString) {
-				details->setObject(kIOPMDriverAssertionOwnerStringKey, _a->ownerString);
-			}
-		}
-	}
+IOReturn PMAssertionsTracker::setAssertionLevel(
+    IOPMDriverAssertionID    _id, 
+    IOPMDriverAssertionLevel _level)
+{
+    if (owner && owner->pmPowerStateQueue) {
+        owner->pmPowerStateQueue->submitPowerEvent(kPowerEventAssertionSetLevel,
+                (void *)_level, _id);
+    }
+
+    return kIOReturnSuccess;    
+}
+
+IOReturn PMAssertionsTracker::handleSetUserAssertionLevels(void * arg0)
+{
+    IOPMDriverAssertionType new_user_levels = *(IOPMDriverAssertionType *) arg0;
+
+    ASSERT_GATED();
+
+    if (new_user_levels != assertionsUser)
+    {
+        assertionsUser = new_user_levels;
+        DLOG("assertionsUser 0x%llx\n", assertionsUser);
+    }
+
+    tabulate();
+    return kIOReturnSuccess;
+}
+
+IOReturn PMAssertionsTracker::setUserAssertionLevels(
+    IOPMDriverAssertionType new_user_levels)
+{
+    if (gIOPMWorkLoop) {
+        gIOPMWorkLoop->runAction(
+            OSMemberFunctionCast(
+                IOWorkLoop::Action,
+                this,
+                &PMAssertionsTracker::handleSetUserAssertionLevels),
+            this,
+            (void *) &new_user_levels, 0, 0, 0);
+    }
+
+    return kIOReturnSuccess;
+}
+
+
+OSArray *PMAssertionsTracker::copyAssertionsArray(void)
+{
+    int count;
+    int i;
+    OSArray     *outArray = NULL;
+
+    if (!assertionsArray ||
+        (0 == (count = assertionsArray->getCount())) ||
+        (NULL == (outArray = OSArray::withCapacity(count))))
+    {
+        goto exit;
+    }
+
+    for (i=0; i<count; i++)
+    {
+        PMAssertStruct  *_a = NULL;
+        OSData          *_d = NULL;
+        OSDictionary    *details = NULL;
+
+        _d = OSDynamicCast(OSData, assertionsArray->getObject(i));
+        if (_d && (_a = (PMAssertStruct *)_d->getBytesNoCopy()))
+        {
+            OSNumber        *_n = NULL;
+
+            details = OSDictionary::withCapacity(7);
+            if (!details)
+                continue;
+
+            outArray->setObject(details);
+            details->release();
+            
+            _n = OSNumber::withNumber(_a->id, 64);
+            if (_n) {            
+                details->setObject(kIOPMDriverAssertionIDKey, _n);
+                _n->release();
+            }
+            _n = OSNumber::withNumber(_a->createdTime, 64);
+            if (_n) {            
+                details->setObject(kIOPMDriverAssertionCreatedTimeKey, _n);
+                _n->release();
+            }
+            _n = OSNumber::withNumber(_a->modifiedTime, 64);
+            if (_n) {            
+                details->setObject(kIOPMDriverAssertionModifiedTimeKey, _n);
+                _n->release();
+            }
+            _n = OSNumber::withNumber((uintptr_t)_a->ownerService, 64);
+            if (_n) {            
+                details->setObject(kIOPMDriverAssertionOwnerServiceKey, _n);
+                _n->release();
+            }
+            _n = OSNumber::withNumber(_a->level, 64);
+            if (_n) {            
+                details->setObject(kIOPMDriverAssertionLevelKey, _n);
+                _n->release();
+            }
+            _n = OSNumber::withNumber(_a->assertionBits, 64);
+            if (_n) {            
+                details->setObject(kIOPMDriverAssertionAssertedKey, _n);
+                _n->release();
+            }
+            
+            if (_a->ownerString) {
+                details->setObject(kIOPMDriverAssertionOwnerStringKey, _a->ownerString);
+            }
+        }
+    }
 
 exit:
-	return os::move(outArray);
-}
-
-IOPMDriverAssertionType
-PMAssertionsTracker::getActivatedAssertions(void)
-{
-	return assertionsCombined;
-}
-
-IOPMDriverAssertionLevel
-PMAssertionsTracker::getAssertionLevel(
-	IOPMDriverAssertionType type)
-{
-	// FIXME: unused and also wrong
-	if (type && ((type & assertionsKernel) == assertionsKernel)) {
-		return kIOPMDriverAssertionLevelOn;
-	} else {
-		return kIOPMDriverAssertionLevelOff;
-	}
+    return outArray;
+}
+
+IOPMDriverAssertionType PMAssertionsTracker::getActivatedAssertions(void)
+{
+    return assertionsCombined;
+}
+
+IOPMDriverAssertionLevel PMAssertionsTracker::getAssertionLevel(
+    IOPMDriverAssertionType type)
+{
+    if (type && ((type & assertionsKernel) == assertionsKernel))
+    {
+        return kIOPMDriverAssertionLevelOn;
+    } else {
+        return kIOPMDriverAssertionLevelOff;
+    }
 }
 
 //*********************************************************************************
@@ -12353,27 +8039,25 @@
 //*********************************************************************************
 
 
-static void
-pmEventTimeStamp(uint64_t *recordTS)
-{
-	clock_sec_t     tsec;
-	clock_usec_t    tusec;
-
-	if (!recordTS) {
-		return;
-	}
-
-	// We assume tsec fits into 32 bits; 32 bits holds enough
-	// seconds for 136 years since the epoch in 1970.
-	clock_get_calendar_microtime(&tsec, &tusec);
-
-
-	// Pack the sec & microsec calendar time into a uint64_t, for fun.
-	*recordTS = 0;
-	*recordTS |= (uint32_t)tusec;
-	*recordTS |= ((uint64_t)tsec << 32);
-
-	return;
+static void pmEventTimeStamp(uint64_t *recordTS)
+{
+    clock_sec_t     tsec;
+    clock_usec_t    tusec;
+
+    if (!recordTS)
+        return;
+    
+    // We assume tsec fits into 32 bits; 32 bits holds enough
+    // seconds for 136 years since the epoch in 1970.
+    clock_get_calendar_microtime(&tsec, &tusec);
+
+
+    // Pack the sec & microsec calendar time into a uint64_t, for fun.
+    *recordTS = 0;
+    *recordTS |= (uint32_t)tusec;
+    *recordTS |= ((uint64_t)tsec << 32);
+
+    return;
 }
 
 // MARK: -
@@ -12393,1134 +8077,50 @@
 
 static IOPMPowerState patriarchPowerStates[2] =
 {
-	{.version = kIOPMPowerStateVersion1, .outputPowerCharacter = ON_POWER },
-	{.version = kIOPMPowerStateVersion1, .outputPowerCharacter = ON_POWER }
+    {1,0,ON_POWER,0,0,0,0,0,0,0,0,0},
+    {1,0,ON_POWER,0,0,0,0,0,0,0,0,0},
 };
 
-void
-IORootParent::initialize( void )
-{
-
-	gIOPMPSExternalConnectedKey = OSSymbol::withCStringNoCopy(kIOPMPSExternalConnectedKey);
-	gIOPMPSExternalChargeCapableKey = OSSymbol::withCStringNoCopy(kIOPMPSExternalChargeCapableKey);
-	gIOPMPSBatteryInstalledKey = OSSymbol::withCStringNoCopy(kIOPMPSBatteryInstalledKey);
-	gIOPMPSIsChargingKey = OSSymbol::withCStringNoCopy(kIOPMPSIsChargingKey);
-	gIOPMPSAtWarnLevelKey = OSSymbol::withCStringNoCopy(kIOPMPSAtWarnLevelKey);
-	gIOPMPSAtCriticalLevelKey = OSSymbol::withCStringNoCopy(kIOPMPSAtCriticalLevelKey);
-	gIOPMPSCurrentCapacityKey = OSSymbol::withCStringNoCopy(kIOPMPSCurrentCapacityKey);
-	gIOPMPSMaxCapacityKey = OSSymbol::withCStringNoCopy(kIOPMPSMaxCapacityKey);
-	gIOPMPSDesignCapacityKey = OSSymbol::withCStringNoCopy(kIOPMPSDesignCapacityKey);
-	gIOPMPSTimeRemainingKey = OSSymbol::withCStringNoCopy(kIOPMPSTimeRemainingKey);
-	gIOPMPSAmperageKey = OSSymbol::withCStringNoCopy(kIOPMPSAmperageKey);
-	gIOPMPSVoltageKey = OSSymbol::withCStringNoCopy(kIOPMPSVoltageKey);
-	gIOPMPSCycleCountKey = OSSymbol::withCStringNoCopy(kIOPMPSCycleCountKey);
-	gIOPMPSMaxErrKey = OSSymbol::withCStringNoCopy(kIOPMPSMaxErrKey);
-	gIOPMPSAdapterInfoKey = OSSymbol::withCStringNoCopy(kIOPMPSAdapterInfoKey);
-	gIOPMPSLocationKey = OSSymbol::withCStringNoCopy(kIOPMPSLocationKey);
-	gIOPMPSErrorConditionKey = OSSymbol::withCStringNoCopy(kIOPMPSErrorConditionKey);
-	gIOPMPSManufacturerKey = OSSymbol::withCStringNoCopy(kIOPMPSManufacturerKey);
-	gIOPMPSManufactureDateKey = OSSymbol::withCStringNoCopy(kIOPMPSManufactureDateKey);
-	gIOPMPSModelKey = OSSymbol::withCStringNoCopy(kIOPMPSModelKey);
-	gIOPMPSSerialKey = OSSymbol::withCStringNoCopy(kIOPMPSSerialKey);
-	gIOPMPSLegacyBatteryInfoKey = OSSymbol::withCStringNoCopy(kIOPMPSLegacyBatteryInfoKey);
-	gIOPMPSBatteryHealthKey = OSSymbol::withCStringNoCopy(kIOPMPSBatteryHealthKey);
-	gIOPMPSHealthConfidenceKey = OSSymbol::withCStringNoCopy(kIOPMPSHealthConfidenceKey);
-	gIOPMPSCapacityEstimatedKey = OSSymbol::withCStringNoCopy(kIOPMPSCapacityEstimatedKey);
-	gIOPMPSBatteryChargeStatusKey = OSSymbol::withCStringNoCopy(kIOPMPSBatteryChargeStatusKey);
-	gIOPMPSBatteryTemperatureKey = OSSymbol::withCStringNoCopy(kIOPMPSBatteryTemperatureKey);
-	gIOPMPSAdapterDetailsKey = OSSymbol::withCStringNoCopy(kIOPMPSAdapterDetailsKey);
-	gIOPMPSChargerConfigurationKey = OSSymbol::withCStringNoCopy(kIOPMPSChargerConfigurationKey);
-	gIOPMPSAdapterDetailsIDKey = OSSymbol::withCStringNoCopy(kIOPMPSAdapterDetailsIDKey);
-	gIOPMPSAdapterDetailsWattsKey = OSSymbol::withCStringNoCopy(kIOPMPSAdapterDetailsWattsKey);
-	gIOPMPSAdapterDetailsRevisionKey = OSSymbol::withCStringNoCopy(kIOPMPSAdapterDetailsRevisionKey);
-	gIOPMPSAdapterDetailsSerialNumberKey = OSSymbol::withCStringNoCopy(kIOPMPSAdapterDetailsSerialNumberKey);
-	gIOPMPSAdapterDetailsFamilyKey = OSSymbol::withCStringNoCopy(kIOPMPSAdapterDetailsFamilyKey);
-	gIOPMPSAdapterDetailsAmperageKey = OSSymbol::withCStringNoCopy(kIOPMPSAdapterDetailsAmperageKey);
-	gIOPMPSAdapterDetailsDescriptionKey = OSSymbol::withCStringNoCopy(kIOPMPSAdapterDetailsDescriptionKey);
-	gIOPMPSAdapterDetailsPMUConfigurationKey = OSSymbol::withCStringNoCopy(kIOPMPSAdapterDetailsPMUConfigurationKey);
-	gIOPMPSAdapterDetailsSourceIDKey = OSSymbol::withCStringNoCopy(kIOPMPSAdapterDetailsSourceIDKey);
-	gIOPMPSAdapterDetailsErrorFlagsKey = OSSymbol::withCStringNoCopy(kIOPMPSAdapterDetailsErrorFlagsKey);
-	gIOPMPSAdapterDetailsSharedSourceKey = OSSymbol::withCStringNoCopy(kIOPMPSAdapterDetailsSharedSourceKey);
-	gIOPMPSAdapterDetailsCloakedKey = OSSymbol::withCStringNoCopy(kIOPMPSAdapterDetailsCloakedKey);
-	gIOPMPSInvalidWakeSecondsKey = OSSymbol::withCStringNoCopy(kIOPMPSInvalidWakeSecondsKey);
-	gIOPMPSPostChargeWaitSecondsKey = OSSymbol::withCStringNoCopy(kIOPMPSPostChargeWaitSecondsKey);
-	gIOPMPSPostDishargeWaitSecondsKey = OSSymbol::withCStringNoCopy(kIOPMPSPostDishargeWaitSecondsKey);
-}
-
-bool
-IORootParent::start( IOService * nub )
-{
-	IOService::start(nub);
-	attachToParent( getRegistryRoot(), gIOPowerPlane );
-	PMinit();
-	registerPowerDriver(this, patriarchPowerStates, 2);
-	makeUsable();
-	return true;
-}
-
-void
-IORootParent::shutDownSystem( void )
-{
-}
-
-void
-IORootParent::restartSystem( void )
-{
-}
-
-void
-IORootParent::sleepSystem( void )
-{
-}
-
-void
-IORootParent::dozeSystem( void )
-{
-}
-
-void
-IORootParent::sleepToDoze( void )
-{
-}
-
-void
-IORootParent::wakeSystem( void )
-{
-}
-
-OSSharedPtr<OSObject>
-IORootParent::copyProperty( const char * aKey) const
-{
-	return IOService::copyProperty(aKey);
-}
-
-uint32_t
-IOPMrootDomain::getWatchdogTimeout()
-{
-	if (gSwdSleepWakeTimeout) {
-		gSwdSleepTimeout = gSwdWakeTimeout = gSwdSleepWakeTimeout;
-	}
-	if ((pmTracer->getTracePhase() < kIOPMTracePointSystemSleep) ||
-	    (pmTracer->getTracePhase() == kIOPMTracePointDarkWakeEntry)) {
-		return gSwdSleepTimeout ? gSwdSleepTimeout : WATCHDOG_SLEEP_TIMEOUT;
-	} else {
-		return gSwdWakeTimeout ? gSwdWakeTimeout : WATCHDOG_WAKE_TIMEOUT;
-	}
-}
-
-
-#if defined(__i386__) || defined(__x86_64__) || (defined(__arm64__) && HIBERNATION)
-IOReturn
-IOPMrootDomain::restartWithStackshot()
-{
-	takeStackshot(true);
-
-	return kIOReturnSuccess;
-}
-
-void
-IOPMrootDomain::sleepWakeDebugTrig(bool wdogTrigger)
-{
-	takeStackshot(wdogTrigger);
-}
-
-void
-IOPMrootDomain::tracePhase2String(uint32_t tracePhase, const char **phaseString, const char **description)
-{
-	switch (tracePhase) {
-	case kIOPMTracePointSleepStarted:
-		*phaseString = "kIOPMTracePointSleepStarted";
-		*description = "starting sleep";
-		break;
-
-	case kIOPMTracePointSleepApplications:
-		*phaseString = "kIOPMTracePointSleepApplications";
-		*description = "notifying applications";
-		break;
-
-	case kIOPMTracePointSleepPriorityClients:
-		*phaseString = "kIOPMTracePointSleepPriorityClients";
-		*description = "notifying clients about upcoming system capability changes";
-		break;
-
-	case kIOPMTracePointSleepWillChangeInterests:
-		*phaseString = "kIOPMTracePointSleepWillChangeInterests";
-		*description = "creating hibernation file or while calling rootDomain's clients about upcoming rootDomain's state changes";
-		break;
-
-	case kIOPMTracePointSleepPowerPlaneDrivers:
-		*phaseString = "kIOPMTracePointSleepPowerPlaneDrivers";
-		*description = "calling power state change callbacks";
-		break;
-
-	case kIOPMTracePointSleepDidChangeInterests:
-		*phaseString = "kIOPMTracePointSleepDidChangeInterests";
-		*description = "calling rootDomain's clients about rootDomain's state changes";
-		break;
-
-	case kIOPMTracePointSleepCapabilityClients:
-		*phaseString = "kIOPMTracePointSleepCapabilityClients";
-		*description = "notifying clients about current system capabilities";
-		break;
-
-	case kIOPMTracePointSleepPlatformActions:
-		*phaseString = "kIOPMTracePointSleepPlatformActions";
-		*description = "calling Quiesce/Sleep action callbacks";
-		break;
-
-	case kIOPMTracePointSleepCPUs:
-	{
-		*phaseString = "kIOPMTracePointSleepCPUs";
-#if defined(__i386__) || defined(__x86_64__)
-		/*
-		 * We cannot use the getCPUNumber() method to get the cpu number, since
-		 * that cpu number is unrelated to the cpu number we need (we need the cpu
-		 * number as enumerated by the scheduler, NOT the CPU number enumerated
-		 * by ACPIPlatform as the CPUs are enumerated in MADT order).
-		 * Instead, pass the Mach processor pointer associated with the current
-		 * shutdown target so its associated cpu_id can be used in
-		 * processor_to_datastring.
-		 */
-		if (currentShutdownTarget != NULL &&
-		    currentShutdownTarget->getMachProcessor() != NULL) {
-			const char *sbuf = processor_to_datastring("halting all non-boot CPUs",
-			    currentShutdownTarget->getMachProcessor());
-			*description = sbuf;
-		} else {
-			*description = "halting all non-boot CPUs";
-		}
-#else
-		*description = "halting all non-boot CPUs";
-#endif
-		break;
-	}
-	case kIOPMTracePointSleepPlatformDriver:
-		*phaseString = "kIOPMTracePointSleepPlatformDriver";
-		*description = "executing platform specific code";
-		break;
-
-	case kIOPMTracePointHibernate:
-		*phaseString = "kIOPMTracePointHibernate";
-		*description = "writing the hibernation image";
-		break;
-
-	case kIOPMTracePointSystemSleep:
-		*phaseString = "kIOPMTracePointSystemSleep";
-		*description = "in EFI/Bootrom after last point of entry to sleep";
-		break;
-
-	case kIOPMTracePointWakePlatformDriver:
-		*phaseString = "kIOPMTracePointWakePlatformDriver";
-		*description = "executing platform specific code";
-		break;
-
-
-	case kIOPMTracePointWakePlatformActions:
-		*phaseString = "kIOPMTracePointWakePlatformActions";
-		*description = "calling Wake action callbacks";
-		break;
-
-	case kIOPMTracePointWakeCPUs:
-		*phaseString = "kIOPMTracePointWakeCPUs";
-		*description = "starting non-boot CPUs";
-		break;
-
-	case kIOPMTracePointWakeWillPowerOnClients:
-		*phaseString = "kIOPMTracePointWakeWillPowerOnClients";
-		*description = "sending kIOMessageSystemWillPowerOn message to kernel and userspace clients";
-		break;
-
-	case kIOPMTracePointWakeWillChangeInterests:
-		*phaseString = "kIOPMTracePointWakeWillChangeInterests";
-		*description = "calling rootDomain's clients about upcoming rootDomain's state changes";
-		break;
-
-	case kIOPMTracePointWakeDidChangeInterests:
-		*phaseString = "kIOPMTracePointWakeDidChangeInterests";
-		*description = "calling rootDomain's clients about completed rootDomain's state changes";
-		break;
-
-	case kIOPMTracePointWakePowerPlaneDrivers:
-		*phaseString = "kIOPMTracePointWakePowerPlaneDrivers";
-		*description = "calling power state change callbacks";
-		break;
-
-	case kIOPMTracePointWakeCapabilityClients:
-		*phaseString = "kIOPMTracePointWakeCapabilityClients";
-		*description = "informing clients about current system capabilities";
-		break;
-
-	case kIOPMTracePointWakeApplications:
-		*phaseString = "kIOPMTracePointWakeApplications";
-		*description = "sending asynchronous kIOMessageSystemHasPoweredOn message to userspace clients";
-		break;
-
-	case kIOPMTracePointDarkWakeEntry:
-		*phaseString = "kIOPMTracePointDarkWakeEntry";
-		*description = "entering darkwake on way to sleep";
-		break;
-
-	case kIOPMTracePointDarkWakeExit:
-		*phaseString = "kIOPMTracePointDarkWakeExit";
-		*description = "entering fullwake from darkwake";
-		break;
-
-	default:
-		*phaseString = NULL;
-		*description = NULL;
-	}
-}
-
-void
-IOPMrootDomain::saveFailureData2File()
-{
-	unsigned int len = 0;
-	char  failureStr[512];
-	errno_t error;
-	char *outbuf;
-	OSNumber *statusCode;
-	uint64_t pmStatusCode = 0;
-	uint32_t phaseData = 0;
-	uint32_t phaseDetail = 0;
-	bool efiFailure = false;
-
-	OSSharedPtr<OSObject> statusCodeProp = copyProperty(kIOPMSleepWakeFailureCodeKey);
-	statusCode = OSDynamicCast(OSNumber, statusCodeProp.get());
-	if (statusCode) {
-		pmStatusCode = statusCode->unsigned64BitValue();
-		phaseData = pmStatusCode & 0xFFFFFFFF;
-		phaseDetail = (pmStatusCode >> 32) & 0xFFFFFFFF;
-		if ((phaseData & 0xFF) == kIOPMTracePointSystemSleep) {
-			LOG("Sleep Wake failure in EFI\n");
-			efiFailure = true;
-			failureStr[0] = 0;
-			snprintf(failureStr, sizeof(failureStr), "Sleep Wake failure in EFI\n\nFailure code:: 0x%08x 0x%08x\n\nPlease IGNORE the below stackshot\n", phaseDetail, phaseData);
-			len = (typeof(len))strnlen(failureStr, sizeof(failureStr));
-		}
-	}
-
-	if (!efiFailure) {
-		if (PEReadNVRAMProperty(kIOSleepWakeFailurePanic, NULL, &len)) {
-			swd_flags |= SWD_BOOT_BY_SW_WDOG;
-			PERemoveNVRAMProperty(kIOSleepWakeFailurePanic);
-			// dump panic will handle saving nvram data
-			return;
-		}
-
-		/* Keeping this around for capturing data during power
-		 * button press */
-
-		if (!PEReadNVRAMProperty(kIOSleepWakeFailureString, NULL, &len)) {
-			DLOG("No sleep wake failure string\n");
-			return;
-		}
-		if (len == 0) {
-			DLOG("Ignoring zero byte SleepWake failure string\n");
-			goto exit;
-		}
-
-		// if PMStatus code is zero, delete stackshot and return
-		if (statusCode) {
-			if (((pmStatusCode & 0xFFFFFFFF) & 0xFF) == 0) {
-				// there was no sleep wake failure
-				// this can happen if delete stackshot was called
-				// before take stackshot completed. Let us delete any
-				// sleep wake failure data in nvram
-				DLOG("Deleting stackshot on successful wake\n");
-				deleteStackshot();
-				return;
-			}
-		}
-
-		if (len > sizeof(failureStr)) {
-			len = sizeof(failureStr);
-		}
-		failureStr[0] = 0;
-		PEReadNVRAMProperty(kIOSleepWakeFailureString, failureStr, &len);
-	}
-	if (failureStr[0] != 0) {
-		error = sleepWakeDebugSaveFile(kSleepWakeFailureStringFile, failureStr, len);
-		if (error) {
-			DLOG("Failed to save SleepWake failure string to file. error:%d\n", error);
-		} else {
-			DLOG("Saved SleepWake failure string to file.\n");
-		}
-	}
-
-	if (!OSCompareAndSwap(0, 1, &gRootDomain->swd_lock)) {
-		goto exit;
-	}
-
-	if (swd_buffer) {
-		unsigned int len = 0;
-		errno_t error;
-		char nvram_var_name_buffer[20];
-		unsigned int concat_len = 0;
-		swd_hdr      *hdr = NULL;
-
-
-		hdr = (swd_hdr *)swd_buffer;
-		outbuf = (char *)hdr + hdr->spindump_offset;
-		OSBoundedArrayRef<char> boundedOutBuf(outbuf, hdr->alloc_size - hdr->spindump_offset);
-
-		for (int i = 0; i < 8; i++) {
-			snprintf(nvram_var_name_buffer, sizeof(nvram_var_name_buffer), "%s%02d", SWD_STACKSHOT_VAR_PREFIX, i + 1);
-			if (!PEReadNVRAMProperty(nvram_var_name_buffer, NULL, &len)) {
-				LOG("No SleepWake blob to read beyond chunk %d\n", i);
-				break;
-			}
-			if (PEReadNVRAMProperty(nvram_var_name_buffer, boundedOutBuf.slice(concat_len, len).data(), &len) == FALSE) {
-				PERemoveNVRAMProperty(nvram_var_name_buffer);
-				LOG("Could not read the property :-(\n");
-				break;
-			}
-			PERemoveNVRAMProperty(nvram_var_name_buffer);
-			concat_len += len;
-		}
-		LOG("Concatenated length for the SWD blob %d\n", concat_len);
-
-		if (concat_len) {
-			error = sleepWakeDebugSaveFile(kSleepWakeStacksFilename, outbuf, concat_len);
-			if (error) {
-				LOG("Failed to save SleepWake zipped data to file. error:%d\n", error);
-			} else {
-				LOG("Saved SleepWake zipped data to file.\n");
-			}
-		} else {
-			// There is a sleep wake failure string but no stackshot
-			// Write a placeholder stacks file so that swd runs
-			snprintf(outbuf, 20, "%s", "No stackshot data\n");
-			error = sleepWakeDebugSaveFile(kSleepWakeStacksFilename, outbuf, 20);
-			if (error) {
-				LOG("Failed to save SleepWake zipped data to file. error:%d\n", error);
-			} else {
-				LOG("Saved SleepWake zipped data to file.\n");
-			}
-		}
-	} else {
-		LOG("No buffer allocated to save failure stackshot\n");
-	}
-
-
-	gRootDomain->swd_lock = 0;
-exit:
-	PERemoveNVRAMProperty(kIOSleepWakeFailureString);
-	return;
-}
-
-
-void
-IOPMrootDomain::getFailureData(thread_t *thread, char *failureStr, size_t strLen)
-{
-	OSSharedPtr<IORegistryIterator>    iter;
-	OSSharedPtr<const OSSymbol>        kextName = NULL;
-	IORegistryEntry *       entry;
-	IOService *             node;
-	bool                    nodeFound = false;
-
-	const void *            callMethod = NULL;
-	const char *            objectName = NULL;
-	const char *            phaseString = NULL;
-	const char *            phaseDescription = NULL;
-	uint64_t                delta;
-
-	IOPMServiceInterestNotifier *notifier = OSDynamicCast(IOPMServiceInterestNotifier, notifierObject.get());
-	uint32_t tracePhase = pmTracer->getTracePhase();
-
-	*thread = NULL;
-
-	delta = get_watchdog_elapsed_time();
-	if ((tracePhase < kIOPMTracePointSystemSleep) || (tracePhase == kIOPMTracePointDarkWakeEntry)) {
-		snprintf(failureStr, strLen, "Sleep transition timed out after %qd seconds", delta);
-	} else {
-		snprintf(failureStr, strLen, "Wake transition timed out after %qd seconds", delta);
-	}
-	tracePhase2String(tracePhase, &phaseString, &phaseDescription);
-
-	if (notifierThread) {
-		if (notifier && (notifier->identifier)) {
-			objectName = notifier->identifier->getCStringNoCopy();
-		}
-		*thread = notifierThread;
-	} else {
-		iter = IORegistryIterator::iterateOver(
-			getPMRootDomain(), gIOPowerPlane, kIORegistryIterateRecursively);
-
-		if (iter) {
-			while ((entry = iter->getNextObject())) {
-				node = OSDynamicCast(IOService, entry);
-				if (!node) {
-					continue;
-				}
-				if (OSDynamicCast(IOPowerConnection, node)) {
-					continue;
-				}
-
-				if (node->getBlockingDriverCall(thread, &callMethod)) {
-					nodeFound = true;
-					break;
-				}
-			}
-		}
-		if (nodeFound) {
-			kextName = copyKextIdentifierWithAddress((vm_address_t) callMethod);
-			if (kextName) {
-				objectName = kextName->getCStringNoCopy();
-			}
-		}
-	}
-	if (phaseDescription) {
-		strlcat(failureStr, " while ", strLen);
-		strlcat(failureStr, phaseDescription, strLen);
-		strlcat(failureStr, ".", strLen);
-	}
-	if (objectName) {
-		strlcat(failureStr, " Suspected bundle: ", strLen);
-		strlcat(failureStr, objectName, strLen);
-		strlcat(failureStr, ".", strLen);
-	}
-	if (*thread) {
-		char threadName[40];
-		snprintf(threadName, sizeof(threadName), " Thread 0x%llx.", thread_tid(*thread));
-		strlcat(failureStr, threadName, strLen);
-	}
-
-	DLOG("%s\n", failureStr);
-}
-
-struct swd_stackshot_compressed_data {
-	z_output_func   zoutput;
-	size_t                  zipped;
-	uint64_t                totalbytes;
-	uint64_t                lastpercent;
-	IOReturn                error;
-	unsigned                outremain;
-	unsigned                outlen;
-	unsigned                writes;
-	Bytef *                 outbuf;
-};
-struct swd_stackshot_compressed_data swd_zip_var = { };
-
-static void *
-swd_zs_alloc(void *__unused ref, u_int items, u_int size)
-{
-	void *result;
-	LOG("Alloc in zipping %d items of size %d\n", items, size);
-
-	result = (void *)(swd_zs_zmem + swd_zs_zoffset);
-	swd_zs_zoffset += ~31L & (31 + (items * size)); // 32b align for vector crc
-	LOG("Offset %zu\n", swd_zs_zoffset);
-	return result;
-}
-
-static int
-swd_zinput(z_streamp strm, Bytef *buf, unsigned size)
-{
-	unsigned len;
-
-	len = strm->avail_in;
-
-	if (len > size) {
-		len = size;
-	}
-	if (len == 0) {
-		return 0;
-	}
-
-	if (strm->next_in != (Bytef *) strm) {
-		memcpy(buf, strm->next_in, len);
-	} else {
-		bzero(buf, len);
-	}
-
-	strm->adler = z_crc32(strm->adler, buf, len);
-
-	strm->avail_in -= len;
-	strm->next_in  += len;
-	strm->total_in += len;
-
-	return (int)len;
-}
-
-static int
-swd_zoutput(z_streamp strm, Bytef *buf, unsigned len)
-{
-	unsigned int i = 0;
-	// if outlen > max size don't add to the buffer
-	assert(buf != NULL);
-	if (strm && buf) {
-		if (swd_zip_var.outlen + len > SWD_COMPRESSED_BUFSIZE) {
-			LOG("No space to GZIP... not writing to NVRAM\n");
-			return len;
-		}
-	}
-	for (i = 0; i < len; i++) {
-		*(swd_zip_var.outbuf + swd_zip_var.outlen + i) = *(buf + i);
-	}
-	swd_zip_var.outlen += len;
-	return len;
-}
-
-static void
-swd_zs_free(void * __unused ref, void * __unused ptr)
-{
-}
-
-static int
-swd_compress(char *inPtr, char *outPtr, size_t numBytes)
-{
-	int wbits = 12;
-	int memlevel = 3;
-
-	if (((unsigned int) numBytes) != numBytes) {
-		return 0;
-	}
-
-	if (!swd_zs.zalloc) {
-		swd_zs.zalloc = swd_zs_alloc;
-		swd_zs.zfree = swd_zs_free;
-		if (deflateInit2(&swd_zs, Z_BEST_SPEED, Z_DEFLATED, wbits + 16, memlevel, Z_DEFAULT_STRATEGY)) {
-			// allocation failed
-			bzero(&swd_zs, sizeof(swd_zs));
-			// swd_zs_zoffset = 0;
-		} else {
-			LOG("PMRD inited the zlib allocation routines\n");
-		}
-	}
-
-	swd_zip_var.zipped = 0;
-	swd_zip_var.totalbytes = 0; // should this be the max that we have?
-	swd_zip_var.lastpercent = 0;
-	swd_zip_var.error = kIOReturnSuccess;
-	swd_zip_var.outremain = 0;
-	swd_zip_var.outlen = 0;
-	swd_zip_var.writes = 0;
-	swd_zip_var.outbuf = (Bytef *)outPtr;
-
-	swd_zip_var.totalbytes = numBytes;
-
-	swd_zs.avail_in = 0;
-	swd_zs.next_in = NULL;
-	swd_zs.avail_out = 0;
-	swd_zs.next_out = NULL;
-
-	deflateResetWithIO(&swd_zs, swd_zinput, swd_zoutput);
-
-	z_stream *zs;
-	int zr;
-	zs = &swd_zs;
-
-	while (swd_zip_var.error >= 0) {
-		if (!zs->avail_in) {
-			zs->next_in = (unsigned char *)inPtr ? (Bytef *)inPtr : (Bytef *)zs; /* zero marker? */
-			zs->avail_in = (unsigned int) numBytes;
-		}
-		if (!zs->avail_out) {
-			zs->next_out = (Bytef *)zs;
-			zs->avail_out = UINT32_MAX;
-		}
-		zr = deflate(zs, Z_NO_FLUSH);
-		if (Z_STREAM_END == zr) {
-			break;
-		}
-		if (zr != Z_OK) {
-			LOG("ZERR %d\n", zr);
-			swd_zip_var.error = zr;
-		} else {
-			if (zs->total_in == numBytes) {
-				break;
-			}
-		}
-	}
-
-	//now flush the stream
-	while (swd_zip_var.error >= 0) {
-		if (!zs->avail_out) {
-			zs->next_out = (Bytef *)zs;
-			zs->avail_out = UINT32_MAX;
-		}
-		zr = deflate(zs, Z_FINISH);
-		if (Z_STREAM_END == zr) {
-			break;
-		}
-		if (zr != Z_OK) {
-			LOG("ZERR %d\n", zr);
-			swd_zip_var.error = zr;
-		} else {
-			if (zs->total_in == numBytes) {
-				LOG("Total output size %d\n", swd_zip_var.outlen);
-				break;
-			}
-		}
-	}
-
-	return swd_zip_var.outlen;
-}
-
-void
-IOPMrootDomain::deleteStackshot()
-{
-	if (!OSCompareAndSwap(0, 1, &gRootDomain->swd_lock)) {
-		// takeStackshot hasn't completed
-		return;
-	}
-	LOG("Deleting any sleepwake failure data in nvram\n");
-
-	PERemoveNVRAMProperty(kIOSleepWakeFailureString);
-	char nvram_var_name_buf[20];
-	for (int i = 0; i < 8; i++) {
-		snprintf(nvram_var_name_buf, sizeof(nvram_var_name_buf), "%s%02d", SWD_STACKSHOT_VAR_PREFIX, i + 1);
-		if (PERemoveNVRAMProperty(nvram_var_name_buf) == false) {
-			LOG("Removing %s returned false\n", nvram_var_name_buf);
-		}
-	}
-	// force NVRAM sync
-	if (PEWriteNVRAMProperty(kIONVRAMSyncNowPropertyKey, kIONVRAMSyncNowPropertyKey, (unsigned int) strlen(kIONVRAMSyncNowPropertyKey)) == false) {
-		DLOG("Failed to force nvram sync\n");
-	}
-	gRootDomain->swd_lock = 0;
-}
-
-void
-IOPMrootDomain::takeStackshot(bool wdogTrigger)
-{
-	swd_hdr *                hdr = NULL;
-	int                      cnt = 0;
-	int                      max_cnt;
-	pid_t                    pid = 0;
-	kern_return_t            kr = KERN_SUCCESS;
-	uint64_t                 flags;
-
-	char *                   dstAddr;
-	uint32_t                 size;
-	uint32_t                 bytesRemaining;
-	unsigned                 bytesWritten = 0;
-
-	char                     failureStr[512];
-	thread_t                 thread = NULL;
-	const char *             swfPanic = "swfPanic";
-
-	uint32_t                 bufSize;
-	int                      success = 0;
-
-#if defined(__i386__) || defined(__x86_64__)
-	const bool               concise = false;
-#else
-	const bool               concise = true;
-#endif
-
-	if (!OSCompareAndSwap(0, 1, &gRootDomain->swd_lock)) {
-		return;
-	}
-
-	failureStr[0] = 0;
-	if ((kIOSleepWakeWdogOff & gIOKitDebug) || systemBooting || systemShutdown || gWillShutdown) {
-		return;
-	}
-
-	if (wdogTrigger) {
-		getFailureData(&thread, failureStr, sizeof(failureStr));
-
-		if (concise || (PEGetCoprocessorVersion() >= kCoprocessorVersion2)) {
-			goto skip_stackshot;
-		}
-	} else {
-		AbsoluteTime now;
-		uint64_t nsec;
-		clock_get_uptime(&now);
-		SUB_ABSOLUTETIME(&now, &gIOLastWakeAbsTime);
-		absolutetime_to_nanoseconds(now, &nsec);
-		snprintf(failureStr, sizeof(failureStr), "Power button pressed during wake transition after %u ms.\n", ((int)((nsec) / NSEC_PER_MSEC)));
-	}
-
-	if (swd_buffer == NULL) {
-		sleepWakeDebugMemAlloc();
-		if (swd_buffer == NULL) {
-			return;
-		}
-	}
-	hdr = (swd_hdr *)swd_buffer;
-	bufSize = hdr->alloc_size;
-
-	dstAddr = (char*)hdr + hdr->spindump_offset;
-	flags = STACKSHOT_KCDATA_FORMAT | STACKSHOT_NO_IO_STATS | STACKSHOT_SAVE_KEXT_LOADINFO | STACKSHOT_ACTIVE_KERNEL_THREADS_ONLY | STACKSHOT_THREAD_WAITINFO | STACKSHOT_INCLUDE_DRIVER_THREADS_IN_KERNEL;
-
-	/* If not wdogTrigger only take kernel tasks stackshot
-	 */
-	if (wdogTrigger) {
-		pid = -1;
-		max_cnt = 3;
-	} else {
-		pid = 0;
-		max_cnt = 2;
-	}
-
-	/* Attempt to take stackshot with all ACTIVE_KERNEL_THREADS
-	 * If we run out of space, take stackshot with only kernel task
-	 */
-	while (success == 0 && cnt < max_cnt) {
-		bytesRemaining = bufSize - hdr->spindump_offset;
-		cnt++;
-		DLOG("Taking snapshot. bytesRemaining: %d\n", bytesRemaining);
-
-		size = bytesRemaining;
-		kr = stack_snapshot_from_kernel(pid, dstAddr, size, flags, 0, 0, &bytesWritten);
-		DLOG("stack_snapshot_from_kernel returned 0x%x. pid: %d bufsize:0x%x flags:0x%llx bytesWritten: %d\n",
-		    kr, pid, size, flags, bytesWritten);
-		if (kr == KERN_INSUFFICIENT_BUFFER_SIZE) {
-			if (pid == -1) {
-				pid = 0;
-			} else if (flags & STACKSHOT_INCLUDE_DRIVER_THREADS_IN_KERNEL) {
-				flags = flags & ~STACKSHOT_INCLUDE_DRIVER_THREADS_IN_KERNEL;
-			} else {
-				LOG("Insufficient buffer size for only kernel task\n");
-				break;
-			}
-		}
-		if (kr == KERN_SUCCESS) {
-			if (bytesWritten == 0) {
-				MSG("Failed to get stackshot(0x%x) bufsize:0x%x flags:0x%llx\n", kr, size, flags);
-				continue;
-			}
-			bytesRemaining -= bytesWritten;
-			hdr->spindump_size = (bufSize - bytesRemaining - hdr->spindump_offset);
-
-			memset(hdr->reason, 0x20, sizeof(hdr->reason));
-
-			// Compress stackshot and save to NVRAM
-			{
-				char *outbuf = (char *)swd_compressed_buffer;
-				int outlen = 0;
-				int num_chunks = 0;
-				int max_chunks = 0;
-				int leftover = 0;
-				char nvram_var_name_buffer[20];
-
-				outlen = swd_compress((char*)hdr + hdr->spindump_offset, outbuf, bytesWritten);
-
-				if (outlen) {
-					max_chunks = outlen / (2096 - 200);
-					leftover = outlen % (2096 - 200);
-
-					if (max_chunks < 8) {
-						for (num_chunks = 0; num_chunks < max_chunks; num_chunks++) {
-							snprintf(nvram_var_name_buffer, sizeof(nvram_var_name_buffer), "%s%02d", SWD_STACKSHOT_VAR_PREFIX, num_chunks + 1);
-							if (PEWriteNVRAMPropertyWithCopy(nvram_var_name_buffer, (outbuf + (num_chunks * (2096 - 200))), (2096 - 200)) == FALSE) {
-								LOG("Failed to update NVRAM %d\n", num_chunks);
-								break;
-							}
-						}
-						if (leftover) {
-							snprintf(nvram_var_name_buffer, sizeof(nvram_var_name_buffer), "%s%02d", SWD_STACKSHOT_VAR_PREFIX, num_chunks + 1);
-							if (PEWriteNVRAMPropertyWithCopy(nvram_var_name_buffer, (outbuf + (num_chunks * (2096 - 200))), leftover) == FALSE) {
-								LOG("Failed to update NVRAM with leftovers\n");
-							}
-						}
-						success = 1;
-						LOG("Successfully saved stackshot to NVRAM\n");
-					} else {
-						if (pid == -1) {
-							LOG("Compressed failure stackshot is too large. size=%d bytes\n", outlen);
-							pid = 0;
-						} else if (flags & STACKSHOT_INCLUDE_DRIVER_THREADS_IN_KERNEL) {
-							LOG("Compressed failure stackshot of kernel+dexts is too large size=%d bytes\n", outlen);
-							flags = flags & ~STACKSHOT_INCLUDE_DRIVER_THREADS_IN_KERNEL;
-						} else {
-							LOG("Compressed failure stackshot of only kernel is too large size=%d bytes\n", outlen);
-							break;
-						}
-					}
-				}
-			}
-		}
-	}
-
-	if (failureStr[0]) {
-		// append sleep-wake failure code
-		char traceCode[80];
-		snprintf(traceCode, sizeof(traceCode), "\nFailure code:: 0x%08x %08x\n",
-		    pmTracer->getTraceData(), pmTracer->getTracePhase());
-		strlcat(failureStr, traceCode, sizeof(failureStr));
-		if (PEWriteNVRAMProperty(kIOSleepWakeFailureString, failureStr, (unsigned int) strnlen(failureStr, sizeof(failureStr))) == false) {
-			DLOG("Failed to write SleepWake failure string\n");
-		}
-	}
-
-	// force NVRAM sync
-	if (PEWriteNVRAMProperty(kIONVRAMSyncNowPropertyKey, kIONVRAMSyncNowPropertyKey, (unsigned int) strlen(kIONVRAMSyncNowPropertyKey)) == false) {
-		DLOG("Failed to force nvram sync\n");
-	}
-
-skip_stackshot:
-	if (wdogTrigger) {
-		if (PEGetCoprocessorVersion() < kCoprocessorVersion2) {
-			if (swd_flags & SWD_BOOT_BY_SW_WDOG) {
-				// If current boot is due to this watch dog trigger restart in previous boot,
-				// then don't trigger again until at least 1 successful sleep & wake.
-				if (!(sleepCnt && (displayWakeCnt || darkWakeCnt))) {
-					LOG("Shutting down due to repeated Sleep/Wake failures\n");
-					updateTasksSuspend(kTasksSuspendSuspended, kTasksSuspendNoChange);
-					PEHaltRestart(kPEHaltCPU);
-					return;
-				}
-			}
-			if (gSwdPanic == 0) {
-				LOG("Calling panic prevented by swd_panic boot-args. Calling restart");
-				updateTasksSuspend(kTasksSuspendSuspended, kTasksSuspendNoChange);
-				PEHaltRestart(kPERestartCPU);
-			}
-		}
-		if (!concise && (PEWriteNVRAMProperty(kIOSleepWakeFailurePanic, swfPanic, (unsigned int) strlen(swfPanic)) == false)) {
-			DLOG("Failed to write SleepWake failure panic key\n");
-		}
-#if defined(__x86_64__)
-		if (thread) {
-			panic_with_thread_context(0, NULL, DEBUGGER_OPTION_ATTEMPTCOREDUMPANDREBOOT, thread, "%s", failureStr);
-		} else
-#endif /* defined(__x86_64__) */
-		{
-			panic_with_options(0, NULL, DEBUGGER_OPTION_ATTEMPTCOREDUMPANDREBOOT, "%s", failureStr);
-		}
-	} else {
-		gRootDomain->swd_lock = 0;
-		return;
-	}
-}
-
-void
-IOPMrootDomain::sleepWakeDebugMemAlloc()
-{
-	vm_size_t    size = SWD_STACKSHOT_SIZE + SWD_COMPRESSED_BUFSIZE + SWD_ZLIB_BUFSIZE;
-
-	swd_hdr      *hdr = NULL;
-	void         *bufPtr = NULL;
-
-	OSSharedPtr<IOBufferMemoryDescriptor>  memDesc;
-
-
-	if (kIOSleepWakeWdogOff & gIOKitDebug) {
-		return;
-	}
-
-	if (!OSCompareAndSwap(0, 1, &gRootDomain->swd_lock)) {
-		return;
-	}
-
-	memDesc = IOBufferMemoryDescriptor::inTaskWithOptions(
-		kernel_task, kIODirectionIn | kIOMemoryMapperNone,
-		size);
-	if (memDesc == NULL) {
-		DLOG("Failed to allocate Memory descriptor for sleepWake debug\n");
-		goto exit;
-	}
-
-	bufPtr = memDesc->getBytesNoCopy();
-
-	// Carve out memory for zlib routines
-	swd_zs_zmem = (vm_offset_t)bufPtr;
-	bufPtr = (char *)bufPtr + SWD_ZLIB_BUFSIZE;
-
-	// Carve out memory for compressed stackshots
-	swd_compressed_buffer = bufPtr;
-	bufPtr = (char *)bufPtr + SWD_COMPRESSED_BUFSIZE;
-
-	// Remaining is used for holding stackshot
-	hdr = (swd_hdr *)bufPtr;
-	memset(hdr, 0, sizeof(swd_hdr));
-
-	hdr->signature = SWD_HDR_SIGNATURE;
-	hdr->alloc_size = SWD_STACKSHOT_SIZE;
-
-	hdr->spindump_offset = sizeof(swd_hdr);
-	swd_buffer = (void *)hdr;
-	swd_memDesc = os::move(memDesc);
-	DLOG("SleepWake debug buffer size:0x%x spindump offset:0x%x\n", hdr->alloc_size, hdr->spindump_offset);
-
-exit:
-	gRootDomain->swd_lock = 0;
-}
-
-void
-IOPMrootDomain::sleepWakeDebugSpinDumpMemAlloc()
-{
-#if UNUSED
-	vm_size_t    size = SWD_SPINDUMP_SIZE;
-
-	swd_hdr      *hdr = NULL;
-
-	OSSharedPtr<IOBufferMemoryDescriptor>  memDesc;
-
-	if (!OSCompareAndSwap(0, 1, &gRootDomain->swd_lock)) {
-		return;
-	}
-
-	memDesc = IOBufferMemoryDescriptor::inTaskWithOptions(
-		kernel_task, kIODirectionIn | kIOMemoryMapperNone,
-		SWD_SPINDUMP_SIZE);
-
-	if (memDesc == NULL) {
-		DLOG("Failed to allocate Memory descriptor for sleepWake debug spindump\n");
-		goto exit;
-	}
-
-
-	hdr = (swd_hdr *)memDesc->getBytesNoCopy();
-	memset(hdr, 0, sizeof(swd_hdr));
-
-	hdr->signature = SWD_HDR_SIGNATURE;
-	hdr->alloc_size = size;
-
-	hdr->spindump_offset = sizeof(swd_hdr);
-	swd_spindump_buffer = (void *)hdr;
-	swd_spindump_memDesc = os::move(memDesc);
-
-exit:
-	gRootDomain->swd_lock = 0;
-#endif /* UNUSED */
-}
-
-void
-IOPMrootDomain::sleepWakeDebugEnableWdog()
-{
-}
-
-bool
-IOPMrootDomain::sleepWakeDebugIsWdogEnabled()
-{
-	return !systemBooting && !systemShutdown && !gWillShutdown;
-}
-
-void
-IOPMrootDomain::sleepWakeDebugSaveSpinDumpFile()
-{
-	swd_hdr *hdr = NULL;
-	errno_t error = EIO;
-
-	if (swd_spindump_buffer && gSpinDumpBufferFull) {
-		hdr = (swd_hdr *)swd_spindump_buffer;
-
-		error = sleepWakeDebugSaveFile("/var/tmp/SleepWakeDelayStacks.dump",
-		    (char*)hdr + hdr->spindump_offset, hdr->spindump_size);
-
-		if (error) {
-			return;
-		}
-
-		sleepWakeDebugSaveFile("/var/tmp/SleepWakeDelayLog.dump",
-		    (char*)hdr + offsetof(swd_hdr, UUID),
-		    sizeof(swd_hdr) - offsetof(swd_hdr, UUID));
-
-		gSpinDumpBufferFull = false;
-	}
-}
-
-errno_t
-IOPMrootDomain::sleepWakeDebugSaveFile(const char *name, char *buf, int len)
-{
-	struct vnode         *vp = NULL;
-	vfs_context_t        ctx = vfs_context_create(vfs_context_current());
-	kauth_cred_t         cred = vfs_context_ucred(ctx);
-	struct vnode_attr    va;
-	errno_t      error = EIO;
-
-	if (vnode_open(name, (O_CREAT | FWRITE | O_NOFOLLOW),
-	    S_IRUSR | S_IRGRP | S_IROTH, VNODE_LOOKUP_NOFOLLOW, &vp, ctx) != 0) {
-		LOG("Failed to open the file %s\n", name);
-		swd_flags |= SWD_FILEOP_ERROR;
-		goto exit;
-	}
-	VATTR_INIT(&va);
-	VATTR_WANTED(&va, va_nlink);
-	/* Don't dump to non-regular files or files with links. */
-	if (vp->v_type != VREG ||
-	    vnode_getattr(vp, &va, ctx) || va.va_nlink != 1) {
-		LOG("Bailing as this is not a regular file\n");
-		swd_flags |= SWD_FILEOP_ERROR;
-		goto exit;
-	}
-	VATTR_INIT(&va);
-	VATTR_SET(&va, va_data_size, 0);
-	vnode_setattr(vp, &va, ctx);
-
-
-	if (buf != NULL) {
-		error = vn_rdwr(UIO_WRITE, vp, buf, len, 0,
-		    UIO_SYSSPACE, IO_NODELOCKED | IO_UNIT, cred, (int *) NULL, vfs_context_proc(ctx));
-		if (error != 0) {
-			LOG("Failed to save sleep wake log. err 0x%x\n", error);
-			swd_flags |= SWD_FILEOP_ERROR;
-		} else {
-			DLOG("Saved %d bytes to file %s\n", len, name);
-		}
-	}
-
-exit:
-	if (vp) {
-		vnode_close(vp, FWRITE, ctx);
-	}
-	if (ctx) {
-		vfs_context_rele(ctx);
-	}
-
-	return error;
-}
-
-#else /* defined(__i386__) || defined(__x86_64__) */
-
-void
-IOPMrootDomain::sleepWakeDebugTrig(bool restart)
-{
-	if (restart) {
-		if (gSwdPanic == 0) {
-			return;
-		}
-		panic("Sleep/Wake hang detected");
-		return;
-	}
-}
-
-void
-IOPMrootDomain::takeStackshot(bool restart)
-{
-#pragma unused(restart)
-}
-
-void
-IOPMrootDomain::deleteStackshot()
-{
-}
-
-void
-IOPMrootDomain::sleepWakeDebugMemAlloc()
-{
-}
-
-void
-IOPMrootDomain::saveFailureData2File()
-{
-}
-
-void
-IOPMrootDomain::sleepWakeDebugEnableWdog()
-{
-}
-
-bool
-IOPMrootDomain::sleepWakeDebugIsWdogEnabled()
-{
-	return false;
-}
-
-void
-IOPMrootDomain::sleepWakeDebugSaveSpinDumpFile()
-{
-}
-
-errno_t
-IOPMrootDomain::sleepWakeDebugSaveFile(const char *name, char *buf, int len)
-{
-	return 0;
-}
-
-#endif /* defined(__i386__) || defined(__x86_64__) */
-
+void IORootParent::initialize( void )
+{
+}
+
+bool IORootParent::start( IOService * nub )
+{
+    IOService::start(nub);
+    attachToParent( getRegistryRoot(), gIOPowerPlane );
+    PMinit();
+    registerPowerDriver(this, patriarchPowerStates, 2);
+    makeUsable();
+    return true;
+}
+
+void IORootParent::shutDownSystem( void )
+{
+}
+
+void IORootParent::restartSystem( void )
+{
+}
+
+void IORootParent::sleepSystem( void )
+{
+}
+
+void IORootParent::dozeSystem( void )
+{
+}
+
+void IORootParent::sleepToDoze( void )
+{
+}
+
+void IORootParent::wakeSystem( void )
+{
+}
+
+OSObject * IORootParent::copyProperty( const char * aKey) const
+{
+    return (IOService::copyProperty(aKey));
+}
+