• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * rand_gen.c
3  *
4  * a random source (random number generator)
5  *
6  * David A. McGrew
7  * Cisco Systems, Inc.
8  */
9 /*
10  *
11  * Copyright(c) 2001-2006 Cisco Systems, Inc.
12  * All rights reserved.
13  *
14  * Redistribution and use in source and binary forms, with or without
15  * modification, are permitted provided that the following conditions
16  * are met:
17  *
18  *   Redistributions of source code must retain the above copyright
19  *   notice, this list of conditions and the following disclaimer.
20  *
21  *   Redistributions in binary form must reproduce the above
22  *   copyright notice, this list of conditions and the following
23  *   disclaimer in the documentation and/or other materials provided
24  *   with the distribution.
25  *
26  *   Neither the name of the Cisco Systems, Inc. nor the names of its
27  *   contributors may be used to endorse or promote products derived
28  *   from this software without specific prior written permission.
29  *
30  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
31  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
32  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
33  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
34  * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
35  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
36  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
37  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
38  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
39  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
40  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
41  * OF THE POSSIBILITY OF SUCH DAMAGE.
42  *
43  */
44 
45 
46 #include <stdio.h>           /* for printf() */
47 #include <unistd.h>          /* for getopt() */
48 #include "crypto_kernel.h"
49 
50 /*
51  * MAX_PRINT_STRING_LEN is defined in datatypes.h, and is the length
52  * of the largest hexadecimal string that can be generated by the
53  * function octet_string_hex_string().
54  */
55 
56 #define BUF_LEN (MAX_PRINT_STRING_LEN/2)
57 
58 void
usage(char * prog_name)59 usage(char *prog_name) {
60   printf("usage: %s -n <num_bytes> [-l][ -d debug_module ]*\n"
61 	 "   -n <num>   output <num> random bytes, where <num>"
62 	 " is between zero and %d\n"
63 	 "   -l         list the avaliable debug modules\n"
64 	 "   -d <mod>   turn on debugging module <mod>\n",
65 	 prog_name, BUF_LEN);
66   exit(255);
67 }
68 
69 int
main(int argc,char * argv[])70 main (int argc, char *argv[]) {
71   extern char *optarg;
72   int q;
73   int num_octets = 0;
74   unsigned do_list_mods = 0;
75   err_status_t status;
76 
77   if (argc == 1)
78     usage(argv[0]);
79 
80   /* initialize kernel - we need to do this before anything else */
81   status = crypto_kernel_init();
82   if (status) {
83     printf("error: crypto_kernel init failed\n");
84     exit(1);
85   }
86 
87   /* process input arguments */
88   while (1) {
89     q = getopt(argc, argv, "ld:n:");
90     if (q == -1)
91       break;
92     switch (q) {
93     case 'd':
94       status = crypto_kernel_set_debug_module(optarg, 1);
95       if (status) {
96 	printf("error: set debug module (%s) failed\n", optarg);
97 	exit(1);
98       }
99       break;
100     case 'l':
101       do_list_mods = 1;
102       break;
103     case 'n':
104       num_octets = atoi(optarg);
105       if (num_octets < 0 || num_octets > BUF_LEN)
106 	usage(argv[0]);
107       break;
108     default:
109       usage(argv[0]);
110     }
111   }
112 
113   if (do_list_mods) {
114     status = crypto_kernel_list_debug_modules();
115     if (status) {
116       printf("error: list of debug modules failed\n");
117       exit(1);
118     }
119   }
120 
121   if (num_octets > 0) {
122     uint8_t buffer[BUF_LEN];
123 
124     status = crypto_get_random(buffer, num_octets);
125     if (status) {
126       printf("error: failure in random source\n");
127     } else {
128       printf("%s\n", octet_string_hex_string(buffer, num_octets));
129     }
130   }
131 
132   status = crypto_kernel_shutdown();
133   if (status) {
134     printf("error: crypto_kernel shutdown failed\n");
135     exit(1);
136   }
137 
138   return 0;
139 }
140 
141