1/* SPDX-License-Identifier: GPL-2.0 */
2#ifndef _NF_CONNTRACK_TIMEOUT_H
3#define _NF_CONNTRACK_TIMEOUT_H
4
5#include <net/net_namespace.h>
6#include <linux/netfilter/nf_conntrack_common.h>
7#include <linux/netfilter/nf_conntrack_tuple_common.h>
8#include <linux/refcount.h>
9#include <net/netfilter/nf_conntrack.h>
10#include <net/netfilter/nf_conntrack_extend.h>
11
12#define CTNL_TIMEOUT_NAME_MAX	32
13
14struct nf_ct_timeout {
15	__u16			l3num;
16	const struct nf_conntrack_l4proto *l4proto;
17	char			data[];
18};
19
20struct nf_conn_timeout {
21	struct nf_ct_timeout __rcu *timeout;
22};
23
24static inline unsigned int *
25nf_ct_timeout_data(const struct nf_conn_timeout *t)
26{
27#ifdef CONFIG_NF_CONNTRACK_TIMEOUT
28	struct nf_ct_timeout *timeout;
29
30	timeout = rcu_dereference(t->timeout);
31	if (timeout == NULL)
32		return NULL;
33
34	return (unsigned int *)timeout->data;
35#else
36	return NULL;
37#endif
38}
39
40static inline
41struct nf_conn_timeout *nf_ct_timeout_find(const struct nf_conn *ct)
42{
43#ifdef CONFIG_NF_CONNTRACK_TIMEOUT
44	return nf_ct_ext_find(ct, NF_CT_EXT_TIMEOUT);
45#else
46	return NULL;
47#endif
48}
49
50static inline
51struct nf_conn_timeout *nf_ct_timeout_ext_add(struct nf_conn *ct,
52					      struct nf_ct_timeout *timeout,
53					      gfp_t gfp)
54{
55#ifdef CONFIG_NF_CONNTRACK_TIMEOUT
56	struct nf_conn_timeout *timeout_ext;
57
58	timeout_ext = nf_ct_ext_add(ct, NF_CT_EXT_TIMEOUT, gfp);
59	if (timeout_ext == NULL)
60		return NULL;
61
62	rcu_assign_pointer(timeout_ext->timeout, timeout);
63
64	return timeout_ext;
65#else
66	return NULL;
67#endif
68};
69
70static inline unsigned int *nf_ct_timeout_lookup(const struct nf_conn *ct)
71{
72	unsigned int *timeouts = NULL;
73#ifdef CONFIG_NF_CONNTRACK_TIMEOUT
74	struct nf_conn_timeout *timeout_ext;
75
76	timeout_ext = nf_ct_timeout_find(ct);
77	if (timeout_ext)
78		timeouts = nf_ct_timeout_data(timeout_ext);
79#endif
80	return timeouts;
81}
82
83#ifdef CONFIG_NF_CONNTRACK_TIMEOUT
84void nf_ct_untimeout(struct net *net, struct nf_ct_timeout *timeout);
85int nf_ct_set_timeout(struct net *net, struct nf_conn *ct, u8 l3num, u8 l4num,
86		      const char *timeout_name);
87void nf_ct_destroy_timeout(struct nf_conn *ct);
88#else
89static inline int nf_ct_set_timeout(struct net *net, struct nf_conn *ct,
90				    u8 l3num, u8 l4num,
91				    const char *timeout_name)
92{
93	return -EOPNOTSUPP;
94}
95
96static inline void nf_ct_destroy_timeout(struct nf_conn *ct)
97{
98	return;
99}
100#endif /* CONFIG_NF_CONNTRACK_TIMEOUT */
101
102#ifdef CONFIG_NF_CONNTRACK_TIMEOUT
103struct nf_ct_timeout_hooks {
104	struct nf_ct_timeout *(*timeout_find_get)(struct net *net, const char *name);
105	void (*timeout_put)(struct nf_ct_timeout *timeout);
106};
107
108extern const struct nf_ct_timeout_hooks __rcu *nf_ct_timeout_hook;
109#endif
110
111#endif /* _NF_CONNTRACK_TIMEOUT_H */
112