tipc/cmdl.c

   1 /*
   2  * cmdl.c       Framework for handling command line options.
   3  *
   4  *              This program is free software; you can redistribute it and/or
   5  *              modify it under the terms of the GNU General Public License
   6  *              as published by the Free Software Foundation; either version
   7  *              2 of the License, or (at your option) any later version.
   8  *
   9  * Authors:     Richard Alpe <richard.alpe@ericsson.com>
  10  */
  11
  12 #include <stdio.h>
  13 #include <string.h>
  14 #include <errno.h>
  15
  16 #include <libmnl/libmnl.h>
  17
  18 #include "cmdl.h"
  19
  20 static const struct cmd *find_cmd(const struct cmd *cmds, char *str)
  21 {
  22         const struct cmd *c;
  23         const struct cmd *match = NULL;
  24
  25         for (c = cmds; c->cmd; c++) {
  26                 if (strstr(c->cmd, str) != c->cmd)
  27                         continue;
  28                 if (match)
  29                         return NULL;
  30                 match = c;
  31         }
  32
  33         return match;
  34 }
  35
  36 static struct opt *find_opt(struct opt *opts, char *str)
  37 {
  38         struct opt *o;
  39         struct opt *match = NULL;
  40
  41         for (o = opts; o->key; o++) {
  42                 if (strstr(o->key, str) != o->key)
  43                         continue;
  44                 if (match)
  45                         return NULL;
  46
  47                 match = o;
  48         }
  49
  50         return match;
  51 }
  52
  53 struct opt *get_opt(struct opt *opts, char *key)
  54 {
  55         struct opt *o;
  56
  57         for (o = opts; o->key; o++) {
  58                 if (strcmp(o->key, key) == 0 && o->val)
  59                         return o;
  60         }
  61
  62         return NULL;
  63 }
  64
  65 bool has_opt(struct opt *opts, char *key)
  66 {
  67         return get_opt(opts, key) ? true : false;
  68 }
  69
  70 char *shift_cmdl(struct cmdl *cmdl)
  71 {
  72         int next;
  73
  74         if (cmdl->optind < cmdl->argc)
  75                 next = (cmdl->optind)++;
  76         else
  77                 next = cmdl->argc;
  78
  79         return cmdl->argv[next];
  80 }
  81
  82 /* Returns the number of options parsed or a negative error code upon failure */
  83 int parse_opts(struct opt *opts, struct cmdl *cmdl)
  84 {
  85         int i;
  86         int cnt = 0;
  87
  88         for (i = cmdl->optind; i < cmdl->argc; i++) {
  89                 struct opt *o;
  90
  91                 o = find_opt(opts, cmdl->argv[i]);
  92                 if (!o) {
  93                         fprintf(stderr, "error, invalid option \"%s\"\n",
  94                                         cmdl->argv[i]);
  95                         return -EINVAL;
  96                 }
  97                 if (o->flag & OPT_KEYVAL) {
  98                         cmdl->optind++;
  99                         i++;
 100                 }
 101                 cnt++;
 102                 o->val = cmdl->argv[i];
 103                 cmdl->optind++;
 104         }
 105
 106         return cnt;
 107 }
 108
 109 int run_cmd(struct nlmsghdr *nlh, const struct cmd *caller,
 110             const struct cmd *cmds, struct cmdl *cmdl, void *data)
 111 {
 112         char *name;
 113         const struct cmd *cmd;
 114
 115         if ((cmdl->optind) >= cmdl->argc) {
 116                 if (caller->help)
 117                         (caller->help)(cmdl);
 118                 return -EINVAL;
 119         }
 120         name = cmdl->argv[cmdl->optind];
 121         (cmdl->optind)++;
 122
 123         cmd = find_cmd(cmds, name);
 124         if (!cmd) {
 125                 /* Show help about last command if we don't find this one */
 126                 if (help_flag && caller->help) {
 127                         (caller->help)(cmdl);
 128                 } else {
 129                         fprintf(stderr, "error, invalid command \"%s\"\n", name);
 130                         fprintf(stderr, "use --help for command help\n");
 131                 }
 132                 return -EINVAL;
 133         }
 134
 135         return (cmd->func)(nlh, cmd, cmdl, data);
 136 }