optimize: assert nat type on nat statement helper

commit 17a39cb0b082fe5117801d0b1a41407eec7b776c upstream.

Add assert() to helper function to expression from NAT statement.

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

diff --git a/src/optimize.c b/src/optimize.c
index e54693fdf67c4519aa97aafbf3ce60555982c01d..7fe0bc936382b07c94ae453ebad03dab14e0f065 100644 (file)
--- a/src/optimize.c
+++ b/src/optimize.c
@@ -853,6 +853,8 @@ static struct expr *stmt_nat_expr(struct stmt *nat_stmt)
 {
        struct expr *nat_expr;
 
+       assert(nat_stmt->ops->type == STMT_NAT);
+
        if (nat_stmt->nat.proto) {
                nat_expr = concat_expr_alloc(&internal_location);
                compound_expr_add(nat_expr, expr_get(nat_stmt->nat.addr));
@@ -863,6 +865,8 @@ static struct expr *stmt_nat_expr(struct stmt *nat_stmt)
                nat_expr = expr_get(nat_stmt->nat.addr);
        }
 
+       assert(nat_expr);
+
        return nat_expr;
 }