util/res.c

   1 /* $Id: res.c,v 1.45 2005-09-13 11:51:11 adam Exp $
   2    Copyright (C) 1995-2005
   3    Index Data ApS
   4
   5 This file is part of the Zebra server.
   6
   7 Zebra is free software; you can redistribute it and/or modify it under
   8 the terms of the GNU General Public License as published by the Free
   9 Software Foundation; either version 2, or (at your option) any later
  10 version.
  11
  12 Zebra is distributed in the hope that it will be useful, but WITHOUT ANY
  13 WARRANTY; without even the implied warranty of MERCHANTABILITY or
  14 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  15 for more details.
  16
  17 You should have received a copy of the GNU General Public License
  18 along with Zebra; see the file LICENSE.zebra.  If not, write to the
  19 Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
  20 02111-1307, USA.
  21 */
  22
  23 #include <stdio.h>
  24 #include <stdlib.h>
  25 #include <string.h>
  26 #include <assert.h>
  27 #ifdef WIN32
  28 #include <io.h>
  29 #endif
  30 #if HAVE_UNISTD_H
  31 #include <unistd.h>
  32 #endif
  33
  34 #include <yaz/yaz-util.h>
  35 #include <idzebra/res.h>
  36
  37 struct res_entry {
  38     char *name;
  39     char *value;
  40     struct res_entry *next;
  41 };
  42
  43 struct res_struct {
  44     struct res_entry *first, *last;
  45     Res def_res;
  46     Res over_res;
  47 };
  48
  49 static struct res_entry *add_entry(Res r)
  50 {
  51     struct res_entry *resp;
  52
  53     if (!r->first)
  54         resp = r->last = r->first =
  55             (struct res_entry *) xmalloc(sizeof(*resp));
  56     else
  57     {
  58         resp = (struct res_entry *) xmalloc(sizeof(*resp));
  59         r->last->next = resp;
  60         r->last = resp;
  61     }
  62     resp->next = NULL;
  63     return resp;
  64 }
  65
  66 static char *xstrdup_env(const char *src)
  67 {
  68     int i = 0;
  69     int j = 0;
  70     char *dst;
  71     int env_strlen = 0;
  72
  73     while (src[i])
  74     {
  75         if (src[i] == '$' && src[i+1] == '{')
  76         {
  77             char envname[128];
  78             char *env_val;
  79             int k = 0;
  80             i = i + 2;
  81             while (k < 127 && src[i] && !strchr(":}\n\r\f", src[i]))
  82                 envname[k++] = src[i++];
  83             envname[k] = '\0';
  84
  85             env_val = getenv(envname);
  86             if (env_val)
  87                 env_strlen += 1 + strlen(env_val);
  88             else
  89                 env_strlen++;
  90             while (src[i] && !strchr("}\n\r\f", src[i]))
  91                 i++;
  92             if (src[i] == '}')
  93                 i++;
  94         }
  95         else
  96             i++;
  97     }
  98     dst = xmalloc(1 + env_strlen + i);
  99     i = 0;
 100     while (src[i])
 101     {
 102         if (src[i] == '$' && src[i+1] == '{')
 103         {
 104             char envname[128];
 105             char *env_val;
 106             int k = 0;
 107             i = i + 2;
 108             while(k < 127 && src[i] && !strchr(":}\n\r\f", src[i]))
 109                 envname[k++] = src[i++];
 110             envname[k] = '\0';
 111             env_val = getenv(envname);
 112             if (env_val)
 113             {
 114                 strcpy(dst+j, env_val);
 115                 j += strlen(env_val);
 116             }
 117             else if (src[i] == ':' && src[i+1] == '-')
 118             {
 119                 i = i + 2;
 120                 while (src[i] && !strchr("}\n\r\f", src[i]))
 121                     dst[j++] = src[i++];
 122             }
 123             while (src[i] && !strchr("}\n\r\f", src[i]))
 124                 i++;
 125             if (src[i] == '}')
 126                 i++;
 127         }
 128         else
 129             dst[j++] = src[i++];
 130     }
 131     dst[j] = '\0';
 132     return dst;
 133 }
 134
 135 ZEBRA_RES res_read_file(Res r, const char *fname)
 136 {
 137     struct res_entry *resp;
 138     char *line;
 139     char *val_buf;
 140     int val_size, val_max = 256;
 141     char fr_buf[1024];
 142     FILE *fr;
 143
 144     assert(r);
 145
 146     fr = fopen(fname, "r");
 147     if (!fr)
 148     {
 149         yaz_log(YLOG_WARN|YLOG_ERRNO, "Cannot open `%s'", fname);
 150         return ZEBRA_FAIL;
 151     }
 152     val_buf = (char*) xmalloc(val_max);
 153     while (1)
 154     {
 155         line = fgets(fr_buf, sizeof(fr_buf)-1, fr);
 156         if (!line)
 157             break;
 158         if (*line != '#')
 159         {
 160             int no = 0;
 161             while (1)
 162             {
 163                 if (fr_buf[no] == 0 || fr_buf[no] == '\n' )
 164                 {
 165                     no = 0;
 166                     break;
 167                 }
 168                 if (strchr(": \t", fr_buf[no]))
 169                     break;
 170                 no++;
 171             }
 172             if (!no)
 173                 continue;
 174             fr_buf[no++] = '\0';
 175             resp = add_entry(r);
 176             resp->name = (char*) xmalloc(no);
 177             strcpy(resp->name, fr_buf);
 178
 179             while (strchr(" \t", fr_buf[no]))
 180                 no++;
 181             val_size = 0;
 182             while (1)
 183             {
 184                 if (fr_buf[no] == '\0' || strchr("\n\r\f", fr_buf[no]))
 185                 {
 186                     while (val_size > 0 &&
 187                               (val_buf[val_size-1] == ' ' ||
 188                                val_buf[val_size-1] == '\t'))
 189                         val_size--;
 190                     val_buf[val_size] = '\0';
 191                     resp->value = xstrdup_env(val_buf);
 192                     yaz_log(YLOG_DEBUG, "(name=%s,value=%s)",
 193                          resp->name, resp->value);
 194                     break;
 195                 }
 196                 else if (fr_buf[no] == '\\' && strchr("\n\r\f", fr_buf[no+1]))
 197                 {
 198                     line = fgets(fr_buf, sizeof(fr_buf)-1, fr);
 199                     if (!line)
 200                     {
 201                         val_buf[val_size] = '\0';
 202                         resp->value = xstrdup_env(val_buf);
 203                         break;
 204                     }
 205                     no = 0;
 206                 }
 207                 else
 208                 {
 209                     val_buf[val_size++] = fr_buf[no++];
 210                     if (val_size+1 >= val_max)
 211                     {
 212                         char *nb;
 213
 214                         nb = (char*) xmalloc(val_max+=1024);
 215                         memcpy(nb, val_buf, val_size);
 216                         xfree(val_buf);
 217                         val_buf = nb;
 218                     }
 219                 }
 220             }
 221         }
 222     }
 223     xfree(val_buf);
 224     fclose(fr);
 225     return ZEBRA_OK;
 226 }
 227
 228 Res res_open(Res def_res, Res over_res)
 229 {
 230     Res r;
 231
 232     r = (Res) xmalloc(sizeof(*r));
 233     r->first = r->last = NULL;
 234     r->def_res = def_res;
 235     r->over_res = over_res;
 236     return r;
 237 }
 238
 239 void res_clear(Res r)
 240 {
 241     struct res_entry *re, *re1;
 242     for (re = r->first; re; re=re1)
 243     {
 244         if (re->name)
 245             xfree(re->name);
 246         if (re->value)
 247             xfree(re->value);
 248         re1 = re->next;
 249         xfree(re);
 250     }
 251     r->first = r->last = NULL;
 252 }
 253
 254 void res_close(Res r)
 255 {
 256     if (!r)
 257         return;
 258     res_clear(r);
 259
 260     xfree(r);
 261 }
 262
 263 const char *res_get_prefix(Res r, const char *name, const char *prefix,
 264                             const char *def)
 265 {
 266     const char *v = 0;;
 267     if (prefix)
 268     {
 269         char rname[128];
 270
 271         if (strlen(name) + strlen(prefix) >= (sizeof(rname)-2))
 272             return 0;
 273         strcpy(rname, prefix);
 274         strcat(rname, ".");
 275         strcat(rname, name);
 276         v = res_get(r, rname);
 277     }
 278     if (!v)
 279         v = res_get(r, name);
 280     if (!v)
 281         v = def;
 282     return v;
 283 }
 284
 285 const char *res_get(Res r, const char *name)
 286 {
 287     struct res_entry *re;
 288     const char *v;
 289
 290     if (!r)
 291         return 0;
 292
 293     v = res_get(r->over_res, name);
 294     if (v)
 295         return v;
 296
 297     for (re = r->first; re; re=re->next)
 298         if (re->value && !yaz_matchstr(re->name, name))
 299             return re->value;
 300
 301     return res_get(r->def_res, name);
 302 }
 303
 304 const char *res_get_def(Res r, const char *name, const char *def)
 305 {
 306     const char *t;
 307
 308     if (!(t = res_get(r, name)))
 309     {
 310         yaz_log(YLOG_DEBUG, "CAUTION: Using default resource %s:%s", name, def);
 311         return def;
 312     }
 313     else
 314         return t;
 315 }
 316
 317 int res_get_match(Res r, const char *name, const char *value, const char *s)
 318 {
 319     const char *cn = res_get(r, name);
 320
 321     if (!cn)
 322         cn = s;
 323     if (cn && !yaz_matchstr(cn, value))
 324         return 1;
 325     return 0;
 326 }
 327
 328 void res_set(Res r, const char *name, const char *value)
 329 {
 330     struct res_entry *re;
 331     assert(r);
 332
 333     for (re = r->first; re; re=re->next)
 334         if (re->value && !yaz_matchstr(re->name, name))
 335         {
 336             xfree(re->value);
 337             re->value = xstrdup_env(value);
 338             return;
 339         }
 340     re = add_entry(r);
 341     re->name = xstrdup(name);
 342     re->value = xstrdup_env(value);
 343 }
 344
 345 int res_trav(Res r, const char *prefix, void *p,
 346               void (*f)(void *p, const char *name, const char *value))
 347 {
 348     struct res_entry *re;
 349     int l = 0;
 350     int no = 0;
 351
 352     if (!r)
 353         return 0;
 354     no = res_trav(r->over_res, prefix, p, f);
 355     if (no)
 356         return no;
 357     if (prefix)
 358         l = strlen(prefix);
 359     for (re = r->first; re; re=re->next)
 360         if (re->value)
 361             if (l==0 || !memcmp(re->name, prefix, l))
 362             {
 363                 (*f)(p, re->name, re->value);
 364                 no++;
 365             }
 366     if (!no)
 367         return res_trav(r->def_res, prefix, p, f);
 368     return no;
 369 }
 370
 371
 372 ZEBRA_RES res_write_file(Res r, const char *fname)
 373 {
 374     struct res_entry *re;
 375     FILE *fr;
 376
 377     assert(r);
 378     fr = fopen(fname, "w");
 379     if (!fr)
 380     {
 381         yaz_log(YLOG_FATAL|YLOG_ERRNO, "Cannot create `%s'", fname);
 382         exit(1);
 383     }
 384
 385     for (re = r->first; re; re=re->next)
 386     {
 387         int no = 0;
 388         int lefts = strlen(re->name)+2;
 389
 390         if (!re->value)
 391             fprintf(fr, "%s\n", re->name);
 392         else
 393         {
 394             fprintf(fr, "%s: ", re->name);
 395             while (lefts + strlen(re->value+no) > 78)
 396             {
 397                 int i = 20;
 398                 int ind = no+ 78-lefts;
 399                 while (--i >= 0)
 400                 {
 401                     if (re->value[ind] == ' ')
 402                         break;
 403                     --ind;
 404                 }
 405                 if (i<0)
 406                     ind = no + 78 - lefts;
 407                 for (i = no; i != ind; i++)
 408                     putc(re->value[i], fr);
 409                 fprintf(fr, "\\\n");
 410                 no=ind;
 411                 lefts = 0;
 412             }
 413             fprintf(fr, "%s\n", re->value+no);
 414         }
 415     }
 416     fclose(fr);
 417     return ZEBRA_OK;
 418 }
 419
 420 ZEBRA_RES res_get_int(Res r, const char *name, int *val)
 421 {
 422     const char *cp = res_get(r, name);
 423     if (cp)
 424     {
 425         if (sscanf(cp, "%d", val) == 1)
 426             return ZEBRA_OK;
 427         yaz_log(YLOG_WARN, "Expected integer for resource %s", name);
 428     }
 429     return ZEBRA_FAIL;
 430 }