ucimap-example.c

   1 /*
   2  * ucimap-example - sample code for the ucimap library
   3  * Copyright (C) 2008-2009 Felix Fietkau <nbd@openwrt.org>
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License version 2
   7  * as published by the Free Software Foundation
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  */
  14 #include <strings.h>
  15 #include <string.h>
  16 #include <stdlib.h>
  17 #include <unistd.h>
  18 #include <ucimap.h>
  19 #include "list.h"
  20
  21 struct list_head ifs;
  22
  23 struct uci_network {
  24         struct ucimap_section_data map;
  25         struct list_head list;
  26         struct list_head alias;
  27
  28         const char *name;
  29         const char *proto;
  30         const char *ifname;
  31         unsigned char *ipaddr;
  32         int test;
  33         bool enabled;
  34         struct ucimap_list *aliases;
  35 };
  36
  37 struct uci_alias {
  38         struct ucimap_section_data map;
  39         struct list_head list;
  40
  41         const char *name;
  42         struct uci_network *interface;
  43 };
  44
  45 static int
  46 network_parse_ip(void *section, struct uci_optmap *om, union ucimap_data *data, const char *str)
  47 {
  48         unsigned char *target;
  49         int tmp[4];
  50         int i;
  51
  52         if (sscanf(str, "%d.%d.%d.%d", &tmp[0], &tmp[1], &tmp[2], &tmp[3]) != 4)
  53                 return -1;
  54
  55         target = malloc(4);
  56         if (!target)
  57                 return -1;
  58
  59         data->ptr = target;
  60         for (i = 0; i < 4; i++)
  61                 target[i] = (char) tmp[i];
  62
  63         return 0;
  64 }
  65
  66 static int
  67 network_format_ip(void *section, struct uci_optmap *om, union ucimap_data *data, char **str)
  68 {
  69         static char buf[16];
  70         unsigned char *ip = (unsigned char *) data->ptr;
  71
  72         if (ip) {
  73                 sprintf(buf, "%d.%d.%d.%d", ip[0], ip[1], ip[2], ip[3]);
  74                 *str = buf;
  75         } else {
  76                 *str = NULL;
  77         }
  78
  79         return 0;
  80 }
  81
  82 static void
  83 network_free_ip(void *section, struct uci_optmap *om, void *ptr)
  84 {
  85         free(ptr);
  86 }
  87
  88 static int
  89 network_init_interface(struct uci_map *map, void *section, struct uci_section *s)
  90 {
  91         struct uci_network *net = section;
  92
  93         INIT_LIST_HEAD(&net->list);
  94         INIT_LIST_HEAD(&net->alias);
  95         net->name = s->e.name;
  96         net->test = -1;
  97         return 0;
  98 }
  99
 100 static int
 101 network_init_alias(struct uci_map *map, void *section, struct uci_section *s)
 102 {
 103         struct uci_alias *alias = section;
 104
 105         INIT_LIST_HEAD(&alias->list);
 106         alias->name = s->e.name;
 107         return 0;
 108 }
 109
 110 static int
 111 network_add_interface(struct uci_map *map, void *section)
 112 {
 113         struct uci_network *net = section;
 114
 115         list_add_tail(&net->list, &ifs);
 116
 117         return 0;
 118 }
 119
 120 static int
 121 network_add_alias(struct uci_map *map, void *section)
 122 {
 123         struct uci_alias *a = section;
 124
 125         if (a->interface)
 126                 list_add_tail(&a->list, &a->interface->alias);
 127
 128         return 0;
 129 }
 130
 131 static struct ucimap_section_data *
 132 network_allocate(struct uci_map *map, struct uci_sectionmap *sm, struct uci_section *s)
 133 {
 134         struct uci_network *p = malloc(sizeof(struct uci_network));
 135         memset(p, 0, sizeof(struct uci_network));
 136         return &p->map;
 137 }
 138
 139 struct my_optmap {
 140         struct uci_optmap map;
 141         int test;
 142 };
 143
 144 static struct uci_sectionmap network_interface;
 145 static struct uci_sectionmap network_alias;
 146
 147 static struct my_optmap network_interface_options[] = {
 148         {
 149                 .map = {
 150                         UCIMAP_OPTION(struct uci_network, proto),
 151                         .type = UCIMAP_STRING,
 152                         .name = "proto",
 153                         .data.s.maxlen = 32,
 154                 }
 155         },
 156         {
 157                 .map = {
 158                         UCIMAP_OPTION(struct uci_network, ifname),
 159                         .type = UCIMAP_STRING,
 160                         .name = "ifname"
 161                 }
 162         },
 163         {
 164                 .map = {
 165                         UCIMAP_OPTION(struct uci_network, ipaddr),
 166                         .type = UCIMAP_CUSTOM,
 167                         .name = "ipaddr",
 168                         .parse = network_parse_ip,
 169                         .format = network_format_ip,
 170                         .free = network_free_ip,
 171                 }
 172         },
 173         {
 174                 .map = {
 175                         UCIMAP_OPTION(struct uci_network, enabled),
 176                         .type = UCIMAP_BOOL,
 177                         .name = "enabled",
 178                 }
 179         },
 180         {
 181                 .map = {
 182                         UCIMAP_OPTION(struct uci_network, test),
 183                         .type = UCIMAP_INT,
 184                         .name = "test"
 185                 }
 186         },
 187         {
 188                 .map = {
 189                         UCIMAP_OPTION(struct uci_network, aliases),
 190                         .type = UCIMAP_LIST | UCIMAP_SECTION | UCIMAP_LIST_AUTO,
 191                         .data.sm = &network_alias
 192                 }
 193         }
 194 };
 195
 196 static struct uci_sectionmap network_interface = {
 197         UCIMAP_SECTION(struct uci_network, map),
 198         .type = "interface",
 199         .alloc = network_allocate,
 200         .init = network_init_interface,
 201         .add = network_add_interface,
 202         .options = &network_interface_options[0].map,
 203         .n_options = ARRAY_SIZE(network_interface_options),
 204         .options_size = sizeof(struct my_optmap)
 205 };
 206
 207 static struct uci_optmap network_alias_options[] = {
 208         {
 209                 UCIMAP_OPTION(struct uci_alias, interface),
 210                 .type = UCIMAP_SECTION,
 211                 .data.sm = &network_interface
 212         }
 213 };
 214
 215 static struct uci_sectionmap network_alias = {
 216         UCIMAP_SECTION(struct uci_alias, map),
 217         .type = "alias",
 218         .options = network_alias_options,
 219         .init = network_init_alias,
 220         .add = network_add_alias,
 221         .n_options = ARRAY_SIZE(network_alias_options),
 222 };
 223
 224 static struct uci_sectionmap *network_smap[] = {
 225         &network_interface,
 226         &network_alias,
 227 };
 228
 229 static struct uci_map network_map = {
 230         .sections = network_smap,
 231         .n_sections = ARRAY_SIZE(network_smap),
 232 };
 233
 234
 235 int main(int argc, char **argv)
 236 {
 237         struct uci_context *ctx;
 238         struct uci_package *pkg;
 239         struct list_head *p;
 240         struct uci_network *net;
 241         struct uci_alias *alias;
 242         bool set = false;
 243         int i;
 244
 245         INIT_LIST_HEAD(&ifs);
 246         ctx = uci_alloc_context();
 247         ucimap_init(&network_map);
 248
 249         if ((argc >= 2) && !strcmp(argv[1], "-s")) {
 250                 uci_set_savedir(ctx, "./test/save");
 251                 set = true;
 252         }
 253
 254         uci_set_confdir(ctx, "./test/config");
 255         uci_load(ctx, "network", &pkg);
 256
 257         ucimap_parse(&network_map, pkg);
 258
 259         list_for_each(p, &ifs) {
 260                 const unsigned char *ipaddr;
 261                 int n_aliases = 0;
 262
 263                 net = list_entry(p, struct uci_network, list);
 264                 ipaddr = net->ipaddr;
 265                 if (!ipaddr)
 266                         ipaddr = (const unsigned char *) "\x00\x00\x00\x00";
 267
 268                 printf("New network section '%s'\n"
 269                         "       type: %s\n"
 270                         "       ifname: %s\n"
 271                         "       ipaddr: %d.%d.%d.%d\n"
 272                         "       test: %d\n"
 273                         "       enabled: %s\n",
 274                         net->name,
 275                         net->proto,
 276                         net->ifname,
 277                         ipaddr[0], ipaddr[1], ipaddr[2], ipaddr[3],
 278                         net->test,
 279                         (net->enabled ? "on" : "off"));
 280
 281                 if (net->aliases->n_items > 0) {
 282                         printf("Configured aliases:");
 283                         for (i = 0; i < net->aliases->n_items; i++) {
 284                                 alias = net->aliases->item[i].ptr;
 285                                 printf(" %s", alias->name);
 286                         }
 287                         printf("\n");
 288                 }
 289                 list_for_each_entry(alias, &net->alias, list) {
 290                         n_aliases++;
 291                         for (i = 0; i < net->aliases->n_items; i++) {
 292                                 if (alias == net->aliases->item[i].ptr)
 293                                         goto next_alias;
 294                         }
 295                         printf("New alias: %s\n", alias->name);
 296 next_alias:
 297                         continue;
 298                 }
 299                 if (set && !strcmp(net->name, "lan")) {
 300                         ucimap_free_item(&net->map, &net->ipaddr);
 301                         ucimap_set_changed(&net->map, &net->ipaddr);
 302                         ucimap_resize_list(&net->map, &net->aliases, n_aliases);
 303                         net->aliases->n_items = 0;
 304                         list_for_each_entry(alias, &net->alias, list) {
 305                                 net->aliases->item[net->aliases->n_items++].ptr = alias;
 306                         }
 307                         ucimap_set_changed(&net->map, &net->aliases);
 308                         ucimap_store_section(&network_map, pkg, &net->map);
 309                         uci_save(ctx, pkg);
 310                 }
 311         }
 312
 313
 314         ucimap_cleanup(&network_map);
 315         uci_free_context(ctx);
 316
 317         return 0;
 318 }