cmd/valtools/ckint.c

/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License, Version 1.0 only
 * (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 2004 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

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


#pragma ident   "%Z%%M% %I% %E% SMI"

#include <stdio.h>
#include <string.h>
#include <signal.h>
#include <stdlib.h>
#include <locale.h>
#include <libintl.h>
#include "usage.h"
#include "libadm.h"

#define BADPID (-2)

static char *prog;
static char *deflt, *prompt, *error, *help;
static int  kpid = BADPID;
static int  signo;
static short    base;

static const char   vusage[] = "b";
static const char   husage[] = "bWh";
static const char   eusage[] = "bWe";

static void
usage(void)
{
    switch (*prog) {
    default:
        (void) fprintf(stderr,
            gettext("usage: %s [options] [-b base]\n"), prog);
        (void) fprintf(stderr, gettext(OPTMESG));
        (void) fprintf(stderr, gettext(STDOPTS));
        break;

    case 'v':
        (void) fprintf(stderr,
            gettext("usage: %s [-b base] input\n"), prog);
        break;

    case 'h':
        (void) fprintf(stderr,
            gettext("usage: %s [options] [-b base]\n"), prog);
        (void) fprintf(stderr, gettext(OPTMESG));
        (void) fprintf(stderr,
            gettext("\t-W width\n\t-h help\n"));
        break;

    case 'e':
        (void) fprintf(stderr,
            gettext("usage: %s [options] [-b base]\n"), prog);
        (void) fprintf(stderr, gettext(OPTMESG));
        (void) fprintf(stderr,
            gettext("\t-W width\n\t-e error\n"));
        break;
    }
    exit(1);
}

/*
 * Given argv[0], return a pointer to the basename of the program.
 */
static char *
prog_name(char *arg0)
{
    char *str;

    /* first strip trailing '/' characters (exec() allows these!) */
    str = arg0 + strlen(arg0);
    while (str > arg0 && *--str == '/')
        *str = '\0';
    if ((str = strrchr(arg0, '/')) != NULL)
        return (str + 1);
    return (arg0);
}

int
main(int argc, char **argv)
{
    int c, n;
    long    intval;

    (void) setlocale(LC_ALL, "");

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

    prog = prog_name(argv[0]);

    while ((c = getopt(argc, argv, "b:d:p:e:h:k:s:QW:?")) != EOF) {
        /* check for invalid option */
        if ((*prog == 'v') && !strchr(vusage, c))
            usage(); /* no valid options */
        if ((*prog == 'e') && !strchr(eusage, c))
            usage();
        if ((*prog == 'h') && !strchr(husage, c))
            usage();

        switch (c) {
        case 'Q':
            ckquit = 0;
            break;

        case 'W':
            ckwidth = atoi(optarg);
            if (ckwidth < 0) {
                (void) fprintf(stderr,
        gettext("%s: ERROR: negative display width specified\n"),
                    prog);
                exit(1);
            }
            break;

        case 'b':
            base = atoi(optarg);
            if ((base < 2) || (base > 36)) {
                (void) fprintf(stderr,
        gettext("%s: ERROR: base must be between 2 and 36\n"),
                    prog);
                exit(1);
            }
            break;

        case 'd':
            deflt = optarg;
            break;

        case 'p':
            prompt = optarg;
            break;

        case 'e':
            error = optarg;
            break;

        case 'h':
            help = optarg;
            break;

        case 'k':
            kpid = atoi(optarg);
            break;

        case 's':
            signo = atoi(optarg);
            break;

        default:
            usage();
        }
    }

    if (signo) {
        if (kpid == BADPID)
            usage();
    } else
        signo = SIGTERM;

    if (*prog == 'v') {
        if (argc != (optind + 1))
            usage();
        exit(ckint_val(argv[optind], base));
    }

    if (optind != argc)
        usage();

    if (*prog == 'e') {
        ckindent = 0;
        ckint_err(base, error);
        exit(0);
    } else if (*prog == 'h') {
        ckindent = 0;
        ckint_hlp(base, help);
        exit(0);
    }

    n = ckint(&intval, base, deflt, error, help, prompt);
    if (n == 3) {
        if (kpid > -2)
            (void) kill(kpid, signo);
        (void) puts("q");
    } else if (n == 0)
        (void) printf("%ld", intval);
    return (n);
}