mcs/common/main.c

	main.c revision 7010c12ad3ac2cada55cf126121a8c46957d3632
18c2aff776a775d34a4c9893a4c72e0434d68e36artem/*
18c2aff776a775d34a4c9893a4c72e0434d68e36artem * CDDL HEADER START
18c2aff776a775d34a4c9893a4c72e0434d68e36artem *
18c2aff776a775d34a4c9893a4c72e0434d68e36artem * The contents of this file are subject to the terms of the
42a7bded1b1244af097afdc88654381a3d3879f9jacobs * Common Development and Distribution License (the "License").
18c2aff776a775d34a4c9893a4c72e0434d68e36artem * You may not use this file except in compliance with the License.
18c2aff776a775d34a4c9893a4c72e0434d68e36artem *
18c2aff776a775d34a4c9893a4c72e0434d68e36artem * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
18c2aff776a775d34a4c9893a4c72e0434d68e36artem * or http://www.opensolaris.org/os/licensing.
18c2aff776a775d34a4c9893a4c72e0434d68e36artem * See the License for the specific language governing permissions
18c2aff776a775d34a4c9893a4c72e0434d68e36artem * and limitations under the License.
92f381329ebf1c2209df9608670666b32b291e05artem *
18c2aff776a775d34a4c9893a4c72e0434d68e36artem * When distributing Covered Code, include this CDDL HEADER in each
18c2aff776a775d34a4c9893a4c72e0434d68e36artem * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
18c2aff776a775d34a4c9893a4c72e0434d68e36artem * If applicable, add the following below this CDDL HEADER, with the
18c2aff776a775d34a4c9893a4c72e0434d68e36artem * fields enclosed by brackets "[]" replaced with your own identifying
18c2aff776a775d34a4c9893a4c72e0434d68e36artem * information: Portions Copyright [yyyy] [name of copyright owner]
18c2aff776a775d34a4c9893a4c72e0434d68e36artem *
18c2aff776a775d34a4c9893a4c72e0434d68e36artem * CDDL HEADER END
18c2aff776a775d34a4c9893a4c72e0434d68e36artem */
18c2aff776a775d34a4c9893a4c72e0434d68e36artem
18c2aff776a775d34a4c9893a4c72e0434d68e36artem/*
18c2aff776a775d34a4c9893a4c72e0434d68e36artem *  Copyright (c) 1988 AT&T
18c2aff776a775d34a4c9893a4c72e0434d68e36artem *    All Rights Reserved
18c2aff776a775d34a4c9893a4c72e0434d68e36artem *
18c2aff776a775d34a4c9893a4c72e0434d68e36artem *
18c2aff776a775d34a4c9893a4c72e0434d68e36artem * Copyright 2006 Sun Microsystems, Inc.  All rights reserved.
18c2aff776a775d34a4c9893a4c72e0434d68e36artem * Use is subject to license terms.
18c2aff776a775d34a4c9893a4c72e0434d68e36artem */
18c2aff776a775d34a4c9893a4c72e0434d68e36artem
18c2aff776a775d34a4c9893a4c72e0434d68e36artem#pragma ident   "%Z%%M% %I% %E% SMI"
18c2aff776a775d34a4c9893a4c72e0434d68e36artem
18c2aff776a775d34a4c9893a4c72e0434d68e36artem#include "stdlib.h"
18c2aff776a775d34a4c9893a4c72e0434d68e36artem#include "conv.h"
18c2aff776a775d34a4c9893a4c72e0434d68e36artem#include "mcs.h"
18c2aff776a775d34a4c9893a4c72e0434d68e36artem#include "extern.h"
18c2aff776a775d34a4c9893a4c72e0434d68e36artem#define OPTUNIT 100
18c2aff776a775d34a4c9893a4c72e0434d68e36artem
18c2aff776a775d34a4c9893a4c72e0434d68e36artemstatic size_t optcnt = 0;
18c2aff776a775d34a4c9893a4c72e0434d68e36artemstatic size_t optbufsz = OPTUNIT;
18c2aff776a775d34a4c9893a4c72e0434d68e36artem
18c2aff776a775d34a4c9893a4c72e0434d68e36artem/*
18c2aff776a775d34a4c9893a4c72e0434d68e36artem * Function prototypes.
18c2aff776a775d34a4c9893a4c72e0434d68e36artem */
18c2aff776a775d34a4c9893a4c72e0434d68e36artemstatic void usage(int);
18c2aff776a775d34a4c9893a4c72e0434d68e36artemstatic void sigexit(int);
18c2aff776a775d34a4c9893a4c72e0434d68e36artemstatic int setup_sectname(char *, int);
18c2aff776a775d34a4c9893a4c72e0434d68e36artemstatic void check_swap();
18c2aff776a775d34a4c9893a4c72e0434d68e36artemstatic void queue(int, char *);
18c2aff776a775d34a4c9893a4c72e0434d68e36artem
18c2aff776a775d34a4c9893a4c72e0434d68e36artemint
18c2aff776a775d34a4c9893a4c72e0434d68e36artemmain(int argc, char ** argv, char ** envp)
18c2aff776a775d34a4c9893a4c72e0434d68e36artem{
18c2aff776a775d34a4c9893a4c72e0434d68e36artem    const char  *opt;
18c2aff776a775d34a4c9893a4c72e0434d68e36artem    char        *str;
18c2aff776a775d34a4c9893a4c72e0434d68e36artem    int     error_count = 0, num_sect = 0, errflag = 0, Dflag = 0;
18c2aff776a775d34a4c9893a4c72e0434d68e36artem    int     c, i, my_prog;
18c2aff776a775d34a4c9893a4c72e0434d68e36artem    Cmd_Info    *cmd_info;
18c2aff776a775d34a4c9893a4c72e0434d68e36artem
18c2aff776a775d34a4c9893a4c72e0434d68e36artem    /*
18c2aff776a775d34a4c9893a4c72e0434d68e36artem     * Check for a binary that better fits this architecture.
18c2aff776a775d34a4c9893a4c72e0434d68e36artem     */
18c2aff776a775d34a4c9893a4c72e0434d68e36artem    (void) conv_check_native(argv, envp);
18c2aff776a775d34a4c9893a4c72e0434d68e36artem
18c2aff776a775d34a4c9893a4c72e0434d68e36artem    /*
18c2aff776a775d34a4c9893a4c72e0434d68e36artem     * mcs(1) and strip() are hard linked together, determine which command
18c2aff776a775d34a4c9893a4c72e0434d68e36artem     * was invoked.
18c2aff776a775d34a4c9893a4c72e0434d68e36artem     */
18c2aff776a775d34a4c9893a4c72e0434d68e36artem    prog = argv[0];
18c2aff776a775d34a4c9893a4c72e0434d68e36artem    if ((str = strrchr(prog, '/')) != NULL)
42a7bded1b1244af097afdc88654381a3d3879f9jacobs        str++;
42a7bded1b1244af097afdc88654381a3d3879f9jacobs    else
18c2aff776a775d34a4c9893a4c72e0434d68e36artem        str = prog;
18c2aff776a775d34a4c9893a4c72e0434d68e36artem
    if (strcmp(str, "mcs") == 0) {
        my_prog = MCS;
        opt = "Da:cdn:pVz?";
    } else if (strcmp(str, "strip") == 0) {
        my_prog = STRIP;
        opt = "DlxV?";
    } else
        exit(1);

    (void) setlocale(LC_ALL, "");
#if !defined(TEXT_DOMAIN)
#define TEXT_DOMAIN "SYS_TEST"
#endif
    (void) textdomain(TEXT_DOMAIN);

    for (i = 0; signum[i]; i++)
        if (signal(signum[i], SIG_IGN) != SIG_IGN)
            (void) signal(signum[i], sigexit);

    if ((Action =
        malloc(optbufsz * sizeof (struct action))) == NULL) {
        error_message(MALLOC_ERROR, PLAIN_ERROR, (char *)0, prog);
        exit(FAILURE);
    }

    /*
     * Allocate command info structure
     */
    cmd_info = (Cmd_Info *) calloc(1, sizeof (Cmd_Info));
    if (cmd_info == NULL) {
        error_message(MALLOC_ERROR, PLAIN_ERROR, (char *)0, prog);
        exit(FAILURE);
    }
    if (my_prog == STRIP)
        cmd_info->flags |= I_AM_STRIP;

    while ((c = getopt(argc, argv, (char *)opt)) != EOF) {
        switch (c) {
        case 'D':
            optcnt++;
            Dflag++;
            break;
        case 'a':
            optcnt++;
            queue(ACT_APPEND, optarg);
            cmd_info->flags |= (MIGHT_CHG | aFLAG);
            cmd_info->str_size += strlen(optarg) + 1;
            break;
        case 'c':
            optcnt++;
            queue(ACT_COMPRESS, NULL);
            cmd_info->flags |= (MIGHT_CHG | cFLAG);
            break;
        case 'd':
            optcnt++;
            if (CHK_OPT(cmd_info, dFLAG) == 0)
                queue(ACT_DELETE, NULL);
            cmd_info->flags |= (MIGHT_CHG | dFLAG);
            break;
        case 'z':
            optcnt++;
            queue(ACT_ZAP, NULL);
            cmd_info->flags |= (MIGHT_CHG | zFLAG);
            break;
        case 'n':
            (void) setup_sectname(optarg, my_prog);
            num_sect++;
            break;
        case 'l':
            optcnt++;
            cmd_info->flags |= lFLAG;
            break;
        case 'p':
            optcnt++;
            queue(ACT_PRINT, NULL);
            cmd_info->flags |= pFLAG;
            break;
        case 'x':
            optcnt++;
            cmd_info->flags |= xFLAG;
            break;
        case 'V':
            cmd_info->flags |= VFLAG;
            (void) fprintf(stderr, "%s: %s %s\n", prog,
                (const char *)SGU_PKG, (const char *)SGU_REL);
            break;
        case '?':
            errflag++;
            break;
        default:
            break;
        }
    }

    if (errflag) {
        usage(my_prog);
        exit(FAILURE);
    }

    if (Dflag)
        check_swap();

    /*
     * strip command may not take any options.
     */
    if (my_prog != STRIP) {
        if (argc == optind &&
            (CHK_OPT(cmd_info, MIGHT_CHG) || CHK_OPT(cmd_info, pFLAG) ||
            argc == 1))
            usage(my_prog);
        else if (!CHK_OPT(cmd_info, MIGHT_CHG) &&
            !CHK_OPT(cmd_info, pFLAG) && !CHK_OPT(cmd_info, VFLAG))
            usage(my_prog);
    }

    /*
     * This version only allows multiple section names
     * only for -d option.
     */
    if ((num_sect >= 2) && (CHK_OPT(cmd_info, pFLAG) ||
        CHK_OPT(cmd_info, aFLAG) ||
        CHK_OPT(cmd_info, cFLAG))) {
        error_message(USAGE_ERROR, PLAIN_ERROR, (char *)0,  prog);
        exit(1);
    }

    /*
     * If no -n was specified,
     * set the default, ".comment".
     * This is for mcs only.
     */
    if (num_sect == 0 && my_prog == MCS) {
        (void) setup_sectname(".comment", MCS);
    }

    /*
     * If I am strip command, then add needed
     * section names.
     */
    if (my_prog == STRIP) {
        (void) setup_sectname(".line", MCS);
        if (CHK_OPT(cmd_info, lFLAG) == 0) {
            (void) setup_sectname(".debug", STRIP);
            (void) setup_sectname(".stab", STRIP);
        }
        if (CHK_OPT(cmd_info, dFLAG) == 0) {
            queue(ACT_DELETE, NULL);
            cmd_info->flags |= MIGHT_CHG;
            cmd_info->flags |= dFLAG;
        }
    }

    (void) elf_version(EV_NONE);
    if (elf_version(EV_CURRENT) == EV_NONE) {
        error_message(ELFVER_ERROR, LIBelf_ERROR, elf_errmsg(-1), prog);
        exit(FAILURE);
    }

    if (CHK_OPT(cmd_info, pFLAG) || CHK_OPT(cmd_info, MIGHT_CHG)) {
        for (; optind < argc; optind++) {
            error_count = error_count +
                (each_file(argv[optind], cmd_info));
        }
    }

    if (Dflag)
        check_swap();
    mcs_exit(error_count);
    /*NOTREACHED*/
    return (0);
}

/*
 * Supplementary functions
 */
static void
queue(int activity, char *string)
{
    if (optcnt > optbufsz) {
        optbufsz = optbufsz * 2;
        if ((Action = realloc((struct action *)Action,
            optbufsz * sizeof (struct action))) == NULL) {
            error_message(MALLOC_ERROR, PLAIN_ERROR, (char *)0, prog);
            mcs_exit(FAILURE);
        }
    }
    Action[actmax].a_action = activity;
    Action[actmax].a_cnt = 0;
    Action[actmax].a_string = string;
    actmax++;
}

/*ARGSUSED0*/
static void
sigexit(int i)
{
    (void) unlink(artmpfile);
    (void) unlink(elftmpfile);
    exit(100);
}

static void
usage(int me)
{
    if (me == MCS)
        (void) fprintf(stderr, gettext(
        "usage: %s [-cdpVz] [-a string] [-n name] file ...\n"), prog);
    else
        (void) fprintf(stderr, gettext(
        "usage: %s [-lVx] file ...\n"), prog);
    mcs_exit(FAILURE);
}

void
mcs_exit(int val)
{
    (void) unlink(artmpfile);
    (void) unlink(elftmpfile);
    exit(val);
}

/*
 * Insert the section name 'name' into the
 * section list.
 */
static int
setup_sectname(char *name, int whoami)
{
    S_Name *new;

    /*
     * Check if the name is already specified or not.
     */
    if ((whoami == MCS) && (sectcmp(name) == 0))
        return (0);

    /*
     * Allocate one
     */
    if ((new = malloc(sizeof (S_Name))) == NULL) {
        error_message(MALLOC_ERROR, PLAIN_ERROR, (char *)0, prog);
        exit(FAILURE);
    }
    new->name = strdup(name);
    if (new->name == NULL) {
        error_message(USAGE_ERROR, PLAIN_ERROR, (char *)0, prog);
        exit(FAILURE);
    }
    if (whoami == STRIP)
        new->flags = SNAME_FLG_STRNCMP;
    new->next = NULL;

    /*
     * Put this one in the list
     */
    new->next = sect_head;
    sect_head = new;

    return (0);
}

/*
 * Check if the 'name' exists in the section list.
 *
 * If found
 *  return 0;
 * else
 *  return 1
 */
int
sectcmp(char *name)
{
    /*
     * Check if the name is already specified or not.
     */
    if (sect_head != NULL) {
        S_Name *p1 = sect_head;
        while (p1 != NULL) {
            if (p1->flags & SNAME_FLG_STRNCMP) {
                if (strncmp(p1->name,
                    name, strlen(p1->name)) == 0)
                    return (0);
            } else if (strcmp(p1->name, name) == 0) {
                return (0); /* silently ignore */
            }
            p1 = p1->next;
        }
    }
    return (1);
}

static void
check_swap()
{
    (void) system("/usr/sbin/swap -s");
}