md_rand.c 21.3 KB
Newer Older
1
/* crypto/rand/md_rand.c */
2
/* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
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
 * 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.]
 */
B
Bodo Möller 已提交
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
/* ====================================================================
 * 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).
 *
 */
111

112
#define ENTROPY_NEEDED 16  /* require 128 bits = 16 bytes of randomness */
113

B
Bodo Möller 已提交
114 115 116 117 118 119 120
#ifndef MD_RAND_DEBUG
# ifndef NDEBUG
#   define NDEBUG
# endif
#endif

#include <assert.h>
121 122
#include <stdio.h>
#include <time.h>
B
Ben Laurie 已提交
123
#include <string.h>
124

125
#include "openssl/e_os.h"
126

127
#include <openssl/crypto.h>
128
#include <openssl/err.h>
129 130

#if !defined(USE_MD5_RAND) && !defined(USE_SHA1_RAND) && !defined(USE_MDC2_RAND) && !defined(USE_MD2_RAND)
U
Ulf Möller 已提交
131
#if !defined(NO_SHA) && !defined(NO_SHA1)
132
#define USE_SHA1_RAND
U
Ulf Möller 已提交
133 134 135
#elif !defined(NO_MD5)
#define USE_MD5_RAND
#elif !defined(NO_MDC2) && !defined(NO_DES)
136 137 138 139
#define USE_MDC2_RAND
#elif !defined(NO_MD2)
#define USE_MD2_RAND
#else
U
Ulf Möller 已提交
140
#error No message digest algorithm available
141 142 143 144 145 146 147 148 149 150
#endif
#endif

/* Changed how the state buffer used.  I now attempt to 'wrap' such
 * that I don't run over the same locations the next time  go through
 * the 1023 bytes - many thanks to
 * Robert J. LeBlanc <rjl@renaissoft.com> for his comments
 */

#if defined(USE_MD5_RAND)
151
#include <openssl/md5.h>
152 153 154 155 156
#define MD_DIGEST_LENGTH	MD5_DIGEST_LENGTH
#define MD_CTX			MD5_CTX
#define MD_Init(a)		MD5_Init(a)
#define MD_Update(a,b,c)	MD5_Update(a,b,c)
#define	MD_Final(a,b)		MD5_Final(a,b)
157
#define	MD(a,b,c)		MD5(a,b,c)
158
#elif defined(USE_SHA1_RAND)
159
#include <openssl/sha.h>
160 161 162 163 164
#define MD_DIGEST_LENGTH	SHA_DIGEST_LENGTH
#define MD_CTX			SHA_CTX
#define MD_Init(a)		SHA1_Init(a)
#define MD_Update(a,b,c)	SHA1_Update(a,b,c)
#define	MD_Final(a,b)		SHA1_Final(a,b)
165
#define	MD(a,b,c)		SHA1(a,b,c)
166
#elif defined(USE_MDC2_RAND)
167
#include <openssl/mdc2.h>
168 169 170 171 172
#define MD_DIGEST_LENGTH	MDC2_DIGEST_LENGTH
#define MD_CTX			MDC2_CTX
#define MD_Init(a)		MDC2_Init(a)
#define MD_Update(a,b,c)	MDC2_Update(a,b,c)
#define	MD_Final(a,b)		MDC2_Final(a,b)
173
#define	MD(a,b,c)		MDC2(a,b,c)
174
#elif defined(USE_MD2_RAND)
175
#include <openssl/md2.h>
176 177 178 179 180
#define MD_DIGEST_LENGTH	MD2_DIGEST_LENGTH
#define MD_CTX			MD2_CTX
#define MD_Init(a)		MD2_Init(a)
#define MD_Update(a,b,c)	MD2_Update(a,b,c)
#define	MD_Final(a,b)		MD2_Final(a,b)
181
#define	MD(a,b,c)		MD2(a,b,c)
182 183
#endif

184
#include <openssl/rand.h>
185

186 187 188 189
#ifdef BN_DEBUG
# define PREDICT
#endif

190 191
/* #define NORAND	1 */
/* #define PREDICT	1 */
192 193 194

#define STATE_SIZE	1023
static int state_num=0,state_index=0;
195
static unsigned char state[STATE_SIZE+MD_DIGEST_LENGTH];
196
static unsigned char md[MD_DIGEST_LENGTH];
197
static long md_count[2]={0,0};
198
static double entropy=0;
U
Ulf Möller 已提交
199
static int initialized=0;
200

201 202 203 204
#ifdef PREDICT
int rand_predictable=0;
#endif

B
Ben Laurie 已提交
205
const char *RAND_version="RAND" OPENSSL_VERSION_PTEXT;
206

207
static void ssleay_rand_cleanup(void);
208
static void ssleay_rand_seed(const void *buf, int num);
209
static void ssleay_rand_add(const void *buf, int num, double add_entropy);
210
static int ssleay_rand_bytes(unsigned char *buf, int num);
211
static int ssleay_rand_pseudo_bytes(unsigned char *buf, int num);
212
static int ssleay_rand_status(void);
213

214
RAND_METHOD rand_ssleay_meth={
215 216 217
	ssleay_rand_seed,
	ssleay_rand_bytes,
	ssleay_rand_cleanup,
218
	ssleay_rand_add,
219
	ssleay_rand_pseudo_bytes,
220
	ssleay_rand_status
221 222
	}; 

U
Ulf Möller 已提交
223
RAND_METHOD *RAND_SSLeay(void)
224
	{
225
	return(&rand_ssleay_meth);
226 227
	}

U
Ulf Möller 已提交
228
static void ssleay_rand_cleanup(void)
229
	{
230
	memset(state,0,sizeof(state));
231 232 233
	state_num=0;
	state_index=0;
	memset(md,0,MD_DIGEST_LENGTH);
234 235
	md_count[0]=0;
	md_count[1]=0;
236
	entropy=0;
237 238
	}

239
static void ssleay_rand_add(const void *buf, int num, double add)
240
	{
B
Bodo Möller 已提交
241 242 243
	int i,j,k,st_idx;
	long md_c[2];
	unsigned char local_md[MD_DIGEST_LENGTH];
244 245 246 247 248 249
	MD_CTX m;

#ifdef NORAND
	return;
#endif

B
Bodo Möller 已提交
250
	/*
U
Ulf Möller 已提交
251
	 * (Based on the rand(3) manpage)
B
Bodo Möller 已提交
252
	 *
253
	 * The input is chopped up into units of 20 bytes (or less for
U
Ulf Möller 已提交
254
	 * the last block).  Each of these blocks is run through the hash
255
	 * function as follows:  The data passed to the hash function
B
Bodo Möller 已提交
256 257 258 259 260
	 * is the current 'md', the same number of bytes from the 'state'
	 * (the location determined by in incremented looping index) as
	 * the current 'block', the new key data 'block', and 'count'
	 * (which is incremented after each use).
	 * The result of this is kept in 'md' and also xored into the
U
Ulf Möller 已提交
261 262
	 * 'state' at the same locations that were used as input into the
         * hash function.
B
Bodo Möller 已提交
263 264
	 */

265 266 267
	CRYPTO_w_lock(CRYPTO_LOCK_RAND);
	st_idx=state_index;

B
Bodo Möller 已提交
268 269 270 271 272 273 274 275 276 277 278
	/* use our own copies of the counters so that even
	 * if a concurrent thread seeds with exactly the
	 * same data and uses the same subarray there's _some_
	 * difference */
	md_c[0] = md_count[0];
	md_c[1] = md_count[1];

	memcpy(local_md, md, sizeof md);

	/* state_index <= state_num <= STATE_SIZE */
	state_index += num;
279
	if (state_index >= STATE_SIZE)
280 281 282 283 284 285 286 287 288
		{
		state_index%=STATE_SIZE;
		state_num=STATE_SIZE;
		}
	else if (state_num < STATE_SIZE)	
		{
		if (state_index > state_num)
			state_num=state_index;
		}
B
Bodo Möller 已提交
289 290 291 292 293 294 295 296
	/* state_index <= state_num <= STATE_SIZE */

	/* state[st_idx], ..., state[(st_idx + num - 1) % STATE_SIZE]
	 * are what we will use now, but other threads may use them
	 * as well */

	md_count[1] += (num / MD_DIGEST_LENGTH) + (num % MD_DIGEST_LENGTH > 0);

297 298 299 300 301 302 303 304
	CRYPTO_w_unlock(CRYPTO_LOCK_RAND);

	for (i=0; i<num; i+=MD_DIGEST_LENGTH)
		{
		j=(num-i);
		j=(j > MD_DIGEST_LENGTH)?MD_DIGEST_LENGTH:j;

		MD_Init(&m);
B
Bodo Möller 已提交
305
		MD_Update(&m,local_md,MD_DIGEST_LENGTH);
306 307 308 309 310 311 312 313 314 315
		k=(st_idx+j)-STATE_SIZE;
		if (k > 0)
			{
			MD_Update(&m,&(state[st_idx]),j-k);
			MD_Update(&m,&(state[0]),k);
			}
		else
			MD_Update(&m,&(state[st_idx]),j);
			
		MD_Update(&m,buf,j);
B
Bodo Möller 已提交
316 317 318
		MD_Update(&m,(unsigned char *)&(md_c[0]),sizeof(md_c));
		MD_Final(local_md,&m);
		md_c[1]++;
319

B
Ben Laurie 已提交
320
		buf=(const char *)buf + j;
321 322 323

		for (k=0; k<j; k++)
			{
B
Bodo Möller 已提交
324 325 326 327 328 329 330 331 332
			/* Parallel threads may interfere with this,
			 * but always each byte of the new state is
			 * the XOR of some previous value of its
			 * and local_md (itermediate values may be lost).
			 * Alway using locking could hurt performance more
			 * than necessary given that conflicts occur only
			 * when the total seeding is longer than the random
			 * state. */
			state[st_idx++]^=local_md[k];
333 334 335 336 337
			if (st_idx >= STATE_SIZE)
				st_idx=0;
			}
		}
	memset((char *)&m,0,sizeof(m));
B
Bodo Möller 已提交
338 339 340 341 342 343 344 345 346 347

	CRYPTO_w_lock(CRYPTO_LOCK_RAND);
	/* Don't just copy back local_md into md -- this could mean that
	 * other thread's seeding remains without effect (except for
	 * the incremented counter).  By XORing it we keep at least as
	 * much entropy as fits into md. */
	for (k = 0; k < sizeof md; k++)
		{
		md[k] ^= local_md[k];
		}
B
Bodo Möller 已提交
348 349
	if (entropy < ENTROPY_NEEDED) /* stop counting when we have enough */
	    entropy += add;
B
Bodo Möller 已提交
350 351 352 353 354
	CRYPTO_w_unlock(CRYPTO_LOCK_RAND);
	
#ifndef THREADS	
	assert(md_c[1] == md_count[1]);
#endif
355 356 357 358 359
	}

static void ssleay_rand_seed(const void *buf, int num)
	{
	ssleay_rand_add(buf, num, num);
360 361
	}

U
Ulf Möller 已提交
362 363 364 365 366 367 368 369 370 371
static void ssleay_rand_initialize(void)
	{
	unsigned long l;
#ifndef GETPID_IS_MEANINGLESS
	pid_t curr_pid = getpid();
#endif
#ifdef DEVRANDOM
	FILE *fh;
#endif

372 373 374 375
#ifdef NORAND
	return;
#endif

U
Ulf Möller 已提交
376 377 378 379 380 381 382 383 384 385 386 387
	CRYPTO_w_unlock(CRYPTO_LOCK_RAND);
	/* put in some default random data, we need more than just this */
#ifndef GETPID_IS_MEANINGLESS
	l=curr_pid;
	RAND_add(&l,sizeof(l),0);
	l=getuid();
	RAND_add(&l,sizeof(l),0);
#endif
	l=time(NULL);
	RAND_add(&l,sizeof(l),0);

#ifdef DEVRANDOM
B
Bodo Möller 已提交
388 389
	/* Use a random entropy pool device. Linux, FreeBSD and OpenBSD
	 * have this. Use /dev/urandom if you can as /dev/random may block
U
Ulf Möller 已提交
390 391 392 393 394 395 396
	 * if it runs out of random entries.  */

	if ((fh = fopen(DEVRANDOM, "r")) != NULL)
		{
		unsigned char tmpbuf[ENTROPY_NEEDED];
		int n;
		
397
		setvbuf(fh, NULL, _IONBF, 0);
U
Ulf Möller 已提交
398 399 400 401 402 403 404 405 406 407 408 409 410 411
		n=fread((unsigned char *)tmpbuf,1,ENTROPY_NEEDED,fh);
		fclose(fh);
		RAND_add(tmpbuf,sizeof tmpbuf,n);
		memset(tmpbuf,0,n);
		}
#endif
#ifdef PURIFY
	memset(state,0,STATE_SIZE);
	memset(md,0,MD_DIGEST_LENGTH);
#endif
	CRYPTO_w_lock(CRYPTO_LOCK_RAND);
	initialized=1;
	}

412
static int ssleay_rand_bytes(unsigned char *buf, int num)
413 414
	{
	int i,j,k,st_num,st_idx;
415
	int ok;
B
Bodo Möller 已提交
416 417
	long md_c[2];
	unsigned char local_md[MD_DIGEST_LENGTH];
418
	MD_CTX m;
A
Andy Polyakov 已提交
419
#ifndef GETPID_IS_MEANINGLESS
420
	pid_t curr_pid = getpid();
B
Bodo Möller 已提交
421
#endif
422 423

#ifdef PREDICT
424 425 426
	if (rand_predictable)
		{
		static unsigned char val=0;
427

428 429 430 431
		for (i=0; i<num; i++)
			buf[i]=val++;
		return(1);
		}
432 433
#endif

B
Bodo Möller 已提交
434
	/*
B
Bodo Möller 已提交
435
	 * (Based on the rand(3) manpage:)
B
Bodo Möller 已提交
436
	 *
437
	 * For each group of 10 bytes (or less), we do the following:
B
Bodo Möller 已提交
438
	 *
439 440 441 442
	 * Input into the hash function the top 10 bytes from the
	 * local 'md' (which is initialized from the global 'md'
	 * before any bytes are generated), the bytes that are
	 * to be overwritten by the random bytes, and bytes from the
B
Bodo Möller 已提交
443
	 * 'state' (incrementing looping index).  From this digest output
444 445
	 * (which is kept in 'md'), the top (up to) 10 bytes are
	 * returned to the caller and the bottom (up to) 10 bytes are xored
B
Bodo Möller 已提交
446 447
	 * into the 'state'.
	 * Finally, after we have finished 'num' random bytes for the
B
Bodo Möller 已提交
448 449 450
	 * caller, 'count' (which is incremented) and the local and global 'md'
	 * are fed into the hash function and the results are kept in the
	 * global 'md'.
B
Bodo Möller 已提交
451 452
	 */

453 454
	CRYPTO_w_lock(CRYPTO_LOCK_RAND);

U
Ulf Möller 已提交
455 456
	if (!initialized)
		ssleay_rand_initialize();
457

458
	ok = (entropy >= ENTROPY_NEEDED);
B
Bodo Möller 已提交
459 460 461 462 463 464 465 466 467 468 469 470 471
	if (!ok)
		{
		/* If the PRNG state is not yet unpredictable, then seeing
		 * the PRNG output may help attackers to determine the new
		 * state; thus we have to decrease the entropy estimate.
		 * Once we've had enough initial seeding we don't bother to
		 * adjust the entropy count, though, because we're not ambitious
		 * to provide *information-theoretic* randomness.
		 */
		entropy -= num;
		if (entropy < 0)
			entropy = 0;
		}
472

473 474
	st_idx=state_index;
	st_num=state_num;
B
Bodo Möller 已提交
475 476 477 478
	md_c[0] = md_count[0];
	md_c[1] = md_count[1];
	memcpy(local_md, md, sizeof md);

479 480
	state_index+=num;
	if (state_index > state_num)
B
Bodo Möller 已提交
481 482 483 484
		state_index %= state_num;

	/* state[st_idx], ..., state[(st_idx + num - 1) % st_num]
	 * are now ours (but other threads may use them too) */
485

B
Bodo Möller 已提交
486
	md_count[0] += 1;
487 488 489 490 491 492 493
	CRYPTO_w_unlock(CRYPTO_LOCK_RAND);

	while (num > 0)
		{
		j=(num >= MD_DIGEST_LENGTH/2)?MD_DIGEST_LENGTH/2:num;
		num-=j;
		MD_Init(&m);
A
Andy Polyakov 已提交
494
#ifndef GETPID_IS_MEANINGLESS
495 496 497 498 499 500
		if (curr_pid) /* just in the first iteration to save time */
			{
			MD_Update(&m,(unsigned char*)&curr_pid,sizeof curr_pid);
			curr_pid = 0;
			}
#endif
B
Bodo Möller 已提交
501 502
		MD_Update(&m,&(local_md[MD_DIGEST_LENGTH/2]),MD_DIGEST_LENGTH/2);
		MD_Update(&m,(unsigned char *)&(md_c[0]),sizeof(md_c));
503 504 505 506 507 508 509 510 511 512 513
#ifndef PURIFY
		MD_Update(&m,buf,j); /* purify complains */
#endif
		k=(st_idx+j)-st_num;
		if (k > 0)
			{
			MD_Update(&m,&(state[st_idx]),j-k);
			MD_Update(&m,&(state[0]),k);
			}
		else
			MD_Update(&m,&(state[st_idx]),j);
B
Bodo Möller 已提交
514
		MD_Final(local_md,&m);
515 516 517

		for (i=0; i<j; i++)
			{
B
Bodo Möller 已提交
518 519
			state[st_idx++]^=local_md[i]; /* may compete with other threads */
			*(buf++)=local_md[i+MD_DIGEST_LENGTH/2];
520 521 522 523 524 525
			if (st_idx >= st_num)
				st_idx=0;
			}
		}

	MD_Init(&m);
B
Bodo Möller 已提交
526 527 528
	MD_Update(&m,(unsigned char *)&(md_c[0]),sizeof(md_c));
	MD_Update(&m,local_md,MD_DIGEST_LENGTH);
	CRYPTO_w_lock(CRYPTO_LOCK_RAND);
529 530
	MD_Update(&m,md,MD_DIGEST_LENGTH);
	MD_Final(md,&m);
B
Bodo Möller 已提交
531 532
	CRYPTO_w_unlock(CRYPTO_LOCK_RAND);

533
	memset(&m,0,sizeof(m));
534 535 536 537 538 539 540
	if (ok)
		return(1);
	else
		{
		RANDerr(RAND_F_SSLEAY_RAND_BYTES,RAND_R_PRNG_NOT_SEEDED);
		return(0);
		}
541 542
	}

543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559
/* pseudo-random bytes that are guaranteed to be unique but not
   unpredictable */
static int ssleay_rand_pseudo_bytes(unsigned char *buf, int num) 
	{
	int ret, err;

	ret = RAND_bytes(buf, num);
	if (ret == 0)
		{
		err = ERR_peek_error();
		if (ERR_GET_LIB(err) == ERR_LIB_RAND &&
		    ERR_GET_REASON(err) == RAND_R_PRNG_NOT_SEEDED)
			(void)ERR_get_error();
		}
	return (ret);
	}

560
static int ssleay_rand_status(void)
U
Ulf Möller 已提交
561
	{
562 563
	int ret;

564 565
	CRYPTO_w_lock(CRYPTO_LOCK_RAND);

U
Ulf Möller 已提交
566 567
	if (!initialized)
		ssleay_rand_initialize();
568
	ret = entropy >= ENTROPY_NEEDED;
569 570 571

	CRYPTO_w_unlock(CRYPTO_LOCK_RAND);

572
	return ret;
U
Ulf Möller 已提交
573 574
	}

575 576
#ifdef WINDOWS
#include <windows.h>
577
#include <openssl/rand.h>
578

579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603
int RAND_event(UINT iMsg, WPARAM wParam, LPARAM lParam)
        {
        double add_entropy=0;
        SYSTEMTIME t;

        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)
604 605
                                add_entropy=.2;
                        lastx=x, lasty=y;
606 607 608 609 610 611 612 613 614 615 616 617 618 619
                        lastdx=dx, lastdy=dy;
                        }
		break;
		}

        GetSystemTime(&t);
        RAND_add(&iMsg, sizeof(iMsg), add_entropy);
	RAND_add(&wParam, sizeof(wParam), 0);
	RAND_add(&lParam, sizeof(lParam), 0);
        RAND_add(&t, sizeof(t), 0);

	return (RAND_status());
	}

620 621 622 623 624 625 626 627 628 629
/*****************************************************************************
 * Initialisation function for the SSL random generator.  Takes the contents
 * of the screen as random seed.
 *
 * 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:
 *
630 631 632 633 634 635 636
 *   (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.
637 638 639
 */
/*
 * I have modified the loading of bytes via RAND_seed() mechanism since
U
Ulf Möller 已提交
640
 * the original would have been very very CPU intensive since RAND_seed()
641 642
 * does an MD5 per 16 bytes of input.  The cost to digest 16 bytes is the same
 * as that to digest 56 bytes.  So under the old system, a screen of
U
Ulf Möller 已提交
643
 * 1024*768*256 would have been CPU cost of approximately 49,000 56 byte MD5
644
 * digests or digesting 2.7 mbytes.  What I have put in place would
U
Ulf Möller 已提交
645
 * be 48 16k MD5 digests, or effectively 48*16+48 MD5 bytes or 816 kbytes
646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694
 * or about 3.5 times as much.
 * - eric 
 */
void RAND_screen(void)
{
  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 */
  hScrDC = CreateDC("DISPLAY", NULL, NULL, NULL);
  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;

  bmbits = Malloc(size);
  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);

	/* Get the MD5 of the bitmap */
695
	MD(bmbits,size,md);
696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712

	/* Seed the random generator with the MD5 digest */
	RAND_seed(md, MD_DIGEST_LENGTH);
	}

    Free(bmbits);
  }

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

  /* Clean up */
  DeleteObject(hBitmap);
  DeleteDC(hMemDC);
  DeleteDC(hScrDC);
}
#endif