]> andersk Git - gssapi-openssh.git/blame - openssh/ssh-rand-helper.c
o Formatting changes
[gssapi-openssh.git] / openssh / ssh-rand-helper.c
CommitLineData
e9a17296 1/*
2 * Copyright (c) 2001-2002 Damien Miller. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
6 * are met:
7 * 1. Redistributions of source code must retain the above copyright
8 * notice, this list of conditions and the following disclaimer.
9 * 2. Redistributions in binary form must reproduce the above copyright
10 * notice, this list of conditions and the following disclaimer in the
11 * documentation and/or other materials provided with the distribution.
12 *
13 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
14 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
15 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
16 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
17 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
18 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
19 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
20 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
21 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
22 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
23 */
24
25#include "includes.h"
26
27#include <openssl/rand.h>
28#include <openssl/sha.h>
29#include <openssl/crypto.h>
30
31/* SunOS 4.4.4 needs this */
32#ifdef HAVE_FLOATINGPOINT_H
33# include <floatingpoint.h>
34#endif /* HAVE_FLOATINGPOINT_H */
35
36#include "misc.h"
37#include "xmalloc.h"
38#include "atomicio.h"
39#include "pathnames.h"
40#include "log.h"
41
42RCSID("$Id$");
43
44/* Number of bytes we write out */
45#define OUTPUT_SEED_SIZE 48
46
47/* Length of on-disk seedfiles */
48#define SEED_FILE_SIZE 1024
49
50/* Maximum number of command-line arguments to read from file */
51#define NUM_ARGS 10
52
53/* Minimum number of usable commands to be considered sufficient */
54#define MIN_ENTROPY_SOURCES 16
55
56/* Path to on-disk seed file (relative to user's home directory */
57#ifndef SSH_PRNG_SEED_FILE
58# define SSH_PRNG_SEED_FILE _PATH_SSH_USER_DIR"/prng_seed"
59#endif
60
ae43c103 61/* Path to PRNG commands list (from pathnames.c) */
62extern char *SSH_PRNG_COMMAND_FILE;
e9a17296 63
e9a17296 64#ifdef HAVE___PROGNAME
65extern char *__progname;
66#else
67char *__progname;
68#endif
69
70#ifndef offsetof
71# define offsetof(type, member) ((size_t) &((type *)0)->member)
72#endif
73
74#define WHITESPACE " \t\n"
75
76#ifndef RUSAGE_SELF
77# define RUSAGE_SELF 0
78#endif
79#ifndef RUSAGE_CHILDREN
80# define RUSAGE_CHILDREN 0
81#endif
82
83#if !defined(PRNGD_SOCKET) && !defined(PRNGD_PORT)
84# define USE_SEED_FILES
85#endif
86
87typedef struct {
88 /* Proportion of data that is entropy */
89 double rate;
90 /* Counter goes positive if this command times out */
91 unsigned int badness;
92 /* Increases by factor of two each timeout */
93 unsigned int sticky_badness;
94 /* Path to executable */
95 char *path;
96 /* argv to pass to executable */
97 char *args[NUM_ARGS]; /* XXX: arbitrary limit */
98 /* full command string (debug) */
99 char *cmdstring;
100} entropy_cmd_t;
101
102/* slow command timeouts (all in milliseconds) */
103/* static int entropy_timeout_default = ENTROPY_TIMEOUT_MSEC; */
104static int entropy_timeout_current = ENTROPY_TIMEOUT_MSEC;
105
106/* this is initialised from a file, by prng_read_commands() */
107static entropy_cmd_t *entropy_cmds = NULL;
108
109/* Prototypes */
110double stir_from_system(void);
111double stir_from_programs(void);
112double stir_gettimeofday(double entropy_estimate);
113double stir_clock(double entropy_estimate);
114double stir_rusage(int who, double entropy_estimate);
e54b3d7c 115double hash_command_output(entropy_cmd_t *src, unsigned char *hash);
e9a17296 116int get_random_bytes_prngd(unsigned char *buf, int len,
117 unsigned short tcp_port, char *socket_path);
118
119/*
120 * Collect 'len' bytes of entropy into 'buf' from PRNGD/EGD daemon
121 * listening either on 'tcp_port', or via Unix domain socket at *
122 * 'socket_path'.
123 * Either a non-zero tcp_port or a non-null socket_path must be
124 * supplied.
125 * Returns 0 on success, -1 on error
126 */
127int
128get_random_bytes_prngd(unsigned char *buf, int len,
129 unsigned short tcp_port, char *socket_path)
130{
131 int fd, addr_len, rval, errors;
132 char msg[2];
133 struct sockaddr_storage addr;
134 struct sockaddr_in *addr_in = (struct sockaddr_in *)&addr;
135 struct sockaddr_un *addr_un = (struct sockaddr_un *)&addr;
136 mysig_t old_sigpipe;
137
138 /* Sanity checks */
139 if (socket_path == NULL && tcp_port == 0)
140 fatal("You must specify a port or a socket");
141 if (socket_path != NULL &&
142 strlen(socket_path) >= sizeof(addr_un->sun_path))
143 fatal("Random pool path is too long");
144 if (len > 255)
145 fatal("Too many bytes to read from PRNGD");
146
147 memset(&addr, '\0', sizeof(addr));
148
149 if (tcp_port != 0) {
150 addr_in->sin_family = AF_INET;
151 addr_in->sin_addr.s_addr = htonl(INADDR_LOOPBACK);
152 addr_in->sin_port = htons(tcp_port);
153 addr_len = sizeof(*addr_in);
154 } else {
155 addr_un->sun_family = AF_UNIX;
156 strlcpy(addr_un->sun_path, socket_path,
157 sizeof(addr_un->sun_path));
158 addr_len = offsetof(struct sockaddr_un, sun_path) +
159 strlen(socket_path) + 1;
160 }
161
162 old_sigpipe = mysignal(SIGPIPE, SIG_IGN);
163
164 errors = 0;
165 rval = -1;
166reopen:
167 fd = socket(addr.ss_family, SOCK_STREAM, 0);
168 if (fd == -1) {
169 error("Couldn't create socket: %s", strerror(errno));
170 goto done;
171 }
172
173 if (connect(fd, (struct sockaddr*)&addr, addr_len) == -1) {
174 if (tcp_port != 0) {
175 error("Couldn't connect to PRNGD port %d: %s",
176 tcp_port, strerror(errno));
177 } else {
178 error("Couldn't connect to PRNGD socket \"%s\": %s",
179 addr_un->sun_path, strerror(errno));
180 }
181 goto done;
182 }
183
184 /* Send blocking read request to PRNGD */
185 msg[0] = 0x02;
186 msg[1] = len;
187
70791e56 188 if (atomicio(vwrite, fd, msg, sizeof(msg)) != sizeof(msg)) {
e9a17296 189 if (errno == EPIPE && errors < 10) {
190 close(fd);
191 errors++;
192 goto reopen;
193 }
194 error("Couldn't write to PRNGD socket: %s",
195 strerror(errno));
196 goto done;
197 }
198
199 if (atomicio(read, fd, buf, len) != len) {
200 if (errno == EPIPE && errors < 10) {
201 close(fd);
202 errors++;
203 goto reopen;
204 }
205 error("Couldn't read from PRNGD socket: %s",
206 strerror(errno));
207 goto done;
208 }
209
210 rval = 0;
211done:
212 mysignal(SIGPIPE, old_sigpipe);
213 if (fd != -1)
214 close(fd);
215 return rval;
216}
217
218double
219stir_gettimeofday(double entropy_estimate)
220{
221 struct timeval tv;
222
223 if (gettimeofday(&tv, NULL) == -1)
224 fatal("Couldn't gettimeofday: %s", strerror(errno));
225
226 RAND_add(&tv, sizeof(tv), entropy_estimate);
227
228 return entropy_estimate;
229}
230
231double
232stir_clock(double entropy_estimate)
233{
234#ifdef HAVE_CLOCK
235 clock_t c;
236
237 c = clock();
238 RAND_add(&c, sizeof(c), entropy_estimate);
239
240 return entropy_estimate;
241#else /* _HAVE_CLOCK */
242 return 0;
243#endif /* _HAVE_CLOCK */
244}
245
246double
247stir_rusage(int who, double entropy_estimate)
248{
249#ifdef HAVE_GETRUSAGE
250 struct rusage ru;
251
252 if (getrusage(who, &ru) == -1)
253 return 0;
254
255 RAND_add(&ru, sizeof(ru), entropy_estimate);
256
257 return entropy_estimate;
258#else /* _HAVE_GETRUSAGE */
259 return 0;
260#endif /* _HAVE_GETRUSAGE */
261}
262
263static int
264timeval_diff(struct timeval *t1, struct timeval *t2)
265{
266 int secdiff, usecdiff;
267
268 secdiff = t2->tv_sec - t1->tv_sec;
269 usecdiff = (secdiff*1000000) + (t2->tv_usec - t1->tv_usec);
270 return (int)(usecdiff / 1000);
271}
272
273double
e54b3d7c 274hash_command_output(entropy_cmd_t *src, unsigned char *hash)
e9a17296 275{
276 char buf[8192];
277 fd_set rdset;
278 int bytes_read, cmd_eof, error_abort, msec_elapsed, p[2];
279 int status, total_bytes_read;
280 static int devnull = -1;
281 pid_t pid;
282 SHA_CTX sha;
283 struct timeval tv_start, tv_current;
284
285 debug3("Reading output from \'%s\'", src->cmdstring);
286
287 if (devnull == -1) {
288 devnull = open("/dev/null", O_RDWR);
289 if (devnull == -1)
290 fatal("Couldn't open /dev/null: %s",
291 strerror(errno));
292 }
293
294 if (pipe(p) == -1)
295 fatal("Couldn't open pipe: %s", strerror(errno));
296
297 (void)gettimeofday(&tv_start, NULL); /* record start time */
298
299 switch (pid = fork()) {
300 case -1: /* Error */
301 close(p[0]);
302 close(p[1]);
303 fatal("Couldn't fork: %s", strerror(errno));
304 /* NOTREACHED */
305 case 0: /* Child */
306 dup2(devnull, STDIN_FILENO);
307 dup2(p[1], STDOUT_FILENO);
308 dup2(p[1], STDERR_FILENO);
309 close(p[0]);
310 close(p[1]);
311 close(devnull);
312
313 execv(src->path, (char**)(src->args));
314
315 debug("(child) Couldn't exec '%s': %s",
316 src->cmdstring, strerror(errno));
317 _exit(-1);
318 default: /* Parent */
319 break;
320 }
321
322 RAND_add(&pid, sizeof(&pid), 0.0);
323
324 close(p[1]);
325
326 /* Hash output from child */
327 SHA1_Init(&sha);
328
329 cmd_eof = error_abort = msec_elapsed = total_bytes_read = 0;
330 while (!error_abort && !cmd_eof) {
331 int ret;
332 struct timeval tv;
333 int msec_remaining;
334
335 (void) gettimeofday(&tv_current, 0);
336 msec_elapsed = timeval_diff(&tv_start, &tv_current);
337 if (msec_elapsed >= entropy_timeout_current) {
338 error_abort=1;
339 continue;
340 }
341 msec_remaining = entropy_timeout_current - msec_elapsed;
342
343 FD_ZERO(&rdset);
344 FD_SET(p[0], &rdset);
345 tv.tv_sec = msec_remaining / 1000;
346 tv.tv_usec = (msec_remaining % 1000) * 1000;
347
348 ret = select(p[0] + 1, &rdset, NULL, NULL, &tv);
349
350 RAND_add(&tv, sizeof(tv), 0.0);
351
352 switch (ret) {
353 case 0:
354 /* timer expired */
355 error_abort = 1;
1c14df9e 356 kill(pid, SIGINT);
e9a17296 357 break;
358 case 1:
359 /* command input */
360 do {
361 bytes_read = read(p[0], buf, sizeof(buf));
362 } while (bytes_read == -1 && errno == EINTR);
363 RAND_add(&bytes_read, sizeof(&bytes_read), 0.0);
364 if (bytes_read == -1) {
365 error_abort = 1;
366 break;
367 } else if (bytes_read) {
368 SHA1_Update(&sha, buf, bytes_read);
369 total_bytes_read += bytes_read;
370 } else {
371 cmd_eof = 1;
372 }
373 break;
374 case -1:
375 default:
376 /* error */
377 debug("Command '%s': select() failed: %s",
378 src->cmdstring, strerror(errno));
379 error_abort = 1;
380 break;
381 }
382 }
383
384 SHA1_Final(hash, &sha);
385
386 close(p[0]);
387
388 debug3("Time elapsed: %d msec", msec_elapsed);
389
390 if (waitpid(pid, &status, 0) == -1) {
391 error("Couldn't wait for child '%s' completion: %s",
ff2d7a98 392 src->cmdstring, strerror(errno));
e9a17296 393 return 0.0;
394 }
395
396 RAND_add(&status, sizeof(&status), 0.0);
397
398 if (error_abort) {
399 /*
400 * Closing p[0] on timeout causes the entropy command to
401 * SIGPIPE. Take whatever output we got, and mark this
402 * command as slow
403 */
404 debug2("Command '%s' timed out", src->cmdstring);
405 src->sticky_badness *= 2;
406 src->badness = src->sticky_badness;
407 return total_bytes_read;
408 }
409
410 if (WIFEXITED(status)) {
411 if (WEXITSTATUS(status) == 0) {
412 return total_bytes_read;
413 } else {
414 debug2("Command '%s' exit status was %d",
415 src->cmdstring, WEXITSTATUS(status));
416 src->badness = src->sticky_badness = 128;
417 return 0.0;
418 }
419 } else if (WIFSIGNALED(status)) {
420 debug2("Command '%s' returned on uncaught signal %d !",
421 src->cmdstring, status);
422 src->badness = src->sticky_badness = 128;
423 return 0.0;
424 } else
425 return 0.0;
426}
427
428double
429stir_from_system(void)
430{
431 double total_entropy_estimate;
432 long int i;
433
434 total_entropy_estimate = 0;
435
436 i = getpid();
437 RAND_add(&i, sizeof(i), 0.5);
438 total_entropy_estimate += 0.1;
439
440 i = getppid();
441 RAND_add(&i, sizeof(i), 0.5);
442 total_entropy_estimate += 0.1;
443
444 i = getuid();
445 RAND_add(&i, sizeof(i), 0.0);
446 i = getgid();
447 RAND_add(&i, sizeof(i), 0.0);
448
449 total_entropy_estimate += stir_gettimeofday(1.0);
450 total_entropy_estimate += stir_clock(0.5);
451 total_entropy_estimate += stir_rusage(RUSAGE_SELF, 2.0);
452
453 return total_entropy_estimate;
454}
455
456double
457stir_from_programs(void)
458{
459 int c;
460 double entropy, total_entropy;
e54b3d7c 461 unsigned char hash[SHA_DIGEST_LENGTH];
e9a17296 462
463 total_entropy = 0;
464 for(c = 0; entropy_cmds[c].path != NULL; c++) {
465 if (!entropy_cmds[c].badness) {
466 /* Hash output from command */
467 entropy = hash_command_output(&entropy_cmds[c],
468 hash);
469
470 /* Scale back estimate by command's rate */
471 entropy *= entropy_cmds[c].rate;
472
473 /* Upper bound of entropy is SHA_DIGEST_LENGTH */
474 if (entropy > SHA_DIGEST_LENGTH)
475 entropy = SHA_DIGEST_LENGTH;
476
477 /* Stir it in */
478 RAND_add(hash, sizeof(hash), entropy);
479
480 debug3("Got %0.2f bytes of entropy from '%s'",
481 entropy, entropy_cmds[c].cmdstring);
482
483 total_entropy += entropy;
484
485 /* Execution time should be a bit unpredictable */
486 total_entropy += stir_gettimeofday(0.05);
487 total_entropy += stir_clock(0.05);
488 total_entropy += stir_rusage(RUSAGE_SELF, 0.1);
489 total_entropy += stir_rusage(RUSAGE_CHILDREN, 0.1);
490 } else {
491 debug2("Command '%s' disabled (badness %d)",
492 entropy_cmds[c].cmdstring,
493 entropy_cmds[c].badness);
494
495 if (entropy_cmds[c].badness > 0)
496 entropy_cmds[c].badness--;
497 }
498 }
499
500 return total_entropy;
501}
502
503/*
504 * prng seedfile functions
505 */
506int
507prng_check_seedfile(char *filename)
508{
509 struct stat st;
510
511 /*
512 * XXX raceable: eg replace seed between this stat and subsequent
513 * open. Not such a problem because we don't really trust the
514 * seed file anyway.
515 * XXX: use secure path checking as elsewhere in OpenSSH
516 */
517 if (lstat(filename, &st) == -1) {
518 /* Give up on hard errors */
519 if (errno != ENOENT)
520 debug("WARNING: Couldn't stat random seed file "
521 "\"%.100s\": %s", filename, strerror(errno));
522 return 0;
523 }
524
525 /* regular file? */
526 if (!S_ISREG(st.st_mode))
527 fatal("PRNG seedfile %.100s is not a regular file",
528 filename);
529
530 /* mode 0600, owned by root or the current user? */
531 if (((st.st_mode & 0177) != 0) || !(st.st_uid == getuid())) {
532 debug("WARNING: PRNG seedfile %.100s must be mode 0600, "
70791e56 533 "owned by uid %li", filename, (long int)getuid());
e9a17296 534 return 0;
535 }
536
537 return 1;
538}
539
540void
541prng_write_seedfile(void)
542{
543 int fd;
e54b3d7c 544 unsigned char seed[SEED_FILE_SIZE];
545 char filename[MAXPATHLEN];
e9a17296 546 struct passwd *pw;
547
548 pw = getpwuid(getuid());
549 if (pw == NULL)
550 fatal("Couldn't get password entry for current user "
70791e56 551 "(%li): %s", (long int)getuid(), strerror(errno));
e9a17296 552
553 /* Try to ensure that the parent directory is there */
554 snprintf(filename, sizeof(filename), "%.512s/%s", pw->pw_dir,
555 _PATH_SSH_USER_DIR);
556 mkdir(filename, 0700);
557
558 snprintf(filename, sizeof(filename), "%.512s/%s", pw->pw_dir,
559 SSH_PRNG_SEED_FILE);
560
561 debug("writing PRNG seed to file %.100s", filename);
562
1c14df9e 563 if (RAND_bytes(seed, sizeof(seed)) <= 0)
564 fatal("PRNG seed extration failed");
e9a17296 565
566 /* Don't care if the seed doesn't exist */
567 prng_check_seedfile(filename);
568
569 if ((fd = open(filename, O_WRONLY|O_TRUNC|O_CREAT, 0600)) == -1) {
570 debug("WARNING: couldn't access PRNG seedfile %.100s "
571 "(%.100s)", filename, strerror(errno));
572 } else {
70791e56 573 if (atomicio(vwrite, fd, &seed, sizeof(seed)) < sizeof(seed))
e9a17296 574 fatal("problem writing PRNG seedfile %.100s "
575 "(%.100s)", filename, strerror(errno));
576 close(fd);
577 }
578}
579
580void
581prng_read_seedfile(void)
582{
583 int fd;
584 char seed[SEED_FILE_SIZE], filename[MAXPATHLEN];
585 struct passwd *pw;
586
587 pw = getpwuid(getuid());
588 if (pw == NULL)
589 fatal("Couldn't get password entry for current user "
70791e56 590 "(%li): %s", (long int)getuid(), strerror(errno));
e9a17296 591
592 snprintf(filename, sizeof(filename), "%.512s/%s", pw->pw_dir,
593 SSH_PRNG_SEED_FILE);
594
595 debug("loading PRNG seed from file %.100s", filename);
596
597 if (!prng_check_seedfile(filename)) {
598 verbose("Random seed file not found or invalid, ignoring.");
599 return;
600 }
601
602 /* open the file and read in the seed */
603 fd = open(filename, O_RDONLY);
604 if (fd == -1)
605 fatal("could not open PRNG seedfile %.100s (%.100s)",
606 filename, strerror(errno));
607
608 if (atomicio(read, fd, &seed, sizeof(seed)) < sizeof(seed)) {
609 verbose("invalid or short read from PRNG seedfile "
610 "%.100s - ignoring", filename);
611 memset(seed, '\0', sizeof(seed));
612 }
613 close(fd);
614
615 /* stir in the seed, with estimated entropy zero */
616 RAND_add(&seed, sizeof(seed), 0.0);
617}
618
619
620/*
621 * entropy command initialisation functions
622 */
623int
624prng_read_commands(char *cmdfilename)
625{
626 char cmd[SEED_FILE_SIZE], *cp, line[1024], path[SEED_FILE_SIZE];
627 double est;
628 entropy_cmd_t *entcmd;
629 FILE *f;
630 int cur_cmd, linenum, num_cmds, arg;
631
632 if ((f = fopen(cmdfilename, "r")) == NULL) {
633 fatal("couldn't read entropy commands file %.100s: %.100s",
634 cmdfilename, strerror(errno));
635 }
636
637 num_cmds = 64;
638 entcmd = xmalloc(num_cmds * sizeof(entropy_cmd_t));
639 memset(entcmd, '\0', num_cmds * sizeof(entropy_cmd_t));
640
641 /* Read in file */
642 cur_cmd = linenum = 0;
643 while (fgets(line, sizeof(line), f)) {
644 linenum++;
645
646 /* Skip leading whitespace, blank lines and comments */
647 cp = line + strspn(line, WHITESPACE);
648 if ((*cp == 0) || (*cp == '#'))
649 continue; /* done with this line */
650
651 /*
652 * The first non-whitespace char should be a double quote
653 * delimiting the commandline
654 */
655 if (*cp != '"') {
656 error("bad entropy command, %.100s line %d",
657 cmdfilename, linenum);
658 continue;
659 }
660
661 /*
662 * First token, command args (incl. argv[0]) in double
663 * quotes
664 */
665 cp = strtok(cp, "\"");
666 if (cp == NULL) {
667 error("missing or bad command string, %.100s "
668 "line %d -- ignored", cmdfilename, linenum);
669 continue;
670 }
671 strlcpy(cmd, cp, sizeof(cmd));
672
673 /* Second token, full command path */
674 if ((cp = strtok(NULL, WHITESPACE)) == NULL) {
675 error("missing command path, %.100s "
676 "line %d -- ignored", cmdfilename, linenum);
677 continue;
678 }
679
680 /* Did configure mark this as dead? */
681 if (strncmp("undef", cp, 5) == 0)
682 continue;
683
684 strlcpy(path, cp, sizeof(path));
685
686 /* Third token, entropy rate estimate for this command */
687 if ((cp = strtok(NULL, WHITESPACE)) == NULL) {
688 error("missing entropy estimate, %.100s "
689 "line %d -- ignored", cmdfilename, linenum);
690 continue;
691 }
692 est = strtod(cp, NULL);
693
694 /* end of line */
695 if ((cp = strtok(NULL, WHITESPACE)) != NULL) {
696 error("garbage at end of line %d in %.100s "
697 "-- ignored", linenum, cmdfilename);
698 continue;
699 }
700
701 /* save the command for debug messages */
702 entcmd[cur_cmd].cmdstring = xstrdup(cmd);
703
704 /* split the command args */
705 cp = strtok(cmd, WHITESPACE);
706 arg = 0;
707 do {
708 entcmd[cur_cmd].args[arg] = xstrdup(cp);
709 arg++;
710 } while(arg < NUM_ARGS && (cp = strtok(NULL, WHITESPACE)));
711
712 if (strtok(NULL, WHITESPACE))
713 error("ignored extra commands (max %d), %.100s "
714 "line %d", NUM_ARGS, cmdfilename, linenum);
715
716 /* Copy the command path and rate estimate */
717 entcmd[cur_cmd].path = xstrdup(path);
718 entcmd[cur_cmd].rate = est;
719
720 /* Initialise other values */
721 entcmd[cur_cmd].sticky_badness = 1;
722
723 cur_cmd++;
724
725 /*
726 * If we've filled the array, reallocate it twice the size
727 * Do this now because even if this we're on the last
728 * command we need another slot to mark the last entry
729 */
730 if (cur_cmd == num_cmds) {
731 num_cmds *= 2;
732 entcmd = xrealloc(entcmd, num_cmds *
733 sizeof(entropy_cmd_t));
734 }
735 }
736
737 /* zero the last entry */
738 memset(&entcmd[cur_cmd], '\0', sizeof(entropy_cmd_t));
739
740 /* trim to size */
741 entropy_cmds = xrealloc(entcmd, (cur_cmd + 1) *
742 sizeof(entropy_cmd_t));
743
744 debug("Loaded %d entropy commands from %.100s", cur_cmd,
745 cmdfilename);
746
747 return cur_cmd < MIN_ENTROPY_SOURCES ? -1 : 0;
748}
749
2980ea68 750void
751usage(void)
752{
753 fprintf(stderr, "Usage: %s [options]\n", __progname);
754 fprintf(stderr, " -v Verbose; display verbose debugging messages.\n");
755 fprintf(stderr, " Multiple -v increases verbosity.\n");
756 fprintf(stderr, " -x Force output in hexidecimal (for debugging)\n");
757 fprintf(stderr, " -X Force output in binary\n");
758 fprintf(stderr, " -b bytes Number of bytes to output (default %d)\n",
759 OUTPUT_SEED_SIZE);
760}
761
e9a17296 762int
763main(int argc, char **argv)
764{
2980ea68 765 unsigned char *buf;
766 int ret, ch, debug_level, output_hex, bytes;
767 extern char *optarg;
768 LogLevel ll;
e9a17296 769
70791e56 770 __progname = ssh_get_progname(argv[0]);
ae43c103 771 init_pathnames();
e9a17296 772 log_init(argv[0], SYSLOG_LEVEL_INFO, SYSLOG_FACILITY_USER, 1);
773
2980ea68 774 ll = SYSLOG_LEVEL_INFO;
775 debug_level = output_hex = 0;
776 bytes = OUTPUT_SEED_SIZE;
777
778 /* Don't write binary data to a tty, unless we are forced to */
779 if (isatty(STDOUT_FILENO))
780 output_hex = 1;
781
782 while ((ch = getopt(argc, argv, "vxXhb:")) != -1) {
783 switch (ch) {
784 case 'v':
785 if (debug_level < 3)
786 ll = SYSLOG_LEVEL_DEBUG1 + debug_level++;
787 break;
788 case 'x':
789 output_hex = 1;
790 break;
791 case 'X':
792 output_hex = 0;
793 break;
794 case 'b':
795 if ((bytes = atoi(optarg)) <= 0)
796 fatal("Invalid number of output bytes");
797 break;
798 case 'h':
799 usage();
800 exit(0);
801 default:
802 error("Invalid commandline option");
803 usage();
804 }
805 }
806
807 log_init(argv[0], ll, SYSLOG_FACILITY_USER, 1);
808
e9a17296 809#ifdef USE_SEED_FILES
810 prng_read_seedfile();
811#endif
812
2980ea68 813 buf = xmalloc(bytes);
814
e9a17296 815 /*
816 * Seed the RNG from wherever we can
817 */
818
819 /* Take whatever is on the stack, but don't credit it */
2980ea68 820 RAND_add(buf, bytes, 0);
e9a17296 821
822 debug("Seeded RNG with %i bytes from system calls",
823 (int)stir_from_system());
824
825#ifdef PRNGD_PORT
2980ea68 826 if (get_random_bytes_prngd(buf, bytes, PRNGD_PORT, NULL) == -1)
e9a17296 827 fatal("Entropy collection failed");
2980ea68 828 RAND_add(buf, bytes, bytes);
e9a17296 829#elif defined(PRNGD_SOCKET)
2980ea68 830 if (get_random_bytes_prngd(buf, bytes, 0, PRNGD_SOCKET) == -1)
e9a17296 831 fatal("Entropy collection failed");
2980ea68 832 RAND_add(buf, bytes, bytes);
e9a17296 833#else
834 /* Read in collection commands */
835 if (prng_read_commands(SSH_PRNG_COMMAND_FILE) == -1)
836 fatal("PRNG initialisation failed -- exiting.");
837 debug("Seeded RNG with %i bytes from programs",
838 (int)stir_from_programs());
839#endif
840
841#ifdef USE_SEED_FILES
842 prng_write_seedfile();
843#endif
844
845 /*
846 * Write the seed to stdout
847 */
848
849 if (!RAND_status())
850 fatal("Not enough entropy in RNG");
851
1c14df9e 852 if (RAND_bytes(buf, bytes) <= 0)
853 fatal("Couldn't extract entropy from PRNG");
e9a17296 854
2980ea68 855 if (output_hex) {
856 for(ret = 0; ret < bytes; ret++)
857 printf("%02x", (unsigned char)(buf[ret]));
858 printf("\n");
859 } else
70791e56 860 ret = atomicio(vwrite, STDOUT_FILENO, buf, bytes);
e9a17296 861
2980ea68 862 memset(buf, '\0', bytes);
863 xfree(buf);
e9a17296 864
2980ea68 865 return ret == bytes ? 0 : 1;
e9a17296 866}
This page took 0.186813 seconds and 5 git commands to generate.