The big extension module removal.
[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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
19  */
20
21 #include <config.h>
22
23 #ifdef USE_RNDEG
24
25 #include <stdio.h>
26 #include <stdlib.h>
27 #include <assert.h>
28 #include <errno.h>
29 #include <sys/time.h>
30 #include <sys/stat.h>
31 #include <string.h>
32 #include <unistd.h>
33 #include <sys/types.h>
34 #include <sys/socket.h>
35 #include <sys/un.h>
36 #include "types.h"
37 #include "util.h"
38 #include "ttyio.h"
39 #include "algorithms.h"
40 #include "cipher.h"
41 #include "i18n.h"
42
43
44 #ifndef offsetof
45 #define offsetof(type, member) ((size_t) &((type *)0)->member)
46 #endif
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     do {
73         do {
74             n = read(fd, (char*)buf + nread, nbytes );
75         } while( n == -1 && errno == EINTR );
76         if( n == -1 )
77             return -1;
78         else if( n == 0 ) {
79             /* EGD probably died. */
80             errno = ECONNRESET;
81             return -1;
82         }
83         nread += n;
84     } while( nread < nbytes );
85     return nbytes;
86 }
87
88
89
90 /****************
91  * Note: we always use the highest level.
92  * TO boost the performance we may want to add some
93  * additional code for level 1
94  *
95  * Using a level of 0 should never block and better add nothing
96  * to the pool.  So this is just a dummy for EGD.
97  */
98 int
99 rndegd_gather_random( void (*add)(const void*, size_t, int), int requester,
100                                           size_t length, int level )
101 {
102     static int fd = -1;
103     int n;
104     byte buffer[256+2];
105     int nbytes;
106     int do_restart = 0;
107
108     if( !length )
109         return 0;
110     if( !level )
111         return 0;
112
113   restart:
114     if( do_restart ) {
115         if( fd != -1 ) {
116             close( fd );
117             fd = -1;
118         }
119     }
120     if( fd == -1 ) {
121         const char *bname = NULL;
122         char *name;
123         struct sockaddr_un addr;
124         int addr_len;
125       
126       #ifdef EGD_SOCKET_NAME
127         bname = EGD_SOCKET_NAME;
128       #endif
129         if ( !bname || !*bname )
130             bname = "=entropy";
131
132         if ( *bname == '=' && bname[1] )
133             name = make_filename( g10_opt_homedir, bname+1 , NULL );
134         else
135             name = make_filename( bname , NULL );
136
137         if ( strlen(name)+1 >= sizeof addr.sun_path ) 
138             g10_log_fatal ("EGD socketname is too long\n");
139
140         memset( &addr, 0, sizeof addr );
141         addr.sun_family = AF_UNIX;
142         strcpy( addr.sun_path, name );    
143         addr_len = offsetof( struct sockaddr_un, sun_path )
144                    + strlen( addr.sun_path );
145
146         fd = socket(AF_UNIX, SOCK_STREAM, 0);
147         if( fd == -1 )
148             g10_log_fatal("can't create unix domain socket: %s\n",
149                                                             strerror(errno) );
150         if( connect( fd, (struct sockaddr*)&addr, addr_len) == -1 )
151             g10_log_fatal("can't connect to `%s': %s\n",
152                                                     name, strerror(errno) );
153         m_free(name);
154     }
155     do_restart = 0;
156
157     nbytes = length < 255? length : 255;
158     /* first time we do it with a non blocking request */
159     buffer[0] = 1; /* non blocking */
160     buffer[1] = nbytes;
161     if( do_write( fd, buffer, 2 ) == -1 )
162         g10_log_fatal("can't write to the EGD: %s\n", strerror(errno) );
163     n = do_read( fd, buffer, 1 );
164     if( n == -1 ) {
165         g10_log_error("read error on EGD: %s\n", strerror(errno));
166         do_restart = 1;
167         goto restart;
168     }
169     n = buffer[0];
170     if( n ) {
171         n = do_read( fd, buffer, n );
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
181     if( length ) {
182         tty_printf(
183          _("Please wait, entropy is being gathered. Do some work if it would\n"
184            "keep you from getting bored, because it will improve the quality\n"
185            "of the entropy.\n") );
186     }
187     while( length ) {
188         nbytes = length < 255? length : 255;
189
190         buffer[0] = 2; /* blocking */
191         buffer[1] = nbytes;
192         if( do_write( fd, buffer, 2 ) == -1 )
193             g10_log_fatal("can't write to the EGD: %s\n", strerror(errno) );
194         n = do_read( fd, buffer, nbytes );
195         if( n == -1 ) {
196             g10_log_error("read error on EGD: %s\n", strerror(errno));
197             do_restart = 1;
198             goto restart;
199         }
200         (*add)( buffer, n, requester );
201         length -= n;
202     }
203     memset(buffer, 0, sizeof(buffer) );
204
205     return 0; /* success */
206 }
207
208 #endif /*USE_RNDEGD*/