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