See ChangeLog: Wed Feb 24 11:07:27 CET 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
38 #ifdef IS_MODULE
39   #define _(a) (a)
40 #else
41   #include "i18n.h"
42 #endif
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     ssize_t 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
68 static int
69 gather_random( void (*add)(const void*, size_t, int), int requester,
70                                           size_t length, int level )
71 {
72     static int fd = -1;
73     int n;
74     int warn=0;
75     byte buffer[256+2];
76
77     if( fd == -1 ) {
78         const char *name = "/tmp/entropy";
79         struct sockaddr_un addr;
80         int addr_len;
81
82         memset( &addr, 0, sizeof addr );
83         addr.sun_family = AF_UNIX;
84         strcpy( addr.sun_path, name );    /* fixme: check that it is long enough */
85         addr_len = offsetof( struct sockaddr_un, sun_path )
86                    + strlen( addr.sun_path );
87
88         fd = socket(AF_UNIX, SOCK_STREAM, 0);
89         if( fd == -1 )
90             g10_log_fatal("can't create unix domain socket: %s\n",
91                                                             strerror(errno) );
92         if( connect( fd, (struct sockaddr*)&addr, addr_len) == -1 )
93             g10_log_fatal("can't connect to `%s': %s\n",
94                                                     name, strerror(errno) );
95     }
96
97
98     while( length ) {
99         fd_set rfds;
100         struct timeval tv;
101         int rc;
102         int nbytes;
103         int cmd;
104
105         nbytes = length < 255? length : 255;
106         /* send request */
107         cmd = level >= 2 ? 2 : 1;
108         buffer[0] = cmd;
109         buffer[1] = nbytes;
110         if( do_write( fd, buffer, 2 ) == -1 )
111             g10_log_fatal("can't write to the EGD: %s\n", strerror(errno) );
112         /* wait on reply */
113         FD_ZERO(&rfds);
114         FD_SET(fd, &rfds);
115         tv.tv_sec = 3;
116         tv.tv_usec = 0;
117         if( !(rc=select(fd+1, &rfds, NULL, NULL, &tv)) ) {
118             if( !warn )
119               #ifdef IS_MODULE
120                 fprintf( stderr,
121               #else
122                 tty_printf(
123               #endif
124                             _(
125 "\n"
126 "Not enough random bytes available.  Please do some other work to give\n"
127 "the OS a chance to collect more entropy! (Need %d more bytes)\n"), length );
128             warn = 0; /*  <--- set to 1 to display the message only once */
129             continue;
130         }
131         else if( rc == -1 ) {
132             g10_log_error("select error on EGD: %s\n", strerror(errno));
133             continue;
134         }
135
136         /* collect reply */
137         do {
138             n = read(fd, buffer, nbytes+2 );
139         } while( n == -1 && errno == EINTR );
140         /* process reply */
141         if( n == -1 )
142             g10_log_error("read error on EGD: %s\n", strerror(errno));
143         else if( n < 2 )
144             g10_log_error("bad EGD reply: too short\n");
145         else if( buffer[0] != cmd )
146             g10_log_error("bad EGD reply: cmd mismatch %d/%d\n",
147                                                         cmd, *buffer );
148         else if( buffer[1] != nbytes )
149             g10_log_error("bad EGD reply: count mismatch %d/%d\n",
150                                                       nbytes, buffer[1] );
151         else {
152             n -= 2;
153             (*add)( buffer+2, n, requester );
154             length -= n;
155         }
156     }
157     memset(buffer, 0, sizeof(buffer) );
158
159     return 0; /* success */
160 }
161
162
163
164 #ifndef IS_MODULE
165 static
166 #endif
167 const char * const gnupgext_version = "RNDEGD ($Revision$)";
168
169 static struct {
170     int class;
171     int version;
172     void *func;
173 } func_table[] = {
174     { 40, 1, gather_random },
175 };
176
177
178 #ifndef IS_MODULE
179 static
180 #endif
181 void *
182 gnupgext_enum_func( int what, int *sequence, int *class, int *vers )
183 {
184     void *ret;
185     int i = *sequence;
186
187     do {
188         if ( i >= DIM(func_table) || i < 0 ) {
189             return NULL;
190         }
191         *class = func_table[i].class;
192         *vers  = func_table[i].version;
193         ret = func_table[i].func;
194         i++;
195     } while ( what && what != *class );
196
197     *sequence = i;
198     return ret;
199 }
200
201 #ifndef IS_MODULE
202 void
203 rndegd_constructor(void)
204 {
205     register_internal_cipher_extension( gnupgext_version,
206                                         gnupgext_enum_func );
207 }
208 #endif
209