| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 // Implementation based on sample code from | 5 // Implementation based on sample code from |
| 6 // http://developer.apple.com/library/mac/#qa/qa1340/_index.html. | 6 // http://developer.apple.com/library/mac/#qa/qa1340/_index.html. |
| 7 | 7 |
| 8 #include "base/system_monitor/system_monitor.h" | 8 #include "base/power_monitor/power_monitor.h" |
| 9 | 9 |
| 10 #include <IOKit/pwr_mgt/IOPMLib.h> | 10 #include <IOKit/pwr_mgt/IOPMLib.h> |
| 11 #include <IOKit/IOMessage.h> | 11 #include <IOKit/IOMessage.h> |
| 12 | 12 |
| 13 namespace base { | 13 namespace base { |
| 14 | 14 |
| 15 namespace { | 15 namespace { |
| 16 | 16 |
| 17 io_connect_t g_system_power_io_port = 0; | 17 io_connect_t g_system_power_io_port = 0; |
| 18 IONotificationPortRef g_notification_port_ref = 0; | 18 IONotificationPortRef g_notification_port_ref = 0; |
| 19 io_object_t g_notifier_object = 0; | 19 io_object_t g_notifier_object = 0; |
| 20 PowerMonitor::Signaler* g_power_signaler = NULL; |
| 20 | 21 |
| 21 void SystemPowerEventCallback(void*, | 22 void SystemPowerEventCallback(void*, |
| 22 io_service_t service, | 23 io_service_t service, |
| 23 natural_t message_type, | 24 natural_t message_type, |
| 24 void* message_argument) { | 25 void* message_argument) { |
| 25 SystemMonitor* sys_monitor = SystemMonitor::Get(); | 26 |
| 26 DCHECK(sys_monitor); | 27 CHECK(g_power_signaler); |
| 27 switch (message_type) { | 28 switch (message_type) { |
| 28 case kIOMessageSystemWillSleep: | 29 case kIOMessageSystemWillSleep: |
| 29 sys_monitor->ProcessPowerMessage(SystemMonitor::SUSPEND_EVENT); | 30 g_power_signaler->ProcessPowerEvent(base::PowerMonitor::SUSPEND_EVENT); |
| 30 IOAllowPowerChange(g_system_power_io_port, | 31 IOAllowPowerChange(g_system_power_io_port, |
| 31 reinterpret_cast<intptr_t>(message_argument)); | 32 reinterpret_cast<intptr_t>(message_argument)); |
| 32 break; | 33 break; |
| 33 | 34 |
| 34 case kIOMessageSystemWillPowerOn: | 35 case kIOMessageSystemWillPowerOn: |
| 35 sys_monitor->ProcessPowerMessage(SystemMonitor::RESUME_EVENT); | 36 g_power_signaler->ProcessPowerEvent(PowerMonitor::RESUME_EVENT); |
| 36 break; | 37 break; |
| 37 } | 38 } |
| 38 } | 39 } |
| 39 | 40 |
| 40 } // namespace | 41 } // namespace |
| 41 | 42 |
| 42 // The reason we can't include this code in the constructor is because | 43 // The reason we can't include this code in the constructor is because |
| 43 // PlatformInit() requires an active runloop and the IO port needs to be | 44 // PlatformInit() requires an active runloop and the IO port needs to be |
| 44 // allocated at sandbox initialization time, before there's a runloop. | 45 // allocated at sandbox initialization time, before there's a runloop. |
| 45 // See crbug.com/83783 . | 46 // See crbug.com/83783 . |
| 46 | 47 |
| 47 // static | 48 // static |
| 48 void SystemMonitor::AllocateSystemIOPorts() { | 49 void PowerMonitor::AllocateSystemIOPorts() { |
| 49 DCHECK_EQ(g_system_power_io_port, 0u); | 50 DCHECK_EQ(g_system_power_io_port, 0u); |
| 51 if (g_system_power_io_port != 0u) |
| 52 return; |
| 50 | 53 |
| 51 // Notification port allocated by IORegisterForSystemPower. | 54 // Notification port allocated by IORegisterForSystemPower. |
| 52 | 55 |
| 53 g_system_power_io_port = IORegisterForSystemPower( | 56 g_system_power_io_port = IORegisterForSystemPower( |
| 54 NULL, &g_notification_port_ref, SystemPowerEventCallback, | 57 NULL, &g_notification_port_ref, SystemPowerEventCallback, |
| 55 &g_notifier_object); | 58 &g_notifier_object); |
| 56 | 59 |
| 57 DCHECK_NE(g_system_power_io_port, 0u); | 60 DCHECK_NE(g_system_power_io_port, 0u); |
| 58 } | 61 } |
| 59 | 62 |
| 60 void SystemMonitor::PlatformInit() { | 63 void PowerMonitor::PlatformInit() { |
| 61 // Need to call AllocateSystemIOPorts() before constructing a SystemMonitor | 64 // Need to call AllocateSystemIOPorts() before creating a PowerMonitor |
| 62 // object. | 65 // object. |
| 63 DCHECK_NE(g_system_power_io_port, 0u); | 66 CHECK_NE(g_system_power_io_port, 0u); |
| 64 if (g_system_power_io_port == 0) | 67 |
| 65 return; | 68 CHECK(!g_power_signaler); |
| 69 g_power_signaler = PowerMonitor::Get()->GetSignalerOnce(); |
| 66 | 70 |
| 67 // Add the notification port to the application runloop | 71 // Add the notification port to the application runloop |
| 68 CFRunLoopAddSource( | 72 CFRunLoopAddSource( |
| 69 CFRunLoopGetCurrent(), | 73 CFRunLoopGetCurrent(), |
| 70 IONotificationPortGetRunLoopSource(g_notification_port_ref), | 74 IONotificationPortGetRunLoopSource(g_notification_port_ref), |
| 71 kCFRunLoopCommonModes); | 75 kCFRunLoopCommonModes); |
| 72 } | 76 } |
| 73 | 77 |
| 74 void SystemMonitor::PlatformDestroy() { | 78 |
| 79 void PowerMonitor::PlatformDestroy() { |
| 75 DCHECK_NE(g_system_power_io_port, 0u); | 80 DCHECK_NE(g_system_power_io_port, 0u); |
| 81 |
| 76 if (g_system_power_io_port == 0) | 82 if (g_system_power_io_port == 0) |
| 77 return; | 83 return; |
| 78 | 84 |
| 79 // Remove the sleep notification port from the application runloop | 85 // Remove the sleep notification port from the application runloop |
| 80 CFRunLoopRemoveSource( | 86 CFRunLoopRemoveSource( |
| 81 CFRunLoopGetCurrent(), | 87 CFRunLoopGetCurrent(), |
| 82 IONotificationPortGetRunLoopSource(g_notification_port_ref), | 88 IONotificationPortGetRunLoopSource(g_notification_port_ref), |
| 83 kCFRunLoopCommonModes); | 89 kCFRunLoopCommonModes); |
| 84 | 90 |
| 85 // Deregister for system sleep notifications | 91 // Deregister for system sleep notifications |
| 86 IODeregisterForSystemPower(&g_notifier_object); | 92 IODeregisterForSystemPower(&g_notifier_object); |
| 87 | 93 |
| 88 // IORegisterForSystemPower implicitly opens the Root Power Domain IOService, | 94 // IORegisterForSystemPower implicitly opens the Root Power Domain IOService, |
| 89 // so we close it here. | 95 // so we close it here. |
| 90 IOServiceClose(g_system_power_io_port); | 96 IOServiceClose(g_system_power_io_port); |
| 91 | 97 |
| 92 g_system_power_io_port = 0; | 98 g_system_power_io_port = 0; |
| 93 | 99 |
| 94 // Destroy the notification port allocated by IORegisterForSystemPower. | 100 // Destroy the notification port allocated by IORegisterForSystemPower. |
| 95 IONotificationPortDestroy(g_notification_port_ref); | 101 IONotificationPortDestroy(g_notification_port_ref); |
| 96 } | 102 } |
| 97 | 103 |
| 98 } // namespace base | 104 } // namespace base |
| OLD | NEW |