Created a shiny new abstraction for the socket handling. Has many

[u/mdw/putty] / noise.c

diff --git a/noise.c b/noise.c
index 383e6c3..af75f9d 100644 (file)
--- a/noise.c
+++ b/noise.c
@@ -8,41 +8,13 @@
 
 #include "putty.h"
 #include "ssh.h"
-
-static char seedpath[2*MAX_PATH+10] = "\0";
+#include "storage.h"
 
 /*
- * Find the random seed file path and store it in `seedpath'.
+ * GetSystemPowerStatus function.
  */
-static void get_seedpath(void) {
-    HKEY rkey;
-    DWORD type, size;
-
-    size = sizeof(seedpath);
-
-    if (RegOpenKey(HKEY_CURRENT_USER, PUTTY_REG_POS, &rkey)==ERROR_SUCCESS) {
-       int ret = RegQueryValueEx(rkey, "RandSeedFile",
-                                 0, &type, seedpath, &size);
-       if (ret != ERROR_SUCCESS || type != REG_SZ)
-           seedpath[0] = '\0';
-       RegCloseKey(rkey);
-    } else
-       seedpath[0] = '\0';
-
-    if (!seedpath[0]) {
-       int len, ret;
-
-       len = GetEnvironmentVariable("HOMEDRIVE", seedpath, sizeof(seedpath));
-       ret = GetEnvironmentVariable("HOMEPATH", seedpath+len,
-                                     sizeof(seedpath)-len);
-       if (ret == 0) {                /* probably win95; store in \WINDOWS */
-           GetWindowsDirectory(seedpath, sizeof(seedpath));
-           len = strlen(seedpath);
-       } else
-           len += ret;
-       strcpy(seedpath+len, "\\PUTTY.RND");
-    }
-}
+typedef BOOL (WINAPI *gsps_t)(LPSYSTEM_POWER_STATUS);
+static gsps_t gsps;
 
 /*
  * This function is called once, at PuTTY startup, and will do some
@@ -52,9 +24,9 @@ static void get_seedpath(void) {
 
 void noise_get_heavy(void (*func) (void *, int)) {
     HANDLE srch;
-    HANDLE seedf;
     WIN32_FIND_DATA finddata;
     char winpath[MAX_PATH+3];
+    HMODULE mod;
 
     GetWindowsDirectory(winpath, sizeof(winpath));
     strcat(winpath, "\\*");
@@ -66,55 +38,21 @@ void noise_get_heavy(void (*func) (void *, int)) {
        FindClose(srch);
     }
 
-    if (!seedpath[0])
-       get_seedpath();
-
-    seedf = CreateFile(seedpath, GENERIC_READ,
-                      FILE_SHARE_READ | FILE_SHARE_WRITE,
-                      NULL, OPEN_EXISTING, 0, NULL);
+    read_random_seed(func);
 
-    if (seedf != INVALID_HANDLE_VALUE) {
-       while (1) {
-           char buf[1024];
-           DWORD len;
-
-           if (ReadFile(seedf, buf, sizeof(buf), &len, NULL) && len)
-               func(buf, len);
-           else
-               break;
-       }
-       CloseHandle(seedf);
+    gsps = NULL;
+    mod = GetModuleHandle("KERNEL32");
+    if (mod) {
+        gsps = (gsps_t)GetProcAddress(mod, "GetSystemPowerStatus");
     }
 }
 
 void random_save_seed(void) {
-    HANDLE seedf;
-
-    if (!seedpath[0])
-       get_seedpath();
-
-    seedf = CreateFile(seedpath, GENERIC_WRITE, 0,
-                      NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
+    int len;
+    void *data;
 
-    if (seedf != INVALID_HANDLE_VALUE) {
-       int len;
-       DWORD lenwritten;
-       void *data;
-
-       random_get_savedata(&data, &len);
-       WriteFile(seedf, data, len, &lenwritten, NULL);
-       CloseHandle(seedf);
-    }
-}
-
-/*
- * This function is called from `putty -cleanup'. It removes the
- * random seed file.
- */
-void random_destroy_seed(void) {
-    if (!seedpath[0])
-       get_seedpath();
-    remove(seedpath);
+    random_get_savedata(&data, &len);
+    write_random_seed(data, len);
 }
 
 /*
@@ -134,10 +72,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));
+    }
 }
 
 /*