See ChangeLog: Wed Dec 23 13:34:22 CET 1998 Werner Koch
[gnupg.git] / cipher / random.c
1 /* random.c  -  random number generator
2  *      Copyright (C) 1998 Free Software Foundation, Inc.
3  *
4  * This file is part of GnuPG.
5  *
6  * GnuPG is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * GnuPG is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
19  */
20
21
22 /****************
23  * This random number generator is modelled after the one described
24  * in Peter Gutmann's Paper: "Software Generation of Practically
25  * Strong Random Numbers".
26  */
27
28
29 #include <config.h>
30 #include <stdio.h>
31 #include <stdlib.h>
32 #include <assert.h>
33 #include <errno.h>
34 #include <string.h>
35 #include <sys/time.h>
36 #include <sys/types.h>
37 #include <sys/stat.h>
38 #ifdef  HAVE_GETHRTIME
39   #include <sys/times.h>
40 #endif
41 #ifdef HAVE_GETTIMEOFDAY
42   #include <sys/times.h>
43 #endif
44 #ifdef HAVE_GETRUSAGE
45   #include <sys/resource.h>
46 #endif
47 #include "util.h"
48 #include "rmd.h"
49 #include "ttyio.h"
50 #include "i18n.h"
51 #include "random.h"
52 #include "rand-internal.h"
53 #include "dynload.h"
54
55
56 #if SIZEOF_UNSIGNED_LONG == 8
57   #define ADD_VALUE 0xa5a5a5a5a5a5a5a5
58 #elif SIZEOF_UNSIGNED_LONG == 4
59   #define ADD_VALUE 0xa5a5a5a5
60 #else
61   #error weird size for an unsigned long
62 #endif
63
64 #define BLOCKLEN  64   /* hash this amount of bytes */
65 #define DIGESTLEN 20   /* into a digest of this length (rmd160) */
66 /* poolblocks is the number of digests which make up the pool
67  * and poolsize must be a multiple of the digest length
68  * to make the AND operations faster, the size should also be
69  * a multiple of ulong
70  */
71 #define POOLBLOCKS 30
72 #define POOLSIZE (POOLBLOCKS*DIGESTLEN)
73 #if (POOLSIZE % SIZEOF_UNSIGNED_LONG)
74   #error Please make sure that poolsize is a multiple of ulong
75 #endif
76 #define POOLWORDS (POOLSIZE / SIZEOF_UNSIGNED_LONG)
77
78
79 static int is_initialized;
80 #define MASK_LEVEL(a) do {if( a > 2 ) a = 2; else if( a < 0 ) a = 0; } while(0)
81 static char *rndpool;   /* allocated size is POOLSIZE+BLOCKLEN */
82 static char *keypool;   /* allocated size is POOLSIZE+BLOCKLEN */
83 static size_t pool_readpos;
84 static size_t pool_writepos;
85 static int pool_filled;
86 static int pool_balance;
87 static int just_mixed;
88
89 static int secure_alloc;
90 static int quick_test;
91 static int faked_rng;
92
93
94 static void read_pool( byte *buffer, size_t length, int level );
95 static void add_randomness( const void *buffer, size_t length, int source );
96 static void random_poll(void);
97 static void read_random_source( int requester, size_t length, int level);
98 static int gather_faked( void (*add)(const void*, size_t, int), int requester,
99                                                     size_t length, int level );
100
101
102 static void
103 initialize()
104 {
105     /* The data buffer is allocated somewhat larger, so that
106      * we can use this extra space (which is allocated in secure memory)
107      * as a temporary hash buffer */
108     rndpool = secure_alloc ? m_alloc_secure_clear(POOLSIZE+BLOCKLEN)
109                            : m_alloc_clear(POOLSIZE+BLOCKLEN);
110     keypool = secure_alloc ? m_alloc_secure_clear(POOLSIZE+BLOCKLEN)
111                            : m_alloc_clear(POOLSIZE+BLOCKLEN);
112     is_initialized = 1;
113
114   #if   USE_RNDLINUX
115     rndlinux_constructor();
116   #elif USE_RNDUNIX
117     rndunix_constructor();
118   #elif USE_RNDW32
119     rndw32_constructor();
120   #elif USE_RNDOS2
121     rndos2_constructor();
122   #elif USE_RNDATARI
123     rndatari_constructor();
124   #elif USE_RNDMVS
125     rndmvs_constructor();
126   #endif
127 }
128
129 void
130 secure_random_alloc()
131 {
132     secure_alloc = 1;
133 }
134
135
136 int
137 quick_random_gen( int onoff )
138 {
139     int last;
140
141     read_random_source(0,0,0); /* init */
142     last = quick_test;
143     if( onoff != -1 )
144         quick_test = onoff;
145     return faked_rng? 1 : last;
146 }
147
148
149 /****************
150  * Fill the buffer with LENGTH bytes of cryptographically strong
151  * random bytes. level 0 is not very strong, 1 is strong enough
152  * for most usage, 2 is good for key generation stuff but may be very slow.
153  */
154 void
155 randomize_buffer( byte *buffer, size_t length, int level )
156 {
157     char *p = get_random_bits( length*8, level, m_is_secure(buffer) );
158     memcpy( buffer, p, length );
159     m_free(p);
160 }
161
162
163
164 /****************
165  * Return a pointer to a randomized buffer of level 0 and LENGTH bits
166  * caller must free the buffer.
167  * Note: The returned value is rounded up to bytes.
168  */
169 byte *
170 get_random_bits( size_t nbits, int level, int secure )
171 {
172     byte *buf;
173     size_t nbytes = (nbits+7)/8;
174
175     if( quick_test && level > 1 )
176         level = 1;
177     MASK_LEVEL(level);
178     buf = secure && secure_alloc ? m_alloc_secure( nbytes ) : m_alloc( nbytes );
179     read_pool( buf, nbytes, level );
180     return buf;
181 }
182
183
184 /****************
185  * Mix the pool
186  */
187 static void
188 mix_pool(byte *pool)
189 {
190     char *hashbuf = pool + POOLSIZE;
191     char *p, *pend;
192     int i, n;
193     RMD160_CONTEXT md;
194
195     rmd160_init( &md );
196  #if DIGESTLEN != 20
197     #error must have a digest length of 20 for ripe-md-160
198  #endif
199     /* loop over the pool */
200     pend = pool + POOLSIZE;
201     memcpy(hashbuf, pend - DIGESTLEN, DIGESTLEN );
202     memcpy(hashbuf+DIGESTLEN, pool, BLOCKLEN-DIGESTLEN);
203     rmd160_mixblock( &md, hashbuf);
204     memcpy(pool, hashbuf, 20 );
205
206     p = pool;
207     for( n=1; n < POOLBLOCKS; n++ ) {
208         memcpy(hashbuf, p, DIGESTLEN );
209
210         p += DIGESTLEN;
211         if( p+DIGESTLEN+BLOCKLEN < pend )
212             memcpy(hashbuf+DIGESTLEN, p+DIGESTLEN, BLOCKLEN-DIGESTLEN);
213         else {
214             char *pp = p+DIGESTLEN;
215             for(i=DIGESTLEN; i < BLOCKLEN; i++ ) {
216                 if( pp >= pend )
217                     pp = pool;
218                 hashbuf[i] = *pp++;
219             }
220         }
221
222         rmd160_mixblock( &md, hashbuf);
223         memcpy(p, hashbuf, 20 );
224     }
225 }
226
227
228 static void
229 read_pool( byte *buffer, size_t length, int level )
230 {
231     int i;
232     ulong *sp, *dp;
233
234     if( length >= POOLSIZE )
235         BUG(); /* not allowed */
236
237     /* for level 2 make sure that there is enough random in the pool */
238     if( level == 2 && pool_balance < length ) {
239         size_t needed;
240
241         if( pool_balance < 0 )
242             pool_balance = 0;
243         needed = length - pool_balance;
244         if( needed > POOLSIZE )
245             BUG();
246         read_random_source( 3, needed, 2 );
247         pool_balance += needed;
248     }
249
250     /* make sure the pool is filled */
251     while( !pool_filled )
252         random_poll();
253
254     /* do always a fast random poll */
255     fast_random_poll();
256
257     if( !level ) { /* no need for cryptographic strong random */
258         /* create a new pool */
259         for(i=0,dp=(ulong*)keypool, sp=(ulong*)rndpool;
260                                     i < POOLWORDS; i++, dp++, sp++ )
261             *dp = *sp + ADD_VALUE;
262         /* must mix both pools */
263         mix_pool(rndpool);
264         mix_pool(keypool);
265         memcpy( buffer, keypool, length );
266     }
267     else {
268         /* mix the pool (if add_randomness() didn't it) */
269         if( !just_mixed )
270             mix_pool(rndpool);
271         /* create a new pool */
272         for(i=0,dp=(ulong*)keypool, sp=(ulong*)rndpool;
273                                     i < POOLWORDS; i++, dp++, sp++ )
274             *dp = *sp + ADD_VALUE;
275         /* and mix both pools */
276         mix_pool(rndpool);
277         mix_pool(keypool);
278         /* read the required data
279          * we use a readpoiter to read from a different postion each
280          * time */
281         while( length-- ) {
282             *buffer++ = keypool[pool_readpos++];
283             if( pool_readpos >= POOLSIZE )
284                 pool_readpos = 0;
285             pool_balance--;
286         }
287         if( pool_balance < 0 )
288             pool_balance = 0;
289         /* and clear the keypool */
290         memset( keypool, 0, POOLSIZE );
291     }
292 }
293
294
295 /****************
296  * Add LENGTH bytes of randomness from buffer to the pool.
297  * source may be used to specify the randomness source.
298  */
299 static void
300 add_randomness( const void *buffer, size_t length, int source )
301 {
302     if( !is_initialized )
303         initialize();
304     while( length-- ) {
305         rndpool[pool_writepos++] = *((byte*)buffer)++;
306         if( pool_writepos >= POOLSIZE ) {
307             if( source > 1 )
308                 pool_filled = 1;
309             pool_writepos = 0;
310             mix_pool(rndpool);
311             just_mixed = !length;
312         }
313     }
314 }
315
316
317
318 static void
319 random_poll()
320 {
321     read_random_source( 2, POOLSIZE/5, 1 );
322 }
323
324
325 void
326 fast_random_poll()
327 {
328     static void (*fnc)( void (*)(const void*, size_t, int), int) = NULL;
329     static int initialized = 0;
330
331     if( !initialized ) {
332         if( !is_initialized )
333             initialize();
334         initialized = 1;
335         fnc = dynload_getfnc_fast_random_poll();
336     }
337     if( fnc ) {
338         (*fnc)( add_randomness, 1 );
339         return;
340     }
341
342     /* fall back to the generic function */
343   #if HAVE_GETHRTIME
344     {   hrtime_t tv;
345         tv = gethrtime();
346         add_randomness( &tv, sizeof(tv), 1 );
347     }
348   #elif HAVE_GETTIMEOFDAY
349     {   struct timeval tv;
350         if( gettimeofday( &tv, NULL ) )
351             BUG();
352         add_randomness( &tv.tv_sec, sizeof(tv.tv_sec), 1 );
353         add_randomness( &tv.tv_usec, sizeof(tv.tv_usec), 1 );
354     }
355   #else /* use times */
356     {   struct tms buf;
357         times( &buf );
358         add_randomness( &buf, sizeof buf, 1 );
359     }
360   #endif
361   #ifdef HAVE_GETRUSAGE
362     {   struct rusage buf;
363         if( getrusage( RUSAGE_SELF, &buf ) )
364             BUG();
365         add_randomness( &buf, sizeof buf, 1 );
366         memset( &buf, 0, sizeof buf );
367     }
368   #endif
369 }
370
371
372
373 static void
374 read_random_source( int requester, size_t length, int level )
375 {
376     static int (*fnc)(void (*)(const void*, size_t, int), int,
377                                                     size_t, int) = NULL;
378     int nbytes;
379     int goodness;
380
381     if( !fnc ) {
382         if( !is_initialized )
383             initialize();
384         fnc = dynload_getfnc_gather_random();
385         if( !fnc ) {
386             faked_rng = 1;
387             fnc = gather_faked;
388         }
389         if( !requester && !length && !level )
390             return; /* init only */
391     }
392     if( (*fnc)( add_randomness, requester, length, level ) < 0 )
393         log_fatal("No way to gather entropy for the RNG\n");
394 }
395
396
397 static int
398 gather_faked( void (*add)(const void*, size_t, int), int requester,
399               size_t length, int level )
400 {
401     static int initialized=0;
402     size_t n;
403     char *buffer, *p;
404
405     if( !initialized ) {
406         log_info(_("WARNING: using insecure random number generator!!\n"));
407         tty_printf(_("The random number generator is only a kludge to let\n"
408                    "it run - it is in no way a strong RNG!\n\n"
409                    "DON'T USE ANY DATA GENERATED BY THIS PROGRAM!!\n\n"));
410         initialized=1;
411       #ifdef HAVE_RAND
412         srand(make_timestamp()*getpid());
413       #else
414         srandom(make_timestamp()*getpid());
415       #endif
416     }
417
418     p = buffer = m_alloc( length );
419     n = length;
420   #ifdef HAVE_RAND
421     while( n-- )
422         *p++ = ((unsigned)(1 + (int) (256.0*rand()/(RAND_MAX+1.0)))-1);
423   #else
424     while( n-- )
425         *p++ = ((unsigned)(1 + (int) (256.0*random()/(RAND_MAX+1.0)))-1);
426   #endif
427     add_randomness( buffer, length, requester );
428     m_free(buffer);
429     return 0; /* okay */
430 }
431