X-Git-Url: https://asedeno.scripts.mit.edu/gitweb/?a=blobdiff_plain;f=noise.c;fp=noise.c;h=3be3c0634b471fed2734fbb08c5bf2335eeb9b16;hb=76746a7d61e8208b91cd3b54dcbd1228cbeff807;hp=88b13a97b0b708b7ae13a2a7e45b14cf63dc68f2;hpb=8805d0b50e12bad71bce45344a076cb0c5a7785c;p=PuTTY.git diff --git a/noise.c b/noise.c index 88b13a97..3be3c063 100644 --- a/noise.c +++ b/noise.c @@ -10,6 +10,12 @@ #include "ssh.h" #include "storage.h" +/* + * GetSystemPowerStatus function. + */ +typedef BOOL (WINAPI *gsps_t)(LPSYSTEM_POWER_STATUS); +gsps_t gsps; + /* * This function is called once, at PuTTY startup, and will do some * seriously silly things like listing directories and getting disk @@ -20,6 +26,7 @@ void noise_get_heavy(void (*func) (void *, int)) { HANDLE srch; WIN32_FIND_DATA finddata; char winpath[MAX_PATH+3]; + HMODULE mod; GetWindowsDirectory(winpath, sizeof(winpath)); strcat(winpath, "\\*"); @@ -32,6 +39,13 @@ void noise_get_heavy(void (*func) (void *, int)) { } read_random_seed(func); + + gsps = NULL; + mod = GetModuleHandle("KERNEL32"); + if (mod) { + gsps = (gsps_t)GetProcAddress(mod, "GetSystemPowerStatus"); + debug(("got gsps=%p\n", gsps)); + } } void random_save_seed(void) { @@ -59,10 +73,13 @@ void noise_get_light(void (*func) (void *, int)) { GetSystemTimeAdjustment(&adjust[0], &adjust[1], &rubbish); func(&adjust, sizeof(adjust)); -#ifndef WIN32S_COMPAT - if (GetSystemPowerStatus(&pwrstat)) - func(&pwrstat, sizeof(pwrstat)); -#endif + /* + * Call GetSystemPowerStatus if present. + */ + if (gsps) { + if (gsps(&pwrstat)) + func(&pwrstat, sizeof(pwrstat)); + } } /*