See ChangeLog: Mon Jul 17 16:35:47 CEST 2000 Werner Koch
[gnupg.git] / cipher / rndegd.c
1 /* rndegd.c  -  interface to the EGD
2  *      Copyright (C) 1999, 2000 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 #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 #ifndef IS_MODULE
36 #include "ttyio.h"
37 #endif
38 #include "dynload.h"
39 #include "cipher.h"
40
41 #ifndef offsetof
42 #define offsetof(type, member) ((size_t) &((type *)0)->member)
43 #endif
44
45
46 /* FIXME: this is duplicated code from util/fileutil
47  * I don't think that this code should go into libgcrypt anyway.
48  */
49 char *
50 my_make_filename( const char *first_part, ... )
51 {
52     va_list arg_ptr ;
53     size_t n;
54     const char *s;
55     char *name, *home, *p;
56
57     va_start( arg_ptr, first_part ) ;
58     n = strlen(first_part)+1;
59     while( (s=va_arg(arg_ptr, const char *)) )
60         n += strlen(s) + 1;
61     va_end(arg_ptr);
62
63     home = NULL;
64     if( *first_part == '~' && first_part[1] == '/'
65                            && (home = getenv("HOME")) && *home )
66         n += strlen(home);
67
68     name = m_alloc(n);
69     p = home ? stpcpy(stpcpy(name,home), first_part+1)
70              : stpcpy(name, first_part);
71     va_start( arg_ptr, first_part ) ;
72     while( (s=va_arg(arg_ptr, const char *)) )
73         p = stpcpy(stpcpy(p,"/"), s);
74     va_end(arg_ptr);
75
76     return name;
77 }
78
79
80
81
82
83 static int
84 do_write( int fd, void *buf, size_t nbytes )
85 {
86     size_t nleft = nbytes;
87     int nwritten;
88
89     while( nleft > 0 ) {
90         nwritten = write( fd, buf, nleft);
91         if( nwritten < 0 ) {
92             if( errno == EINTR )
93                 continue;
94             return -1;
95         }
96         nleft -= nwritten;
97         buf = (char*)buf + nwritten;
98     }
99     return 0;
100 }
101
102 static int
103 do_read( int fd, void *buf, size_t nbytes )
104 {
105     int n, nread = 0;
106
107     do {
108         do {
109             n = read(fd, (char*)buf + nread, nbytes );
110         } while( n == -1 && errno == EINTR );
111         if( n == -1 )
112             return -1;
113         nread += n;
114     } while( nread < nbytes );
115     return nbytes;
116 }
117
118
119
120 /****************
121  * Note: we always use the highest level.
122  * TO boost the performance we may want to add some
123  * additional code for level 1
124  *
125  * Using a level of 0 should never block and better add nothing
126  * to the pool.  So this is just a dummy for EGD.
127  */
128 static int
129 gather_random( void (*add)(const void*, size_t, int), int requester,
130                                           size_t length, int level )
131 {
132     static int fd = -1;
133     int n;
134     byte buffer[256+2];
135     int nbytes;
136     int do_restart = 0;
137
138     if( !length )
139         return 0;
140     if( !level )
141         return 0;
142
143   restart:
144     if( do_restart ) {
145         if( fd != -1 ) {
146             close( fd );
147             fd = -1;
148         }
149     }
150     if( fd == -1 ) {
151         #warning Fixme: make the filename configurable
152         char *name = my_make_filename( "~/.gnupg-test", "entropy", NULL );
153         struct sockaddr_un addr;
154         int addr_len;
155
156         memset( &addr, 0, sizeof addr );
157         addr.sun_family = AF_UNIX;
158         strcpy( addr.sun_path, name );    /* fixme: check that it is long enough */
159         addr_len = offsetof( struct sockaddr_un, sun_path )
160                    + strlen( addr.sun_path );
161
162         fd = socket(AF_UNIX, SOCK_STREAM, 0);
163         if( fd == -1 )
164             g10_log_fatal("can't create unix domain socket: %s\n",
165                                                             strerror(errno) );
166         if( connect( fd, (struct sockaddr*)&addr, addr_len) == -1 )
167             g10_log_fatal("can't connect to `%s': %s\n",
168                                                     name, strerror(errno) );
169         g10_free(name);
170     }
171     do_restart = 0;
172
173     nbytes = length < 255? length : 255;
174     /* first time we do it with a non blocking request */
175     buffer[0] = 1; /* non blocking */
176     buffer[1] = nbytes;
177     if( do_write( fd, buffer, 2 ) == -1 )
178         g10_log_fatal("can't write to the EGD: %s\n", strerror(errno) );
179     n = do_read( fd, buffer, 1 );
180     if( n == -1 ) {
181         g10_log_error("read error on EGD: %s\n", strerror(errno));
182         do_restart = 1;
183         goto restart;
184     }
185     n = buffer[0];
186     if( n ) {
187         n = do_read( fd, buffer, n );
188         if( n == -1 ) {
189             g10_log_error("read error on EGD: %s\n", strerror(errno));
190             do_restart = 1;
191             goto restart;
192         }
193         (*add)( buffer, n, requester );
194         length -= n;
195     }
196
197     if( length ) {
198       #ifdef IS_MODULE
199         fprintf( stderr,
200       #else
201         tty_printf(
202       #endif
203          _("Please wait, entropy is being gathered. Do some work if it would\n"
204            "keep you from getting bored, because it will improve the quality\n"
205            "of the entropy.\n") );
206     }
207     while( length ) {
208         nbytes = length < 255? length : 255;
209
210         buffer[0] = 2; /* blocking */
211         buffer[1] = nbytes;
212         if( do_write( fd, buffer, 2 ) == -1 )
213             g10_log_fatal("can't write to the EGD: %s\n", strerror(errno) );
214         n = do_read( fd, buffer, nbytes );
215         if( n == -1 ) {
216             g10_log_error("read error on EGD: %s\n", strerror(errno));
217             do_restart = 1;
218             goto restart;
219         }
220         (*add)( buffer, n, requester );
221         length -= n;
222     }
223     memset(buffer, 0, sizeof(buffer) );
224
225     return 0; /* success */
226 }
227
228
229
230 #ifndef IS_MODULE
231 static
232 #endif
233 const char * const gnupgext_version = "RNDEGD ($Revision$)";
234
235 static struct {
236     int class;
237     int version;
238     void *func;
239 } func_table[] = {
240     { 40, 1, gather_random },
241 };
242
243
244 #ifndef IS_MODULE
245 static
246 #endif
247 void *
248 gnupgext_enum_func( int what, int *sequence, int *class, int *vers )
249 {
250     void *ret;
251     int i = *sequence;
252
253     do {
254         if ( i >= DIM(func_table) || i < 0 ) {
255             return NULL;
256         }
257         *class = func_table[i].class;
258         *vers  = func_table[i].version;
259         ret = func_table[i].func;
260         i++;
261     } while ( what && what != *class );
262
263     *sequence = i;
264     return ret;
265 }
266
267 #ifndef IS_MODULE
268 void
269 rndegd_constructor(void)
270 {
271     register_internal_cipher_extension( gnupgext_version,
272                                         gnupgext_enum_func );
273 }
274 #endif
275