Improve.
[libgcrypt.git] / cipher / rndegd.c
1 /* rndegd.c  -  interface to the EGD
2  *      Copyright (C) 1999, 2000, 2002 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 #include <config.h>
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include <assert.h>
25 #include <errno.h>
26 #include <sys/time.h>
27 #include <sys/stat.h>
28 #include <string.h>
29 #include <unistd.h>
30 #include <sys/types.h>
31 #include <sys/socket.h>
32 #include <sys/un.h>
33 #include "types.h"
34 #include "g10lib.h"
35 #include "cipher.h"
36
37 #ifndef offsetof
38 #define offsetof(type, member) ((size_t) &((type *)0)->member)
39 #endif
40
41 static int egd_socket = -1;
42
43 /* FIXME: this is duplicated code from util/fileutil
44  * I don't think that this code should go into libgcrypt anyway.
45  */
46 char *
47 my_make_filename( const char *first_part, ... )
48 {
49     va_list arg_ptr ;
50     size_t n;
51     const char *s;
52     char *name, *home, *p;
53
54     va_start( arg_ptr, first_part ) ;
55     n = strlen(first_part)+1;
56     while( (s=va_arg(arg_ptr, const char *)) )
57         n += strlen(s) + 1;
58     va_end(arg_ptr);
59
60     home = NULL;
61     if( *first_part == '~' && first_part[1] == '/'
62                            && (home = getenv("HOME")) && *home )
63         n += strlen(home);
64
65     name = gcry_malloc(n);
66     p = home ? stpcpy(stpcpy(name,home), first_part+1)
67              : stpcpy(name, first_part);
68     va_start( arg_ptr, first_part ) ;
69     while( (s=va_arg(arg_ptr, const char *)) )
70         p = stpcpy(stpcpy(p,"/"), s);
71     va_end(arg_ptr);
72
73     return name;
74 }
75
76
77 static int
78 do_write( int fd, void *buf, size_t nbytes )
79 {
80     size_t nleft = nbytes;
81     int nwritten;
82
83     while( nleft > 0 ) {
84         nwritten = write( fd, buf, nleft);
85         if( nwritten < 0 ) {
86             if( errno == EINTR )
87                 continue;
88             return -1;
89         }
90         nleft -= nwritten;
91         buf = (char*)buf + nwritten;
92     }
93     return 0;
94 }
95
96 static int
97 do_read( int fd, void *buf, size_t nbytes )
98 {
99     int n, nread = 0;
100
101     do {
102         do {
103             n = read(fd, (char*)buf + nread, nbytes );
104         } while( n == -1 && errno == EINTR );
105         if( n == -1)
106             return nread? nread:-1;
107         if( n == 0)
108             return -1;
109         nread += n;
110         nbytes -= n;
111     } while( nread < nbytes );
112     return nbytes;
113 }
114
115
116 /* Connect to the EGD and return the file descriptor.  Return -1 on
117    error.  With NOFAIL set to true, silently fail and return the
118    error, otherwise print an error message and die. */
119 int
120 rndegd_connect_socket (int nofail)
121 {
122   int fd;
123   const char *bname = NULL;
124   char *name;
125   struct sockaddr_un addr;
126   int addr_len;
127
128   if (egd_socket != -1)
129     {
130       close (egd_socket);
131       egd_socket = -1;
132     }
133
134 #ifdef EGD_SOCKET_NAME
135   bname = EGD_SOCKET_NAME;
136 #endif
137   if ( !bname || !*bname )
138     name = my_make_filename ("~/.gnupg", "entropy", NULL); /* FIXME?  */
139   else
140     name = my_make_filename (bname, NULL);
141
142   if (strlen(name)+1 >= sizeof addr.sun_path)
143     log_fatal ("EGD socketname is too long\n");
144   
145   memset( &addr, 0, sizeof addr );
146   addr.sun_family = AF_UNIX;
147   strcpy( addr.sun_path, name );          
148   addr_len = (offsetof( struct sockaddr_un, sun_path )
149               + strlen( addr.sun_path ));
150   
151   fd = socket(AF_UNIX, SOCK_STREAM, 0);
152   if (fd == -1 && !nofail)
153     log_fatal("can't create unix domain socket: %s\n",
154               strerror(errno) );
155   else if (connect (fd, (struct sockaddr*)&addr, addr_len) == -1)
156     {
157       if (!nofail)
158         log_fatal("can't connect to `%s': %s\n",
159                   name, strerror(errno) );
160       close (fd);
161       fd = -1;
162     }
163   gcry_free(name);
164   if (fd != -1)
165     egd_socket = fd;
166   return fd;
167 }
168
169 /****************
170  * Note: we always use the highest level.
171  * TO boost the performance we may want to add some
172  * additional code for level 1
173  *
174  * Using a level of 0 should never block and better add nothing
175  * to the pool.  So this is just a dummy for EGD.
176  */
177 int
178 rndegd_gather_random( void (*add)(const void*, size_t, int), int requester,
179                       size_t length, int level )
180 {
181     int fd = egd_socket;
182     int n;
183     byte buffer[256+2];
184     int nbytes;
185     int do_restart = 0;
186
187     if( !length )
188         return 0;
189     if( !level )
190         return 0;
191
192   restart:
193     if (fd == -1 || do_restart)
194       fd = rndegd_connect_socket (0);
195
196     do_restart = 0;
197
198     nbytes = length < 255? length : 255;
199     /* first time we do it with a non blocking request */
200     buffer[0] = 1; /* non blocking */
201     buffer[1] = nbytes;
202     if( do_write( fd, buffer, 2 ) == -1 )
203         log_fatal("can't write to the EGD: %s\n", strerror(errno) );
204     n = do_read( fd, buffer, 1 );
205     if( n == -1 ) {
206         log_error("read error on EGD: %s\n", strerror(errno));
207         do_restart = 1;
208         goto restart;
209     }
210     n = buffer[0];
211     if( n ) {
212         n = do_read( fd, buffer, n );
213         if( n == -1 ) {
214             log_error("read error on EGD: %s\n", strerror(errno));
215             do_restart = 1;
216             goto restart;
217         }
218         (*add)( buffer, n, requester );
219         length -= n;
220     }
221
222     if( length ) {
223         log_info (
224          _("Please wait, entropy is being gathered. Do some work if it would\n"
225            "keep you from getting bored, because it will improve the quality\n"
226            "of the entropy.\n") );
227     }
228     while( length ) {
229         nbytes = length < 255? length : 255;
230
231         buffer[0] = 2; /* blocking */
232         buffer[1] = nbytes;
233         if( do_write( fd, buffer, 2 ) == -1 )
234             log_fatal("can't write to the EGD: %s\n", strerror(errno) );
235         n = do_read( fd, buffer, nbytes );
236         if( n == -1 ) {
237             log_error("read error on EGD: %s\n", strerror(errno));
238             do_restart = 1;
239             goto restart;
240         }
241         (*add)( buffer, n, requester );
242         length -= n;
243     }
244     memset(buffer, 0, sizeof(buffer) );
245
246     return 0; /* success */
247 }