added option file handling
[libgcrypt.git] / cipher / random.c
1 /* random.c  -  random number generator
2  *      Copyright (c) 1997 by Werner Koch (dd9jn)
3  *
4  * This file is part of G10.
5  *
6  * G10 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  * G10 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 #include <config.h>
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include <errno.h>
25 #include <sys/types.h>
26 #include <sys/stat.h>
27 #include <fcntl.h>
28 #include "util.h"
29 #include "cipher.h"
30
31 struct cache {
32     int len;
33     byte buffer[100]; /* fixme: should be allocalted with m_alloc_secure()*/
34 };
35
36 static struct cache cache[3];
37 #define MASK_LEVEL(a) do {if( a > 2 ) a = 2; else if( a < 0 ) a = 0; } while(0)
38
39
40 static int open_device( const char *name, int minor );
41 static void fill_buffer( byte *buffer, size_t length, int level );
42
43 /****************
44  * Fill the buffer with LENGTH bytes of cryptologic strong
45  * random bytes. level 0 is not very strong, 1 is strong enough
46  * for most usage, 2 is good for key generation stuff but may be very slow.
47  */
48 void
49 randomize_buffer( byte *buffer, size_t length, int level )
50 {
51     for( ; length; length-- )
52         *buffer++ = get_random_byte(level);
53 }
54
55
56 byte
57 get_random_byte( int level )
58 {
59     MASK_LEVEL(level);
60     if( !cache[level].len ) {
61         fill_buffer(cache[level].buffer, DIM(cache[level].buffer), level );
62         cache[level].len = DIM(cache[level].buffer);
63     }
64
65     return cache[level].buffer[--cache[level].len];
66 }
67
68
69
70
71 static int
72 open_device( const char *name, int minor )
73 {
74     int fd;
75     struct stat sb;
76
77     fd = open( name, O_RDONLY );
78     if( fd == -1 )
79         log_fatal("can't open %s: %s\n", name, strerror(errno) );
80     if( fstat( fd, &sb ) )
81         log_fatal("stat() off %s failed: %s\n", name, strerror(errno) );
82     if( !S_ISCHR(sb.st_mode)
83       #ifdef __linux__
84         || (sb.st_rdev >> 8) != 1
85         || (sb.st_rdev & 0xff) != minor
86       #endif
87       )
88         log_fatal("invalid random device!\n" );
89     return fd;
90 }
91
92
93 static void
94 fill_buffer( byte *buffer, size_t length, int level )
95 {
96     FILE *fp;
97     static int fd_urandom = -1;
98     static int fd_random = -1;
99     int fd;
100     int n;
101
102     if( level == 2 ) {
103         if( fd_random == -1 )
104             fd_random = open_device( "/dev/random", 8 );
105         fd = fd_random;
106     }
107     else {
108         if( fd_urandom == -1 )
109             fd_urandom = open_device( "/dev/urandom", 9 );
110         fd = fd_urandom;
111     }
112
113
114     do {
115         do {
116             n = read(fd, buffer, length );
117         } while( n == -1 && errno == EINTR );
118         if( n == -1 )
119             log_fatal("read error on random device: %s\n", strerror(errno) );
120         buffer += n;
121         length -= n;
122     } while( length );
123 }
124