michael@0: /*
michael@0:  * rand_gen.c
michael@0:  *
michael@0:  * a random source (random number generator)
michael@0:  *
michael@0:  * David A. McGrew
michael@0:  * Cisco Systems, Inc.
michael@0:  */
michael@0: /*
michael@0:  *	
michael@0:  * Copyright(c) 2001-2006 Cisco Systems, Inc.
michael@0:  * All rights reserved.
michael@0:  * 
michael@0:  * Redistribution and use in source and binary forms, with or without
michael@0:  * modification, are permitted provided that the following conditions
michael@0:  * are met:
michael@0:  * 
michael@0:  *   Redistributions of source code must retain the above copyright
michael@0:  *   notice, this list of conditions and the following disclaimer.
michael@0:  * 
michael@0:  *   Redistributions in binary form must reproduce the above
michael@0:  *   copyright notice, this list of conditions and the following
michael@0:  *   disclaimer in the documentation and/or other materials provided
michael@0:  *   with the distribution.
michael@0:  * 
michael@0:  *   Neither the name of the Cisco Systems, Inc. nor the names of its
michael@0:  *   contributors may be used to endorse or promote products derived
michael@0:  *   from this software without specific prior written permission.
michael@0:  * 
michael@0:  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
michael@0:  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
michael@0:  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
michael@0:  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
michael@0:  * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
michael@0:  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
michael@0:  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
michael@0:  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
michael@0:  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
michael@0:  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
michael@0:  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
michael@0:  * OF THE POSSIBILITY OF SUCH DAMAGE.
michael@0:  *
michael@0:  */
michael@0: 
michael@0: 
michael@0: #include <stdio.h>           /* for printf() */
michael@0: #include <unistd.h>          /* for getopt() */
michael@0: #include "crypto_kernel.h"
michael@0: 
michael@0: /*
michael@0:  * MAX_PRINT_STRING_LEN is defined in datatypes.h, and is the length
michael@0:  * of the largest hexadecimal string that can be generated by the
michael@0:  * function octet_string_hex_string().
michael@0:  */
michael@0: 
michael@0: #define BUF_LEN (MAX_PRINT_STRING_LEN/2)
michael@0: 
michael@0: void
michael@0: usage(char *prog_name) {
michael@0:   printf("usage: %s -n <num_bytes> [-l][ -d debug_module ]*\n"
michael@0: 	 "   -n <num>   output <num> random bytes, where <num>"
michael@0: 	 " is between zero and %d\n"
michael@0: 	 "   -l         list the avaliable debug modules\n"
michael@0: 	 "   -d <mod>   turn on debugging module <mod>\n", 
michael@0: 	 prog_name, BUF_LEN);
michael@0:   exit(255);
michael@0: }
michael@0: 
michael@0: int
michael@0: main (int argc, char *argv[]) {
michael@0:   extern char *optarg;
michael@0:   int q;
michael@0:   int num_octets = 0;
michael@0:   unsigned do_list_mods = 0;
michael@0:   err_status_t status;
michael@0: 
michael@0:   if (argc == 1)
michael@0:     usage(argv[0]);
michael@0: 
michael@0:   /* initialize kernel - we need to do this before anything else */ 
michael@0:   status = crypto_kernel_init();
michael@0:   if (status) {
michael@0:     printf("error: crypto_kernel init failed\n");
michael@0:     exit(1);
michael@0:   }
michael@0: 
michael@0:   /* process input arguments */
michael@0:   while (1) {
michael@0:     q = getopt(argc, argv, "ld:n:");
michael@0:     if (q == -1) 
michael@0:       break;
michael@0:     switch (q) {
michael@0:     case 'd':
michael@0:       status = crypto_kernel_set_debug_module(optarg, 1);
michael@0:       if (status) {
michael@0: 	printf("error: set debug module (%s) failed\n", optarg);
michael@0: 	exit(1);
michael@0:       }
michael@0:       break;
michael@0:     case 'l':
michael@0:       do_list_mods = 1;
michael@0:       break;
michael@0:     case 'n':
michael@0:       num_octets = atoi(optarg);
michael@0:       if (num_octets < 0 || num_octets > BUF_LEN)
michael@0: 	usage(argv[0]);
michael@0:       break;
michael@0:     default:
michael@0:       usage(argv[0]);
michael@0:     }    
michael@0:   }
michael@0: 
michael@0:   if (do_list_mods) {
michael@0:     status = crypto_kernel_list_debug_modules();
michael@0:     if (status) {
michael@0:       printf("error: list of debug modules failed\n");
michael@0:       exit(1);
michael@0:     }
michael@0:   }
michael@0: 
michael@0:   if (num_octets > 0) {
michael@0:     uint8_t buffer[BUF_LEN];
michael@0:     
michael@0:     status = crypto_get_random(buffer, num_octets);
michael@0:     if (status) {
michael@0:       printf("error: failure in random source\n");
michael@0:     } else {
michael@0:       printf("%s\n", octet_string_hex_string(buffer, num_octets));
michael@0:     }
michael@0:   }
michael@0: 
michael@0:   status = crypto_kernel_shutdown();
michael@0:   if (status) {
michael@0:     printf("error: crypto_kernel shutdown failed\n");
michael@0:     exit(1);
michael@0:   }
michael@0:   
michael@0:   return 0;
michael@0: }
michael@0: