src/config.c

   1 /* Copyright (C) 2020 SovietPony
   2  *
   3  * This program is free software: you can redistribute it and/or modify
   4  * it under the terms of the GNU General Public License as published by
   5  * the Free Software Foundation, version 3 of the License ONLY.
   6  *
   7  * This program is distributed in the hope that it will be useful,
   8  * but WITHOUT ANY WARRANTY; without even the implied warranty of
   9  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  10  * GNU General Public License for more details.
  11  *
  12  * You should have received a copy of the GNU General Public License
  13  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  14  */
  15
  16 #include "glob.h"
  17 #include <stdio.h>
  18 #include <stdlib.h>
  19 #include <string.h>
  20 #include <assert.h>
  21 #include <ctype.h>
  22 #include "system.h"
  23 #include "files.h"
  24 #include "input.h"
  25 #include "cp866.h"
  26
  27 static FILE *f;
  28 static int ch;
  29
  30 const cfg_t *CFG_find_entry (const char *key, const cfg_t *cfg) {
  31   assert(key != NULL);
  32   if (cfg != NULL) {
  33     int i = 0;
  34     while (cfg[i].cfg && cp866_strcasecmp(cfg[i].cfg, key) != 0) {
  35       i++;
  36     }
  37     return cfg[i].cfg ? &cfg[i] : NULL;
  38   } else {
  39     return NULL;
  40   }
  41 }
  42
  43 int CFG_update_key (const char *key, const char *value, const cfg_t *cfg) {
  44   assert(key != NULL);
  45   assert(value != NULL);
  46   const cfg_t *entry = CFG_find_entry(key, cfg);
  47   if (entry != NULL) {
  48     void *p = entry->p;
  49     switch (entry->t) {
  50       case Y_BYTE: *(byte*)p = atoi(value); break;
  51       case Y_WORD: *(word*)p = atoi(value); break;
  52       case Y_DWORD: *(dword*)p = atoi(value); break;
  53       case Y_STRING: strcpy(p, value); break; // TODO fix this security problem
  54       case Y_SW_ON: *(byte*)p = cp866_strcasecmp(value, "on") == 0 ? 1 : 0; break;
  55       case Y_SW_OFF: *(byte*)p = cp866_strcasecmp(value, "off") == 0 ? 1 : 0; break;
  56       case Y_FILES: F_addwad(value); break;
  57       case Y_KEY: *(int*)p = I_string_to_key(value); break;
  58       default: assert(0); // unknown type -> something broken
  59     }
  60     //logo("CFG_update_key: [%s] = [%s]\n", key, value);
  61     return 1;
  62   } else {
  63     return 0;
  64   }
  65 }
  66
  67 /* --- parser --- */
  68
  69 int CFG_open_iterator (const char *name) {
  70   assert(f == NULL);
  71   f = fopen(name, "rb");
  72   if (f != NULL) {
  73     ch = fgetc(f);
  74   }
  75   return f != NULL;
  76 }
  77
  78 static void CFG_skip_space (void) {
  79   while (feof(f) == 0 && isspace(ch)) {
  80     ch = fgetc(f);
  81   }
  82 }
  83
  84 static void CFG_skip_line (void) {
  85   while (feof(f) == 0 && ch != '\n' && ch != '\r') {
  86     ch = fgetc(f);
  87   }
  88   while (feof(f) == 0 && ch == '\n' && ch == '\r') {
  89     ch = fgetc(f);
  90   }
  91 }
  92
  93 int CFG_scan_iterator (char *key, int keylen, char *value, int valuelen) {
  94   assert(key != NULL);
  95   assert(keylen > 0);
  96   assert(value != NULL);
  97   assert(valuelen > 0);
  98   int i;
  99   int found = 0;
 100   while (feof(f) == 0 && found == 0) {
 101     CFG_skip_space();
 102     if (ch == ';') {
 103       CFG_skip_line();
 104     } else if (feof(f) == 0) {
 105       found = 1;
 106       i = 0;
 107       while (feof(f) == 0 && isspace(ch) == 0 && ch != '=') {
 108         if (i < keylen - 1) {
 109           key[i] = ch;
 110           i += 1;
 111         }
 112         ch = fgetc(f);
 113       }
 114       key[i] = 0;
 115       CFG_skip_space();
 116       if (feof(f) == 0 && ch == '=') {
 117         ch = fgetc(f);
 118         CFG_skip_space();
 119       }
 120       i = 0;
 121       while (feof(f) == 0 && ch != '\n' && ch != '\r') {
 122         if (i < valuelen - 1) {
 123           value[i] = ch;
 124           i += 1;
 125         }
 126         ch = fgetc(f);
 127       }
 128       value[i] = 0;
 129       CFG_skip_line();
 130     }
 131   }
 132   return found;
 133 }
 134
 135 void CFG_close_iterator (void) {
 136   assert(f != NULL);
 137   fclose(f);
 138   f = NULL;
 139 }
 140
 141 /* --- reader --- */
 142
 143 int CFG_read_config (const char *name, int n, const cfg_t **cfg) {
 144   assert(name != NULL);
 145   assert(n >= 0);
 146   assert(cfg != NULL);
 147   int i;
 148   char key[64];
 149   char value[64];
 150   assert(name != NULL);
 151   if (CFG_open_iterator(name)) {
 152     while (CFG_scan_iterator(key, 64, value, 64)) {
 153       i = 0;
 154       while (i < n && CFG_update_key(key, value, cfg[i]) == 0) {
 155         i++;
 156       }
 157     }
 158     CFG_close_iterator();
 159     return 1;
 160   } else {
 161     return 0;
 162   }
 163 }
 164
 165
 166
 167 /* --- writer --- */
 168
 169 static void CFG_write_key_value (FILE *f, const char *key, const char *value) {
 170   assert(f != NULL);
 171   assert(key != NULL);
 172   assert(value != NULL);
 173   fwrite(key, strlen(key), 1, f);
 174   fwrite("=", 1, 1, f);
 175   fwrite(value, strlen(value), 1, f);
 176   fwrite("\n", 1, 1, f);
 177 }
 178
 179 static int CFG_write_entry (FILE *f, const cfg_t *entry) {
 180   assert(f != NULL);
 181   assert(entry != NULL);
 182   char buf[16];
 183   const char *str;
 184   const char *key = entry->cfg;
 185   if (key != NULL) {
 186     switch (entry->t) {
 187       case Y_BYTE:
 188         snprintf(buf, 16, "%i", *(byte*)entry->p);
 189         CFG_write_key_value(f, key, buf);
 190         break;
 191       case Y_WORD:
 192         snprintf(buf, 16, "%i", *(word*)entry->p);
 193         CFG_write_key_value(f, key, buf);
 194         break;
 195       case Y_DWORD:
 196         snprintf(buf, 16, "%i", *(dword*)entry->p);
 197         CFG_write_key_value(f, key, buf);
 198         break;
 199       case Y_STRING:
 200         CFG_write_key_value(f, key, entry->p);
 201         break;
 202       case Y_SW_ON:
 203       case Y_SW_OFF:
 204         str = *(byte*)entry->p ? "on" : "off";
 205         CFG_write_key_value(f, key, str);
 206         break;
 207       case Y_KEY:
 208         str = I_key_to_string(*(int*)entry->p);
 209         CFG_write_key_value(f, key, str);
 210         break;
 211       case Y_FILES: return 1; // ignore
 212       case 0: return 0; // end
 213       default: assert(0); // unknown type -> something broken
 214     }
 215   }
 216   return entry->t == 0 ? 0 : 1;
 217 }
 218
 219 int CFG_update_config (const char *old, const char *new, int n, const cfg_t **cfg, const char *msg) {
 220   assert(old != NULL);
 221   assert(new != NULL);
 222   assert(n >= 0);
 223   assert(cfg != NULL);
 224   int i, j;
 225   char key[64];
 226   char value[64];
 227   FILE *nf = fopen(new, "wb");
 228   if (nf != NULL) {
 229     if (msg != NULL) {
 230       fwrite("; ", 2, 1, nf);
 231       fwrite(msg, strlen(msg), 1, nf);
 232       fwrite("\n", 1, 1, nf);
 233     }
 234     if (CFG_open_iterator(old)) {
 235       while (CFG_scan_iterator(key, 64, value, 64)) {
 236         i = 0;
 237         while (i < n && CFG_find_entry(key, cfg[i]) == NULL) {
 238           i++;
 239         }
 240         if (i >= n) {
 241           CFG_write_key_value(nf, key, value);
 242         }
 243       }
 244       CFG_close_iterator();
 245     }
 246     for (j = 0; j < n; j++) {
 247       if (cfg[j] != NULL) {
 248         i = 0;
 249         while (CFG_write_entry(nf, &cfg[j][i])) {
 250           i++;
 251         }
 252       }
 253     }
 254     fclose(nf);
 255   }
 256   return nf != NULL;
 257 }