Browse Source

Import RegisterPowerSettingNotification().

Iain Patterson 7 years ago
parent
commit
d340e13eff
2 changed files with 12 additions and 0 deletions
  1. 9 0
      imports.cpp
  2. 3 0
      imports.h

+ 9 - 0
imports.cpp

@@ -82,6 +82,15 @@ int get_imports() {
   }
   else if (error != ERROR_MOD_NOT_FOUND) return 5;
 
+  imports.user32 = get_dll(_T("user32.dll"), &error);
+  if (imports.user32) {
+    imports.RegisterPowerSettingNotification = (RegisterPowerSettingNotification_ptr) get_import(imports.user32, "RegisterPowerSettingNotification", &error);
+    if (! imports.RegisterPowerSettingNotification) {
+      if (error != ERROR_PROC_NOT_FOUND) return 9;
+    }
+  }
+  else if (error != ERROR_MOD_NOT_FOUND) return 8;
+
   return 0;
 }
 

+ 3 - 0
imports.h

@@ -6,15 +6,18 @@ typedef BOOL (WINAPI *SleepConditionVariableCS_ptr)(PCONDITION_VARIABLE, PCRITIC
 typedef void (WINAPI *WakeConditionVariable_ptr)(PCONDITION_VARIABLE);
 typedef BOOL (WINAPI *CreateWellKnownSid_ptr)(WELL_KNOWN_SID_TYPE, SID *, SID *, unsigned long *);
 typedef BOOL (WINAPI *IsWellKnownSid_ptr)(SID *, WELL_KNOWN_SID_TYPE);
+typedef HANDLE (WINAPI *RegisterPowerSettingNotification_ptr)(HANDLE, LPCGUID, DWORD);
 
 typedef struct {
   HMODULE kernel32;
   HMODULE advapi32;
+  HMODULE user32;
   AttachConsole_ptr AttachConsole;
   SleepConditionVariableCS_ptr SleepConditionVariableCS;
   WakeConditionVariable_ptr WakeConditionVariable;
   CreateWellKnownSid_ptr CreateWellKnownSid;
   IsWellKnownSid_ptr IsWellKnownSid;
+  RegisterPowerSettingNotification_ptr RegisterPowerSettingNotification;
 } imports_t;
 
 HMODULE get_dll(const TCHAR *, unsigned long *);