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