]> mj.ucw.cz Git - libucw.git/blob - lib/conf-intr.c
5895bd6b97c2bfc75f86fe353a42f72f9cee7509
[libucw.git] / lib / conf-intr.c
1 /*
2  *      UCW Library -- Configuration files: interpreter
3  *
4  *      (c) 2001--2006 Robert Spalek <robert@ucw.cz>
5  *      (c) 2003--2006 Martin Mares <mj@ucw.cz>
6  *
7  *      This software may be freely distributed and used according to the terms
8  *      of the GNU Lesser General Public License.
9  */
10
11 #include "lib/lib.h"
12 #include "lib/conf.h"
13 #include "lib/getopt.h"
14 #include "lib/conf-internal.h"
15 #include "lib/clists.h"
16
17 #include <string.h>
18 #include <stdio.h>
19
20 #define TRY(f)  do { byte *_msg = f; if (_msg) return _msg; } while (0)
21
22 /* Register size of and parser for each basic type */
23
24 static byte *
25 cf_parse_string(byte *str, byte **ptr)
26 {
27   *ptr = cf_strdup(str);
28   return NULL;
29 }
30
31 typedef byte *cf_basic_parser(byte *str, void *ptr);
32 static struct {
33   uns size;
34   void *parser;
35 } parsers[] = {
36   { sizeof(int), cf_parse_int },
37   { sizeof(u64), cf_parse_u64 },
38   { sizeof(double), cf_parse_double },
39   { sizeof(u32), cf_parse_ip },
40   { sizeof(byte*), cf_parse_string },
41   { sizeof(int), NULL },                        // lookups are parsed extra
42   { 0, NULL },                                  // user-defined types are parsed extra
43 };
44
45 inline uns
46 cf_type_size(enum cf_type type, struct cf_user_type *utype)
47 {
48   if (type < CT_USER)
49     return parsers[type].size;
50   else
51     return utype->size;
52 }
53
54 static byte *
55 cf_parse_lookup(byte *str, int *ptr, byte **t)
56 {
57   byte **n = t;
58   uns total_len = 0;
59   while (*n && strcasecmp(*n, str)) {
60     total_len += strlen(*n) + 2;
61     n++;
62   }
63   if (*n) {
64     *ptr = n - t;
65     return NULL;
66   }
67   byte *err = cf_malloc(total_len + strlen(str) + 60), *c = err;
68   c += sprintf(err, "Invalid value %s, possible values are: ", str);
69   for (n=t; *n; n++)
70     c+= sprintf(c, "%s, ", *n);
71   if (*t)
72     c[-2] = 0;
73   *ptr = -1;
74   return err;
75 }
76
77 static byte *
78 cf_parse_ary(uns number, byte **pars, void *ptr, enum cf_type type, union cf_union *u)
79 {
80   for (uns i=0; i<number; i++)
81   {
82     byte *msg;
83     uns size = cf_type_size(type, u->utype);
84     if (type < CT_LOOKUP)
85       msg = ((cf_basic_parser*) parsers[type].parser) (pars[i], ptr + i * size);
86     else if (type == CT_LOOKUP)
87       msg = cf_parse_lookup(pars[i], ptr + i * size, u->lookup);
88     else if (type == CT_USER)
89       msg = u->utype->parser(pars[i], ptr + i * size);
90     else
91       ASSERT(0);
92     if (msg)
93       return cf_printf("Cannot parse item %d: %s", i+1, msg);
94   }
95   return NULL;
96 }
97
98 /* Interpreter */
99
100 #define T(x) #x,
101 byte *cf_op_names[] = { CF_OPERATIONS };
102 #undef T
103
104 static byte *
105 interpret_set_dynamic(struct cf_item *item, int number, byte **pars, void **ptr)
106 {
107   enum cf_type type = item->type;
108   cf_journal_block(ptr, sizeof(void*));
109   // boundary checks done by the caller
110   uns size = cf_type_size(item->type, item->u.utype);
111   ASSERT(size >= sizeof(uns));
112   *ptr = cf_malloc((number+1) * size) + size;
113   * (uns*) (*ptr - size) = number;
114   return cf_parse_ary(number, pars, *ptr, type, &item->u);
115 }
116
117 static byte *
118 interpret_add_dynamic(struct cf_item *item, int number, byte **pars, int *processed, void **ptr, enum cf_operation op)
119 {
120   enum cf_type type = item->type;
121   void *old_p = *ptr;
122   uns size = cf_type_size(item->type, item->u.utype);
123   ASSERT(size >= sizeof(uns));
124   int old_nr = old_p ? * (int*) (old_p - size) : 0;
125   int taken = MIN(number, ABS(item->number)-old_nr);
126   *processed = taken;
127   // stretch the dynamic array
128   void *new_p = cf_malloc((old_nr + taken + 1) * size) + size;
129   * (uns*) (new_p - size) = old_nr + taken;
130   cf_journal_block(ptr, sizeof(void*));
131   *ptr = new_p;
132   if (op == OP_APPEND) {
133     memcpy(new_p, old_p, old_nr * size);
134     return cf_parse_ary(taken, pars, new_p + old_nr * size, type, &item->u);
135   } else if (op == OP_PREPEND) {
136     memcpy(new_p + taken * size, old_p, old_nr * size);
137     return cf_parse_ary(taken, pars, new_p, type, &item->u);
138   } else
139     return cf_printf("Dynamic arrays do not support operation %s", cf_op_names[op]);
140 }
141
142 static byte *interpret_set_item(struct cf_item *item, int number, byte **pars, int *processed, void *ptr, uns allow_dynamic);
143
144 static byte *
145 interpret_section(struct cf_section *sec, int number, byte **pars, int *processed, void *ptr, uns allow_dynamic)
146 {
147   cf_add_dirty(sec, ptr);
148   *processed = 0;
149   for (struct cf_item *ci=sec->cfg; ci->cls; ci++)
150   {
151     int taken;
152     byte *msg = interpret_set_item(ci, number, pars, &taken, ptr + (addr_int_t) ci->ptr, allow_dynamic && !ci[1].cls);
153     if (msg)
154       return cf_printf("Item %s: %s", ci->name, msg);
155     *processed += taken;
156     number -= taken;
157     pars += taken;
158     if (!number)                // stop parsing, because many parsers would otherwise complain that number==0
159       break;
160   }
161   return NULL;
162 }
163
164 static void
165 add_to_list(cnode *where, cnode *new_node, enum cf_operation op)
166 {
167   switch (op)
168   {
169     case OP_EDIT:               // edition has been done in-place
170       break;
171     case OP_REMOVE:
172       CF_JOURNAL_VAR(where->prev->next);
173       CF_JOURNAL_VAR(where->next->prev);
174       clist_remove(where);
175       break;
176     case OP_AFTER:              // implementation dependend (prepend_head = after(list)), and where==list, see clists.h:74
177     case OP_PREPEND:
178     case OP_COPY:
179       CF_JOURNAL_VAR(where->next->prev);
180       CF_JOURNAL_VAR(where->next);
181       clist_insert_after(new_node, where);
182       break;
183     case OP_BEFORE:             // implementation dependend (append_tail = before(list))
184     case OP_APPEND:
185     case OP_SET:
186       CF_JOURNAL_VAR(where->prev->next);
187       CF_JOURNAL_VAR(where->prev);
188       clist_insert_before(new_node, where);
189       break;
190     default:
191       ASSERT(0);
192   }
193 }
194
195 static byte *
196 interpret_add_list(struct cf_item *item, int number, byte **pars, int *processed, void *ptr, enum cf_operation op)
197 {
198   if (op >= OP_REMOVE)
199     return cf_printf("You have to open a block for operation %s", cf_op_names[op]);
200   if (!number)
201     return "Nothing to add to the list";
202   struct cf_section *sec = item->u.sec;
203   *processed = 0;
204   while (number > 0)
205   {
206     void *node = cf_malloc(sec->size);
207     cf_init_section(item->name, sec, node, 1);
208     add_to_list(ptr, node, op);
209     int taken;
210     /* If the node contains any dynamic attribute at the end, we suppress
211      * auto-repetition here and pass the flag inside instead.  */
212     TRY( interpret_section(sec, number, pars, &taken, node, sec->flags & SEC_FLAG_DYNAMIC) );
213     *processed += taken;
214     number -= taken;
215     pars += taken;
216     if (sec->flags & SEC_FLAG_DYNAMIC)
217       break;
218   }
219   return NULL;
220 }
221
222 static byte *
223 interpret_set_item(struct cf_item *item, int number, byte **pars, int *processed, void *ptr, uns allow_dynamic)
224 {
225   int taken;
226   switch (item->cls)
227   {
228     case CC_STATIC:
229       if (!number)
230         return "Missing value";
231       taken = MIN(number, item->number);
232       *processed = taken;
233       uns size = cf_type_size(item->type, item->u.utype);
234       cf_journal_block(ptr, taken * size);
235       return cf_parse_ary(taken, pars, ptr, item->type, &item->u);
236     case CC_DYNAMIC:
237       if (!allow_dynamic)
238         return "Dynamic array cannot be used here";
239       taken = MIN(number, ABS(item->number));
240       *processed = taken;
241       return interpret_set_dynamic(item, taken, pars, ptr);
242     case CC_PARSER:
243       if (item->number < 0 && !allow_dynamic)
244         return "Parsers with variable number of parameters cannot be used here";
245       if (item->number > 0 && number < item->number)
246         return "Not enough parameters available for the parser";
247       taken = MIN(number, ABS(item->number));
248       *processed = taken;
249       for (int i=0; i<taken; i++)
250         pars[i] = cf_strdup(pars[i]);
251       return item->u.par(taken, pars, ptr);
252     case CC_SECTION:
253       return interpret_section(item->u.sec, number, pars, processed, ptr, allow_dynamic);
254     case CC_LIST:
255       if (!allow_dynamic)
256         return "Lists cannot be used here";
257       return interpret_add_list(item, number, pars, processed, ptr, OP_SET);
258     default:
259       ASSERT(0);
260   }
261 }
262
263 static byte *
264 interpret_clear(struct cf_item *item, void *ptr)
265 {
266   if (item->cls == CC_LIST) {
267     cf_journal_block(ptr, sizeof(clist));
268     clist_init(ptr);
269   } else if (item->cls == CC_DYNAMIC) {
270     cf_journal_block(ptr, sizeof(void *));
271     * (void**) ptr = NULL;
272   } else
273     return "The item is not a list or a dynamic array";
274   return NULL;
275 }
276
277 static int
278 cmp_items(void *i1, void *i2, struct cf_item *item)
279 {
280   ASSERT(item->cls == CC_STATIC);
281   i1 += (addr_int_t) item->ptr;
282   i2 += (addr_int_t) item->ptr;
283   if (item->type == CT_STRING)
284     return strcmp(* (byte**) i1, * (byte**) i2);
285   else                          // all numeric types
286     return memcmp(i1, i2, cf_type_size(item->type, item->u.utype));
287 }
288
289 static void *
290 find_list_node(clist *list, void *query, struct cf_section *sec, u32 mask)
291 {
292   CLIST_FOR_EACH(cnode *, n, *list)
293   {
294     uns found = 1;
295     for (uns i=0; i<32; i++)
296       if (mask & (1<<i))
297         if (cmp_items(n, query, sec->cfg+i))
298         {
299           found = 0;
300           break;
301         }
302     if (found)
303       return n;
304   }
305   return NULL;
306 }
307
308 static byte *
309 record_selector(struct cf_item *item, struct cf_section *sec, u32 *mask)
310 {
311   uns nr = sec->flags & SEC_FLAG_NUMBER;
312   if (item >= sec->cfg && item < sec->cfg + nr) // setting an attribute relative to this section
313   {
314     uns i = item - sec->cfg;
315     if (i >= 32)
316       return "Cannot select list nodes by this attribute";
317     if (sec->cfg[i].cls != CC_STATIC)
318       return "Selection can only be done based on basic attributes";
319     *mask |= 1 << i;
320   }
321   return NULL;
322 }
323
324 #define MAX_STACK_SIZE  10
325 static struct item_stack {
326   struct cf_section *sec;       // nested section
327   void *base_ptr;               // because original pointers are often relative
328   enum cf_operation op;         // it is performed when a closing brace is encountered
329   void *list;                   // list the operations should be done on
330   u32 mask;                     // bit array of selectors searching in a list
331   struct cf_item *item;         // cf_item of the list
332 } stack[MAX_STACK_SIZE];
333 static uns level;
334
335 static byte *
336 opening_brace(struct cf_item *item, void *ptr, enum cf_operation op)
337 {
338   if (level >= MAX_STACK_SIZE-1)
339     return "Too many nested sections";
340   stack[++level] = (struct item_stack) {
341     .sec = NULL,
342     .base_ptr = NULL,
343     .op = op & OP_MASK,
344     .list = NULL,
345     .mask = 0,
346     .item = NULL,
347   };
348   if (!item)                    // unknown is ignored; we just need to trace recursion
349     return NULL;
350   stack[level].sec = item->u.sec;
351   if (item->cls == CC_SECTION)
352   {
353     stack[level].base_ptr = ptr;
354     stack[level].op = OP_EDIT | OP_2ND; // this list operation does nothing
355   }
356   else if (item->cls == CC_LIST)
357   {
358     stack[level].base_ptr = cf_malloc(item->u.sec->size);
359     cf_init_section(item->name, item->u.sec, stack[level].base_ptr, 1);
360     stack[level].list = ptr;
361     stack[level].item = item;
362     if ((op & OP_MASK) < OP_REMOVE) {
363       add_to_list(ptr, stack[level].base_ptr, op & OP_MASK);
364       stack[level].op |= OP_2ND;
365     } else
366       stack[level].op |= OP_1ST;
367   }
368   else
369     return "Opening brace can only be used on sections and lists";
370   return NULL;
371 }
372
373 static byte *
374 closing_brace(struct item_stack *st, enum cf_operation op, int number, byte **pars)
375 {
376   if (st->op == OP_CLOSE)       // top-level
377     return "Unmatched } parenthesis";
378   if (!st->sec) {               // dummy run on unknown section
379     if (!(op & OP_OPEN))
380       level--;
381     return NULL;
382   }
383   enum cf_operation pure_op = st->op & OP_MASK;
384   if (st->op & OP_1ST)
385   {
386     st->list = find_list_node(st->list, st->base_ptr, st->sec, st->mask);
387     if (!st->list)
388       return "Cannot find a node matching the query";
389     if (pure_op != OP_REMOVE)
390     {
391       if (pure_op == OP_EDIT)
392         st->base_ptr = st->list;
393       else if (pure_op == OP_AFTER || pure_op == OP_BEFORE)
394         cf_init_section(st->item->name, st->sec, st->base_ptr, 1);
395       else if (pure_op == OP_COPY) {
396         if (st->sec->flags & SEC_FLAG_CANT_COPY)
397           return cf_printf("Item %s cannot be copied", st->item->name);
398         memcpy(st->base_ptr, st->list, st->sec->size);  // strings and dynamic arrays are shared
399         if (st->sec->copy)
400           TRY( st->sec->copy(st->base_ptr, st->list) );
401       } else
402         ASSERT(0);
403       if (op & OP_OPEN) {       // stay at the same recursion level
404         st->op = (st->op | OP_2ND) & ~OP_1ST;
405         add_to_list(st->list, st->base_ptr, pure_op);
406         return NULL;
407       }
408       int taken;                // parse parameters on 1 line immediately
409       TRY( interpret_section(st->sec, number, pars, &taken, st->base_ptr, 1) );
410       number -= taken;
411       pars += taken;
412       // and fall-thru to the 2nd phase
413     }
414     add_to_list(st->list, st->base_ptr, pure_op);
415   }
416   level--;
417   if (number)
418     return "No parameters expected after the }";
419   else if (op & OP_OPEN)
420     return "No { is expected";
421   else
422     return NULL;
423 }
424
425 static struct cf_item *
426 find_item(struct cf_section *curr_sec, byte *name, byte **msg, void **ptr)
427 {
428   *msg = NULL;
429   if (name[0] == '^')                           // absolute name instead of relative
430     name++, curr_sec = &cf_sections, *ptr = NULL;
431   if (!curr_sec)                                // don't even search in an unknown section
432     return NULL;
433   while (1)
434   {
435     if (curr_sec != &cf_sections)
436       cf_add_dirty(curr_sec, *ptr);
437     byte *c = strchr(name, '.');
438     if (c)
439       *c++ = 0;
440     struct cf_item *ci = cf_find_subitem(curr_sec, name);
441     if (!ci->cls)
442     {
443       if (!(curr_sec->flags & SEC_FLAG_UNKNOWN))        // ignore silently unknown top-level sections and unknown attributes in flagged sections
444         *msg = cf_printf("Unknown item %s", name);
445       return NULL;
446     }
447     *ptr += (addr_int_t) ci->ptr;
448     if (!c)
449       return ci;
450     if (ci->cls != CC_SECTION)
451     {
452       *msg = cf_printf("Item %s is not a section", name);
453       return NULL;
454     }
455     curr_sec = ci->u.sec;
456     name = c;
457   }
458 }
459
460 byte *
461 cf_interpret_line(byte *name, enum cf_operation op, int number, byte **pars)
462 {
463   byte *msg;
464   if ((op & OP_MASK) == OP_CLOSE)
465     return closing_brace(stack+level, op, number, pars);
466   void *ptr = stack[level].base_ptr;
467   struct cf_item *item = find_item(stack[level].sec, name, &msg, &ptr);
468   if (msg)
469     return msg;
470   if (stack[level].op & OP_1ST)
471     TRY( record_selector(item, stack[level].sec, &stack[level].mask) );
472   if (op & OP_OPEN) {           // the operation will be performed after the closing brace
473     if (number)
474       return "Cannot open a block after a parameter has been passed on a line";
475     return opening_brace(item, ptr, op);
476   }
477   if (!item)                    // ignored item in an unknown section
478     return NULL;
479   op &= OP_MASK;
480
481   int taken;                    // process as many parameters as possible
482   if (op == OP_CLEAR)
483     taken = 0, msg = interpret_clear(item, ptr);
484   else if (op == OP_SET)
485     msg = interpret_set_item(item, number, pars, &taken, ptr, 1);
486   else if (item->cls == CC_DYNAMIC)
487     msg = interpret_add_dynamic(item, number, pars, &taken, ptr, op);
488   else if (item->cls == CC_LIST)
489     msg = interpret_add_list(item, number, pars, &taken, ptr, op);
490   else
491     return cf_printf("Operation %s not supported on attribute %s", cf_op_names[op], name);
492   if (msg)
493     return msg;
494   if (taken < number)
495     return cf_printf("Too many parameters: %d>%d", number, taken);
496
497   return NULL;
498 }
499
500 byte *
501 cf_find_item(byte *name, struct cf_item *item)
502 {
503   byte *msg;
504   void *ptr;
505   struct cf_item *ci = find_item(&cf_sections, name, &msg, &ptr);
506   if (msg)
507     return msg;
508   if (ci) {
509     *item = *ci;
510     item->ptr = ptr;
511   } else
512     bzero(item, sizeof(struct cf_item));
513   return NULL;
514 }
515
516 byte *
517 cf_write_item(struct cf_item *item, enum cf_operation op, int number, byte **pars)
518 {
519   byte *msg;
520   int taken;
521   switch (op) {
522     case OP_SET:
523       msg = interpret_set_item(item, number, pars, &taken, item->ptr, 1);
524       break;
525     case OP_CLEAR:
526       taken = 0;
527       msg = interpret_clear(item, item->ptr);
528       break;
529     case OP_APPEND:
530     case OP_PREPEND:
531       if (item->cls == CC_DYNAMIC)
532         msg = interpret_add_dynamic(item, number, pars, &taken, item->ptr, op);
533       else if (item->cls == CC_LIST)
534         msg = interpret_add_list(item, number, pars, &taken, item->ptr, op);
535       else
536         return "The attribute class does not support append/prepend";
537       break;
538     default:
539       return "Unsupported operation";
540   }
541   if (msg)
542     return msg;
543   if (taken < number)
544     return "Too many parameters";
545   return NULL;
546 }
547
548 void
549 cf_init_stack(void)
550 {
551   static uns initialized = 0;
552   if (!initialized++) {
553     cf_sections.flags |= SEC_FLAG_UNKNOWN;
554     cf_sections.size = 0;                       // size of allocated array used to be stored here
555     cf_init_section(NULL, &cf_sections, NULL, 0);
556   }
557   level = 0;
558   stack[0] = (struct item_stack) {
559     .sec = &cf_sections,
560     .base_ptr = NULL,
561     .op = OP_CLOSE,
562     .list = NULL,
563     .mask = 0,
564     .item = NULL
565   };
566 }
567
568 int
569 cf_check_stack(void)
570 {
571   if (level > 0) {
572     log(L_ERROR, "Unterminated block");
573     return 1;
574   }
575   return 0;
576 }
577