int nft_expr_inner_parse(const struct nft_ctx *ctx, const struct nlattr *nla,
                         struct nft_expr_info *info);
-int nft_expr_clone(struct nft_expr *dst, struct nft_expr *src);
+int nft_expr_clone(struct nft_expr *dst, struct nft_expr *src, gfp_t gfp);
 void nft_expr_destroy(const struct nft_ctx *ctx, struct nft_expr *expr);
 int nft_expr_dump(struct sk_buff *skb, unsigned int attr,
                  const struct nft_expr *expr, bool reset);
                                                struct nft_regs *regs,
                                                const struct nft_pktinfo *pkt);
        int                             (*clone)(struct nft_expr *dst,
-                                                const struct nft_expr *src);
+                                                const struct nft_expr *src, gfp_t gfp);
        unsigned int                    size;
 
        int                             (*init)(const struct nft_ctx *ctx,
 
        return ERR_PTR(err);
 }
 
-int nft_expr_clone(struct nft_expr *dst, struct nft_expr *src)
+int nft_expr_clone(struct nft_expr *dst, struct nft_expr *src, gfp_t gfp)
 {
        int err;
 
                return -EINVAL;
 
        dst->ops = src->ops;
-       err = src->ops->clone(dst, src);
+       err = src->ops->clone(dst, src, gfp);
        if (err < 0)
                return err;
 
                if (!expr)
                        goto err_expr;
 
-               err = nft_expr_clone(expr, set->exprs[i]);
+               err = nft_expr_clone(expr, set->exprs[i], GFP_KERNEL_ACCOUNT);
                if (err < 0) {
                        kfree(expr);
                        goto err_expr;
 
        for (i = 0; i < num_exprs; i++) {
                expr = nft_setelem_expr_at(elem_expr, elem_expr->size);
-               err = nft_expr_clone(expr, expr_array[i]);
+               err = nft_expr_clone(expr, expr_array[i], GFP_KERNEL_ACCOUNT);
                if (err < 0)
                        goto err_elem_expr_setup;
 
 
        nft_connlimit_do_destroy(ctx, priv);
 }
 
-static int nft_connlimit_clone(struct nft_expr *dst, const struct nft_expr *src)
+static int nft_connlimit_clone(struct nft_expr *dst, const struct nft_expr *src, gfp_t gfp)
 {
        struct nft_connlimit *priv_dst = nft_expr_priv(dst);
        struct nft_connlimit *priv_src = nft_expr_priv(src);
 
-       priv_dst->list = kmalloc(sizeof(*priv_dst->list), GFP_ATOMIC);
+       priv_dst->list = kmalloc(sizeof(*priv_dst->list), gfp);
        if (!priv_dst->list)
                return -ENOMEM;
 
 
        nft_counter_do_destroy(priv);
 }
 
-static int nft_counter_clone(struct nft_expr *dst, const struct nft_expr *src)
+static int nft_counter_clone(struct nft_expr *dst, const struct nft_expr *src, gfp_t gfp)
 {
        struct nft_counter_percpu_priv *priv = nft_expr_priv(src);
        struct nft_counter_percpu_priv *priv_clone = nft_expr_priv(dst);
 
        nft_counter_fetch(priv, &total);
 
-       cpu_stats = alloc_percpu_gfp(struct nft_counter, GFP_ATOMIC);
+       cpu_stats = alloc_percpu_gfp(struct nft_counter, gfp);
        if (cpu_stats == NULL)
                return -ENOMEM;
 
 
 
        for (i = 0; i < priv->num_exprs; i++) {
                expr = nft_setelem_expr_at(elem_expr, elem_expr->size);
-               if (nft_expr_clone(expr, priv->expr_array[i]) < 0)
+               if (nft_expr_clone(expr, priv->expr_array[i], GFP_ATOMIC) < 0)
                        return -1;
 
                elem_expr->size += priv->expr_array[i]->ops->size;
 
        kfree(priv->last);
 }
 
-static int nft_last_clone(struct nft_expr *dst, const struct nft_expr *src)
+static int nft_last_clone(struct nft_expr *dst, const struct nft_expr *src, gfp_t gfp)
 {
        struct nft_last_priv *priv_dst = nft_expr_priv(dst);
        struct nft_last_priv *priv_src = nft_expr_priv(src);
 
-       priv_dst->last = kzalloc(sizeof(*priv_dst->last), GFP_ATOMIC);
+       priv_dst->last = kzalloc(sizeof(*priv_dst->last), gfp);
        if (!priv_dst->last)
                return -ENOMEM;
 
 
 }
 
 static int nft_limit_clone(struct nft_limit_priv *priv_dst,
-                          const struct nft_limit_priv *priv_src)
+                          const struct nft_limit_priv *priv_src, gfp_t gfp)
 {
        priv_dst->tokens_max = priv_src->tokens_max;
        priv_dst->rate = priv_src->rate;
        priv_dst->burst = priv_src->burst;
        priv_dst->invert = priv_src->invert;
 
-       priv_dst->limit = kmalloc(sizeof(*priv_dst->limit), GFP_ATOMIC);
+       priv_dst->limit = kmalloc(sizeof(*priv_dst->limit), gfp);
        if (!priv_dst->limit)
                return -ENOMEM;
 
        nft_limit_destroy(ctx, &priv->limit);
 }
 
-static int nft_limit_pkts_clone(struct nft_expr *dst, const struct nft_expr *src)
+static int nft_limit_pkts_clone(struct nft_expr *dst, const struct nft_expr *src,
+                               gfp_t gfp)
 {
        struct nft_limit_priv_pkts *priv_dst = nft_expr_priv(dst);
        struct nft_limit_priv_pkts *priv_src = nft_expr_priv(src);
 
        priv_dst->cost = priv_src->cost;
 
-       return nft_limit_clone(&priv_dst->limit, &priv_src->limit);
+       return nft_limit_clone(&priv_dst->limit, &priv_src->limit, gfp);
 }
 
 static struct nft_expr_type nft_limit_type;
        nft_limit_destroy(ctx, priv);
 }
 
-static int nft_limit_bytes_clone(struct nft_expr *dst, const struct nft_expr *src)
+static int nft_limit_bytes_clone(struct nft_expr *dst, const struct nft_expr *src,
+                                gfp_t gfp)
 {
        struct nft_limit_priv *priv_dst = nft_expr_priv(dst);
        struct nft_limit_priv *priv_src = nft_expr_priv(src);
 
-       return nft_limit_clone(priv_dst, priv_src);
+       return nft_limit_clone(priv_dst, priv_src, gfp);
 }
 
 static const struct nft_expr_ops nft_limit_bytes_ops = {
 
        return nft_quota_do_destroy(ctx, priv);
 }
 
-static int nft_quota_clone(struct nft_expr *dst, const struct nft_expr *src)
+static int nft_quota_clone(struct nft_expr *dst, const struct nft_expr *src, gfp_t gfp)
 {
        struct nft_quota *priv_dst = nft_expr_priv(dst);
        struct nft_quota *priv_src = nft_expr_priv(src);
        priv_dst->quota = priv_src->quota;
        priv_dst->flags = priv_src->flags;
 
-       priv_dst->consumed = kmalloc(sizeof(*priv_dst->consumed), GFP_ATOMIC);
+       priv_dst->consumed = kmalloc(sizeof(*priv_dst->consumed), gfp);
        if (!priv_dst->consumed)
                return -ENOMEM;