OSThermalNotification.c [plain text]
#include <dispatch/dispatch.h>
#include <libkern/OSThermalNotification.h>
#include <notify.h>
#define OSThermalStatusName "com.apple.system.thermalstatus"
const char * const kOSThermalNotificationName = OSThermalStatusName;
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",
"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_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);
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)
{
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 (tableReady == kOSThermalMitigationCount) {
thermalLevelsReady = true;
return _OSThermalNotificationLevelForBehavior(kOSThermalMitigationNone);
}
else {
thermalLevelsReady = false;
}
return OSThermalNotificationLevelAny;
}