rand_win.c 23.3 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111
/* crypto/rand/rand_win.c */
/* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
 * All rights reserved.
 *
 * This package is an SSL implementation written
 * by Eric Young (eay@cryptsoft.com).
 * The implementation was written so as to conform with Netscapes SSL.
 * 
 * This library is free for commercial and non-commercial use as long as
 * the following conditions are aheared to.  The following conditions
 * apply to all code found in this distribution, be it the RC4, RSA,
 * lhash, DES, etc., code; not just the SSL code.  The SSL documentation
 * included with this distribution is covered by the same copyright terms
 * except that the holder is Tim Hudson (tjh@cryptsoft.com).
 * 
 * Copyright remains Eric Young's, and as such any Copyright notices in
 * the code are not to be removed.
 * If this package is used in a product, Eric Young should be given attribution
 * as the author of the parts of the library used.
 * This can be in the form of a textual message at program startup or
 * in documentation (online or textual) provided with the package.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *    "This product includes cryptographic software written by
 *     Eric Young (eay@cryptsoft.com)"
 *    The word 'cryptographic' can be left out if the rouines from the library
 *    being used are not cryptographic related :-).
 * 4. If you include any Windows specific code (or a derivative thereof) from 
 *    the apps directory (application code) you must include an acknowledgement:
 *    "This product includes software written by Tim Hudson (tjh@cryptsoft.com)"
 * 
 * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 * 
 * The licence and distribution terms for any publically available version or
 * derivative of this code cannot be changed.  i.e. this code cannot simply be
 * copied and put under another distribution licence
 * [including the GNU Public Licence.]
 */
/* ====================================================================
 * Copyright (c) 1998-2000 The OpenSSL Project.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer. 
 *
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in
 *    the documentation and/or other materials provided with the
 *    distribution.
 *
 * 3. All advertising materials mentioning features or use of this
 *    software must display the following acknowledgment:
 *    "This product includes software developed by the OpenSSL Project
 *    for use in the OpenSSL Toolkit. (http://www.openssl.org/)"
 *
 * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
 *    endorse or promote products derived from this software without
 *    prior written permission. For written permission, please contact
 *    openssl-core@openssl.org.
 *
 * 5. Products derived from this software may not be called "OpenSSL"
 *    nor may "OpenSSL" appear in their names without prior written
 *    permission of the OpenSSL Project.
 *
 * 6. Redistributions of any form whatsoever must retain the following
 *    acknowledgment:
 *    "This product includes software developed by the OpenSSL Project
 *    for use in the OpenSSL Toolkit (http://www.openssl.org/)"
 *
 * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
 * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
 * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
 * OF THE POSSIBILITY OF SUCH DAMAGE.
 * ====================================================================
 *
 * This product includes cryptographic software written by Eric Young
 * (eay@cryptsoft.com).  This product includes software written by Tim
 * Hudson (tjh@cryptsoft.com).
 *
 */

112 113 114
#include "cryptlib.h"
#include <openssl/rand.h>
#include "rand_lcl.h"
115

116
#if defined(OPENSSL_SYS_WINDOWS) || defined(OPENSSL_SYS_WIN32)
117
#include <windows.h>
118 119
#ifndef _WIN32_WINNT
# define _WIN32_WINNT 0x0400
120
#endif
121 122 123 124 125 126 127
#include <wincrypt.h>
#include <tlhelp32.h>

/* Intel hardware RNG CSP -- available from
 * http://developer.intel.com/design/security/rng/redist_license.htm
 */
#define PROV_INTEL_SEC 22
R
Richard Levitte 已提交
128
#define INTEL_DEF_PROV TEXT("Intel Hardware Cryptographic Service Provider")
129 130 131 132

static void readtimer(void);
static void readscreen(void);

133 134 135 136 137 138 139
/* It appears like CURSORINFO, PCURSORINFO and LPCURSORINFO are only defined
   when WINVER is 0x0500 and up, which currently only happens on Win2000.
   Unfortunately, those are typedefs, so they're a little bit difficult to
   detect properly.  On the other hand, the macro CURSOR_SHOWING is defined
   within the same conditional, so it can be use to detect the absence of said
   typedefs. */

R
Richard Levitte 已提交
140
#ifndef CURSOR_SHOWING
141 142 143 144 145 146 147 148 149 150 151 152 153
/*
 * Information about the global cursor.
 */
typedef struct tagCURSORINFO
{
    DWORD   cbSize;
    DWORD   flags;
    HCURSOR hCursor;
    POINT   ptScreenPos;
} CURSORINFO, *PCURSORINFO, *LPCURSORINFO;

#define CURSOR_SHOWING     0x00000001
#endif /* CURSOR_SHOWING */
R
Richard Levitte 已提交
154

155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171
typedef BOOL (WINAPI *CRYPTACQUIRECONTEXT)(HCRYPTPROV *, LPCTSTR, LPCTSTR,
				    DWORD, DWORD);
typedef BOOL (WINAPI *CRYPTGENRANDOM)(HCRYPTPROV, DWORD, BYTE *);
typedef BOOL (WINAPI *CRYPTRELEASECONTEXT)(HCRYPTPROV, DWORD);

typedef HWND (WINAPI *GETFOREGROUNDWINDOW)(VOID);
typedef BOOL (WINAPI *GETCURSORINFO)(PCURSORINFO);
typedef DWORD (WINAPI *GETQUEUESTATUS)(UINT);

typedef HANDLE (WINAPI *CREATETOOLHELP32SNAPSHOT)(DWORD, DWORD);
typedef BOOL (WINAPI *HEAP32FIRST)(LPHEAPENTRY32, DWORD, DWORD);
typedef BOOL (WINAPI *HEAP32NEXT)(LPHEAPENTRY32);
typedef BOOL (WINAPI *HEAP32LIST)(HANDLE, LPHEAPLIST32);
typedef BOOL (WINAPI *PROCESS32)(HANDLE, LPPROCESSENTRY32);
typedef BOOL (WINAPI *THREAD32)(HANDLE, LPTHREADENTRY32);
typedef BOOL (WINAPI *MODULE32)(HANDLE, LPMODULEENTRY32);

172
#include <lmcons.h>
R
Richard Levitte 已提交
173
#ifndef OPENSSL_SYS_WINCE
174
#include <lmstats.h>
R
Richard Levitte 已提交
175
#endif
176 177 178 179 180
#if 1 /* The NET API is Unicode only.  It requires the use of the UNICODE
       * macro.  When UNICODE is defined LPTSTR becomes LPWSTR.  LMSTR was
       * was added to the Platform SDK to allow the NET API to be used in
       * non-Unicode applications provided that Unicode strings were still
       * used for input.  LMSTR is defined as LPWSTR.
181
       */
182
typedef NET_API_STATUS (NET_API_FUNCTION * NETSTATGET)
183
        (LPWSTR, LPWSTR, DWORD, DWORD, LPBYTE*);
184
typedef NET_API_STATUS (NET_API_FUNCTION * NETFREE)(LPBYTE);
185
#endif /* 1 */
186

187 188 189 190 191 192 193 194
int RAND_poll(void)
{
	MEMORYSTATUS m;
	HCRYPTPROV hProvider = 0;
	BYTE buf[64];
	DWORD w;
	HWND h;

195
	HMODULE advapi, kernel, user, netapi;
196 197 198
	CRYPTACQUIRECONTEXT acquire = 0;
	CRYPTGENRANDOM gen = 0;
	CRYPTRELEASECONTEXT release = 0;
199 200 201
#if 1 /* There was previously a problem with NETSTATGET.  Currently, this
       * section is still experimental, but if all goes well, this conditional
       * will be removed
202
       */
203 204
	NETSTATGET netstatget = 0;
	NETFREE netfree = 0;
205 206 207 208 209 210 211 212
#endif /* 1 */

	/* Determine the OS version we are on so we can turn off things 
	 * that do not work properly.
	 */
        OSVERSIONINFO osverinfo ;
        osverinfo.dwOSVersionInfoSize = sizeof(OSVERSIONINFO) ;
        GetVersionEx( &osverinfo ) ;
213

214
#if defined(OPENSSL_SYS_WINCE) && WCEPLATFORM!=MS_HPC_PRO
R
Richard Levitte 已提交
215 216 217 218 219 220 221 222 223 224
	/* poll the CryptoAPI PRNG */
	/* The CryptoAPI returns sizeof(buf) bytes of randomness */
	if (CryptAcquireContext(&hProvider, 0, 0, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT))
		{
		if (CryptGenRandom(hProvider, sizeof(buf), buf))
			RAND_add(buf, sizeof(buf), sizeof(buf));
		CryptReleaseContext(hProvider, 0); 
		}
#endif

225
	/* load functions dynamically - not available on all systems */
R
Richard Levitte 已提交
226 227 228 229
	advapi = LoadLibrary(TEXT("ADVAPI32.DLL"));
	kernel = LoadLibrary(TEXT("KERNEL32.DLL"));
	user = LoadLibrary(TEXT("USER32.DLL"));
	netapi = LoadLibrary(TEXT("NETAPI32.DLL"));
230

R
Richard Levitte 已提交
231
#ifndef OPENSSL_SYS_WINCE
232 233 234
#if 1 /* There was previously a problem with NETSTATGET.  Currently, this
       * section is still experimental, but if all goes well, this conditional
       * will be removed
235
       */
236 237
	if (netapi)
		{
R
Richard Levitte 已提交
238 239
		netstatget = (NETSTATGET) GetProcAddress(netapi,TEXT("NetStatisticsGet"));
		netfree = (NETFREE) GetProcAddress(netapi,TEXT("NetApiBufferFree"));
240 241 242 243 244
		}

	if (netstatget && netfree)
		{
		LPBYTE outbuf;
245 246 247 248 249 250
		/* NetStatisticsGet() is a Unicode only function
 		 * STAT_WORKSTATION_0 contains 45 fields and STAT_SERVER_0
		 * contains 17 fields.  We treat each field as a source of
		 * one byte of entropy.
                 */

251 252
		if (netstatget(NULL, L"LanmanWorkstation", 0, 0, &outbuf) == 0)
			{
253
			RAND_add(outbuf, sizeof(STAT_WORKSTATION_0), 45);
254 255 256 257
			netfree(outbuf);
			}
		if (netstatget(NULL, L"LanmanServer", 0, 0, &outbuf) == 0)
			{
258
			RAND_add(outbuf, sizeof(STAT_SERVER_0), 17);
259 260 261 262
			netfree(outbuf);
			}
		}

263 264
	if (netapi)
		FreeLibrary(netapi);
265
#endif /* 1 */
R
Richard Levitte 已提交
266
#endif /* !OPENSSL_SYS_WINCE */
267
 
R
Richard Levitte 已提交
268
#ifndef OPENSSL_SYS_WINCE
269 270 271 272
        /* It appears like this can cause an exception deep within ADVAPI32.DLL
         * at random times on Windows 2000.  Reported by Jeffrey Altman.  
         * Only use it on NT.
	 */
B
Bodo Möller 已提交
273 274 275 276
	/* Wolfgang Marczy <WMarczy@topcall.co.at> reports that
	 * the RegQueryValueEx call below can hang on NT4.0 (SP6).
	 * So we don't use this at all for now. */
#if 0
277 278
        if ( osverinfo.dwPlatformId == VER_PLATFORM_WIN32_NT &&
		osverinfo.dwMajorVersion < 5)
279
		{
280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298
		/* Read Performance Statistics from NT/2000 registry
		 * The size of the performance data can vary from call
		 * to call so we must guess the size of the buffer to use
		 * and increase its size if we get an ERROR_MORE_DATA
		 * return instead of ERROR_SUCCESS.
		 */
		LONG   rc=ERROR_MORE_DATA;
		char * buf=NULL;
		DWORD bufsz=0;
		DWORD length;

		while (rc == ERROR_MORE_DATA)
			{
			buf = realloc(buf,bufsz+8192);
			if (!buf)
				break;
			bufsz += 8192;

			length = bufsz;
R
Richard Levitte 已提交
299
			rc = RegQueryValueEx(HKEY_PERFORMANCE_DATA, TEXT("Global"),
300 301 302 303 304 305
				NULL, NULL, buf, &length);
			}
		if (rc == ERROR_SUCCESS)
			{
                        /* For entropy count assume only least significant
			 * byte of each DWORD is random.
306
			 */
307 308
			RAND_add(&length, sizeof(length), 0);
			RAND_add(buf, length, length / 4.0);
309 310 311 312 313 314 315 316 317

			/* Close the Registry Key to allow Windows to cleanup/close
			 * the open handle
			 * Note: The 'HKEY_PERFORMANCE_DATA' key is implicitly opened
			 *       when the RegQueryValueEx above is done.  However, if
			 *       it is not explicitly closed, it can cause disk
			 *       partition manipulation problems.
			 */
			RegCloseKey(HKEY_PERFORMANCE_DATA);
318 319 320
			}
		if (buf)
			free(buf);
321
		}
B
Bodo Möller 已提交
322
#endif
R
Richard Levitte 已提交
323
#endif /* !OPENSSL_SYS_WINCE */
324 325 326 327

	if (advapi)
		{
		acquire = (CRYPTACQUIRECONTEXT) GetProcAddress(advapi,
R
Richard Levitte 已提交
328
			TEXT("CryptAcquireContextA"));
329
		gen = (CRYPTGENRANDOM) GetProcAddress(advapi,
R
Richard Levitte 已提交
330
			TEXT("CryptGenRandom"));
331
		release = (CRYPTRELEASECONTEXT) GetProcAddress(advapi,
R
Richard Levitte 已提交
332
			TEXT("CryptReleaseContext"));
333 334 335 336 337
		}

	if (acquire && gen && release)
		{
		/* poll the CryptoAPI PRNG */
338
                /* The CryptoAPI returns sizeof(buf) bytes of randomness */
339 340 341 342 343
		if (acquire(&hProvider, 0, 0, PROV_RSA_FULL,
			CRYPT_VERIFYCONTEXT))
			{
			if (gen(hProvider, sizeof(buf), buf) != 0)
				{
344
				RAND_add(buf, sizeof(buf), 0);
B
Bodo Möller 已提交
345
#if 0
346
				printf("randomness from PROV_RSA_FULL\n");
347
#endif
348 349 350 351 352 353 354 355 356
				}
			release(hProvider, 0); 
			}
		
		/* poll the Pentium PRG with CryptoAPI */
		if (acquire(&hProvider, 0, INTEL_DEF_PROV, PROV_INTEL_SEC, 0))
			{
			if (gen(hProvider, sizeof(buf), buf) != 0)
				{
357
				RAND_add(buf, sizeof(buf), sizeof(buf));
B
Bodo Möller 已提交
358
#if 0
359 360 361 362 363 364 365
				printf("randomness from PROV_INTEL_SEC\n");
#endif
				}
			release(hProvider, 0);
			}
		}

366 367 368
        if (advapi)
		FreeLibrary(advapi);

369 370 371 372 373 374 375 376 377
	/* timer data */
	readtimer();
	
	/* memory usage statistics */
	GlobalMemoryStatus(&m);
	RAND_add(&m, sizeof(m), 1);

	/* process ID */
	w = GetCurrentProcessId();
378
	RAND_add(&w, sizeof(w), 1);
379 380 381 382 383

	if (user)
		{
		GETCURSORINFO cursor;
		GETFOREGROUNDWINDOW win;
384
		GETQUEUESTATUS queue;
385

R
Richard Levitte 已提交
386 387 388
		win = (GETFOREGROUNDWINDOW) GetProcAddress(user, TEXT("GetForegroundWindow"));
		cursor = (GETCURSORINFO) GetProcAddress(user, TEXT("GetCursorInfo"));
		queue = (GETQUEUESTATUS) GetProcAddress(user, TEXT("GetQueueStatus"));
389 390

		if (win)
391
			{
392 393 394
			/* window handle */
			h = win();
			RAND_add(&h, sizeof(h), 0);
395
			}
396 397 398 399 400 401 402 403 404 405
		if (cursor)
			{
			/* unfortunately, its not safe to call GetCursorInfo()
			 * on NT4 even though it exists in SP3 (or SP6) and
			 * higher.
			 */
			if ( osverinfo.dwPlatformId == VER_PLATFORM_WIN32_NT &&
				osverinfo.dwMajorVersion < 5)
				cursor = 0;
			}
406 407 408
		if (cursor)
			{
			/* cursor position */
409
                        /* assume 2 bytes of entropy */
410 411 412
			CURSORINFO ci;
			ci.cbSize = sizeof(CURSORINFO);
			if (cursor(&ci))
413
				RAND_add(&ci, ci.cbSize, 2);
414 415 416 417 418
			}

		if (queue)
			{
			/* message queue status */
419
                        /* assume 1 byte of entropy */
420
			w = queue(QS_ALLEVENTS);
421
			RAND_add(&w, sizeof(w), 1);
422
			}
423 424

		FreeLibrary(user);
425 426 427 428
		}

	/* Toolhelp32 snapshot: enumerate processes, threads, modules and heap
	 * http://msdn.microsoft.com/library/psdk/winbase/toolhelp_5pfd.htm
429
	 * (Win 9x and 2000 only, not available on NT)
430 431 432
	 *
	 * This seeding method was proposed in Peter Gutmann, Software
	 * Generation of Practically Strong Random Numbers,
R
Richard Levitte 已提交
433
	 * http://www.usenix.org/publications/library/proceedings/sec98/gutmann.html
434
     * revised version at http://www.cryptoengines.com/~peter/06_random.pdf
435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458
	 * (The assignment of entropy estimates below is arbitrary, but based
	 * on Peter's analysis the full poll appears to be safe. Additional
	 * interactive seeding is encouraged.)
	 */

	if (kernel)
		{
		CREATETOOLHELP32SNAPSHOT snap;
		HANDLE handle;

		HEAP32FIRST heap_first;
		HEAP32NEXT heap_next;
		HEAP32LIST heaplist_first, heaplist_next;
		PROCESS32 process_first, process_next;
		THREAD32 thread_first, thread_next;
		MODULE32 module_first, module_next;

		HEAPLIST32 hlist;
		HEAPENTRY32 hentry;
		PROCESSENTRY32 p;
		THREADENTRY32 t;
		MODULEENTRY32 m;

		snap = (CREATETOOLHELP32SNAPSHOT)
R
Richard Levitte 已提交
459 460 461 462 463 464 465 466 467 468 469
			GetProcAddress(kernel, TEXT("CreateToolhelp32Snapshot"));
		heap_first = (HEAP32FIRST) GetProcAddress(kernel, TEXT("Heap32First"));
		heap_next = (HEAP32NEXT) GetProcAddress(kernel, TEXT("Heap32Next"));
		heaplist_first = (HEAP32LIST) GetProcAddress(kernel, TEXT("Heap32ListFirst"));
		heaplist_next = (HEAP32LIST) GetProcAddress(kernel, TEXT("Heap32ListNext"));
		process_first = (PROCESS32) GetProcAddress(kernel, TEXT("Process32First"));
		process_next = (PROCESS32) GetProcAddress(kernel, TEXT("Process32Next"));
		thread_first = (THREAD32) GetProcAddress(kernel, TEXT("Thread32First"));
		thread_next = (THREAD32) GetProcAddress(kernel, TEXT("Thread32Next"));
		module_first = (MODULE32) GetProcAddress(kernel, TEXT("Module32First"));
		module_next = (MODULE32) GetProcAddress(kernel, TEXT("Module32Next"));
470 471 472 473 474 475 476 477

		if (snap && heap_first && heap_next && heaplist_first &&
			heaplist_next && process_first && process_next &&
			thread_first && thread_next && module_first &&
			module_next && (handle = snap(TH32CS_SNAPALL,0))
			!= NULL)
			{
			/* heap list and heap walking */
478 479 480 481 482 483 484
                        /* HEAPLIST32 contains 3 fields that will change with
                         * each entry.  Consider each field a source of 1 byte
                         * of entropy.
                         * HEAPENTRY32 contains 5 fields that will change with 
                         * each entry.  Consider each field a source of 1 byte
                         * of entropy.
                         */
485 486 487 488
			hlist.dwSize = sizeof(HEAPLIST32);		
			if (heaplist_first(handle, &hlist))
				do
					{
489
					RAND_add(&hlist, hlist.dwSize, 3);
490 491 492 493
					hentry.dwSize = sizeof(HEAPENTRY32);
					if (heap_first(&hentry,
						hlist.th32ProcessID,
						hlist.th32HeapID))
R
Richard Levitte 已提交
494
						{
U
Ulf Möller 已提交
495
						int entrycnt = 80;
496 497
						do
							RAND_add(&hentry,
498
								hentry.dwSize, 5);
R
Richard Levitte 已提交
499 500 501
						while (heap_next(&hentry)
							&& --entrycnt > 0);
						}
502 503
					} while (heaplist_next(handle,
						&hlist));
504
			
505
			/* process walking */
506 507 508 509
                        /* PROCESSENTRY32 contains 9 fields that will change
                         * with each entry.  Consider each field a source of
                         * 1 byte of entropy.
                         */
510 511 512
			p.dwSize = sizeof(PROCESSENTRY32);
			if (process_first(handle, &p))
				do
513
					RAND_add(&p, p.dwSize, 9);
514
				while (process_next(handle, &p));
515

516
			/* thread walking */
517 518 519 520
                        /* THREADENTRY32 contains 6 fields that will change
                         * with each entry.  Consider each field a source of
                         * 1 byte of entropy.
                         */
521 522 523
			t.dwSize = sizeof(THREADENTRY32);
			if (thread_first(handle, &t))
				do
524
					RAND_add(&t, t.dwSize, 6);
525
				while (thread_next(handle, &t));
526

527
			/* module walking */
528 529 530 531
                        /* MODULEENTRY32 contains 9 fields that will change
                         * with each entry.  Consider each field a source of
                         * 1 byte of entropy.
                         */
532 533 534
			m.dwSize = sizeof(MODULEENTRY32);
			if (module_first(handle, &m))
				do
535
					RAND_add(&m, m.dwSize, 9);
536
				while (module_next(handle, &m));
537

538 539
			CloseHandle(handle);
			}
540 541

		FreeLibrary(kernel);
542 543
		}

B
Bodo Möller 已提交
544
#if 0
545
	printf("Exiting RAND_poll\n");
546 547
#endif

548 549
	return(1);
}
550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581

int RAND_event(UINT iMsg, WPARAM wParam, LPARAM lParam)
        {
        double add_entropy=0;

        switch (iMsg)
                {
        case WM_KEYDOWN:
                        {
                        static WPARAM key;
                        if (key != wParam)
                                add_entropy = 0.05;
                        key = wParam;
                        }
                        break;
	case WM_MOUSEMOVE:
                        {
                        static int lastx,lasty,lastdx,lastdy;
                        int x,y,dx,dy;

                        x=LOWORD(lParam);
                        y=HIWORD(lParam);
                        dx=lastx-x;
                        dy=lasty-y;
                        if (dx != 0 && dy != 0 && dx-lastdx != 0 && dy-lastdy != 0)
                                add_entropy=.2;
                        lastx=x, lasty=y;
                        lastdx=dx, lastdy=dy;
                        }
		break;
		}

582
	readtimer();
583 584 585
        RAND_add(&iMsg, sizeof(iMsg), add_entropy);
	RAND_add(&wParam, sizeof(wParam), 0);
	RAND_add(&lParam, sizeof(lParam), 0);
586
 
587 588 589 590
	return (RAND_status());
	}


591 592 593 594 595 596 597 598 599 600
void RAND_screen(void) /* function available for backward compatibility */
{
	RAND_poll();
	readscreen();
}


/* feed timing information to the PRNG */
static void readtimer(void)
{
601
	DWORD w;
602 603
	LARGE_INTEGER l;
	static int have_perfc = 1;
R
Richard Levitte 已提交
604
#if defined(_MSC_VER) && !defined(OPENSSL_SYS_WINCE)
605
	static int have_tsc = 1;
606
	DWORD cyclecount;
607 608 609 610

	if (have_tsc) {
	  __try {
	    __asm {
U
Ulf Möller 已提交
611
	      _emit 0x0f
D
Typo.  
Dr. Stephen Henson 已提交
612
	      _emit 0x31
613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637
	      mov cyclecount, eax
	      }
	    RAND_add(&cyclecount, sizeof(cyclecount), 1);
	  } __except(EXCEPTION_EXECUTE_HANDLER) {
	    have_tsc = 0;
	  }
	}
#else
# define have_tsc 0
#endif

	if (have_perfc) {
	  if (QueryPerformanceCounter(&l) == 0)
	    have_perfc = 0;
	  else
	    RAND_add(&l, sizeof(l), 0);
	}

	if (!have_tsc && !have_perfc) {
	  w = GetTickCount();
	  RAND_add(&w, sizeof(w), 0);
	}
}

/* feed screen contents to PRNG */
638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653
/*****************************************************************************
 *
 * Created 960901 by Gertjan van Oosten, gertjan@West.NL, West Consulting B.V.
 *
 * Code adapted from
 * <URL:http://www.microsoft.com/kb/developr/win_dk/q97193.htm>;
 * the original copyright message is:
 *
 *   (C) Copyright Microsoft Corp. 1993.  All rights reserved.
 *
 *   You have a royalty-free right to use, modify, reproduce and
 *   distribute the Sample Files (and/or any modified version) in
 *   any way you find useful, provided that you agree that
 *   Microsoft has no warranty obligations or liability for any
 *   Sample Application Files which are modified.
 */
654 655

static void readscreen(void)
656
{
R
Richard Levitte 已提交
657
#ifndef OPENSSL_SYS_WINCE
658 659 660 661 662 663 664 665 666 667 668 669 670
  HDC		hScrDC;		/* screen DC */
  HDC		hMemDC;		/* memory DC */
  HBITMAP	hBitmap;	/* handle for our bitmap */
  HBITMAP	hOldBitmap;	/* handle for previous bitmap */
  BITMAP	bm;		/* bitmap properties */
  unsigned int	size;		/* size of bitmap */
  char		*bmbits;	/* contents of bitmap */
  int		w;		/* screen width */
  int		h;		/* screen height */
  int		y;		/* y-coordinate of screen lines to grab */
  int		n = 16;		/* number of screen lines to grab at a time */

  /* Create a screen DC and a memory DC compatible to screen DC */
R
Richard Levitte 已提交
671
  hScrDC = CreateDC(TEXT("DISPLAY"), NULL, NULL, NULL);
672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687
  hMemDC = CreateCompatibleDC(hScrDC);

  /* Get screen resolution */
  w = GetDeviceCaps(hScrDC, HORZRES);
  h = GetDeviceCaps(hScrDC, VERTRES);

  /* Create a bitmap compatible with the screen DC */
  hBitmap = CreateCompatibleBitmap(hScrDC, w, n);

  /* Select new bitmap into memory DC */
  hOldBitmap = SelectObject(hMemDC, hBitmap);

  /* Get bitmap properties */
  GetObject(hBitmap, sizeof(BITMAP), (LPSTR)&bm);
  size = (unsigned int)bm.bmWidthBytes * bm.bmHeight * bm.bmPlanes;

688
  bmbits = OPENSSL_malloc(size);
689 690 691 692 693 694 695 696 697 698 699 700
  if (bmbits) {
    /* Now go through the whole screen, repeatedly grabbing n lines */
    for (y = 0; y < h-n; y += n)
    	{
	unsigned char md[MD_DIGEST_LENGTH];

	/* Bitblt screen DC to memory DC */
	BitBlt(hMemDC, 0, 0, w, n, hScrDC, 0, y, SRCCOPY);

	/* Copy bitmap bits from memory DC to bmbits */
	GetBitmapBits(hBitmap, size, bmbits);

701
	/* Get the hash of the bitmap */
702 703
	MD(bmbits,size,md);

704 705
	/* Seed the random generator with the hash value */
	RAND_add(md, MD_DIGEST_LENGTH, 0);
706 707
	}

708
    OPENSSL_free(bmbits);
709 710 711 712 713 714 715 716 717
  }

  /* Select old bitmap back into memory DC */
  hBitmap = SelectObject(hMemDC, hOldBitmap);

  /* Clean up */
  DeleteObject(hBitmap);
  DeleteDC(hMemDC);
  DeleteDC(hScrDC);
R
Richard Levitte 已提交
718
#endif /* !OPENSSL_SYS_WINCE */
719 720 721
}

#endif