netfilter: don't use INIT_RCU_HEAD()
[linux-2.6.git] / net / netfilter / nf_conntrack_ecache.c
1 /* Event cache for netfilter. */
2
3 /* (C) 1999-2001 Paul `Rusty' Russell
4  * (C) 2002-2006 Netfilter Core Team <coreteam@netfilter.org>
5  * (C) 2003,2004 USAGI/WIDE Project <http://www.linux-ipv6.org>
6  *
7  * This program is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License version 2 as
9  * published by the Free Software Foundation.
10  */
11
12 #include <linux/types.h>
13 #include <linux/netfilter.h>
14 #include <linux/skbuff.h>
15 #include <linux/vmalloc.h>
16 #include <linux/stddef.h>
17 #include <linux/err.h>
18 #include <linux/percpu.h>
19 #include <linux/kernel.h>
20 #include <linux/netdevice.h>
21
22 #include <net/netfilter/nf_conntrack.h>
23 #include <net/netfilter/nf_conntrack_core.h>
24 #include <net/netfilter/nf_conntrack_extend.h>
25
26 static DEFINE_MUTEX(nf_ct_ecache_mutex);
27
28 struct nf_ct_event_notifier *nf_conntrack_event_cb __read_mostly;
29 EXPORT_SYMBOL_GPL(nf_conntrack_event_cb);
30
31 struct nf_exp_event_notifier *nf_expect_event_cb __read_mostly;
32 EXPORT_SYMBOL_GPL(nf_expect_event_cb);
33
34 /* deliver cached events and clear cache entry - must be called with locally
35  * disabled softirqs */
36 void nf_ct_deliver_cached_events(struct nf_conn *ct)
37 {
38         unsigned long events;
39         struct nf_ct_event_notifier *notify;
40         struct nf_conntrack_ecache *e;
41
42         rcu_read_lock();
43         notify = rcu_dereference(nf_conntrack_event_cb);
44         if (notify == NULL)
45                 goto out_unlock;
46
47         e = nf_ct_ecache_find(ct);
48         if (e == NULL)
49                 goto out_unlock;
50
51         events = xchg(&e->cache, 0);
52
53         if (nf_ct_is_confirmed(ct) && !nf_ct_is_dying(ct) && events) {
54                 struct nf_ct_event item = {
55                         .ct     = ct,
56                         .pid    = 0,
57                         .report = 0
58                 };
59                 int ret;
60                 /* We make a copy of the missed event cache without taking
61                  * the lock, thus we may send missed events twice. However,
62                  * this does not harm and it happens very rarely. */
63                 unsigned long missed = e->missed;
64
65                 ret = notify->fcn(events | missed, &item);
66                 if (unlikely(ret < 0 || missed)) {
67                         spin_lock_bh(&ct->lock);
68                         if (ret < 0)
69                                 e->missed |= events;
70                         else
71                                 e->missed &= ~missed;
72                         spin_unlock_bh(&ct->lock);
73                 } 
74         }
75
76 out_unlock:
77         rcu_read_unlock();
78 }
79 EXPORT_SYMBOL_GPL(nf_ct_deliver_cached_events);
80
81 int nf_conntrack_register_notifier(struct nf_ct_event_notifier *new)
82 {
83         int ret = 0;
84         struct nf_ct_event_notifier *notify;
85
86         mutex_lock(&nf_ct_ecache_mutex);
87         notify = rcu_dereference(nf_conntrack_event_cb);
88         if (notify != NULL) {
89                 ret = -EBUSY;
90                 goto out_unlock;
91         }
92         rcu_assign_pointer(nf_conntrack_event_cb, new);
93         mutex_unlock(&nf_ct_ecache_mutex);
94         return ret;
95
96 out_unlock:
97         mutex_unlock(&nf_ct_ecache_mutex);
98         return ret;
99 }
100 EXPORT_SYMBOL_GPL(nf_conntrack_register_notifier);
101
102 void nf_conntrack_unregister_notifier(struct nf_ct_event_notifier *new)
103 {
104         struct nf_ct_event_notifier *notify;
105
106         mutex_lock(&nf_ct_ecache_mutex);
107         notify = rcu_dereference(nf_conntrack_event_cb);
108         BUG_ON(notify != new);
109         rcu_assign_pointer(nf_conntrack_event_cb, NULL);
110         mutex_unlock(&nf_ct_ecache_mutex);
111 }
112 EXPORT_SYMBOL_GPL(nf_conntrack_unregister_notifier);
113
114 int nf_ct_expect_register_notifier(struct nf_exp_event_notifier *new)
115 {
116         int ret = 0;
117         struct nf_exp_event_notifier *notify;
118
119         mutex_lock(&nf_ct_ecache_mutex);
120         notify = rcu_dereference(nf_expect_event_cb);
121         if (notify != NULL) {
122                 ret = -EBUSY;
123                 goto out_unlock;
124         }
125         rcu_assign_pointer(nf_expect_event_cb, new);
126         mutex_unlock(&nf_ct_ecache_mutex);
127         return ret;
128
129 out_unlock:
130         mutex_unlock(&nf_ct_ecache_mutex);
131         return ret;
132 }
133 EXPORT_SYMBOL_GPL(nf_ct_expect_register_notifier);
134
135 void nf_ct_expect_unregister_notifier(struct nf_exp_event_notifier *new)
136 {
137         struct nf_exp_event_notifier *notify;
138
139         mutex_lock(&nf_ct_ecache_mutex);
140         notify = rcu_dereference(nf_expect_event_cb);
141         BUG_ON(notify != new);
142         rcu_assign_pointer(nf_expect_event_cb, NULL);
143         mutex_unlock(&nf_ct_ecache_mutex);
144 }
145 EXPORT_SYMBOL_GPL(nf_ct_expect_unregister_notifier);
146
147 #define NF_CT_EVENTS_DEFAULT 1
148 static int nf_ct_events __read_mostly = NF_CT_EVENTS_DEFAULT;
149 static int nf_ct_events_retry_timeout __read_mostly = 15*HZ;
150
151 #ifdef CONFIG_SYSCTL
152 static struct ctl_table event_sysctl_table[] = {
153         {
154                 .procname       = "nf_conntrack_events",
155                 .data           = &init_net.ct.sysctl_events,
156                 .maxlen         = sizeof(unsigned int),
157                 .mode           = 0644,
158                 .proc_handler   = proc_dointvec,
159         },
160         {
161                 .procname       = "nf_conntrack_events_retry_timeout",
162                 .data           = &init_net.ct.sysctl_events_retry_timeout,
163                 .maxlen         = sizeof(unsigned int),
164                 .mode           = 0644,
165                 .proc_handler   = proc_dointvec_jiffies,
166         },
167         {}
168 };
169 #endif /* CONFIG_SYSCTL */
170
171 static struct nf_ct_ext_type event_extend __read_mostly = {
172         .len    = sizeof(struct nf_conntrack_ecache),
173         .align  = __alignof__(struct nf_conntrack_ecache),
174         .id     = NF_CT_EXT_ECACHE,
175 };
176
177 #ifdef CONFIG_SYSCTL
178 static int nf_conntrack_event_init_sysctl(struct net *net)
179 {
180         struct ctl_table *table;
181
182         table = kmemdup(event_sysctl_table, sizeof(event_sysctl_table),
183                         GFP_KERNEL);
184         if (!table)
185                 goto out;
186
187         table[0].data = &net->ct.sysctl_events;
188         table[1].data = &net->ct.sysctl_events_retry_timeout;
189
190         net->ct.event_sysctl_header =
191                 register_net_sysctl_table(net,
192                                           nf_net_netfilter_sysctl_path, table);
193         if (!net->ct.event_sysctl_header) {
194                 printk(KERN_ERR "nf_ct_event: can't register to sysctl.\n");
195                 goto out_register;
196         }
197         return 0;
198
199 out_register:
200         kfree(table);
201 out:
202         return -ENOMEM;
203 }
204
205 static void nf_conntrack_event_fini_sysctl(struct net *net)
206 {
207         struct ctl_table *table;
208
209         table = net->ct.event_sysctl_header->ctl_table_arg;
210         unregister_net_sysctl_table(net->ct.event_sysctl_header);
211         kfree(table);
212 }
213 #else
214 static int nf_conntrack_event_init_sysctl(struct net *net)
215 {
216         return 0;
217 }
218
219 static void nf_conntrack_event_fini_sysctl(struct net *net)
220 {
221 }
222 #endif /* CONFIG_SYSCTL */
223
224 int nf_conntrack_ecache_init(struct net *net)
225 {
226         int ret;
227
228         net->ct.sysctl_events = nf_ct_events;
229         net->ct.sysctl_events_retry_timeout = nf_ct_events_retry_timeout;
230
231         if (net_eq(net, &init_net)) {
232                 ret = nf_ct_extend_register(&event_extend);
233                 if (ret < 0) {
234                         printk(KERN_ERR "nf_ct_event: Unable to register "
235                                         "event extension.\n");
236                         goto out_extend_register;
237                 }
238         }
239
240         ret = nf_conntrack_event_init_sysctl(net);
241         if (ret < 0)
242                 goto out_sysctl;
243
244         return 0;
245
246 out_sysctl:
247         if (net_eq(net, &init_net))
248                 nf_ct_extend_unregister(&event_extend);
249 out_extend_register:
250         return ret;
251 }
252
253 void nf_conntrack_ecache_fini(struct net *net)
254 {
255         nf_conntrack_event_fini_sysctl(net);
256         if (net_eq(net, &init_net))
257                 nf_ct_extend_unregister(&event_extend);
258 }