Begin destabilisation in the wake of 0.53! This checkin contains the
[u/mdw/putty] / noise.c
CommitLineData
374330e2 1/*
2 * Noise generation for PuTTY's cryptographic random number
3 * generator.
4 */
5
6#include <windows.h>
7#include <stdio.h>
8
9#include "putty.h"
10#include "ssh.h"
d5859615 11#include "storage.h"
374330e2 12
13/*
a4e14164 14 * GetSystemPowerStatus function.
15 */
32874aea 16typedef BOOL(WINAPI * gsps_t) (LPSYSTEM_POWER_STATUS);
75cab814 17static gsps_t gsps;
a4e14164 18
19/*
374330e2 20 * This function is called once, at PuTTY startup, and will do some
21 * seriously silly things like listing directories and getting disk
22 * free space and a process snapshot.
23 */
24
32874aea 25void noise_get_heavy(void (*func) (void *, int))
26{
374330e2 27 HANDLE srch;
374330e2 28 WIN32_FIND_DATA finddata;
32874aea 29 char winpath[MAX_PATH + 3];
a4e14164 30 HMODULE mod;
374330e2 31
32 GetWindowsDirectory(winpath, sizeof(winpath));
33 strcat(winpath, "\\*");
34 srch = FindFirstFile(winpath, &finddata);
35 if (srch != INVALID_HANDLE_VALUE) {
36 do {
37 func(&finddata, sizeof(finddata));
38 } while (FindNextFile(srch, &finddata));
39 FindClose(srch);
40 }
41
d5859615 42 read_random_seed(func);
a4e14164 43
44 gsps = NULL;
45 mod = GetModuleHandle("KERNEL32");
46 if (mod) {
32874aea 47 gsps = (gsps_t) GetProcAddress(mod, "GetSystemPowerStatus");
a4e14164 48 }
374330e2 49}
50
32874aea 51void random_save_seed(void)
52{
d5859615 53 int len;
54 void *data;
374330e2 55
93b581bd 56 if (random_active) {
57 random_get_savedata(&data, &len);
58 write_random_seed(data, len);
59 }
de3df031 60}
61
62/*
374330e2 63 * This function is called every time the random pool needs
64 * stirring, and will acquire the system time in all available
65 * forms and the battery status.
66 */
32874aea 67void noise_get_light(void (*func) (void *, int))
68{
374330e2 69 SYSTEMTIME systime;
70 DWORD adjust[2];
71 BOOL rubbish;
72 SYSTEM_POWER_STATUS pwrstat;
73
74 GetSystemTime(&systime);
75 func(&systime, sizeof(systime));
76
77 GetSystemTimeAdjustment(&adjust[0], &adjust[1], &rubbish);
78 func(&adjust, sizeof(adjust));
79
a4e14164 80 /*
81 * Call GetSystemPowerStatus if present.
82 */
83 if (gsps) {
32874aea 84 if (gsps(&pwrstat))
85 func(&pwrstat, sizeof(pwrstat));
a4e14164 86 }
374330e2 87}
88
89/*
7d6ee6ff 90 * This function is called on a timer, and it will monitor
91 * frequently changing quantities such as the state of physical and
92 * virtual memory, the state of the process's message queue, which
93 * window is in the foreground, which owns the clipboard, etc.
94 */
32874aea 95void noise_regular(void)
96{
7d6ee6ff 97 HWND w;
98 DWORD z;
99 POINT pt;
100 MEMORYSTATUS memstat;
101 FILETIME times[4];
102
32874aea 103 w = GetForegroundWindow();
104 random_add_noise(&w, sizeof(w));
105 w = GetCapture();
106 random_add_noise(&w, sizeof(w));
107 w = GetClipboardOwner();
108 random_add_noise(&w, sizeof(w));
109 z = GetQueueStatus(QS_ALLEVENTS);
110 random_add_noise(&z, sizeof(z));
7d6ee6ff 111
32874aea 112 GetCursorPos(&pt);
113 random_add_noise(&pt, sizeof(pt));
7d6ee6ff 114
32874aea 115 GlobalMemoryStatus(&memstat);
116 random_add_noise(&memstat, sizeof(memstat));
7d6ee6ff 117
32874aea 118 GetThreadTimes(GetCurrentThread(), times, times + 1, times + 2,
119 times + 3);
7d6ee6ff 120 random_add_noise(&times, sizeof(times));
32874aea 121 GetProcessTimes(GetCurrentProcess(), times, times + 1, times + 2,
122 times + 3);
7d6ee6ff 123 random_add_noise(&times, sizeof(times));
124}
125
126/*
374330e2 127 * This function is called on every keypress or mouse move, and
128 * will add the current Windows time and performance monitor
129 * counter to the noise pool. It gets the scan code or mouse
130 * position passed in.
131 */
f7f27309 132void noise_ultralight(unsigned long data)
32874aea 133{
374330e2 134 DWORD wintime;
135 LARGE_INTEGER perftime;
136
137 random_add_noise(&data, sizeof(DWORD));
138
139 wintime = GetTickCount();
140 random_add_noise(&wintime, sizeof(DWORD));
141
142 if (QueryPerformanceCounter(&perftime))
143 random_add_noise(&perftime, sizeof(perftime));
144}