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