]> git.ipfire.org Git - thirdparty/nftables.git/commitdiff
evaluate: add object to the cache
authorPablo Neira Ayuso <pablo@netfilter.org>
Thu, 15 Apr 2021 12:00:26 +0000 (14:00 +0200)
committerPablo Neira Ayuso <pablo@netfilter.org>
Sun, 2 May 2021 21:30:35 +0000 (23:30 +0200)
If the cache does not contain this object that is defined in this batch,
add it to the cache. This allows for references to this new object in
the same batch.

This patch also adds missing handle_merge() to set the object name,
otherwise object name is NULL and obj_cache_find() crashes.

Signed-off-by: Pablo Neira Ayuso <pablo@netfilter.org>
src/evaluate.c

index f4c1acef0b167eaa125629a6c04fcce768e2958c..c5adf2cab7b5efc078be57fd18da4e47801278c2 100644 (file)
@@ -4246,6 +4246,15 @@ static int ct_timeout_evaluate(struct eval_ctx *ctx, struct obj *obj)
 
 static int obj_evaluate(struct eval_ctx *ctx, struct obj *obj)
 {
+       struct table *table;
+
+       table = table_lookup_global(ctx);
+       if (!table)
+               return table_not_found(ctx);
+
+       if (!obj_cache_find(table, obj->handle.obj.name, obj->type))
+               obj_cache_add(obj_get(obj), table);
+
        switch (obj->type) {
        case NFT_OBJECT_CT_TIMEOUT:
                return ct_timeout_evaluate(ctx, obj);
@@ -4332,6 +4341,7 @@ static int cmd_evaluate_add(struct eval_ctx *ctx, struct cmd *cmd)
        case CMD_OBJ_SECMARK:
        case CMD_OBJ_CT_EXPECT:
        case CMD_OBJ_SYNPROXY:
+               handle_merge(&cmd->object->handle, &cmd->handle);
                return obj_evaluate(ctx, cmd->object);
        default:
                BUG("invalid command object type %u\n", cmd->obj);