#ifndef _NF_CONNTRACK_COUNT_H
#define _NF_CONNTRACK_COUNT_H

#include <linux/list.h>
#include <linux/spinlock.h>
#include <net/netfilter/nf_conntrack_tuple.h>
#include <net/netfilter/nf_conntrack_zones.h>

struct nf_conncount_data;

struct nf_conncount_list {
        spinlock_t list_lock;
        u32 last_gc;            /* jiffies at most recent gc */
        struct list_head head;  /* connections with the same filtering key */
        unsigned int count;     /* length of list */
        unsigned int last_gc_count; /* length of list at most recent gc */
};

struct nf_conncount_data *nf_conncount_init(struct net *net, unsigned int keylen);
void nf_conncount_destroy(struct net *net, struct nf_conncount_data *data);

unsigned int nf_conncount_count_skb(struct net *net,
                                    const struct sk_buff *skb,
                                    u16 l3num,
                                    struct nf_conncount_data *data,
                                    const u32 *key);

int nf_conncount_add_skb(struct net *net, const struct sk_buff *skb,
                         u16 l3num, struct nf_conncount_list *list);

void nf_conncount_list_init(struct nf_conncount_list *list);

bool nf_conncount_gc_list(struct net *net,
                          struct nf_conncount_list *list);

void nf_conncount_cache_free(struct nf_conncount_list *list);

#endif