summaryrefslogtreecommitdiff
path: root/usr/src/common/cmdparse/cmdparse.c
diff options
context:
space:
mode:
Diffstat (limited to 'usr/src/common/cmdparse/cmdparse.c')
-rw-r--r--usr/src/common/cmdparse/cmdparse.c681
1 files changed, 681 insertions, 0 deletions
diff --git a/usr/src/common/cmdparse/cmdparse.c b/usr/src/common/cmdparse/cmdparse.c
new file mode 100644
index 0000000..7a9e575
--- /dev/null
+++ b/usr/src/common/cmdparse/cmdparse.c
@@ -0,0 +1,681 @@
+/*
+ * 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.
+ */
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <sys/types.h>
+#include <unistd.h>
+#include <libintl.h>
+#include <errno.h>
+#include <string.h>
+#include <assert.h>
+#include <getopt.h>
+#include <cmdparse.h>
+
+
+/* Usage types */
+#define GENERAL_USAGE 1
+#define DETAIL_USAGE 2
+
+/* printable ascii character set len */
+#define MAXOPTIONS (uint_t)('~' - '!' + 1)
+
+/*
+ * MAXOPTIONSTRING is the max length of the options string used in getopt and
+ * will be the printable character set + ':' for each character,
+ * providing for options with arguments. e.g. "t:Cs:hglr:"
+ */
+#define MAXOPTIONSTRING MAXOPTIONS * 2
+
+/* standard command options table to support -?, -V */
+struct option standardCmdOptions[] = {
+ {"help", no_argument, NULL, '?'},
+ {"version", no_argument, NULL, 'V'},
+ {NULL, 0, NULL, 0}
+};
+
+/* standard subcommand options table to support -? */
+struct option standardSubCmdOptions[] = {
+ {"help", no_argument, NULL, '?'},
+ {NULL, 0, NULL, 0}
+};
+
+/* forward declarations */
+static int getSubcommandProps(char *, subCommandProps_t **);
+static char *getExecBasename(char *);
+static void usage(uint_t);
+static void subUsage(uint_t, subCommandProps_t *);
+static char *getLongOption(int);
+static char *getOptionArgDesc(int);
+
+/* global data */
+static struct option *_longOptions;
+static subCommandProps_t *_subCommandProps;
+static optionTbl_t *_clientOptionTbl;
+static char *commandName;
+
+
+/*
+ * input:
+ * subCommand - subcommand value
+ * output:
+ * subCommandProps - pointer to subCommandProps_t structure allocated by caller
+ *
+ * On successful return, subCommandProps contains the properties for the value
+ * in subCommand. On failure, the contents of subCommandProps is unspecified.
+ *
+ * Returns:
+ * zero on success
+ * non-zero on failure
+ *
+ */
+static int
+getSubcommandProps(char *subCommand, subCommandProps_t **subCommandProps)
+{
+ subCommandProps_t *sp;
+ int len;
+
+ for (sp = _subCommandProps; sp->name; sp++) {
+ len = strlen(subCommand);
+ if (len == strlen(sp->name) &&
+ strncasecmp(subCommand, sp->name, len) == 0) {
+ *subCommandProps = sp;
+ return (0);
+ }
+ }
+ return (1);
+}
+
+/*
+ * input:
+ * shortOption - short option character for which to return the
+ * associated long option string
+ *
+ * Returns:
+ * on success, long option name
+ * on failure, NULL
+ */
+static char *
+getLongOption(int shortOption)
+{
+ struct option *op;
+ for (op = _longOptions; op->name; op++) {
+ if (shortOption == op->val) {
+ return (op->name);
+ }
+ }
+ return (NULL);
+}
+
+/*
+ * input
+ * shortOption - short option character for which to return the
+ * option argument
+ * Returns:
+ * on success, argument string
+ * on failure, NULL
+ */
+static char *
+getOptionArgDesc(int shortOption)
+{
+ optionTbl_t *op;
+ for (op = _clientOptionTbl; op->name; op++) {
+ if (op->val == shortOption &&
+ op->has_arg == required_argument) {
+ return (op->argDesc);
+ }
+ }
+ return (NULL);
+}
+
+
+/*
+ * Print usage for a subcommand.
+ *
+ * input:
+ * usage type - GENERAL_USAGE, DETAIL_USAGE
+ * subcommand - pointer to subCommandProps_t structure
+ *
+ * Returns:
+ * none
+ *
+ */
+static void
+subUsage(uint_t usageType, subCommandProps_t *subcommand)
+{
+ int i;
+ char *optionArgDesc;
+ char *longOpt;
+
+ if (usageType == GENERAL_USAGE) {
+ (void) printf("%s:\t%s %s [", gettext("Usage"), commandName,
+ subcommand->name);
+ for (i = 0; standardSubCmdOptions[i].name; i++) {
+ (void) printf("-%c", standardSubCmdOptions[i].val);
+ if (standardSubCmdOptions[i+1].name)
+ (void) printf(",");
+ }
+ (void) fprintf(stdout, "]\n");
+ return;
+ }
+
+ /* print subcommand usage */
+ (void) printf("\n%s:\t%s %s ", gettext("Usage"), commandName,
+ subcommand->name);
+
+ /* print options if applicable */
+ if (subcommand->optionString != NULL) {
+ if (subcommand->required) {
+ (void) printf("%s", gettext("<"));
+ } else {
+ (void) printf("%s", gettext("["));
+ }
+ (void) printf("%s", gettext("OPTIONS"));
+ if (subcommand->required) {
+ (void) printf("%s ", gettext(">"));
+ } else {
+ (void) printf("%s ", gettext("]"));
+ }
+ }
+
+ /* print operand requirements */
+ if (!(subcommand->operand & OPERAND_NONE) &&
+ !(subcommand->operand & OPERAND_MANDATORY)) {
+ (void) printf(gettext("["));
+ }
+
+ if (subcommand->operand & OPERAND_MANDATORY) {
+ (void) printf(gettext("<"));
+ }
+
+ if (!(subcommand->operand & OPERAND_NONE)) {
+ assert(subcommand->operandDefinition);
+ (void) printf("%s", subcommand->operandDefinition);
+ }
+
+ if (subcommand->operand & OPERAND_MULTIPLE) {
+ (void) printf(gettext(" ..."));
+ }
+
+ if (subcommand->operand & OPERAND_MANDATORY) {
+ (void) printf(gettext(">"));
+ }
+
+ if (!(subcommand->operand & OPERAND_NONE) &&
+ !(subcommand->operand & OPERAND_MANDATORY)) {
+ (void) printf(gettext("]"));
+ }
+
+ /* print options for subcommand */
+ if (subcommand->optionString != NULL) {
+ (void) printf("\n\t%s:", gettext("OPTIONS"));
+ for (i = 0; i < strlen(subcommand->optionString); i++) {
+ assert((longOpt = getLongOption(
+ subcommand->optionString[i])) != NULL);
+ (void) printf("\n\t\t-%c, --%s ",
+ subcommand->optionString[i],
+ longOpt);
+ optionArgDesc =
+ getOptionArgDesc(subcommand->optionString[i]);
+ if (optionArgDesc != NULL) {
+ (void) printf("<%s>", optionArgDesc);
+ }
+ if (subcommand->exclusive &&
+ strchr(subcommand->exclusive,
+ subcommand->optionString[i])) {
+ (void) printf(" (%s)", gettext("exclusive"));
+ }
+ }
+ }
+ (void) fprintf(stdout, "\n");
+ if (subcommand->helpText) {
+ (void) printf("%s\n", subcommand->helpText);
+ }
+}
+
+/*
+ * input:
+ * type of usage statement to print
+ *
+ * Returns:
+ * return value of subUsage
+ */
+static void
+usage(uint_t usageType)
+{
+ int i;
+ subCommandProps_t *sp;
+
+ /* print general command usage */
+ (void) printf("%s:\t%s ", gettext("Usage"), commandName);
+
+ for (i = 0; standardCmdOptions[i].name; i++) {
+ (void) printf("-%c", standardCmdOptions[i].val);
+ if (standardCmdOptions[i+1].name)
+ (void) printf(",");
+ }
+
+ if (usageType == GENERAL_USAGE) {
+ for (i = 0; standardSubCmdOptions[i].name; i++) {
+ (void) printf(",--%s", standardSubCmdOptions[i].name);
+ if (standardSubCmdOptions[i+1].name)
+ (void) printf(",");
+ }
+ }
+
+ (void) fprintf(stdout, "\n");
+
+
+ /* print all subcommand usage */
+ for (sp = _subCommandProps; sp->name; sp++) {
+ subUsage(usageType, sp);
+ }
+}
+
+/*
+ * input:
+ * execFullName - exec name of program (argv[0])
+ *
+ * Returns:
+ * command name portion of execFullName
+ */
+static char *
+getExecBasename(char *execFullname)
+{
+ char *lastSlash, *execBasename;
+
+ /* guard against '/' at end of command invocation */
+ for (;;) {
+ lastSlash = strrchr(execFullname, '/');
+ if (lastSlash == NULL) {
+ execBasename = execFullname;
+ break;
+ } else {
+ execBasename = lastSlash + 1;
+ if (*execBasename == '\0') {
+ *lastSlash = '\0';
+ continue;
+ }
+ break;
+ }
+ }
+ return (execBasename);
+}
+
+/*
+ * cmdParse is a parser that checks syntax of the input command against
+ * various rules tables.
+ *
+ * It provides usage feedback based upon the passed rules tables by calling
+ * two usage functions, usage, subUsage
+ *
+ * When syntax is successfully validated, the associated function is called
+ * using the subcommands table functions.
+ *
+ * Syntax is as follows:
+ * command subcommand [<options>] [<operand>]
+ *
+ * There are two standard short and long options assumed:
+ * -?, --help Provides usage on a command or subcommand
+ * and stops further processing of the arguments
+ *
+ * -V, --version Provides version information on the command
+ * and stops further processing of the arguments
+ *
+ * These options are loaded by this function.
+ *
+ * input:
+ * argc, argv from main
+ * syntax rules tables (synTables_t structure)
+ * callArgs - void * passed by caller to be passed to subcommand function
+ *
+ * output:
+ * funcRet - pointer to int that holds subcommand function return value
+ *
+ * Returns:
+ *
+ * zero on successful syntax parse and function call
+ *
+ * 1 on unsuccessful syntax parse (no function has been called)
+ * This could be due to a version or help call or simply a
+ * general usage call.
+ *
+ * -1 check errno, call failed
+ *
+ * This module is not MT-safe.
+ *
+ */
+int
+cmdParse(int argc, char *argv[], synTables_t synTable, void *callArgs,
+ int *funcRet)
+{
+ int getoptargc;
+ char **getoptargv;
+ int opt;
+ int operInd;
+ int i, j;
+ int len;
+ int requiredOptionCnt = 0, requiredOptionEntered = 0;
+ char *availOptions;
+ char *versionString;
+ char optionStringAll[MAXOPTIONSTRING + 1];
+ subCommandProps_t *subcommand;
+ cmdOptions_t cmdOptions[MAXOPTIONS + 1];
+ optionTbl_t *optionTbl;
+ struct option *lp;
+ struct option intLongOpt[MAXOPTIONS + 1];
+
+ /*
+ * Check for NULLs on mandatory input arguments
+ *
+ * Note: longOptionTbl can be NULL in the case
+ * where there is no caller defined options
+ *
+ */
+ assert(synTable.versionString);
+ assert(synTable.subCommandPropsTbl);
+ assert(funcRet);
+
+ versionString = synTable.versionString;
+
+ /* set global command name */
+ commandName = getExecBasename(argv[0]);
+
+ /* Set unbuffered output */
+ setbuf(stdout, NULL);
+
+ /* load globals */
+ _subCommandProps = synTable.subCommandPropsTbl;
+ _clientOptionTbl = synTable.longOptionTbl;
+
+ /* There must be at least two arguments */
+ if (argc < 2) {
+ usage(GENERAL_USAGE);
+ return (1);
+ }
+
+ (void) memset(&intLongOpt[0], 0, sizeof (intLongOpt));
+
+ /*
+ * load standard subcommand options to internal long options table
+ * Two separate getopt_long(3C) tables are used.
+ */
+ for (i = 0; standardSubCmdOptions[i].name; i++) {
+ intLongOpt[i].name = standardSubCmdOptions[i].name;
+ intLongOpt[i].has_arg = standardSubCmdOptions[i].has_arg;
+ intLongOpt[i].flag = standardSubCmdOptions[i].flag;
+ intLongOpt[i].val = standardSubCmdOptions[i].val;
+ }
+
+ /*
+ * copy caller's long options into internal long options table
+ * We do this for two reasons:
+ * 1) We need to use the getopt_long option structure internally
+ * 2) We need to prepend the table with the standard option
+ * for all subcommands (currently -?)
+ */
+ for (optionTbl = synTable.longOptionTbl;
+ optionTbl && optionTbl->name; optionTbl++, i++) {
+ if (i > MAXOPTIONS - 1) {
+ /* option table too long */
+ assert(0);
+ }
+ intLongOpt[i].name = optionTbl->name;
+ intLongOpt[i].has_arg = optionTbl->has_arg;
+ intLongOpt[i].flag = NULL;
+ intLongOpt[i].val = optionTbl->val;
+ }
+
+ /* set option table global */
+ _longOptions = &intLongOpt[0];
+
+
+ /*
+ * Check for help/version request immediately following command
+ * '+' in option string ensures POSIX compliance in getopt_long()
+ * which means that processing will stop at first non-option
+ * argument.
+ */
+ while ((opt = getopt_long(argc, argv, "+?V", standardCmdOptions,
+ NULL)) != EOF) {
+ switch (opt) {
+ case '?':
+ /*
+ * getopt can return a '?' when no
+ * option letters match string. Check for
+ * the 'real' '?' in optopt.
+ */
+ if (optopt == '?') {
+ usage(DETAIL_USAGE);
+ exit(0);
+ } else {
+ usage(GENERAL_USAGE);
+ return (1);
+ }
+ break;
+ case 'V':
+ (void) fprintf(stdout, "%s: %s %s\n",
+ commandName, gettext("Version"),
+ versionString);
+ exit(0);
+ break;
+ default:
+ break;
+ }
+ }
+
+ /*
+ * subcommand is always in the second argument. If there is no
+ * recognized subcommand in the second argument, print error,
+ * general usage and then return.
+ */
+ if (getSubcommandProps(argv[1], &subcommand) != 0) {
+ (void) printf("%s: %s\n", commandName,
+ gettext("invalid subcommand"));
+ usage(GENERAL_USAGE);
+ return (1);
+ }
+
+ getoptargv = argv;
+ getoptargv++;
+ getoptargc = argc;
+ getoptargc -= 1;
+
+ (void) memset(optionStringAll, 0, sizeof (optionStringAll));
+ (void) memset(&cmdOptions[0], 0, sizeof (cmdOptions));
+
+ j = 0;
+ /*
+ * Build optionStringAll from long options table
+ */
+ for (lp = _longOptions; lp->name; lp++, j++) {
+ /* sanity check on string length */
+ if (j + 1 >= sizeof (optionStringAll)) {
+ /* option table too long */
+ assert(0);
+ }
+ optionStringAll[j] = lp->val;
+ if (lp->has_arg == required_argument) {
+ optionStringAll[++j] = ':';
+ }
+ }
+
+ i = 0;
+ /*
+ * Run getopt for all arguments against all possible options
+ * Store all options/option arguments in an array for retrieval
+ * later.
+ *
+ * Once all options are retrieved, a validity check against
+ * subcommand table is performed.
+ */
+ while ((opt = getopt_long(getoptargc, getoptargv, optionStringAll,
+ _longOptions, NULL)) != EOF) {
+ switch (opt) {
+ case '?':
+ subUsage(DETAIL_USAGE, subcommand);
+ /*
+ * getopt can return a '?' when no
+ * option letters match string. Check for
+ * the 'real' '?' in optopt.
+ */
+ if (optopt == '?') {
+ exit(0);
+ } else {
+ exit(1);
+ }
+ default:
+ cmdOptions[i].optval = opt;
+ if (optarg) {
+ len = strlen(optarg);
+ if (len > sizeof (cmdOptions[i].optarg)
+ - 1) {
+ (void) printf("%s: %s\n",
+ commandName,
+ gettext("option too long"));
+ errno = EINVAL;
+ return (-1);
+ }
+ (void) strncpy(cmdOptions[i].optarg,
+ optarg, len);
+ }
+ i++;
+ break;
+ }
+ }
+
+ /*
+ * increment past last option
+ */
+ operInd = optind + 1;
+
+ /*
+ * Check validity of given options, if any were given
+ */
+
+ /* get option string for this subcommand */
+ availOptions = subcommand->optionString;
+
+ /* Get count of required options */
+ if (subcommand->required) {
+ requiredOptionCnt = strlen(subcommand->required);
+ }
+
+ if (cmdOptions[0].optval != 0) { /* options were input */
+ if (availOptions == NULL) { /* no options permitted */
+ (void) printf("%s: %s\n", commandName,
+ gettext("no options permitted"));
+ subUsage(DETAIL_USAGE, subcommand);
+ return (1);
+ }
+ for (i = 0; cmdOptions[i].optval; i++) {
+ /* is the option in the available option string? */
+ if (!(strchr(availOptions, cmdOptions[i].optval))) {
+ (void) printf("%s: '-%c': %s\n", commandName,
+ cmdOptions[i].optval,
+ gettext("invalid option"));
+ subUsage(DETAIL_USAGE, subcommand);
+ return (1);
+ /* increment required options entered */
+ } else if (subcommand->required &&
+ (strchr(subcommand->required,
+ cmdOptions[i].optval))) {
+ requiredOptionEntered++;
+ /* Check for exclusive options */
+ } else if (cmdOptions[1].optval != 0 &&
+ subcommand->exclusive &&
+ strchr(subcommand->exclusive,
+ cmdOptions[i].optval)) {
+ (void) printf("%s: '-%c': %s\n",
+ commandName, cmdOptions[i].optval,
+ gettext("is an exclusive option"));
+ subUsage(DETAIL_USAGE, subcommand);
+ return (1);
+ }
+ }
+ } else { /* no options were input */
+ if (availOptions != NULL && subcommand->required) {
+ (void) printf("%s: %s\n", commandName,
+ gettext("at least one option required"));
+ subUsage(DETAIL_USAGE, subcommand);
+ return (1);
+ }
+ }
+
+ /* Were all required options entered? */
+ if (requiredOptionEntered != requiredOptionCnt) {
+ (void) printf("%s: %s: %s\n", commandName,
+ gettext("Following option(s) required"),
+ subcommand->required);
+ subUsage(DETAIL_USAGE, subcommand);
+ return (1);
+ }
+
+
+ /*
+ * If there are no operands,
+ * check to see if this is okay
+ */
+ if ((operInd == argc) &&
+ (subcommand->operand & OPERAND_MANDATORY)) {
+ (void) printf("%s: %s %s\n", commandName, subcommand->name,
+ gettext("requires an operand"));
+ subUsage(DETAIL_USAGE, subcommand);
+ return (1);
+ }
+
+ /*
+ * If there are more operands,
+ * check to see if this is okay
+ */
+ if ((argc > operInd) &&
+ (subcommand->operand & OPERAND_NONE)) {
+ (void) fprintf(stderr, "%s: %s %s\n", commandName,
+ subcommand->name, gettext("takes no operands"));
+ subUsage(DETAIL_USAGE, subcommand);
+ return (1);
+ }
+
+ /*
+ * If there is more than one more operand,
+ * check to see if this is okay
+ */
+ if ((argc > operInd) && ((argc - operInd) != 1) &&
+ (subcommand->operand & OPERAND_SINGLE)) {
+ (void) printf("%s: %s %s\n", commandName,
+ subcommand->name, gettext("accepts only a single operand"));
+ subUsage(DETAIL_USAGE, subcommand);
+ return (1);
+ }
+
+ /* Finished syntax checks */
+
+
+ /* Call appropriate function */
+ *funcRet = subcommand->handler(argc - operInd, &argv[operInd],
+ &cmdOptions[0], callArgs);
+
+ return (0);
+}
generated by cgit v1.2.3 (git 2.39.1) at 2024-11-23 08:09:17 +0000