See ChangeLog: Fri Jul 23 13:53:03 CEST 1999 Werner Koch
[libgcrypt.git] / cipher / rndlinux.c
1 /* rndlinux.c  -  raw random number for OSes with /dev/random
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 #include <config.h>
23 #include <stdio.h>
24 #include <stdlib.h>
25 #include <assert.h>
26 #include <errno.h>
27 #include <sys/time.h>
28 #include <sys/types.h>
29 #include <sys/stat.h>
30 #ifdef HAVE_GETTIMEOFDAY
31   #include <sys/times.h>
32 #endif
33 #include <string.h>
34 #include <unistd.h>
35 #include <fcntl.h>
36 #if 0
37   #ifdef HAVE_LINUX_RANDOM_H
38     #include <sys/ioctl.h>
39     #include <asm/types.h>
40     #include <linux/random.h>
41   #endif
42 #endif
43 #include "types.h"
44 #include "util.h"
45 #include "ttyio.h"
46 #include "dynload.h"
47
48 #ifdef IS_MODULE
49   #define _(a) (a)
50 #else
51   #include "i18n.h"
52 #endif
53
54 static int open_device( const char *name, int minor );
55 static int gather_random( void (*add)(const void*, size_t, int), int requester,
56                                           size_t length, int level );
57
58 #if 0
59 #ifdef HAVE_DEV_RANDOM_IOCTL
60 static ulong
61 get_entropy_count( int fd )
62 {
63     ulong count;
64
65     if( ioctl( fd, RNDGETENTCNT, &count ) == -1 )
66         g10_log_fatal("ioctl(RNDGETENTCNT) failed: %s\n", strerror(errno) );
67     return count;
68 }
69 #endif
70 #endif
71
72 /****************
73  * Used to open the Linux and xBSD /dev/random devices
74  */
75 static int
76 open_device( const char *name, int minor )
77 {
78     int fd;
79     struct stat sb;
80
81     fd = open( name, O_RDONLY );
82     if( fd == -1 )
83         g10_log_fatal("can't open %s: %s\n", name, strerror(errno) );
84     if( fstat( fd, &sb ) )
85         g10_log_fatal("stat() off %s failed: %s\n", name, strerror(errno) );
86     if( !S_ISCHR(sb.st_mode) )
87         g10_log_fatal("invalid random device!\n" );
88     return fd;
89 }
90
91
92 static int
93 gather_random( void (*add)(const void*, size_t, int), int requester,
94                                           size_t length, int level )
95 {
96     static int fd_urandom = -1;
97     static int fd_random = -1;
98     int fd;
99     int n;
100     int warn=0;
101     byte buffer[768];
102
103     if( level >= 2 ) {
104         if( fd_random == -1 )
105             fd_random = open_device( NAME_OF_DEV_RANDOM, 8 );
106         fd = fd_random;
107     }
108     else {
109         if( fd_urandom == -1 )
110             fd_urandom = open_device( NAME_OF_DEV_URANDOM, 9 );
111         fd = fd_urandom;
112     }
113
114   #if 0
115   #ifdef HAVE_DEV_RANDOM_IOCTL
116     g10_log_info("entropy count of %d is %lu\n", fd, get_entropy_count(fd) );
117   #endif
118   #endif
119     while( length ) {
120         fd_set rfds;
121         struct timeval tv;
122         int rc;
123
124         FD_ZERO(&rfds);
125         FD_SET(fd, &rfds);
126         tv.tv_sec = 3;
127         tv.tv_usec = 0;
128         if( !(rc=select(fd+1, &rfds, NULL, NULL, &tv)) ) {
129             if( !warn )
130               #ifdef IS_MODULE
131                 fprintf(stderr,
132               #else
133                 tty_printf(
134               #endif
135 _("\n"
136 "Not enough random bytes available.  Please do some other work to give\n"
137 "the OS a chance to collect more entropy! (Need %d more bytes)\n"), length );
138             warn = 1;
139             continue;
140         }
141         else if( rc == -1 ) {
142           #ifdef IS_MODULE
143             fprintf(stderr,
144           #else
145             tty_printf(
146           #endif
147                        "select() error: %s\n", strerror(errno));
148             continue;
149         }
150
151         do {
152             int nbytes = length < sizeof(buffer)? length : sizeof(buffer);
153             n = read(fd, buffer, nbytes );
154             if( n >= 0 && n > nbytes ) {
155                 g10_log_error("bogus read from random device (n=%d)\n", n );
156                 n = nbytes;
157             }
158         } while( n == -1 && errno == EINTR );
159         if( n == -1 )
160             g10_log_fatal("read error on random device: %s\n", strerror(errno));
161         (*add)( buffer, n, requester );
162         length -= n;
163     }
164     memset(buffer, 0, sizeof(buffer) );
165
166     return 0; /* success */
167 }
168
169
170
171 #ifndef IS_MODULE
172 static
173 #endif
174 const char * const gnupgext_version = "RNDLINUX ($Revision$)";
175
176 static struct {
177     int class;
178     int version;
179     void *func;
180 } func_table[] = {
181     { 40, 1, gather_random },
182 };
183
184
185
186 /****************
187  * Enumerate the names of the functions together with informations about
188  * this function. Set sequence to an integer with a initial value of 0 and
189  * do not change it.
190  * If what is 0 all kind of functions are returned.
191  * Return values: class := class of function:
192  *                         10 = message digest algorithm info function
193  *                         11 = integer with available md algorithms
194  *                         20 = cipher algorithm info function
195  *                         21 = integer with available cipher algorithms
196  *                         30 = public key algorithm info function
197  *                         31 = integer with available pubkey algorithms
198  *                         40 = get gather_random function
199  *                         41 = get fast_random_poll function
200  *                version = interface version of the function/pointer
201  *                          (currently this is 1 for all functions)
202  */
203
204 #ifndef IS_MODULE
205 static
206 #endif
207 void *
208 gnupgext_enum_func( int what, int *sequence, int *class, int *vers )
209 {
210     void *ret;
211     int i = *sequence;
212
213     do {
214         if ( i >= DIM(func_table) || i < 0 ) {
215             return NULL;
216         }
217         *class = func_table[i].class;
218         *vers  = func_table[i].version;
219         ret = func_table[i].func;
220         i++;
221     } while ( what && what != *class );
222
223     *sequence = i;
224     return ret;
225 }
226
227 #ifndef IS_MODULE
228 void
229 rndlinux_constructor(void)
230 {
231     register_internal_cipher_extension( gnupgext_version,
232                                         gnupgext_enum_func );
233 }
234 #endif
235