Rename source files and improve readability
[src/app-framework-main.git] / src / main-wgtpkg-sign.c
diff --git a/src/main-wgtpkg-sign.c b/src/main-wgtpkg-sign.c
new file mode 100644 (file)
index 0000000..d40ddaf
--- /dev/null
@@ -0,0 +1,254 @@
+/*
+ Copyright (C) 2015-2020 IoT.bzh
+
+ author: José Bollo <jose.bollo@iot.bzh>
+
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+
+     http://www.apache.org/licenses/LICENSE-2.0
+
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License.
+*/
+
+#define _GNU_SOURCE
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <limits.h>
+#include <errno.h>
+#include <getopt.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+
+#include <libxml/tree.h>
+
+#include "verbose.h"
+#include "wgtpkg-files.h"
+#include "wgtpkg-workdir.h"
+#include "wgtpkg-digsig.h"
+#include "wgtpkg-xmlsec.h"
+
+#if !defined(MAXCERT)
+#define MAXCERT 20
+#endif
+#if !defined(DEFAULT_KEY_FILE)
+#define DEFAULT_KEY_FILE "key.pem"
+#endif
+#if !defined(DEFAULT_CERT_FILE)
+#define DEFAULT_CERT_FILE "cert.pem"
+#endif
+
+const char appname[] = "wgtpkg-sign";
+
+static unsigned int get_number(const char *value)
+{
+       char *end;
+       unsigned long int val;
+
+       val = strtoul(value, &end, 10);
+       if (*end || 0 == val || val >= UINT_MAX || *value == '-') {
+               ERROR("bad number value %s", value);
+               exit(1);
+       }
+       return (unsigned int)val;
+}
+
+static void make_realpath(char **x)
+{
+       char *p = realpath(*x, NULL);
+       if (p == NULL) {
+               ERROR("realpath failed for %s", *x);
+               exit(1);
+       }
+       *x = p;
+}
+
+static void version()
+{
+       printf(
+               "\n"
+               "  %s  version="AFM_VERSION"\n"
+               "\n"
+               "  Copyright (C) 2015-2020 \"IoT.bzh\"\n"
+               "  AFB comes with ABSOLUTELY NO WARRANTY.\n"
+               "  Licence Apache 2\n"
+               "\n",
+               appname
+       );
+}
+
+static void usage()
+{
+       printf(
+               "usage: %s [-f] [-k keyfile] [-c certfile]... [-d number | -a] directory\n"
+               "\n"
+               "   -k keyfile       the private key to use for author signing\n"
+               "   -c certfile      the certificate(s) to use for author signing\n"
+               "   -d number        the number of the distributor signature (zero for automatic)\n"
+               "   -a               the author signature\n"
+               "   -f               force overwriting\n"
+               "   -q               quiet\n"
+               "   -v               verbose\n"
+               "   -V               version\n"
+               "\n",
+               appname
+       );
+}
+
+static struct option options_l[] = {
+       { "author",      no_argument,       NULL, 'a' },
+       { "certificate", required_argument, NULL, 'c' },
+       { "distributor", required_argument, NULL, 'd' },
+       { "force",       no_argument,       NULL, 'f' },
+       { "help",        no_argument,       NULL, 'h' },
+       { "key",         required_argument, NULL, 'k' },
+       { "quiet",       no_argument,       NULL, 'q' },
+       { "verbose",     no_argument,       NULL, 'v' },
+       { "version",     no_argument,       NULL, 'V' },
+       { NULL, 0, NULL, 0 }
+};
+
+static const char options_s[] = "ac:d:fhk:qvV";
+
+/* install the widgets of the list */
+int main(int ac, char **av)
+{
+       int i, force, ncert, author;
+       unsigned int number;
+       char *keyfile, *certfiles[MAXCERT+1], *directory;
+       struct stat s;
+
+       LOGUSER(appname);
+
+       force = ncert = author = 0;
+       number = UINT_MAX;
+       keyfile = directory = NULL;
+       for (;;) {
+               i = getopt_long(ac, av, options_s, options_l, NULL);
+               if (i < 0)
+                       break;
+               switch (i) {
+               case 'c':
+                       if (ncert == MAXCERT) {
+                               ERROR("maximum count of certificates reached");
+                               return 1;
+                       }
+                       certfiles[ncert++] = optarg;
+                       break;
+               case 'k':
+                       if (keyfile) {
+                               ERROR("key already set");
+                               return 1;
+                       }
+                       keyfile = optarg;
+                       break;
+               case 'd':
+                       if (number != UINT_MAX) {
+                               ERROR("number already set");
+                               return 1;
+                       }
+                       number = get_number(optarg);
+                       break;
+               case 'f':
+                       force = 1;
+                       break;
+               case 'a':
+                       author = 1;
+                       break;
+               case 'h':
+                       usage();
+                       return 0;
+               case 'V':
+                       version();
+                       return 0;
+               case 'q':
+                       if (verbosity)
+                               verbosity--;
+                       break;
+               case 'v':
+                       verbosity++;
+                       break;
+               case ':':
+                       ERROR("missing argument");
+                       return 1;
+               default:
+                       ERROR("unrecognized option");
+                       return 1;
+               }
+       }
+
+       /* remaining arguments and final checks */
+       if (optind >= ac) {
+               ERROR("no directory set");
+               return 1;
+       }
+       directory = av[optind++];
+       if (optind < ac) {
+               ERROR("extra parameters found");
+               return 1;
+       }
+
+       /* set default values */
+       if (keyfile == NULL)
+               keyfile = DEFAULT_KEY_FILE;
+       if (ncert == 0)
+               certfiles[ncert++] = DEFAULT_CERT_FILE;
+
+       /* check values */
+       if (stat(directory, &s)) {
+               ERROR("can't find directory %s", directory);
+               return 1;
+       }
+       if (!S_ISDIR(s.st_mode)) {
+               ERROR("%s isn't a directory", directory);
+               return 1;
+       }
+       if (access(keyfile, R_OK) != 0) {
+               ERROR("can't access private key %s", keyfile);
+               return 1;
+       }
+       for(i = 0 ; i < ncert ; i++)
+               if (access(certfiles[i], R_OK) != 0) {
+                       ERROR("can't access certificate %s", certfiles[i]);
+                       return 1;
+               }
+
+       /* init xmlsec module */
+       if (xmlsec_init())
+               return 1;
+
+       /* compute absolutes paths */
+       make_realpath(&keyfile);
+       for(i = 0 ; i < ncert ; i++)
+               make_realpath(&certfiles[i]);
+
+       /* set and enter the workdir */
+       if (set_workdir(directory, 0))
+               return 1;
+
+       if (fill_files())
+               return 1;
+
+       if (author)
+               number = 0;
+       else if (number == UINT_MAX)
+               for (number = 1; get_signature(number) != NULL ; number++);
+
+       if (!force && get_signature(number) != NULL) {
+               ERROR("can't overwrite existing signature %s", get_signature(number)->name);
+               return 1;
+       }
+
+       NOTICE("-- SIGNING content of directory %s for number %u", directory, number);
+
+       certfiles[ncert] = NULL;
+       return !!create_digsig(number, keyfile, (const char**)certfiles);
+}
+