Initial import from FreeBSD RELENG_4:
[dragonfly.git] / usr.sbin / pkg_install / lib / plist.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  * General packing list routines.
18  *
19  */
20
21 #include <sys/cdefs.h>
22 __FBSDID("$FreeBSD: src/usr.sbin/pkg_install/lib/plist.c,v 1.29.2.10 2002/08/31 19:38:55 obrien Exp $");
23
24 #include "lib.h"
25 #include <err.h>
26 #include <md5.h>
27
28 /* Add an item to a packing list */
29 void
30 add_plist(Package *p, plist_t type, const char *arg)
31 {
32     PackingList tmp;
33
34     tmp = new_plist_entry();
35     tmp->name = copy_string(arg);
36     tmp->type = type;
37
38     if (!p->head)
39         p->head = p->tail = tmp;
40     else {
41         tmp->prev = p->tail;
42         p->tail->next = tmp;
43         p->tail = tmp;
44     }
45     switch (type) {
46     case PLIST_NAME:
47         p->name = tmp->name;
48         break;
49
50     case PLIST_ORIGIN:
51         p->origin = tmp->name;
52         break;
53
54     default:
55         break;
56     }
57 }
58
59 void
60 add_plist_top(Package *p, plist_t type, const char *arg)
61 {
62     PackingList tmp;
63
64     tmp = new_plist_entry();
65     tmp->name = copy_string(arg);
66     tmp->type = type;
67
68     if (!p->head)
69         p->head = p->tail = tmp;
70     else {
71         tmp->next = p->head;
72         p->head->prev = tmp;
73         p->head = tmp;
74     }
75 }
76
77 /* Return the last (most recent) entry in a packing list */
78 PackingList
79 last_plist(Package *p)
80 {
81     return p->tail;
82 }
83
84 /* Mark all items in a packing list to prevent iteration over them */
85 void
86 mark_plist(Package *pkg)
87 {
88     PackingList p = pkg->head;
89
90     while (p) {
91         p->marked = TRUE;
92         p = p->next;
93     }
94 }
95
96 /* Find a given item in a packing list and, if so, return it (else NULL) */
97 PackingList
98 find_plist(Package *pkg, plist_t type)
99 {
100     PackingList p = pkg->head;
101
102     while (p) {
103         if (p->type == type)
104             return p;
105         p = p->next;
106     }
107     return NULL;
108 }
109
110 /* Look for a specific boolean option argument in the list */
111 char *
112 find_plist_option(Package *pkg, const char *name)
113 {
114     PackingList p = pkg->head;
115
116     while (p) {
117         if (p->type == PLIST_OPTION && !strcmp(p->name, name))
118             return p->name;
119         p = p->next;
120     }
121     return NULL;
122 }
123
124 /*
125  * Delete plist item 'type' in the list (if 'name' is non-null, match it
126  * too.)  If 'all' is set, delete all items, not just the first occurance.
127  */
128 void
129 delete_plist(Package *pkg, Boolean all, plist_t type, const char *name)
130 {
131     PackingList p = pkg->head;
132
133     while (p) {
134         PackingList pnext = p->next;
135
136         if (p->type == type && (!name || !strcmp(name, p->name))) {
137             free(p->name);
138             if (p->prev)
139                 p->prev->next = pnext;
140             else
141                 pkg->head = pnext;
142             if (pnext)
143                 pnext->prev = p->prev;
144             else
145                 pkg->tail = p->prev;
146             free(p);
147             if (!all)
148                 return;
149             p = pnext;
150         }
151         else
152             p = p->next;
153     }
154 }
155
156 /* Allocate a new packing list entry */
157 PackingList
158 new_plist_entry(void)
159 {
160     PackingList ret;
161
162     ret = (PackingList)malloc(sizeof(struct _plist));
163     bzero(ret, sizeof(struct _plist));
164     return ret;
165 }
166
167 /* Free an entire packing list */
168 void
169 free_plist(Package *pkg)
170 {
171     PackingList p = pkg->head;
172
173     while (p) {
174         PackingList p1 = p->next;
175
176         free(p->name);
177         free(p);
178         p = p1;
179     }
180     pkg->head = pkg->tail = NULL;
181 }
182
183 /*
184  * For an ascii string denoting a plist command, return its code and
185  * optionally its argument(s)
186  */
187 int
188 plist_cmd(const char *s, char **arg)
189 {
190     char cmd[FILENAME_MAX + 20];        /* 20 == fudge for max cmd len */
191     char *cp;
192     const char *sp;
193
194     strcpy(cmd, s);
195     str_lowercase(cmd);
196     cp = cmd;
197     sp = s;
198     while (*cp) {
199         if (isspace(*cp)) {
200             *cp = '\0';
201             while (isspace(*sp)) /* Never sure if macro, increment later */
202                 ++sp;
203             break;
204         }
205         ++cp, ++sp;
206     }
207     if (arg)
208         (const char *)*arg = sp;
209     if (!strcmp(cmd, "cwd"))
210         return PLIST_CWD;
211     else if (!strcmp(cmd, "srcdir"))
212         return PLIST_SRC;
213     else if (!strcmp(cmd, "cd"))
214         return PLIST_CWD;
215     else if (!strcmp(cmd, "exec"))
216         return PLIST_CMD;
217     else if (!strcmp(cmd, "unexec"))
218         return PLIST_UNEXEC;
219     else if (!strcmp(cmd, "mode"))
220         return PLIST_CHMOD;
221     else if (!strcmp(cmd, "owner"))
222         return PLIST_CHOWN;
223     else if (!strcmp(cmd, "group"))
224         return PLIST_CHGRP;
225     else if (!strcmp(cmd, "comment")) {
226         if (!strncmp(*arg, "ORIGIN:", 7)) {
227             *arg += 7;
228             return PLIST_ORIGIN;
229         } else if (!strncmp(*arg, "DEPORIGIN:", 10)) {
230             *arg += 10;
231             return PLIST_DEPORIGIN;
232         }
233         return PLIST_COMMENT;
234     } else if (!strcmp(cmd, "ignore"))
235         return PLIST_IGNORE;
236     else if (!strcmp(cmd, "ignore_inst"))
237         return PLIST_IGNORE_INST;
238     else if (!strcmp(cmd, "name"))
239         return PLIST_NAME;
240     else if (!strcmp(cmd, "display"))
241         return PLIST_DISPLAY;
242     else if (!strcmp(cmd, "pkgdep"))
243         return PLIST_PKGDEP;
244     else if (!strcmp(cmd, "mtree"))
245         return PLIST_MTREE;
246     else if (!strcmp(cmd, "dirrm"))
247         return PLIST_DIR_RM;
248     else if (!strcmp(cmd, "option"))
249         return PLIST_OPTION;
250     else
251         return FAIL;
252 }
253
254 /* Read a packing list from a file */
255 void
256 read_plist(Package *pkg, FILE *fp)
257 {
258     char *cp, pline[FILENAME_MAX];
259     int cmd, major, minor;
260
261     pkg->fmtver_maj = 1;
262     pkg->fmtver_mnr = 0;
263     pkg->origin = NULL;
264     while (fgets(pline, FILENAME_MAX, fp)) {
265         int len = strlen(pline);
266
267         while (len && isspace(pline[len - 1]))
268             pline[--len] = '\0';
269         if (!len)
270             continue;
271         cp = pline;
272         if (pline[0] != CMD_CHAR) {
273             cmd = PLIST_FILE;
274             goto bottom;
275         }
276         cmd = plist_cmd(pline + 1, &cp);
277         if (cmd == FAIL) {
278             cleanup(0);
279             errx(2, "%s: bad command '%s'", __func__, pline);
280         }
281         if (*cp == '\0') {
282             cp = NULL;
283             goto bottom;
284         }
285         if (cmd == PLIST_COMMENT && sscanf(cp, "PKG_FORMAT_REVISION:%d.%d\n",
286                                            &major, &minor) == 2) {
287             pkg->fmtver_maj = major;
288             pkg->fmtver_mnr = minor;
289             if (verscmp(pkg, PLIST_FMT_VER_MAJOR, PLIST_FMT_VER_MINOR) <= 0)
290                 goto bottom;
291
292             warnx("plist format revision (%d.%d) is higher than supported"
293                   "(%d.%d)", pkg->fmtver_maj, pkg->fmtver_mnr,
294                   PLIST_FMT_VER_MAJOR, PLIST_FMT_VER_MINOR);
295             if (pkg->fmtver_maj > PLIST_FMT_VER_MAJOR) {
296                 cleanup(0);
297                 exit(2);
298             }
299         }
300 bottom:
301         add_plist(pkg, cmd, cp);
302     }
303 }
304
305 /* Write a packing list to a file, converting commands to ascii equivs */
306 void
307 write_plist(Package *pkg, FILE *fp)
308 {
309     PackingList plist = pkg->head;
310
311     while (plist) {
312         switch(plist->type) {
313         case PLIST_FILE:
314             fprintf(fp, "%s\n", plist->name);
315             break;
316
317         case PLIST_CWD:
318             fprintf(fp, "%ccwd %s\n", CMD_CHAR, plist->name);
319             break;
320
321         case PLIST_SRC:
322             fprintf(fp, "%csrcdir %s\n", CMD_CHAR, plist->name);
323             break;
324
325         case PLIST_CMD:
326             fprintf(fp, "%cexec %s\n", CMD_CHAR, plist->name);
327             break;
328
329         case PLIST_UNEXEC:
330             fprintf(fp, "%cunexec %s\n", CMD_CHAR, plist->name);
331             break;
332
333         case PLIST_CHMOD:
334             fprintf(fp, "%cmode %s\n", CMD_CHAR, plist->name ? plist->name : "");
335             break;
336
337         case PLIST_CHOWN:
338             fprintf(fp, "%cowner %s\n", CMD_CHAR, plist->name ? plist->name : "");
339             break;
340
341         case PLIST_CHGRP:
342             fprintf(fp, "%cgroup %s\n", CMD_CHAR, plist->name ? plist->name : "");
343             break;
344
345         case PLIST_COMMENT:
346             fprintf(fp, "%ccomment %s\n", CMD_CHAR, plist->name);
347             break;
348
349         case PLIST_IGNORE:
350         case PLIST_IGNORE_INST:         /* a one-time non-ignored file */
351             fprintf(fp, "%cignore\n", CMD_CHAR);
352             break;
353
354         case PLIST_NAME:
355             fprintf(fp, "%cname %s\n", CMD_CHAR, plist->name);
356             break;
357
358         case PLIST_DISPLAY:
359             fprintf(fp, "%cdisplay %s\n", CMD_CHAR, plist->name);
360             break;
361
362         case PLIST_PKGDEP:
363             fprintf(fp, "%cpkgdep %s\n", CMD_CHAR, plist->name);
364             break;
365
366         case PLIST_MTREE:
367             fprintf(fp, "%cmtree %s\n", CMD_CHAR, plist->name);
368             break;
369
370         case PLIST_DIR_RM:
371             fprintf(fp, "%cdirrm %s\n", CMD_CHAR, plist->name);
372             break;
373
374         case PLIST_OPTION:
375             fprintf(fp, "%coption %s\n", CMD_CHAR, plist->name);
376             break;
377
378         case PLIST_ORIGIN:
379             fprintf(fp, "%ccomment ORIGIN:%s\n", CMD_CHAR, plist->name);
380             break;
381
382         case PLIST_DEPORIGIN:
383             fprintf(fp, "%ccomment DEPORIGIN:%s\n", CMD_CHAR, plist->name);
384             break;
385
386         default:
387             cleanup(0);
388             errx(2, "%s: unknown command type %d (%s)", __func__,
389                 plist->type, plist->name);
390             break;
391         }
392         plist = plist->next;
393     }
394 }
395
396 /*
397  * Delete the results of a package installation.
398  *
399  * This is here rather than in the pkg_delete code because pkg_add needs to
400  * run it too in cases of failure.
401  */
402 int
403 delete_package(Boolean ign_err, Boolean nukedirs, Package *pkg)
404 {
405     PackingList p;
406     const char *Where = ".", *last_file = "";
407     Boolean fail = SUCCESS;
408     Boolean preserve;
409     char tmp[FILENAME_MAX], *name = NULL;
410
411     preserve = find_plist_option(pkg, "preserve") ? TRUE : FALSE;
412     for (p = pkg->head; p; p = p->next) {
413         switch (p->type)  {
414         case PLIST_NAME:
415             name = p->name;
416             break;
417
418         case PLIST_IGNORE:
419             p = p->next;
420             break;
421
422         case PLIST_CWD:
423             Where = p->name;
424             if (Verbose)
425                 printf("Change working directory to %s\n", Where);
426             break;
427
428         case PLIST_UNEXEC:
429             format_cmd(tmp, p->name, Where, last_file);
430             if (Verbose)
431                 printf("Execute '%s'\n", tmp);
432             if (!Fake && system(tmp)) {
433                 warnx("unexec command for '%s' failed", tmp);
434                 fail = FAIL;
435             }
436             break;
437
438         case PLIST_FILE:
439             last_file = p->name;
440             sprintf(tmp, "%s/%s", Where, p->name);
441             if (isdir(tmp) && fexists(tmp) && !issymlink(tmp)) {
442                 warnx("cannot delete specified file '%s' - it is a directory!\n"
443            "this packing list is incorrect - ignoring delete request", tmp);
444             }
445             else {
446                 if (p->next && p->next->type == PLIST_COMMENT && !strncmp(p->next->name, "MD5:", 4)) {
447                     char *cp = NULL, buf[33];
448
449                     /*
450                      * For packing lists whose version is 1.1 or greater, the md5
451                      * hash for a symlink is calculated on the string returned
452                      * by readlink().
453                      */
454                     if (issymlink(tmp) && verscmp(pkg, 1, 0) > 0) {
455                         int len;
456                         char linkbuf[FILENAME_MAX];
457
458                         if ((len = readlink(tmp, linkbuf, FILENAME_MAX)) > 0)
459                              cp = MD5Data((unsigned char *)linkbuf, len, buf);
460                     } else if (isfile(tmp) || verscmp(pkg, 1, 1) < 0)
461                         cp = MD5File(tmp, buf);
462
463                     if (cp != NULL) {
464                         /* Mismatch? */
465                         if (strcmp(cp, p->next->name + 4)) {
466                             warnx("'%s' fails original MD5 checksum - %s",
467                                        tmp, Force ? "deleted anyway." : "not deleted.");
468                             if (!Force) {
469                                 fail = FAIL;
470                                 continue;
471                             }
472                         }
473                     }
474                 }
475                 if (Verbose)
476                     printf("Delete file %s\n", tmp);
477                 if (!Fake) {
478                     if (delete_hierarchy(tmp, ign_err, nukedirs))
479                         fail = FAIL;
480                     if (preserve && name) {
481                         char tmp2[FILENAME_MAX];
482                             
483                         if (make_preserve_name(tmp2, FILENAME_MAX, name, tmp)) {
484                             if (fexists(tmp2)) {
485                                 if (rename(tmp2, tmp))
486                                    warn("preserve: unable to restore %s as %s",
487                                         tmp2, tmp);
488                             }
489                         }
490                     }
491                 }
492             }
493             break;
494
495         case PLIST_DIR_RM:
496             sprintf(tmp, "%s/%s", Where, p->name);
497             if (!isdir(tmp) && fexists(tmp)) {
498                 warnx("cannot delete specified directory '%s' - it is a file!\n"
499         "this packing list is incorrect - ignoring delete request", tmp);
500             }
501             else {
502                 if (Verbose)
503                     printf("Delete directory %s\n", tmp);
504                 if (!Fake && delete_hierarchy(tmp, ign_err, FALSE)) {
505                     warnx("unable to completely remove directory '%s'", tmp);
506                     fail = FAIL;
507                 }
508             }
509             last_file = p->name;
510             break;
511
512         default:
513             break;
514         }
515     }
516     return fail;
517 }
518
519 #ifdef DEBUG
520 #define RMDIR(dir) vsystem("%s %s", RMDIR_CMD, dir)
521 #define REMOVE(dir,ie) vsystem("%s %s%s", REMOVE_CMD, (ie ? "-f " : ""), dir)
522 #else
523 #define RMDIR rmdir
524 #define REMOVE(file,ie) (remove(file) && !(ie))
525 #endif
526
527 /* Selectively delete a hierarchy */
528 int
529 delete_hierarchy(const char *dir, Boolean ign_err, Boolean nukedirs)
530 {
531     char *cp1, *cp2;
532
533     cp1 = cp2 = strdup(dir);
534     if (!fexists(dir)) {
535         if (!ign_err)
536             warnx("%s '%s' doesn't really exist",
537                 isdir(dir) ? "directory" : "file", dir);
538         return !ign_err;
539     }
540     else if (nukedirs) {
541         if (vsystem("%s -r%s %s", REMOVE_CMD, (ign_err ? "f" : ""), dir))
542             return 1;
543     }
544     else if (isdir(dir) && !issymlink(dir)) {
545         if (RMDIR(dir) && !ign_err)
546             return 1;
547     }
548     else {
549         if (REMOVE(dir, ign_err))
550             return 1;
551     }
552
553     if (!nukedirs)
554         return 0;
555     while (cp2) {
556         if ((cp2 = strrchr(cp1, '/')) != NULL)
557             *cp2 = '\0';
558         if (!isemptydir(dir))
559             return 0;
560         if (RMDIR(dir) && !ign_err) {
561             if (!fexists(dir))
562                 warnx("directory '%s' doesn't really exist", dir);
563             else
564                 return 1;
565         }
566         /* back up the pathname one component */
567         if (cp2) {
568             cp1 = strdup(dir);
569         }
570     }
571     return 0;
572 }