2004-08-23 Moritz Schulte <moritz@g10code.com>
[libgcrypt.git] / cipher / rndlinux.c
1 /* rndlinux.c  -  raw random number for OSes with /dev/random
2  * Copyright (C) 1998, 2001, 2002, 2003  Free Software Foundation, Inc.
3  *
4  * This file is part of Libgcrypt.
5  *
6  * Libgcrypt is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU Lesser General Public License as
8  * published by the Free Software Foundation; either version 2.1 of
9  * the License, or (at your option) any later version.
10  *
11  * Libgcrypt 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 Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License 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 <errno.h>
26 #include <sys/time.h>
27 #include <sys/types.h>
28 #include <sys/stat.h>
29 #ifdef HAVE_GETTIMEOFDAY
30 # include <sys/times.h>
31 #endif
32 #include <string.h>
33 #include <unistd.h>
34 #include <fcntl.h>
35 #include "types.h"
36 #include "g10lib.h"
37 #include "rand-internal.h"
38
39 static int open_device( const char *name, int minor );
40 int _gcry_rndlinux_gather_random (void (*add)(const void*, size_t, int),
41                                   int requester,
42                                   size_t length, int level );
43
44 /*
45  * Used to open the /dev/random devices (Linux, xBSD, Solaris (if it exists)).
46  */
47 static int
48 open_device( const char *name, int minor )
49 {
50   int fd;
51
52   fd = open( name, O_RDONLY );
53   if( fd == -1 )
54     log_fatal ("can't open %s: %s\n", name, strerror(errno) );
55
56   /* We used to do the follwing check, however it turned out that this
57      is not portable since more OSes provide a random device which is
58      sometimes implemented as anoteher device type. 
59      
60      struct stat sb;
61
62      if( fstat( fd, &sb ) )
63         log_fatal("stat() off %s failed: %s\n", name, strerror(errno) );
64      if( (!S_ISCHR(sb.st_mode)) && (!S_ISFIFO(sb.st_mode)) )
65         log_fatal("invalid random device!\n" );
66   */
67   return fd;
68 }
69
70
71 int
72 _gcry_rndlinux_gather_random (void (*add)(const void*, size_t, int),
73                               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     {
85       if( fd_random == -1 )
86         fd_random = open_device( NAME_OF_DEV_RANDOM, 8 );
87       fd = fd_random;
88     }
89   else
90     {
91       if( fd_urandom == -1 )
92         fd_urandom = open_device( NAME_OF_DEV_URANDOM, 9 );
93       fd = fd_urandom;
94     }
95
96   while (length)
97     {
98       fd_set rfds;
99       struct timeval tv;
100       int rc;
101       
102       FD_ZERO(&rfds);
103       FD_SET(fd, &rfds);
104       tv.tv_sec = 3;
105       tv.tv_usec = 0;
106       if( !(rc=select(fd+1, &rfds, NULL, NULL, &tv)) )
107         {
108           if( !warn )
109             {
110               _gcry_random_progress ("need_entropy", 'X', 0, (int)length);
111               warn = 1;
112             }
113           continue;
114         }
115         else if( rc == -1 )
116           {
117             log_error ("select() error: %s\n", strerror(errno));
118             continue;
119           }
120
121         do 
122           {
123             int nbytes = length < sizeof(buffer)? length : sizeof(buffer);
124             n = read(fd, buffer, nbytes );
125             if( n >= 0 && n > nbytes ) 
126               {
127                 log_error("bogus read from random device (n=%d)\n", n );
128                 n = nbytes;
129               }
130           } 
131         while( n == -1 && errno == EINTR );
132         if( n == -1 )
133           log_fatal("read error on random device: %s\n", strerror(errno));
134         (*add)( buffer, n, requester );
135         length -= n;
136     }
137   memset(buffer, 0, sizeof(buffer) );
138
139   return 0; /* success */
140 }