usr.sbin/pkg_install/add/extract.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  * This is the package extraction code for the add module.
  18  *
  19  */
  20
  21 #include <sys/cdefs.h>
  22 __FBSDID("$FreeBSD$");
  23
  24 #include <ctype.h>
  25 #include <err.h>
  26 #include "lib.h"
  27 #include "add.h"
  28
  29
  30 #define STARTSTRING "/usr/bin/tar cf -"
  31 #define TOOBIG(str) \
  32     (((int)strlen(str) + FILENAME_MAX + where_count > maxargs) ||\
  33         ((int)strlen(str) + FILENAME_MAX + perm_count > maxargs))
  34
  35 #define PUSHOUT(todir) /* push out string */ \
  36     if (where_count > (int)sizeof(STARTSTRING)-1) { \
  37         strcat(where_args, "|/usr/bin/tar --unlink -xpPf - -C "); \
  38         strcat(where_args, todir); \
  39         if (system(where_args)) { \
  40             cleanup(0); \
  41             errx(2, "%s: can not invoke %ld byte tar pipeline: %s", \
  42                  __func__, (long)strlen(where_args), where_args); \
  43         } \
  44         strcpy(where_args, STARTSTRING); \
  45         where_count = sizeof(STARTSTRING)-1; \
  46     } \
  47     if (perm_count) { \
  48         apply_perms(todir, perm_args); \
  49         perm_args[0] = 0;\
  50         perm_count = 0; \
  51     }
  52
  53 static void
  54 rollback(const char *name, const char *home, PackingList start, PackingList stop)
  55 {
  56     PackingList q;
  57     char try[FILENAME_MAX], bup[FILENAME_MAX];
  58     const char *dir;
  59     char *prefix = NULL;
  60
  61     dir = home;
  62     for (q = start; q != stop; q = q->next) {
  63         if (q->type == PLIST_FILE) {
  64             snprintf(try, FILENAME_MAX, "%s/%s", dir, q->name);
  65             if (make_preserve_name(bup, FILENAME_MAX, name, try) && fexists(bup)) {
  66                 (void)chflags(try, 0);
  67                 (void)unlink(try);
  68                 if (rename(bup, try))
  69                     warnx("rollback: unable to rename %s back to %s", bup, try);
  70             }
  71         }
  72         else if (q->type == PLIST_CWD) {
  73             if (!prefix)
  74                 prefix = q->name;
  75             if (q->name == NULL)
  76                 q->name = prefix;
  77             else if (strcmp(q->name, "."))
  78                 dir = q->name;
  79             else
  80                 dir = home;
  81         }
  82     }
  83 }
  84
  85 #define add_char(buf, len, pos, ch) do {\
  86     if ((pos) < (len)) { \
  87         buf[(pos)] = (ch); \
  88         buf[(pos) + 1] = '\0'; \
  89     } \
  90     ++(pos); \
  91 } while (0)
  92
  93 static int
  94 add_arg(char *buf, int len, const char *str)
  95 {
  96     int i = 0;
  97
  98     add_char(buf, len, i, ' ');
  99     for (; *str != '\0'; ++str) {
 100         if (!isalnum(*str) && *str != '/' && *str != '.' && *str != '-')
 101             add_char(buf, len, i, '\\');
 102         add_char(buf, len, i, *str);
 103     }
 104     return (i);
 105 }
 106
 107 void
 108 extract_plist(const char *home, Package *pkg)
 109 {
 110     PackingList p = pkg->head;
 111     char *last_file, *prefix = NULL;
 112     char *where_args, *perm_args, *last_chdir;
 113     long maxargs;
 114     int where_count = 0, perm_count = 0, add_count;
 115     Boolean preserve;
 116
 117     maxargs = sysconf(_SC_ARG_MAX) / 2; /* Just use half the argument space */
 118     where_args = alloca(maxargs);
 119     if (!where_args) {
 120         cleanup(0);
 121         errx(2, "%s: can't get argument list space", __func__);
 122     }
 123     perm_args = alloca(maxargs);
 124     if (!perm_args) {
 125         cleanup(0);
 126         errx(2, "%s: can't get argument list space", __func__);
 127     }
 128
 129     strcpy(where_args, STARTSTRING);
 130     where_count = sizeof(STARTSTRING)-1;
 131     perm_args[0] = 0;
 132
 133     last_chdir = 0;
 134     preserve = find_plist_option(pkg, "preserve") ? TRUE : FALSE;
 135
 136     /* Reset the world */
 137     Owner = NULL;
 138     Group = NULL;
 139     Mode = NULL;
 140     last_file = NULL;
 141     Directory = (char *)home;
 142
 143     /* Do it */
 144     while (p) {
 145         char cmd[FILENAME_MAX];
 146
 147         switch(p->type) {
 148         case PLIST_NAME:
 149             PkgName = p->name;
 150             if (Verbose)
 151                 printf("extract: Package name is %s\n", p->name);
 152             break;
 153
 154         case PLIST_FILE:
 155             last_file = p->name;
 156             if (Verbose)
 157                 printf("extract: %s/%s\n", Directory, p->name);
 158             if (!Fake) {
 159                 char try[FILENAME_MAX];
 160
 161                 /* first try to rename it into place */
 162                 snprintf(try, FILENAME_MAX, "%s/%s", Directory, p->name);
 163                 if (fexists(try)) {
 164                     (void)chflags(try, 0);      /* XXX hack - if truly immutable, rename fails */
 165                     if (preserve && PkgName) {
 166                         char pf[FILENAME_MAX];
 167
 168                         if (make_preserve_name(pf, FILENAME_MAX, PkgName, try)) {
 169                             if (rename(try, pf)) {
 170                                 warnx(
 171                                 "unable to back up %s to %s, aborting pkg_add",
 172                                 try, pf);
 173                                 rollback(PkgName, home, pkg->head, p);
 174                                 return;
 175                             }
 176                         }
 177                     }
 178                 }
 179                 if (rename(p->name, try) == 0) {
 180                     /* try to add to list of perms to be changed and run in bulk. */
 181                     if (p->name[0] == '/' || TOOBIG(p->name)) {
 182                         PUSHOUT(Directory);
 183                     }
 184                     add_count = add_arg(&perm_args[perm_count], maxargs - perm_count, p->name);
 185                     if (add_count < 0 || add_count >= maxargs - perm_count) {
 186                         cleanup(0);
 187                         errx(2, "%s: oops, miscounted strings!", __func__);
 188                     }
 189                     perm_count += add_count;
 190                 }
 191                 else {
 192                     /* rename failed, try copying with a big tar command */
 193                     if (last_chdir != Directory) {
 194                         if (last_chdir == NULL) {
 195                             PUSHOUT(Directory);
 196                         } else {
 197                             PUSHOUT(last_chdir);
 198                         }
 199                         last_chdir = Directory;
 200                     }
 201                     else if (p->name[0] == '/' || TOOBIG(p->name)) {
 202                         PUSHOUT(Directory);
 203                     }
 204                     add_count = add_arg(&where_args[where_count], maxargs - where_count, p->name);
 205                     if (add_count < 0 || add_count >= maxargs - where_count) {
 206                         cleanup(0);
 207                         errx(2, "%s: oops, miscounted strings!", __func__);
 208                     }
 209                     where_count += add_count;
 210                     add_count = add_arg(&perm_args[perm_count], maxargs - perm_count, p->name);
 211                     if (add_count < 0 || add_count >= maxargs - perm_count) {
 212                         cleanup(0);
 213                         errx(2, "%s: oops, miscounted strings!", __func__);
 214                     }
 215                     perm_count += add_count;
 216                 }
 217             }
 218             break;
 219
 220         case PLIST_CWD:
 221             if (!prefix)
 222                 prefix = p->name;
 223             if (p->name == NULL)
 224                 p->name = strdup(prefix);
 225             if (Verbose)
 226                 printf("extract: CWD to %s\n", p->name);
 227             PUSHOUT(Directory);
 228             if (strcmp(p->name, ".")) {
 229                 if (!Fake && make_hierarchy(p->name, TRUE) == FAIL) {
 230                     cleanup(0);
 231                     errx(2, "%s: unable to cwd to '%s'", __func__, p->name);
 232                 }
 233                 Directory = p->name;
 234             }
 235             else
 236                 Directory = (char *)home;
 237             break;
 238
 239         case PLIST_CMD:
 240             if ((strstr(p->name, "%B") || strstr(p->name, "%F") ||
 241                  strstr(p->name, "%f")) && last_file == NULL) {
 242                 cleanup(0);
 243                 errx(2, "%s: no last file specified for '%s' command",
 244                     __func__, p->name);
 245             }
 246             if (strstr(p->name, "%D") && Directory == NULL) {
 247                 cleanup(0);
 248                 errx(2, "%s: no directory specified for '%s' command",
 249                     __func__, p->name);
 250             }
 251             format_cmd(cmd, FILENAME_MAX, p->name, Directory, last_file);
 252             PUSHOUT(Directory);
 253             if (Verbose)
 254                 printf("extract: execute '%s'\n", cmd);
 255             if (!Fake && system(cmd))
 256                 warnx("command '%s' failed", cmd);
 257             break;
 258
 259         case PLIST_CHMOD:
 260             PUSHOUT(Directory);
 261             Mode = p->name;
 262             break;
 263
 264         case PLIST_CHOWN:
 265             PUSHOUT(Directory);
 266             Owner = p->name;
 267             break;
 268
 269         case PLIST_CHGRP:
 270             PUSHOUT(Directory);
 271             Group = p->name;
 272             break;
 273
 274         case PLIST_COMMENT: /* FALLTHROUGH */
 275         case PLIST_NOINST:
 276             break;
 277
 278         case PLIST_IGNORE:
 279             p = p->next;
 280             break;
 281
 282         default:
 283             break;
 284         }
 285         p = p->next;
 286     }
 287     PUSHOUT(Directory);
 288 }