pkg.c revision 257309
1/*- 2 * Copyright (c) 2012-2013 Baptiste Daroussin <bapt@FreeBSD.org> 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 24 * SUCH DAMAGE. 25 */ 26 27#include <sys/cdefs.h> 28__FBSDID("$FreeBSD: stable/10/usr.sbin/pkg/pkg.c 257309 2013-10-29 07:33:53Z bapt $"); 29 30#include <sys/param.h> 31#include <sys/wait.h> 32 33#include <archive.h> 34#include <archive_entry.h> 35#include <err.h> 36#include <errno.h> 37#include <fcntl.h> 38#include <fetch.h> 39#include <paths.h> 40#include <stdbool.h> 41#include <stdlib.h> 42#include <stdio.h> 43#include <string.h> 44#include <time.h> 45#include <unistd.h> 46 47#include "dns_utils.h" 48#include "config.h" 49 50static int 51extract_pkg_static(int fd, char *p, int sz) 52{ 53 struct archive *a; 54 struct archive_entry *ae; 55 char *end; 56 int ret, r; 57 58 ret = -1; 59 a = archive_read_new(); 60 if (a == NULL) { 61 warn("archive_read_new"); 62 return (ret); 63 } 64 archive_read_support_filter_all(a); 65 archive_read_support_format_tar(a); 66 67 if (lseek(fd, 0, 0) == -1) { 68 warn("lseek"); 69 goto cleanup; 70 } 71 72 if (archive_read_open_fd(a, fd, 4096) != ARCHIVE_OK) { 73 warnx("archive_read_open_fd: %s", archive_error_string(a)); 74 goto cleanup; 75 } 76 77 ae = NULL; 78 while ((r = archive_read_next_header(a, &ae)) == ARCHIVE_OK) { 79 end = strrchr(archive_entry_pathname(ae), '/'); 80 if (end == NULL) 81 continue; 82 83 if (strcmp(end, "/pkg-static") == 0) { 84 r = archive_read_extract(a, ae, 85 ARCHIVE_EXTRACT_OWNER | ARCHIVE_EXTRACT_PERM | 86 ARCHIVE_EXTRACT_TIME | ARCHIVE_EXTRACT_ACL | 87 ARCHIVE_EXTRACT_FFLAGS | ARCHIVE_EXTRACT_XATTR); 88 strlcpy(p, archive_entry_pathname(ae), sz); 89 break; 90 } 91 } 92 93 if (r == ARCHIVE_OK) 94 ret = 0; 95 else 96 warnx("fail to extract pkg-static"); 97 98cleanup: 99 archive_read_free(a); 100 return (ret); 101 102} 103 104static int 105install_pkg_static(char *path, char *pkgpath) 106{ 107 int pstat; 108 pid_t pid; 109 110 switch ((pid = fork())) { 111 case -1: 112 return (-1); 113 case 0: 114 execl(path, "pkg-static", "add", pkgpath, (char *)NULL); 115 _exit(1); 116 default: 117 break; 118 } 119 120 while (waitpid(pid, &pstat, 0) == -1) 121 if (errno != EINTR) 122 return (-1); 123 124 if (WEXITSTATUS(pstat)) 125 return (WEXITSTATUS(pstat)); 126 else if (WIFSIGNALED(pstat)) 127 return (128 & (WTERMSIG(pstat))); 128 return (pstat); 129} 130 131static int 132bootstrap_pkg(void) 133{ 134 struct url *u; 135 FILE *remote; 136 FILE *config; 137 char *site; 138 struct dns_srvinfo *mirrors, *current; 139 /* To store _https._tcp. + hostname + \0 */ 140 char zone[MAXHOSTNAMELEN + 13]; 141 char url[MAXPATHLEN]; 142 char conf[MAXPATHLEN]; 143 char tmppkg[MAXPATHLEN]; 144 const char *packagesite, *mirror_type; 145 char buf[10240]; 146 char pkgstatic[MAXPATHLEN]; 147 int fd, retry, ret, max_retry; 148 struct url_stat st; 149 off_t done, r; 150 time_t now; 151 time_t last; 152 153 done = 0; 154 last = 0; 155 max_retry = 3; 156 ret = -1; 157 remote = NULL; 158 config = NULL; 159 current = mirrors = NULL; 160 161 printf("Bootstrapping pkg please wait\n"); 162 163 if (config_string(PACKAGESITE, &packagesite) != 0) { 164 warnx("No PACKAGESITE defined"); 165 return (-1); 166 } 167 if (config_string(MIRROR_TYPE, &mirror_type) != 0) { 168 warnx("No MIRROR_TYPE defined"); 169 return (-1); 170 } 171 snprintf(url, MAXPATHLEN, "%s/Latest/pkg.txz", packagesite); 172 173 snprintf(tmppkg, MAXPATHLEN, "%s/pkg.txz.XXXXXX", 174 getenv("TMPDIR") ? getenv("TMPDIR") : _PATH_TMP); 175 176 if ((fd = mkstemp(tmppkg)) == -1) { 177 warn("mkstemp()"); 178 return (-1); 179 } 180 181 retry = max_retry; 182 183 u = fetchParseURL(url); 184 while (remote == NULL) { 185 if (retry == max_retry) { 186 if (strcmp(u->scheme, "file") != 0 && 187 strcasecmp(mirror_type, "srv") == 0) { 188 snprintf(zone, sizeof(zone), 189 "_%s._tcp.%s", u->scheme, u->host); 190 mirrors = dns_getsrvinfo(zone); 191 current = mirrors; 192 } 193 } 194 195 if (mirrors != NULL) { 196 strlcpy(u->host, current->host, sizeof(u->host)); 197 u->port = current->port; 198 } 199 200 remote = fetchXGet(u, &st, ""); 201 if (remote == NULL) { 202 --retry; 203 if (retry <= 0) 204 goto fetchfail; 205 if (mirrors == NULL) { 206 sleep(1); 207 } else { 208 current = current->next; 209 if (current == NULL) 210 current = mirrors; 211 } 212 } 213 } 214 215 if (remote == NULL) 216 goto fetchfail; 217 218 while (done < st.size) { 219 if ((r = fread(buf, 1, sizeof(buf), remote)) < 1) 220 break; 221 222 if (write(fd, buf, r) != r) { 223 warn("write()"); 224 goto cleanup; 225 } 226 227 done += r; 228 now = time(NULL); 229 if (now > last || done == st.size) 230 last = now; 231 } 232 233 if (ferror(remote)) 234 goto fetchfail; 235 236 if ((ret = extract_pkg_static(fd, pkgstatic, MAXPATHLEN)) == 0) 237 ret = install_pkg_static(pkgstatic, tmppkg); 238 239 snprintf(conf, MAXPATHLEN, "%s/etc/pkg.conf", 240 getenv("LOCALBASE") ? getenv("LOCALBASE") : _LOCALBASE); 241 242 if (access(conf, R_OK) == -1) { 243 site = strrchr(url, '/'); 244 if (site == NULL) 245 goto cleanup; 246 site[0] = '\0'; 247 site = strrchr(url, '/'); 248 if (site == NULL) 249 goto cleanup; 250 site[0] = '\0'; 251 252 config = fopen(conf, "w+"); 253 if (config == NULL) 254 goto cleanup; 255 fprintf(config, "packagesite: %s\n", url); 256 fclose(config); 257 } 258 259 goto cleanup; 260 261fetchfail: 262 warnx("Error fetching %s: %s", url, fetchLastErrString); 263 fprintf(stderr, "A pre-built version of pkg could not be found for your system.\n"); 264 fprintf(stderr, "Consider changing PACKAGESITE or installing it from ports: 'ports-mgmt/pkg'.\n"); 265 266cleanup: 267 if (remote != NULL) 268 fclose(remote); 269 close(fd); 270 unlink(tmppkg); 271 272 return (ret); 273} 274 275static const char confirmation_message[] = 276"The package management tool is not yet installed on your system.\n" 277"Do you want to fetch and install it now? [y/N]: "; 278 279static int 280pkg_query_yes_no(void) 281{ 282 int ret, c; 283 284 c = getchar(); 285 286 if (c == 'y' || c == 'Y') 287 ret = 1; 288 else 289 ret = 0; 290 291 while (c != '\n' && c != EOF) 292 c = getchar(); 293 294 return (ret); 295} 296 297int 298main(__unused int argc, char *argv[]) 299{ 300 char pkgpath[MAXPATHLEN]; 301 char pkgstatic[MAXPATHLEN]; 302 bool yes = false; 303 int fd, ret; 304 305 snprintf(pkgpath, MAXPATHLEN, "%s/sbin/pkg", 306 getenv("LOCALBASE") ? getenv("LOCALBASE") : _LOCALBASE); 307 308 if (access(pkgpath, X_OK) == -1) { 309 /* 310 * To allow 'pkg -N' to be used as a reliable test for whether 311 * a system is configured to use pkg, don't bootstrap pkg 312 * when that argument is given as argv[1]. 313 */ 314 if (argv[1] != NULL && strcmp(argv[1], "-N") == 0) 315 errx(EXIT_FAILURE, "pkg is not installed"); 316 317 if (argc > 2 && strcmp(argv[1], "add") == 0 && 318 access(argv[2], R_OK) == 0) { 319 fd = open(argv[2], O_RDONLY); 320 if (fd == -1) 321 err(EXIT_FAILURE, "Unable to open %s", argv[2]); 322 323 if ((ret = extract_pkg_static(fd, pkgstatic, MAXPATHLEN)) == 0) 324 ret = install_pkg_static(pkgstatic, argv[2]); 325 close(fd); 326 if (ret != 0) 327 exit(EXIT_FAILURE); 328 exit(EXIT_SUCCESS); 329 } 330 /* 331 * Do not ask for confirmation if either of stdin or stdout is 332 * not tty. Check the environment to see if user has answer 333 * tucked in there already. 334 */ 335 config_init(); 336 config_bool(ASSUME_ALWAYS_YES, &yes); 337 if (!yes) { 338 printf("%s", confirmation_message); 339 if (!isatty(fileno(stdin))) 340 exit(EXIT_FAILURE); 341 342 if (pkg_query_yes_no() == 0) 343 exit(EXIT_FAILURE); 344 } 345 if (bootstrap_pkg() != 0) 346 exit(EXIT_FAILURE); 347 config_finish(); 348 } 349 350 execv(pkgpath, argv); 351 352 /* NOT REACHED */ 353 return (EXIT_FAILURE); 354} 355