Commit | Line | Data |
---|---|---|
f6ebe77f HW |
1 | #include <linux/kernel.h> |
2 | #include <linux/init.h> | |
3 | #include <linux/module.h> | |
4 | #include <linux/proc_fs.h> | |
5 | #include <linux/skbuff.h> | |
6 | #include <linux/netfilter.h> | |
bbd86b9f | 7 | #include <linux/seq_file.h> |
f6ebe77f | 8 | #include <net/protocol.h> |
f01ffbd6 | 9 | #include <net/netfilter/nf_log.h> |
f6ebe77f HW |
10 | |
11 | #include "nf_internals.h" | |
12 | ||
a5d29264 | 13 | /* Internal logging interface, which relies on the real |
f6ebe77f HW |
14 | LOG target modules */ |
15 | ||
16 | #define NF_LOG_PREFIXLEN 128 | |
17625274 | 17 | #define NFLOGGER_NAME_LEN 64 |
f6ebe77f | 18 | |
5962815a | 19 | static struct nf_logger __rcu *loggers[NFPROTO_NUMPROTO][NF_LOG_TYPE_MAX] __read_mostly; |
9b73534d | 20 | static DEFINE_MUTEX(nf_log_mutex); |
f6ebe77f | 21 | |
ca735b3a | 22 | static struct nf_logger *__find_logger(int pf, const char *str_logger) |
f6ebe77f | 23 | { |
5962815a PNA |
24 | struct nf_logger *log; |
25 | int i; | |
26 | ||
27 | for (i = 0; i < NF_LOG_TYPE_MAX; i++) { | |
28 | if (loggers[pf][i] == NULL) | |
29 | continue; | |
f6ebe77f | 30 | |
5962815a PNA |
31 | log = rcu_dereference_protected(loggers[pf][i], |
32 | lockdep_is_held(&nf_log_mutex)); | |
18082746 | 33 | if (!strncasecmp(str_logger, log->name, strlen(log->name))) |
5962815a | 34 | return log; |
ca735b3a | 35 | } |
d72367b6 | 36 | |
ca735b3a | 37 | return NULL; |
601e68e1 | 38 | } |
f6ebe77f | 39 | |
30e0c6a6 G |
40 | void nf_log_set(struct net *net, u_int8_t pf, const struct nf_logger *logger) |
41 | { | |
42 | const struct nf_logger *log; | |
43 | ||
30e0c6a6 G |
44 | if (pf == NFPROTO_UNSPEC) |
45 | return; | |
46 | ||
47 | mutex_lock(&nf_log_mutex); | |
48 | log = rcu_dereference_protected(net->nf.nf_loggers[pf], | |
49 | lockdep_is_held(&nf_log_mutex)); | |
50 | if (log == NULL) | |
51 | rcu_assign_pointer(net->nf.nf_loggers[pf], logger); | |
52 | ||
53 | mutex_unlock(&nf_log_mutex); | |
54 | } | |
55 | EXPORT_SYMBOL(nf_log_set); | |
56 | ||
57 | void nf_log_unset(struct net *net, const struct nf_logger *logger) | |
58 | { | |
59 | int i; | |
60 | const struct nf_logger *log; | |
61 | ||
30e0c6a6 G |
62 | mutex_lock(&nf_log_mutex); |
63 | for (i = 0; i < NFPROTO_NUMPROTO; i++) { | |
64 | log = rcu_dereference_protected(net->nf.nf_loggers[i], | |
65 | lockdep_is_held(&nf_log_mutex)); | |
66 | if (log == logger) | |
67 | RCU_INIT_POINTER(net->nf.nf_loggers[i], NULL); | |
68 | } | |
69 | mutex_unlock(&nf_log_mutex); | |
70 | synchronize_rcu(); | |
71 | } | |
72 | EXPORT_SYMBOL(nf_log_unset); | |
73 | ||
d93cf068 | 74 | /* return EEXIST if the same logger is registered, 0 on success. */ |
ca735b3a | 75 | int nf_log_register(u_int8_t pf, struct nf_logger *logger) |
f6ebe77f | 76 | { |
b6f0a365 | 77 | int i; |
ca735b3a | 78 | |
30e0c6a6 | 79 | if (pf >= ARRAY_SIZE(init_net.nf.nf_loggers)) |
ca735b3a EL |
80 | return -EINVAL; |
81 | ||
9b73534d | 82 | mutex_lock(&nf_log_mutex); |
ca735b3a EL |
83 | |
84 | if (pf == NFPROTO_UNSPEC) { | |
ca735b3a | 85 | for (i = NFPROTO_UNSPEC; i < NFPROTO_NUMPROTO; i++) |
5962815a | 86 | rcu_assign_pointer(loggers[i][logger->type], logger); |
ca735b3a EL |
87 | } else { |
88 | /* register at end of list to honor first register win */ | |
5962815a | 89 | rcu_assign_pointer(loggers[pf][logger->type], logger); |
ca735b3a EL |
90 | } |
91 | ||
9b73534d | 92 | mutex_unlock(&nf_log_mutex); |
f6ebe77f | 93 | |
ca735b3a | 94 | return 0; |
f6ebe77f | 95 | } |
ca735b3a | 96 | EXPORT_SYMBOL(nf_log_register); |
f6ebe77f | 97 | |
ca735b3a | 98 | void nf_log_unregister(struct nf_logger *logger) |
f6ebe77f HW |
99 | { |
100 | int i; | |
101 | ||
9b73534d | 102 | mutex_lock(&nf_log_mutex); |
30e0c6a6 | 103 | for (i = 0; i < NFPROTO_NUMPROTO; i++) |
5962815a | 104 | RCU_INIT_POINTER(loggers[i][logger->type], NULL); |
9b73534d | 105 | mutex_unlock(&nf_log_mutex); |
f6ebe77f | 106 | } |
e92ad99c | 107 | EXPORT_SYMBOL(nf_log_unregister); |
f6ebe77f | 108 | |
30e0c6a6 G |
109 | int nf_log_bind_pf(struct net *net, u_int8_t pf, |
110 | const struct nf_logger *logger) | |
ca735b3a | 111 | { |
30e0c6a6 | 112 | if (pf >= ARRAY_SIZE(net->nf.nf_loggers)) |
9ef0298a | 113 | return -EINVAL; |
ca735b3a EL |
114 | mutex_lock(&nf_log_mutex); |
115 | if (__find_logger(pf, logger->name) == NULL) { | |
116 | mutex_unlock(&nf_log_mutex); | |
117 | return -ENOENT; | |
118 | } | |
30e0c6a6 | 119 | rcu_assign_pointer(net->nf.nf_loggers[pf], logger); |
ca735b3a EL |
120 | mutex_unlock(&nf_log_mutex); |
121 | return 0; | |
122 | } | |
123 | EXPORT_SYMBOL(nf_log_bind_pf); | |
124 | ||
30e0c6a6 | 125 | void nf_log_unbind_pf(struct net *net, u_int8_t pf) |
ca735b3a | 126 | { |
30e0c6a6 | 127 | if (pf >= ARRAY_SIZE(net->nf.nf_loggers)) |
9ef0298a | 128 | return; |
ca735b3a | 129 | mutex_lock(&nf_log_mutex); |
30e0c6a6 | 130 | RCU_INIT_POINTER(net->nf.nf_loggers[pf], NULL); |
ca735b3a EL |
131 | mutex_unlock(&nf_log_mutex); |
132 | } | |
133 | EXPORT_SYMBOL(nf_log_unbind_pf); | |
134 | ||
960649d1 PNA |
135 | void nf_logger_request_module(int pf, enum nf_log_type type) |
136 | { | |
137 | if (loggers[pf][type] == NULL) | |
138 | request_module("nf-logger-%u-%u", pf, type); | |
139 | } | |
140 | EXPORT_SYMBOL_GPL(nf_logger_request_module); | |
141 | ||
fab4085f PNA |
142 | int nf_logger_find_get(int pf, enum nf_log_type type) |
143 | { | |
144 | struct nf_logger *logger; | |
145 | int ret = -ENOENT; | |
146 | ||
147 | logger = loggers[pf][type]; | |
148 | if (logger == NULL) | |
149 | request_module("nf-logger-%u-%u", pf, type); | |
150 | ||
151 | rcu_read_lock(); | |
152 | logger = rcu_dereference(loggers[pf][type]); | |
153 | if (logger == NULL) | |
154 | goto out; | |
155 | ||
156 | if (logger && try_module_get(logger->me)) | |
157 | ret = 0; | |
158 | out: | |
159 | rcu_read_unlock(); | |
160 | return ret; | |
161 | } | |
162 | EXPORT_SYMBOL_GPL(nf_logger_find_get); | |
163 | ||
164 | void nf_logger_put(int pf, enum nf_log_type type) | |
165 | { | |
166 | struct nf_logger *logger; | |
167 | ||
168 | BUG_ON(loggers[pf][type] == NULL); | |
169 | ||
170 | rcu_read_lock(); | |
171 | logger = rcu_dereference(loggers[pf][type]); | |
172 | module_put(logger->me); | |
173 | rcu_read_unlock(); | |
174 | } | |
175 | EXPORT_SYMBOL_GPL(nf_logger_put); | |
176 | ||
30e0c6a6 G |
177 | void nf_log_packet(struct net *net, |
178 | u_int8_t pf, | |
f6ebe77f HW |
179 | unsigned int hooknum, |
180 | const struct sk_buff *skb, | |
181 | const struct net_device *in, | |
182 | const struct net_device *out, | |
7b2f9631 | 183 | const struct nf_loginfo *loginfo, |
f6ebe77f HW |
184 | const char *fmt, ...) |
185 | { | |
186 | va_list args; | |
187 | char prefix[NF_LOG_PREFIXLEN]; | |
7b2f9631 | 188 | const struct nf_logger *logger; |
601e68e1 | 189 | |
f6ebe77f | 190 | rcu_read_lock(); |
fab4085f PNA |
191 | if (loginfo != NULL) |
192 | logger = rcu_dereference(loggers[pf][loginfo->type]); | |
193 | else | |
194 | logger = rcu_dereference(net->nf.nf_loggers[pf]); | |
195 | ||
f6ebe77f HW |
196 | if (logger) { |
197 | va_start(args, fmt); | |
198 | vsnprintf(prefix, sizeof(prefix), fmt, args); | |
199 | va_end(args); | |
8cdb46da | 200 | logger->logfn(net, pf, hooknum, skb, in, out, loginfo, prefix); |
f6ebe77f HW |
201 | } |
202 | rcu_read_unlock(); | |
203 | } | |
204 | EXPORT_SYMBOL(nf_log_packet); | |
205 | ||
27fd8d90 PNA |
206 | #define S_SIZE (1024 - (sizeof(unsigned int) + 1)) |
207 | ||
208 | struct nf_log_buf { | |
209 | unsigned int count; | |
210 | char buf[S_SIZE + 1]; | |
211 | }; | |
212 | static struct nf_log_buf emergency, *emergency_ptr = &emergency; | |
213 | ||
214 | __printf(2, 3) int nf_log_buf_add(struct nf_log_buf *m, const char *f, ...) | |
215 | { | |
216 | va_list args; | |
217 | int len; | |
218 | ||
219 | if (likely(m->count < S_SIZE)) { | |
220 | va_start(args, f); | |
221 | len = vsnprintf(m->buf + m->count, S_SIZE - m->count, f, args); | |
222 | va_end(args); | |
223 | if (likely(m->count + len < S_SIZE)) { | |
224 | m->count += len; | |
225 | return 0; | |
226 | } | |
227 | } | |
228 | m->count = S_SIZE; | |
229 | printk_once(KERN_ERR KBUILD_MODNAME " please increase S_SIZE\n"); | |
230 | return -1; | |
231 | } | |
232 | EXPORT_SYMBOL_GPL(nf_log_buf_add); | |
233 | ||
234 | struct nf_log_buf *nf_log_buf_open(void) | |
235 | { | |
236 | struct nf_log_buf *m = kmalloc(sizeof(*m), GFP_ATOMIC); | |
237 | ||
238 | if (unlikely(!m)) { | |
239 | local_bh_disable(); | |
240 | do { | |
241 | m = xchg(&emergency_ptr, NULL); | |
242 | } while (!m); | |
243 | } | |
244 | m->count = 0; | |
245 | return m; | |
246 | } | |
247 | EXPORT_SYMBOL_GPL(nf_log_buf_open); | |
248 | ||
249 | void nf_log_buf_close(struct nf_log_buf *m) | |
250 | { | |
251 | m->buf[m->count] = 0; | |
252 | printk("%s\n", m->buf); | |
253 | ||
254 | if (likely(m != &emergency)) | |
255 | kfree(m); | |
256 | else { | |
257 | emergency_ptr = m; | |
258 | local_bh_enable(); | |
259 | } | |
260 | } | |
261 | EXPORT_SYMBOL_GPL(nf_log_buf_close); | |
262 | ||
f6ebe77f HW |
263 | #ifdef CONFIG_PROC_FS |
264 | static void *seq_start(struct seq_file *seq, loff_t *pos) | |
265 | { | |
30e0c6a6 G |
266 | struct net *net = seq_file_net(seq); |
267 | ||
6440fe05 | 268 | mutex_lock(&nf_log_mutex); |
f6ebe77f | 269 | |
30e0c6a6 | 270 | if (*pos >= ARRAY_SIZE(net->nf.nf_loggers)) |
f6ebe77f HW |
271 | return NULL; |
272 | ||
273 | return pos; | |
274 | } | |
275 | ||
276 | static void *seq_next(struct seq_file *s, void *v, loff_t *pos) | |
277 | { | |
30e0c6a6 G |
278 | struct net *net = seq_file_net(s); |
279 | ||
f6ebe77f HW |
280 | (*pos)++; |
281 | ||
30e0c6a6 | 282 | if (*pos >= ARRAY_SIZE(net->nf.nf_loggers)) |
f6ebe77f HW |
283 | return NULL; |
284 | ||
285 | return pos; | |
286 | } | |
287 | ||
288 | static void seq_stop(struct seq_file *s, void *v) | |
289 | { | |
6440fe05 | 290 | mutex_unlock(&nf_log_mutex); |
f6ebe77f HW |
291 | } |
292 | ||
293 | static int seq_show(struct seq_file *s, void *v) | |
294 | { | |
295 | loff_t *pos = v; | |
296 | const struct nf_logger *logger; | |
5962815a | 297 | int i, ret; |
30e0c6a6 | 298 | struct net *net = seq_file_net(s); |
f6ebe77f | 299 | |
30e0c6a6 | 300 | logger = rcu_dereference_protected(net->nf.nf_loggers[*pos], |
0e60ebe0 | 301 | lockdep_is_held(&nf_log_mutex)); |
f6ebe77f HW |
302 | |
303 | if (!logger) | |
c7a913cd EL |
304 | ret = seq_printf(s, "%2lld NONE (", *pos); |
305 | else | |
306 | ret = seq_printf(s, "%2lld %s (", *pos, logger->name); | |
307 | ||
308 | if (ret < 0) | |
309 | return ret; | |
310 | ||
5962815a PNA |
311 | for (i = 0; i < NF_LOG_TYPE_MAX; i++) { |
312 | if (loggers[*pos][i] == NULL) | |
313 | continue; | |
314 | ||
315 | logger = rcu_dereference_protected(loggers[*pos][i], | |
316 | lockdep_is_held(&nf_log_mutex)); | |
317 | ret = seq_printf(s, "%s", logger->name); | |
6440fe05 | 318 | if (ret < 0) |
c7a913cd | 319 | return ret; |
5962815a | 320 | if (i == 0 && loggers[*pos][i + 1] != NULL) { |
c7a913cd | 321 | ret = seq_printf(s, ","); |
6440fe05 | 322 | if (ret < 0) |
c7a913cd | 323 | return ret; |
c7a913cd EL |
324 | } |
325 | } | |
601e68e1 | 326 | |
c7a913cd | 327 | return seq_printf(s, ")\n"); |
f6ebe77f HW |
328 | } |
329 | ||
56b3d975 | 330 | static const struct seq_operations nflog_seq_ops = { |
f6ebe77f HW |
331 | .start = seq_start, |
332 | .next = seq_next, | |
333 | .stop = seq_stop, | |
334 | .show = seq_show, | |
335 | }; | |
336 | ||
337 | static int nflog_open(struct inode *inode, struct file *file) | |
338 | { | |
30e0c6a6 G |
339 | return seq_open_net(inode, file, &nflog_seq_ops, |
340 | sizeof(struct seq_net_private)); | |
f6ebe77f HW |
341 | } |
342 | ||
da7071d7 | 343 | static const struct file_operations nflog_file_ops = { |
f6ebe77f HW |
344 | .owner = THIS_MODULE, |
345 | .open = nflog_open, | |
346 | .read = seq_read, | |
347 | .llseek = seq_lseek, | |
30e0c6a6 | 348 | .release = seq_release_net, |
f6ebe77f HW |
349 | }; |
350 | ||
17625274 | 351 | |
f6ebe77f HW |
352 | #endif /* PROC_FS */ |
353 | ||
17625274 | 354 | #ifdef CONFIG_SYSCTL |
17625274 EL |
355 | static char nf_log_sysctl_fnames[NFPROTO_NUMPROTO-NFPROTO_UNSPEC][3]; |
356 | static struct ctl_table nf_log_sysctl_table[NFPROTO_NUMPROTO+1]; | |
f6ebe77f | 357 | |
fe2c6338 | 358 | static int nf_log_proc_dostring(struct ctl_table *table, int write, |
24955619 | 359 | void __user *buffer, size_t *lenp, loff_t *ppos) |
17625274 EL |
360 | { |
361 | const struct nf_logger *logger; | |
24955619 PM |
362 | char buf[NFLOGGER_NAME_LEN]; |
363 | size_t size = *lenp; | |
17625274 EL |
364 | int r = 0; |
365 | int tindex = (unsigned long)table->extra1; | |
30e0c6a6 | 366 | struct net *net = current->nsproxy->net_ns; |
17625274 EL |
367 | |
368 | if (write) { | |
24955619 PM |
369 | if (size > sizeof(buf)) |
370 | size = sizeof(buf); | |
371 | if (copy_from_user(buf, buffer, size)) | |
372 | return -EFAULT; | |
373 | ||
374 | if (!strcmp(buf, "NONE")) { | |
30e0c6a6 | 375 | nf_log_unbind_pf(net, tindex); |
17625274 EL |
376 | return 0; |
377 | } | |
378 | mutex_lock(&nf_log_mutex); | |
24955619 | 379 | logger = __find_logger(tindex, buf); |
17625274 EL |
380 | if (logger == NULL) { |
381 | mutex_unlock(&nf_log_mutex); | |
382 | return -ENOENT; | |
383 | } | |
30e0c6a6 | 384 | rcu_assign_pointer(net->nf.nf_loggers[tindex], logger); |
17625274 EL |
385 | mutex_unlock(&nf_log_mutex); |
386 | } else { | |
266d07cb | 387 | mutex_lock(&nf_log_mutex); |
30e0c6a6 | 388 | logger = rcu_dereference_protected(net->nf.nf_loggers[tindex], |
0e60ebe0 | 389 | lockdep_is_held(&nf_log_mutex)); |
17625274 EL |
390 | if (!logger) |
391 | table->data = "NONE"; | |
392 | else | |
393 | table->data = logger->name; | |
8d65af78 | 394 | r = proc_dostring(table, write, buffer, lenp, ppos); |
266d07cb | 395 | mutex_unlock(&nf_log_mutex); |
17625274 EL |
396 | } |
397 | ||
398 | return r; | |
399 | } | |
400 | ||
30e0c6a6 | 401 | static int netfilter_log_sysctl_init(struct net *net) |
f6ebe77f | 402 | { |
ca735b3a | 403 | int i; |
30e0c6a6 G |
404 | struct ctl_table *table; |
405 | ||
406 | table = nf_log_sysctl_table; | |
407 | if (!net_eq(net, &init_net)) { | |
408 | table = kmemdup(nf_log_sysctl_table, | |
409 | sizeof(nf_log_sysctl_table), | |
410 | GFP_KERNEL); | |
411 | if (!table) | |
412 | goto err_alloc; | |
413 | } else { | |
414 | for (i = NFPROTO_UNSPEC; i < NFPROTO_NUMPROTO; i++) { | |
415 | snprintf(nf_log_sysctl_fnames[i], | |
416 | 3, "%d", i); | |
417 | nf_log_sysctl_table[i].procname = | |
418 | nf_log_sysctl_fnames[i]; | |
419 | nf_log_sysctl_table[i].data = NULL; | |
420 | nf_log_sysctl_table[i].maxlen = | |
421 | NFLOGGER_NAME_LEN * sizeof(char); | |
422 | nf_log_sysctl_table[i].mode = 0644; | |
423 | nf_log_sysctl_table[i].proc_handler = | |
424 | nf_log_proc_dostring; | |
425 | nf_log_sysctl_table[i].extra1 = | |
426 | (void *)(unsigned long) i; | |
427 | } | |
17625274 EL |
428 | } |
429 | ||
30e0c6a6 G |
430 | net->nf.nf_log_dir_header = register_net_sysctl(net, |
431 | "net/netfilter/nf_log", | |
432 | table); | |
433 | if (!net->nf.nf_log_dir_header) | |
434 | goto err_reg; | |
17625274 EL |
435 | |
436 | return 0; | |
30e0c6a6 G |
437 | |
438 | err_reg: | |
439 | if (!net_eq(net, &init_net)) | |
440 | kfree(table); | |
441 | err_alloc: | |
442 | return -ENOMEM; | |
443 | } | |
444 | ||
445 | static void netfilter_log_sysctl_exit(struct net *net) | |
446 | { | |
447 | struct ctl_table *table; | |
448 | ||
449 | table = net->nf.nf_log_dir_header->ctl_table_arg; | |
450 | unregister_net_sysctl_table(net->nf.nf_log_dir_header); | |
451 | if (!net_eq(net, &init_net)) | |
452 | kfree(table); | |
17625274 EL |
453 | } |
454 | #else | |
30e0c6a6 | 455 | static int netfilter_log_sysctl_init(struct net *net) |
17625274 EL |
456 | { |
457 | return 0; | |
458 | } | |
30e0c6a6 G |
459 | |
460 | static void netfilter_log_sysctl_exit(struct net *net) | |
461 | { | |
462 | } | |
17625274 EL |
463 | #endif /* CONFIG_SYSCTL */ |
464 | ||
30e0c6a6 | 465 | static int __net_init nf_log_net_init(struct net *net) |
17625274 | 466 | { |
30e0c6a6 G |
467 | int ret = -ENOMEM; |
468 | ||
f6ebe77f | 469 | #ifdef CONFIG_PROC_FS |
8eeee8b1 | 470 | if (!proc_create("nf_log", S_IRUGO, |
30e0c6a6 G |
471 | net->nf.proc_netfilter, &nflog_file_ops)) |
472 | return ret; | |
62243927 | 473 | #endif |
30e0c6a6 G |
474 | ret = netfilter_log_sysctl_init(net); |
475 | if (ret < 0) | |
476 | goto out_sysctl; | |
ca735b3a | 477 | |
30e0c6a6 G |
478 | return 0; |
479 | ||
480 | out_sysctl: | |
e778f56e | 481 | #ifdef CONFIG_PROC_FS |
6d11cfdb | 482 | remove_proc_entry("nf_log", net->nf.proc_netfilter); |
e778f56e | 483 | #endif |
30e0c6a6 G |
484 | return ret; |
485 | } | |
486 | ||
487 | static void __net_exit nf_log_net_exit(struct net *net) | |
488 | { | |
489 | netfilter_log_sysctl_exit(net); | |
e778f56e | 490 | #ifdef CONFIG_PROC_FS |
30e0c6a6 | 491 | remove_proc_entry("nf_log", net->nf.proc_netfilter); |
e778f56e | 492 | #endif |
30e0c6a6 G |
493 | } |
494 | ||
495 | static struct pernet_operations nf_log_net_ops = { | |
496 | .init = nf_log_net_init, | |
497 | .exit = nf_log_net_exit, | |
498 | }; | |
499 | ||
500 | int __init netfilter_log_init(void) | |
501 | { | |
5962815a | 502 | return register_pernet_subsys(&nf_log_net_ops); |
f6ebe77f | 503 | } |