fix double free bug
[project/uci.git] / file.c
diff --git a/file.c b/file.c
index 580ca18..54003c1 100644 (file)
--- a/file.c
+++ b/file.c
 #include <stdio.h>
 #include <ctype.h>
 
-#define LINEBUF        128
+#define LINEBUF        32
 #define LINEBUF_MAX    4096
 
 /*
  * Fetch a new line from the input stream and resize buffer if necessary
  */
-static void uci_getln(struct uci_context *ctx)
+static void uci_getln(struct uci_context *ctx, int offset)
 {
        struct uci_parse_context *pctx = ctx->pctx;
        char *p;
@@ -39,13 +39,13 @@ static void uci_getln(struct uci_context *ctx)
                pctx->bufsz = LINEBUF;
        }
 
-       ofs = 0;
+       ofs = offset;
        do {
                p = &pctx->buf[ofs];
                p[ofs] = 0;
 
                p = fgets(p, pctx->bufsz - ofs, pctx->file);
-               if (!p || !p[ofs])
+               if (!p || !*p)
                        return;
 
                ofs += strlen(p);
@@ -67,12 +67,18 @@ static void uci_getln(struct uci_context *ctx)
 }
 
 /*
- * Clean up all extra memory used by the parser
+ * Clean up all extra memory used by the parser and exporter
  */
-static void uci_parse_cleanup(struct uci_context *ctx)
+static void uci_file_cleanup(struct uci_context *ctx)
 {
        struct uci_parse_context *pctx;
 
+       if (ctx->buf) {
+               free(ctx->buf);
+               ctx->buf = NULL;
+               ctx->bufsz = 0;
+       }
+
        pctx = ctx->pctx;
        if (!pctx)
                return;
@@ -90,14 +96,41 @@ static void uci_parse_cleanup(struct uci_context *ctx)
        free(pctx);
 }
 
+/* 
+ * parse a character escaped by '\'
+ * returns true if the escaped character is to be parsed
+ * returns false if the escaped character is to be ignored
+ */
+static inline bool parse_backslash(struct uci_context *ctx, char **str)
+{
+       /* skip backslash */
+       *str += 1;
+
+       /* undecoded backslash at the end of line, fetch the next line */
+       if (!**str) {
+               *str += 1;
+               uci_getln(ctx, *str - ctx->pctx->buf);
+               return false;
+       }
+
+       /* FIXME: decode escaped char, necessary? */
+       return true;
+}
+
 /*
  * move the string pointer forward until a non-whitespace character or
  * EOL is reached
  */
-static void skip_whitespace(char **str)
+static void skip_whitespace(struct uci_context *ctx, char **str)
 {
+restart:
        while (**str && isspace(**str))
                *str += 1;
+
+       if (**str == '\\') {
+               if (!parse_backslash(ctx, str))
+                       goto restart;
+       }
 }
 
 static inline void addc(char **dest, char **src)
@@ -107,14 +140,6 @@ static inline void addc(char **dest, char **src)
        *src += 1;
 }
 
-static inline void parse_backslash(char **str, char **target)
-{
-       /* skip backslash */
-       *str += 1;
-       /* FIXME: decode escaped characters? */
-       addc(target, str);
-}
-
 /*
  * parse a double quoted string argument from the command line
  */
@@ -127,13 +152,14 @@ static void parse_double_quote(struct uci_context *ctx, char **str, char **targe
 
        while ((c = **str)) {
                switch(c) {
-               case '\\':
-                       parse_backslash(str, target);
-                       continue;
                case '"':
                        **target = 0;
                        *str += 1;
                        return;
+               case '\\':
+                       if (!parse_backslash(ctx, str))
+                               continue;
+                       /* fall through */
                default:
                        addc(target, str);
                        break;
@@ -175,9 +201,6 @@ static void parse_str(struct uci_context *ctx, char **str, char **target)
 {
        do {
                switch(**str) {
-               case '\\':
-                       parse_backslash(str, target);
-                       continue;
                case '\'':
                        parse_single_quote(ctx, str, target);
                        break;
@@ -186,6 +209,10 @@ static void parse_str(struct uci_context *ctx, char **str, char **target)
                        break;
                case 0:
                        goto done;
+               case '\\':
+                       if (!parse_backslash(ctx, str))
+                               continue;
+                       /* fall through */
                default:
                        addc(target, str);
                        break;
@@ -214,7 +241,7 @@ static char *next_arg(struct uci_context *ctx, char **str, bool required)
        char *ptr;
 
        val = ptr = *str;
-       skip_whitespace(str);
+       skip_whitespace(ctx, str);
        parse_str(ctx, str, &ptr);
        if (required && !*val) {
                ctx->pctx->reason = "insufficient arguments";
@@ -278,6 +305,30 @@ ignore:
 }
 
 /*
+ * parse the 'package' uci command (next config package)
+ */
+static void uci_parse_package(struct uci_context *ctx, char **str)
+{
+       char *name = NULL;
+
+       /* command string null-terminated by strtok */
+       *str += strlen(*str) + 1;
+
+       UCI_TRAP_SAVE(ctx, error);
+       name = next_arg(ctx, str, true);
+       assert_eol(ctx, str);
+       ctx->pctx->name = name;
+       uci_switch_config(ctx);
+       UCI_TRAP_RESTORE(ctx);
+       return;
+
+error:
+       if (name)
+               free(name);
+       UCI_THROW(ctx, ctx->errno);
+}
+
+/*
  * parse the 'config' uci command (open a section)
  */
 static void uci_parse_config(struct uci_context *ctx, char **str)
@@ -362,6 +413,10 @@ static void uci_parse_line(struct uci_context *ctx)
                word = strtok_r(word, " \t", &pbrk);
 
                switch(word[0]) {
+                       case 'p':
+                               if ((word[1] == 0) || !strcmp(word + 1, "ackage"))
+                                       uci_parse_package(ctx, &word);
+                               break;
                        case 'c':
                                if ((word[1] == 0) || !strcmp(word + 1, "onfig"))
                                        uci_parse_config(ctx, &word);
@@ -379,12 +434,92 @@ static void uci_parse_line(struct uci_context *ctx)
        }
 }
 
+/* max number of characters that escaping adds to the string */
+#define UCI_QUOTE_ESCAPE       "'\\'"
+
+/*
+ * escape an uci string for export
+ */
+static char *uci_escape(struct uci_context *ctx, char *str)
+{
+       char *s, *p, *t;
+       int pos = 0;
+
+       if (!ctx->buf) {
+               ctx->bufsz = LINEBUF;
+               ctx->buf = malloc(LINEBUF);
+       }
+
+       s = str;
+       p = strchr(str, '\'');
+       if (!p)
+               return str;
+
+       do {
+               int len = p - s;
+               if (len > 0) {
+                       if (p + sizeof(UCI_QUOTE_ESCAPE) - str >= ctx->bufsz) {
+                               ctx->bufsz *= 2;
+                               ctx->buf = realloc(ctx->buf, ctx->bufsz);
+                               if (!ctx->buf)
+                                       UCI_THROW(ctx, UCI_ERR_MEM);
+                       }
+                       memcpy(&ctx->buf[pos], s, len);
+                       pos += len;
+               }
+               strcpy(&ctx->buf[pos], UCI_QUOTE_ESCAPE);
+               pos += sizeof(UCI_QUOTE_ESCAPE);
+               s = p + 1;
+       } while ((p = strchr(s, '\'')));
+
+       return ctx->buf;
+}
+
+
+/*
+ * export a single config package to a file stream
+ */
+static void uci_export_config(struct uci_config *cfg, FILE *stream)
+{
+       struct uci_context *ctx = cfg->ctx;
+       struct uci_section *s;
+       struct uci_option *o;
+
+       fprintf(stream, "package '%s'\n", uci_escape(ctx, cfg->name));
+       uci_foreach_entry(section, &cfg->sections, s) {
+               fprintf(stream, "\nconfig '%s'", uci_escape(ctx, s->type));
+               fprintf(stream, " '%s'\n", uci_escape(ctx, s->name));
+               uci_foreach_entry(option, &s->options, o) {
+                       fprintf(stream, "\toption '%s'", uci_escape(ctx, o->name));
+                       fprintf(stream, " '%s'\n", uci_escape(ctx, o->value));
+               }
+       }
+       fprintf(stream, "\n");
+}
+
+int uci_export(struct uci_context *ctx, FILE *stream, struct uci_config *cfg)
+{
+       UCI_HANDLE_ERR(ctx);
+       UCI_ASSERT(ctx, stream != NULL);
+
+       if (cfg) {
+               uci_export_config(cfg, stream);
+               goto done;
+       }
+
+       uci_foreach_entry(config, &ctx->root, cfg) {
+               uci_export_config(cfg, stream);
+       }
+done:
+       return 0;
+}
+
 int uci_import(struct uci_context *ctx, FILE *stream, const char *name, struct uci_config **cfg)
 {
        struct uci_parse_context *pctx;
 
        /* make sure no memory from previous parse attempts is leaked */
-       uci_parse_cleanup(ctx);
+       uci_file_cleanup(ctx);
 
        pctx = (struct uci_parse_context *) uci_malloc(ctx, sizeof(struct uci_parse_context));
        ctx->pctx = pctx;
@@ -399,7 +534,7 @@ int uci_import(struct uci_context *ctx, FILE *stream, const char *name, struct u
                pctx->name = name;
 
        while (!feof(pctx->file)) {
-               uci_getln(ctx);
+               uci_getln(ctx, 0);
                if (pctx->buf[0])
                        uci_parse_line(ctx);
        }
@@ -411,7 +546,7 @@ int uci_import(struct uci_context *ctx, FILE *stream, const char *name, struct u
        uci_switch_config(ctx);
 
        /* no error happened, we can get rid of the parser context now */
-       uci_parse_cleanup(ctx);
+       uci_file_cleanup(ctx);
 
        return 0;
 }