d3fe50d341368e81a0c27efe1475d67e32a27532
[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(WINDOWS) || defined(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 /* Intel hardware RNG CSP -- available from
125  * http://developer.intel.com/design/security/rng/redist_license.htm
126  */
127 #define PROV_INTEL_SEC 22
128 #define INTEL_DEF_PROV "Intel Hardware Cryptographic Service Provider"
129
130 static void readtimer(void);
131 static void readscreen(void);
132
133 /* It appears like CURSORINFO, PCURSORINFO and LPCURSORINFO are only defined
134    when WINVER is 0x0500 and up, which currently only happens on Win2000.
135    Unfortunately, those are typedefs, so they're a little bit difficult to
136    detect properly.  On the other hand, the macro CURSOR_SHOWING is defined
137    within the same conditional, so it can be use to detect the absence of said
138    typedefs. */
139
140 #ifndef CURSOR_SHOWING
141 /*
142  * Information about the global cursor.
143  */
144 typedef struct tagCURSORINFO
145 {
146     DWORD   cbSize;
147     DWORD   flags;
148     HCURSOR hCursor;
149     POINT   ptScreenPos;
150 } CURSORINFO, *PCURSORINFO, *LPCURSORINFO;
151
152 #define CURSOR_SHOWING     0x00000001
153 #endif /* CURSOR_SHOWING */
154
155 typedef BOOL (WINAPI *CRYPTACQUIRECONTEXT)(HCRYPTPROV *, LPCTSTR, LPCTSTR,
156                                     DWORD, DWORD);
157 typedef BOOL (WINAPI *CRYPTGENRANDOM)(HCRYPTPROV, DWORD, BYTE *);
158 typedef BOOL (WINAPI *CRYPTRELEASECONTEXT)(HCRYPTPROV, DWORD);
159
160 typedef HWND (WINAPI *GETFOREGROUNDWINDOW)(VOID);
161 typedef BOOL (WINAPI *GETCURSORINFO)(PCURSORINFO);
162 typedef DWORD (WINAPI *GETQUEUESTATUS)(UINT);
163
164 typedef HANDLE (WINAPI *CREATETOOLHELP32SNAPSHOT)(DWORD, DWORD);
165 typedef BOOL (WINAPI *HEAP32FIRST)(LPHEAPENTRY32, DWORD, DWORD);
166 typedef BOOL (WINAPI *HEAP32NEXT)(LPHEAPENTRY32);
167 typedef BOOL (WINAPI *HEAP32LIST)(HANDLE, LPHEAPLIST32);
168 typedef BOOL (WINAPI *PROCESS32)(HANDLE, LPPROCESSENTRY32);
169 typedef BOOL (WINAPI *THREAD32)(HANDLE, LPTHREADENTRY32);
170 typedef BOOL (WINAPI *MODULE32)(HANDLE, LPMODULEENTRY32);
171
172 #include <lmcons.h>
173 #include <lmstats.h>
174 #if 0 /* Some compilers use LMSTR, others (VC6, for example) use LPTSTR.
175        * This part is disabled until a fix is found.
176        */
177 typedef NET_API_STATUS (NET_API_FUNCTION * NETSTATGET)
178         (LMSTR, LMSTR, DWORD, DWORD, LPBYTE*);
179 typedef NET_API_STATUS (NET_API_FUNCTION * NETFREE)(LPBYTE);
180 #endif /* 0 */
181
182 int RAND_poll(void)
183 {
184         MEMORYSTATUS m;
185         HCRYPTPROV hProvider = 0;
186         BYTE buf[64];
187         DWORD w;
188         HWND h;
189
190         HMODULE advapi, kernel, user, netapi;
191         CRYPTACQUIRECONTEXT acquire = 0;
192         CRYPTGENRANDOM gen = 0;
193         CRYPTRELEASECONTEXT release = 0;
194 #if 0 /* This part is disabled until a fix for the problem with the
195        * definition of NETSTATGET is found.
196        */
197         NETSTATGET netstatget = 0;
198         NETFREE netfree = 0;
199 #endif /* 0 */
200
201         /* load functions dynamically - not available on all systems */
202         advapi = LoadLibrary("ADVAPI32.DLL");
203         kernel = LoadLibrary("KERNEL32.DLL");
204         user = LoadLibrary("USER32.DLL");
205         netapi = LoadLibrary("NETAPI32.DLL");
206
207 #if 0 /* This part is disabled until a fix for the problem with the
208        * definition of NETSTATGET is found.  Also, note that VC6 doesn't
209        * understand strings starting with L".
210        */
211         if (netapi)
212                 {
213                 netstatget = (NETSTATGET) GetProcAddress(netapi,"NetStatisticsGet");
214                 netfree = (NETFREE) GetProcAddress(netapi,"NetApiBufferFree");
215                 }
216
217         if (netstatget && netfree)
218                 {
219                 LPBYTE outbuf;
220                 /* NetStatisticsGet() is a Unicode only function */
221                 if (netstatget(NULL, L"LanmanWorkstation", 0, 0, &outbuf) == 0)
222                         {
223                         RAND_add(outbuf, sizeof(STAT_WORKSTATION_0), 0);
224                         netfree(outbuf);
225                         }
226                 if (netstatget(NULL, L"LanmanServer", 0, 0, &outbuf) == 0)
227                         {
228                         RAND_add(outbuf, sizeof(STAT_SERVER_0), 0);
229                         netfree(outbuf);
230                         }
231                 }
232
233         if (netapi)
234                 FreeLibrary(netapi);
235 #endif /* 0 */
236  
237 #if 0 /* It appears like this can cause an exception deep within ADVAPI32.DLL
238        * at random times.  Reported by Jeffrey Altman.
239        */
240         /* Read Performance Statistics from NT/2000 registry */
241         /* The size of the performance data can vary from call to call */
242         /* so we must guess the size of the buffer to use and increase */
243         /* its size if we get an ERROR_MORE_DATA return instead of     */
244         /* ERROR_SUCCESS.                                              */
245         {
246         LONG   rc=ERROR_MORE_DATA;
247         char * buf=NULL;
248         DWORD bufsz=0;
249         DWORD length;
250
251         while (rc == ERROR_MORE_DATA)
252                 {
253                 buf = realloc(buf,bufsz+8192);
254                 if (!buf)
255                         break;
256                 bufsz += 8192;
257
258                 length = bufsz;
259                 rc = RegQueryValueEx(HKEY_PERFORMANCE_DATA, "Global",
260                         NULL, NULL, buf, &length);
261                 }
262         if (rc == ERROR_SUCCESS)
263                 {
264                 RAND_add(&length, sizeof(length), 0);
265                 RAND_add(buf, length, 0);
266                 }
267         if (buf)
268                 free(buf);
269         }
270 #endif /* 0 */
271
272         if (advapi)
273                 {
274                 acquire = (CRYPTACQUIRECONTEXT) GetProcAddress(advapi,
275                         "CryptAcquireContextA");
276                 gen = (CRYPTGENRANDOM) GetProcAddress(advapi,
277                         "CryptGenRandom");
278                 release = (CRYPTRELEASECONTEXT) GetProcAddress(advapi,
279                         "CryptReleaseContext");
280                 }
281
282         if (acquire && gen && release)
283                 {
284                 /* poll the CryptoAPI PRNG */
285                 if (acquire(&hProvider, 0, 0, PROV_RSA_FULL,
286                         CRYPT_VERIFYCONTEXT))
287                         {
288                         if (gen(hProvider, sizeof(buf), buf) != 0)
289                                 {
290                                 RAND_add(buf, sizeof(buf), 0);
291 #ifdef DEBUG
292                                 printf("randomness from PROV_RSA_FULL\n");
293 #endif
294                                 }
295                         release(hProvider, 0); 
296                         }
297                 
298                 /* poll the Pentium PRG with CryptoAPI */
299                 if (acquire(&hProvider, 0, INTEL_DEF_PROV, PROV_INTEL_SEC, 0))
300                         {
301                         if (gen(hProvider, sizeof(buf), buf) != 0)
302                                 {
303                                 RAND_add(buf, sizeof(buf), 0);
304 #ifdef DEBUG
305                                 printf("randomness from PROV_INTEL_SEC\n");
306 #endif
307                                 }
308                         release(hProvider, 0);
309                         }
310                 }
311
312         if (advapi)
313                 FreeLibrary(advapi);
314
315         /* timer data */
316         readtimer();
317         
318         /* memory usage statistics */
319         GlobalMemoryStatus(&m);
320         RAND_add(&m, sizeof(m), 1);
321
322         /* process ID */
323         w = GetCurrentProcessId();
324         RAND_add(&w, sizeof(w), 0);
325
326         if (user)
327                 {
328                 GETCURSORINFO cursor;
329                 GETFOREGROUNDWINDOW win;
330                 GETQUEUESTATUS queue;
331
332                 win = (GETFOREGROUNDWINDOW) GetProcAddress(user, "GetForegroundWindow");
333                 cursor = (GETCURSORINFO) GetProcAddress(user, "GetCursorInfo");
334                 queue = (GETQUEUESTATUS) GetProcAddress(user, "GetQueueStatus");
335
336                 if (win)
337                 {
338                         /* window handle */
339                         h = win();
340                         RAND_add(&h, sizeof(h), 0);
341                 }
342
343                 if (cursor)
344                         {
345                         /* unfortunately, its not safe to call GetCursorInfo()
346                          * on NT4 even though it exists in SP3 (or SP6) and
347                          * higher.
348                          */
349                         OSVERSIONINFO osverinfo ;
350                         osverinfo.dwOSVersionInfoSize = sizeof(OSVERSIONINFO) ;
351                         GetVersionEx( &osverinfo ) ;
352
353                         if ( osverinfo.dwPlatformId == VER_PLATFORM_WIN32_NT &&
354                                 osverinfo.dwMajorVersion < 5)
355                                 cursor = 0;
356                         }
357
358                 if (cursor)
359                         {
360                         /* cursor position */
361                         CURSORINFO ci;
362                         ci.cbSize = sizeof(CURSORINFO);
363                         if (cursor(&ci))
364                                 RAND_add(&ci, ci.cbSize, 0);
365                         }
366
367                 if (queue)
368                         {
369                         /* message queue status */
370                         w = queue(QS_ALLEVENTS);
371                         RAND_add(&w, sizeof(w), 0);
372                         }
373
374                 FreeLibrary(user);
375                 }
376
377         /* Toolhelp32 snapshot: enumerate processes, threads, modules and heap
378          * http://msdn.microsoft.com/library/psdk/winbase/toolhelp_5pfd.htm
379          * (Win 9x and 2000 only, not available on NT)
380          *
381          * This seeding method was proposed in Peter Gutmann, Software
382          * Generation of Practically Strong Random Numbers,
383          * http://www.usenix.org/publications/library/proceedings/sec98/gutmann.html
384      * revised version at http://www.cryptoengines.com/~peter/06_random.pdf
385          * (The assignment of entropy estimates below is arbitrary, but based
386          * on Peter's analysis the full poll appears to be safe. Additional
387          * interactive seeding is encouraged.)
388          */
389
390         if (kernel)
391                 {
392                 CREATETOOLHELP32SNAPSHOT snap;
393                 HANDLE handle;
394
395                 HEAP32FIRST heap_first;
396                 HEAP32NEXT heap_next;
397                 HEAP32LIST heaplist_first, heaplist_next;
398                 PROCESS32 process_first, process_next;
399                 THREAD32 thread_first, thread_next;
400                 MODULE32 module_first, module_next;
401
402                 HEAPLIST32 hlist;
403                 HEAPENTRY32 hentry;
404                 PROCESSENTRY32 p;
405                 THREADENTRY32 t;
406                 MODULEENTRY32 m;
407
408                 snap = (CREATETOOLHELP32SNAPSHOT)
409                   GetProcAddress(kernel, "CreateToolhelp32Snapshot");
410                 heap_first = (HEAP32FIRST) GetProcAddress(kernel, "Heap32First");
411                 heap_next = (HEAP32NEXT) GetProcAddress(kernel, "Heap32Next");
412                 heaplist_first = (HEAP32LIST) GetProcAddress(kernel, "Heap32ListFirst");
413                 heaplist_next = (HEAP32LIST) GetProcAddress(kernel, "Heap32ListNext");
414                 process_first = (PROCESS32) GetProcAddress(kernel, "Process32First");
415                 process_next = (PROCESS32) GetProcAddress(kernel, "Process32Next");
416                 thread_first = (THREAD32) GetProcAddress(kernel, "Thread32First");
417                 thread_next = (THREAD32) GetProcAddress(kernel, "Thread32Next");
418                 module_first = (MODULE32) GetProcAddress(kernel, "Module32First");
419                 module_next = (MODULE32) GetProcAddress(kernel, "Module32Next");
420
421                 if (snap && heap_first && heap_next && heaplist_first &&
422                         heaplist_next && process_first && process_next &&
423                         thread_first && thread_next && module_first &&
424                         module_next && (handle = snap(TH32CS_SNAPALL,0))
425                         != NULL)
426                         {
427                         /* heap list and heap walking */
428                         hlist.dwSize = sizeof(HEAPLIST32);              
429                         if (heaplist_first(handle, &hlist))
430                                 do
431                                         {
432                                         RAND_add(&hlist, hlist.dwSize, 0);
433                                         hentry.dwSize = sizeof(HEAPENTRY32);
434                                         if (heap_first(&hentry,
435                                                 hlist.th32ProcessID,
436                                                 hlist.th32HeapID))
437                                                 {
438                                                 int entrycnt = 50;
439                                                 do
440                                                         RAND_add(&hentry,
441                                                                 hentry.dwSize, 0);
442                                                 while (heap_next(&hentry)
443                                                         && --entrycnt > 0);
444                                                 }
445                                         } while (heaplist_next(handle,
446                                                 &hlist));
447
448                         /* process walking */
449                         p.dwSize = sizeof(PROCESSENTRY32);
450                         if (process_first(handle, &p))
451                                 do
452                                         RAND_add(&p, p.dwSize, 0);
453                                 while (process_next(handle, &p));
454                         
455                         /* thread walking */
456                         t.dwSize = sizeof(THREADENTRY32);
457                         if (thread_first(handle, &t))
458                                 do
459                                         RAND_add(&t, t.dwSize, 0);
460                                 while (thread_next(handle, &t));
461                         
462                         /* module walking */
463                         m.dwSize = sizeof(MODULEENTRY32);
464                         if (module_first(handle, &m))
465                                 do
466                                         RAND_add(&m, m.dwSize, 1);
467                                 while (module_next(handle, &m));
468                         
469                         CloseHandle(handle);
470                         }
471
472                 FreeLibrary(kernel);
473                 }
474
475 #ifdef DEBUG
476         printf("Exiting RAND_poll\n");
477 #endif
478
479         return(1);
480 }
481
482 int RAND_event(UINT iMsg, WPARAM wParam, LPARAM lParam)
483         {
484         double add_entropy=0;
485
486         switch (iMsg)
487                 {
488         case WM_KEYDOWN:
489                         {
490                         static WPARAM key;
491                         if (key != wParam)
492                                 add_entropy = 0.05;
493                         key = wParam;
494                         }
495                         break;
496         case WM_MOUSEMOVE:
497                         {
498                         static int lastx,lasty,lastdx,lastdy;
499                         int x,y,dx,dy;
500
501                         x=LOWORD(lParam);
502                         y=HIWORD(lParam);
503                         dx=lastx-x;
504                         dy=lasty-y;
505                         if (dx != 0 && dy != 0 && dx-lastdx != 0 && dy-lastdy != 0)
506                                 add_entropy=.2;
507                         lastx=x, lasty=y;
508                         lastdx=dx, lastdy=dy;
509                         }
510                 break;
511                 }
512
513         readtimer();
514         RAND_add(&iMsg, sizeof(iMsg), add_entropy);
515         RAND_add(&wParam, sizeof(wParam), 0);
516         RAND_add(&lParam, sizeof(lParam), 0);
517  
518         return (RAND_status());
519         }
520
521
522 void RAND_screen(void) /* function available for backward compatibility */
523 {
524         RAND_poll();
525         readscreen();
526 }
527
528
529 /* feed timing information to the PRNG */
530 static void readtimer(void)
531 {
532         DWORD w;
533         LARGE_INTEGER l;
534         static int have_perfc = 1;
535 #ifndef __GNUC__
536         static int have_tsc = 1;
537         DWORD cyclecount;
538
539         if (have_tsc) {
540           __try {
541             __asm {
542               rdtsc
543               mov cyclecount, eax
544               }
545             RAND_add(&cyclecount, sizeof(cyclecount), 1);
546           } __except(EXCEPTION_EXECUTE_HANDLER) {
547             have_tsc = 0;
548           }
549         }
550 #else
551 # define have_tsc 0
552 #endif
553
554         if (have_perfc) {
555           if (QueryPerformanceCounter(&l) == 0)
556             have_perfc = 0;
557           else
558             RAND_add(&l, sizeof(l), 0);
559         }
560
561         if (!have_tsc && !have_perfc) {
562           w = GetTickCount();
563           RAND_add(&w, sizeof(w), 0);
564         }
565 }
566
567 /* feed screen contents to PRNG */
568 /*****************************************************************************
569  *
570  * Created 960901 by Gertjan van Oosten, gertjan@West.NL, West Consulting B.V.
571  *
572  * Code adapted from
573  * <URL:http://www.microsoft.com/kb/developr/win_dk/q97193.htm>;
574  * the original copyright message is:
575  *
576  *   (C) Copyright Microsoft Corp. 1993.  All rights reserved.
577  *
578  *   You have a royalty-free right to use, modify, reproduce and
579  *   distribute the Sample Files (and/or any modified version) in
580  *   any way you find useful, provided that you agree that
581  *   Microsoft has no warranty obligations or liability for any
582  *   Sample Application Files which are modified.
583  */
584
585 static void readscreen(void)
586 {
587   HDC           hScrDC;         /* screen DC */
588   HDC           hMemDC;         /* memory DC */
589   HBITMAP       hBitmap;        /* handle for our bitmap */
590   HBITMAP       hOldBitmap;     /* handle for previous bitmap */
591   BITMAP        bm;             /* bitmap properties */
592   unsigned int  size;           /* size of bitmap */
593   char          *bmbits;        /* contents of bitmap */
594   int           w;              /* screen width */
595   int           h;              /* screen height */
596   int           y;              /* y-coordinate of screen lines to grab */
597   int           n = 16;         /* number of screen lines to grab at a time */
598
599   /* Create a screen DC and a memory DC compatible to screen DC */
600   hScrDC = CreateDC("DISPLAY", NULL, NULL, NULL);
601   hMemDC = CreateCompatibleDC(hScrDC);
602
603   /* Get screen resolution */
604   w = GetDeviceCaps(hScrDC, HORZRES);
605   h = GetDeviceCaps(hScrDC, VERTRES);
606
607   /* Create a bitmap compatible with the screen DC */
608   hBitmap = CreateCompatibleBitmap(hScrDC, w, n);
609
610   /* Select new bitmap into memory DC */
611   hOldBitmap = SelectObject(hMemDC, hBitmap);
612
613   /* Get bitmap properties */
614   GetObject(hBitmap, sizeof(BITMAP), (LPSTR)&bm);
615   size = (unsigned int)bm.bmWidthBytes * bm.bmHeight * bm.bmPlanes;
616
617   bmbits = OPENSSL_malloc(size);
618   if (bmbits) {
619     /* Now go through the whole screen, repeatedly grabbing n lines */
620     for (y = 0; y < h-n; y += n)
621         {
622         unsigned char md[MD_DIGEST_LENGTH];
623
624         /* Bitblt screen DC to memory DC */
625         BitBlt(hMemDC, 0, 0, w, n, hScrDC, 0, y, SRCCOPY);
626
627         /* Copy bitmap bits from memory DC to bmbits */
628         GetBitmapBits(hBitmap, size, bmbits);
629
630         /* Get the hash of the bitmap */
631         MD(bmbits,size,md);
632
633         /* Seed the random generator with the hash value */
634         RAND_add(md, MD_DIGEST_LENGTH, 0);
635         }
636
637     OPENSSL_free(bmbits);
638   }
639
640   /* Select old bitmap back into memory DC */
641   hBitmap = SelectObject(hMemDC, hOldBitmap);
642
643   /* Clean up */
644   DeleteObject(hBitmap);
645   DeleteDC(hMemDC);
646   DeleteDC(hScrDC);
647 }
648
649 #else /* Unix version */
650
651 #include <time.h>
652
653 int RAND_poll(void)
654 {
655         unsigned long l;
656         pid_t curr_pid = getpid();
657 #ifdef DEVRANDOM
658         FILE *fh;
659 #endif
660
661 #ifdef DEVRANDOM
662         /* Use a random entropy pool device. Linux, FreeBSD and OpenBSD
663          * have this. Use /dev/urandom if you can as /dev/random may block
664          * if it runs out of random entries.  */
665
666         if ((fh = fopen(DEVRANDOM, "r")) != NULL)
667                 {
668                 unsigned char tmpbuf[ENTROPY_NEEDED];
669                 int n;
670                 
671                 setvbuf(fh, NULL, _IONBF, 0);
672                 n=fread((unsigned char *)tmpbuf,1,ENTROPY_NEEDED,fh);
673                 fclose(fh);
674                 RAND_add(tmpbuf,sizeof tmpbuf,n);
675                 memset(tmpbuf,0,n);
676                 }
677 #endif
678
679         /* put in some default random data, we need more than just this */
680         l=curr_pid;
681         RAND_add(&l,sizeof(l),0);
682         l=getuid();
683         RAND_add(&l,sizeof(l),0);
684
685         l=time(NULL);
686         RAND_add(&l,sizeof(l),0);
687
688 #ifdef DEVRANDOM
689         return 1;
690 #endif
691         return 0;
692 }
693
694 #endif