usr.sbin/pkg/pkg.c

   1 /*-
   2  * SPDX-License-Identifier: BSD-2-Clause
   3  *
   4  * Copyright (c) 2012-2014 Baptiste Daroussin <bapt@FreeBSD.org>
   5  * Copyright (c) 2013 Bryan Drewery <bdrewery@FreeBSD.org>
   6  * All rights reserved.
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted provided that the following conditions
  10  * are met:
  11  * 1. Redistributions of source code must retain the above copyright
  12  *    notice, this list of conditions and the following disclaimer.
  13  * 2. Redistributions in binary form must reproduce the above copyright
  14  *    notice, this list of conditions and the following disclaimer in the
  15  *    documentation and/or other materials provided with the distribution.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  27  * SUCH DAMAGE.
  28  */
  29
  30 #include <sys/cdefs.h>
  31 __FBSDID("$FreeBSD$");
  32
  33 #include <sys/param.h>
  34 #include <sys/queue.h>
  35 #include <sys/types.h>
  36 #include <sys/wait.h>
  37
  38 #include <archive.h>
  39 #include <archive_entry.h>
  40 #include <dirent.h>
  41 #include <err.h>
  42 #include <errno.h>
  43 #include <fcntl.h>
  44 #include <fetch.h>
  45 #include <getopt.h>
  46 #include <libutil.h>
  47 #include <paths.h>
  48 #include <stdbool.h>
  49 #include <stdlib.h>
  50 #include <stdio.h>
  51 #include <string.h>
  52 #include <ucl.h>
  53
  54 #include <openssl/err.h>
  55 #include <openssl/ssl.h>
  56
  57 #include "dns_utils.h"
  58 #include "config.h"
  59 #include "hash.h"
  60
  61 struct sig_cert {
  62         char *name;
  63         unsigned char *sig;
  64         int siglen;
  65         unsigned char *cert;
  66         int certlen;
  67         bool trusted;
  68 };
  69
  70 struct pubkey {
  71         unsigned char *sig;
  72         int siglen;
  73 };
  74
  75 typedef enum {
  76         HASH_UNKNOWN,
  77         HASH_SHA256,
  78 } hash_t;
  79
  80 struct fingerprint {
  81         hash_t type;
  82         char *name;
  83         char hash[BUFSIZ];
  84         STAILQ_ENTRY(fingerprint) next;
  85 };
  86
  87 static const char *bootstrap_names []  = {
  88         "pkg.pkg",
  89         "pkg.txz",
  90         NULL
  91 };
  92
  93 STAILQ_HEAD(fingerprint_list, fingerprint);
  94
  95 static int debug;
  96
  97 static int
  98 extract_pkg_static(int fd, char *p, int sz)
  99 {
 100         struct archive *a;
 101         struct archive_entry *ae;
 102         char *end;
 103         int ret, r;
 104
 105         ret = -1;
 106         a = archive_read_new();
 107         if (a == NULL) {
 108                 warn("archive_read_new");
 109                 return (ret);
 110         }
 111         archive_read_support_filter_all(a);
 112         archive_read_support_format_tar(a);
 113
 114         if (lseek(fd, 0, 0) == -1) {
 115                 warn("lseek");
 116                 goto cleanup;
 117         }
 118
 119         if (archive_read_open_fd(a, fd, 4096) != ARCHIVE_OK) {
 120                 warnx("archive_read_open_fd: %s", archive_error_string(a));
 121                 goto cleanup;
 122         }
 123
 124         ae = NULL;
 125         while ((r = archive_read_next_header(a, &ae)) == ARCHIVE_OK) {
 126                 end = strrchr(archive_entry_pathname(ae), '/');
 127                 if (end == NULL)
 128                         continue;
 129
 130                 if (strcmp(end, "/pkg-static") == 0) {
 131                         r = archive_read_extract(a, ae,
 132                             ARCHIVE_EXTRACT_OWNER | ARCHIVE_EXTRACT_PERM |
 133                             ARCHIVE_EXTRACT_TIME | ARCHIVE_EXTRACT_ACL |
 134                             ARCHIVE_EXTRACT_FFLAGS | ARCHIVE_EXTRACT_XATTR);
 135                         strlcpy(p, archive_entry_pathname(ae), sz);
 136                         break;
 137                 }
 138         }
 139
 140         if (r == ARCHIVE_OK)
 141                 ret = 0;
 142         else
 143                 warnx("failed to extract pkg-static: %s",
 144                     archive_error_string(a));
 145
 146 cleanup:
 147         archive_read_free(a);
 148         return (ret);
 149
 150 }
 151
 152 static int
 153 install_pkg_static(const char *path, const char *pkgpath, bool force)
 154 {
 155         int pstat;
 156         pid_t pid;
 157
 158         switch ((pid = fork())) {
 159         case -1:
 160                 return (-1);
 161         case 0:
 162                 if (force)
 163                         execl(path, "pkg-static", "add", "-f", pkgpath,
 164                             (char *)NULL);
 165                 else
 166                         execl(path, "pkg-static", "add", pkgpath,
 167                             (char *)NULL);
 168                 _exit(1);
 169         default:
 170                 break;
 171         }
 172
 173         while (waitpid(pid, &pstat, 0) == -1)
 174                 if (errno != EINTR)
 175                         return (-1);
 176
 177         if (WEXITSTATUS(pstat))
 178                 return (WEXITSTATUS(pstat));
 179         else if (WIFSIGNALED(pstat))
 180                 return (128 & (WTERMSIG(pstat)));
 181         return (pstat);
 182 }
 183
 184 static int
 185 fetch_to_fd(const char *url, char *path, const char *fetchOpts)
 186 {
 187         struct url *u;
 188         struct dns_srvinfo *mirrors, *current;
 189         struct url_stat st;
 190         FILE *remote;
 191         /* To store _https._tcp. + hostname + \0 */
 192         int fd;
 193         int retry, max_retry;
 194         ssize_t r;
 195         char buf[10240];
 196         char zone[MAXHOSTNAMELEN + 13];
 197         static const char *mirror_type = NULL;
 198
 199         max_retry = 3;
 200         current = mirrors = NULL;
 201         remote = NULL;
 202
 203         if (mirror_type == NULL && config_string(MIRROR_TYPE, &mirror_type)
 204             != 0) {
 205                 warnx("No MIRROR_TYPE defined");
 206                 return (-1);
 207         }
 208
 209         if ((fd = mkstemp(path)) == -1) {
 210                 warn("mkstemp()");
 211                 return (-1);
 212         }
 213
 214         retry = max_retry;
 215
 216         if ((u = fetchParseURL(url)) == NULL) {
 217                 warn("fetchParseURL('%s')", url);
 218                 return (-1);
 219         }
 220
 221         while (remote == NULL) {
 222                 if (retry == max_retry) {
 223                         if (strcmp(u->scheme, "file") != 0 &&
 224                             strcasecmp(mirror_type, "srv") == 0) {
 225                                 snprintf(zone, sizeof(zone),
 226                                     "_%s._tcp.%s", u->scheme, u->host);
 227                                 mirrors = dns_getsrvinfo(zone);
 228                                 current = mirrors;
 229                         }
 230                 }
 231
 232                 if (mirrors != NULL) {
 233                         strlcpy(u->host, current->host, sizeof(u->host));
 234                         u->port = current->port;
 235                 }
 236
 237                 remote = fetchXGet(u, &st, fetchOpts);
 238                 if (remote == NULL) {
 239                         --retry;
 240                         if (retry <= 0)
 241                                 goto fetchfail;
 242                         if (mirrors != NULL) {
 243                                 current = current->next;
 244                                 if (current == NULL)
 245                                         current = mirrors;
 246                         }
 247                 }
 248         }
 249
 250         while ((r = fread(buf, 1, sizeof(buf), remote)) > 0) {
 251                 if (write(fd, buf, r) != r) {
 252                         warn("write()");
 253                         goto fetchfail;
 254                 }
 255         }
 256
 257         if (r != 0) {
 258                 warn("An error occurred while fetching pkg(8)");
 259                 goto fetchfail;
 260         }
 261
 262         if (ferror(remote))
 263                 goto fetchfail;
 264
 265         goto cleanup;
 266
 267 fetchfail:
 268         if (fd != -1) {
 269                 close(fd);
 270                 fd = -1;
 271                 unlink(path);
 272         }
 273
 274 cleanup:
 275         if (remote != NULL)
 276                 fclose(remote);
 277
 278         return fd;
 279 }
 280
 281 static struct fingerprint *
 282 parse_fingerprint(ucl_object_t *obj)
 283 {
 284         const ucl_object_t *cur;
 285         ucl_object_iter_t it = NULL;
 286         const char *function, *fp, *key;
 287         struct fingerprint *f;
 288         hash_t fct = HASH_UNKNOWN;
 289
 290         function = fp = NULL;
 291
 292         while ((cur = ucl_iterate_object(obj, &it, true))) {
 293                 key = ucl_object_key(cur);
 294                 if (cur->type != UCL_STRING)
 295                         continue;
 296                 if (strcasecmp(key, "function") == 0) {
 297                         function = ucl_object_tostring(cur);
 298                         continue;
 299                 }
 300                 if (strcasecmp(key, "fingerprint") == 0) {
 301                         fp = ucl_object_tostring(cur);
 302                         continue;
 303                 }
 304         }
 305
 306         if (fp == NULL || function == NULL)
 307                 return (NULL);
 308
 309         if (strcasecmp(function, "sha256") == 0)
 310                 fct = HASH_SHA256;
 311
 312         if (fct == HASH_UNKNOWN) {
 313                 warnx("Unsupported hashing function: %s", function);
 314                 return (NULL);
 315         }
 316
 317         f = calloc(1, sizeof(struct fingerprint));
 318         f->type = fct;
 319         strlcpy(f->hash, fp, sizeof(f->hash));
 320
 321         return (f);
 322 }
 323
 324 static void
 325 free_fingerprint_list(struct fingerprint_list* list)
 326 {
 327         struct fingerprint *fingerprint, *tmp;
 328
 329         STAILQ_FOREACH_SAFE(fingerprint, list, next, tmp) {
 330                 free(fingerprint->name);
 331                 free(fingerprint);
 332         }
 333         free(list);
 334 }
 335
 336 static struct fingerprint *
 337 load_fingerprint(const char *dir, const char *filename)
 338 {
 339         ucl_object_t *obj = NULL;
 340         struct ucl_parser *p = NULL;
 341         struct fingerprint *f;
 342         char path[MAXPATHLEN];
 343
 344         f = NULL;
 345
 346         snprintf(path, MAXPATHLEN, "%s/%s", dir, filename);
 347
 348         p = ucl_parser_new(0);
 349         if (!ucl_parser_add_file(p, path)) {
 350                 warnx("%s: %s", path, ucl_parser_get_error(p));
 351                 ucl_parser_free(p);
 352                 return (NULL);
 353         }
 354
 355         obj = ucl_parser_get_object(p);
 356
 357         if (obj->type == UCL_OBJECT)
 358                 f = parse_fingerprint(obj);
 359
 360         if (f != NULL)
 361                 f->name = strdup(filename);
 362
 363         ucl_object_unref(obj);
 364         ucl_parser_free(p);
 365
 366         return (f);
 367 }
 368
 369 static struct fingerprint_list *
 370 load_fingerprints(const char *path, int *count)
 371 {
 372         DIR *d;
 373         struct dirent *ent;
 374         struct fingerprint *finger;
 375         struct fingerprint_list *fingerprints;
 376
 377         *count = 0;
 378
 379         fingerprints = calloc(1, sizeof(struct fingerprint_list));
 380         if (fingerprints == NULL)
 381                 return (NULL);
 382         STAILQ_INIT(fingerprints);
 383
 384         if ((d = opendir(path)) == NULL) {
 385                 free(fingerprints);
 386
 387                 return (NULL);
 388         }
 389
 390         while ((ent = readdir(d))) {
 391                 if (strcmp(ent->d_name, ".") == 0 ||
 392                     strcmp(ent->d_name, "..") == 0)
 393                         continue;
 394                 finger = load_fingerprint(path, ent->d_name);
 395                 if (finger != NULL) {
 396                         STAILQ_INSERT_TAIL(fingerprints, finger, next);
 397                         ++(*count);
 398                 }
 399         }
 400
 401         closedir(d);
 402
 403         return (fingerprints);
 404 }
 405
 406 static EVP_PKEY *
 407 load_public_key_file(const char *file)
 408 {
 409         EVP_PKEY *pkey;
 410         BIO *bp;
 411         char errbuf[1024];
 412
 413         bp = BIO_new_file(file, "r");
 414         if (!bp)
 415                 errx(EXIT_FAILURE, "Unable to read %s", file);
 416
 417         if ((pkey = PEM_read_bio_PUBKEY(bp, NULL, NULL, NULL)) == NULL)
 418                 warnx("ici: %s", ERR_error_string(ERR_get_error(), errbuf));
 419
 420         BIO_free(bp);
 421
 422         return (pkey);
 423 }
 424
 425 static EVP_PKEY *
 426 load_public_key_buf(const unsigned char *cert, int certlen)
 427 {
 428         EVP_PKEY *pkey;
 429         BIO *bp;
 430         char errbuf[1024];
 431
 432         bp = BIO_new_mem_buf(__DECONST(void *, cert), certlen);
 433
 434         if ((pkey = PEM_read_bio_PUBKEY(bp, NULL, NULL, NULL)) == NULL)
 435                 warnx("%s", ERR_error_string(ERR_get_error(), errbuf));
 436
 437         BIO_free(bp);
 438
 439         return (pkey);
 440 }
 441
 442 static bool
 443 rsa_verify_cert(int fd, const char *sigfile, const unsigned char *key,
 444     int keylen, unsigned char *sig, int siglen)
 445 {
 446         EVP_MD_CTX *mdctx;
 447         EVP_PKEY *pkey;
 448         char *sha256;
 449         char errbuf[1024];
 450         bool ret;
 451
 452         sha256 = NULL;
 453         pkey = NULL;
 454         mdctx = NULL;
 455         ret = false;
 456
 457         SSL_load_error_strings();
 458
 459         /* Compute SHA256 of the package. */
 460         if (lseek(fd, 0, 0) == -1) {
 461                 warn("lseek");
 462                 goto cleanup;
 463         }
 464         if ((sha256 = sha256_fd(fd)) == NULL) {
 465                 warnx("Error creating SHA256 hash for package");
 466                 goto cleanup;
 467         }
 468
 469         if (sigfile != NULL) {
 470                 if ((pkey = load_public_key_file(sigfile)) == NULL) {
 471                         warnx("Error reading public key");
 472                         goto cleanup;
 473                 }
 474         } else {
 475                 if ((pkey = load_public_key_buf(key, keylen)) == NULL) {
 476                         warnx("Error reading public key");
 477                         goto cleanup;
 478                 }
 479         }
 480
 481         /* Verify signature of the SHA256(pkg) is valid. */
 482         if ((mdctx = EVP_MD_CTX_create()) == NULL) {
 483                 warnx("%s", ERR_error_string(ERR_get_error(), errbuf));
 484                 goto error;
 485         }
 486
 487         if (EVP_DigestVerifyInit(mdctx, NULL, EVP_sha256(), NULL, pkey) != 1) {
 488                 warnx("%s", ERR_error_string(ERR_get_error(), errbuf));
 489                 goto error;
 490         }
 491         if (EVP_DigestVerifyUpdate(mdctx, sha256, strlen(sha256)) != 1) {
 492                 warnx("%s", ERR_error_string(ERR_get_error(), errbuf));
 493                 goto error;
 494         }
 495
 496         if (EVP_DigestVerifyFinal(mdctx, sig, siglen) != 1) {
 497                 warnx("%s", ERR_error_string(ERR_get_error(), errbuf));
 498                 goto error;
 499         }
 500
 501         ret = true;
 502         printf("done\n");
 503         goto cleanup;
 504
 505 error:
 506         printf("failed\n");
 507
 508 cleanup:
 509         free(sha256);
 510         if (pkey)
 511                 EVP_PKEY_free(pkey);
 512         if (mdctx)
 513                 EVP_MD_CTX_destroy(mdctx);
 514         ERR_free_strings();
 515
 516         return (ret);
 517 }
 518
 519 static struct pubkey *
 520 read_pubkey(int fd)
 521 {
 522         struct pubkey *pk;
 523         char *sigb;
 524         size_t sigsz;
 525         FILE *sig;
 526         char buf[4096];
 527         int r;
 528
 529         if (lseek(fd, 0, 0) == -1) {
 530                 warn("lseek");
 531                 return (NULL);
 532         }
 533
 534         sigsz = 0;
 535         sigb = NULL;
 536         sig = open_memstream(&sigb, &sigsz);
 537         if (sig == NULL)
 538                 err(EXIT_FAILURE, "open_memstream()");
 539
 540         while ((r = read(fd, buf, sizeof(buf))) >0) {
 541                 fwrite(buf, 1, r, sig);
 542         }
 543
 544         fclose(sig);
 545         pk = calloc(1, sizeof(struct pubkey));
 546         pk->siglen = sigsz;
 547         pk->sig = calloc(1, pk->siglen);
 548         memcpy(pk->sig, sigb, pk->siglen);
 549         free(sigb);
 550
 551         return (pk);
 552 }
 553
 554 static struct sig_cert *
 555 parse_cert(int fd) {
 556         int my_fd;
 557         struct sig_cert *sc;
 558         FILE *fp, *sigfp, *certfp, *tmpfp;
 559         char *line;
 560         char *sig, *cert;
 561         size_t linecap, sigsz, certsz;
 562         ssize_t linelen;
 563
 564         sc = NULL;
 565         line = NULL;
 566         linecap = 0;
 567         sig = cert = NULL;
 568         sigfp = certfp = tmpfp = NULL;
 569
 570         if (lseek(fd, 0, 0) == -1) {
 571                 warn("lseek");
 572                 return (NULL);
 573         }
 574
 575         /* Duplicate the fd so that fclose(3) does not close it. */
 576         if ((my_fd = dup(fd)) == -1) {
 577                 warnx("dup");
 578                 return (NULL);
 579         }
 580
 581         if ((fp = fdopen(my_fd, "rb")) == NULL) {
 582                 warn("fdopen");
 583                 close(my_fd);
 584                 return (NULL);
 585         }
 586
 587         sigsz = certsz = 0;
 588         sigfp = open_memstream(&sig, &sigsz);
 589         if (sigfp == NULL)
 590                 err(EXIT_FAILURE, "open_memstream()");
 591         certfp = open_memstream(&cert, &certsz);
 592         if (certfp == NULL)
 593                 err(EXIT_FAILURE, "open_memstream()");
 594
 595         while ((linelen = getline(&line, &linecap, fp)) > 0) {
 596                 if (strcmp(line, "SIGNATURE\n") == 0) {
 597                         tmpfp = sigfp;
 598                         continue;
 599                 } else if (strcmp(line, "CERT\n") == 0) {
 600                         tmpfp = certfp;
 601                         continue;
 602                 } else if (strcmp(line, "END\n") == 0) {
 603                         break;
 604                 }
 605                 if (tmpfp != NULL)
 606                         fwrite(line, 1, linelen, tmpfp);
 607         }
 608
 609         fclose(fp);
 610         fclose(sigfp);
 611         fclose(certfp);
 612
 613         sc = calloc(1, sizeof(struct sig_cert));
 614         sc->siglen = sigsz -1; /* Trim out unrelated trailing newline */
 615         sc->sig = sig;
 616
 617         sc->certlen = certsz;
 618         sc->cert = cert;
 619
 620         return (sc);
 621 }
 622
 623 static bool
 624 verify_pubsignature(int fd_pkg, int fd_sig)
 625 {
 626         struct pubkey *pk;
 627         const char *pubkey;
 628         bool ret;
 629
 630         pk = NULL;
 631         pubkey = NULL;
 632         ret = false;
 633         if (config_string(PUBKEY, &pubkey) != 0) {
 634                 warnx("No CONFIG_PUBKEY defined");
 635                 goto cleanup;
 636         }
 637
 638         if ((pk = read_pubkey(fd_sig)) == NULL) {
 639                 warnx("Error reading signature");
 640                 goto cleanup;
 641         }
 642
 643         /* Verify the signature. */
 644         printf("Verifying signature with public key %s... ", pubkey);
 645         if (rsa_verify_cert(fd_pkg, pubkey, NULL, 0, pk->sig,
 646             pk->siglen) == false) {
 647                 fprintf(stderr, "Signature is not valid\n");
 648                 goto cleanup;
 649         }
 650
 651         ret = true;
 652
 653 cleanup:
 654         if (pk) {
 655                 free(pk->sig);
 656                 free(pk);
 657         }
 658
 659         return (ret);
 660 }
 661
 662 static bool
 663 verify_signature(int fd_pkg, int fd_sig)
 664 {
 665         struct fingerprint_list *trusted, *revoked;
 666         struct fingerprint *fingerprint;
 667         struct sig_cert *sc;
 668         bool ret;
 669         int trusted_count, revoked_count;
 670         const char *fingerprints;
 671         char path[MAXPATHLEN];
 672         char *hash;
 673
 674         hash = NULL;
 675         sc = NULL;
 676         trusted = revoked = NULL;
 677         ret = false;
 678
 679         /* Read and parse fingerprints. */
 680         if (config_string(FINGERPRINTS, &fingerprints) != 0) {
 681                 warnx("No CONFIG_FINGERPRINTS defined");
 682                 goto cleanup;
 683         }
 684
 685         snprintf(path, MAXPATHLEN, "%s/trusted", fingerprints);
 686         if ((trusted = load_fingerprints(path, &trusted_count)) == NULL) {
 687                 warnx("Error loading trusted certificates");
 688                 goto cleanup;
 689         }
 690
 691         if (trusted_count == 0 || trusted == NULL) {
 692                 fprintf(stderr, "No trusted certificates found.\n");
 693                 goto cleanup;
 694         }
 695
 696         snprintf(path, MAXPATHLEN, "%s/revoked", fingerprints);
 697         if ((revoked = load_fingerprints(path, &revoked_count)) == NULL) {
 698                 warnx("Error loading revoked certificates");
 699                 goto cleanup;
 700         }
 701
 702         /* Read certificate and signature in. */
 703         if ((sc = parse_cert(fd_sig)) == NULL) {
 704                 warnx("Error parsing certificate");
 705                 goto cleanup;
 706         }
 707         /* Explicitly mark as non-trusted until proven otherwise. */
 708         sc->trusted = false;
 709
 710         /* Parse signature and pubkey out of the certificate */
 711         hash = sha256_buf(sc->cert, sc->certlen);
 712
 713         /* Check if this hash is revoked */
 714         if (revoked != NULL) {
 715                 STAILQ_FOREACH(fingerprint, revoked, next) {
 716                         if (strcasecmp(fingerprint->hash, hash) == 0) {
 717                                 fprintf(stderr, "The package was signed with "
 718                                     "revoked certificate %s\n",
 719                                     fingerprint->name);
 720                                 goto cleanup;
 721                         }
 722                 }
 723         }
 724
 725         STAILQ_FOREACH(fingerprint, trusted, next) {
 726                 if (strcasecmp(fingerprint->hash, hash) == 0) {
 727                         sc->trusted = true;
 728                         sc->name = strdup(fingerprint->name);
 729                         break;
 730                 }
 731         }
 732
 733         if (sc->trusted == false) {
 734                 fprintf(stderr, "No trusted fingerprint found matching "
 735                     "package's certificate\n");
 736                 goto cleanup;
 737         }
 738
 739         /* Verify the signature. */
 740         printf("Verifying signature with trusted certificate %s... ", sc->name);
 741         if (rsa_verify_cert(fd_pkg, NULL, sc->cert, sc->certlen, sc->sig,
 742             sc->siglen) == false) {
 743                 fprintf(stderr, "Signature is not valid\n");
 744                 goto cleanup;
 745         }
 746
 747         ret = true;
 748
 749 cleanup:
 750         free(hash);
 751         if (trusted)
 752                 free_fingerprint_list(trusted);
 753         if (revoked)
 754                 free_fingerprint_list(revoked);
 755         if (sc) {
 756                 free(sc->cert);
 757                 free(sc->sig);
 758                 free(sc->name);
 759                 free(sc);
 760         }
 761
 762         return (ret);
 763 }
 764
 765 static int
 766 bootstrap_pkg(bool force, const char *fetchOpts)
 767 {
 768         int fd_pkg, fd_sig;
 769         int ret;
 770         char url[MAXPATHLEN];
 771         char tmppkg[MAXPATHLEN];
 772         char tmpsig[MAXPATHLEN];
 773         const char *packagesite;
 774         const char *signature_type;
 775         char pkgstatic[MAXPATHLEN];
 776         const char *bootstrap_name;
 777
 778         fd_sig = -1;
 779         ret = -1;
 780
 781         if (config_string(PACKAGESITE, &packagesite) != 0) {
 782                 warnx("No PACKAGESITE defined");
 783                 return (-1);
 784         }
 785
 786         if (config_string(SIGNATURE_TYPE, &signature_type) != 0) {
 787                 warnx("Error looking up SIGNATURE_TYPE");
 788                 return (-1);
 789         }
 790
 791         printf("Bootstrapping pkg from %s, please wait...\n", packagesite);
 792
 793         /* Support pkg+http:// for PACKAGESITE which is the new format
 794            in 1.2 to avoid confusion on why http://pkg.FreeBSD.org has
 795            no A record. */
 796         if (strncmp(URL_SCHEME_PREFIX, packagesite,
 797             strlen(URL_SCHEME_PREFIX)) == 0)
 798                 packagesite += strlen(URL_SCHEME_PREFIX);
 799         for (int j = 0; bootstrap_names[j] != NULL; j++) {
 800                 bootstrap_name = bootstrap_names[j];
 801
 802                 snprintf(url, MAXPATHLEN, "%s/Latest/%s", packagesite, bootstrap_name);
 803                 snprintf(tmppkg, MAXPATHLEN, "%s/%s.XXXXXX",
 804                     getenv("TMPDIR") ? getenv("TMPDIR") : _PATH_TMP,
 805                     bootstrap_name);
 806                 if ((fd_pkg = fetch_to_fd(url, tmppkg, fetchOpts)) != -1)
 807                         break;
 808                 bootstrap_name = NULL;
 809         }
 810         if (bootstrap_name == NULL)
 811                 goto fetchfail;
 812
 813         if (signature_type != NULL &&
 814             strcasecmp(signature_type, "NONE") != 0) {
 815                 if (strcasecmp(signature_type, "FINGERPRINTS") == 0) {
 816
 817                         snprintf(tmpsig, MAXPATHLEN, "%s/%s.sig.XXXXXX",
 818                             getenv("TMPDIR") ? getenv("TMPDIR") : _PATH_TMP,
 819                             bootstrap_name);
 820                         snprintf(url, MAXPATHLEN, "%s/Latest/%s.sig",
 821                             packagesite, bootstrap_name);
 822
 823                         if ((fd_sig = fetch_to_fd(url, tmpsig, fetchOpts)) == -1) {
 824                                 fprintf(stderr, "Signature for pkg not "
 825                                     "available.\n");
 826                                 goto fetchfail;
 827                         }
 828
 829                         if (verify_signature(fd_pkg, fd_sig) == false)
 830                                 goto cleanup;
 831                 } else if (strcasecmp(signature_type, "PUBKEY") == 0) {
 832
 833                         snprintf(tmpsig, MAXPATHLEN,
 834                             "%s/%s.pubkeysig.XXXXXX",
 835                             getenv("TMPDIR") ? getenv("TMPDIR") : _PATH_TMP,
 836                             bootstrap_name);
 837                         snprintf(url, MAXPATHLEN, "%s/Latest/%s.pubkeysig",
 838                             packagesite, bootstrap_name);
 839
 840                         if ((fd_sig = fetch_to_fd(url, tmpsig, fetchOpts)) == -1) {
 841                                 fprintf(stderr, "Signature for pkg not "
 842                                     "available.\n");
 843                                 goto fetchfail;
 844                         }
 845
 846                         if (verify_pubsignature(fd_pkg, fd_sig) == false)
 847                                 goto cleanup;
 848                 } else {
 849                         warnx("Signature type %s is not supported for "
 850                             "bootstrapping.", signature_type);
 851                         goto cleanup;
 852                 }
 853         }
 854
 855         if ((ret = extract_pkg_static(fd_pkg, pkgstatic, MAXPATHLEN)) == 0)
 856                 ret = install_pkg_static(pkgstatic, tmppkg, force);
 857
 858         goto cleanup;
 859
 860 fetchfail:
 861         warnx("Error fetching %s: %s", url, fetchLastErrString);
 862         if (fetchLastErrCode == FETCH_RESOLV) {
 863                 fprintf(stderr, "Address resolution failed for %s.\n", packagesite);
 864                 fprintf(stderr, "Consider changing PACKAGESITE.\n");
 865         } else {
 866                 fprintf(stderr, "A pre-built version of pkg could not be found for "
 867                     "your system.\n");
 868                 fprintf(stderr, "Consider changing PACKAGESITE or installing it from "
 869                     "ports: 'ports-mgmt/pkg'.\n");
 870         }
 871
 872 cleanup:
 873         if (fd_sig != -1) {
 874                 close(fd_sig);
 875                 unlink(tmpsig);
 876         }
 877
 878         if (fd_pkg != -1) {
 879                 close(fd_pkg);
 880                 unlink(tmppkg);
 881         }
 882
 883         return (ret);
 884 }
 885
 886 static const char confirmation_message[] =
 887 "The package management tool is not yet installed on your system.\n"
 888 "Do you want to fetch and install it now? [y/N]: ";
 889
 890 static const char non_interactive_message[] =
 891 "The package management tool is not yet installed on your system.\n"
 892 "Please set ASSUME_ALWAYS_YES=yes environment variable to be able to bootstrap "
 893 "in non-interactive (stdin not being a tty)\n";
 894
 895 static const char args_bootstrap_message[] =
 896 "Too many arguments\n"
 897 "Usage: pkg [-4|-6] bootstrap [-f] [-y]\n";
 898
 899 static const char args_add_message[] =
 900 "Too many arguments\n"
 901 "Usage: pkg add [-f] [-y] {pkg.txz}\n";
 902
 903 static int
 904 pkg_query_yes_no(void)
 905 {
 906         int ret, c;
 907
 908         fflush(stdout);
 909         c = getchar();
 910
 911         if (c == 'y' || c == 'Y')
 912                 ret = 1;
 913         else
 914                 ret = 0;
 915
 916         while (c != '\n' && c != EOF)
 917                 c = getchar();
 918
 919         return (ret);
 920 }
 921
 922 static int
 923 bootstrap_pkg_local(const char *pkgpath, bool force)
 924 {
 925         char path[MAXPATHLEN];
 926         char pkgstatic[MAXPATHLEN];
 927         const char *signature_type;
 928         int fd_pkg, fd_sig, ret;
 929
 930         fd_sig = -1;
 931         ret = -1;
 932
 933         fd_pkg = open(pkgpath, O_RDONLY);
 934         if (fd_pkg == -1)
 935                 err(EXIT_FAILURE, "Unable to open %s", pkgpath);
 936
 937         if (config_string(SIGNATURE_TYPE, &signature_type) != 0) {
 938                 warnx("Error looking up SIGNATURE_TYPE");
 939                 goto cleanup;
 940         }
 941         if (signature_type != NULL &&
 942             strcasecmp(signature_type, "NONE") != 0) {
 943                 if (strcasecmp(signature_type, "FINGERPRINTS") == 0) {
 944
 945                         snprintf(path, sizeof(path), "%s.sig", pkgpath);
 946
 947                         if ((fd_sig = open(path, O_RDONLY)) == -1) {
 948                                 fprintf(stderr, "Signature for pkg not "
 949                                     "available.\n");
 950                                 goto cleanup;
 951                         }
 952
 953                         if (verify_signature(fd_pkg, fd_sig) == false)
 954                                 goto cleanup;
 955
 956                 } else if (strcasecmp(signature_type, "PUBKEY") == 0) {
 957
 958                         snprintf(path, sizeof(path), "%s.pubkeysig", pkgpath);
 959
 960                         if ((fd_sig = open(path, O_RDONLY)) == -1) {
 961                                 fprintf(stderr, "Signature for pkg not "
 962                                     "available.\n");
 963                                 goto cleanup;
 964                         }
 965
 966                         if (verify_pubsignature(fd_pkg, fd_sig) == false)
 967                                 goto cleanup;
 968
 969                 } else {
 970                         warnx("Signature type %s is not supported for "
 971                             "bootstrapping.", signature_type);
 972                         goto cleanup;
 973                 }
 974         }
 975
 976         if ((ret = extract_pkg_static(fd_pkg, pkgstatic, MAXPATHLEN)) == 0)
 977                 ret = install_pkg_static(pkgstatic, pkgpath, force);
 978
 979 cleanup:
 980         close(fd_pkg);
 981         if (fd_sig != -1)
 982                 close(fd_sig);
 983
 984         return (ret);
 985 }
 986
 987 #define PKG_NAME        "pkg"
 988 #define PKG_DEVEL_NAME  PKG_NAME "-devel"
 989 #define PKG_PKG         PKG_NAME "."
 990
 991 static bool
 992 pkg_is_pkg_pkg(const char *pkg)
 993 {
 994         char *vstart, *basename;
 995         size_t namelen;
 996
 997         /* Strip path. */
 998         if ((basename = strrchr(pkg, '/')) != NULL)
 999                 pkg = basename + 1;
1000
1001         /*
1002          * Chop off the final "-" (version delimiter) and check the name that
1003          * precedes it.  If we didn't have a version delimiter, it must be the
1004          * pkg.$archive short form but we'll check it anyways.  pkg-devel short
1005          * form will look like a pkg archive with 'devel' version, but that's
1006          * OK.  We otherwise assumed that non-pkg packages will always have a
1007          * version component.
1008          */
1009         vstart = strrchr(pkg, '-');
1010         if (vstart == NULL) {
1011                 return (strlen(pkg) > sizeof(PKG_PKG) - 1 &&
1012                     strncmp(pkg, PKG_PKG, sizeof(PKG_PKG) - 1) == 0);
1013         }
1014
1015         namelen = vstart - pkg;
1016         if (namelen == sizeof(PKG_NAME) - 1 &&
1017             strncmp(pkg, PKG_NAME, sizeof(PKG_NAME) - 1) == 0)
1018                 return (true);
1019         if (namelen == sizeof(PKG_DEVEL_NAME) - 1 &&
1020             strncmp(pkg, PKG_DEVEL_NAME, sizeof(PKG_DEVEL_NAME) - 1) == 0)
1021                 return (true);
1022         return (false);
1023 }
1024
1025 int
1026 main(int argc, char *argv[])
1027 {
1028         char pkgpath[MAXPATHLEN];
1029         const char *pkgarg, *repo_name;
1030         bool activation_test, add_pkg, bootstrap_only, force, yes;
1031         signed char ch;
1032         const char *fetchOpts;
1033         char *command;
1034
1035         activation_test = false;
1036         add_pkg = false;
1037         bootstrap_only = false;
1038         command = NULL;
1039         fetchOpts = "";
1040         force = false;
1041         pkgarg = NULL;
1042         repo_name = NULL;
1043         yes = false;
1044
1045         struct option longopts[] = {
1046                 { "debug",              no_argument,            NULL,   'd' },
1047                 { "force",              no_argument,            NULL,   'f' },
1048                 { "only-ipv4",          no_argument,            NULL,   '4' },
1049                 { "only-ipv6",          no_argument,            NULL,   '6' },
1050                 { "yes",                no_argument,            NULL,   'y' },
1051                 { NULL,                 0,                      NULL,   0   },
1052         };
1053
1054         snprintf(pkgpath, MAXPATHLEN, "%s/sbin/pkg", getlocalbase());
1055
1056         while ((ch = getopt_long(argc, argv, "-:dfr::yN46", longopts, NULL)) != -1) {
1057                 switch (ch) {
1058                 case 'd':
1059                         debug++;
1060                         break;
1061                 case 'f':
1062                         force = true;
1063                         break;
1064                 case 'N':
1065                         activation_test = true;
1066                         break;
1067                 case 'y':
1068                         yes = true;
1069                         break;
1070                 case '4':
1071                         fetchOpts = "4";
1072                         break;
1073                 case '6':
1074                         fetchOpts = "6";
1075                         break;
1076                 case 'r':
1077                         /*
1078                          * The repository can only be specified for an explicit
1079                          * bootstrap request at this time, so that we don't
1080                          * confuse the user if they're trying to use a verb that
1081                          * has some other conflicting meaning but we need to
1082                          * bootstrap.
1083                          *
1084                          * For that reason, we specify that -r has an optional
1085                          * argument above and process the next index ourselves.
1086                          * This is mostly significant because getopt(3) will
1087                          * otherwise eat the next argument, which could be
1088                          * something we need to try and make sense of.
1089                          *
1090                          * At worst this gets us false positives that we ignore
1091                          * in other contexts, and we have to do a little fudging
1092                          * in order to support separating -r from the reponame
1093                          * with a space since it's not actually optional in
1094                          * the bootstrap/add sense.
1095                          */
1096                         if (add_pkg || bootstrap_only) {
1097                                 if (optarg != NULL) {
1098                                         repo_name = optarg;
1099                                 } else if (optind < argc) {
1100                                         repo_name = argv[optind];
1101                                 }
1102
1103                                 if (repo_name == NULL || *repo_name == '\0') {
1104                                         fprintf(stderr,
1105                                             "Must specify a repository with -r!\n");
1106                                         exit(EXIT_FAILURE);
1107                                 }
1108
1109                                 if (optarg == NULL) {
1110                                         /* Advance past repo name. */
1111                                         optreset = 1;
1112                                         optind++;
1113                                 }
1114                         }
1115                         break;
1116                 case 1:
1117                         // Non-option arguments, first one is the command
1118                         if (command == NULL) {
1119                                 command = argv[optind-1];
1120                                 if (strcmp(command, "add") == 0) {
1121                                         add_pkg = true;
1122                                 }
1123                                 else if (strcmp(command, "bootstrap") == 0) {
1124                                         bootstrap_only = true;
1125                                 }
1126                         }
1127                         // bootstrap doesn't accept other arguments
1128                         else if (bootstrap_only) {
1129                                 fprintf(stderr, args_bootstrap_message);
1130                                 exit(EXIT_FAILURE);
1131                         }
1132                         else if (add_pkg && pkgarg != NULL) {
1133                                 /*
1134                                  * Additional arguments also means it's not a
1135                                  * local bootstrap request.
1136                                  */
1137                                 add_pkg = false;
1138                         }
1139                         else if (add_pkg) {
1140                                 /*
1141                                  * If it's not a request for pkg or pkg-devel,
1142                                  * then we must assume they were trying to
1143                                  * install some other local package and we
1144                                  * should try to bootstrap from the repo.
1145                                  */
1146                                 if (!pkg_is_pkg_pkg(argv[optind-1])) {
1147                                         add_pkg = false;
1148                                 } else {
1149                                         pkgarg = argv[optind-1];
1150                                 }
1151                         }
1152                         break;
1153                 default:
1154                         break;
1155                 }
1156         }
1157         if (debug > 1)
1158                 fetchDebug = 1;
1159
1160         if ((bootstrap_only && force) || access(pkgpath, X_OK) == -1) {
1161                 /*
1162                  * To allow 'pkg -N' to be used as a reliable test for whether
1163                  * a system is configured to use pkg, don't bootstrap pkg
1164                  * when that option is passed.
1165                  */
1166                 if (activation_test)
1167                         errx(EXIT_FAILURE, "pkg is not installed");
1168
1169                 config_init(repo_name);
1170
1171                 if (add_pkg) {
1172                         if (pkgarg == NULL) {
1173                                 fprintf(stderr, "Path to pkg.txz required\n");
1174                                 exit(EXIT_FAILURE);
1175                         }
1176                         if (access(pkgarg, R_OK) == -1) {
1177                                 fprintf(stderr, "No such file: %s\n", pkgarg);
1178                                 exit(EXIT_FAILURE);
1179                         }
1180                         if (bootstrap_pkg_local(pkgarg, force) != 0)
1181                                 exit(EXIT_FAILURE);
1182                         exit(EXIT_SUCCESS);
1183                 }
1184                 /*
1185                  * Do not ask for confirmation if either of stdin or stdout is
1186                  * not tty. Check the environment to see if user has answer
1187                  * tucked in there already.
1188                  */
1189                 if (!yes)
1190                         config_bool(ASSUME_ALWAYS_YES, &yes);
1191                 if (!yes) {
1192                         if (!isatty(fileno(stdin))) {
1193                                 fprintf(stderr, non_interactive_message);
1194                                 exit(EXIT_FAILURE);
1195                         }
1196
1197                         printf("%s", confirmation_message);
1198                         if (pkg_query_yes_no() == 0)
1199                                 exit(EXIT_FAILURE);
1200                 }
1201                 if (bootstrap_pkg(force, fetchOpts) != 0)
1202                         exit(EXIT_FAILURE);
1203                 config_finish();
1204
1205                 if (bootstrap_only)
1206                         exit(EXIT_SUCCESS);
1207         } else if (bootstrap_only) {
1208                 printf("pkg already bootstrapped at %s\n", pkgpath);
1209                 exit(EXIT_SUCCESS);
1210         }
1211
1212         execv(pkgpath, argv);
1213
1214         /* NOT REACHED */
1215         return (EXIT_FAILURE);
1216 }