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
  51 enum cert_attr {
  52         CERT_ATTR_SIGNATURE,
  53         CERT_ATTR_PAYLOAD,
  54         CERT_ATTR_MAX
  55 };
  56
  57 static const struct blob_attr_info cert_policy[CERT_ATTR_MAX] = {
  58         [CERT_ATTR_SIGNATURE] = { .type = BLOB_ATTR_BINARY },
  59         [CERT_ATTR_PAYLOAD] = { .type = BLOB_ATTR_NESTED },
  60 };
  61
  62 enum cert_cont_attr {
  63         CERT_CT_ATTR_PAYLOAD,
  64         CERT_CT_ATTR_MAX
  65 };
  66
  67 static const struct blobmsg_policy cert_cont_policy[CERT_CT_ATTR_MAX] = {
  68         [CERT_CT_ATTR_PAYLOAD] = { .name = "ucert", .type = BLOBMSG_TYPE_TABLE },
  69 };
  70
  71 enum cert_payload_attr {
  72         CERT_PL_ATTR_CERTTYPE,
  73         CERT_PL_ATTR_CERTID,
  74         CERT_PL_ATTR_VALIDFROMTIME,
  75         CERT_PL_ATTR_EXPIRETIME,
  76         CERT_PL_ATTR_PUBKEY,
  77         CERT_PL_ATTR_KEY_FINGERPRINT,
  78         CERT_PL_ATTR_MAX
  79 };
  80
  81 enum certtype_id {
  82         CERTTYPE_UNSPEC,
  83         CERTTYPE_AUTH,
  84         CERTTYPE_REVOKE
  85 };
  86
  87 static const struct blobmsg_policy cert_payload_policy[CERT_PL_ATTR_MAX] = {
  88         [CERT_PL_ATTR_CERTTYPE] = { .name = "certtype", .type = BLOBMSG_TYPE_INT32 },
  89         [CERT_PL_ATTR_CERTID] = { .name = "certid", .type = BLOBMSG_TYPE_INT32 },
  90         [CERT_PL_ATTR_VALIDFROMTIME] = { .name = "validfrom", .type = BLOBMSG_TYPE_INT64 },
  91         [CERT_PL_ATTR_EXPIRETIME] = { .name = "expiresat", .type = BLOBMSG_TYPE_INT64 },
  92         [CERT_PL_ATTR_PUBKEY] = { .name = "pubkey", .type = BLOBMSG_TYPE_STRING },
  93         [CERT_PL_ATTR_KEY_FINGERPRINT] = { .name = "fingerprint", .type = BLOBMSG_TYPE_STRING },
  94 };
  95
  96 struct cert_object {
  97         struct list_head list;
  98         struct blob_attr *cert[CERT_ATTR_MAX];
  99 };
 100
 101 static int write_file(const char *filename, void *buf, size_t len, bool append) {
 102         FILE *f;
 103         size_t outlen;
 104
 105         f = fopen(filename, append?"a":"w");
 106         if (!f)
 107                 return 1;
 108
 109         outlen = fwrite(buf, 1, len, f);
 110         fclose(f);
 111         return (outlen == len);
 112 }
 113
 114 static int cert_load(const char *certfile, struct list_head *chain) {
 115         FILE *f;
 116         struct blob_attr *certtb[CERT_ATTR_MAX];
 117         struct blob_attr *bufpt;
 118         struct cert_object *cobj;
 119         char filebuf[CERT_BUF_LEN];
 120         int ret = 0, pret = 0;
 121         int len, pos = 0;
 122
 123         f = fopen(certfile, "r");
 124         if (!f)
 125                 return 1;
 126
 127         len = fread(&filebuf, 1, CERT_BUF_LEN - 1, f);
 128         if (len < 64)
 129                 return 1;
 130
 131         ret = ferror(f) || !feof(f);
 132         fclose(f);
 133         if (ret)
 134                 return 1;
 135
 136         bufpt = (struct blob_attr *)filebuf;
 137         do {
 138                 pret = blob_parse(bufpt, certtb, cert_policy, CERT_ATTR_MAX);
 139                 if (pret <= 0)
 140                         /* no attributes found */
 141                         break;
 142
 143                 if (pos + blob_pad_len(bufpt) > len)
 144                         /* blob exceeds filebuffer */
 145                         break;
 146                 else
 147                         pos += blob_pad_len(bufpt);
 148
 149                 cobj = calloc(1, sizeof(*cobj));
 150                 memcpy(cobj->cert, &certtb, sizeof(certtb));
 151                 list_add_tail(&cobj->list, chain);
 152                 ret += pret;
 153                 bufpt = blob_next(bufpt);
 154         /* repeat parsing while there is still enough remaining data in buffer */
 155         } while(len > pos + sizeof(struct blob_attr));
 156
 157         return (ret <= 0);
 158 }
 159
 160 static int cert_append(const char *certfile, const char *sigfile) {
 161         FILE *fs;
 162         char filebuf[CERT_BUF_LEN];
 163         struct blob_buf sigbuf;
 164         struct stat st;
 165         int len;
 166         int ret;
 167
 168         if (stat(certfile, &st) != 0) {
 169                 fprintf(stderr, "certfile %s doesn't exist, can't append.\n", certfile);
 170                 return -1;
 171         }
 172
 173         fs = fopen(sigfile, "r");
 174         if (!fs)
 175                 return 1;
 176
 177         len = fread(&filebuf, 1, CERT_BUF_LEN - 1, fs);
 178         ret = ferror(fs) || !feof(fs) || (len < 64);
 179         fclose(fs);
 180         if (ret)
 181                 return 1;
 182
 183         blob_buf_init(&sigbuf, 0);
 184         blob_put(&sigbuf, CERT_ATTR_SIGNATURE, filebuf, len);
 185         ret = write_file(certfile, sigbuf.head, blob_raw_len(sigbuf.head), true);
 186         blob_buf_free(&sigbuf);
 187         return ret;
 188 }
 189
 190 static int cert_verify_blob(struct blob_attr *cert[CERT_ATTR_MAX],
 191                        const char *pubkeyfile, const char *pubkeydir) {
 192         int i;
 193         char msgfname[256], sigfname[256];
 194         int ret;
 195         char tmpdir[] = "/tmp/ucert-XXXXXX";
 196
 197         if (mkdtemp(tmpdir) == NULL)
 198                 return errno;
 199
 200         snprintf(msgfname, sizeof(msgfname) - 1, "%s/%s", tmpdir, "payload");
 201         snprintf(sigfname, sizeof(sigfname) - 1, "%s/%s", tmpdir, "payload.sig");
 202
 203         for (i = 0; i < CERT_ATTR_MAX; i++) {
 204                 struct blob_attr *v = cert[i];
 205
 206                 if (!v)
 207                         break;
 208
 209                 switch(cert_policy[i].type) {
 210                 case BLOB_ATTR_BINARY:
 211                         write_file(sigfname, blob_data(v), blob_len(v), false);
 212                         break;
 213                 case BLOB_ATTR_NESTED:
 214                         write_file(msgfname, blob_data(v), blob_len(v), false);
 215                         break;
 216                 }
 217         }
 218
 219         ret = usign_v(msgfname, pubkeyfile, pubkeydir, sigfname, quiet);
 220
 221         unlink(msgfname);
 222         unlink(sigfname);
 223         rmdir(tmpdir);
 224
 225         return ret;
 226 }
 227
 228 static int chain_verify(const char *msgfile, const char *pubkeyfile,
 229                         const char *pubkeydir, struct list_head *chain) {
 230         struct cert_object *cobj;
 231         struct blob_attr *containertb[CERT_CT_ATTR_MAX];
 232         struct blob_attr *payloadtb[CERT_PL_ATTR_MAX];
 233         char tmpdir[] = "/tmp/ucert-XXXXXX";
 234         char chainedpubkey[256] = {0};
 235         char extsigfile[256] = {0};
 236         int ret = 1;
 237         int checkmsg = 0;
 238
 239         if (mkdtemp(tmpdir) == NULL)
 240                 return errno;
 241
 242         if (msgfile)
 243                 checkmsg = -1;
 244
 245         list_for_each_entry(cobj, chain, list) {
 246                 /* blob has payload, verify that using signature */
 247                 if (cobj->cert[CERT_ATTR_PAYLOAD]) {
 248                         struct timeval tv;
 249                         uint64_t validfrom;
 250                         uint64_t expiresat;
 251                         uint32_t certtype;
 252
 253                         ret = cert_verify_blob(cobj->cert, chainedpubkey[0]?chainedpubkey:pubkeyfile, pubkeydir);
 254                         if (ret)
 255                                 goto clean_and_return;
 256
 257                         blobmsg_parse(cert_cont_policy,
 258                                       ARRAY_SIZE(cert_cont_policy),
 259                                       containertb,
 260                                       blob_data(cobj->cert[CERT_ATTR_PAYLOAD]),
 261                                       blob_len(cobj->cert[CERT_ATTR_PAYLOAD]));
 262                         if (!containertb[CERT_CT_ATTR_PAYLOAD]) {
 263                                 ret = 1;
 264                                 fprintf(stderr, "no ucert in signed payload\n");
 265                                 goto clean_and_return;
 266                         }
 267                         blobmsg_parse(cert_payload_policy,
 268                                       ARRAY_SIZE(cert_payload_policy),
 269                                       payloadtb,
 270                                       blobmsg_data(containertb[CERT_CT_ATTR_PAYLOAD]),
 271                                       blobmsg_data_len(containertb[CERT_CT_ATTR_PAYLOAD]));
 272
 273                         if (!payloadtb[CERT_PL_ATTR_CERTTYPE] ||
 274                             !payloadtb[CERT_PL_ATTR_VALIDFROMTIME] ||
 275                             !payloadtb[CERT_PL_ATTR_EXPIRETIME] ||
 276                             !payloadtb[CERT_PL_ATTR_PUBKEY]) {
 277                                 ret = 1;
 278                                 fprintf(stderr, "missing mandatory ucert attributes\n");
 279                                 goto clean_and_return;
 280                         }
 281                         certtype = blobmsg_get_u32(payloadtb[CERT_PL_ATTR_CERTTYPE]);
 282                         validfrom = blobmsg_get_u64(payloadtb[CERT_PL_ATTR_VALIDFROMTIME]);
 283                         expiresat = blobmsg_get_u64(payloadtb[CERT_PL_ATTR_EXPIRETIME]);
 284
 285                         if (certtype != CERTTYPE_AUTH) {
 286                                 ret = 2;
 287                                 fprintf(stderr, "wrong certificate type\n");
 288                                 goto clean_and_return;
 289                         }
 290
 291                         gettimeofday(&tv, NULL);
 292                         if (tv.tv_sec < validfrom ||
 293                             tv.tv_sec >= expiresat) {
 294                                 ret = 3;
 295                                 fprintf(stderr, "certificate expired\n");
 296                                 goto clean_and_return;
 297                         }
 298
 299                         snprintf(chainedpubkey, sizeof(chainedpubkey) - 1, "%s/%s", tmpdir, "chained-pubkey");
 300                         write_file(chainedpubkey,
 301                                    blobmsg_data(payloadtb[CERT_PL_ATTR_PUBKEY]),
 302                                    blobmsg_data_len(payloadtb[CERT_PL_ATTR_PUBKEY]),
 303                                    false);
 304                 } else {
 305                 /* blob doesn't have payload, verify message using signature */
 306                         if (msgfile) {
 307                                 snprintf(extsigfile, sizeof(extsigfile) - 1, "%s/%s", tmpdir, "ext-sig");
 308                                 write_file(extsigfile,
 309                                            blob_data(cobj->cert[CERT_ATTR_SIGNATURE]),
 310                                            blob_len(cobj->cert[CERT_ATTR_SIGNATURE]),
 311                                            false);
 312                                 checkmsg = ret = usign_v(msgfile,
 313                                               chainedpubkey[0]?chainedpubkey:pubkeyfile,
 314                                               pubkeydir, extsigfile, quiet);
 315                                 unlink(extsigfile);
 316                         } else {
 317                                 fprintf(stderr, "stray trailing signature without anything to verify!\n");
 318                                 ret = 1;
 319                         };
 320                 }
 321         }
 322
 323         if (checkmsg == -1)
 324                 fprintf(stderr, "missing signature to verify message!\n");
 325
 326 clean_and_return:
 327         if (chainedpubkey[0])
 328                 unlink(chainedpubkey);
 329         rmdir(tmpdir);
 330         return ret | checkmsg;
 331 }
 332
 333 static void cert_dump_blob(struct blob_attr *cert[CERT_ATTR_MAX]) {
 334         int i;
 335
 336         for (i = 0; i < CERT_ATTR_MAX; i++) {
 337                 struct blob_attr *v = cert[i];
 338
 339                 if (!v)
 340                         continue;
 341
 342                 switch(cert_policy[i].type) {
 343                 case BLOB_ATTR_BINARY:
 344                         fprintf(stdout, "signature:\n---\n%s---\n", (char *) blob_data(v));
 345                         break;
 346                 case BLOB_ATTR_NESTED:
 347                         fprintf(stdout, "payload:\n---\n%s\n---\n", blobmsg_format_json_indent(blob_data(v), false, 0));
 348                         break;
 349                 }
 350         }
 351 }
 352
 353 static int cert_dump(const char *certfile) {
 354         struct cert_object *cobj;
 355         static LIST_HEAD(certchain);
 356         unsigned int count = 0;
 357
 358         if (cert_load(certfile, &certchain)) {
 359                 fprintf(stderr, "cannot parse cert\n");
 360                 return 1;
 361         }
 362
 363         list_for_each_entry(cobj, &certchain, list) {
 364                 fprintf(stderr, "=== CHAIN ELEMENT %02u ===\n", ++count);
 365                 cert_dump_blob(cobj->cert);
 366                 fprintf(stderr, "========================\n");
 367         }
 368
 369         return 0;
 370 }
 371
 372 static int cert_issue(const char *certfile, const char *pubkeyfile, const char *seckeyfile) {
 373         struct blob_buf certbuf;
 374         struct blob_buf payloadbuf;
 375         struct timeval tv;
 376         struct stat st;
 377         int pklen, siglen;
 378         int revoker = 1;
 379         void *c;
 380         FILE *pkf, *sigf;
 381         char pkb[512];
 382         char sigb[512];
 383         char fname[256], sfname[256];
 384         char pkfp[17];
 385         char tmpdir[] = "/tmp/ucert-XXXXXX";
 386
 387         if (stat(certfile, &st) == 0) {
 388                 fprintf(stderr, "certfile %s exists, won't overwrite.\n", certfile);
 389                 return -1;
 390         }
 391
 392         pkf = fopen(pubkeyfile, "r");
 393         if (!pkf)
 394                 return -1;
 395
 396         pklen = fread(pkb, 1, 512, pkf);
 397         pkb[pklen] = '\0';
 398
 399         if (pklen < 32)
 400                 return -1;
 401
 402         fclose(pkf);
 403
 404         if (usign_f_pubkey(pkfp, pubkeyfile))
 405                 return -1;
 406
 407         gettimeofday(&tv, NULL);
 408
 409         if (mkdtemp(tmpdir) == NULL)
 410                 return errno;
 411
 412         while (revoker >= 0) {
 413                 blob_buf_init(&payloadbuf, 0);
 414                 c = blobmsg_open_table(&payloadbuf, "ucert");
 415                 blobmsg_add_u32(&payloadbuf, "certtype", revoker?CERTTYPE_REVOKE:CERTTYPE_AUTH);
 416                 blobmsg_add_u64(&payloadbuf, "validfrom", tv.tv_sec);
 417                 if (!revoker) {
 418                         blobmsg_add_u64(&payloadbuf, "expiresat", tv.tv_sec + 60 * 60 * 24 * 365);
 419                         blobmsg_add_string(&payloadbuf, "pubkey", pkb);
 420                 } else {
 421                         blobmsg_add_string(&payloadbuf, "fingerprint", pkfp);
 422                 }
 423
 424                 blobmsg_close_table(&payloadbuf, c);
 425
 426                 snprintf(fname, sizeof(fname) - 1, "%s/%s", tmpdir, revoker?"revoker":"payload");
 427                 write_file(fname, blob_data(payloadbuf.head), blob_len(payloadbuf.head), false);
 428
 429                 snprintf(sfname, sizeof(sfname) - 1, "%s/%s", tmpdir, revoker?"revoker.sig":"payload.sig");
 430                 if (usign_s(fname, seckeyfile, sfname, quiet))
 431                         return 1;
 432
 433                 sigf = fopen(sfname, "r");
 434                 if (!sigf)
 435                         return 1;
 436
 437                 siglen = fread(sigb, 1, 1024, sigf);
 438                 if (siglen < 1)
 439                         return 1;
 440
 441                 sigb[siglen] = '\0';
 442                 fclose(sigf);
 443
 444                 unlink(fname);
 445                 unlink(sfname);
 446
 447                 blob_buf_init(&certbuf, 0);
 448                 blob_put(&certbuf, CERT_ATTR_SIGNATURE, sigb, siglen);
 449                 blob_put(&certbuf, CERT_ATTR_PAYLOAD, blob_data(payloadbuf.head), blob_len(payloadbuf.head));
 450                 snprintf(fname, sizeof(fname) - 1, "%s%s", certfile, revoker?".revoke":"");
 451                 write_file(fname, certbuf.head, blob_raw_len(certbuf.head), false);
 452                 blob_buf_free(&certbuf);
 453                 blob_buf_free(&payloadbuf);
 454
 455                 revoker--;
 456         }
 457
 458         rmdir(tmpdir);
 459
 460         return 0;
 461 }
 462
 463 static int cert_process_revoker(const char *certfile) {
 464         fprintf(stderr, "not implemented\n");
 465         return 1;
 466 }
 467
 468 static int cert_verify(const char *certfile, const char *pubkeyfile, const char *pubkeydir, const char *msgfile) {
 469         static LIST_HEAD(certchain);
 470
 471         if (cert_load(certfile, &certchain)) {
 472                 fprintf(stderr, "cannot parse cert\n");
 473                 return 1;
 474         }
 475
 476         return chain_verify(msgfile, pubkeyfile, pubkeydir, &certchain);
 477 }
 478
 479 static int usage(const char *cmd)
 480 {
 481         fprintf(stderr,
 482                 "Usage: %s <command> <options>\n"
 483                 "Commands:\n"
 484                 "  -A:                  append (needs -c and -x)\n"
 485                 "  -D:                  dump\n"
 486                 "  -I:                  issue cert and revoker (needs -c and -p and -s)\n"
 487                 "  -R:                  process revoker certificate (needs -c)\n"
 488                 "  -V:                  verify (needs -c and -p|-P)\n"
 489                 "Options:\n"
 490                 "  -c <file>:           certificate file\n"
 491                 "  -m <file>:           message file (verify only)\n"
 492                 "  -p <file>:           public key file\n"
 493                 "  -P <path>:           public key directory (verify only)\n"
 494                 "  -q:                  quiet (do not print verification result, use return code only)\n"
 495                 "  -s <file>:           secret key file (issue only)\n"
 496                 "  -x <file>:           signature file\n"
 497                 "\n",
 498                 cmd);
 499         return 1;
 500 }
 501
 502 int main(int argc, char *argv[]) {
 503         int ch;
 504         const char *msgfile = NULL;
 505         const char *sigfile = NULL;
 506         const char *pubkeyfile = NULL;
 507         const char *pubkeydir = NULL;
 508         const char *certfile = NULL;
 509         const char *seckeyfile = NULL;
 510
 511         quiet = false;
 512         while ((ch = getopt(argc, argv, "ADIRVc:m:p:P:qs:x:")) != -1) {
 513                 switch (ch) {
 514                 case 'A':
 515                         cmd = CMD_APPEND;
 516                         break;
 517                 case 'D':
 518                         cmd = CMD_DUMP;
 519                         break;
 520                 case 'I':
 521                         cmd = CMD_ISSUE;
 522                         break;
 523                 case 'R':
 524                         cmd = CMD_REVOKE;
 525                         break;
 526                 case 'V':
 527                         cmd = CMD_VERIFY;
 528                         break;
 529                 case 'c':
 530                         certfile = optarg;
 531                         break;
 532                 case 'm':
 533                         msgfile = optarg;
 534                         break;
 535                 case 'p':
 536                         pubkeyfile = optarg;
 537                         break;
 538                 case 'P':
 539                         pubkeydir = optarg;
 540                         break;
 541                 case 'q':
 542                         quiet = true;
 543                         break;
 544                 case 's':
 545                         seckeyfile = optarg;
 546                         break;
 547                 case 'x':
 548                         sigfile = optarg;
 549                         break;
 550                 default:
 551                         return usage(argv[0]);
 552                 }
 553         }
 554
 555         switch (cmd) {
 556         case CMD_APPEND:
 557                 if (certfile && sigfile)
 558                         return cert_append(certfile, sigfile);
 559                 else
 560                         return usage(argv[0]);
 561         case CMD_DUMP:
 562                 if (certfile)
 563                         return cert_dump(certfile);
 564                 else
 565                         return usage(argv[0]);
 566         case CMD_ISSUE:
 567                 if (certfile && pubkeyfile && seckeyfile)
 568                         return cert_issue(certfile, pubkeyfile, seckeyfile);
 569                 else
 570                         return usage(argv[0]);
 571         case CMD_REVOKE:
 572                 if (certfile)
 573                         return cert_process_revoker(certfile);
 574                 else
 575                         return usage(argv[0]);
 576         case CMD_VERIFY:
 577                 if (certfile && (pubkeyfile || pubkeydir))
 578                         return cert_verify(certfile, pubkeyfile, pubkeydir, msgfile);
 579                 else
 580                         return usage(argv[0]);
 581         case CMD_NONE:
 582                 return usage(argv[0]);
 583         }
 584
 585         /* unreachable */
 586         return usage(argv[0]);
 587 }