cli.c

   1 /*
   2  * Copyright (C) 2008 Felix Fietkau <nbd@openwrt.org>
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License version 2
   6  * as published by the Free Software Foundation
   7  *
   8  * This program is distributed in the hope that it will be useful,
   9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11  * GNU General Public License for more details.
  12  */
  13 #include <strings.h>
  14 #include <stdlib.h>
  15 #include <unistd.h>
  16 #include "uci.h"
  17
  18 static const char *appname = "uci";
  19
  20 static struct uci_context *ctx;
  21 enum {
  22         CMD_GET,
  23         CMD_SET,
  24         CMD_DEL
  25 };
  26
  27 static void uci_usage(int argc, char **argv)
  28 {
  29         fprintf(stderr,
  30                 "Usage: %s [<options>] <command> [<arguments>]\n\n"
  31                 "Commands:\n"
  32                 "\texport   [<config>]\n"
  33                 "\tshow     [<config>[.<section>[.<option>]]]\n"
  34                 "\tget      <config>.<section>[.<option>]\n"
  35                 "\tset      <config>.<section>[.<option>]=<value>\n"
  36                 "\n"
  37                 "Options:\n"
  38                 "\t-s       force strict mode (stop on parser errors)\n"
  39                 "\t-S       disable strict mode\n"
  40                 "\n",
  41                 argv[0]
  42         );
  43         exit(255);
  44 }
  45
  46 static void uci_show_section(struct uci_section *p)
  47 {
  48         struct uci_element *e;
  49         const char *cname, *sname;
  50
  51         cname = p->package->e.name;
  52         sname = p->e.name;
  53         printf("%s.%s=%s\n", cname, sname, p->type);
  54         uci_foreach_element(&p->options, e) {
  55                 printf("%s.%s.%s=%s\n", cname, sname, e->name, uci_to_option(e)->value);
  56         }
  57 }
  58
  59 static void uci_show_package(struct uci_package *p, char *section)
  60 {
  61         struct uci_element *e;
  62
  63         uci_foreach_element( &p->sections, e) {
  64                 if (!section || !strcmp(e->name, section))
  65                         uci_show_section(uci_to_section(e));
  66         }
  67 }
  68
  69 static int uci_show(int argc, char **argv)
  70 {
  71         char *section = (argc > 2 ? argv[2] : NULL);
  72         struct uci_package *package;
  73         char **configs = NULL;
  74         char **p;
  75
  76         if ((uci_list_configs(ctx, &configs) != UCI_OK) || !configs) {
  77                 uci_perror(ctx, appname);
  78                 return 1;
  79         }
  80
  81         if (argc >= 2) {
  82                 if (uci_load(ctx, argv[1], &package) != UCI_OK) {
  83                         uci_perror(ctx, appname);
  84                         return 1;
  85                 }
  86                 uci_show_package(package, section);
  87                 uci_unload(ctx, package);
  88                 return 0;
  89         }
  90
  91         for (p = configs; *p; p++) {
  92                 if ((argc < 2) || !strcmp(argv[1], *p)) {
  93                         if (uci_load(ctx, *p, &package) != UCI_OK) {
  94                                 uci_perror(ctx, appname);
  95                                 return 1;
  96                         }
  97                         uci_show_package(package, section);
  98                         uci_unload(ctx, package);
  99                 }
 100         }
 101
 102         return 0;
 103 }
 104
 105 static int uci_do_export(int argc, char **argv)
 106 {
 107         char **configs = NULL;
 108         char **p;
 109
 110         if ((uci_list_configs(ctx, &configs) != UCI_OK) || !configs) {
 111                 uci_perror(ctx, appname);
 112                 return 1;
 113         }
 114
 115         for (p = configs; *p; p++) {
 116                 if ((argc < 2) || !strcmp(argv[1], *p)) {
 117                         struct uci_package *package = NULL;
 118                         int ret;
 119
 120                         ret = uci_load(ctx, *p, &package);
 121                         if (ret)
 122                                 continue;
 123                         uci_export(ctx, stdout, package, true);
 124                         uci_unload(ctx, package);
 125                 }
 126         }
 127         return 0;
 128 }
 129
 130 static int uci_do_cmd(int cmd, int argc, char **argv)
 131 {
 132         char *package = NULL;
 133         char *section = NULL;
 134         char *option = NULL;
 135         char *value = NULL;
 136         struct uci_package *p = NULL;
 137         struct uci_element *e = NULL;
 138
 139         if (argc != 2)
 140                 return 255;
 141
 142         if (uci_parse_tuple(ctx, argv[1], &package, &section, &option, (cmd == CMD_SET ? &value : NULL)) != UCI_OK)
 143                 return 1;
 144
 145         if (uci_load(ctx, package, &p) != UCI_OK) {
 146                 uci_perror(ctx, appname);
 147                 return 1;
 148         }
 149
 150         if (uci_lookup(ctx, &e, p, section, option) != UCI_OK)
 151                 return 1;
 152
 153         switch(cmd) {
 154         case CMD_GET:
 155                 switch(e->type) {
 156                 case UCI_TYPE_SECTION:
 157                         value = uci_to_section(e)->type;
 158                         break;
 159                 case UCI_TYPE_OPTION:
 160                         value = uci_to_option(e)->value;
 161                         break;
 162                 default:
 163                         /* should not happen */
 164                         return 1;
 165                 }
 166                 /* throw the value to stdout */
 167                 printf("%s\n", value);
 168                 break;
 169         case CMD_SET:
 170                 if (uci_set(ctx, p, section, option, value) != UCI_OK) {
 171                         uci_perror(ctx, appname);
 172                         return 1;
 173                 }
 174                 break;
 175         case CMD_DEL:
 176                 if (uci_del(ctx, p, section, option) != UCI_OK) {
 177                         uci_perror(ctx, appname);
 178                         return 1;
 179                 }
 180                 break;
 181         }
 182
 183         /* no save necessary for get */
 184         if (cmd == CMD_GET)
 185                 return 0;
 186
 187         /* save changes, but don't commit them yet */
 188         if (uci_save(ctx, p) != UCI_OK) {
 189                 uci_perror(ctx, appname);
 190                 return 1;
 191         }
 192
 193         return 0;
 194 }
 195
 196 static int uci_cmd(int argc, char **argv)
 197 {
 198         int cmd;
 199
 200         if (!strcasecmp(argv[0], "show"))
 201                 return uci_show(argc, argv);
 202         if (!strcasecmp(argv[0], "export"))
 203                 return uci_do_export(argc, argv);
 204
 205         if (!strcasecmp(argv[0], "get"))
 206                 cmd = CMD_GET;
 207         else if (!strcasecmp(argv[0], "set"))
 208                 cmd = CMD_SET;
 209         else if (!strcasecmp(argv[0], "del"))
 210                 cmd = CMD_DEL;
 211         else
 212                 return 255;
 213         return uci_do_cmd(cmd, argc, argv);
 214 }
 215
 216 int main(int argc, char **argv)
 217 {
 218         int ret;
 219         int c;
 220
 221         ctx = uci_alloc_context();
 222         if (!ctx) {
 223                 fprintf(stderr, "Out of memory\n");
 224                 return 1;
 225         }
 226
 227         while((c = getopt(argc, argv, "sS")) != -1) {
 228                 switch(c) {
 229                         case 's':
 230                                 ctx->flags |= UCI_FLAG_STRICT;
 231                                 break;
 232                         case 'S':
 233                                 ctx->flags &= ~UCI_FLAG_STRICT;
 234                                 ctx->flags |= UCI_FLAG_PERROR;
 235                                 break;
 236                         default:
 237                                 uci_usage(argc, argv);
 238                                 break;
 239                 }
 240         }
 241         if (optind > 1)
 242                 argv[optind - 1] = argv[0];
 243         argv += optind - 1;
 244         argc -= optind - 1;
 245
 246         if (argc < 2)
 247                 uci_usage(argc, argv);
 248         ret = uci_cmd(argc - 1, argv + 1);
 249         if (ret == 255)
 250                 uci_usage(argc, argv);
 251
 252         uci_free_context(ctx);
 253
 254         return ret;
 255 }