Mix/Src/random.c

/* Mixmaster version 3  --  (C) 1999 Anonymizer Inc.

   Mixmaster may be redistributed and modified under certain conditions.
   This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF
   ANY KIND, either express or implied. See the file COPYRIGHT for
   details.

   Randomness
   $Id: random.c,v 1.3 2002/09/18 23:26:16 rabbi Exp $ */


#include "mix3.h"
#include "crypto.h"
#include <fcntl.h>
#ifdef POSIX
#include <sys/time.h>
#include <unistd.h>
#else /* end of POSIX */
#include <io.h>
#include <process.h>
#endif /* else if not POSIX */
#ifdef WIN32
#include <windows.h>
#endif /* WIN32 */
#include <assert.h>
#include <string.h>

int rnd_state = RND_NOTSEEDED;

#ifdef USE_OPENSSL
int rnd_init(void)
{
  char r[PATHMAX];
  int n;
  LOCK *rndlock;

  if (rnd_state == RND_SEEDED)
    return(0);
  rndlock = lockfile(MIXRAND);
  mixfile(r, MIXRAND);
  n = RAND_load_file(r, 1024);
  if (n < 256 && rnd_seed() == -1)
    goto err;
  rnd_time();
  RAND_write_file(r);
  rnd_state = RND_SEEDED;
 err:
  unlockfile(rndlock);
  return (rnd_state == RND_SEEDED ? 0 : -1);
}

int rnd_final(void)
{
  int err = 0;
  char r[PATHMAX];
  LOCK *rndlock;

  if (rnd_state != RND_SEEDED)
    return(-1);

  rnd_update(NULL, 0);
  rndlock = lockfile(MIXRAND);
  mixfile(r, MIXRAND);
  RAND_load_file(r, 1024);      /* add seed file again in case other instances
                                   of the program have used it */
  if (RAND_write_file(r) < 1)
    err = -1;
  unlockfile(rndlock);
  RAND_cleanup();
  return (err);
}

int rnd_add(byte *b, int l)
{
  RAND_seed(b, l);
  return (0);
}
#endif /* USE_OPENSSL */

void rnd_time(void)
{
  int pid;

#ifdef WIN32
  SYSTEMTIME t;
#endif /* WIN32 */

#ifdef HAVE_GETTIMEOFDAY
  struct timeval tv;

  gettimeofday(&tv, 0);
  rnd_add((byte *) &tv, sizeof(tv));
#elif defined(WIN32) /* end of HAVE_GETTIMEOFDAY */
  GetSystemTime(&t);
  rnd_add((byte *) &t, sizeof(t));
#else /* end of defined(WIN32) */
  rnd_add((byte *) time(NULL), sizeof(time_t));
#endif /* else if not defined(WIN32), HAVE_GETTIMEOFDAY */
  pid = getpid();
  rnd_add((byte *) &pid, sizeof(pid));
}

void rnd_update(byte *seed, int l)
{
  int fd = -1;
  byte b[512];

  rnd_time();
  if (seed)
    rnd_add(seed, l);
#ifdef DEV_URANDOM
  fd = open(DEV_URANDOM, O_RDONLY);
  if (fd != -1) {
    ssize_t ret;

    ret = read(fd, b, sizeof(b));
    if (ret > 0) {
      rnd_add(b, ret);
    }
    close(fd);
  }
#endif /* DEV_URANDOM */
}

int rnd_bytes(byte *b, int n)
{
  /* we frequently need to get small amounts of random data.
     speed up by pre-generating dating data */

  static byte rand[BUFSIZE];
  static int idx = BUFSIZE;

  if (rnd_state != RND_SEEDED)
    rnd_error();

  if (n + idx < BUFSIZE) {
    memcpy(b, rand + idx, n);
    idx += n;
  } else
    RAND_bytes(b, n);

  if (idx + 256 > BUFSIZE) {
    RAND_bytes(rand, BUFSIZE);
    idx = 0;
  }
  return (0);
}

int rnd_number(int n)
{
  int r;

  assert(n > 0);
  if (n > 65535)
    do
      r = rnd_byte() * 65536 +
        rnd_byte() * 256 + rnd_byte();
    while (r >= n);
  else if (n > 255)
    do
      r = rnd_byte() * 256 + rnd_byte();
    while (r >= n);
  else
    do
      r = rnd_byte();
    while (r >= n);
  return r;
}

byte rnd_byte()
{
  byte b;

  rnd_bytes(&b, 1);
  return b;
}

void rnd_initialized(void)
{
  rnd_state = RND_SEEDED;
}

#ifdef WIN32

#define NEEDED 256

int rnd_mouse(UINT i, WPARAM w, LPARAM l)
{
  static int entropy = 0;
  static int x, y, dx, dy;
  int newx, newy, newdx, newdy;
  int rnd[4];

  if (i == WM_MOUSEMOVE) {
    newx = LOWORD(l);
    newy = HIWORD(l);
    newdx = x - newx;
    newdy = y - newy;
    if (dx != 0 && dy != 0 && dx - newdx != 0 && dy - newdy != 0) {
      entropy++;
      if (entropy >= NEEDED)
        rnd_state = RND_SEEDED;
    }
    x = newx, y = newy, dx = newdx, dy = newdy;
    rnd[0] = x; rnd[1] = y; rnd[2] = dx; rnd[3] = dy;
    rnd_update((byte*)rnd, 4 * sizeof(int));
  }
  return (rnd_state == RND_SEEDED ? 100 : entropy * 100 / NEEDED);
}
#endif /* WIN32 */
1	rabbi	1	/* Mixmaster version 3 -- (C) 1999 Anonymizer Inc.
2
3			Mixmaster may be redistributed and modified under certain conditions.
4			This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF
5			ANY KIND, either express or implied. See the file COPYRIGHT for
6			details.
7
8			Randomness
9	rabbi	262	$Id: random.c,v 1.3 2002/09/18 23:26:16 rabbi Exp $ */
10	rabbi	1
11
12			#include "mix3.h"
13			#include "crypto.h"
14			#include <fcntl.h>
15			#ifdef POSIX
16			#include <sys/time.h>
17			#include <unistd.h>
18	rabbi	262	#else /* end of POSIX */
19	rabbi	1	#include <io.h>
20			#include <process.h>
21	rabbi	262	#endif /* else if not POSIX */
22	rabbi	1	#ifdef WIN32
23			#include <windows.h>
24	rabbi	262	#endif /* WIN32 */
25	rabbi	1	#include <assert.h>
26	weaselp	120	#include <string.h>
27	rabbi	1
28			int rnd_state = RND_NOTSEEDED;
29
30			#ifdef USE_OPENSSL
31			int rnd_init(void)
32			{
33			char r[PATHMAX];
34			int n;
35			LOCK *rndlock;
36
37			if (rnd_state == RND_SEEDED)
38			return(0);
39			rndlock = lockfile(MIXRAND);
40			mixfile(r, MIXRAND);
41			n = RAND_load_file(r, 1024);
42			if (n < 256 && rnd_seed() == -1)
43			goto err;
44			rnd_time();
45			RAND_write_file(r);
46			rnd_state = RND_SEEDED;
47			err:
48			unlockfile(rndlock);
49			return (rnd_state == RND_SEEDED ? 0 : -1);
50			}
51
52			int rnd_final(void)
53			{
54			int err = 0;
55			char r[PATHMAX];
56			LOCK *rndlock;
57
58			if (rnd_state != RND_SEEDED)
59			return(-1);
60
61			rnd_update(NULL, 0);
62			rndlock = lockfile(MIXRAND);
63			mixfile(r, MIXRAND);
64			RAND_load_file(r, 1024); /* add seed file again in case other instances
65			of the program have used it */
66			if (RAND_write_file(r) < 1)
67			err = -1;
68			unlockfile(rndlock);
69			RAND_cleanup();
70			return (err);
71			}
72
73			int rnd_add(byte *b, int l)
74			{
75			RAND_seed(b, l);
76			return (0);
77			}
78	rabbi	262	#endif /* USE_OPENSSL */
79	rabbi	1
80			void rnd_time(void)
81			{
82			int pid;
83
84			#ifdef WIN32
85			SYSTEMTIME t;
86	rabbi	262	#endif /* WIN32 */
87	rabbi	1
88			#ifdef HAVE_GETTIMEOFDAY
89			struct timeval tv;
90
91			gettimeofday(&tv, 0);
92			rnd_add((byte *) &tv, sizeof(tv));
93	rabbi	262	#elif defined(WIN32) /* end of HAVE_GETTIMEOFDAY */
94	rabbi	1	GetSystemTime(&t);
95			rnd_add((byte *) &t, sizeof(t));
96	rabbi	262	#else /* end of defined(WIN32) */
97	rabbi	1	rnd_add((byte *) time(NULL), sizeof(time_t));
98	rabbi	262	#endif /* else if not defined(WIN32), HAVE_GETTIMEOFDAY */
99	rabbi	1	pid = getpid();
100			rnd_add((byte *) &pid, sizeof(pid));
101			}
102
103			void rnd_update(byte *seed, int l)
104			{
105			int fd = -1;
106			byte b[512];
107
108			rnd_time();
109			if (seed)
110			rnd_add(seed, l);
111			#ifdef DEV_URANDOM
112			fd = open(DEV_URANDOM, O_RDONLY);
113			if (fd != -1) {
114	weaselp	120	ssize_t ret;
115
116			ret = read(fd, b, sizeof(b));
117			if (ret > 0) {
118			rnd_add(b, ret);
119			}
120	rabbi	1	close(fd);
121			}
122	rabbi	262	#endif /* DEV_URANDOM */
123	rabbi	1	}
124
125			int rnd_bytes(byte *b, int n)
126			{
127			/* we frequently need to get small amounts of random data.
128			speed up by pre-generating dating data */
129
130			static byte rand[BUFSIZE];
131			static int idx = BUFSIZE;
132
133			if (rnd_state != RND_SEEDED)
134			rnd_error();
135
136			if (n + idx < BUFSIZE) {
137			memcpy(b, rand + idx, n);
138			idx += n;
139			} else
140			RAND_bytes(b, n);
141
142			if (idx + 256 > BUFSIZE) {
143			RAND_bytes(rand, BUFSIZE);
144			idx = 0;
145			}
146			return (0);
147			}
148
149			int rnd_number(int n)
150			{
151			int r;
152
153			assert(n > 0);
154			if (n > 65535)
155			do
156			r = rnd_byte() * 65536 +
157			rnd_byte() * 256 + rnd_byte();
158			while (r >= n);
159			else if (n > 255)
160			do
161			r = rnd_byte() * 256 + rnd_byte();
162			while (r >= n);
163			else
164			do
165			r = rnd_byte();
166			while (r >= n);
167			return r;
168			}
169
170			byte rnd_byte()
171			{
172			byte b;
173
174			rnd_bytes(&b, 1);
175			return b;
176			}
177
178			void rnd_initialized(void)
179			{
180			rnd_state = RND_SEEDED;
181			}
182
183			#ifdef WIN32
184
185			#define NEEDED 256
186
187			int rnd_mouse(UINT i, WPARAM w, LPARAM l)
188			{
189			static int entropy = 0;
190			static int x, y, dx, dy;
191			int newx, newy, newdx, newdy;
192			int rnd[4];
193
194			if (i == WM_MOUSEMOVE) {
195			newx = LOWORD(l);
196			newy = HIWORD(l);
197			newdx = x - newx;
198			newdy = y - newy;
199			if (dx != 0 && dy != 0 && dx - newdx != 0 && dy - newdy != 0) {
200			entropy++;
201			if (entropy >= NEEDED)
202			rnd_state = RND_SEEDED;
203			}
204			x = newx, y = newy, dx = newdx, dy = newdy;
205			rnd[0] = x; rnd[1] = y; rnd[2] = dx; rnd[3] = dy;
206			rnd_update((byte)rnd, 4 sizeof(int));
207			}
208			return (rnd_state == RND_SEEDED ? 100 : entropy * 100 / NEEDED);
209			}
210	rabbi	262	#endif /* WIN32 */