usr.sbin/pkg_install/lib/pen.c

   1 /*
   2  * FreeBSD install - a package for the installation and maintainance
   3  * of non-core utilities.
   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  * Jordan K. Hubbard
  15  * 18 July 1993
  16  *
  17  * Routines for managing the "play pen".
  18  *
  19  */
  20
  21 #include <sys/cdefs.h>
  22 __FBSDID("$FreeBSD$");
  23
  24 #include "lib.h"
  25 #include <err.h>
  26 #include <libgen.h>
  27 #include <sys/signal.h>
  28 #include <sys/param.h>
  29 #include <sys/mount.h>
  30
  31 /* For keeping track of where we are */
  32 static char PenLocation[FILENAME_MAX];
  33 static char Previous[FILENAME_MAX];
  34
  35 char *
  36 where_playpen(void)
  37 {
  38     return PenLocation;
  39 }
  40
  41 /* Find a good place to play. */
  42 static char *
  43 find_play_pen(char *pen, off_t sz)
  44 {
  45     char *cp;
  46     struct stat sb;
  47
  48     if (pen[0] && isdir(dirname(pen)) == TRUE && (min_free(dirname(pen)) >= sz))
  49         return pen;
  50     else if ((cp = getenv("PKG_TMPDIR")) != NULL && stat(cp, &sb) != FAIL && (min_free(cp) >= sz))
  51         sprintf(pen, "%s/instmp.XXXXXX", cp);
  52     else if ((cp = getenv("TMPDIR")) != NULL && stat(cp, &sb) != FAIL && (min_free(cp) >= sz))
  53         sprintf(pen, "%s/instmp.XXXXXX", cp);
  54     else if (stat("/var/tmp", &sb) != FAIL && min_free("/var/tmp") >= sz)
  55         strcpy(pen, "/var/tmp/instmp.XXXXXX");
  56     else if (stat("/tmp", &sb) != FAIL && min_free("/tmp") >= sz)
  57         strcpy(pen, "/tmp/instmp.XXXXXX");
  58     else if ((stat("/usr/tmp", &sb) == SUCCESS || mkdir("/usr/tmp", 01777) == SUCCESS) && min_free("/usr/tmp") >= sz)
  59         strcpy(pen, "/usr/tmp/instmp.XXXXXX");
  60     else {
  61         cleanup(0);
  62         errx(2,
  63 "%s: can't find enough temporary space to extract the files, please set your\n"
  64 "PKG_TMPDIR environment variable to a location with at least %ld bytes\n"
  65 "free", __func__, (long)sz);
  66         return NULL;
  67     }
  68     return pen;
  69 }
  70
  71 #define MAX_STACK       20
  72 static char *pstack[MAX_STACK];
  73 static int pdepth = -1;
  74
  75 static void
  76 pushPen(const char *pen)
  77 {
  78     if (++pdepth == MAX_STACK)
  79         errx(2, "%s: stack overflow.\n", __func__);
  80     pstack[pdepth] = strdup(pen);
  81 }
  82
  83 static void
  84 popPen(char *pen)
  85 {
  86     if (pdepth == -1) {
  87         pen[0] = '\0';
  88         return;
  89     }
  90     strcpy(pen, pstack[pdepth]);
  91     free(pstack[pdepth--]);
  92 }
  93
  94 /*
  95  * Make a temporary directory to play in and chdir() to it, returning
  96  * pathname of previous working directory.
  97  */
  98 char *
  99 make_playpen(char *pen, off_t sz)
 100 {
 101     if (!find_play_pen(pen, sz))
 102         return NULL;
 103
 104     if (!mkdtemp(pen)) {
 105         cleanup(0);
 106         errx(2, "%s: can't mktemp '%s'", __func__, pen);
 107     }
 108     if (chmod(pen, 0700) == FAIL) {
 109         cleanup(0);
 110         errx(2, "%s: can't mkdir '%s'", __func__, pen);
 111     }
 112
 113     if (Verbose) {
 114         if (sz)
 115             fprintf(stderr, "Requested space: %d bytes, free space: %lld bytes in %s\n", (int)sz, (long long)min_free(pen), pen);
 116     }
 117
 118     if (min_free(pen) < sz) {
 119         rmdir(pen);
 120         cleanup(0);
 121         errx(2, "%s: not enough free space to create '%s'.\n"
 122              "Please set your PKG_TMPDIR environment variable to a location\n"
 123              "with more space and\ntry the command again", __func__, pen);
 124     }
 125
 126     if (!getcwd(Previous, FILENAME_MAX)) {
 127         upchuck("getcwd");
 128         return NULL;
 129     }
 130
 131     if (chdir(pen) == FAIL) {
 132         cleanup(0);
 133         errx(2, "%s: can't chdir to '%s'", __func__, pen);
 134     }
 135
 136     if (PenLocation[0])
 137         pushPen(PenLocation);
 138
 139     strcpy(PenLocation, pen);
 140     return Previous;
 141 }
 142
 143 /* Convenience routine for getting out of playpen */
 144 void
 145 leave_playpen()
 146 {
 147     void (*oldsig)(int);
 148
 149     /* Don't interrupt while we're cleaning up */
 150     oldsig = signal(SIGINT, SIG_IGN);
 151     if (Previous[0]) {
 152         if (chdir(Previous) == FAIL) {
 153             cleanup(0);
 154             errx(2, "%s: can't chdir back to '%s'", __func__, Previous);
 155         }
 156         Previous[0] = '\0';
 157     }
 158     if (PenLocation[0]) {
 159         if (PenLocation[0] == '/' && vsystem("/bin/rm -rf %s", PenLocation))
 160             warnx("couldn't remove temporary dir '%s'", PenLocation);
 161         popPen(PenLocation);
 162     }
 163     signal(SIGINT, oldsig);
 164 }
 165
 166 off_t
 167 min_free(const char *tmpdir)
 168 {
 169     struct statfs buf;
 170
 171     if (statfs(tmpdir, &buf) != 0) {
 172         warn("statfs");
 173         return -1;
 174     }
 175     return (off_t)buf.f_bavail * (off_t)buf.f_bsize;
 176 }