From 514b5a9e58022166e1999a4c06792fce5da507aa Mon Sep 17 00:00:00 2001
From: Richard Levitte <levitte@lp.se>
Date: Wed, 1 Oct 2008 14:02:49 +0000
Subject: [PATCH] Add a first draft of a command line tool. It doesn't take any
 options yet (except for an optional salt), and it doesn't actually talk to
 the Yubikey yet.

---
 ykpersonalize.c | 146 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 146 insertions(+)
 create mode 100644 ykpersonalize.c

diff --git a/ykpersonalize.c b/ykpersonalize.c
new file mode 100644
index 0000000..b8f39ce
--- /dev/null
+++ b/ykpersonalize.c
@@ -0,0 +1,146 @@
+/* -*- mode:C; c-file-style: "bsd" -*- */
+/*
+ * Copyright (c) 2008, Yubico AB
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are
+ * met:
+ *
+ *     * Redistributions of source code must retain the above copyright
+ *       notice, this list of conditions and the following disclaimer.
+ *
+ *     * 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.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "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 COPYRIGHT
+ * OWNER OR CONTRIBUTORS 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.
+ */
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <stddef.h>
+#include <string.h>
+#include <unistd.h>
+#include <errno.h>
+
+#include <ykpers.h>
+
+const char *usage =
+"Usage: ykpersonalize [options]\n"
+"-sfile    save configuration to file instead of key\n"
+"          (if file is -, send to stdout)\n"
+"-ifile    read configuration from file\n"
+"          (if file is -, read from stdin)\n"
+"-ooption  change configuration option.  Possible option arguments are:\n"
+"          salt=ssssssss   Salt to be used for key generation.  If none\n"
+"                          is given, a unique random one will be generated\n"
+""
+"-v        verbose\n"
+"-h        help (this text)\n"
+;
+const char *optstring = "hi:o:s:v";
+
+static int reader(char *buf, size_t count, void *stream)
+{
+	return (int)fread(buf, 1, count, (FILE *)stream);
+}
+static int writer(const char *buf, size_t count, void *stream)
+{
+	return (int)fwrite(buf, 1, count, (FILE *)stream);
+}
+
+main(int argc, char **argv)
+{
+	char c;
+	FILE *inf = NULL;
+	FILE *outf = NULL;
+	bool verbose = false;
+	CONFIG *cfg = ykp_create_config();
+
+	/* Options */
+	char *salt = NULL;
+
+	while((c = getopt(argc, argv, optstring)) != -1) {
+		switch (c) {
+		case 'i':
+			if (strcmp(optarg, "-") == 0)
+				inf = stdin;
+			else
+				inf = fopen(optarg, "r");
+			if (inf == NULL) {
+				fprintf(stderr,
+					"Couldn't open %s for reading: %s\n",
+					optarg,
+					strerror(errno));
+				exit(1);
+			}
+			break;
+		case 's':
+			if (strcmp(optarg, "-") == 0)
+				outf = stdout;
+			else
+				outf = fopen(optarg, "r");
+			if (outf == NULL) {
+				fprintf(stderr,
+					"Couldn't open %s for writing: %s\n",
+					optarg,
+					strerror(errno));
+				exit(1);
+			}
+			break;
+		case 'o':
+			if (strncmp(optarg, "salt=", 5) == 0)
+				salt = strdup(optarg+5);
+			else {
+				fprintf(stderr, "Unknown option '%s'\n",
+					optarg);
+				fprintf(stderr, usage);
+				exit(1);
+			}
+			break;
+		case 'v':
+			verbose = true;
+		case 'h':
+		default:
+			fprintf(stderr, usage);
+			exit(0);
+			break;
+		}
+	}
+
+	if (inf) {
+		ykp_read_config(cfg, reader, inf);
+		fclose(inf);
+	} else {
+		char passphrasebuf[256]; size_t passphraselen;
+		fprintf(stderr, "Passphrase to create AES key: ");
+		fflush(stderr);
+		fgets(passphrasebuf, sizeof(passphrasebuf), stdin);
+		passphraselen = strlen(passphrasebuf);
+		if (passphrasebuf[passphraselen - 1] == '\n')
+			passphrasebuf[passphraselen - 1] == '\0';
+		ykp_AES_key_from_passphrase(cfg, passphrasebuf, salt);
+	}
+
+	if (outf) {
+		ykp_write_config(cfg, writer, outf);
+		fclose(outf);
+	} else {
+		/* Output to key, and that's a different story! */
+	}
+
+	if (salt)
+		free(salt);
+}
-- 
2.39.5