* random.c: Fix prototype of the fast random gatherer. Noted by Joe
[gnupg.git] / cipher / rndlinux.c
1 /* rndlinux.c  -  raw random number for OSes with /dev/random
2  *      Copyright (C) 1998, 1999, 2000, 2001 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
19  * USA.
20  */
21
22
23 #include <config.h>
24
25 #ifdef USE_RNDLINUX
26
27 #include <stdio.h>
28 #include <stdlib.h>
29 #include <assert.h>
30 #include <errno.h>
31 #include <sys/time.h>
32 #include <sys/types.h>
33 #include <sys/stat.h>
34 #ifdef HAVE_GETTIMEOFDAY
35 #include <sys/times.h>
36 #endif
37 #include <string.h>
38 #include <unistd.h>
39 #include <fcntl.h>
40 #if 0
41 #include <sys/ioctl.h>
42 #include <asm/types.h>
43 #include <linux/random.h>
44 #endif
45 #include "types.h"
46 #include "util.h"
47 #include "ttyio.h"
48 #include "algorithms.h"
49
50 #include "i18n.h"
51
52 static int open_device( const char *name, int minor );
53
54
55 #if 0
56 #ifdef HAVE_DEV_RANDOM_IOCTL
57 static ulong
58 get_entropy_count( int fd )
59 {
60     ulong count;
61
62     if( ioctl( fd, RNDGETENTCNT, &count ) == -1 )
63         g10_log_fatal("ioctl(RNDGETENTCNT) failed: %s\n", strerror(errno) );
64     return count;
65 }
66 #endif
67 #endif
68
69 /****************
70  * Used to open the /dev/random devices (Linux, xBSD, Solaris (if it exists), ...)
71  */
72 static int
73 open_device( const char *name, int minor )
74 {
75     int fd;
76     struct stat sb;
77
78     fd = open( name, O_RDONLY );
79     if( fd == -1 )
80         g10_log_fatal("can't open %s: %s\n", name, strerror(errno) );
81     if( fstat( fd, &sb ) )
82         g10_log_fatal("stat() off %s failed: %s\n", name, strerror(errno) );
83     /* Don't check device type for better portability */
84     /*  if( (!S_ISCHR(sb.st_mode)) && (!S_ISFIFO(sb.st_mode)) )
85           g10_log_fatal("invalid random device!\n" ); */
86     return fd;
87 }
88
89
90 /****************
91  * Note:  Using a level of 0 should never block and better add nothing
92  * to the pool.  This is easy to accomplish with /dev/urandom.
93  */
94 int
95 rndlinux_gather_random( void (*add)(const void*, size_t, int), int requester,
96                                           size_t length, int level )
97 {
98     static int fd_urandom = -1;
99     static int fd_random = -1;
100     int fd;
101     int n;
102     int warn=0;
103     byte buffer[768];
104
105     if( level >= 2 ) {
106         if( fd_random == -1 )
107             fd_random = open_device( NAME_OF_DEV_RANDOM, 8 );
108         fd = fd_random;
109     }
110     else {
111         /* this will also be used for elve 0 but by using /dev/urandom
112          * we can be sure that oit will never block. */
113         if( fd_urandom == -1 )
114             fd_urandom = open_device( NAME_OF_DEV_URANDOM, 9 );
115         fd = fd_urandom;
116     }
117
118 #if 0
119 #ifdef HAVE_DEV_RANDOM_IOCTL
120     g10_log_info("entropy count of %d is %lu\n", fd, get_entropy_count(fd) );
121 #endif
122 #endif
123     while( length ) {
124         fd_set rfds;
125         struct timeval tv;
126         int rc;
127
128         FD_ZERO(&rfds);
129         FD_SET(fd, &rfds);
130         tv.tv_sec = 3;
131         tv.tv_usec = 0;
132         if( !(rc=select(fd+1, &rfds, NULL, NULL, &tv)) ) {
133             if( !warn )
134                 tty_printf(
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"), (int)length );
138             warn = 1;
139             continue;
140         }
141         else if( rc == -1 ) {
142             tty_printf(
143                        "select() error: %s\n", strerror(errno));
144             continue;
145         }
146
147         do {
148             int nbytes = length < sizeof(buffer)? length : sizeof(buffer);
149             n = read(fd, buffer, nbytes );
150             if( n >= 0 && n > nbytes ) {
151                 g10_log_error("bogus read from random device (n=%d)\n", n );
152                 n = nbytes;
153             }
154         } while( n == -1 && errno == EINTR );
155         if( n == -1 )
156             g10_log_fatal("read error on random device: %s\n", strerror(errno));
157         (*add)( buffer, n, requester );
158         length -= n;
159     }
160     wipememory(buffer, sizeof(buffer) );
161
162     return 0; /* success */
163 }
164
165 #endif /*USE_RNDLINUX*/