asdcplib 1.9.45

1 files changed, 292 insertions, 0 deletions

diff --git a/asdcplib/src/kmrandgen.cpp b/asdcplib/src/kmrandgen.cpp
new file mode 100644
index 00000000..560fa6ec
--- /dev/null
+++ b/asdcplib/src/kmrandgen.cpp
@@ -0,0 +1,292 @@
+/*
+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
+//