Code cleanups.
[libgcrypt.git] / cipher / rndegd.c
1 /* rndegd.c  -  interface to the EGD
2  *      Copyright (C) 1999, 2000, 2002, 2003 Free Software Foundation, Inc.
3  *
4  * This file is part of Libgcrypt.
5  *
6  * Libgcrypt is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU Lesser General Public License as
8  * published by the Free Software Foundation; either version 2.1 of
9  * the License, or (at your option) any later version.
10  *
11  * Libgcrypt 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 Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License 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 <errno.h>
25 #include <sys/time.h>
26 #include <sys/stat.h>
27 #include <string.h>
28 #include <unistd.h>
29 #include <sys/types.h>
30 #include <sys/socket.h>
31 #include <sys/un.h>
32 #include "types.h"
33 #include "g10lib.h"
34 #include "cipher.h"
35 #include "rand-internal.h"
36
37 #ifndef offsetof
38 #define offsetof(type, member) ((size_t) &((type *)0)->member)
39 #endif
40
41 static int egd_socket = -1;
42
43 /* Allocate a new filename from FIRST_PART and SECOND_PART and to
44    tilde expansion for first_part.  SECOND_PART might be NULL.
45  */
46 static char *
47 my_make_filename (const char *first_part, const char *second_part)
48 {
49   size_t n;
50   char *name, *home, *p;
51
52   n = strlen(first_part)+1;
53   if (second_part)
54     n += strlen (second_part) + 1;
55
56   home = NULL;
57   if( *first_part == '~' && first_part[1] == '/'
58       && (home = getenv("HOME")) && *home )
59     n += strlen(home);
60
61   name = gcry_xmalloc(n);
62   p = (home
63        ? stpcpy (stpcpy (name, home), first_part+1 )
64        : stpcpy (name, first_part) );
65
66   if (second_part)
67     strcpy (stpcpy(p,"/"), second_part);
68
69   return name;
70 }
71
72
73 static int
74 do_write( int fd, void *buf, size_t nbytes )
75 {
76   size_t nleft = nbytes;
77   int nwritten;
78   
79   while( nleft > 0 ) 
80     {
81       nwritten = write( fd, buf, nleft);
82       if( nwritten < 0 )
83         {
84           if( errno == EINTR )
85             continue;
86           return -1;
87         }
88       nleft -= nwritten;
89       buf = (char*)buf + nwritten;
90     }
91   return 0;
92 }
93
94 static int
95 do_read( int fd, void *buf, size_t nbytes )
96 {
97   int n, nread = 0;
98
99   do
100     {
101       do
102         {
103           n = read(fd, (char*)buf + nread, nbytes );
104         } 
105       while( n == -1 && errno == EINTR );
106       if( n == -1)
107         return nread? nread:-1;
108       if( n == 0)
109         return -1;
110       nread += n;
111       nbytes -= n;
112     } 
113   while( nread < nbytes );
114   return nread;
115 }
116
117
118 /* Connect to the EGD and return the file descriptor.  Return -1 on
119    error.  With NOFAIL set to true, silently fail and return the
120    error, otherwise print an error message and die. */
121 int
122 _gcry_rndegd_connect_socket (int nofail)
123 {
124   int fd;
125   const char *bname = NULL;
126   char *name;
127   struct sockaddr_un addr;
128   int addr_len;
129
130   if (egd_socket != -1)
131     {
132       close (egd_socket);
133       egd_socket = -1;
134     }
135
136 #ifdef EGD_SOCKET_NAME
137   bname = EGD_SOCKET_NAME;
138 #endif
139   if ( !bname || !*bname )
140     name = my_make_filename ("~/.gnupg", "entropy");
141   else
142     name = my_make_filename (bname, NULL);
143
144   if (strlen(name)+1 >= sizeof addr.sun_path)
145     log_fatal ("EGD socketname is too long\n");
146   
147   memset( &addr, 0, sizeof addr );
148   addr.sun_family = AF_UNIX;
149   strcpy( addr.sun_path, name );          
150   addr_len = (offsetof( struct sockaddr_un, sun_path )
151               + strlen( addr.sun_path ));
152   
153   fd = socket(AF_UNIX, SOCK_STREAM, 0);
154   if (fd == -1 && !nofail)
155     log_fatal("can't create unix domain socket: %s\n", strerror(errno) );
156   else if (connect (fd, (struct sockaddr*)&addr, addr_len) == -1)
157     {
158       if (!nofail)
159         log_fatal("can't connect to EGD socket `%s': %s\n",
160                   name, strerror(errno) );
161       close (fd);
162       fd = -1;
163     }
164   gcry_free(name);
165   if (fd != -1)
166     egd_socket = fd;
167   return fd;
168 }
169
170 /****************
171  * Note: We always use the highest level.
172  * To boost the performance we may want to add some
173  * additional code for level 1
174  *
175  * Using a level of 0 should never block and better add nothing
176  * to the pool.  So this is just a dummy for EGD.
177  */
178 int
179 _gcry_rndegd_gather_random (void (*add)(const void*, size_t,
180                                         enum random_origins),
181                             enum random_origins origin,
182                             size_t length, int level )
183 {
184   int fd = egd_socket;
185   int n;
186   byte buffer[256+2];
187   int nbytes;
188   int do_restart = 0;
189
190   if( !length )
191     return 0;
192   if( !level )
193     return 0;
194
195  restart:
196   if (fd == -1 || do_restart)
197     fd = _gcry_rndegd_connect_socket (0);
198
199   do_restart = 0;
200
201   nbytes = length < 255? length : 255;
202   /* First time we do it with a non blocking request */
203   buffer[0] = 1; /* non blocking */
204   buffer[1] = nbytes;
205   if( do_write( fd, buffer, 2 ) == -1 )
206     log_fatal("can't write to the EGD: %s\n", strerror(errno) );
207   n = do_read( fd, buffer, 1 );
208   if( n == -1 )
209     {
210       log_error("read error on EGD: %s\n", strerror(errno));
211       do_restart = 1;
212       goto restart;
213     }
214   n = buffer[0];
215   if( n )
216     {
217       n = do_read( fd, buffer, n );
218       if( n == -1 )
219         {
220           log_error("read error on EGD: %s\n", strerror(errno));
221           do_restart = 1;
222           goto restart;
223         }
224       (*add)( buffer, n, origin );
225       length -= n;
226     }
227
228   if( length )
229     {
230       log_info (
231       _("Please wait, entropy is being gathered. Do some work if it would\n"
232         "keep you from getting bored, because it will improve the quality\n"
233         "of the entropy.\n") );
234     }
235   while( length )
236     {
237       nbytes = length < 255? length : 255;
238
239       buffer[0] = 2; /* blocking */
240       buffer[1] = nbytes;
241       if( do_write( fd, buffer, 2 ) == -1 )
242         log_fatal("can't write to the EGD: %s\n", strerror(errno) );
243       n = do_read( fd, buffer, nbytes );
244       if( n == -1 )
245         {
246           log_error("read error on EGD: %s\n", strerror(errno));
247           do_restart = 1;
248           goto restart;
249         }
250       (*add)( buffer, n, origin );
251       length -= n;
252     }
253   memset(buffer, 0, sizeof(buffer) );
254
255   return 0; /* success */
256 }