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