ucw/conf-test.c

   1 /*
   2  *      Insane tester of reading configuration files
   3  *
   4  *      (c) 2006 Robert Spalek <robert@ucw.cz>
   5  *      (c) 2012--2014 Martin Mares <mj@ucw.cz>
   6  */
   7
   8 #include <ucw/lib.h>
   9 #include <ucw/conf.h>
  10 #include <ucw/getopt.h>
  11 #include <ucw/clists.h>
  12 #include <ucw/fastbuf.h>
  13 #include <ucw/gary.h>
  14
  15 #include <stdlib.h>
  16 #include <stdio.h>
  17 #include <time.h>
  18
  19 static int verbose;
  20 static int reload;
  21
  22 struct sub_sect_1 {
  23   cnode n;
  24   char *name;
  25   time_t t;
  26   char *level;
  27   int confidence[2];
  28   double *list;
  29 };
  30
  31 static struct sub_sect_1 sec1 = { {}, "Charlie", 0, "WBAFC", { 0, -1}, NULL };
  32
  33 static char *
  34 init_sec_1(struct sub_sect_1 *s)
  35 {
  36   if (s == &sec1)                       // this is a static variable; skip clearing
  37     return NULL;
  38   s->name = "unknown";
  39   s->level = "default";
  40   s->confidence[0] = 5;
  41   s->confidence[1] = 6;
  42   // leave s->list==NULL
  43   return NULL;
  44 }
  45
  46 static char *
  47 commit_sec_1(struct sub_sect_1 *s)
  48 {
  49   if (s->confidence[0] < 0 || s->confidence[0] > 10)
  50     return "Well, this can't be";
  51   return NULL;
  52 }
  53
  54 static char *
  55 time_parser(uint number, char **pars, time_t *ptr)
  56 {
  57   *ptr = number ? atoi(pars[0]) : time(NULL);
  58   return NULL;
  59 }
  60
  61 static struct cf_section cf_sec_1 = {
  62   CF_TYPE(struct sub_sect_1),
  63   CF_INIT(init_sec_1),
  64   CF_COMMIT(commit_sec_1),
  65 #define F(x)    PTR_TO(struct sub_sect_1, x)
  66   CF_ITEMS {
  67     CF_STRING("name", F(name)),
  68     //CF_PARSER("t", F(t), time_parser, 0),
  69     CF_STRING("level", F(level)),
  70     CF_INT_ARY("confidence", F(confidence[0]), 2),              // XXX: the [0] is needed for the sake of type checking
  71     CF_DOUBLE_DYN("list", F(list), 100),
  72     CF_END
  73   }
  74 #undef F
  75 };
  76
  77 static uint nr1 = 15;
  78 static int *nrs1;
  79 static int nrs2[5];
  80 static char *str1 = "no worries";
  81 static char **str2;
  82 static u64 u1 = 0xCafeBeefDeadC00ll;
  83 static double d1 = -1.1;
  84 static clist secs;
  85 static time_t t1, t2;
  86 static u32 ip;
  87 static int *look;
  88 static u16 numbers[10] = { 2, 100, 1, 5 };
  89 static u32 bitmap1 = 0xff;
  90 static u32 bitmap2 = 3;
  91
  92 static char *
  93 parse_u16(char *string, u16 *ptr)
  94 {
  95   uint a;
  96   char *msg = cf_parse_int(string, &a);
  97   if (msg)
  98     return msg;
  99   if (a >= (1<<16))
 100     return "Come on, man, this doesn't fit to 16 bits";
 101   *ptr = a;
 102   return NULL;
 103 }
 104
 105 static void
 106 dump_u16(struct fastbuf *fb, u16 *ptr)
 107 {
 108   bprintf(fb, "%d ", *ptr);
 109 }
 110
 111 static struct cf_user_type u16_type = {
 112   .size = sizeof(u16),
 113   .name = "u16",
 114   .parser = (cf_parser1*) parse_u16,
 115   .dumper = (cf_dumper1*) dump_u16
 116 };
 117
 118 static char *
 119 init_top(void *ptr UNUSED)
 120 {
 121   for (uint i=0; i<5; i++)
 122   {
 123     struct sub_sect_1 *s = xmalloc(sizeof(struct sub_sect_1));  // XXX: cannot by cf_malloc(), because it's deleted when cf_reload()'ed
 124     cf_init_section("slaves", &cf_sec_1, s, 1);
 125     s->confidence[1] = i;
 126     clist_add_tail(&secs, &s->n);
 127   }
 128   return NULL;
 129 }
 130
 131 static char *
 132 commit_top(void *ptr UNUSED)
 133 {
 134   if (nr1 != 15)
 135     return "Don't touch my variable!";
 136   return NULL;
 137 }
 138
 139 static const char * const alphabet[] = { "alpha", "beta", "gamma", "delta", NULL };
 140 static struct cf_section cf_top = {
 141   CF_INIT(init_top),
 142   CF_COMMIT(commit_top),
 143   CF_ITEMS {
 144     CF_UINT("nr1", &nr1),
 145     CF_INT_DYN("nrs1", &nrs1, 1000),
 146     CF_INT_ARY("nrs2", nrs2, 5),
 147     CF_STRING("str1", &str1),
 148     CF_STRING_DYN("str2", &str2, 20),
 149     CF_U64("u1", &u1),
 150     CF_DOUBLE("d1", &d1),
 151     CF_PARSER("FirstTime", &t1, time_parser, -1),
 152     CF_PARSER("SecondTime", &t2, time_parser, 1),
 153     CF_SECTION("master", &sec1, &cf_sec_1),
 154     CF_LIST("slaves", &secs, &cf_sec_1),
 155     CF_IP("ip", &ip),
 156     CF_LOOKUP_DYN("look", &look, alphabet, 1000),
 157     CF_USER_ARY("numbers", numbers, &u16_type, 10),
 158     CF_BITMAP_INT("bitmap1", &bitmap1),
 159     CF_BITMAP_LOOKUP("bitmap2", &bitmap2, ((const char* const[]) {
 160           "one", "two", "three", "four", "five", "six", "seven", "eight",
 161           "nine", "ten", "eleven", "twelve", "thirteen", "fourteen", "fifteen", "seventeen",
 162           "eighteen", "nineteen", "twenty", NULL        // hidden joke here
 163           })),
 164     CF_END
 165   }
 166 };
 167
 168 static char short_opts[] = CF_SHORT_OPTS "rv";
 169 static struct option long_opts[] = {
 170         CF_LONG_OPTS
 171         {"reload",      0, 0, 'r'},
 172         {"verbose",     0, 0, 'v'},
 173         {NULL,          0, 0, 0}
 174 };
 175
 176 static char *help = "\
 177 Usage: conf-test [ctxt] [nojournal] <options>\n\
 178 \n\
 179 Options:\n" CF_USAGE "\
 180 -r, --reload\t\tReload configuration\n\
 181 -v, --verbose\t\tBe verbose\n\
 182 ";
 183
 184 static void NONRET
 185 usage(char *msg, ...)
 186 {
 187   va_list va;
 188   va_start(va, msg);
 189   if (msg)
 190     vfprintf(stderr, msg, va);
 191   fputs(help, stderr);
 192   exit(1);
 193 }
 194
 195 int
 196 main(int argc, char *argv[])
 197 {
 198   log_init(argv[0]);
 199   struct cf_context *cc = NULL, *prev = NULL;
 200
 201   // Special arguments which have to be parsed before cf_getopt()
 202   while (argc > 1) {
 203     if (!strcmp(argv[1], "ctxt")) {
 204       cc = cf_new_context();
 205       prev = cf_switch_context(cc);
 206       argc--, argv++;
 207     } else if (!strcmp(argv[1], "nojournal")) {
 208       cf_set_journalling(0);
 209       argc--, argv++;
 210     } else
 211       break;
 212   }
 213
 214   cf_declare_section("top", &cf_top, 0);
 215   cf_def_file = "ucw/conf-test.cf";
 216
 217   // Create and initialize dynamic arrays
 218   GARY_INIT(nrs1, 6);
 219   memcpy(nrs1, (int []) { 5, 5, 4, 3, 2, 1 }, 6 * sizeof(int));
 220   GARY_INIT(str2, 2);
 221   str2[0] = "Alice";
 222   str2[1] = "Bob";
 223   GARY_INIT(look, 2);
 224   look[0] = 2;
 225   look[1] = 1;
 226   GARY_INIT(sec1.list, 3);
 227   memcpy(sec1.list, (double []) { 1e4, -1e-4, 8 }, 3 * sizeof(double));
 228
 229   int opt;
 230   while ((opt = cf_getopt(argc, argv, short_opts, long_opts, NULL)) >= 0)
 231     switch (opt) {
 232       case 'r': reload++; break;
 233       case 'v': verbose++; break;
 234       default: usage("unknown option %c\n", opt);
 235     }
 236   if (optind < argc)
 237     usage("too many parameters (%d more)\n", argc-optind);
 238
 239   if (reload) {
 240     cf_reload(NULL);
 241     cf_reload(NULL);
 242   }
 243
 244   if (verbose) {
 245     struct fastbuf *out = bfdopen(1, 1<<14);
 246     cf_dump_sections(out);
 247     bclose(out);
 248   }
 249
 250   if (cc) {
 251     cf_switch_context(prev);
 252     cf_delete_context(cc);
 253   }
 254
 255   return 0;
 256 }