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