Merge branch 'linus' into tracing/urgent
[linux-2.6] / net / netfilter / nf_sockopt.c
1 #include <linux/kernel.h>
2 #include <linux/init.h>
3 #include <linux/module.h>
4 #include <linux/skbuff.h>
5 #include <linux/netfilter.h>
6 #include <linux/mutex.h>
7 #include <net/sock.h>
8
9 #include "nf_internals.h"
10
11 /* Sockopts only registered and called from user context, so
12    net locking would be overkill.  Also, [gs]etsockopt calls may
13    sleep. */
14 static DEFINE_MUTEX(nf_sockopt_mutex);
15 static LIST_HEAD(nf_sockopts);
16
17 /* Do exclusive ranges overlap? */
18 static inline int overlap(int min1, int max1, int min2, int max2)
19 {
20         return max1 > min2 && min1 < max2;
21 }
22
23 /* Functions to register sockopt ranges (exclusive). */
24 int nf_register_sockopt(struct nf_sockopt_ops *reg)
25 {
26         struct nf_sockopt_ops *ops;
27         int ret = 0;
28
29         if (mutex_lock_interruptible(&nf_sockopt_mutex) != 0)
30                 return -EINTR;
31
32         list_for_each_entry(ops, &nf_sockopts, list) {
33                 if (ops->pf == reg->pf
34                     && (overlap(ops->set_optmin, ops->set_optmax,
35                                 reg->set_optmin, reg->set_optmax)
36                         || overlap(ops->get_optmin, ops->get_optmax,
37                                    reg->get_optmin, reg->get_optmax))) {
38                         NFDEBUG("nf_sock overlap: %u-%u/%u-%u v %u-%u/%u-%u\n",
39                                 ops->set_optmin, ops->set_optmax,
40                                 ops->get_optmin, ops->get_optmax,
41                                 reg->set_optmin, reg->set_optmax,
42                                 reg->get_optmin, reg->get_optmax);
43                         ret = -EBUSY;
44                         goto out;
45                 }
46         }
47
48         list_add(&reg->list, &nf_sockopts);
49 out:
50         mutex_unlock(&nf_sockopt_mutex);
51         return ret;
52 }
53 EXPORT_SYMBOL(nf_register_sockopt);
54
55 void nf_unregister_sockopt(struct nf_sockopt_ops *reg)
56 {
57         mutex_lock(&nf_sockopt_mutex);
58         list_del(&reg->list);
59         mutex_unlock(&nf_sockopt_mutex);
60 }
61 EXPORT_SYMBOL(nf_unregister_sockopt);
62
63 static struct nf_sockopt_ops *nf_sockopt_find(struct sock *sk, int pf,
64                 int val, int get)
65 {
66         struct nf_sockopt_ops *ops;
67
68         if (!net_eq(sock_net(sk), &init_net))
69                 return ERR_PTR(-ENOPROTOOPT);
70
71         if (mutex_lock_interruptible(&nf_sockopt_mutex) != 0)
72                 return ERR_PTR(-EINTR);
73
74         list_for_each_entry(ops, &nf_sockopts, list) {
75                 if (ops->pf == pf) {
76                         if (!try_module_get(ops->owner))
77                                 goto out_nosup;
78
79                         if (get) {
80                                 if (val >= ops->get_optmin &&
81                                                 val < ops->get_optmax)
82                                         goto out;
83                         } else {
84                                 if (val >= ops->set_optmin &&
85                                                 val < ops->set_optmax)
86                                         goto out;
87                         }
88                         module_put(ops->owner);
89                 }
90         }
91 out_nosup:
92         ops = ERR_PTR(-ENOPROTOOPT);
93 out:
94         mutex_unlock(&nf_sockopt_mutex);
95         return ops;
96 }
97
98 /* Call get/setsockopt() */
99 static int nf_sockopt(struct sock *sk, int pf, int val,
100                       char __user *opt, int *len, int get)
101 {
102         struct nf_sockopt_ops *ops;
103         int ret;
104
105         ops = nf_sockopt_find(sk, pf, val, get);
106         if (IS_ERR(ops))
107                 return PTR_ERR(ops);
108
109         if (get)
110                 ret = ops->get(sk, val, opt, len);
111         else
112                 ret = ops->set(sk, val, opt, *len);
113
114         module_put(ops->owner);
115         return ret;
116 }
117
118 int nf_setsockopt(struct sock *sk, int pf, int val, char __user *opt,
119                   int len)
120 {
121         return nf_sockopt(sk, pf, val, opt, &len, 0);
122 }
123 EXPORT_SYMBOL(nf_setsockopt);
124
125 int nf_getsockopt(struct sock *sk, int pf, int val, char __user *opt, int *len)
126 {
127         return nf_sockopt(sk, pf, val, opt, len, 1);
128 }
129 EXPORT_SYMBOL(nf_getsockopt);
130
131 #ifdef CONFIG_COMPAT
132 static int compat_nf_sockopt(struct sock *sk, int pf, int val,
133                              char __user *opt, int *len, int get)
134 {
135         struct nf_sockopt_ops *ops;
136         int ret;
137
138         ops = nf_sockopt_find(sk, pf, val, get);
139         if (IS_ERR(ops))
140                 return PTR_ERR(ops);
141
142         if (get) {
143                 if (ops->compat_get)
144                         ret = ops->compat_get(sk, val, opt, len);
145                 else
146                         ret = ops->get(sk, val, opt, len);
147         } else {
148                 if (ops->compat_set)
149                         ret = ops->compat_set(sk, val, opt, *len);
150                 else
151                         ret = ops->set(sk, val, opt, *len);
152         }
153
154         module_put(ops->owner);
155         return ret;
156 }
157
158 int compat_nf_setsockopt(struct sock *sk, int pf,
159                 int val, char __user *opt, int len)
160 {
161         return compat_nf_sockopt(sk, pf, val, opt, &len, 0);
162 }
163 EXPORT_SYMBOL(compat_nf_setsockopt);
164
165 int compat_nf_getsockopt(struct sock *sk, int pf,
166                 int val, char __user *opt, int *len)
167 {
168         return compat_nf_sockopt(sk, pf, val, opt, len, 1);
169 }
170 EXPORT_SYMBOL(compat_nf_getsockopt);
171 #endif