/*	$NetBSD: extract.c,v 1.34 2005/01/06 11:59:35 agc Exp $	*/

#include <sys/cdefs.h>
#ifndef lint
#if 0
static const char *rcsid = "FreeBSD - Id: extract.c,v 1.17 1997/10/08 07:45:35 charnier Exp";
#else
__RCSID("$NetBSD: extract.c,v 1.34 2005/01/06 11:59:35 agc Exp $");
#endif
#endif

/*
 * FreeBSD install - a package for the installation and maintainance
 * of non-core utilities.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * Jordan K. Hubbard
 * 18 July 1993
 *
 * This is the package extraction code for the add module.
 *
 */

#include <err.h>
#include "lib.h"
#include "add.h"

#define TAR_ARGS	" cf - "
#define TARX_CMD	"|" TAR_CMD " xpf - -C "

/* 
 * This macro is used to determine if the 'where_args'  buffer is big enough to add the
 * current string (usually a filename) plus some extra commands (the contents of TARX_CMD, and
 * the directory name stored in 'Directory').
 * 
 * The string " 'str'" will be added so we need room for the string plus 3 chars plus the other arguments.
 * 
 * In addition, we will add " 'srt'" to the perm_args buffer so we need to ensure that there is room
 * for that.
 */
#define TOOBIG(str) ((strlen(str) + 3 + strlen(TARX_CMD) + strlen(Directory) + where_count  >= maxargs) \
		|| (strlen(str) + 3 + perm_count >= maxargs))

#define PUSHOUT(todir) /* push out string */				\
        if (where_count > sizeof(TAR_CMD) + sizeof(TAR_ARGS)-1) {		\
		    strlcat(where_args, TARX_CMD, maxargs);		\
		    strlcat(where_args, todir, maxargs);		\
		    if (system(where_args)) {				\
			cleanup(0);					\
			errx(2, "can not invoke %lu byte %s pipeline: %s", \
				(u_long)strlen(where_args), TAR_CMD,	\
				where_args);				\
		    }							\
		    strlcpy(where_args, TAR_CMD TAR_ARGS, maxargs);	\
		    where_count = strlen(where_args);			\
	}								\
	if (perm_count) {						\
		    apply_perms(todir, perm_args);			\
		    perm_args[0] = 0;					\
		    perm_count = 0;					\
	}


static void
rollback(char *name, char *home, plist_t *start, plist_t *stop)
{
	plist_t *q;
	char    try[MaxPathSize], bup[MaxPathSize], *dir;

	dir = home;
	for (q = start; q != stop; q = q->next) {
		if (q->type == PLIST_FILE) {
			(void) snprintf(try, sizeof(try), "%s/%s", dir, q->name);
			if (make_preserve_name(bup, sizeof(bup), name, try) && fexists(bup)) {
				(void) chflags(try, 0);
				(void) unlink(try);
				if (rename(bup, try))
					warnx("rollback: unable to rename %s back to %s", bup, try);
			}
		} else if (q->type == PLIST_CWD) {
			if (strcmp(q->name, "."))
				dir = q->name;
			else
				dir = home;
		}
	}
}


/*
 * Return 0 on error, 1 for success.
 */
int
extract_plist(char *home, package_t *pkg)
{
	plist_t *p = pkg->head;
	char   *last_file;
	char   *where_args, *perm_args, *last_chdir;
	int     maxargs, where_count = 0, perm_count = 0, add_count;
	Boolean preserve;

	maxargs = sysconf(_SC_ARG_MAX) / 2;	/* Just use half the argument space */
	where_args = malloc(maxargs);
	if (!where_args) {
		cleanup(0);
		errx(2, "can't get argument list space");
	}
	perm_args = malloc(maxargs);
	if (!perm_args) {
		cleanup(0);
		errx(2, "can't get argument list space");
	}
	strlcpy(where_args, TAR_CMD TAR_ARGS, maxargs);
	/*
	 * we keep track of how many characters are stored in 'where_args' with 'where_count'.
	 * Note this doesn't include the trailing null character.
	 */
	where_count = strlen(where_args);

	perm_args[0] = 0;
	/*
	 * we keep track of how many characters are stored in 'perm__args' with 'perm_count'.
	 * Note this doesn't include the trailing null character.
	 */
	perm_count = 0;

	last_chdir = 0;
	preserve = find_plist_option(pkg, "preserve") ? TRUE : FALSE;

	/* Reset the world */
	Owner = NULL;
	Group = NULL;
	Mode = NULL;
	last_file = NULL;
	Directory = home;

	/* Open Package Database for writing */
	if (!pkgdb_open(ReadWrite)) {
		cleanup(0);
		err(EXIT_FAILURE, "can't open pkgdb");
	}
	/* Do it */
	while (p) {
		char    cmd[MaxPathSize];

		switch (p->type) {
		case PLIST_NAME:
			PkgName = p->name;
			if (Verbose)
				printf("extract: Package name is %s\n", p->name);
			break;

		case PLIST_FILE:
			last_file = p->name;
			if (Verbose)
				printf("extract: %s/%s\n", Directory, p->name);
			if (!Fake) {
				char    try[MaxPathSize];

				if (strrchr(p->name, '\'')) {
					cleanup(0);
					errx(2, "Bogus filename \"%s\"", p->name);
				}

				/* first try to rename it into place */
				(void) snprintf(try, sizeof(try), "%s/%s", Directory, p->name);
				if (fexists(try)) {
					(void) chflags(try, 0);	/* XXX hack - if truly immutable, rename fails */
					if (preserve && PkgName) {
						char    pf[MaxPathSize];

						if (make_preserve_name(pf, sizeof(pf), PkgName, try)) {
							if (rename(try, pf)) {
								warnx(
								    "unable to back up %s to %s, aborting pkg_add",
								    try, pf);
								rollback(PkgName, home, pkg->head, p);
								free(perm_args);
								free(where_args);
								return 0;
							}
						}
					}
				}
				if (rename(p->name, try) == 0) {
					/* note in pkgdb */
					{
						char   *s, t[MaxPathSize];
						int     rc;

						(void) snprintf(t, sizeof(t), "%s/%s", Directory, p->name);

						s = pkgdb_retrieve(t);
#ifdef PKGDB_DEBUG
						printf("pkgdb_retrieve(\"%s\")=\"%s\"\n", t, s);	/* pkgdb-debug - HF */
#endif
						if (s)
							warnx("Overwriting %s - pkg %s bogus/conflicting?", t, s);
						else {
							rc = pkgdb_store(t, PkgName);
#ifdef PKGDB_DEBUG
							printf("pkgdb_store(\"%s\", \"%s\") = %d\n", t, PkgName, rc);	/* pkgdb-debug - HF */
#endif

						}
					}

					/* try to add to list of perms to be changed and run in bulk. */
					if (p->name[0] == '/' || TOOBIG(p->name)) {
						PUSHOUT(Directory);
					}
					/* note, if the following line is modified, TOOBIG must be adjusted accordingly */
					add_count = snprintf(&perm_args[perm_count], maxargs - perm_count, "'%s' ", p->name);
					if (add_count > maxargs - perm_count) {
						cleanup(0);
						errx(2, "oops, miscounted strings!");
					}
					perm_count += add_count;
				} else {
					/* rename failed, try copying with a big tar command */
					if (last_chdir != Directory) {
						PUSHOUT(last_chdir);
						last_chdir = Directory;
					} else if (p->name[0] == '/' || TOOBIG(p->name)) {
						PUSHOUT(Directory);
					}
					/* note, if the following line is modified, TOOBIG must be adjusted accordingly */
					add_count = snprintf(&where_args[where_count], maxargs - where_count, " '%s'", p->name);
					if (add_count > maxargs - where_count) {
						cleanup(0);
						errx(2, "oops, miscounted strings!");
					}
					where_count += add_count;
					/* note, if the following line is modified, TOOBIG must be adjusted accordingly */
					add_count = snprintf(&perm_args[perm_count],
					    maxargs - perm_count,
					    "'%s' ", p->name);
					if (add_count > maxargs - perm_count) {
						cleanup(0);
						errx(2, "oops, miscounted strings!");
					}
					perm_count += add_count;

					/* note in pkgdb */
					/* XXX would be better to store in PUSHOUT, but
					 * that would probably affect too much code I prefer
					 * not to touch - HF */
					{
						char   *s, t[MaxPathSize], *u;
						int     rc;

						if (p->name[0] == '/')
							u = p->name;
						else {
							(void) snprintf(t, sizeof(t), "%s/%s", Directory, p->name);
							u = t;
						}

						s = pkgdb_retrieve(t);
#ifdef PKGDB_DEBUG
						printf("pkgdb_retrieve(\"%s\")=\"%s\"\n", t, s);	/* pkgdb-debug - HF */
#endif
						if (s)
							warnx("Overwriting %s - pkg %s bogus/conflicting?", t, s);
						else {
							rc = pkgdb_store(t, PkgName);
#ifdef PKGDB_DEBUG
							printf("pkgdb_store(\"%s\", \"%s\") = %d\n", t, PkgName, rc);	/* pkgdb-debug - HF */
#endif
						}
					}
				}
			}
			break;

		case PLIST_CWD:
			if (Verbose)
				printf("extract: CWD to %s\n", p->name);
			PUSHOUT(Directory);
			if (strcmp(p->name, ".")) {
				if (!Fake && make_hierarchy(p->name) == FAIL) {
					cleanup(0);
					errx(2, "unable to make directory '%s'", p->name);
				}
				Directory = p->name;
			} else
				Directory = home;
			break;

		case PLIST_CMD:
			format_cmd(cmd, sizeof(cmd), p->name, Directory, last_file);
			PUSHOUT(Directory);
			printf("Executing '%s'\n", cmd);
			if (!Fake && system(cmd))
				warnx("command '%s' failed", cmd);
			break;

		case PLIST_CHMOD:
			PUSHOUT(Directory);
			Mode = p->name;
			break;

		case PLIST_CHOWN:
			PUSHOUT(Directory);
			Owner = p->name;
			break;

		case PLIST_CHGRP:
			PUSHOUT(Directory);
			Group = p->name;
			break;

		case PLIST_COMMENT:
			break;

		case PLIST_IGNORE:
			p = p->next;
			break;

		default:
			break;
		}
		p = p->next;
	}
	PUSHOUT(Directory);
	pkgdb_close();
	free(perm_args);
	free(where_args);
	return 1;
}
