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