optimize: wrap code to build concatenation in helper function

Move code to build concatenations into helper function, this routine
includes support for expansion of implicit sets containing singleton
values. This is preparation work to reuse existing code in a follow up
patch.

Signed-off-by: Pablo Neira Ayuso <pablo@netfilter.org>

diff --git a/src/optimize.c b/src/optimize.c
index 289c442dc915eb833f0347e154c10a8242d3d6f5..8cec04c02c4fba70f5039559b5a46691431f26be 100644 (file)
--- a/src/optimize.c
+++ b/src/optimize.c
@@ -552,20 +552,19 @@ static void merge_stmts(const struct optimize_ctx *ctx,
        }
 }
 
-static void __merge_concat_stmts(const struct optimize_ctx *ctx, uint32_t i,
-                                const struct merge *merge, struct expr *set)
+static void __merge_concat(const struct optimize_ctx *ctx, uint32_t i,
+                          const struct merge *merge, struct list_head *concat_list)
 {
-       struct expr *concat, *next, *expr, *concat_clone, *clone, *elem;
+       struct expr *concat, *next, *expr, *concat_clone, *clone;
        LIST_HEAD(pending_list);
-       LIST_HEAD(concat_list);
        struct stmt *stmt_a;
        uint32_t k;
 
        concat = concat_expr_alloc(&internal_location);
-       list_add(&concat->list, &concat_list);
+       list_add(&concat->list, concat_list);
 
        for (k = 0; k < merge->num_stmts; k++) {
-               list_for_each_entry_safe(concat, next, &concat_list, list) {
+               list_for_each_entry_safe(concat, next, concat_list, list) {
                        stmt_a = ctx->stmt_matrix[i][merge->stmt[k]];
                        switch (stmt_a->expr->right->etype) {
                        case EXPR_SET:
@@ -590,8 +589,17 @@ static void __merge_concat_stmts(const struct optimize_ctx *ctx, uint32_t i,
                                break;
                        }
                }
-               list_splice_init(&pending_list, &concat_list);
+               list_splice_init(&pending_list, concat_list);
        }
+}
+
+static void __merge_concat_stmts(const struct optimize_ctx *ctx, uint32_t i,
+                                const struct merge *merge, struct expr *set)
+{
+       struct expr *concat, *next, *elem;
+       LIST_HEAD(concat_list);
+
+       __merge_concat(ctx, i, merge, &concat_list);
 
        list_for_each_entry_safe(concat, next, &concat_list, list) {
                list_del(&concat->list);