1 files changed, 68 insertions, 0 deletions
diff --git a/sha3sum.c b/sha3sum.c
new file mode 100644
index 0000000..a34f8aa
--- /dev/null
+++ b/sha3sum.c
@@ -0,0 +1,68 @@
+/* See LICENSE file for copyright and license details. */
+#include "common.h"
+#include "arg.h"
+
+#include <alloca.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+
+static void
+usage(void)
+{
+	/*
+	 * Since our main in this file validates the arguments,
+	 * we can add -a here but leave it left out in common.c,
+	 * which we like because the other commands do not have -a.
+	 */
+
+	fprintf(stderr, "usage: %s [-u | -l | -b | -c] [-a bits] [-R rate] [-C capacity] "
+	                "[-N output-size] [-S state-size] [-W word-size] "
+	                "[-Z squeeze-count] [-vxz] [file ...]\n", argv0);
+	exit(2);
+}
+
+int
+main(int argc, char *argv[])
+{
+	int bits = 224, orig_argc = argc;
+	char **orig_argv = alloca((argc + 1) * sizeof(*argv));
+	struct libkeccak_generalised_spec spec;
+
+	libkeccak_generalised_spec_initialise(&spec);
+	memcpy(orig_argv, argv, (argc + 1) * sizeof(*argv));
+
+	ARGBEGIN {
+	case 'R':
+	case 'C':
+	case 'N':
+	case 'O':
+	case 'S':
+	case 'B':
+	case 'W':
+	case 'Z':
+		(void) EARGF(usage());
+		break;
+	case 'u':
+	case 'l':
+	case 'b':
+	case 'c':
+	case 'v':
+	case 'x':
+	case 'z':
+		break;
+	case 'a':
+		bits = atoi(EARGF(usage()));
+		if (bits != 224 && bits != 256 && bits != 384 && bits != 512) {
+			fprintf(stderr, "%s: valid arguments for -a are 224 (default), 256, 384, and 512\n", argv0);
+			return 2;
+		}
+		break;
+	default:
+		usage();
+	} ARGEND;
+
+	libkeccak_spec_sha3((struct libkeccak_spec *)&spec, bits);
+	return run(orig_argc, orig_argv, &spec, LIBKECCAK_SHA3_SUFFIX, 1);
+}