ucert.c

   1 /*
   2  * Copyright (C) 2018 Daniel Golle <daniel@makrotopia.org>
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License version 3
   6  * as published by the Free Software Foundation
   7  *
   8  * This program is distributed in the hope that it will be useful,
   9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11  * GNU General Public License for more details.
  12  */
  13
  14 #define _GNU_SOURCE
  15
  16 #include <fcntl.h>
  17 #include <dlfcn.h>
  18 #include <stdio.h>
  19 #include <stdbool.h>
  20 #include <stdlib.h>
  21 #include <string.h>
  22 #include <getopt.h>
  23 #include <stdint.h>
  24 #include <unistd.h>
  25 #include <inttypes.h>
  26 #include <sys/stat.h>
  27 #include <sys/wait.h>
  28
  29 #include <json-c/json.h>
  30 #include <libubox/blob.h>
  31 #include <libubox/utils.h>
  32 #include <libubox/list.h>
  33 #include <libubox/vlist.h>
  34 #include <libubox/blobmsg_json.h>
  35
  36 #include "usign.h"
  37
  38 #define CERT_BUF_LEN 4096
  39
  40 static enum {
  41         CMD_APPEND,
  42         CMD_DUMP,
  43         CMD_ISSUE,
  44         CMD_REVOKE,
  45         CMD_VERIFY,
  46         CMD_NONE,
  47 } cmd = CMD_NONE;
  48
  49 static bool quiet;
  50 #ifndef UCERT_STRIP_MESSAGES
  51 #define DPRINTF(format, ...) if (!quiet) fprintf(stderr, "%s(%d): " format, __func__, __LINE__, ## __VA_ARGS__)
  52 #else
  53 #define DPRINTF(format, ...)
  54 #endif
  55
  56 /*
  57  * ucert structure
  58  * |               BLOB                    |
  59  * |    SIGNATURE    |       PAYLOAD       |
  60  * |                 |[ BLOBMSG CONTAINER ]|
  61  * |                 |[[T,i,v,e,f,pubkey ]]|
  62  */
  63 enum cert_attr {
  64         CERT_ATTR_SIGNATURE,
  65         CERT_ATTR_PAYLOAD,
  66         CERT_ATTR_MAX
  67 };
  68
  69 static const struct blob_attr_info cert_policy[CERT_ATTR_MAX] = {
  70         [CERT_ATTR_SIGNATURE] = { .type = BLOB_ATTR_BINARY },
  71         [CERT_ATTR_PAYLOAD] = { .type = BLOB_ATTR_NESTED },
  72 };
  73
  74 enum cert_cont_attr {
  75         CERT_CT_ATTR_PAYLOAD,
  76         CERT_CT_ATTR_MAX
  77 };
  78
  79 static const struct blobmsg_policy cert_cont_policy[CERT_CT_ATTR_MAX] = {
  80         [CERT_CT_ATTR_PAYLOAD] = { .name = "ucert", .type = BLOBMSG_TYPE_TABLE },
  81 };
  82
  83 enum cert_payload_attr {
  84         CERT_PL_ATTR_CERTTYPE,
  85         CERT_PL_ATTR_CERTID,
  86         CERT_PL_ATTR_VALIDFROMTIME,
  87         CERT_PL_ATTR_EXPIRETIME,
  88         CERT_PL_ATTR_PUBKEY,
  89         CERT_PL_ATTR_KEY_FINGERPRINT,
  90         CERT_PL_ATTR_MAX
  91 };
  92
  93 enum certtype_id {
  94         CERTTYPE_UNSPEC,
  95         CERTTYPE_AUTH,
  96         CERTTYPE_REVOKE
  97 };
  98
  99 static const struct blobmsg_policy cert_payload_policy[CERT_PL_ATTR_MAX] = {
 100         [CERT_PL_ATTR_CERTTYPE] = { .name = "certtype", .type = BLOBMSG_TYPE_INT32 },
 101         [CERT_PL_ATTR_CERTID] = { .name = "certid", .type = BLOBMSG_TYPE_INT32 },
 102         [CERT_PL_ATTR_VALIDFROMTIME] = { .name = "validfrom", .type = BLOBMSG_TYPE_INT64 },
 103         [CERT_PL_ATTR_EXPIRETIME] = { .name = "expiresat", .type = BLOBMSG_TYPE_INT64 },
 104         [CERT_PL_ATTR_PUBKEY] = { .name = "pubkey", .type = BLOBMSG_TYPE_STRING },
 105         [CERT_PL_ATTR_KEY_FINGERPRINT] = { .name = "fingerprint", .type = BLOBMSG_TYPE_STRING },
 106 };
 107
 108 /* list to store certificate chain at runtime */
 109 struct cert_object {
 110         struct list_head list;
 111         struct blob_attr *cert[CERT_ATTR_MAX];
 112 };
 113
 114 /* write buffer to file */
 115 static int write_file(const char *filename, void *buf, size_t len, bool append) {
 116         FILE *f;
 117         size_t outlen;
 118
 119         f = fopen(filename, append?"a":"w");
 120         if (!f)
 121                 return 1;
 122
 123         outlen = fwrite(buf, 1, len, f);
 124         fclose(f);
 125         return (outlen == len);
 126 }
 127
 128 /* load certfile into list */
 129 static int cert_load(const char *certfile, struct list_head *chain) {
 130         FILE *f;
 131         struct blob_attr *certtb[CERT_ATTR_MAX];
 132         struct blob_attr *bufpt;
 133         struct cert_object *cobj;
 134         char filebuf[CERT_BUF_LEN];
 135         int ret = 0, pret = 0;
 136         int len, pos = 0;
 137
 138         f = fopen(certfile, "r");
 139         if (!f)
 140                 return 1;
 141
 142         len = fread(&filebuf, 1, CERT_BUF_LEN - 1, f);
 143         if (len < 64)
 144                 return 1;
 145
 146         ret = ferror(f) || !feof(f);
 147         fclose(f);
 148         if (ret)
 149                 return 1;
 150
 151         bufpt = (struct blob_attr *)filebuf;
 152         do {
 153                 pret = blob_parse(bufpt, certtb, cert_policy, CERT_ATTR_MAX);
 154                 if (pret <= 0)
 155                         /* no attributes found */
 156                         break;
 157
 158                 if (pos + blob_pad_len(bufpt) > len)
 159                         /* blob exceeds filebuffer */
 160                         break;
 161                 else
 162                         pos += blob_pad_len(bufpt);
 163
 164                 if (!certtb[CERT_ATTR_SIGNATURE])
 165                         /* no signature -> drop */
 166                         break;
 167
 168                 cobj = calloc(1, sizeof(*cobj));
 169                 cobj->cert[CERT_ATTR_SIGNATURE] = blob_memdup(certtb[CERT_ATTR_SIGNATURE]);
 170                 if (certtb[CERT_ATTR_PAYLOAD])
 171                         cobj->cert[CERT_ATTR_PAYLOAD] = blob_memdup(certtb[CERT_ATTR_PAYLOAD]);
 172
 173                 list_add_tail(&cobj->list, chain);
 174                 ret += pret;
 175                 bufpt = blob_next(bufpt);
 176         /* repeat parsing while there is still enough remaining data in buffer */
 177         } while(len > pos + sizeof(struct blob_attr));
 178
 179         return (ret <= 0);
 180 }
 181
 182 #ifdef UCERT_FULL
 183 /* append signature to certfile */
 184 static int cert_append(const char *certfile, const char *sigfile) {
 185         FILE *fs;
 186         char filebuf[CERT_BUF_LEN];
 187         struct blob_buf sigbuf;
 188         int len;
 189         int ret;
 190
 191         fs = fopen(sigfile, "r");
 192         if (!fs)
 193                 return 1;
 194
 195         len = fread(&filebuf, 1, CERT_BUF_LEN - 1, fs);
 196         ret = ferror(fs) || !feof(fs) || (len < 64);
 197         fclose(fs);
 198         if (ret)
 199                 return 1;
 200
 201         blob_buf_init(&sigbuf, 0);
 202         blob_put(&sigbuf, CERT_ATTR_SIGNATURE, filebuf, len);
 203         ret = write_file(certfile, sigbuf.head, blob_raw_len(sigbuf.head), true);
 204         blob_buf_free(&sigbuf);
 205         return ret;
 206 }
 207 #endif
 208
 209 /* verify the signature of a single chain element */
 210 static int cert_verify_blob(struct blob_attr *cert[CERT_ATTR_MAX],
 211                        const char *pubkeyfile, const char *pubkeydir) {
 212         int i;
 213         char msgfname[256], sigfname[256];
 214         int ret;
 215         char tmpdir[] = "/tmp/ucert-XXXXXX";
 216
 217         if (mkdtemp(tmpdir) == NULL)
 218                 return errno;
 219
 220         snprintf(msgfname, sizeof(msgfname) - 1, "%s/%s", tmpdir, "payload");
 221         snprintf(sigfname, sizeof(sigfname) - 1, "%s/%s", tmpdir, "payload.sig");
 222
 223         for (i = 0; i < CERT_ATTR_MAX; i++) {
 224                 struct blob_attr *v = cert[i];
 225
 226                 if (!v)
 227                         break;
 228
 229                 switch(cert_policy[i].type) {
 230                 case BLOB_ATTR_BINARY:
 231                         write_file(sigfname, blob_data(v), blob_len(v), false);
 232                         break;
 233                 case BLOB_ATTR_NESTED:
 234                         write_file(msgfname, blob_data(v), blob_len(v), false);
 235                         break;
 236                 }
 237         }
 238
 239         ret = usign_v(msgfname, pubkeyfile, pubkeydir, sigfname, quiet);
 240
 241         unlink(msgfname);
 242         unlink(sigfname);
 243         rmdir(tmpdir);
 244
 245         return ret;
 246 }
 247
 248 /* verify cert chain (and message) */
 249 static int chain_verify(const char *msgfile, const char *pubkeyfile,
 250                         const char *pubkeydir, struct list_head *chain) {
 251         struct cert_object *cobj;
 252         struct blob_attr *containertb[CERT_CT_ATTR_MAX];
 253         struct blob_attr *payloadtb[CERT_PL_ATTR_MAX];
 254         char tmpdir[] = "/tmp/ucert-XXXXXX";
 255         char chainedpubkey[256] = {0};
 256         char chainedfp[17] = {0};
 257         char extsigfile[256] = {0};
 258         int ret = 1;
 259         int checkmsg = 0;
 260         struct timeval tv;
 261
 262         if (mkdtemp(tmpdir) == NULL)
 263                 return errno;
 264
 265         if (msgfile)
 266                 checkmsg = -1;
 267
 268         gettimeofday(&tv, NULL);
 269
 270         list_for_each_entry(cobj, chain, list) {
 271                 /* blob has payload, verify that using signature */
 272                 if (cobj->cert[CERT_ATTR_PAYLOAD]) {
 273                         uint64_t validfrom;
 274                         uint64_t expiresat;
 275                         uint32_t certtype;
 276
 277                         ret = cert_verify_blob(cobj->cert, chainedpubkey[0]?chainedpubkey:pubkeyfile, pubkeydir);
 278                         if (ret)
 279                                 goto clean_and_return;
 280
 281                         blobmsg_parse(cert_cont_policy,
 282                                       ARRAY_SIZE(cert_cont_policy),
 283                                       containertb,
 284                                       blob_data(cobj->cert[CERT_ATTR_PAYLOAD]),
 285                                       blob_len(cobj->cert[CERT_ATTR_PAYLOAD]));
 286                         if (!containertb[CERT_CT_ATTR_PAYLOAD]) {
 287                                 ret = 1;
 288                                 DPRINTF("no ucert in signed payload\n");
 289                                 goto clean_and_return;
 290                         }
 291                         blobmsg_parse(cert_payload_policy,
 292                                       ARRAY_SIZE(cert_payload_policy),
 293                                       payloadtb,
 294                                       blobmsg_data(containertb[CERT_CT_ATTR_PAYLOAD]),
 295                                       blobmsg_data_len(containertb[CERT_CT_ATTR_PAYLOAD]));
 296
 297                         if (!payloadtb[CERT_PL_ATTR_CERTTYPE] ||
 298                             !payloadtb[CERT_PL_ATTR_VALIDFROMTIME] ||
 299                             !payloadtb[CERT_PL_ATTR_EXPIRETIME] ||
 300                             !payloadtb[CERT_PL_ATTR_PUBKEY]) {
 301                                 ret = 1;
 302                                 DPRINTF("missing mandatory ucert attributes\n");
 303                                 goto clean_and_return;
 304                         }
 305                         certtype = blobmsg_get_u32(payloadtb[CERT_PL_ATTR_CERTTYPE]);
 306                         validfrom = blobmsg_get_u64(payloadtb[CERT_PL_ATTR_VALIDFROMTIME]);
 307                         expiresat = blobmsg_get_u64(payloadtb[CERT_PL_ATTR_EXPIRETIME]);
 308
 309                         if (certtype != CERTTYPE_AUTH) {
 310                                 ret = 2;
 311                                 DPRINTF("wrong certificate type\n");
 312                                 goto clean_and_return;
 313                         }
 314
 315                         if (tv.tv_sec < validfrom ||
 316                             tv.tv_sec >= expiresat) {
 317                                 ret = 3;
 318                                 DPRINTF("certificate expired\n");
 319                                 goto clean_and_return;
 320                         }
 321
 322                         snprintf(chainedpubkey, sizeof(chainedpubkey) - 1, "%s/%s", tmpdir, "chained-pubkey");
 323                         write_file(chainedpubkey,
 324                                    blobmsg_data(payloadtb[CERT_PL_ATTR_PUBKEY]),
 325                                    blobmsg_data_len(payloadtb[CERT_PL_ATTR_PUBKEY]),
 326                                    false);
 327
 328                         if (usign_f_pubkey(chainedfp, chainedpubkey)) {
 329                                 DPRINTF("cannot get fingerprint for chained key\n");
 330                                 ret = 2;
 331                                 goto clean_and_return;
 332                         }
 333                         if (pubkeydir && _usign_key_is_revoked(chainedfp, pubkeydir)) {
 334                                 DPRINTF("key %s has been revoked!\n", chainedfp);
 335                                 ret = 4;
 336                                 goto clean_and_return;
 337                         }
 338                 } else {
 339                 /* blob doesn't have payload, verify message using signature */
 340                         if (msgfile) {
 341                                 snprintf(extsigfile, sizeof(extsigfile) - 1, "%s/%s", tmpdir, "ext-sig");
 342                                 write_file(extsigfile,
 343                                            blob_data(cobj->cert[CERT_ATTR_SIGNATURE]),
 344                                            blob_len(cobj->cert[CERT_ATTR_SIGNATURE]),
 345                                            false);
 346                                 checkmsg = ret = usign_v(msgfile,
 347                                               chainedpubkey[0]?chainedpubkey:pubkeyfile,
 348                                               pubkeydir, extsigfile, quiet);
 349                                 unlink(extsigfile);
 350                         } else {
 351                                 DPRINTF("stray trailing signature without anything to verify!\n");
 352                                 ret = 1;
 353                         };
 354                 }
 355         }
 356
 357         if (checkmsg == -1)
 358                 DPRINTF("missing signature to verify message!\n");
 359
 360 clean_and_return:
 361         if (chainedpubkey[0])
 362                 unlink(chainedpubkey);
 363         rmdir(tmpdir);
 364         return ret | checkmsg;
 365 }
 366
 367 #ifdef UCERT_FULL
 368 /* dump single chain element to console */
 369 static void cert_dump_blob(struct blob_attr *cert[CERT_ATTR_MAX]) {
 370         int i;
 371
 372         for (i = 0; i < CERT_ATTR_MAX; i++) {
 373                 struct blob_attr *v = cert[i];
 374
 375                 if (!v)
 376                         continue;
 377
 378                 switch(cert_policy[i].type) {
 379                 case BLOB_ATTR_BINARY:
 380                         fprintf(stdout, "signature:\n---\n%s---\n", (char *) blob_data(v));
 381                         break;
 382                 case BLOB_ATTR_NESTED:
 383                         fprintf(stdout, "payload:\n---\n%s\n---\n", blobmsg_format_json_indent(blob_data(v), false, 0));
 384                         break;
 385                 }
 386         }
 387 }
 388
 389 /* dump certfile to console */
 390 static int cert_dump(const char *certfile) {
 391         struct cert_object *cobj;
 392         static LIST_HEAD(certchain);
 393         unsigned int count = 0;
 394
 395         if (cert_load(certfile, &certchain)) {
 396                 DPRINTF("cannot parse cert\n");
 397                 return 1;
 398         }
 399
 400         list_for_each_entry(cobj, &certchain, list) {
 401                 fprintf(stdout, "=== CHAIN ELEMENT %02u ===\n", ++count);
 402                 cert_dump_blob(cobj->cert);
 403         }
 404
 405         return 0;
 406 }
 407
 408 /* issue an auth certificate for pubkey */
 409 static int cert_issue(const char *certfile, const char *pubkeyfile, const char *seckeyfile) {
 410         struct blob_buf certbuf;
 411         struct blob_buf payloadbuf;
 412         struct timeval tv;
 413         int pklen, siglen;
 414         int revoker = 1;
 415         void *c;
 416         FILE *pkf, *sigf;
 417         char pkb[512];
 418         char sigb[512];
 419         char fname[256], sfname[256];
 420         char pkfp[17];
 421         char tmpdir[] = "/tmp/ucert-XXXXXX";
 422
 423         pkf = fopen(pubkeyfile, "r");
 424         if (!pkf)
 425                 return -1;
 426
 427         pklen = fread(pkb, 1, 512, pkf);
 428         pkb[pklen] = '\0';
 429
 430         if (pklen < 32)
 431                 return -1;
 432
 433         fclose(pkf);
 434
 435         if (usign_f_pubkey(pkfp, pubkeyfile))
 436                 return -1;
 437
 438         gettimeofday(&tv, NULL);
 439
 440         if (mkdtemp(tmpdir) == NULL)
 441                 return errno;
 442
 443         while (revoker >= 0) {
 444                 blob_buf_init(&payloadbuf, 0);
 445                 c = blobmsg_open_table(&payloadbuf, "ucert");
 446                 blobmsg_add_u32(&payloadbuf, "certtype", revoker?CERTTYPE_REVOKE:CERTTYPE_AUTH);
 447                 blobmsg_add_u64(&payloadbuf, "validfrom", tv.tv_sec);
 448                 if (!revoker) {
 449                         blobmsg_add_u64(&payloadbuf, "expiresat", tv.tv_sec + 60 * 60 * 24 * 365);
 450                         blobmsg_add_string(&payloadbuf, "pubkey", pkb);
 451                 } else {
 452                         blobmsg_add_string(&payloadbuf, "fingerprint", pkfp);
 453                 }
 454
 455                 blobmsg_close_table(&payloadbuf, c);
 456
 457                 snprintf(fname, sizeof(fname) - 1, "%s/%s", tmpdir, revoker?"revoker":"payload");
 458                 write_file(fname, blob_data(payloadbuf.head), blob_len(payloadbuf.head), false);
 459
 460                 snprintf(sfname, sizeof(sfname) - 1, "%s/%s", tmpdir, revoker?"revoker.sig":"payload.sig");
 461                 if (usign_s(fname, seckeyfile, sfname, quiet))
 462                         return 1;
 463
 464                 sigf = fopen(sfname, "r");
 465                 if (!sigf)
 466                         return 1;
 467
 468                 siglen = fread(sigb, 1, 1024, sigf);
 469                 if (siglen < 1)
 470                         return 1;
 471
 472                 sigb[siglen] = '\0';
 473                 fclose(sigf);
 474
 475                 unlink(fname);
 476                 unlink(sfname);
 477
 478                 blob_buf_init(&certbuf, 0);
 479                 blob_put(&certbuf, CERT_ATTR_SIGNATURE, sigb, siglen);
 480                 blob_put(&certbuf, CERT_ATTR_PAYLOAD, blob_data(payloadbuf.head), blob_len(payloadbuf.head));
 481                 snprintf(fname, sizeof(fname) - 1, "%s%s", certfile, revoker?".revoke":"");
 482                 write_file(fname, certbuf.head, blob_raw_len(certbuf.head), true);
 483                 blob_buf_free(&certbuf);
 484                 blob_buf_free(&payloadbuf);
 485
 486                 revoker--;
 487         }
 488
 489         rmdir(tmpdir);
 490
 491         return 0;
 492 }
 493 #endif
 494
 495 /* process revoker certificate */
 496 static int cert_process_revoker(const char *certfile, const char *pubkeydir) {
 497         static LIST_HEAD(certchain);
 498         struct cert_object *cobj;
 499         struct blob_attr *containertb[CERT_CT_ATTR_MAX];
 500         struct blob_attr *payloadtb[CERT_PL_ATTR_MAX];
 501         struct stat st;
 502         struct timeval tv;
 503         uint64_t validfrom;
 504         uint32_t certtype;
 505         char *fingerprint;
 506         char rfname[512];
 507
 508         int ret;
 509
 510         if (cert_load(certfile, &certchain)) {
 511                 DPRINTF("cannot parse cert\n");
 512                 return 1;
 513         }
 514
 515         gettimeofday(&tv, NULL);
 516
 517         list_for_each_entry(cobj, &certchain, list) {
 518                 if (!cobj->cert[CERT_ATTR_PAYLOAD])
 519                         return 2;
 520
 521                 /* blob has payload, verify that using signature */
 522                 ret = cert_verify_blob(cobj->cert, NULL, pubkeydir);
 523                 if (ret)
 524                         return ret;
 525
 526                 blobmsg_parse(cert_cont_policy,
 527                               ARRAY_SIZE(cert_cont_policy),
 528                               containertb,
 529                               blob_data(cobj->cert[CERT_ATTR_PAYLOAD]),
 530                               blob_len(cobj->cert[CERT_ATTR_PAYLOAD]));
 531                 if (!containertb[CERT_CT_ATTR_PAYLOAD]) {
 532                         DPRINTF("no ucert in signed payload\n");
 533                         return 2;
 534                 }
 535
 536                 blobmsg_parse(cert_payload_policy,
 537                               ARRAY_SIZE(cert_payload_policy),
 538                               payloadtb,
 539                               blobmsg_data(containertb[CERT_CT_ATTR_PAYLOAD]),
 540                               blobmsg_data_len(containertb[CERT_CT_ATTR_PAYLOAD]));
 541
 542                 if (!payloadtb[CERT_PL_ATTR_CERTTYPE] ||
 543                     !payloadtb[CERT_PL_ATTR_VALIDFROMTIME] ||
 544                     !payloadtb[CERT_PL_ATTR_KEY_FINGERPRINT]) {
 545                         DPRINTF("missing mandatory ucert attributes\n");
 546                         return 2;
 547                 }
 548
 549                 certtype = blobmsg_get_u32(payloadtb[CERT_PL_ATTR_CERTTYPE]);
 550                 validfrom = blobmsg_get_u64(payloadtb[CERT_PL_ATTR_VALIDFROMTIME]);
 551                 fingerprint = blobmsg_get_string(payloadtb[CERT_PL_ATTR_KEY_FINGERPRINT]);
 552
 553                 if (certtype != CERTTYPE_REVOKE) {
 554                         DPRINTF("wrong certificate type\n");
 555                         return 2;
 556                 }
 557
 558                 if (tv.tv_sec < validfrom) {
 559                         return 3;
 560                 }
 561
 562                 snprintf(rfname, sizeof(rfname)-1, "%s/%s", pubkeydir, fingerprint);
 563                 /* check if entry in pubkeydir exists */
 564                 if (stat(rfname, &st) == 0) {
 565                         if (_usign_key_is_revoked(fingerprint, pubkeydir)) {
 566                                 DPRINTF("existing revoker deadlink for key %s\n", fingerprint);
 567                                 continue;
 568                         };
 569
 570                         /* remove any other entry */
 571                         if (unlink(rfname))
 572                                 return -1;
 573                 }
 574
 575                 ret = symlink(".revoked.", rfname);
 576                 if (ret)
 577                         return ret;
 578
 579                 DPRINTF("created revoker deadlink for key %s\n", fingerprint);
 580         };
 581
 582         return ret;
 583 }
 584
 585 /* load and verify certfile (and message) */
 586 static int cert_verify(const char *certfile, const char *pubkeyfile, const char *pubkeydir, const char *msgfile) {
 587         static LIST_HEAD(certchain);
 588
 589         if (cert_load(certfile, &certchain)) {
 590                 DPRINTF("cannot parse cert\n");
 591                 return 1;
 592         }
 593
 594         return chain_verify(msgfile, pubkeyfile, pubkeydir, &certchain);
 595 }
 596
 597 /* output help */
 598 static int usage(const char *cmd)
 599 {
 600 #ifndef UCERT_STRIP_MESSAGES
 601         fprintf(stderr,
 602                 "Usage: %s <command> <options>\n"
 603                 "Commands:\n"
 604 #ifdef UCERT_FULL
 605                 "  -A:                  append signature (needs -c and -x)\n"
 606                 "  -D:                  dump (needs -c)\n"
 607                 "  -I:                  issue cert and revoker (needs -c and -p and -s)\n"
 608 #endif /* UCERT_FULL */
 609                 "  -R:                  process revoker certificate (needs -c and -P)\n"
 610                 "  -V:                  verify (needs -c and -p|-P, may have -m)\n"
 611                 "Options:\n"
 612                 "  -c <file>:           certificate file\n"
 613                 "  -m <file>:           message file (verify only)\n"
 614                 "  -p <file>:           public key file\n"
 615                 "  -P <path>:           public key directory (verify only)\n"
 616                 "  -q:                  quiet (do not print verification result, use return code only)\n"
 617 #ifdef UCERT_FULL
 618                 "  -s <file>:           secret key file (issue only)\n"
 619                 "  -x <file>:           signature file (append only)\n"
 620 #endif /* UCERT_FULL */
 621                 "\n",
 622                 cmd);
 623 #endif /* UCERT_STRIP_MESSAGES */
 624         return 1;
 625 }
 626
 627 /* parse command line options and call functions */
 628 int main(int argc, char *argv[]) {
 629         int ch;
 630         const char *msgfile = NULL;
 631         const char *pubkeyfile = NULL;
 632         const char *pubkeydir = NULL;
 633         const char *certfile = NULL;
 634 #ifdef UCERT_FULL
 635         const char *sigfile = NULL;
 636         const char *seckeyfile = NULL;
 637 #endif
 638
 639         quiet = false;
 640         while ((ch = getopt(argc, argv,
 641                 "RVc:m:p:P:q"
 642 #ifdef UCERT_FULL
 643                 "ADIs:x:"
 644 #endif
 645                )) != -1) {
 646                 switch (ch) {
 647 #ifdef UCERT_FULL
 648                 case 'A':
 649                         if (cmd != CMD_NONE)
 650                                 return usage(argv[0]);
 651                         cmd = CMD_APPEND;
 652                         break;
 653                 case 'D':
 654                         if (cmd != CMD_NONE)
 655                                 return usage(argv[0]);
 656                         cmd = CMD_DUMP;
 657                         break;
 658                 case 'I':
 659                         if (cmd != CMD_NONE)
 660                                 return usage(argv[0]);
 661                         cmd = CMD_ISSUE;
 662                         break;
 663 #endif
 664                 case 'R':
 665                         if (cmd != CMD_NONE)
 666                                 return usage(argv[0]);
 667                         cmd = CMD_REVOKE;
 668                         break;
 669                 case 'V':
 670                         if (cmd != CMD_NONE)
 671                                 return usage(argv[0]);
 672                         cmd = CMD_VERIFY;
 673                         break;
 674                 case 'c':
 675                         if (certfile || cmd == CMD_NONE)
 676                                 return usage(argv[0]);
 677                         certfile = optarg;
 678                         break;
 679                 case 'm':
 680                         if (msgfile || cmd != CMD_VERIFY)
 681                                 return usage(argv[0]);
 682                         msgfile = optarg;
 683                         break;
 684                 case 'p':
 685                         if (pubkeyfile || (cmd != CMD_VERIFY && cmd != CMD_ISSUE) || cmd == CMD_NONE)
 686                                 return usage(argv[0]);
 687                         pubkeyfile = optarg;
 688                         break;
 689                 case 'P':
 690                         if (pubkeydir || (cmd != CMD_VERIFY && cmd != CMD_REVOKE) || cmd == CMD_NONE)
 691                                 return usage(argv[0]);
 692                         pubkeydir = optarg;
 693                         break;
 694                 case 'q':
 695                         if (quiet)
 696                                 return usage(argv[0]);
 697                         quiet = true;
 698                         break;
 699 #ifdef UCERT_FULL
 700                 case 's':
 701                         if (seckeyfile || cmd != CMD_ISSUE || cmd == CMD_NONE)
 702                                 return usage(argv[0]);
 703                         seckeyfile = optarg;
 704                         break;
 705                 case 'x':
 706                         if (sigfile || cmd != CMD_APPEND || cmd == CMD_NONE)
 707                                 return usage(argv[0]);
 708                         sigfile = optarg;
 709                         break;
 710 #endif
 711                 default:
 712                         return usage(argv[0]);
 713                 }
 714         }
 715
 716         switch (cmd) {
 717 #ifdef UCERT_FULL
 718         case CMD_APPEND:
 719                 if (certfile && sigfile)
 720                         return cert_append(certfile, sigfile);
 721                 else
 722                         return usage(argv[0]);
 723         case CMD_DUMP:
 724                 if (certfile)
 725                         return cert_dump(certfile);
 726                 else
 727                         return usage(argv[0]);
 728         case CMD_ISSUE:
 729                 if (certfile && pubkeyfile && seckeyfile)
 730                         return cert_issue(certfile, pubkeyfile, seckeyfile);
 731                 else
 732                         return usage(argv[0]);
 733 #endif
 734         case CMD_REVOKE:
 735                 if (certfile && pubkeydir)
 736                         return cert_process_revoker(certfile, pubkeydir);
 737                 else
 738                         return usage(argv[0]);
 739         case CMD_VERIFY:
 740                 if (certfile && (pubkeyfile || pubkeydir))
 741                         return cert_verify(certfile, pubkeyfile, pubkeydir, msgfile);
 742                 else
 743                         return usage(argv[0]);
 744         default:
 745                 return usage(argv[0]);
 746         }
 747
 748         /* unreachable */
 749         return usage(argv[0]);
 750 }