Rewrite RAND_egd
[openssl.git] / crypto / rand / rand_egd.c
1 /*
2  * Copyright 2000-2017 The OpenSSL Project Authors. All Rights Reserved.
3  *
4  * Licensed under the OpenSSL license (the "License").  You may not use
5  * this file except in compliance with the License.  You can obtain a copy
6  * in the file LICENSE in the source distribution or at
7  * https://www.openssl.org/source/license.html
8  */
9
10 #include <openssl/opensslconf.h>
11 #ifdef OPENSSL_NO_EGD
12 NON_EMPTY_TRANSLATION_UNIT
13 #else
14
15 # include <openssl/crypto.h>
16 # include <openssl/e_os2.h>
17 # include <openssl/rand.h>
18
19 /*
20  * Query an EGD
21  */
22
23 # if defined(OPENSSL_SYS_WIN32) || defined(OPENSSL_SYS_VMS) || defined(OPENSSL_SYS_MSDOS) || defined(OPENSSL_SYS_VXWORKS) || defined(OPENSSL_SYS_VOS) || defined(OPENSSL_SYS_UEFI)
24 int RAND_query_egd_bytes(const char *path, unsigned char *buf, int bytes)
25 {
26     return -1;
27 }
28
29 int RAND_egd(const char *path)
30 {
31     return (-1);
32 }
33
34 int RAND_egd_bytes(const char *path, int bytes)
35 {
36     return (-1);
37 }
38
39 # else
40
41 #  include <openssl/opensslconf.h>
42 #  include OPENSSL_UNISTD
43 #  include <stddef.h>
44 #  include <sys/types.h>
45 #  include <sys/socket.h>
46 #  ifndef NO_SYS_UN_H
47 #   ifdef OPENSSL_SYS_VXWORKS
48 #    include <streams/un.h>
49 #   else
50 #    include <sys/un.h>
51 #   endif
52 #  else
53 struct sockaddr_un {
54     short sun_family;           /* AF_UNIX */
55     char sun_path[108];         /* path name (gag) */
56 };
57 #  endif                         /* NO_SYS_UN_H */
58 #  include <string.h>
59 #  include <errno.h>
60
61 int RAND_query_egd_bytes(const char *path, unsigned char *buf, int bytes)
62 {
63     FILE *fp = NULL;
64     struct sockaddr_un addr;
65     int mybuffer, ret = -1, i, numbytes, fd;
66     unsigned char tempbuf[255];
67
68     if (bytes > (int)sizeof(tempbuf))
69         return -1;
70
71     /* Make socket. */
72     memset(&addr, 0, sizeof(addr));
73     addr.sun_family = AF_UNIX;
74     if (strlen(path) >= sizeof(addr.sun_path))
75         return (-1);
76     strcpy(addr.sun_path, path);
77     i = offsetof(struct sockaddr_un, sun_path) + strlen(path);
78     fd = socket(AF_UNIX, SOCK_STREAM, 0);
79     if (fd == -1 || (fp = fdopen(fd, "r+")) == NULL)
80         return (-1);
81
82     /* Try to connect */
83     for ( ; ; ) {
84         if (connect(fd, (struct sockaddr *)&addr, i) == 0)
85             break;
86 #  ifdef EISCONN
87         if (errno == EISCONN)
88             break;
89 #  endif
90         switch (errno) {
91 #  ifdef EINTR
92         case EINTR:
93 #  endif
94 #  ifdef EAGAIN
95         case EAGAIN:
96 #  endif
97 #  ifdef EINPROGRESS
98         case EINPROGRESS:
99 #  endif
100 #  ifdef EALREADY
101         case EALREADY:
102 #  endif
103             /* No error, try again */
104             break;
105         default:
106             ret = -1;
107             goto err;
108         }
109     }
110
111     /* Make request, see how many bytes we can get back. */
112     tempbuf[0] = 1;
113     tempbuf[1] = bytes;
114     if (fwrite(tempbuf, sizeof(char), 2, fp) != 2 || fflush(fp) == EOF)
115         goto err;
116     if (fread(tempbuf, sizeof(char), 1, fp) != 1 || tempbuf[0] == 0)
117         goto err;
118     numbytes = tempbuf[0];
119
120     /* Which buffer are we using? */
121     mybuffer = buf == NULL;
122     if (mybuffer)
123         buf = tempbuf;
124
125     /* Read bytes. */
126     i = fread(buf, sizeof(char), numbytes, fp);
127     if (i < numbytes)
128         goto err;
129     ret = numbytes;
130     if (mybuffer)
131         RAND_seed(tempbuf, i);
132
133  err:
134     if (fp != NULL)
135         fclose(fp);
136     return ret;
137 }
138
139 int RAND_egd_bytes(const char *path, int bytes)
140 {
141     int num;
142
143     num = RAND_query_egd_bytes(path, NULL, bytes);
144     if (num < 0)
145         return -1;
146     if (RAND_status() != 1)
147         return -1;
148     return num;
149 }
150
151 int RAND_egd(const char *path)
152 {
153     return (RAND_egd_bytes(path, 255));
154 }
155
156 # endif
157
158 #endif