Libc-763.11.tar.gz

[apple/libc.git] / sys / OSThermalNotification.c

diff --git a/sys/OSThermalNotification.c b/sys/OSThermalNotification.c
index cd4d2b7852001d3d519fa63528930f00ddf8f6ff..618bf58ef787079b5cae2b2a137536bbbdccdaa8 100644 (file)
--- a/sys/OSThermalNotification.c
+++ b/sys/OSThermalNotification.c
@@ -21,24 +21,55 @@
  * @APPLE_LICENSE_HEADER_END@
  */
 
+#include <dispatch/dispatch.h>
 #include <libkern/OSThermalNotification.h>
 #include <notify.h>
 
-const char *kOSThermalNotificationName = "com.apple.system.thermalstatus";
+#define OSThermalStatusName "com.apple.system.thermalstatus"
 
-OSThermalNotificationLevel OSThermalNotificationCurrentLevel(void)
-{
-       uint64_t val;
-       int token;
-
-       if (NOTIFY_STATUS_OK != notify_register_check(kOSThermalNotificationName, &token))
-               return OSThermalNotificationLevelAny;
+const char * const kOSThermalNotificationName = OSThermalStatusName; 
 
-       if (NOTIFY_STATUS_OK != notify_get_state(token, &val))
-               return OSThermalNotificationLevelAny;
+static const char * const kOSThermalMitigationNames[kOSThermalMitigationCount] = {
+       OSThermalStatusName,
+       "com.apple.system.thermalmitigation.70percenttorch",
+       "com.apple.system.thermalmitigation.70percentbacklight",
+       "com.apple.system.thermalmitigation.50percenttorch",
+       "com.apple.system.thermalmitigation.50percentbacklight",
+       "com.apple.system.thermalmitigation.disabletorch",
+       "com.apple.system.thermalmitigation.25percentbacklight",
+       "com.apple.system.thermalmitigation.disablemapshalo",
+       "com.apple.system.thermalmitigation.appterminate",
+       "com.apple.system.thermalmitigation.devicerestart"
+};
 
-       if (NOTIFY_STATUS_OK != notify_cancel(token))
-               return OSThermalNotificationLevelAny;
+static int tokens[kOSThermalMitigationCount];
+static dispatch_once_t predicates[kOSThermalMitigationCount];
 
+OSThermalNotificationLevel _OSThermalNotificationLevelForBehavior(int behavior)
+{
+       uint64_t val = OSThermalNotificationLevelAny;
+       if (behavior >= 0 && behavior < kOSThermalMitigationCount) {
+               dispatch_once(&predicates[behavior], ^{
+                               (void)notify_register_check(kOSThermalMitigationNames[behavior], &tokens[behavior]);
+               });
+               (void)notify_get_state(tokens[behavior], &val);
+       }
        return (OSThermalNotificationLevel)val;
 }
+
+void _OSThermalNotificationSetLevelForBehavior(int level, int behavior)
+{
+       uint64_t val = (uint64_t)level;
+       if (behavior >= 0 && behavior < kOSThermalMitigationCount) {
+               dispatch_once(&predicates[behavior], ^{
+                               (void)notify_register_check(kOSThermalMitigationNames[behavior], &tokens[behavior]);
+               });
+               (void)notify_set_state(tokens[behavior], val);
+       }
+}
+
+
+OSThermalNotificationLevel OSThermalNotificationCurrentLevel(void)
+{
+       return _OSThermalNotificationLevelForBehavior(kOSThermalMitigationNone);
+}