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