Indesc structure is now bound to scopes. Fixed issues with glob includes
incorrectly increase the inclusion depth value. The entire scope tree gets 
cleaned up at the end of parsing now to properly support displaying of 
error messages.

---
 include/nftables.h |  10 +++--
 include/parser.h   |  15 +++----
 include/rule.h     |  15 +++++--
 src/parser_bison.y |  64 ++++++++++++++++++++--------
 src/rule.c         |  56 +++++++++++++++++++++----
 src/scanner.l      | 120 +++++++++++++++++++++--------------------------------
 6 files changed, 164 insertions(+), 116 deletions(-)

diff --git a/include/nftables.h b/include/nftables.h
index 5e209b4..28d6043 100644
--- a/include/nftables.h
+++ b/include/nftables.h
@@ -118,13 +118,15 @@ struct input_descriptor {
        struct location                 location;
        enum input_descriptor_types     type;
        const char                      *name;
-       const char                      *data;
+       union {
+               const char              *data;
+               FILE                    *fp;
+       };
        unsigned int                    lineno;
        unsigned int                    column;
-       off_t                           token_offset;
-       off_t                           line_offset;
+       off_t                                   token_offset;
+       off_t                                   line_offset;
 };
-
 void ct_label_table_init(void);
 void mark_table_init(void);
 void gmp_init(void);
diff --git a/include/parser.h b/include/parser.h
index 1f3eb5f..eabb2f4 100644
--- a/include/parser.h
+++ b/include/parser.h
@@ -4,26 +4,19 @@
 #include <list.h>
 #include <rule.h> // FIXME
 
-#define MAX_INCLUDE_DEPTH              64
+#define MAX_INCLUDE_DEPTH              16
 #define TABSIZE                                8
 
 #define YYLTYPE                                struct location
 #define YYLTYPE_IS_TRIVIAL             0
 #define YYENABLE_NLS                   0
 
-#define SCOPE_NEST_MAX                 (MAX_INCLUDE_DEPTH + 3)
-
 struct parser_state {
-       struct input_descriptor         *indesc;
-       struct input_descriptor         indescs[MAX_INCLUDE_DEPTH];
-       unsigned int                    indesc_idx;
-
        struct list_head                *msgs;
        unsigned int                    nerrs;
 
-       struct scope                    top_scope;
-       struct scope                    *scopes[SCOPE_NEST_MAX];
-       unsigned int                    scope;
+       struct scope                    *top_scope; // top-level scope
+       struct scope                    *scope; // current scope
 
        struct list_head                *cmds;
        struct eval_ctx                 ectx;
@@ -37,7 +30,9 @@ extern void parser_destroy(struct parser_state *state);
 
 extern void open_scope(struct parser_state *state, struct scope *scope);
 extern void close_scope(struct parser_state *state);
+extern struct scope *toplevel_scope(const struct parser_state *state);
 extern struct scope *current_scope(const struct parser_state *state);
+extern struct input_descriptor* get_indesc(struct parser_state *state);
 
 extern int nft_parse(struct nft_ctx *ctx, void *, struct parser_state *state);
 
diff --git a/include/rule.h b/include/rule.h
index 8991025..c2ff692 100644
--- a/include/rule.h
+++ b/include/rule.h
@@ -82,12 +82,19 @@ extern void handle_free(struct handle *h);
  * @parent:    pointer to parent scope
  * @symbols:   symbols bound in the scope
  */
+struct input_descriptor;
 struct scope {
-       const struct scope      *parent;
+       struct list_head        list;
+       struct scope    *parent;
+       struct list_head                scopes;
+       struct input_descriptor *indesc;
+       int depth;
        struct list_head        symbols;
 };
 
-extern struct scope *scope_init(struct scope *scope, const struct scope 
*parent);
+extern struct input_descriptor* indesc_alloc(void);
+extern struct scope *scope_alloc(struct scope *parent, 
+                                                                struct 
input_descriptor *parent_indesc);
 extern void scope_release(const struct scope *scope);
 
 /**
@@ -137,7 +144,7 @@ struct table {
        struct list_head        list;
        struct handle           handle;
        struct location         location;
-       struct scope            scope;
+       struct scope            *scope;
        struct list_head        chains;
        struct list_head        sets;
        struct list_head        objs;
@@ -190,7 +197,7 @@ struct chain {
        int                     policy;
        const char              *type;
        const char              *dev;
-       struct scope            scope;
+       struct scope            *scope;
        struct list_head        rules;
 };
 
diff --git a/src/parser_bison.y b/src/parser_bison.y
index 4500aac..9982c5b 100755
--- a/src/parser_bison.y
+++ b/src/parser_bison.y
@@ -36,14 +36,23 @@
 
 #include "parser_bison.h"
 
+struct scope *toplevel_scope(const struct parser_state *state)
+{
+       return state->top_scope;
+}
+
+struct scope *current_scope(const struct parser_state *state)
+{
+       return state->scope;
+}
+
 void parser_init(struct nft_ctx *nft, struct parser_state *state,
                 struct list_head *msgs, struct list_head *cmds)
 {
        memset(state, 0, sizeof(*state));
-       init_list_head(&state->top_scope.symbols);
        state->msgs = msgs;
        state->cmds = cmds;
-       state->scopes[0] = scope_init(&state->top_scope, NULL);
+       state->top_scope = state->scope = scope_alloc(NULL, NULL);
        state->ectx.cache = &nft->cache;
        state->ectx.msgs = msgs;
        state->ectx.nf_sock = nft->nf_sock;
@@ -53,7 +62,7 @@ void parser_init(struct nft_ctx *nft, struct parser_state 
*state,
 
 void parser_destroy(struct parser_state *state)
 {
-       scope_release(state->scopes[0]);
+       scope_release(state->top_scope);
 }
 
 static void yyerror(struct location *loc, struct nft_ctx *nft, void *scanner,
@@ -62,35 +71,50 @@ static void yyerror(struct location *loc, struct nft_ctx 
*nft, void *scanner,
        erec_queue(error(loc, "%s", s), state->msgs);
 }
 
-static struct scope *toplevel_scope(const struct parser_state *state)
+void open_scope(struct parser_state *state, struct scope *scope)
 {
-       return state->scopes[0];
+       list_add_tail(&scope->list, &state->scope->scopes);
+       state->scope = scope;
 }
 
-struct scope *current_scope(const struct parser_state *state)
+void open_scope_glob(struct parser_state *state, struct scope *scope)
 {
-       return state->scopes[state->scope];
+       /* add the scope to the head of the current scope level list */
+       list_add(&scope->list, state->scope->list.prev);
+       state->scope = scope;
 }
 
-void open_scope(struct parser_state *state, struct scope *scope)
+void close_scope(struct parser_state *state)
 {
-       assert(state->scope < array_size(state->scopes) - 1);
-       scope_init(scope, current_scope(state));
-       state->scopes[++state->scope] = scope;
+       struct scope *scope = state->scope;
+       assert(state->scope->parent != NULL);
+
+       /* no additional globs, backtrack */
+       if (list_is_last(&scope->list, &scope->parent->scopes))
+               state->scope = scope->parent;
+       else
+               state->scope = list_entry(scope->list.next, struct scope, list);
 }
 
-void close_scope(struct parser_state *state)
+
+struct input_descriptor* get_indesc(struct parser_state *state)
 {
-       assert(state->scope > 0);
-       scope_release(state->scopes[state->scope]);
-       state->scope--;
+       struct scope *scope = state->scope;
+       while (scope) {
+               if (scope->indesc)
+                       return scope->indesc;
+               scope = scope->parent;
+       }
+       /* should never happen */
+       assert(false);
+       return NULL;
 }
 
 static void location_init(void *scanner, struct parser_state *state,
                          struct location *loc)
 {
        memset(loc, 0, sizeof(*loc));
-       loc->indesc = state->indesc;
+       loc->indesc = get_indesc(state);
 }
 
 static void location_update(struct location *loc, struct location *rhs, int n)
@@ -1407,8 +1431,10 @@ describe_cmd             :       primary_expr
 
 table_block_alloc      :       /* empty */
                        {
+                               struct scope *scope = 
scope_alloc(current_scope(state), NULL);
                                $$ = table_alloc();
-                               open_scope(state, &$$->scope);
+                               $$->scope->parent = current_scope(state);
+                               open_scope(state, scope);
                        }
                        ;
 
@@ -1518,8 +1544,10 @@ table_block              :       /* empty */     { $$ = 
$<table>-1; }
 
 chain_block_alloc      :       /* empty */
                        {
+                               struct scope *scope = 
scope_alloc(current_scope(state), NULL);
                                $$ = chain_alloc(NULL);
-                               open_scope(state, &$$->scope);
+                               $$->scope->parent = current_scope(state);
+                               open_scope(state, scope);
                        }
                        ;
 
diff --git a/src/rule.c b/src/rule.c
index d40ec50..aff6992 100644
--- a/src/rule.c
+++ b/src/rule.c
@@ -495,15 +495,42 @@ struct rule *rule_lookup(const struct chain *chain, 
uint64_t handle)
        return NULL;
 }
 
-struct scope *scope_init(struct scope *scope, const struct scope *parent)
+struct input_descriptor* indesc_alloc(void)
 {
+       struct input_descriptor* indesc = xzalloc(sizeof(struct 
input_descriptor));
+       return indesc;
+}
+
+struct scope *scope_alloc(struct scope *parent, 
+                                                 struct input_descriptor 
*parent_indesc)
+{
+       struct scope *scope = xzalloc(sizeof(*scope));
+       init_list_head(&scope->list);
+       init_list_head(&scope->symbols);
+       init_list_head(&scope->scopes);
        scope->parent = parent;
+       scope->indesc = parent_indesc;
+       if (parent != NULL)
+               scope->depth = parent->depth + 1;
        return scope;
 }
 
-void scope_release(const struct scope *scope)
+static void scope_indesc_release(const struct input_descriptor *indesc)
+{
+       if (indesc == NULL)
+               return;
+       xfree(indesc->name);
+       if (indesc->type == INDESC_FILE) {
+               fclose(indesc->fp);
+       }
+}
+
+static void single_scope_release(const struct scope *scope)
 {
        struct symbol *sym, *next;
+       
+       if (scope == NULL)
+               return;
 
        list_for_each_entry_safe(sym, next, &scope->symbols, list) {
                assert(sym->refcnt == 1);
@@ -512,6 +539,18 @@ void scope_release(const struct scope *scope)
                expr_free(sym->expr);
                xfree(sym);
        }
+       scope_indesc_release(scope->indesc);
+       xfree(scope->indesc);
+}
+
+void scope_release(const struct scope *scope)
+{
+       struct scope *other_scope, *next;
+       list_for_each_entry_safe(other_scope, next, &scope->scopes, list) {
+               scope_release(other_scope);
+       }
+       single_scope_release(scope);
+       xfree(scope);
 }
 
 void symbol_bind(struct scope *scope, const char *identifier, struct expr 
*expr)
@@ -635,10 +674,9 @@ struct chain *chain_alloc(const char *name)
        chain = xzalloc(sizeof(*chain));
        chain->refcnt = 1;
        init_list_head(&chain->rules);
-       init_list_head(&chain->scope.symbols);
+       chain->scope = scope_alloc(NULL, NULL);
        if (name != NULL)
-               chain->handle.chain.name = xstrdup(name);
-
+       chain->handle.chain.name = xstrdup(name);
        chain->policy = -1;
        return chain;
 }
@@ -658,7 +696,8 @@ void chain_free(struct chain *chain)
        list_for_each_entry_safe(rule, next, &chain->rules, list)
                rule_free(rule);
        handle_free(&chain->handle);
-       scope_release(&chain->scope);
+       list_del(&chain->scope->list);
+       scope_release(chain->scope);
        xfree(chain->type);
        if (chain->dev != NULL)
                xfree(chain->dev);
@@ -816,7 +855,7 @@ struct table *table_alloc(void)
        init_list_head(&table->sets);
        init_list_head(&table->objs);
        init_list_head(&table->flowtables);
-       init_list_head(&table->scope.symbols);
+       table->scope = scope_alloc(NULL, NULL);
        table->refcnt = 1;
 
        return table;
@@ -837,7 +876,8 @@ void table_free(struct table *table)
        list_for_each_entry_safe(obj, nobj, &table->objs, list)
                obj_free(obj);
        handle_free(&table->handle);
-       scope_release(&table->scope);
+       list_del(&table->scope->list);
+       scope_release(table->scope);
        xfree(table);
 }
 
diff --git a/src/scanner.l b/src/scanner.l
index 96bb307..e190811 100755
--- a/src/scanner.l
+++ b/src/scanner.l
@@ -56,41 +56,43 @@
        }                                                                       
\
 }
 
-static void scanner_pop_buffer(yyscan_t scanner);
-
-
 static void init_pos(struct parser_state *state)
 {
-       state->indesc->lineno           = 1;
-       state->indesc->column           = 1;
-       state->indesc->token_offset     = 0;
-       state->indesc->line_offset      = 0;
+       struct input_descriptor *indesc = get_indesc(state);
+/*     printf("initializing pos to %p, name %s\n", indesc, indesc->name); */
+       indesc->lineno          = 1;
+       indesc->column          = 1;
+       indesc->token_offset    = 0;
+       indesc->line_offset     = 0;
 }
 
 static void update_pos(struct parser_state *state, struct location *loc,
                       int len)
 {
-       loc->indesc                     = state->indesc;
-       loc->first_line                 = state->indesc->lineno;
-       loc->last_line                  = state->indesc->lineno;
-       loc->first_column               = state->indesc->column;
-       loc->last_column                = state->indesc->column + len - 1;
-       state->indesc->column           += len;
+       struct input_descriptor *indesc = get_indesc(state);
+       loc->indesc                     = indesc;
+       loc->first_line                 = indesc->lineno;
+       loc->last_line                  = indesc->lineno;
+       loc->first_column               = indesc->column;
+       loc->last_column                = indesc->column + len - 1;
+       indesc->column                  += len;
 }
 
 static void update_offset(struct parser_state *state, struct location *loc,
                          unsigned int len)
 {
-       state->indesc->token_offset     += len;
-       loc->token_offset               = state->indesc->token_offset;
-       loc->line_offset                = state->indesc->line_offset;
+       struct input_descriptor *indesc = get_indesc(state);
+       indesc->token_offset    += len;
+       loc->token_offset               = indesc->token_offset;
+       loc->line_offset                = indesc->line_offset;
 }
 
 static void reset_pos(struct parser_state *state, struct location *loc)
 {
-       state->indesc->line_offset      = state->indesc->token_offset;
-       state->indesc->lineno           += 1;
-       state->indesc->column           = 1;
+       struct input_descriptor *indesc = get_indesc(state);
+       indesc->line_offset     = indesc->token_offset;
+       indesc->lineno          += 1;
+       indesc->column          = 1;
 }
 
 #define YY_USER_ACTION {                                       \
@@ -620,7 +622,7 @@ addrstring  ({macaddr}|{ip4addr}|{ip6addr})
                                unsigned int diff;
 
                                diff = TABSIZE - strlen("\t");
-                               diff -= (state->indesc->column -
+                               diff -= (get_indesc(state)->column -
                                         strlen("\t") - 1) % TABSIZE;
 
                                update_pos(state, yylloc, diff);
@@ -631,15 +633,9 @@ addrstring ({macaddr}|{ip4addr}|{ip6addr})
 
 <<EOF>>                {
                                struct parser_state *state = 
yyget_extra(yyscanner);
-                               struct scope *scope = current_scope(state);
                                update_pos(state, yylloc, 1);
-                               /* only close scope if file was included, skip 
buffer parsing
-                                  since it does not create new scope */
-                               if (state->indesc && state->indesc->name) {
-                                       close_scope(state);
-                                       xfree(scope);
-                               }
-                               scanner_pop_buffer(yyscanner);
+                               close_scope(state);
+                               yypop_buffer_state(yyscanner);
                                if (YY_CURRENT_BUFFER == NULL)
                                        return TOKEN_EOF;
                        }
@@ -648,26 +644,15 @@ addrstring        ({macaddr}|{ip4addr}|{ip6addr})
 
 %%
 
-static void scanner_pop_buffer(yyscan_t scanner)
-{
-       struct parser_state *state = yyget_extra(scanner);
-
-       yypop_buffer_state(scanner);
-       if (state->indesc && state->indesc->name) {
-               xfree(state->indesc->name);
-               state->indesc->name = NULL;
-       }
-       state->indesc = &state->indescs[--state->indesc_idx - 1];
-}
-
 static struct error_record *scanner_push_file(void *scanner, const char 
*filename,
-                                             FILE *f, const struct location 
*loc)
+                                             FILE *f, const struct location 
*loc, struct scope *parent)
 {
        struct parser_state *state = yyget_extra(scanner);
-       struct scope *scope = NULL;
+       struct scope *scope = scope_alloc(parent, indesc_alloc());
        YY_BUFFER_STATE b;
 
-       if (state->indesc_idx == MAX_INCLUDE_DEPTH) {
+       if (scope->depth == MAX_INCLUDE_DEPTH) {
+               scope_release(scope);
                fclose(f);
                return error(loc, "Include nested too deeply, max %u levels",
                             MAX_INCLUDE_DEPTH);
@@ -676,20 +661,21 @@ static struct error_record *scanner_push_file(void 
*scanner, const char *filenam
        b = yy_create_buffer(f, YY_BUF_SIZE, scanner);
        yypush_buffer_state(b, scanner);
 
-       state->indesc = &state->indescs[state->indesc_idx++];
        if (loc != NULL)
-               state->indesc->location = *loc;
-       state->indesc->type     = INDESC_FILE;
-       state->indesc->name     = xstrdup(filename);
+               scope->indesc->location = *loc;
+       scope->indesc->type     = INDESC_FILE;
+       scope->indesc->name     = xstrdup(filename);
+       scope->indesc->fp       = f;
+       if (current_scope(state)->parent == parent)
+               open_scope_glob(state, scope);
+       else
+               open_scope(state, scope);
        init_pos(state);
-       scope = xzalloc(sizeof(*scope));
-       init_list_head(&scope->symbols);
-       open_scope(state, scope);
        return NULL;
 }
 
 static int include_file(void *scanner, const char *filename,
-                       const struct location *loc)
+                       const struct location *loc, struct scope *parent)
 {
        struct parser_state *state = yyget_extra(scanner);
        struct error_record *erec;
@@ -702,7 +688,7 @@ static int include_file(void *scanner, const char *filename,
                goto err;
        }
 
-       erec = scanner_push_file(scanner, filename, f, loc);
+       erec = scanner_push_file(scanner, filename, f, loc, parent);
        if (erec != NULL)
                goto err;
        return 0;
@@ -716,6 +702,7 @@ static int include_glob(void *scanner, const char *pattern,
 {
        struct parser_state *state = yyget_extra(scanner);
        struct error_record *erec = NULL;
+       struct scope *scope = current_scope(state);
        bool wildcard = false;
        glob_t glob_data;
        unsigned int i;
@@ -775,7 +762,7 @@ static int include_glob(void *scanner, const char *pattern,
                        if (len == 0 || path[len - 1] == '/')
                                continue;
 
-                       ret = include_file(scanner, path, loc);
+                       ret = include_file(scanner, path, loc, scope);
                        if (ret != 0)
                                goto err;
                }
@@ -812,7 +799,8 @@ err:
 int scanner_read_file(void *scanner, const char *filename,
                      const struct location *loc)
 {
-       return include_file(scanner, filename, loc);
+       struct parser_state *state = yyget_extra(scanner);
+       return include_file(scanner, filename, loc, current_scope(state));
 }
 
 static bool search_in_include_path(const char *filename)
@@ -878,12 +866,14 @@ void scanner_push_buffer(void *scanner, const struct 
input_descriptor *indesc,
                         const char *buffer)
 {
        struct parser_state *state = yyget_extra(scanner);
+       struct scope *scope = scope_alloc(current_scope(state), indesc_alloc());
        YY_BUFFER_STATE b;
 
-       state->indesc = &state->indescs[state->indesc_idx++];
-       memcpy(state->indesc, indesc, sizeof(*state->indesc));
-       state->indesc->data = buffer;
-       state->indesc->name = NULL;
+       open_scope(state, scope);
+       scope->indesc->type = indesc->type;
+       scope->indesc->name = xstrdup(indesc->name);
+       scope->indesc->data = buffer;
+       scope->indesc->name = NULL;
 
        b = yy_scan_string(buffer, scanner);
        assert(b != NULL);
@@ -894,8 +884,6 @@ void *scanner_init(struct parser_state *state)
 {
        yyscan_t scanner;
 
-       state->indesc = state->indescs;
-
        yylex_init_extra(state, &scanner);
        yyset_out(NULL, scanner);
 
@@ -904,17 +892,5 @@ void *scanner_init(struct parser_state *state)
 
 void scanner_destroy(void *scanner)
 {
-       struct parser_state *state = yyget_extra(scanner);
-
-       do {
-               struct input_descriptor *inpdesc =
-                       &state->indescs[state->indesc_idx];
-               if (inpdesc && inpdesc->name) {
-                       xfree(inpdesc->name);
-                       inpdesc->name = NULL;
-               }
-               yypop_buffer_state(scanner);
-       } while (state->indesc_idx--);
-
        yylex_destroy(scanner);
 }
-- 
2.13.6

--
To unsubscribe from this list: send the line "unsubscribe netfilter-devel" in
the body of a message to majord...@vger.kernel.org
More majordomo info at  http://vger.kernel.org/majordomo-info.html

Reply via email to