diff options
| author | Carl Hetherington <cth@carlh.net> | 2016-01-16 15:58:56 +0000 |
|---|---|---|
| committer | Carl Hetherington <cth@carlh.net> | 2016-01-16 15:58:56 +0000 |
| commit | e04eced93da1907e803c065a953ffe0d6540e1cb (patch) | |
| tree | cab897dea7c1a50c141ff948ff21e401038d0c07 /asdcplib/src/kmrandgen.cpp | |
| parent | 485794f8322b090a22da841961025b19642e42a2 (diff) | |
Use asdclib-cth rather than in-tree copy.
Diffstat (limited to 'asdcplib/src/kmrandgen.cpp')
| -rw-r--r-- | asdcplib/src/kmrandgen.cpp | 292 |
1 files changed, 0 insertions, 292 deletions
diff --git a/asdcplib/src/kmrandgen.cpp b/asdcplib/src/kmrandgen.cpp deleted file mode 100644 index 560fa6e..0000000 --- a/asdcplib/src/kmrandgen.cpp +++ /dev/null @@ -1,292 +0,0 @@ -/* -Copyright (c) 2005-2009, John Hurst -All rights reserved. - -Redistribution and use in source and binary forms, with or without -modification, are permitted provided that the following conditions -are met: -1. Redistributions of source code must retain the above copyright - notice, this list of conditions and the following disclaimer. -2. Redistributions in binary form must reproduce the above copyright - notice, this list of conditions and the following disclaimer in the - documentation and/or other materials provided with the distribution. -3. The name of the author may not be used to endorse or promote products - derived from this software without specific prior written permission. - -THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR -IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES -OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. -IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, -INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT -NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, -DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY -THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT -(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF -THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. -*/ - /*! \file kmrandgen.cpp - \version $Id: kmrandgen.cpp,v 1.8 2009/03/04 17:52:45 jhurst Exp $ - \brief psuedo-random number generation utility - */ - -#include "AS_DCP.h" -#include <KM_fileio.h> -#include <KM_prng.h> -#include <ctype.h> - -using namespace Kumu; - -const ui32_t RandBlockSize = 16; -const char* PROGRAM_NAME = "kmrandgen"; - -// Increment the iterator, test for an additional non-option command line argument. -// Causes the caller to return if there are no remaining arguments or if the next -// argument begins with '-'. -#define TEST_EXTRA_ARG(i,c) if ( ++i >= argc || argv[(i)][0] == '-' ) \ - { \ - fprintf(stderr, "Argument not found for option -%c.\n", (c)); \ - return; \ - } - -// -void -banner(FILE* stream = stdout) -{ - fprintf(stream, "\n\ -%s (asdcplib %s)\n\n\ -Copyright (c) 2003-2009 John Hurst\n\n\ -%s is part of the asdcp DCP tools package.\n\ -asdcplib may be copied only under the terms of the license found at\n\ -the top of every file in the asdcplib distribution kit.\n\n\ -Specify the -h (help) option for further information about %s\n\n", - PROGRAM_NAME, Kumu::Version(), PROGRAM_NAME, PROGRAM_NAME); -} - -// -void -usage(FILE* stream = stdout) -{ - fprintf(stream, "\ -USAGE: %s [-b|-B|-c|-x] [-n] [-s <size>] [-v]\n\ -\n\ - %s [-h|-help] [-V]\n\ -\n\ - -b - Output a stream of binary data\n\ - -B - Output a Base64 string\n\ - -c - Output a C-language struct containing the values\n\ - -h | -help - Show help\n\ - -n - Suppress newlines\n\ - -s <size> - Number of random bytes to generate (default 32)\n\ - -v - Verbose. Prints informative messages to stderr\n\ - -V - Show version information\n\ - -x - Output hexadecimal (default)\n\ -\n\ - NOTES: o There is no option grouping, all options must be distinct arguments.\n\ - o All option arguments must be separated from the option by whitespace.\n\ -\n", PROGRAM_NAME, PROGRAM_NAME); -} - -enum OutputFormat_t { - OF_HEX, - OF_BINARY, - OF_BASE64, - OF_CSTRUCT -}; - -// -class CommandOptions -{ - CommandOptions(); - -public: - bool error_flag; // true if the given options are in error or not complete - bool no_newline_flag; // - bool verbose_flag; // true if the verbose option was selected - bool version_flag; // true if the version display option was selected - bool help_flag; // true if the help display option was selected - OutputFormat_t format; // - ui32_t request_size; - - // - CommandOptions(int argc, const char** argv) : - error_flag(true), no_newline_flag(false), verbose_flag(false), - version_flag(false), help_flag(false), format(OF_HEX), request_size(RandBlockSize*2) - { - for ( int i = 1; i < argc; i++ ) - { - - if ( (strcmp( argv[i], "-help") == 0) ) - { - help_flag = true; - continue; - } - - if ( argv[i][0] == '-' && isalpha(argv[i][1]) && argv[i][2] == 0 ) - { - switch ( argv[i][1] ) - { - case 'b': format = OF_BINARY; break; - case 'B': format = OF_BASE64; break; - case 'c': format = OF_CSTRUCT; break; - case 'n': no_newline_flag = true; break; - case 'h': help_flag = true; break; - - case 's': - TEST_EXTRA_ARG(i, 's'); - request_size = abs(atoi(argv[i])); - break; - - case 'v': verbose_flag = true; break; - case 'V': version_flag = true; break; - case 'x': format = OF_HEX; break; - - default: - fprintf(stderr, "Unrecognized option: %s\n", argv[i]); - return; - } - } - else - { - fprintf(stderr, "Unrecognized option: %s\n", argv[i]); - return; - } - } - - if ( help_flag || version_flag ) - return; - - if ( request_size == 0 ) - { - fprintf(stderr, "Please use a non-zero request size\n"); - return; - } - - error_flag = false; - } -}; - - -// -int -main(int argc, const char** argv) -{ - CommandOptions Options(argc, argv); - - if ( Options.version_flag ) - banner(); - - if ( Options.help_flag ) - usage(); - - if ( Options.version_flag || Options.help_flag ) - return 0; - - if ( Options.error_flag ) - { - fprintf(stderr, "There was a problem. Type %s -h for help.\n", PROGRAM_NAME); - return 3; - } - - FortunaRNG RandGen; - ByteString Buf(Kumu::Kilobyte); - - if ( Options.verbose_flag ) - fprintf(stderr, "Generating %d random byte%s.\n", Options.request_size, (Options.request_size == 1 ? "" : "s")); - - if ( Options.format == OF_BINARY ) - { - if ( KM_FAILURE(Buf.Capacity(Options.request_size)) ) - { - fprintf(stderr, "randbuf: %s\n", RESULT_ALLOC.Label()); - return 1; - } - - RandGen.FillRandom(Buf.Data(), Options.request_size); - fwrite((byte_t*)Buf.Data(), 1, Options.request_size, stdout); - } - else if ( Options.format == OF_CSTRUCT ) - { - ui32_t line_count = 0; - byte_t* p = Buf.Data(); - printf("byte_t rand_buf[%u] = {\n", Options.request_size); - - if ( Options.request_size > 128 ) - fputs(" // 0x00000000\n", stdout); - - while ( Options.request_size > 0 ) - { - if ( line_count > 0 && (line_count % (RandBlockSize*8)) == 0 ) - fprintf(stdout, " // 0x%08x\n", line_count); - - RandGen.FillRandom(p, RandBlockSize); - fputc(' ', stdout); - - for ( ui32_t i = 0; i < RandBlockSize && Options.request_size > 0; i++, Options.request_size-- ) - printf(" 0x%02x,", p[i]); - - fputc('\n', stdout); - line_count += RandBlockSize; - } - - fputs("};", stdout); - - if ( ! Options.no_newline_flag ) - fputc('\n', stdout); - } - else if ( Options.format == OF_BASE64 ) - { - if ( KM_FAILURE(Buf.Capacity(Options.request_size)) ) - { - fprintf(stderr, "randbuf: %s\n", RESULT_ALLOC.Label()); - return 1; - } - - ByteString Strbuf; - ui32_t e_len = base64_encode_length(Options.request_size) + 1; - - if ( KM_FAILURE(Strbuf.Capacity(e_len)) ) - { - fprintf(stderr, "strbuf: %s\n", RESULT_ALLOC.Label()); - return 1; - } - - RandGen.FillRandom(Buf.Data(), Options.request_size); - - if ( base64encode(Buf.RoData(), Options.request_size, (char*)Strbuf.Data(), Strbuf.Capacity()) == 0 ) - { - fprintf(stderr, "encode error\n"); - return 2; - } - - fputs((const char*)Strbuf.RoData(), stdout); - - if ( ! Options.no_newline_flag ) - fputs("\n", stdout); - } - else // OF_HEX - { - byte_t* p = Buf.Data(); - char hex_buf[64]; - - while ( Options.request_size > 0 ) - { - ui32_t x_len = xmin(Options.request_size, RandBlockSize); - RandGen.FillRandom(p, RandBlockSize); - bin2hex(p, x_len, hex_buf, 64); - fputs(hex_buf, stdout); - - if ( ! Options.no_newline_flag ) - fputc('\n', stdout); - - Options.request_size -= x_len; - } - - } - - return 0; -} - - -// -// end kmrandgen.cpp -// |
