svr4pkg/pkgscripts/cmdexec.c

/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */

/*
 * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

/* Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T */
/* All Rights Reserved */


#include <stdio.h>
#include <ctype.h>
#include <string.h>
#include <locale.h>
#include <libintl.h>
#include <stdlib.h>
#include <unistd.h>
#include <pkglib.h>

#define COMMAND '!'
#define LSIZE 256

#define ERR_NOTROOT "You must be \"root\" for %s to execute properly."

static void usage(void);
static int  docmd(char *cmd, char *file, char *input);

int
main(int argc, char *argv[])
{
    FILE    *fpout, *fp;
    char    line[LSIZE],
        *pt,
        *keyword,   /* keyword = install || remove */
        *input,     /* sed input file */
        *cmd,
        *srcfile,   /* sed data file */
        *destfile;  /* target file to be updated */
    int flag;
    char    *prog;

    (void) setlocale(LC_ALL, "");

#if !defined(TEXT_DOMAIN)   /* Should be defined by cc -D */
#define TEXT_DOMAIN "SYS_TEST"
#endif
    (void) textdomain(TEXT_DOMAIN);

    prog = set_prog_name(argv[0]);

    if (getuid()) {
        progerr(gettext(ERR_NOTROOT), prog);
        exit(1);
    }

    if (argc != 5)
        usage();

    cmd = argv[1];
    keyword = argv[2];
    srcfile = argv[3];
    destfile = argv[4];

    srcfile = argv[3];
    if ((fp = fopen(srcfile, "r")) == NULL) {
        progerr(gettext("unable to open %s"), srcfile);
        exit(1);
    }

    input = tempnam(NULL, "sedinp");
    if ((fpout = fopen(input, "w")) == NULL) {
        progerr(gettext("unable to open %s"), input);
        exit(2);
    }

    flag = (-1);
    while (fgets(line, LSIZE, fp)) {
        for (pt = line; isspace(*pt); /* void */)
            ++pt;
        if (*pt == '#')
            continue;
        if (*pt == COMMAND) {
            if (flag > 0)
                break; /* no more lines to read */
            pt = strtok(pt+1, " \t\n");
            if (!pt) {
                progerr(gettext("null token after '!'"));
                exit(1);
            }
            flag = (strcmp(pt, keyword) ? 0 : 1);
        } else if (flag == 1) { /* bug # 1083359 */
            (void) fputs(line, fpout);
        }
    }
    (void) fclose(fpout);
    if (flag > 0) {
        if (docmd(cmd, destfile, input)) {
            progerr(gettext("command failed <%s>"), cmd);
            exit(1);
        }
    }
    (void) unlink(input);
    return (0);
}

static int
docmd(char *cmd, char *file, char *input)
{
    char *tempout;
    char command[256];

    tempout = tempnam(NULL, "temp1");
    if (!tempout)
        return (-1);

    (void) sprintf(command, "%s -f %s <%s >%s", cmd, input, file, tempout);
    if (system(command))
        return (-1);

    (void) sprintf(command, "cp %s %s", tempout, file);
    if (system(command))
        return (-1);

    (void) unlink(tempout);
    free(tempout);
    return (0);
}

static void
usage(void)
{
    (void) fprintf(stderr, gettext("usage: %s cmd keyword src dest\n"),
        get_prog_name());
    exit(2);
}