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