history.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 /*
  16  * This file contains the code for handling uci config history files
  17  */
  18
  19 #define _GNU_SOURCE
  20 #include <sys/types.h>
  21 #include <sys/stat.h>
  22 #include <sys/file.h>
  23 #include <stdbool.h>
  24 #include <unistd.h>
  25 #include <fcntl.h>
  26 #include <stdio.h>
  27 #include <ctype.h>
  28
  29 /* record a change that was done to a package */
  30 void
  31 uci_add_history(struct uci_context *ctx, struct uci_list *list, int cmd, const char *section, const char *option, const char *value)
  32 {
  33         struct uci_history *h;
  34         int size = strlen(section) + 1;
  35         char *ptr;
  36
  37         if (value)
  38                 size += strlen(value) + 1;
  39
  40         h = uci_alloc_element(ctx, history, option, size);
  41         ptr = uci_dataptr(h);
  42         h->cmd = cmd;
  43         h->section = strcpy(ptr, section);
  44         if (value) {
  45                 ptr += strlen(ptr) + 1;
  46                 h->value = strcpy(ptr, value);
  47         }
  48         uci_list_add(list, &h->e.list);
  49 }
  50
  51 void
  52 uci_free_history(struct uci_history *h)
  53 {
  54         if (!h)
  55                 return;
  56         if ((h->section != NULL) &&
  57                 (h->section != uci_dataptr(h))) {
  58                 free(h->section);
  59                 free(h->value);
  60         }
  61         uci_free_element(&h->e);
  62 }
  63
  64
  65 int uci_set_savedir(struct uci_context *ctx, const char *dir)
  66 {
  67         char *sdir;
  68
  69         UCI_HANDLE_ERR(ctx);
  70         UCI_ASSERT(ctx, dir != NULL);
  71
  72         sdir = uci_strdup(ctx, dir);
  73         if (ctx->savedir != uci_savedir)
  74                 free(ctx->savedir);
  75         ctx->savedir = sdir;
  76         return 0;
  77 }
  78
  79 int uci_add_history_path(struct uci_context *ctx, const char *dir)
  80 {
  81         struct uci_element *e;
  82
  83         UCI_HANDLE_ERR(ctx);
  84         UCI_ASSERT(ctx, dir != NULL);
  85         e = uci_alloc_generic(ctx, UCI_TYPE_PATH, dir, sizeof(struct uci_element));
  86         uci_list_add(&ctx->history_path, &e->list);
  87
  88         return 0;
  89 }
  90
  91 static inline int uci_parse_history_tuple(struct uci_context *ctx, char **buf, struct uci_ptr *ptr)
  92 {
  93         int c = UCI_CMD_CHANGE;
  94
  95         switch(**buf) {
  96         case '-':
  97                 c = UCI_CMD_REMOVE;
  98                 break;
  99         case '@':
 100                 c = UCI_CMD_RENAME;
 101                 break;
 102         case '+':
 103                 /* UCI_CMD_ADD is used for anonymous sections or list values */
 104                 c = UCI_CMD_ADD;
 105                 break;
 106         case '|':
 107                 c = UCI_CMD_LIST_ADD;
 108                 break;
 109         }
 110
 111         if (c != UCI_CMD_CHANGE)
 112                 *buf += 1;
 113
 114         UCI_INTERNAL(uci_parse_ptr, ctx, ptr, *buf);
 115
 116         if (!ptr->section)
 117                 goto error;
 118         if (ptr->flags & UCI_LOOKUP_EXTENDED)
 119                 goto error;
 120
 121         switch(c) {
 122         case UCI_CMD_RENAME:
 123                 if (!ptr->value || !uci_validate_name(ptr->value))
 124                         goto error;
 125                 break;
 126         case UCI_CMD_LIST_ADD:
 127                 if (!ptr->option)
 128                         goto error;
 129         }
 130
 131         return c;
 132
 133 error:
 134         UCI_THROW(ctx, UCI_ERR_INVAL);
 135         return 0;
 136 }
 137
 138 static void uci_parse_history_line(struct uci_context *ctx, struct uci_package *p, char *buf)
 139 {
 140         struct uci_element *e = NULL;
 141         struct uci_ptr ptr;
 142         int cmd;
 143
 144         cmd = uci_parse_history_tuple(ctx, &buf, &ptr);
 145         if (strcmp(ptr.package, p->e.name) != 0)
 146                 goto error;
 147
 148         if (ctx->flags & UCI_FLAG_SAVED_HISTORY)
 149                 uci_add_history(ctx, &p->saved_history, cmd, ptr.section, ptr.option, ptr.value);
 150
 151         switch(cmd) {
 152         case UCI_CMD_RENAME:
 153                 UCI_INTERNAL(uci_rename, ctx, &ptr);
 154                 break;
 155         case UCI_CMD_REMOVE:
 156                 UCI_INTERNAL(uci_delete, ctx, &ptr);
 157                 break;
 158         case UCI_CMD_LIST_ADD:
 159                 UCI_INTERNAL(uci_add_list, ctx, &ptr);
 160                 break;
 161         case UCI_CMD_ADD:
 162         case UCI_CMD_CHANGE:
 163                 UCI_INTERNAL(uci_set, ctx, &ptr);
 164                 e = ptr.last;
 165                 if (!ptr.option && e && (cmd == UCI_CMD_ADD))
 166                         uci_to_section(e)->anonymous = true;
 167                 break;
 168         }
 169         return;
 170 error:
 171         UCI_THROW(ctx, UCI_ERR_PARSE);
 172 }
 173
 174 /* returns the number of changes that were successfully parsed */
 175 static int uci_parse_history(struct uci_context *ctx, FILE *stream, struct uci_package *p)
 176 {
 177         struct uci_parse_context *pctx;
 178         int changes = 0;
 179
 180         /* make sure no memory from previous parse attempts is leaked */
 181         uci_cleanup(ctx);
 182
 183         pctx = (struct uci_parse_context *) uci_malloc(ctx, sizeof(struct uci_parse_context));
 184         ctx->pctx = pctx;
 185         pctx->file = stream;
 186
 187         while (!feof(pctx->file)) {
 188                 uci_getln(ctx, 0);
 189                 if (!pctx->buf[0])
 190                         continue;
 191
 192                 /*
 193                  * ignore parse errors in single lines, we want to preserve as much
 194                  * history as possible
 195                  */
 196                 UCI_TRAP_SAVE(ctx, error);
 197                 uci_parse_history_line(ctx, p, pctx->buf);
 198                 UCI_TRAP_RESTORE(ctx);
 199                 changes++;
 200 error:
 201                 continue;
 202         }
 203
 204         /* no error happened, we can get rid of the parser context now */
 205         uci_cleanup(ctx);
 206         return changes;
 207 }
 208
 209 /* returns the number of changes that were successfully parsed */
 210 static int uci_load_history_file(struct uci_context *ctx, struct uci_package *p, char *filename, FILE **f, bool flush)
 211 {
 212         FILE *stream = NULL;
 213         int changes = 0;
 214
 215         UCI_TRAP_SAVE(ctx, done);
 216         stream = uci_open_stream(ctx, filename, SEEK_SET, flush, false);
 217         if (p)
 218                 changes = uci_parse_history(ctx, stream, p);
 219         UCI_TRAP_RESTORE(ctx);
 220 done:
 221         if (f)
 222                 *f = stream;
 223         else if (stream)
 224                 uci_close_stream(stream);
 225         return changes;
 226 }
 227
 228 /* returns the number of changes that were successfully parsed */
 229 static int uci_load_history(struct uci_context *ctx, struct uci_package *p, bool flush)
 230 {
 231         struct uci_element *e;
 232         char *filename = NULL;
 233         FILE *f = NULL;
 234         int changes = 0;
 235
 236         if (!p->has_history)
 237                 return 0;
 238
 239         uci_foreach_element(&ctx->history_path, e) {
 240                 if ((asprintf(&filename, "%s/%s", e->name, p->e.name) < 0) || !filename)
 241                         UCI_THROW(ctx, UCI_ERR_MEM);
 242
 243                 uci_load_history_file(ctx, p, filename, NULL, false);
 244                 free(filename);
 245         }
 246
 247         if ((asprintf(&filename, "%s/%s", ctx->savedir, p->e.name) < 0) || !filename)
 248                 UCI_THROW(ctx, UCI_ERR_MEM);
 249
 250         changes = uci_load_history_file(ctx, p, filename, &f, flush);
 251         if (flush && f && (changes > 0)) {
 252                 rewind(f);
 253                 ftruncate(fileno(f), 0);
 254         }
 255         if (filename)
 256                 free(filename);
 257         uci_close_stream(f);
 258         ctx->err = 0;
 259         return changes;
 260 }
 261
 262 static void uci_filter_history(struct uci_context *ctx, const char *name, const char *section, const char *option)
 263 {
 264         struct uci_parse_context *pctx;
 265         struct uci_element *e, *tmp;
 266         struct uci_list list;
 267         char *filename = NULL;
 268         struct uci_ptr ptr;
 269         FILE *f = NULL;
 270
 271         uci_list_init(&list);
 272         uci_alloc_parse_context(ctx);
 273         pctx = ctx->pctx;
 274
 275         if ((asprintf(&filename, "%s/%s", ctx->savedir, name) < 0) || !filename)
 276                 UCI_THROW(ctx, UCI_ERR_MEM);
 277
 278         UCI_TRAP_SAVE(ctx, done);
 279         f = uci_open_stream(ctx, filename, SEEK_SET, true, false);
 280         pctx->file = f;
 281         while (!feof(f)) {
 282                 struct uci_element *e;
 283                 char *buf;
 284
 285                 uci_getln(ctx, 0);
 286                 buf = pctx->buf;
 287                 if (!buf[0])
 288                         continue;
 289
 290                 /* NB: need to allocate the element before the call to
 291                  * uci_parse_history_tuple, otherwise the original string
 292                  * gets modified before it is saved */
 293                 e = uci_alloc_generic(ctx, UCI_TYPE_HISTORY, pctx->buf, sizeof(struct uci_element));
 294                 uci_list_add(&list, &e->list);
 295
 296                 uci_parse_history_tuple(ctx, &buf, &ptr);
 297                 if (section) {
 298                         if (!ptr.section || (strcmp(section, ptr.section) != 0))
 299                                 continue;
 300                 }
 301                 if (option) {
 302                         if (!ptr.option || (strcmp(option, ptr.option) != 0))
 303                                 continue;
 304                 }
 305                 /* match, drop this element again */
 306                 uci_free_element(e);
 307         }
 308
 309         /* rebuild the history file */
 310         rewind(f);
 311         ftruncate(fileno(f), 0);
 312         uci_foreach_element_safe(&list, tmp, e) {
 313                 fprintf(f, "%s\n", e->name);
 314                 uci_free_element(e);
 315         }
 316         UCI_TRAP_RESTORE(ctx);
 317
 318 done:
 319         if (filename)
 320                 free(filename);
 321         uci_close_stream(pctx->file);
 322         uci_foreach_element_safe(&list, tmp, e) {
 323                 uci_free_element(e);
 324         }
 325         uci_cleanup(ctx);
 326 }
 327
 328 int uci_revert(struct uci_context *ctx, struct uci_ptr *ptr)
 329 {
 330         char *package = NULL;
 331         char *section = NULL;
 332         char *option = NULL;
 333
 334         UCI_HANDLE_ERR(ctx);
 335         expand_ptr(ctx, ptr, false);
 336         UCI_ASSERT(ctx, ptr->p->has_history);
 337
 338         /*
 339          * - flush unwritten changes
 340          * - save the package name
 341          * - unload the package
 342          * - filter the history
 343          * - reload the package
 344          */
 345         UCI_TRAP_SAVE(ctx, error);
 346         UCI_INTERNAL(uci_save, ctx, ptr->p);
 347
 348         /* NB: need to clone package, section and option names,
 349          * as they may get freed on uci_free_package() */
 350         package = uci_strdup(ctx, ptr->p->e.name);
 351         if (ptr->section)
 352                 section = uci_strdup(ctx, ptr->section);
 353         if (ptr->option)
 354                 option = uci_strdup(ctx, ptr->option);
 355
 356         uci_free_package(&ptr->p);
 357         uci_filter_history(ctx, package, section, option);
 358
 359         UCI_INTERNAL(uci_load, ctx, package, &ptr->p);
 360         UCI_TRAP_RESTORE(ctx);
 361         ctx->err = 0;
 362
 363 error:
 364         if (package)
 365                 free(package);
 366         if (section)
 367                 free(section);
 368         if (option)
 369                 free(option);
 370         if (ctx->err)
 371                 UCI_THROW(ctx, ctx->err);
 372         return 0;
 373 }
 374
 375 int uci_save(struct uci_context *ctx, struct uci_package *p)
 376 {
 377         FILE *f = NULL;
 378         char *filename = NULL;
 379         struct uci_element *e, *tmp;
 380         struct stat statbuf;
 381
 382         UCI_HANDLE_ERR(ctx);
 383         UCI_ASSERT(ctx, p != NULL);
 384
 385         /*
 386          * if the config file was outside of the /etc/config path,
 387          * don't save the history to a file, update the real file
 388          * directly.
 389          * does not modify the uci_package pointer
 390          */
 391         if (!p->has_history)
 392                 return uci_commit(ctx, &p, false);
 393
 394         if (uci_list_empty(&p->history))
 395                 return 0;
 396
 397         if (stat(ctx->savedir, &statbuf) < 0)
 398                 mkdir(ctx->savedir, UCI_DIRMODE);
 399         else if ((statbuf.st_mode & S_IFMT) != S_IFDIR)
 400                 UCI_THROW(ctx, UCI_ERR_IO);
 401
 402         if ((asprintf(&filename, "%s/%s", ctx->savedir, p->e.name) < 0) || !filename)
 403                 UCI_THROW(ctx, UCI_ERR_MEM);
 404
 405         ctx->err = 0;
 406         UCI_TRAP_SAVE(ctx, done);
 407         f = uci_open_stream(ctx, filename, SEEK_END, true, true);
 408         UCI_TRAP_RESTORE(ctx);
 409
 410         uci_foreach_element_safe(&p->history, tmp, e) {
 411                 struct uci_history *h = uci_to_history(e);
 412                 char *prefix = "";
 413
 414                 switch(h->cmd) {
 415                 case UCI_CMD_REMOVE:
 416                         prefix = "-";
 417                         break;
 418                 case UCI_CMD_RENAME:
 419                         prefix = "@";
 420                         break;
 421                 case UCI_CMD_ADD:
 422                         prefix = "+";
 423                         break;
 424                 case UCI_CMD_LIST_ADD:
 425                         prefix = "|";
 426                         break;
 427                 default:
 428                         break;
 429                 }
 430
 431                 fprintf(f, "%s%s.%s", prefix, p->e.name, h->section);
 432                 if (e->name)
 433                         fprintf(f, ".%s", e->name);
 434
 435                 if (h->cmd == UCI_CMD_REMOVE)
 436                         fprintf(f, "\n");
 437                 else
 438                         fprintf(f, "=%s\n", h->value);
 439                 uci_free_history(h);
 440         }
 441
 442 done:
 443         uci_close_stream(f);
 444         if (filename)
 445                 free(filename);
 446         if (ctx->err)
 447                 UCI_THROW(ctx, ctx->err);
 448
 449         return 0;
 450 }
 451
 452