Fix some signed/unsigned warnings.
[oweals/openssl.git] / crypto / rand / rand_win.c
1 /* crypto/rand/rand_win.c */
2 /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
3  * All rights reserved.
4  *
5  * This package is an SSL implementation written
6  * by Eric Young (eay@cryptsoft.com).
7  * The implementation was written so as to conform with Netscapes SSL.
8  * 
9  * This library is free for commercial and non-commercial use as long as
10  * the following conditions are aheared to.  The following conditions
11  * apply to all code found in this distribution, be it the RC4, RSA,
12  * lhash, DES, etc., code; not just the SSL code.  The SSL documentation
13  * included with this distribution is covered by the same copyright terms
14  * except that the holder is Tim Hudson (tjh@cryptsoft.com).
15  * 
16  * Copyright remains Eric Young's, and as such any Copyright notices in
17  * the code are not to be removed.
18  * If this package is used in a product, Eric Young should be given attribution
19  * as the author of the parts of the library used.
20  * This can be in the form of a textual message at program startup or
21  * in documentation (online or textual) provided with the package.
22  * 
23  * Redistribution and use in source and binary forms, with or without
24  * modification, are permitted provided that the following conditions
25  * are met:
26  * 1. Redistributions of source code must retain the copyright
27  *    notice, this list of conditions and the following disclaimer.
28  * 2. Redistributions in binary form must reproduce the above copyright
29  *    notice, this list of conditions and the following disclaimer in the
30  *    documentation and/or other materials provided with the distribution.
31  * 3. All advertising materials mentioning features or use of this software
32  *    must display the following acknowledgement:
33  *    "This product includes cryptographic software written by
34  *     Eric Young (eay@cryptsoft.com)"
35  *    The word 'cryptographic' can be left out if the rouines from the library
36  *    being used are not cryptographic related :-).
37  * 4. If you include any Windows specific code (or a derivative thereof) from 
38  *    the apps directory (application code) you must include an acknowledgement:
39  *    "This product includes software written by Tim Hudson (tjh@cryptsoft.com)"
40  * 
41  * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
42  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
43  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
44  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
45  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
46  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
47  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
48  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
49  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
50  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
51  * SUCH DAMAGE.
52  * 
53  * The licence and distribution terms for any publically available version or
54  * derivative of this code cannot be changed.  i.e. this code cannot simply be
55  * copied and put under another distribution licence
56  * [including the GNU Public Licence.]
57  */
58 /* ====================================================================
59  * Copyright (c) 1998-2000 The OpenSSL Project.  All rights reserved.
60  *
61  * Redistribution and use in source and binary forms, with or without
62  * modification, are permitted provided that the following conditions
63  * are met:
64  *
65  * 1. Redistributions of source code must retain the above copyright
66  *    notice, this list of conditions and the following disclaimer. 
67  *
68  * 2. Redistributions in binary form must reproduce the above copyright
69  *    notice, this list of conditions and the following disclaimer in
70  *    the documentation and/or other materials provided with the
71  *    distribution.
72  *
73  * 3. All advertising materials mentioning features or use of this
74  *    software must display the following acknowledgment:
75  *    "This product includes software developed by the OpenSSL Project
76  *    for use in the OpenSSL Toolkit. (http://www.openssl.org/)"
77  *
78  * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
79  *    endorse or promote products derived from this software without
80  *    prior written permission. For written permission, please contact
81  *    openssl-core@openssl.org.
82  *
83  * 5. Products derived from this software may not be called "OpenSSL"
84  *    nor may "OpenSSL" appear in their names without prior written
85  *    permission of the OpenSSL Project.
86  *
87  * 6. Redistributions of any form whatsoever must retain the following
88  *    acknowledgment:
89  *    "This product includes software developed by the OpenSSL Project
90  *    for use in the OpenSSL Toolkit (http://www.openssl.org/)"
91  *
92  * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
93  * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
94  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
95  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
96  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
97  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
98  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
99  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
100  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
101  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
102  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
103  * OF THE POSSIBILITY OF SUCH DAMAGE.
104  * ====================================================================
105  *
106  * This product includes cryptographic software written by Eric Young
107  * (eay@cryptsoft.com).  This product includes software written by Tim
108  * Hudson (tjh@cryptsoft.com).
109  *
110  */
111
112 #include "cryptlib.h"
113 #include <openssl/rand.h>
114 #include "rand_lcl.h"
115
116 #if defined(OPENSSL_SYS_WINDOWS) || defined(OPENSSL_SYS_WIN32)
117 #include <windows.h>
118 #ifndef _WIN32_WINNT
119 # define _WIN32_WINNT 0x0400
120 #endif
121 #include <wincrypt.h>
122 #include <tlhelp32.h>
123
124 /* Limit the time spent walking through the heap, processes, threads and modules to
125    a maximum of 1000 miliseconds each, unless CryptoGenRandom failed */
126 #define MAXDELAY 1000
127
128 /* Intel hardware RNG CSP -- available from
129  * http://developer.intel.com/design/security/rng/redist_license.htm
130  */
131 #define PROV_INTEL_SEC 22
132 #define INTEL_DEF_PROV L"Intel Hardware Cryptographic Service Provider"
133
134 static void readtimer(void);
135 static void readscreen(void);
136
137 /* It appears like CURSORINFO, PCURSORINFO and LPCURSORINFO are only defined
138    when WINVER is 0x0500 and up, which currently only happens on Win2000.
139    Unfortunately, those are typedefs, so they're a little bit difficult to
140    detect properly.  On the other hand, the macro CURSOR_SHOWING is defined
141    within the same conditional, so it can be use to detect the absence of said
142    typedefs. */
143
144 #ifndef CURSOR_SHOWING
145 /*
146  * Information about the global cursor.
147  */
148 typedef struct tagCURSORINFO
149 {
150     DWORD   cbSize;
151     DWORD   flags;
152     HCURSOR hCursor;
153     POINT   ptScreenPos;
154 } CURSORINFO, *PCURSORINFO, *LPCURSORINFO;
155
156 #define CURSOR_SHOWING     0x00000001
157 #endif /* CURSOR_SHOWING */
158
159 typedef BOOL (WINAPI *CRYPTACQUIRECONTEXTW)(HCRYPTPROV *, LPCWSTR, LPCWSTR,
160                                     DWORD, DWORD);
161 typedef BOOL (WINAPI *CRYPTGENRANDOM)(HCRYPTPROV, DWORD, BYTE *);
162 typedef BOOL (WINAPI *CRYPTRELEASECONTEXT)(HCRYPTPROV, DWORD);
163
164 typedef HWND (WINAPI *GETFOREGROUNDWINDOW)(VOID);
165 typedef BOOL (WINAPI *GETCURSORINFO)(PCURSORINFO);
166 typedef DWORD (WINAPI *GETQUEUESTATUS)(UINT);
167
168 typedef HANDLE (WINAPI *CREATETOOLHELP32SNAPSHOT)(DWORD, DWORD);
169 typedef BOOL (WINAPI *CLOSETOOLHELP32SNAPSHOT)(HANDLE);
170 typedef BOOL (WINAPI *HEAP32FIRST)(LPHEAPENTRY32, DWORD, DWORD);
171 typedef BOOL (WINAPI *HEAP32NEXT)(LPHEAPENTRY32);
172 typedef BOOL (WINAPI *HEAP32LIST)(HANDLE, LPHEAPLIST32);
173 typedef BOOL (WINAPI *PROCESS32)(HANDLE, LPPROCESSENTRY32);
174 typedef BOOL (WINAPI *THREAD32)(HANDLE, LPTHREADENTRY32);
175 typedef BOOL (WINAPI *MODULE32)(HANDLE, LPMODULEENTRY32);
176
177 #include <lmcons.h>
178 #ifndef OPENSSL_SYS_WINCE
179 #include <lmstats.h>
180 #endif
181 #if 1 /* The NET API is Unicode only.  It requires the use of the UNICODE
182        * macro.  When UNICODE is defined LPTSTR becomes LPWSTR.  LMSTR was
183        * was added to the Platform SDK to allow the NET API to be used in
184        * non-Unicode applications provided that Unicode strings were still
185        * used for input.  LMSTR is defined as LPWSTR.
186        */
187 typedef NET_API_STATUS (NET_API_FUNCTION * NETSTATGET)
188         (LPWSTR, LPWSTR, DWORD, DWORD, LPBYTE*);
189 typedef NET_API_STATUS (NET_API_FUNCTION * NETFREE)(LPBYTE);
190 #endif /* 1 */
191
192 int RAND_poll(void)
193 {
194         MEMORYSTATUS m;
195         HCRYPTPROV hProvider = 0;
196         BYTE buf[64];
197         DWORD w;
198         HWND h;
199         int good = 0;
200
201         HMODULE advapi, kernel, user, netapi;
202         CRYPTACQUIRECONTEXTW acquire = 0;
203         CRYPTGENRANDOM gen = 0;
204         CRYPTRELEASECONTEXT release = 0;
205 #if 1 /* There was previously a problem with NETSTATGET.  Currently, this
206        * section is still experimental, but if all goes well, this conditional
207        * will be removed
208        */
209         NETSTATGET netstatget = 0;
210         NETFREE netfree = 0;
211 #endif /* 1 */
212
213         /* Determine the OS version we are on so we can turn off things 
214          * that do not work properly.
215          */
216         OSVERSIONINFO osverinfo ;
217         osverinfo.dwOSVersionInfoSize = sizeof(OSVERSIONINFO) ;
218         GetVersionEx( &osverinfo ) ;
219
220 #if defined(OPENSSL_SYS_WINCE) && WCEPLATFORM!=MS_HPC_PRO
221 #ifndef CryptAcquireContext
222 #define CryptAcquireContext CryptAcquireContextW
223 #endif
224         /* poll the CryptoAPI PRNG */
225         /* The CryptoAPI returns sizeof(buf) bytes of randomness */
226         if (CryptAcquireContext(&hProvider, 0, 0, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT))
227                 {
228                 if (CryptGenRandom(hProvider, sizeof(buf), buf))
229                         RAND_add(buf, sizeof(buf), sizeof(buf));
230                 CryptReleaseContext(hProvider, 0); 
231                 }
232 #endif
233
234 #ifndef OPENSSL_SYS_WINCE
235         /*
236          * None of below libraries are present on Windows CE, which is
237          * why we #ifndef the whole section. This also excuses us from
238          * handling the GetProcAddress issue. The trouble is that in
239          * real Win32 API GetProcAddress is available in ANSI flavor
240          * only. In WinCE on the other hand GetProcAddress is a macro
241          * most commonly defined as GetProcAddressW, which accepts
242          * Unicode argument. If we were to call GetProcAddress under
243          * WinCE, I'd recommend to either redefine GetProcAddress as
244          * GetProcAddressA (there seem to be one in common CE spec) or
245          * implement own shim routine, which would accept ANSI argument
246          * and expand it to Unicode.
247          */
248
249         /* load functions dynamically - not available on all systems */
250         advapi = LoadLibrary(TEXT("ADVAPI32.DLL"));
251         kernel = LoadLibrary(TEXT("KERNEL32.DLL"));
252         user = LoadLibrary(TEXT("USER32.DLL"));
253         netapi = LoadLibrary(TEXT("NETAPI32.DLL"));
254
255 #if 1 /* There was previously a problem with NETSTATGET.  Currently, this
256        * section is still experimental, but if all goes well, this conditional
257        * will be removed
258        */
259         if (netapi)
260                 {
261                 netstatget = (NETSTATGET) GetProcAddress(netapi,"NetStatisticsGet");
262                 netfree = (NETFREE) GetProcAddress(netapi,"NetApiBufferFree");
263                 }
264
265         if (netstatget && netfree)
266                 {
267                 LPBYTE outbuf;
268                 /* NetStatisticsGet() is a Unicode only function
269                  * STAT_WORKSTATION_0 contains 45 fields and STAT_SERVER_0
270                  * contains 17 fields.  We treat each field as a source of
271                  * one byte of entropy.
272                  */
273
274                 if (netstatget(NULL, L"LanmanWorkstation", 0, 0, &outbuf) == 0)
275                         {
276                         RAND_add(outbuf, sizeof(STAT_WORKSTATION_0), 45);
277                         netfree(outbuf);
278                         }
279                 if (netstatget(NULL, L"LanmanServer", 0, 0, &outbuf) == 0)
280                         {
281                         RAND_add(outbuf, sizeof(STAT_SERVER_0), 17);
282                         netfree(outbuf);
283                         }
284                 }
285
286         if (netapi)
287                 FreeLibrary(netapi);
288 #endif /* 1 */
289
290         /* It appears like this can cause an exception deep within ADVAPI32.DLL
291          * at random times on Windows 2000.  Reported by Jeffrey Altman.  
292          * Only use it on NT.
293          */
294         /* Wolfgang Marczy <WMarczy@topcall.co.at> reports that
295          * the RegQueryValueEx call below can hang on NT4.0 (SP6).
296          * So we don't use this at all for now. */
297 #if 0
298         if ( osverinfo.dwPlatformId == VER_PLATFORM_WIN32_NT &&
299                 osverinfo.dwMajorVersion < 5)
300                 {
301                 /* Read Performance Statistics from NT/2000 registry
302                  * The size of the performance data can vary from call
303                  * to call so we must guess the size of the buffer to use
304                  * and increase its size if we get an ERROR_MORE_DATA
305                  * return instead of ERROR_SUCCESS.
306                  */
307                 LONG   rc=ERROR_MORE_DATA;
308                 char * buf=NULL;
309                 DWORD bufsz=0;
310                 DWORD length;
311
312                 while (rc == ERROR_MORE_DATA)
313                         {
314                         buf = realloc(buf,bufsz+8192);
315                         if (!buf)
316                                 break;
317                         bufsz += 8192;
318
319                         length = bufsz;
320                         rc = RegQueryValueEx(HKEY_PERFORMANCE_DATA, TEXT("Global"),
321                                 NULL, NULL, buf, &length);
322                         }
323                 if (rc == ERROR_SUCCESS)
324                         {
325                         /* For entropy count assume only least significant
326                          * byte of each DWORD is random.
327                          */
328                         RAND_add(&length, sizeof(length), 0);
329                         RAND_add(buf, length, length / 4.0);
330
331                         /* Close the Registry Key to allow Windows to cleanup/close
332                          * the open handle
333                          * Note: The 'HKEY_PERFORMANCE_DATA' key is implicitly opened
334                          *       when the RegQueryValueEx above is done.  However, if
335                          *       it is not explicitly closed, it can cause disk
336                          *       partition manipulation problems.
337                          */
338                         RegCloseKey(HKEY_PERFORMANCE_DATA);
339                         }
340                 if (buf)
341                         free(buf);
342                 }
343 #endif
344
345         if (advapi)
346                 {
347                 /*
348                  * If it's available, then it's available in both ANSI
349                  * and UNICODE flavors even in Win9x, documentation says.
350                  * We favor Unicode...
351                  */
352                 acquire = (CRYPTACQUIRECONTEXTW) GetProcAddress(advapi,
353                         "CryptAcquireContextW");
354                 gen = (CRYPTGENRANDOM) GetProcAddress(advapi,
355                         "CryptGenRandom");
356                 release = (CRYPTRELEASECONTEXT) GetProcAddress(advapi,
357                         "CryptReleaseContext");
358                 }
359
360         if (acquire && gen && release)
361                 {
362                 /* poll the CryptoAPI PRNG */
363                 /* The CryptoAPI returns sizeof(buf) bytes of randomness */
364                 if (acquire(&hProvider, 0, 0, PROV_RSA_FULL,
365                         CRYPT_VERIFYCONTEXT))
366                         {
367                         if (gen(hProvider, sizeof(buf), buf) != 0)
368                                 {
369                                 RAND_add(buf, sizeof(buf), 0);
370                                 good = 1;
371 #if 0
372                                 printf("randomness from PROV_RSA_FULL\n");
373 #endif
374                                 }
375                         release(hProvider, 0); 
376                         }
377                 
378                 /* poll the Pentium PRG with CryptoAPI */
379                 if (acquire(&hProvider, 0, INTEL_DEF_PROV, PROV_INTEL_SEC, 0))
380                         {
381                         if (gen(hProvider, sizeof(buf), buf) != 0)
382                                 {
383                                 RAND_add(buf, sizeof(buf), sizeof(buf));
384                                 good = 1;
385 #if 0
386                                 printf("randomness from PROV_INTEL_SEC\n");
387 #endif
388                                 }
389                         release(hProvider, 0);
390                         }
391                 }
392
393         if (advapi)
394                 FreeLibrary(advapi);
395
396         if (user)
397                 {
398                 GETCURSORINFO cursor;
399                 GETFOREGROUNDWINDOW win;
400                 GETQUEUESTATUS queue;
401
402                 win = (GETFOREGROUNDWINDOW) GetProcAddress(user, "GetForegroundWindow");
403                 cursor = (GETCURSORINFO) GetProcAddress(user, "GetCursorInfo");
404                 queue = (GETQUEUESTATUS) GetProcAddress(user, "GetQueueStatus");
405
406                 if (win)
407                         {
408                         /* window handle */
409                         h = win();
410                         RAND_add(&h, sizeof(h), 0);
411                         }
412                 if (cursor)
413                         {
414                         /* unfortunately, its not safe to call GetCursorInfo()
415                          * on NT4 even though it exists in SP3 (or SP6) and
416                          * higher.
417                          */
418                         if ( osverinfo.dwPlatformId == VER_PLATFORM_WIN32_NT &&
419                                 osverinfo.dwMajorVersion < 5)
420                                 cursor = 0;
421                         }
422                 if (cursor)
423                         {
424                         /* cursor position */
425                         /* assume 2 bytes of entropy */
426                         CURSORINFO ci;
427                         ci.cbSize = sizeof(CURSORINFO);
428                         if (cursor(&ci))
429                                 RAND_add(&ci, ci.cbSize, 2);
430                         }
431
432                 if (queue)
433                         {
434                         /* message queue status */
435                         /* assume 1 byte of entropy */
436                         w = queue(QS_ALLEVENTS);
437                         RAND_add(&w, sizeof(w), 1);
438                         }
439
440                 FreeLibrary(user);
441                 }
442
443         /* Toolhelp32 snapshot: enumerate processes, threads, modules and heap
444          * http://msdn.microsoft.com/library/psdk/winbase/toolhelp_5pfd.htm
445          * (Win 9x and 2000 only, not available on NT)
446          *
447          * This seeding method was proposed in Peter Gutmann, Software
448          * Generation of Practically Strong Random Numbers,
449          * http://www.usenix.org/publications/library/proceedings/sec98/gutmann.html
450          * revised version at http://www.cryptoengines.com/~peter/06_random.pdf
451          * (The assignment of entropy estimates below is arbitrary, but based
452          * on Peter's analysis the full poll appears to be safe. Additional
453          * interactive seeding is encouraged.)
454          */
455
456         if (kernel)
457                 {
458                 CREATETOOLHELP32SNAPSHOT snap;
459                 CLOSETOOLHELP32SNAPSHOT close_snap;
460                 HANDLE handle;
461
462                 HEAP32FIRST heap_first;
463                 HEAP32NEXT heap_next;
464                 HEAP32LIST heaplist_first, heaplist_next;
465                 PROCESS32 process_first, process_next;
466                 THREAD32 thread_first, thread_next;
467                 MODULE32 module_first, module_next;
468
469                 HEAPLIST32 hlist;
470                 HEAPENTRY32 hentry;
471                 PROCESSENTRY32 p;
472                 THREADENTRY32 t;
473                 MODULEENTRY32 m;
474                 DWORD stoptime = 0;
475
476                 snap = (CREATETOOLHELP32SNAPSHOT)
477                         GetProcAddress(kernel, "CreateToolhelp32Snapshot");
478                 close_snap = (CLOSETOOLHELP32SNAPSHOT)
479                         GetProcAddress(kernel, "CloseToolhelp32Snapshot");
480                 heap_first = (HEAP32FIRST) GetProcAddress(kernel, "Heap32First");
481                 heap_next = (HEAP32NEXT) GetProcAddress(kernel, "Heap32Next");
482                 heaplist_first = (HEAP32LIST) GetProcAddress(kernel, "Heap32ListFirst");
483                 heaplist_next = (HEAP32LIST) GetProcAddress(kernel, "Heap32ListNext");
484                 process_first = (PROCESS32) GetProcAddress(kernel, "Process32First");
485                 process_next = (PROCESS32) GetProcAddress(kernel, "Process32Next");
486                 thread_first = (THREAD32) GetProcAddress(kernel, "Thread32First");
487                 thread_next = (THREAD32) GetProcAddress(kernel, "Thread32Next");
488                 module_first = (MODULE32) GetProcAddress(kernel, "Module32First");
489                 module_next = (MODULE32) GetProcAddress(kernel, "Module32Next");
490
491                 if (snap && heap_first && heap_next && heaplist_first &&
492                         heaplist_next && process_first && process_next &&
493                         thread_first && thread_next && module_first &&
494                         module_next && (handle = snap(TH32CS_SNAPALL,0))
495                         != INVALID_HANDLE_VALUE)
496                         {
497                         /* heap list and heap walking */
498                         /* HEAPLIST32 contains 3 fields that will change with
499                          * each entry.  Consider each field a source of 1 byte
500                          * of entropy.
501                          * HEAPENTRY32 contains 5 fields that will change with 
502                          * each entry.  Consider each field a source of 1 byte
503                          * of entropy.
504                          */
505                         hlist.dwSize = sizeof(HEAPLIST32);              
506                         if (good) stoptime = GetTickCount() + MAXDELAY;
507                         if (heaplist_first(handle, &hlist))
508                                 do
509                                         {
510                                         RAND_add(&hlist, hlist.dwSize, 3);
511                                         hentry.dwSize = sizeof(HEAPENTRY32);
512                                         if (heap_first(&hentry,
513                                                 hlist.th32ProcessID,
514                                                 hlist.th32HeapID))
515                                                 {
516                                                 int entrycnt = 80;
517                                                 do
518                                                         RAND_add(&hentry,
519                                                                 hentry.dwSize, 5);
520                                                 while (heap_next(&hentry)
521                                                         && --entrycnt > 0);
522                                                 }
523                                         } while (heaplist_next(handle,
524                                                 &hlist) && GetTickCount() < stoptime);
525
526                         /* process walking */
527                         /* PROCESSENTRY32 contains 9 fields that will change
528                          * with each entry.  Consider each field a source of
529                          * 1 byte of entropy.
530                          */
531                         p.dwSize = sizeof(PROCESSENTRY32);
532                 
533                         if (good) stoptime = GetTickCount() + MAXDELAY;
534                         if (process_first(handle, &p))
535                                 do
536                                         RAND_add(&p, p.dwSize, 9);
537                                 while (process_next(handle, &p) && GetTickCount() < stoptime);
538
539                         /* thread walking */
540                         /* THREADENTRY32 contains 6 fields that will change
541                          * with each entry.  Consider each field a source of
542                          * 1 byte of entropy.
543                          */
544                         t.dwSize = sizeof(THREADENTRY32);
545                         if (good) stoptime = GetTickCount() + MAXDELAY;
546                         if (thread_first(handle, &t))
547                                 do
548                                         RAND_add(&t, t.dwSize, 6);
549                                 while (thread_next(handle, &t) && GetTickCount() < stoptime);
550
551                         /* module walking */
552                         /* MODULEENTRY32 contains 9 fields that will change
553                          * with each entry.  Consider each field a source of
554                          * 1 byte of entropy.
555                          */
556                         m.dwSize = sizeof(MODULEENTRY32);
557                         if (good) stoptime = GetTickCount() + MAXDELAY;
558                         if (module_first(handle, &m))
559                                 do
560                                         RAND_add(&m, m.dwSize, 9);
561                                 while (module_next(handle, &m)
562                                                 && (GetTickCount() < stoptime));
563                         if (close_snap)
564                                 close_snap(handle);
565                         else
566                                 CloseHandle(handle);
567
568                         }
569
570                 FreeLibrary(kernel);
571                 }
572 #endif /* !OPENSSL_SYS_WINCE */
573
574         /* timer data */
575         readtimer();
576         
577         /* memory usage statistics */
578         GlobalMemoryStatus(&m);
579         RAND_add(&m, sizeof(m), 1);
580
581         /* process ID */
582         w = GetCurrentProcessId();
583         RAND_add(&w, sizeof(w), 1);
584
585 #if 0
586         printf("Exiting RAND_poll\n");
587 #endif
588
589         return(1);
590 }
591
592 int RAND_event(UINT iMsg, WPARAM wParam, LPARAM lParam)
593         {
594         double add_entropy=0;
595
596         switch (iMsg)
597                 {
598         case WM_KEYDOWN:
599                         {
600                         static WPARAM key;
601                         if (key != wParam)
602                                 add_entropy = 0.05;
603                         key = wParam;
604                         }
605                         break;
606         case WM_MOUSEMOVE:
607                         {
608                         static int lastx,lasty,lastdx,lastdy;
609                         int x,y,dx,dy;
610
611                         x=LOWORD(lParam);
612                         y=HIWORD(lParam);
613                         dx=lastx-x;
614                         dy=lasty-y;
615                         if (dx != 0 && dy != 0 && dx-lastdx != 0 && dy-lastdy != 0)
616                                 add_entropy=.2;
617                         lastx=x, lasty=y;
618                         lastdx=dx, lastdy=dy;
619                         }
620                 break;
621                 }
622
623         readtimer();
624         RAND_add(&iMsg, sizeof(iMsg), add_entropy);
625         RAND_add(&wParam, sizeof(wParam), 0);
626         RAND_add(&lParam, sizeof(lParam), 0);
627  
628         return (RAND_status());
629         }
630
631
632 void RAND_screen(void) /* function available for backward compatibility */
633 {
634         RAND_poll();
635         if (GetVersion() >= 0x80000000 || !OPENSSL_isservice())
636                 readscreen();
637 }
638
639
640 /* feed timing information to the PRNG */
641 static void readtimer(void)
642 {
643         DWORD w;
644         LARGE_INTEGER l;
645         static int have_perfc = 1;
646 #if defined(_MSC_VER) && defined(_M_X86)
647         static int have_tsc = 1;
648         DWORD cyclecount;
649
650         if (have_tsc) {
651           __try {
652             __asm {
653               _emit 0x0f
654               _emit 0x31
655               mov cyclecount, eax
656               }
657             RAND_add(&cyclecount, sizeof(cyclecount), 1);
658           } __except(EXCEPTION_EXECUTE_HANDLER) {
659             have_tsc = 0;
660           }
661         }
662 #else
663 # define have_tsc 0
664 #endif
665
666         if (have_perfc) {
667           if (QueryPerformanceCounter(&l) == 0)
668             have_perfc = 0;
669           else
670             RAND_add(&l, sizeof(l), 0);
671         }
672
673         if (!have_tsc && !have_perfc) {
674           w = GetTickCount();
675           RAND_add(&w, sizeof(w), 0);
676         }
677 }
678
679 /* feed screen contents to PRNG */
680 /*****************************************************************************
681  *
682  * Created 960901 by Gertjan van Oosten, gertjan@West.NL, West Consulting B.V.
683  *
684  * Code adapted from
685  * <URL:http://support.microsoft.com/default.aspx?scid=kb;[LN];97193>;
686  * the original copyright message is:
687  *
688  *   (C) Copyright Microsoft Corp. 1993.  All rights reserved.
689  *
690  *   You have a royalty-free right to use, modify, reproduce and
691  *   distribute the Sample Files (and/or any modified version) in
692  *   any way you find useful, provided that you agree that
693  *   Microsoft has no warranty obligations or liability for any
694  *   Sample Application Files which are modified.
695  */
696
697 static void readscreen(void)
698 {
699 #if !defined(OPENSSL_SYS_WINCE) && !defined(OPENSSL_SYS_WIN32_CYGWIN)
700   HDC           hScrDC;         /* screen DC */
701   HDC           hMemDC;         /* memory DC */
702   HBITMAP       hBitmap;        /* handle for our bitmap */
703   HBITMAP       hOldBitmap;     /* handle for previous bitmap */
704   BITMAP        bm;             /* bitmap properties */
705   unsigned int  size;           /* size of bitmap */
706   char          *bmbits;        /* contents of bitmap */
707   int           w;              /* screen width */
708   int           h;              /* screen height */
709   int           y;              /* y-coordinate of screen lines to grab */
710   int           n = 16;         /* number of screen lines to grab at a time */
711
712   /* Create a screen DC and a memory DC compatible to screen DC */
713   hScrDC = CreateDC(TEXT("DISPLAY"), NULL, NULL, NULL);
714   hMemDC = CreateCompatibleDC(hScrDC);
715
716   /* Get screen resolution */
717   w = GetDeviceCaps(hScrDC, HORZRES);
718   h = GetDeviceCaps(hScrDC, VERTRES);
719
720   /* Create a bitmap compatible with the screen DC */
721   hBitmap = CreateCompatibleBitmap(hScrDC, w, n);
722
723   /* Select new bitmap into memory DC */
724   hOldBitmap = SelectObject(hMemDC, hBitmap);
725
726   /* Get bitmap properties */
727   GetObject(hBitmap, sizeof(BITMAP), (LPSTR)&bm);
728   size = (unsigned int)bm.bmWidthBytes * bm.bmHeight * bm.bmPlanes;
729
730   bmbits = OPENSSL_malloc(size);
731   if (bmbits) {
732     /* Now go through the whole screen, repeatedly grabbing n lines */
733     for (y = 0; y < h-n; y += n)
734         {
735         unsigned char md[MD_DIGEST_LENGTH];
736
737         /* Bitblt screen DC to memory DC */
738         BitBlt(hMemDC, 0, 0, w, n, hScrDC, 0, y, SRCCOPY);
739
740         /* Copy bitmap bits from memory DC to bmbits */
741         GetBitmapBits(hBitmap, size, bmbits);
742
743         /* Get the hash of the bitmap */
744         MD(bmbits,size,md);
745
746         /* Seed the random generator with the hash value */
747         RAND_add(md, MD_DIGEST_LENGTH, 0);
748         }
749
750     OPENSSL_free(bmbits);
751   }
752
753   /* Select old bitmap back into memory DC */
754   hBitmap = SelectObject(hMemDC, hOldBitmap);
755
756   /* Clean up */
757   DeleteObject(hBitmap);
758   DeleteDC(hMemDC);
759   DeleteDC(hScrDC);
760 #endif /* !OPENSSL_SYS_WINCE */
761 }
762
763 #endif