xref: /freebsd/contrib/unbound/compat/getentropy_osx.c (revision a0ee8cc636cd5c2374ec44ca71226564ea0bca95)
1 /*	$OpenBSD: getentropy_osx.c,v 1.3 2014/07/12 14:48:00 deraadt Exp $	*/
2 
3 /*
4  * Copyright (c) 2014 Theo de Raadt <deraadt@openbsd.org>
5  * Copyright (c) 2014 Bob Beck <beck@obtuse.com>
6  *
7  * Permission to use, copy, modify, and distribute this software for any
8  * purpose with or without fee is hereby granted, provided that the above
9  * copyright notice and this permission notice appear in all copies.
10  *
11  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
12  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
13  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
14  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
15  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
16  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
17  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
18  */
19 #include "config.h"
20 
21 #include <sys/types.h>
22 #include <sys/param.h>
23 #include <sys/ioctl.h>
24 #include <sys/resource.h>
25 #include <sys/syscall.h>
26 #include <sys/sysctl.h>
27 #include <sys/statvfs.h>
28 #include <sys/socket.h>
29 #include <sys/mount.h>
30 #include <sys/mman.h>
31 #include <sys/stat.h>
32 #include <sys/time.h>
33 #include <stdlib.h>
34 #include <stdint.h>
35 #include <stdio.h>
36 #include <termios.h>
37 #include <fcntl.h>
38 #include <signal.h>
39 #include <string.h>
40 #include <errno.h>
41 #include <unistd.h>
42 #include <time.h>
43 #include <mach/mach_time.h>
44 #include <mach/mach_host.h>
45 #include <mach/host_info.h>
46 #include <sys/socketvar.h>
47 #include <sys/vmmeter.h>
48 #include <netinet/in.h>
49 #include <netinet/tcp.h>
50 #include <netinet/udp.h>
51 #include <netinet/ip_var.h>
52 #include <netinet/tcp_var.h>
53 #include <netinet/udp_var.h>
54 #include <CommonCrypto/CommonDigest.h>
55 #define SHA512_Update(a, b, c)	(CC_SHA512_Update((a), (b), (c)))
56 #define SHA512_Init(xxx) (CC_SHA512_Init((xxx)))
57 #define SHA512_Final(xxx, yyy) (CC_SHA512_Final((xxx), (yyy)))
58 #define SHA512_CTX CC_SHA512_CTX
59 #define SHA512_DIGEST_LENGTH CC_SHA512_DIGEST_LENGTH
60 
61 #define REPEAT 5
62 #define min(a, b) (((a) < (b)) ? (a) : (b))
63 
64 #define HX(a, b) \
65 	do { \
66 		if ((a)) \
67 			HD(errno); \
68 		else \
69 			HD(b); \
70 	} while (0)
71 
72 #define HR(x, l) (SHA512_Update(&ctx, (char *)(x), (l)))
73 #define HD(x)	 (SHA512_Update(&ctx, (char *)&(x), sizeof (x)))
74 #define HF(x)    (SHA512_Update(&ctx, (char *)&(x), sizeof (void*)))
75 
76 int	getentropy(void *buf, size_t len);
77 
78 #ifdef CAN_REFERENCE_MAIN
79 extern int main(int, char *argv[]);
80 #endif
81 static int gotdata(char *buf, size_t len);
82 static int getentropy_urandom(void *buf, size_t len);
83 static int getentropy_fallback(void *buf, size_t len);
84 
85 int
86 getentropy(void *buf, size_t len)
87 {
88 	int ret = -1;
89 
90 	if (len > 256) {
91 		errno = EIO;
92 		return -1;
93 	}
94 
95 	/*
96 	 * Try to get entropy with /dev/urandom
97 	 *
98 	 * This can fail if the process is inside a chroot or if file
99 	 * descriptors are exhausted.
100 	 */
101 	ret = getentropy_urandom(buf, len);
102 	if (ret != -1)
103 		return (ret);
104 
105 	/*
106 	 * Entropy collection via /dev/urandom and sysctl have failed.
107 	 *
108 	 * No other API exists for collecting entropy, and we have
109 	 * no failsafe way to get it on OSX that is not sensitive
110 	 * to resource exhaustion.
111 	 *
112 	 * We have very few options:
113 	 *     - Even syslog_r is unsafe to call at this low level, so
114 	 *	 there is no way to alert the user or program.
115 	 *     - Cannot call abort() because some systems have unsafe
116 	 *	 corefiles.
117 	 *     - Could raise(SIGKILL) resulting in silent program termination.
118 	 *     - Return EIO, to hint that arc4random's stir function
119 	 *       should raise(SIGKILL)
120 	 *     - Do the best under the circumstances....
121 	 *
122 	 * This code path exists to bring light to the issue that OSX
123 	 * does not provide a failsafe API for entropy collection.
124 	 *
125 	 * We hope this demonstrates that OSX should consider
126 	 * providing a new failsafe API which works in a chroot or
127 	 * when file descriptors are exhausted.
128 	 */
129 #undef FAIL_INSTEAD_OF_TRYING_FALLBACK
130 #ifdef FAIL_INSTEAD_OF_TRYING_FALLBACK
131 	raise(SIGKILL);
132 #endif
133 	ret = getentropy_fallback(buf, len);
134 	if (ret != -1)
135 		return (ret);
136 
137 	errno = EIO;
138 	return (ret);
139 }
140 
141 /*
142  * Basic sanity checking; wish we could do better.
143  */
144 static int
145 gotdata(char *buf, size_t len)
146 {
147 	char	any_set = 0;
148 	size_t	i;
149 
150 	for (i = 0; i < len; ++i)
151 		any_set |= buf[i];
152 	if (any_set == 0)
153 		return -1;
154 	return 0;
155 }
156 
157 static int
158 getentropy_urandom(void *buf, size_t len)
159 {
160 	struct stat st;
161 	size_t i;
162 	int fd, flags;
163 	int save_errno = errno;
164 
165 start:
166 
167 	flags = O_RDONLY;
168 #ifdef O_NOFOLLOW
169 	flags |= O_NOFOLLOW;
170 #endif
171 #ifdef O_CLOEXEC
172 	flags |= O_CLOEXEC;
173 #endif
174 	fd = open("/dev/urandom", flags, 0);
175 	if (fd == -1) {
176 		if (errno == EINTR)
177 			goto start;
178 		goto nodevrandom;
179 	}
180 #ifndef O_CLOEXEC
181 	fcntl(fd, F_SETFD, fcntl(fd, F_GETFD) | FD_CLOEXEC);
182 #endif
183 
184 	/* Lightly verify that the device node looks sane */
185 	if (fstat(fd, &st) == -1 || !S_ISCHR(st.st_mode)) {
186 		close(fd);
187 		goto nodevrandom;
188 	}
189 	for (i = 0; i < len; ) {
190 		size_t wanted = len - i;
191 		ssize_t ret = read(fd, (char*)buf + i, wanted);
192 
193 		if (ret == -1) {
194 			if (errno == EAGAIN || errno == EINTR)
195 				continue;
196 			close(fd);
197 			goto nodevrandom;
198 		}
199 		i += ret;
200 	}
201 	close(fd);
202 	if (gotdata(buf, len) == 0) {
203 		errno = save_errno;
204 		return 0;		/* satisfied */
205 	}
206 nodevrandom:
207 	errno = EIO;
208 	return -1;
209 }
210 
211 static int tcpmib[] = { CTL_NET, AF_INET, IPPROTO_TCP, TCPCTL_STATS };
212 static int udpmib[] = { CTL_NET, AF_INET, IPPROTO_UDP, UDPCTL_STATS };
213 static int ipmib[] = { CTL_NET, AF_INET, IPPROTO_IP, IPCTL_STATS };
214 static int kmib[] = { CTL_KERN, KERN_USRSTACK };
215 static int hwmib[] = { CTL_HW, HW_USERMEM };
216 
217 static int
218 getentropy_fallback(void *buf, size_t len)
219 {
220 	uint8_t results[SHA512_DIGEST_LENGTH];
221 	int save_errno = errno, e, pgs = getpagesize(), faster = 0, repeat;
222 	static int cnt;
223 	struct timespec ts;
224 	struct timeval tv;
225 	struct rusage ru;
226 	sigset_t sigset;
227 	struct stat st;
228 	SHA512_CTX ctx;
229 	static pid_t lastpid;
230 	pid_t pid;
231 	size_t i, ii, m;
232 	char *p;
233 	struct tcpstat tcpstat;
234 	struct udpstat udpstat;
235 	struct ipstat ipstat;
236 	u_int64_t mach_time;
237 	unsigned int idata;
238 	void *addr;
239 
240 	pid = getpid();
241 	if (lastpid == pid) {
242 		faster = 1;
243 		repeat = 2;
244 	} else {
245 		faster = 0;
246 		lastpid = pid;
247 		repeat = REPEAT;
248 	}
249 	for (i = 0; i < len; ) {
250 		int j;
251 		SHA512_Init(&ctx);
252 		for (j = 0; j < repeat; j++) {
253 			HX((e = gettimeofday(&tv, NULL)) == -1, tv);
254 			if (e != -1) {
255 				cnt += (int)tv.tv_sec;
256 				cnt += (int)tv.tv_usec;
257 			}
258 
259 			mach_time = mach_absolute_time();
260 			HD(mach_time);
261 
262 			ii = sizeof(addr);
263 			HX(sysctl(kmib, sizeof(kmib) / sizeof(kmib[0]),
264 			    &addr, &ii, NULL, 0) == -1, addr);
265 
266 			ii = sizeof(idata);
267 			HX(sysctl(hwmib, sizeof(hwmib) / sizeof(hwmib[0]),
268 			    &idata, &ii, NULL, 0) == -1, idata);
269 
270 			ii = sizeof(tcpstat);
271 			HX(sysctl(tcpmib, sizeof(tcpmib) / sizeof(tcpmib[0]),
272 			    &tcpstat, &ii, NULL, 0) == -1, tcpstat);
273 
274 			ii = sizeof(udpstat);
275 			HX(sysctl(udpmib, sizeof(udpmib) / sizeof(udpmib[0]),
276 			    &udpstat, &ii, NULL, 0) == -1, udpstat);
277 
278 			ii = sizeof(ipstat);
279 			HX(sysctl(ipmib, sizeof(ipmib) / sizeof(ipmib[0]),
280 			    &ipstat, &ii, NULL, 0) == -1, ipstat);
281 
282 			HX((pid = getpid()) == -1, pid);
283 			HX((pid = getsid(pid)) == -1, pid);
284 			HX((pid = getppid()) == -1, pid);
285 			HX((pid = getpgid(0)) == -1, pid);
286 			HX((e = getpriority(0, 0)) == -1, e);
287 
288 			if (!faster) {
289 				ts.tv_sec = 0;
290 				ts.tv_nsec = 1;
291 				(void) nanosleep(&ts, NULL);
292 			}
293 
294 			HX(sigpending(&sigset) == -1, sigset);
295 			HX(sigprocmask(SIG_BLOCK, NULL, &sigset) == -1,
296 			    sigset);
297 
298 #ifdef CAN_REFERENCE_MAIN
299 			HF(main);		/* an addr in program */
300 #endif
301 			HF(getentropy);	/* an addr in this library */
302 			HF(printf);		/* an addr in libc */
303 			p = (char *)&p;
304 			HD(p);		/* an addr on stack */
305 			p = (char *)&errno;
306 			HD(p);		/* the addr of errno */
307 
308 			if (i == 0) {
309 				struct sockaddr_storage ss;
310 				struct statvfs stvfs;
311 				struct termios tios;
312 				struct statfs stfs;
313 				socklen_t ssl;
314 				off_t off;
315 
316 				/*
317 				 * Prime-sized mappings encourage fragmentation;
318 				 * thus exposing some address entropy.
319 				 */
320 				struct mm {
321 					size_t	npg;
322 					void	*p;
323 				} mm[] =	 {
324 					{ 17, MAP_FAILED }, { 3, MAP_FAILED },
325 					{ 11, MAP_FAILED }, { 2, MAP_FAILED },
326 					{ 5, MAP_FAILED }, { 3, MAP_FAILED },
327 					{ 7, MAP_FAILED }, { 1, MAP_FAILED },
328 					{ 57, MAP_FAILED }, { 3, MAP_FAILED },
329 					{ 131, MAP_FAILED }, { 1, MAP_FAILED },
330 				};
331 
332 				for (m = 0; m < sizeof mm/sizeof(mm[0]); m++) {
333 					HX(mm[m].p = mmap(NULL,
334 					    mm[m].npg * pgs,
335 					    PROT_READ|PROT_WRITE,
336 					    MAP_PRIVATE|MAP_ANON, -1,
337 					    (off_t)0), mm[m].p);
338 					if (mm[m].p != MAP_FAILED) {
339 						size_t mo;
340 
341 						/* Touch some memory... */
342 						p = mm[m].p;
343 						mo = cnt %
344 						    (mm[m].npg * pgs - 1);
345 						p[mo] = 1;
346 						cnt += (int)((long)(mm[m].p)
347 						    / pgs);
348 					}
349 
350 					/* Check cnts and times... */
351 					mach_time = mach_absolute_time();
352 					HD(mach_time);
353 					cnt += (int)mach_time;
354 
355 					HX((e = getrusage(RUSAGE_SELF,
356 					    &ru)) == -1, ru);
357 					if (e != -1) {
358 						cnt += (int)ru.ru_utime.tv_sec;
359 						cnt += (int)ru.ru_utime.tv_usec;
360 					}
361 				}
362 
363 				for (m = 0; m < sizeof mm/sizeof(mm[0]); m++) {
364 					if (mm[m].p != MAP_FAILED)
365 						munmap(mm[m].p, mm[m].npg * pgs);
366 					mm[m].p = MAP_FAILED;
367 				}
368 
369 				HX(stat(".", &st) == -1, st);
370 				HX(statvfs(".", &stvfs) == -1, stvfs);
371 				HX(statfs(".", &stfs) == -1, stfs);
372 
373 				HX(stat("/", &st) == -1, st);
374 				HX(statvfs("/", &stvfs) == -1, stvfs);
375 				HX(statfs("/", &stfs) == -1, stfs);
376 
377 				HX((e = fstat(0, &st)) == -1, st);
378 				if (e == -1) {
379 					if (S_ISREG(st.st_mode) ||
380 					    S_ISFIFO(st.st_mode) ||
381 					    S_ISSOCK(st.st_mode)) {
382 						HX(fstatvfs(0, &stvfs) == -1,
383 						    stvfs);
384 						HX(fstatfs(0, &stfs) == -1,
385 						    stfs);
386 						HX((off = lseek(0, (off_t)0,
387 						    SEEK_CUR)) < 0, off);
388 					}
389 					if (S_ISCHR(st.st_mode)) {
390 						HX(tcgetattr(0, &tios) == -1,
391 						    tios);
392 					} else if (S_ISSOCK(st.st_mode)) {
393 						memset(&ss, 0, sizeof ss);
394 						ssl = sizeof(ss);
395 						HX(getpeername(0,
396 						    (void *)&ss, &ssl) == -1,
397 						    ss);
398 					}
399 				}
400 
401 				HX((e = getrusage(RUSAGE_CHILDREN,
402 				    &ru)) == -1, ru);
403 				if (e != -1) {
404 					cnt += (int)ru.ru_utime.tv_sec;
405 					cnt += (int)ru.ru_utime.tv_usec;
406 				}
407 			} else {
408 				/* Subsequent hashes absorb previous result */
409 				HD(results);
410 			}
411 
412 			HX((e = gettimeofday(&tv, NULL)) == -1, tv);
413 			if (e != -1) {
414 				cnt += (int)tv.tv_sec;
415 				cnt += (int)tv.tv_usec;
416 			}
417 
418 			HD(cnt);
419 		}
420 
421 		SHA512_Final(results, &ctx);
422 		memcpy((char*)buf + i, results, min(sizeof(results), len - i));
423 		i += min(sizeof(results), len - i);
424 	}
425 	memset(results, 0, sizeof results);
426 	if (gotdata(buf, len) == 0) {
427 		errno = save_errno;
428 		return 0;		/* satisfied */
429 	}
430 	errno = EIO;
431 	return -1;
432 }
433