Libc-1353.11.2.tar.gz

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

diff --git a/sys/OSThermalNotification.c b/sys/OSThermalNotification.c
index 618bf58ef787079b5cae2b2a137536bbbdccdaa8..36b0042a96426db35f8adc5a13caab8e9c43f83f 100644 (file)
--- a/sys/OSThermalNotification.c
+++ b/sys/OSThermalNotification.c
@@ -21,13 +21,25 @@
  * @APPLE_LICENSE_HEADER_END@
  */
 
-#include <dispatch/dispatch.h>
 #include <libkern/OSThermalNotification.h>
+
+#include <TargetConditionals.h>
+
+#define OSThermalPressureLevelName             "com.apple.system.thermalpressurelevel"
+const char * const kOSThermalNotificationPressureLevelName = OSThermalPressureLevelName;
+
+#if TARGET_OS_IPHONE
+
+#include <dispatch/dispatch.h>
 #include <notify.h>
 
+#define OSThermalAlert      "com.apple.system.thermalalert"
+#define OSThermalDecision   "com.apple.system.thermaldecision"
 #define OSThermalStatusName "com.apple.system.thermalstatus"
 
-const char * const kOSThermalNotificationName = OSThermalStatusName; 
+const char * const kOSThermalNotificationAlert    = OSThermalAlert;
+const char * const kOSThermalNotificationDecision = OSThermalDecision;
+const char * const kOSThermalNotificationName     = OSThermalStatusName;
 
 static const char * const kOSThermalMitigationNames[kOSThermalMitigationCount] = {
        OSThermalStatusName,
@@ -39,18 +51,20 @@ static const char * const kOSThermalMitigationNames[kOSThermalMitigationCount] =
        "com.apple.system.thermalmitigation.25percentbacklight",
        "com.apple.system.thermalmitigation.disablemapshalo",
        "com.apple.system.thermalmitigation.appterminate",
-       "com.apple.system.thermalmitigation.devicerestart"
+       "com.apple.system.thermalmitigation.devicerestart",
+       "com.apple.system.thermalmitigation.thermaltableready"
 };
 
 static int tokens[kOSThermalMitigationCount];
 static dispatch_once_t predicates[kOSThermalMitigationCount];
+static bool thermalLevelsReady = false;
 
 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_register_check(kOSThermalMitigationNames[behavior], &tokens[behavior]);
                });
                (void)notify_get_state(tokens[behavior], &val);
        }
@@ -62,14 +76,49 @@ 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_register_check(kOSThermalMitigationNames[behavior], &tokens[behavior]);
                });
                (void)notify_set_state(tokens[behavior], val);
+
+               // Note:  
+               // - We are ready when we program in the appterminate value. 
+               // - Assumes that user programs kOSThermalMitigationNone level less than 
+               //   kOSThermalMitigationAppTerminate & kOSThermalMitigationDeviceRestart  
+               if (behavior == kOSThermalMitigationAppTerminate) {
+                       dispatch_once(&predicates[kOSThermalMitigationThermalTableReady], ^{
+                               (void)notify_register_check(kOSThermalMitigationNames[kOSThermalMitigationThermalTableReady], &tokens[kOSThermalMitigationThermalTableReady]);
+                       });
+                       (void)notify_set_state(tokens[kOSThermalMitigationThermalTableReady], kOSThermalMitigationCount);
+               }
        }
 }
 
 
 OSThermalNotificationLevel OSThermalNotificationCurrentLevel(void)
 {
-       return _OSThermalNotificationLevelForBehavior(kOSThermalMitigationNone);
+       if (thermalLevelsReady) {
+               return _OSThermalNotificationLevelForBehavior(kOSThermalMitigationNone);
+       }
+
+       uint64_t tableReady = 0;
+
+       dispatch_once(&predicates[kOSThermalMitigationThermalTableReady], ^{
+               (void)notify_register_check(kOSThermalMitigationNames[kOSThermalMitigationThermalTableReady], &tokens[kOSThermalMitigationThermalTableReady]);
+       });
+       (void)notify_get_state(tokens[kOSThermalMitigationThermalTableReady], &tableReady);
+
+       // If we are ready then optimize this so we don't call dispatch everytime.
+       if (tableReady == kOSThermalMitigationCount) {
+               thermalLevelsReady = true;
+               return _OSThermalNotificationLevelForBehavior(kOSThermalMitigationNone); 
+       }
+       else {
+               // Allow reset so we can dynamically change the table without thermal trap screen appearing. 
+               thermalLevelsReady = false;
+       }
+
+       // Not ready returns -1, which should not be equal or greater than any other thermal state. 
+       return OSThermalNotificationLevelAny;
 }
+
+#endif // TARGET_OS_IPHONE