add an extra check
[project/uci.git] / list.c
1 /*
2  * libuci - Library for the Unified Configuration Interface
3  * Copyright (C) 2008 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 lesser general public license version 2.1
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
15 #include <glob.h>
16
17 /* initialize a list head/item */
18 static inline void uci_list_init(struct uci_list *ptr)
19 {
20         ptr->prev = ptr;
21         ptr->next = ptr;
22 }
23
24 /* inserts a new list entry after a given entry */
25 static inline void uci_list_insert(struct uci_list *list, struct uci_list *ptr)
26 {
27         list->next->prev = ptr;
28         ptr->prev = list;
29         ptr->next = list->next;
30         list->next = ptr;
31 }
32
33 /* inserts a new list entry at the tail of the list */
34 static inline void uci_list_add(struct uci_list *head, struct uci_list *ptr)
35 {
36         /* NB: head->prev points at the tail */
37         uci_list_insert(head->prev, ptr);
38 }
39
40 static inline void uci_list_del(struct uci_list *ptr)
41 {
42         struct uci_list *next, *prev;
43
44         next = ptr->next;
45         prev = ptr->prev;
46
47         prev->next = next;
48         next->prev = prev;
49
50         uci_list_init(ptr);
51 }
52
53 /* 
54  * uci_alloc_generic allocates a new uci_element with payload
55  * payload is appended to the struct to save memory and reduce fragmentation
56  */
57 static struct uci_element *
58 uci_alloc_generic(struct uci_context *ctx, int type, const char *name, int size)
59 {
60         struct uci_element *e;
61         void *ptr;
62
63         ptr = uci_malloc(ctx, size + strlen(name) + 1);
64         e = (struct uci_element *) ptr;
65         e->type = type;
66         e->name = (char *) ptr + size;
67         strcpy(e->name, name);
68         uci_list_init(&e->list);
69
70         return e;
71 }
72
73 static void
74 uci_free_element(struct uci_element *e)
75 {
76         if (!uci_list_empty(&e->list))
77                 uci_list_del(&e->list);
78         free(e);
79 }
80
81 static struct uci_option *
82 uci_alloc_option(struct uci_section *s, const char *name, const char *value)
83 {
84         struct uci_package *p = s->package;
85         struct uci_context *ctx = p->ctx;
86         struct uci_option *o;
87
88         o = uci_alloc_element(ctx, option, name, strlen(value) + 1);
89         o->value = uci_dataptr(o);
90         o->section = s;
91         strcpy(o->value, value);
92         uci_list_add(&s->options, &o->e.list);
93
94         return o;
95 }
96
97 static inline void
98 uci_free_option(struct uci_option *o)
99 {
100         uci_free_element(&o->e);
101 }
102
103 static struct uci_section *
104 uci_alloc_section(struct uci_package *p, const char *type, const char *name)
105 {
106         struct uci_context *ctx = p->ctx;
107         struct uci_section *s;
108         char buf[16];
109
110         if (!name || !name[0]) {
111                 snprintf(buf, 16, "cfg%d", p->n_section);
112                 name = buf;
113         }
114
115         s = uci_alloc_element(ctx, section, name, strlen(type) + 1);
116         s->type = uci_dataptr(s);
117         s->package = p;
118         strcpy(s->type, type);
119         uci_list_init(&s->options);
120         uci_list_add(&p->sections, &s->e.list);
121
122         return s;
123 }
124
125 static void
126 uci_free_section(struct uci_section *s)
127 {
128         struct uci_element *o, *tmp;
129
130         uci_foreach_element_safe(&s->options, tmp, o) {
131                 uci_free_option(uci_to_option(o));
132         }
133         uci_free_element(&s->e);
134 }
135
136 static struct uci_package *
137 uci_alloc_package(struct uci_context *ctx, const char *name)
138 {
139         struct uci_package *p;
140
141         p = uci_alloc_element(ctx, package, name, 0);
142         p->ctx = ctx;
143         uci_list_init(&p->sections);
144         uci_list_init(&p->history);
145         return p;
146 }
147
148 static void
149 uci_free_package(struct uci_package *p)
150 {
151         struct uci_element *e, *tmp;
152
153         if(!p)
154                 return;
155
156         if (p->path)
157                 free(p->path);
158         uci_foreach_element_safe(&p->sections, tmp, e) {
159                 uci_free_section(uci_to_section(e));
160         }
161         uci_free_element(&p->e);
162 }
163
164 /* record a change that was done to a package */
165 static inline void
166 uci_add_history(struct uci_context *ctx, struct uci_package *p, int cmd, char *section, char *option, char *value)
167 {
168         struct uci_history *h = (struct uci_history *) uci_malloc(ctx, sizeof(struct uci_history));
169
170         uci_list_init(&h->list);
171         h->cmd = cmd;
172         h->section = section;
173         h->option = option;
174         h->value = value;
175         uci_list_add(&p->history, &h->list);
176 }
177
178
179 static struct uci_element *uci_lookup_list(struct uci_context *ctx, struct uci_list *list, const char *name)
180 {
181         struct uci_element *e;
182
183         uci_foreach_element(list, e) {
184                 if (!strcmp(e->name, name))
185                         return e;
186         }
187         UCI_THROW(ctx, UCI_ERR_NOTFOUND);
188 }
189
190 int uci_lookup(struct uci_context *ctx, struct uci_element **res, struct uci_package *p, char *section, char *option)
191 {
192         struct uci_element *e;
193         struct uci_section *s;
194         struct uci_option *o;
195
196         UCI_HANDLE_ERR(ctx);
197         UCI_ASSERT(ctx, res != NULL);
198         UCI_ASSERT(ctx, p != NULL);
199
200         e = uci_lookup_list(ctx, &p->sections, section);
201         if (!option)
202                 goto found;
203
204         s = uci_to_section(e);
205         e = uci_lookup_list(ctx, &s->options, option);
206
207 found:
208         *res = e;
209         return 0;
210 }
211
212 int uci_set_element_value(struct uci_context *ctx, struct uci_element **element, char *value)
213 {
214         int size;
215         char *str;
216         struct uci_list *list;
217         struct uci_element *e;
218
219         UCI_HANDLE_ERR(ctx);
220         UCI_ASSERT(ctx, value != NULL);
221         UCI_ASSERT(ctx, element != NULL);
222         UCI_ASSERT(ctx, *element != NULL);
223
224         /* what the 'value' of an element means depends on the type
225          * for a section, the 'value' means its type
226          * for an option, the 'value' means its value string
227          * when changing the value, shrink the element to its actual size
228          * (it may have been allocated with a bigger size, to include
229          *  its buffer)
230          * then duplicate the string passed on the command line and
231          * insert it into the structure.
232          */
233         e = *element;
234         list = e->list.prev;
235         switch(e->type) {
236         case UCI_TYPE_SECTION:
237                 size = sizeof(struct uci_section);
238                 break;
239         case UCI_TYPE_OPTION:
240                 size = sizeof(struct uci_option);
241                 break;
242         default:
243                 UCI_THROW(ctx, UCI_ERR_INVAL);
244                 break;
245         }
246
247         uci_list_del(&e->list);
248         e = uci_realloc(ctx, e, size);
249         str = uci_strdup(ctx, value);
250         uci_list_insert(list, &e->list);
251         *element = e;
252
253         switch(e->type) {
254         case UCI_TYPE_SECTION:
255                 uci_to_section(e)->type = value;
256                 break;
257         case UCI_TYPE_OPTION:
258                 uci_to_option(e)->value = value;
259                 break;
260         default:
261                 break;
262         }
263
264         return 0;
265 }
266
267 int uci_set(struct uci_context *ctx, char *package, char *section, char *option, char *value)
268 {
269         struct uci_element *e;
270         struct uci_package *p = NULL;
271         struct uci_section *s = NULL;
272         struct uci_option *o = NULL;
273         struct uci_history *h;
274
275         UCI_HANDLE_ERR(ctx);
276         UCI_ASSERT(ctx, package != NULL);
277         UCI_ASSERT(ctx, section != NULL);
278
279         /*
280          * look up the package, section and option (if set)
281          * if the section/option is to be modified and it is not found
282          * create a new element in the appropriate list
283          */
284         UCI_TRAP_SAVE(ctx, notfound);
285         e = uci_lookup_list(ctx, &ctx->root, package);
286         if (!e)
287                 goto notfound;
288
289         p = uci_to_package(e);
290         e = uci_lookup_list(ctx, &p->sections, section);
291         if (!e)
292                 goto notfound;
293         s = uci_to_section(e);
294
295         if (option) {
296                 e = uci_lookup_list(ctx, &s->options, option);
297                 if (!e)
298                         goto notfound;
299                 o = uci_to_option(e);
300         }
301         UCI_TRAP_RESTORE(ctx);
302
303         /* 
304          * no unknown element was supplied, assume that we can just update 
305          * an existing entry
306          */
307         if (o)
308                 e = &o->e;
309         else
310                 e = &s->e;
311
312         uci_add_history(ctx, p, UCI_CMD_CHANGE, section, option, value);
313         return uci_set_element_value(ctx, &e, value);
314
315 notfound:
316         /* 
317          * the entry that we need to update was not found,
318          * check if the search failed prematurely.
319          * this can happen if the package was not found, or if
320          * an option was supplied, but the section wasn't found
321          */
322         if (!p || (!s && option))
323                 UCI_THROW(ctx, ctx->errno);
324
325         /* now add the missing entry */
326         uci_add_history(ctx, p, UCI_CMD_ADD, section, option, value);
327         if (s)
328                 uci_alloc_option(s, option, value);
329         else
330                 uci_alloc_section(p, section, value);
331
332         return 0;
333 }
334
335 int uci_unload(struct uci_context *ctx, struct uci_package *p)
336 {
337         struct uci_element *e;
338
339         UCI_HANDLE_ERR(ctx);
340         UCI_ASSERT(ctx, p != NULL);
341
342         uci_free_package(p);
343         return 0;
344 }
345