Add vbutil_key

utility/vbutil_key.c

+#include <getopt.h>
+#include <inttypes.h>  /* For PRIu64 */
+#include <stdio.h>
+#include <stdlib.h>
+
+#include "cryptolib.h"
+#include "host_common.h"
+#include "vboot_common.h"
+
+
+/* Command line options */
+enum {
+  OPT_IN = 1000,
+  OPT_OUT,
+  OPT_KEY_VERSION,
+  OPT_ALGORITHM,
+  OPT_MODE_PACK,
+  OPT_MODE_UNPACK,
+};
+
+static struct option long_opts[] = {
+  {"in", 1, 0,                        OPT_IN                      },
+  {"out", 1, 0,                       OPT_OUT                     },
+  {"version", 1, 0,                   OPT_KEY_VERSION             },
+  {"algorithm", 1, 0,                 OPT_ALGORITHM               },
+  {"pack", 0, 0,                      OPT_MODE_PACK               },
+  {"unpack", 0, 0,                    OPT_MODE_UNPACK             },
+  {NULL, 0, 0, 0}
+};
+
+
+/* Print help and return error */
+static int PrintHelp(void) {
+  int i;
+
+  puts("vbutil_key - Verified boot key utility\n"
+       "\n"
+       "Usage:  vbutil_key <--pack|--unpack> [OPTIONS]\n"
+       "\n"
+       "For '--pack', required OPTIONS are:\n"
+       "  --in <infile>                 Input key in .keyb format\n"
+       "  --out <outfile>               Output file for .vbpubk format\n"
+       "  --version <number>            Key version number\n"
+       "  --algorithm <algoid>          Signing algorithm for key, one of:");
+
+  for (i = 0; i < kNumAlgorithms; i++)
+    printf("                                %d (%s)\n", i, algo_strings[i]);
+
+  puts("\n"
+       "For '--unpack', required OPTIONS are:\n"
+       "  --in <infile>                 Input key in .vbpubk format\n"
+       "Optional OPTIONS are:\n"
+       "  --out <outfile>               Output file for .keyb format\n"
+       "");
+  return 1;
+}
+
+
+/* Pack a .keyb file into a .vbpubk */
+static int Pack(const char *infile, const char *outfile, uint64_t algorithm,
+                uint64_t version) {
+  VbPublicKey* key;
+
+  if (!infile || !outfile) {
+    fprintf(stderr, "vbutil_key: Must specify --in and --out\n");
+    return 1;
+  }
+
+  key = PublicKeyReadKeyb(infile, algorithm, version);
+  if (!key) {
+    fprintf(stderr, "vbutil_key: Error reading key.\n");
+    return 1;
+  }
+
+  if (0 != PublicKeyWrite(outfile, key)) {
+    fprintf(stderr, "vbutil_key: Error writing key.\n");
+    return 1;
+  }
+
+  Free(key);
+  return 0;
+}
+
+
+/* Unpack a .vbpubk */
+static int Unpack(const char *infile, const char *outfile) {
+  VbPublicKey* key;
+
+  if (!infile) {
+    fprintf(stderr, "vbutil_key: Must specify --in\n");
+    return 1;
+  }
+
+  key = PublicKeyRead(infile);
+  if (!key) {
+    fprintf(stderr, "vbutil_key: Error reading key.\n");
+    return 1;
+  }
+
+  printf("Key file:    %s\n", infile);
+  printf("Algorithm:   %" PRIu64 " %s\n", key->algorithm,
+         (key->algorithm < kNumAlgorithms ?
+          algo_strings[key->algorithm] : "(invalid)"));
+  printf("Version:     %" PRIu64 "\n", key->key_version);
+
+  /* TODO: write key data, if any */
+
+  Free(key);
+  return 0;
+}
+
+
+int main(int argc, char* argv[]) {
+
+  char *infile = NULL;
+  char *outfile = NULL;
+  int mode = 0;
+  int parse_error = 0;
+  uint64_t version = 1;
+  uint64_t algorithm = kNumAlgorithms;
+  char* e;
+  int i;
+
+  while ((i = getopt_long(argc, argv, "", long_opts, NULL)) != -1) {
+    switch (i) {
+      case '?':
+        /* Unhandled option */
+        printf("Unknown option\n");
+        parse_error = 1;
+        break;
+
+      case OPT_IN:
+        infile = optarg;
+        break;
+
+      case OPT_OUT:
+        outfile = optarg;
+        break;
+
+      case OPT_KEY_VERSION:
+        version = strtoul(optarg, &e, 0);
+        if (!*optarg || (e && *e)) {
+          printf("Invalid --version\n");
+          parse_error = 1;
+        }
+        break;
+
+      case OPT_ALGORITHM:
+        algorithm = strtoul(optarg, &e, 0);
+        if (!*optarg || (e && *e)) {
+          printf("Invalid --algorithm\n");
+          parse_error = 1;
+        }
+        break;
+
+      case OPT_MODE_PACK:
+      case OPT_MODE_UNPACK:
+        mode = i;
+        break;
+    }
+  }
+
+  if (parse_error)
+    return PrintHelp();
+
+  switch(mode) {
+    case OPT_MODE_PACK:
+      return Pack(infile, outfile, algorithm, version);
+    case OPT_MODE_UNPACK:
+      return Unpack(infile, outfile);
+    default:
+      printf("Must specify a mode.\n");
+      return PrintHelp();
+  }
+}