release/sysinstall/package.c

   1 /*
   2  * The new sysinstall program.
   3  *
   4  * This is probably the last program in the `sysinstall' line - the next
   5  * generation being essentially a complete rewrite.
   6  *
   7  * $FreeBSD: src/release/sysinstall/package.c,v 1.90.2.9 2002/04/23 20:35:08 obrien Exp $
   8  *
   9  * Copyright (c) 1995
  10  *      Jordan Hubbard.  All rights reserved.
  11  *
  12  * Redistribution and use in source and binary forms, with or without
  13  * modification, are permitted provided that the following conditions
  14  * are met:
  15  * 1. Redistributions of source code must retain the above copyright
  16  *    notice, this list of conditions and the following disclaimer,
  17  *    verbatim and that no modifications are made prior to this
  18  *    point in the file.
  19  * 2. Redistributions in binary form must reproduce the above copyright
  20  *    notice, this list of conditions and the following disclaimer in the
  21  *    documentation and/or other materials provided with the distribution.
  22  *
  23  * THIS SOFTWARE IS PROVIDED BY JORDAN HUBBARD ``AS IS'' AND
  24  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  25  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  26  * ARE DISCLAIMED.  IN NO EVENT SHALL JORDAN HUBBARD OR HIS PETS BE LIABLE
  27  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  29  * OR SERVICES; LOSS OF USE, DATA, LIFE OR PROFITS; OR BUSINESS INTERRUPTION)
  30  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  31  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  32  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  33  * SUCH DAMAGE.
  34  *
  35  */
  36
  37 #include "sysinstall.h"
  38 #include <sys/errno.h>
  39 #include <sys/time.h>
  40 #include <sys/param.h>
  41 #include <sys/mount.h>
  42 #include <sys/stat.h>
  43
  44 static Boolean sigpipe_caught;
  45
  46 static void
  47 catch_pipe(int sig)
  48 {
  49     sigpipe_caught = TRUE;
  50 }
  51
  52 extern PkgNode Top;
  53
  54 /* Like package_extract, but assumes current media device and chases deps */
  55 int
  56 package_add(char *name)
  57 {
  58     PkgNodePtr tmp;
  59     int i;
  60
  61     if (!mediaVerify())
  62         return DITEM_FAILURE;
  63
  64     if (!DEVICE_INIT(mediaDevice))
  65         return DITEM_FAILURE;
  66
  67     i = index_initialize("packages/INDEX");
  68     if (DITEM_STATUS(i) != DITEM_SUCCESS)
  69         return i;
  70
  71     tmp = index_search(&Top, name, &tmp);
  72     if (tmp)
  73         return index_extract(mediaDevice, &Top, tmp, FALSE);
  74     else {
  75         msgConfirm("Sorry, package %s was not found in the INDEX.", name);
  76         return DITEM_FAILURE;
  77     }
  78 }
  79
  80 /* For use by dispatch */
  81 int
  82 packageAdd(dialogMenuItem *self)
  83 {
  84     char *cp;
  85
  86     cp = variable_get(VAR_PACKAGE);
  87     if (!cp) {
  88         msgDebug("packageAdd:  No package name passed in package variable\n");
  89         return DITEM_FAILURE;
  90     }
  91     else
  92         return package_add(cp);
  93 }
  94
  95 Boolean
  96 package_exists(char *name)
  97 {
  98     char fname[FILENAME_MAX];
  99     int status /* = vsystem("pkg_info -e %s", name) */;
 100
 101     /* XXX KLUDGE ALERT!  This makes evil assumptions about how XXX
 102      * packages register themselves and should *really be done with
 103      * `pkg_info -e <name>' except that this it's too slow for an
 104      * item check routine.. :-(
 105      */
 106     snprintf(fname, FILENAME_MAX, "/var/db/pkg/%s", name);
 107     status = access(fname, R_OK);
 108     if (isDebug())
 109         msgDebug("package check for %s returns %s.\n", name, status ? "failure" : "success");
 110     return !status;
 111 }
 112
 113 /* Extract a package based on a namespec and a media device */
 114 int
 115 package_extract(Device *dev, char *name, Boolean depended)
 116 {
 117     char path[511];
 118     int ret, last_msg = 0;
 119     FILE *fp;
 120
 121     /* Check to make sure it's not already there */
 122     if (package_exists(name))
 123         return DITEM_SUCCESS;
 124
 125     if (!DEVICE_INIT(dev)) {
 126         msgConfirm("Unable to initialize media type for package extract.");
 127         return DITEM_FAILURE;
 128     }
 129
 130     /* If necessary, initialize the ldconfig hints */
 131     if (!file_readable("/var/run/ld-elf.so.hints"))
 132         vsystem("ldconfig /usr/lib /usr/lib/compat /usr/local/lib /usr/X11R6/lib");
 133     if (!file_readable("/var/run/ld.so.hints"))
 134         vsystem("ldconfig -aout /usr/lib/aout /usr/lib/compat/aout /usr/local/lib/aout /usr/X11R6/lib/aout");
 135
 136     /* Be initially optimistic */
 137     ret = DITEM_SUCCESS;
 138     /* Make a couple of paranoid locations for temp files to live if user specified none */
 139     if (!variable_get(VAR_PKG_TMPDIR)) {
 140         /* Set it to a location with as much space as possible */
 141         variable_set2(VAR_PKG_TMPDIR, "/var/tmp", 0);
 142     }
 143     Mkdir(variable_get(VAR_PKG_TMPDIR));
 144     vsystem("chmod 1777 %s", variable_get(VAR_PKG_TMPDIR));
 145
 146     if (!index(name, '/')) {
 147         if (!strpbrk(name, "-_"))
 148             sprintf(path, "packages/Latest/%s.tgz", name);
 149         else
 150             sprintf(path, "packages/All/%s%s", name, strstr(name, ".tgz") ? "" : ".tgz");
 151     }
 152     else
 153         sprintf(path, "%s%s", name, strstr(name, ".tgz") ? "" : ".tgz");
 154
 155     /* We have a path, call the device strategy routine to get the file */
 156     fp = DEVICE_GET(dev, path, TRUE);
 157     if (fp) {
 158         int i = 0, tot, pfd[2];
 159         pid_t pid;
 160         WINDOW *w = savescr();
 161
 162         sigpipe_caught = FALSE;
 163         signal(SIGPIPE, catch_pipe);
 164
 165         dialog_clear_norefresh();
 166         msgNotify("Adding %s%s\nfrom %s", path, depended ? " (as a dependency)" : "", dev->name);
 167         pipe(pfd);
 168         pid = fork();
 169         if (!pid) {
 170             dup2(pfd[0], 0); close(pfd[0]);
 171             dup2(DebugFD, 1); dup2(1, 2);
 172             close(pfd[1]);
 173
 174             /* Prevent pkg_add from wanting to interact in bad ways */
 175             setenv("PACKAGE_BUILDING", "t", 1);
 176             setenv("BATCH", "t", 1);
 177
 178             if (isDebug())
 179                 i = execl("/usr/sbin/pkg_add", "/usr/sbin/pkg_add", "-v", "-",
 180                     (char *)0);
 181             else
 182                 i = execl("/usr/sbin/pkg_add", "/usr/sbin/pkg_add", "-",
 183                     (char *)0);
 184         }
 185         else {
 186             char buf[BUFSIZ];
 187             struct timeval start, stop;
 188
 189             close(pfd[0]);
 190             tot = 0;
 191             (void)gettimeofday(&start, (struct timezone *)0);
 192
 193             while (!sigpipe_caught && (i = fread(buf, 1, BUFSIZ, fp)) > 0) {
 194                 int seconds;
 195
 196                 tot += i;
 197                 /* Print statistics about how we're doing */
 198                 (void) gettimeofday(&stop, (struct timezone *)0);
 199                 stop.tv_sec = stop.tv_sec - start.tv_sec;
 200                 stop.tv_usec = stop.tv_usec - start.tv_usec;
 201                 if (stop.tv_usec < 0)
 202                     stop.tv_sec--, stop.tv_usec += 1000000;
 203                 seconds = stop.tv_sec + (stop.tv_usec / 1000000.0);
 204                 if (!seconds)
 205                     seconds = 1;
 206                 if (seconds != last_msg) {
 207                     last_msg = seconds;
 208                     msgInfo("%10d bytes read from package %s @ %4.1f KBytes/second", tot, name, (tot / seconds) / 1000.0);
 209                 }
 210                 /* Write it out */
 211                 if (sigpipe_caught || write(pfd[1], buf, i) != i) {
 212                     msgInfo("Write failure to pkg_add!  Package may be corrupt.");
 213                     break;
 214                 }
 215             }
 216             close(pfd[1]);
 217             fclose(fp);
 218             if (sigpipe_caught)
 219                 msgInfo("pkg_add(1) apparently did not like the %s package.", name);
 220             else if (i == -1)
 221                 msgInfo("I/O error while reading in the %s package.", name);
 222             else
 223                 msgInfo("Package %s read successfully - waiting for pkg_add(1)", name);
 224             refresh();
 225             i = waitpid(pid, &tot, 0);
 226             dialog_clear_norefresh();
 227             if (sigpipe_caught || i < 0 || WEXITSTATUS(tot)) {
 228                 ret = DITEM_FAILURE;
 229                 if (variable_get(VAR_NO_CONFIRM))
 230                     msgNotify("Add of package %s aborted, error code %d -\n"
 231                                "Please check the debug screen for more info.", name, WEXITSTATUS(tot));
 232                 else
 233                     msgConfirm("Add of package %s aborted, error code %d -\n"
 234                                "Please check the debug screen for more info.", name, WEXITSTATUS(tot));
 235             }
 236             else
 237                 msgNotify("Package %s was added successfully", name);
 238
 239             /* Now catch any stragglers */
 240             while (wait3(&tot, WNOHANG, NULL) > 0);
 241
 242             sleep(1);
 243             restorescr(w);
 244         }
 245     }
 246     else {
 247         dialog_clear_norefresh();
 248         if (variable_get(VAR_NO_CONFIRM))
 249             msgNotify("Unable to fetch package %s from selected media.\n"
 250                       "No package add will be done.", name);
 251         else
 252             msgConfirm("Unable to fetch package %s from selected media.\n"
 253                        "No package add will be done.", name);
 254         ret = DITEM_FAILURE;
 255     }
 256     signal(SIGPIPE, SIG_IGN);
 257     return ret;
 258 }