A lot of cleanups as well as minor API changes.
[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 );
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 static int
46 set_cloexec_flag (int fd)
47 {
48   int oldflags;
49
50   oldflags= fcntl (fd, F_GETFD, 0);
51   if (oldflags < 0)
52     return oldflags;
53   oldflags |= FD_CLOEXEC;
54   return fcntl (fd, F_SETFD, oldflags);
55 }
56
57
58
59 /*
60  * Used to open the /dev/random devices (Linux, xBSD, Solaris (if it exists)).
61  */
62 static int
63 open_device ( const char *name )
64 {
65   int fd;
66
67   fd = open ( name, O_RDONLY );
68   if ( fd == -1 )
69     log_fatal ("can't open %s: %s\n", name, strerror(errno) );
70
71   if (set_cloexec_flag (fd))
72     log_error ("error setting FD_CLOEXEC on fd %d: %s\n",
73                fd, strerror (errno));
74
75   /* We used to do the following check, however it turned out that this
76      is not portable since more OSes provide a random device which is
77      sometimes implemented as another device type. 
78      
79      struct stat sb;
80
81      if( fstat( fd, &sb ) )
82         log_fatal("stat() off %s failed: %s\n", name, strerror(errno) );
83      if( (!S_ISCHR(sb.st_mode)) && (!S_ISFIFO(sb.st_mode)) )
84         log_fatal("invalid random device!\n" );
85   */
86   return fd;
87 }
88
89
90 int
91 _gcry_rndlinux_gather_random (void (*add)(const void*, size_t, int),
92                               int requester,
93                               size_t length, int level )
94 {
95   static int fd_urandom = -1;
96   static int fd_random = -1;
97   int fd;
98   int n;
99   int warn=0;
100   byte buffer[768];
101
102   if( level >= 2 )
103     {
104       if( fd_random == -1 )
105         fd_random = open_device ( NAME_OF_DEV_RANDOM );
106       fd = fd_random;
107     }
108   else
109     {
110       if( fd_urandom == -1 )
111         fd_urandom = open_device ( NAME_OF_DEV_URANDOM );
112       fd = fd_urandom;
113     }
114
115   while (length)
116     {
117       fd_set rfds;
118       struct timeval tv;
119       int rc;
120       
121       FD_ZERO(&rfds);
122       FD_SET(fd, &rfds);
123       tv.tv_sec = 3;
124       tv.tv_usec = 0;
125       if( !(rc=select(fd+1, &rfds, NULL, NULL, &tv)) )
126         {
127           if( !warn )
128             {
129               _gcry_random_progress ("need_entropy", 'X', 0, (int)length);
130               warn = 1;
131             }
132           continue;
133         }
134         else if( rc == -1 )
135           {
136             log_error ("select() error: %s\n", strerror(errno));
137             continue;
138           }
139
140         do 
141           {
142             int nbytes = length < sizeof(buffer)? length : sizeof(buffer);
143             n = read(fd, buffer, nbytes );
144             if( n >= 0 && n > nbytes ) 
145               {
146                 log_error("bogus read from random device (n=%d)\n", n );
147                 n = nbytes;
148               }
149           } 
150         while( n == -1 && errno == EINTR );
151         if( n == -1 )
152           log_fatal("read error on random device: %s\n", strerror(errno));
153         (*add)( buffer, n, requester );
154         length -= n;
155     }
156   memset(buffer, 0, sizeof(buffer) );
157
158   return 0; /* success */
159 }