blob: ab07711cf2f40e161cd0602678ce3b4555ab68b3 [file] [log] [blame]
Remi Denis-Courmontba113a92008-09-22 20:05:19 -07001/*
2 * File: socket.c
3 *
4 * Phonet sockets
5 *
6 * Copyright (C) 2008 Nokia Corporation.
7 *
8 * Contact: Remi Denis-Courmont <remi.denis-courmont@nokia.com>
9 * Original author: Sakari Ailus <sakari.ailus@nokia.com>
10 *
11 * This program is free software; you can redistribute it and/or
12 * modify it under the terms of the GNU General Public License
13 * version 2 as published by the Free Software Foundation.
14 *
15 * This program is distributed in the hope that it will be useful, but
16 * WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18 * General Public License for more details.
19 *
20 * You should have received a copy of the GNU General Public License
21 * along with this program; if not, write to the Free Software
22 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
23 * 02110-1301 USA
24 */
25
Tejun Heo5a0e3ad2010-03-24 17:04:11 +090026#include <linux/gfp.h>
Remi Denis-Courmontba113a92008-09-22 20:05:19 -070027#include <linux/kernel.h>
28#include <linux/net.h>
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -070029#include <linux/poll.h>
Remi Denis-Courmontba113a92008-09-22 20:05:19 -070030#include <net/sock.h>
31#include <net/tcp_states.h>
32
33#include <linux/phonet.h>
34#include <net/phonet/phonet.h>
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -070035#include <net/phonet/pep.h>
Remi Denis-Courmontba113a92008-09-22 20:05:19 -070036#include <net/phonet/pn_dev.h>
37
38static int pn_socket_release(struct socket *sock)
39{
40 struct sock *sk = sock->sk;
41
42 if (sk) {
43 sock->sk = NULL;
44 sk->sk_prot->close(sk, 0);
45 }
46 return 0;
47}
48
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +000049#define PN_HASHSIZE 16
50#define PN_HASHMASK (PN_HASHSIZE-1)
51
52
Remi Denis-Courmontba113a92008-09-22 20:05:19 -070053static struct {
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +000054 struct hlist_head hlist[PN_HASHSIZE];
Rémi Denis-Courmont44f4d5a2011-04-12 23:27:36 +000055 struct mutex lock;
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +000056} pnsocks;
57
58void __init pn_sock_init(void)
59{
60 unsigned i;
61
62 for (i = 0; i < PN_HASHSIZE; i++)
63 INIT_HLIST_HEAD(pnsocks.hlist + i);
Rémi Denis-Courmont44f4d5a2011-04-12 23:27:36 +000064 mutex_init(&pnsocks.lock);
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +000065}
66
67static struct hlist_head *pn_hash_list(u16 obj)
68{
69 return pnsocks.hlist + (obj & PN_HASHMASK);
70}
Remi Denis-Courmontba113a92008-09-22 20:05:19 -070071
72/*
73 * Find address based on socket address, match only certain fields.
74 * Also grab sock if it was found. Remember to sock_put it later.
75 */
Rémi Denis-Courmont52404882008-12-03 15:42:56 -080076struct sock *pn_find_sock_by_sa(struct net *net, const struct sockaddr_pn *spn)
Remi Denis-Courmontba113a92008-09-22 20:05:19 -070077{
78 struct hlist_node *node;
79 struct sock *sknode;
80 struct sock *rval = NULL;
81 u16 obj = pn_sockaddr_get_object(spn);
82 u8 res = spn->spn_resource;
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +000083 struct hlist_head *hlist = pn_hash_list(obj);
Remi Denis-Courmontba113a92008-09-22 20:05:19 -070084
Rémi Denis-Courmont44f4d5a2011-04-12 23:27:36 +000085 rcu_read_lock();
86 sk_for_each_rcu(sknode, node, hlist) {
Remi Denis-Courmontba113a92008-09-22 20:05:19 -070087 struct pn_sock *pn = pn_sk(sknode);
88 BUG_ON(!pn->sobject); /* unbound socket */
89
Rémi Denis-Courmont52404882008-12-03 15:42:56 -080090 if (!net_eq(sock_net(sknode), net))
91 continue;
Remi Denis-Courmontba113a92008-09-22 20:05:19 -070092 if (pn_port(obj)) {
93 /* Look up socket by port */
94 if (pn_port(pn->sobject) != pn_port(obj))
95 continue;
96 } else {
97 /* If port is zero, look up by resource */
98 if (pn->resource != res)
99 continue;
100 }
Joe Perchesf64f9e72009-11-29 16:55:45 -0800101 if (pn_addr(pn->sobject) &&
102 pn_addr(pn->sobject) != pn_addr(obj))
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700103 continue;
104
105 rval = sknode;
106 sock_hold(sknode);
107 break;
108 }
Rémi Denis-Courmont44f4d5a2011-04-12 23:27:36 +0000109 rcu_read_unlock();
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700110
111 return rval;
Rémi Denis-Courmontf14001f2009-10-14 00:48:27 +0000112}
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700113
Rémi Denis-Courmontf14001f2009-10-14 00:48:27 +0000114/* Deliver a broadcast packet (only in bottom-half) */
115void pn_deliver_sock_broadcast(struct net *net, struct sk_buff *skb)
116{
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +0000117 struct hlist_head *hlist = pnsocks.hlist;
118 unsigned h;
Rémi Denis-Courmontf14001f2009-10-14 00:48:27 +0000119
Rémi Denis-Courmont44f4d5a2011-04-12 23:27:36 +0000120 rcu_read_lock();
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +0000121 for (h = 0; h < PN_HASHSIZE; h++) {
122 struct hlist_node *node;
123 struct sock *sknode;
Rémi Denis-Courmontf14001f2009-10-14 00:48:27 +0000124
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +0000125 sk_for_each(sknode, node, hlist) {
126 struct sk_buff *clone;
Rémi Denis-Courmontf14001f2009-10-14 00:48:27 +0000127
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +0000128 if (!net_eq(sock_net(sknode), net))
129 continue;
130 if (!sock_flag(sknode, SOCK_BROADCAST))
131 continue;
132
133 clone = skb_clone(skb, GFP_ATOMIC);
134 if (clone) {
135 sock_hold(sknode);
136 sk_receive_skb(sknode, clone, 0);
137 }
Rémi Denis-Courmont21912d12009-10-15 12:29:14 -0700138 }
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +0000139 hlist++;
Rémi Denis-Courmontf14001f2009-10-14 00:48:27 +0000140 }
Rémi Denis-Courmont44f4d5a2011-04-12 23:27:36 +0000141 rcu_read_unlock();
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700142}
143
144void pn_sock_hash(struct sock *sk)
145{
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +0000146 struct hlist_head *hlist = pn_hash_list(pn_sk(sk)->sobject);
147
Rémi Denis-Courmont44f4d5a2011-04-12 23:27:36 +0000148 mutex_lock(&pnsocks.lock);
149 sk_add_node_rcu(sk, hlist);
150 mutex_unlock(&pnsocks.lock);
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700151}
152EXPORT_SYMBOL(pn_sock_hash);
153
154void pn_sock_unhash(struct sock *sk)
155{
Rémi Denis-Courmont44f4d5a2011-04-12 23:27:36 +0000156 mutex_lock(&pnsocks.lock);
157 sk_del_node_init_rcu(sk);
158 mutex_unlock(&pnsocks.lock);
Rémi Denis-Courmont7417fa82010-09-15 12:30:12 +0000159 pn_sock_unbind_all_res(sk);
Rémi Denis-Courmont44f4d5a2011-04-12 23:27:36 +0000160 synchronize_rcu();
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700161}
162EXPORT_SYMBOL(pn_sock_unhash);
163
Rémi Denis-Courmont582b0b62009-09-23 03:17:10 +0000164static DEFINE_MUTEX(port_mutex);
165
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700166static int pn_socket_bind(struct socket *sock, struct sockaddr *addr, int len)
167{
168 struct sock *sk = sock->sk;
169 struct pn_sock *pn = pn_sk(sk);
170 struct sockaddr_pn *spn = (struct sockaddr_pn *)addr;
171 int err;
172 u16 handle;
173 u8 saddr;
174
175 if (sk->sk_prot->bind)
176 return sk->sk_prot->bind(sk, addr, len);
177
178 if (len < sizeof(struct sockaddr_pn))
179 return -EINVAL;
180 if (spn->spn_family != AF_PHONET)
181 return -EAFNOSUPPORT;
182
183 handle = pn_sockaddr_get_object((struct sockaddr_pn *)addr);
184 saddr = pn_addr(handle);
Rémi Denis-Courmont52404882008-12-03 15:42:56 -0800185 if (saddr && phonet_address_lookup(sock_net(sk), saddr))
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700186 return -EADDRNOTAVAIL;
187
188 lock_sock(sk);
189 if (sk->sk_state != TCP_CLOSE || pn_port(pn->sobject)) {
190 err = -EINVAL; /* attempt to rebind */
191 goto out;
192 }
Rémi Denis-Courmont582b0b62009-09-23 03:17:10 +0000193 WARN_ON(sk_hashed(sk));
194 mutex_lock(&port_mutex);
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700195 err = sk->sk_prot->get_port(sk, pn_port(handle));
196 if (err)
Rémi Denis-Courmont582b0b62009-09-23 03:17:10 +0000197 goto out_port;
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700198
199 /* get_port() sets the port, bind() sets the address if applicable */
200 pn->sobject = pn_object(saddr, pn_port(pn->sobject));
201 pn->resource = spn->spn_resource;
202
203 /* Enable RX on the socket */
204 sk->sk_prot->hash(sk);
Rémi Denis-Courmont582b0b62009-09-23 03:17:10 +0000205out_port:
206 mutex_unlock(&port_mutex);
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700207out:
208 release_sock(sk);
209 return err;
210}
211
212static int pn_socket_autobind(struct socket *sock)
213{
214 struct sockaddr_pn sa;
215 int err;
216
217 memset(&sa, 0, sizeof(sa));
218 sa.spn_family = AF_PHONET;
219 err = pn_socket_bind(sock, (struct sockaddr *)&sa,
220 sizeof(struct sockaddr_pn));
221 if (err != -EINVAL)
222 return err;
223 BUG_ON(!pn_port(pn_sk(sock->sk)->sobject));
224 return 0; /* socket was already bound */
225}
226
Kumar Sanghvib3d62552010-10-12 20:14:43 +0000227static int pn_socket_connect(struct socket *sock, struct sockaddr *addr,
228 int len, int flags)
229{
230 struct sock *sk = sock->sk;
Rémi Denis-Courmont297edb62011-03-08 22:44:12 +0000231 struct pn_sock *pn = pn_sk(sk);
Kumar Sanghvib3d62552010-10-12 20:14:43 +0000232 struct sockaddr_pn *spn = (struct sockaddr_pn *)addr;
Rémi Denis-Courmont297edb62011-03-08 22:44:12 +0000233 struct task_struct *tsk = current;
234 long timeo = sock_rcvtimeo(sk, flags & O_NONBLOCK);
Kumar Sanghvib3d62552010-10-12 20:14:43 +0000235 int err;
236
Rémi Denis-Courmont297edb62011-03-08 22:44:12 +0000237 if (pn_socket_autobind(sock))
238 return -ENOBUFS;
Kumar Sanghvib3d62552010-10-12 20:14:43 +0000239 if (len < sizeof(struct sockaddr_pn))
240 return -EINVAL;
241 if (spn->spn_family != AF_PHONET)
242 return -EAFNOSUPPORT;
243
244 lock_sock(sk);
245
246 switch (sock->state) {
247 case SS_UNCONNECTED:
Rémi Denis-Courmont297edb62011-03-08 22:44:12 +0000248 if (sk->sk_state != TCP_CLOSE) {
249 err = -EISCONN;
250 goto out;
251 }
Kumar Sanghvib3d62552010-10-12 20:14:43 +0000252 break;
253 case SS_CONNECTING:
Rémi Denis-Courmont297edb62011-03-08 22:44:12 +0000254 err = -EALREADY;
255 goto out;
256 default:
257 err = -EISCONN;
258 goto out;
Kumar Sanghvib3d62552010-10-12 20:14:43 +0000259 }
Kumar Sanghvib3d62552010-10-12 20:14:43 +0000260
Rémi Denis-Courmont297edb62011-03-08 22:44:12 +0000261 pn->dobject = pn_sockaddr_get_object(spn);
262 pn->resource = pn_sockaddr_get_resource(spn);
Kumar Sanghvib3d62552010-10-12 20:14:43 +0000263 sock->state = SS_CONNECTING;
Rémi Denis-Courmont297edb62011-03-08 22:44:12 +0000264
Kumar Sanghvib3d62552010-10-12 20:14:43 +0000265 err = sk->sk_prot->connect(sk, addr, len);
Rémi Denis-Courmont297edb62011-03-08 22:44:12 +0000266 if (err) {
Kumar Sanghvib3d62552010-10-12 20:14:43 +0000267 sock->state = SS_UNCONNECTED;
Rémi Denis-Courmont297edb62011-03-08 22:44:12 +0000268 pn->dobject = 0;
Kumar Sanghvib3d62552010-10-12 20:14:43 +0000269 goto out;
270 }
271
Rémi Denis-Courmont297edb62011-03-08 22:44:12 +0000272 while (sk->sk_state == TCP_SYN_SENT) {
273 DEFINE_WAIT(wait);
Kumar Sanghvib3d62552010-10-12 20:14:43 +0000274
Rémi Denis-Courmont297edb62011-03-08 22:44:12 +0000275 if (!timeo) {
276 err = -EINPROGRESS;
277 goto out;
278 }
279 if (signal_pending(tsk)) {
280 err = sock_intr_errno(timeo);
281 goto out;
282 }
Kumar Sanghvib3d62552010-10-12 20:14:43 +0000283
Rémi Denis-Courmont297edb62011-03-08 22:44:12 +0000284 prepare_to_wait_exclusive(sk_sleep(sk), &wait,
285 TASK_INTERRUPTIBLE);
286 release_sock(sk);
287 timeo = schedule_timeout(timeo);
288 lock_sock(sk);
289 finish_wait(sk_sleep(sk), &wait);
Kumar Sanghvib3d62552010-10-12 20:14:43 +0000290 }
Kumar Sanghvib3d62552010-10-12 20:14:43 +0000291
Rémi Denis-Courmont297edb62011-03-08 22:44:12 +0000292 if ((1 << sk->sk_state) & (TCPF_SYN_RECV|TCPF_ESTABLISHED))
293 err = 0;
294 else if (sk->sk_state == TCP_CLOSE_WAIT)
295 err = -ECONNRESET;
296 else
297 err = -ECONNREFUSED;
298 sock->state = err ? SS_UNCONNECTED : SS_CONNECTED;
Kumar Sanghvib3d62552010-10-12 20:14:43 +0000299out:
300 release_sock(sk);
301 return err;
302}
Kumar Sanghvib3d62552010-10-12 20:14:43 +0000303
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700304static int pn_socket_accept(struct socket *sock, struct socket *newsock,
305 int flags)
306{
307 struct sock *sk = sock->sk;
308 struct sock *newsk;
309 int err;
310
Rémi Denis-Courmontf7ae8d52011-03-08 22:44:10 +0000311 if (unlikely(sk->sk_state != TCP_LISTEN))
312 return -EINVAL;
313
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700314 newsk = sk->sk_prot->accept(sk, flags, &err);
315 if (!newsk)
316 return err;
317
318 lock_sock(newsk);
319 sock_graft(newsk, newsock);
320 newsock->state = SS_CONNECTED;
321 release_sock(newsk);
322 return 0;
323}
324
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700325static int pn_socket_getname(struct socket *sock, struct sockaddr *addr,
326 int *sockaddr_len, int peer)
327{
328 struct sock *sk = sock->sk;
329 struct pn_sock *pn = pn_sk(sk);
330
331 memset(addr, 0, sizeof(struct sockaddr_pn));
332 addr->sa_family = AF_PHONET;
333 if (!peer) /* Race with bind() here is userland's problem. */
334 pn_sockaddr_set_object((struct sockaddr_pn *)addr,
335 pn->sobject);
336
337 *sockaddr_len = sizeof(struct sockaddr_pn);
338 return 0;
339}
340
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700341static unsigned int pn_socket_poll(struct file *file, struct socket *sock,
342 poll_table *wait)
343{
344 struct sock *sk = sock->sk;
345 struct pep_sock *pn = pep_sk(sk);
346 unsigned int mask = 0;
347
Eric Dumazet43815482010-04-29 11:01:49 +0000348 poll_wait(file, sk_sleep(sk), wait);
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700349
Rémi Denis-Courmontf7ae8d52011-03-08 22:44:10 +0000350 if (sk->sk_state == TCP_CLOSE)
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700351 return POLLERR;
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700352 if (!skb_queue_empty(&sk->sk_receive_queue))
353 mask |= POLLIN | POLLRDNORM;
Rémi Denis-Courmontc41bd97f82008-10-05 11:15:43 -0700354 if (!skb_queue_empty(&pn->ctrlreq_queue))
355 mask |= POLLPRI;
356 if (!mask && sk->sk_state == TCP_CLOSE_WAIT)
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700357 return POLLHUP;
358
Rémi Denis-Courmont01b38602010-08-30 12:57:05 +0000359 if (sk->sk_state == TCP_ESTABLISHED &&
360 atomic_read(&sk->sk_wmem_alloc) < sk->sk_sndbuf &&
361 atomic_read(&pn->tx_credits))
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700362 mask |= POLLOUT | POLLWRNORM | POLLWRBAND;
363
364 return mask;
365}
366
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700367static int pn_socket_ioctl(struct socket *sock, unsigned int cmd,
368 unsigned long arg)
369{
370 struct sock *sk = sock->sk;
371 struct pn_sock *pn = pn_sk(sk);
372
373 if (cmd == SIOCPNGETOBJECT) {
374 struct net_device *dev;
375 u16 handle;
376 u8 saddr;
377
378 if (get_user(handle, (__u16 __user *)arg))
379 return -EFAULT;
380
381 lock_sock(sk);
382 if (sk->sk_bound_dev_if)
383 dev = dev_get_by_index(sock_net(sk),
384 sk->sk_bound_dev_if);
385 else
386 dev = phonet_device_get(sock_net(sk));
387 if (dev && (dev->flags & IFF_UP))
388 saddr = phonet_address_get(dev, pn_addr(handle));
389 else
390 saddr = PN_NO_ADDR;
391 release_sock(sk);
392
393 if (dev)
394 dev_put(dev);
395 if (saddr == PN_NO_ADDR)
396 return -EHOSTUNREACH;
397
398 handle = pn_object(saddr, pn_port(pn->sobject));
399 return put_user(handle, (__u16 __user *)arg);
400 }
401
402 return sk->sk_prot->ioctl(sk, cmd, arg);
403}
404
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700405static int pn_socket_listen(struct socket *sock, int backlog)
406{
407 struct sock *sk = sock->sk;
408 int err = 0;
409
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700410 if (pn_socket_autobind(sock))
411 return -ENOBUFS;
412
413 lock_sock(sk);
Rémi Denis-Courmont96241542011-02-24 23:14:56 +0000414 if (sock->state != SS_UNCONNECTED) {
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700415 err = -EINVAL;
416 goto out;
417 }
418
Rémi Denis-Courmont96241542011-02-24 23:14:56 +0000419 if (sk->sk_state != TCP_LISTEN) {
420 sk->sk_state = TCP_LISTEN;
421 sk->sk_ack_backlog = 0;
422 }
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700423 sk->sk_max_ack_backlog = backlog;
424out:
425 release_sock(sk);
426 return err;
427}
428
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700429static int pn_socket_sendmsg(struct kiocb *iocb, struct socket *sock,
430 struct msghdr *m, size_t total_len)
431{
432 struct sock *sk = sock->sk;
433
434 if (pn_socket_autobind(sock))
435 return -EAGAIN;
436
437 return sk->sk_prot->sendmsg(iocb, sk, m, total_len);
438}
439
440const struct proto_ops phonet_dgram_ops = {
441 .family = AF_PHONET,
442 .owner = THIS_MODULE,
443 .release = pn_socket_release,
444 .bind = pn_socket_bind,
445 .connect = sock_no_connect,
446 .socketpair = sock_no_socketpair,
447 .accept = sock_no_accept,
448 .getname = pn_socket_getname,
449 .poll = datagram_poll,
450 .ioctl = pn_socket_ioctl,
451 .listen = sock_no_listen,
452 .shutdown = sock_no_shutdown,
453 .setsockopt = sock_no_setsockopt,
454 .getsockopt = sock_no_getsockopt,
455#ifdef CONFIG_COMPAT
456 .compat_setsockopt = sock_no_setsockopt,
457 .compat_getsockopt = sock_no_getsockopt,
458#endif
459 .sendmsg = pn_socket_sendmsg,
460 .recvmsg = sock_common_recvmsg,
461 .mmap = sock_no_mmap,
462 .sendpage = sock_no_sendpage,
463};
464
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700465const struct proto_ops phonet_stream_ops = {
466 .family = AF_PHONET,
467 .owner = THIS_MODULE,
468 .release = pn_socket_release,
469 .bind = pn_socket_bind,
Kumar Sanghvib3d62552010-10-12 20:14:43 +0000470 .connect = pn_socket_connect,
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700471 .socketpair = sock_no_socketpair,
472 .accept = pn_socket_accept,
473 .getname = pn_socket_getname,
474 .poll = pn_socket_poll,
475 .ioctl = pn_socket_ioctl,
476 .listen = pn_socket_listen,
477 .shutdown = sock_no_shutdown,
Rémi Denis-Courmont02a47612008-10-05 11:16:16 -0700478 .setsockopt = sock_common_setsockopt,
479 .getsockopt = sock_common_getsockopt,
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700480#ifdef CONFIG_COMPAT
Rémi Denis-Courmont02a47612008-10-05 11:16:16 -0700481 .compat_setsockopt = compat_sock_common_setsockopt,
482 .compat_getsockopt = compat_sock_common_getsockopt,
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700483#endif
484 .sendmsg = pn_socket_sendmsg,
485 .recvmsg = sock_common_recvmsg,
486 .mmap = sock_no_mmap,
487 .sendpage = sock_no_sendpage,
488};
489EXPORT_SYMBOL(phonet_stream_ops);
490
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700491/* allocate port for a socket */
492int pn_sock_get_port(struct sock *sk, unsigned short sport)
493{
494 static int port_cur;
Rémi Denis-Courmont52404882008-12-03 15:42:56 -0800495 struct net *net = sock_net(sk);
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700496 struct pn_sock *pn = pn_sk(sk);
497 struct sockaddr_pn try_sa;
498 struct sock *tmpsk;
499
500 memset(&try_sa, 0, sizeof(struct sockaddr_pn));
501 try_sa.spn_family = AF_PHONET;
Rémi Denis-Courmont582b0b62009-09-23 03:17:10 +0000502 WARN_ON(!mutex_is_locked(&port_mutex));
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700503 if (!sport) {
504 /* search free port */
Remi Denis-Courmont87ab4e22008-09-22 20:08:39 -0700505 int port, pmin, pmax;
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700506
Remi Denis-Courmont87ab4e22008-09-22 20:08:39 -0700507 phonet_get_local_port_range(&pmin, &pmax);
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700508 for (port = pmin; port <= pmax; port++) {
509 port_cur++;
510 if (port_cur < pmin || port_cur > pmax)
511 port_cur = pmin;
512
513 pn_sockaddr_set_port(&try_sa, port_cur);
Rémi Denis-Courmont52404882008-12-03 15:42:56 -0800514 tmpsk = pn_find_sock_by_sa(net, &try_sa);
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700515 if (tmpsk == NULL) {
516 sport = port_cur;
517 goto found;
518 } else
519 sock_put(tmpsk);
520 }
521 } else {
522 /* try to find specific port */
523 pn_sockaddr_set_port(&try_sa, sport);
Rémi Denis-Courmont52404882008-12-03 15:42:56 -0800524 tmpsk = pn_find_sock_by_sa(net, &try_sa);
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700525 if (tmpsk == NULL)
526 /* No sock there! We can use that port... */
527 goto found;
528 else
529 sock_put(tmpsk);
530 }
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700531 /* the port must be in use already */
532 return -EADDRINUSE;
533
534found:
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700535 pn->sobject = pn_object(pn_addr(pn->sobject), sport);
536 return 0;
537}
538EXPORT_SYMBOL(pn_sock_get_port);
Rémi Denis-Courmontc1dc13e2009-07-21 01:57:57 +0000539
Randy Dunlapae6e2ae2009-08-17 10:35:49 +0000540#ifdef CONFIG_PROC_FS
Rémi Denis-Courmontc1dc13e2009-07-21 01:57:57 +0000541static struct sock *pn_sock_get_idx(struct seq_file *seq, loff_t pos)
542{
543 struct net *net = seq_file_net(seq);
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +0000544 struct hlist_head *hlist = pnsocks.hlist;
Rémi Denis-Courmontc1dc13e2009-07-21 01:57:57 +0000545 struct hlist_node *node;
546 struct sock *sknode;
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +0000547 unsigned h;
Rémi Denis-Courmontc1dc13e2009-07-21 01:57:57 +0000548
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +0000549 for (h = 0; h < PN_HASHSIZE; h++) {
Rémi Denis-Courmont44f4d5a2011-04-12 23:27:36 +0000550 sk_for_each_rcu(sknode, node, hlist) {
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +0000551 if (!net_eq(net, sock_net(sknode)))
552 continue;
553 if (!pos)
554 return sknode;
555 pos--;
556 }
557 hlist++;
Rémi Denis-Courmontc1dc13e2009-07-21 01:57:57 +0000558 }
559 return NULL;
560}
561
562static struct sock *pn_sock_get_next(struct seq_file *seq, struct sock *sk)
563{
564 struct net *net = seq_file_net(seq);
565
566 do
567 sk = sk_next(sk);
568 while (sk && !net_eq(net, sock_net(sk)));
569
570 return sk;
571}
572
573static void *pn_sock_seq_start(struct seq_file *seq, loff_t *pos)
Rémi Denis-Courmont44f4d5a2011-04-12 23:27:36 +0000574 __acquires(rcu)
Rémi Denis-Courmontc1dc13e2009-07-21 01:57:57 +0000575{
Rémi Denis-Courmont44f4d5a2011-04-12 23:27:36 +0000576 rcu_read_lock();
Rémi Denis-Courmontc1dc13e2009-07-21 01:57:57 +0000577 return *pos ? pn_sock_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
578}
579
580static void *pn_sock_seq_next(struct seq_file *seq, void *v, loff_t *pos)
581{
582 struct sock *sk;
583
584 if (v == SEQ_START_TOKEN)
585 sk = pn_sock_get_idx(seq, 0);
586 else
587 sk = pn_sock_get_next(seq, v);
588 (*pos)++;
589 return sk;
590}
591
592static void pn_sock_seq_stop(struct seq_file *seq, void *v)
Rémi Denis-Courmont44f4d5a2011-04-12 23:27:36 +0000593 __releases(rcu)
Rémi Denis-Courmontc1dc13e2009-07-21 01:57:57 +0000594{
Rémi Denis-Courmont44f4d5a2011-04-12 23:27:36 +0000595 rcu_read_unlock();
Rémi Denis-Courmontc1dc13e2009-07-21 01:57:57 +0000596}
597
598static int pn_sock_seq_show(struct seq_file *seq, void *v)
599{
600 int len;
601
602 if (v == SEQ_START_TOKEN)
603 seq_printf(seq, "%s%n", "pt loc rem rs st tx_queue rx_queue "
604 " uid inode ref pointer drops", &len);
605 else {
606 struct sock *sk = v;
607 struct pn_sock *pn = pn_sk(sk);
608
609 seq_printf(seq, "%2d %04X:%04X:%02X %02X %08X:%08X %5d %lu "
Dan Rosenberg71338aa2011-05-23 12:17:35 +0000610 "%d %pK %d%n",
Rémi Denis-Courmonta8059512011-02-24 23:14:57 +0000611 sk->sk_protocol, pn->sobject, pn->dobject,
612 pn->resource, sk->sk_state,
Rémi Denis-Courmontc1dc13e2009-07-21 01:57:57 +0000613 sk_wmem_alloc_get(sk), sk_rmem_alloc_get(sk),
614 sock_i_uid(sk), sock_i_ino(sk),
615 atomic_read(&sk->sk_refcnt), sk,
616 atomic_read(&sk->sk_drops), &len);
617 }
618 seq_printf(seq, "%*s\n", 127 - len, "");
619 return 0;
620}
621
622static const struct seq_operations pn_sock_seq_ops = {
623 .start = pn_sock_seq_start,
624 .next = pn_sock_seq_next,
625 .stop = pn_sock_seq_stop,
626 .show = pn_sock_seq_show,
627};
628
629static int pn_sock_open(struct inode *inode, struct file *file)
630{
Rémi Denis-Courmontcb7d9e72009-08-11 03:12:07 +0000631 return seq_open_net(inode, file, &pn_sock_seq_ops,
632 sizeof(struct seq_net_private));
Rémi Denis-Courmontc1dc13e2009-07-21 01:57:57 +0000633}
634
635const struct file_operations pn_sock_seq_fops = {
636 .owner = THIS_MODULE,
637 .open = pn_sock_open,
638 .read = seq_read,
639 .llseek = seq_lseek,
Rémi Denis-Courmontcb7d9e72009-08-11 03:12:07 +0000640 .release = seq_release_net,
Rémi Denis-Courmontc1dc13e2009-07-21 01:57:57 +0000641};
Randy Dunlapae6e2ae2009-08-17 10:35:49 +0000642#endif
Rémi Denis-Courmont4e3d16c2010-09-15 12:30:11 +0000643
644static struct {
645 struct sock *sk[256];
646} pnres;
647
648/*
649 * Find and hold socket based on resource.
650 */
651struct sock *pn_find_sock_by_res(struct net *net, u8 res)
652{
653 struct sock *sk;
654
655 if (!net_eq(net, &init_net))
656 return NULL;
657
658 rcu_read_lock();
659 sk = rcu_dereference(pnres.sk[res]);
660 if (sk)
661 sock_hold(sk);
662 rcu_read_unlock();
663 return sk;
664}
665
666static DEFINE_MUTEX(resource_mutex);
667
668int pn_sock_bind_res(struct sock *sk, u8 res)
669{
670 int ret = -EADDRINUSE;
671
672 if (!net_eq(sock_net(sk), &init_net))
673 return -ENOIOCTLCMD;
674 if (!capable(CAP_SYS_ADMIN))
675 return -EPERM;
676 if (pn_socket_autobind(sk->sk_socket))
677 return -EAGAIN;
678
679 mutex_lock(&resource_mutex);
680 if (pnres.sk[res] == NULL) {
681 sock_hold(sk);
682 rcu_assign_pointer(pnres.sk[res], sk);
683 ret = 0;
684 }
685 mutex_unlock(&resource_mutex);
686 return ret;
687}
688
689int pn_sock_unbind_res(struct sock *sk, u8 res)
690{
691 int ret = -ENOENT;
692
693 if (!capable(CAP_SYS_ADMIN))
694 return -EPERM;
695
696 mutex_lock(&resource_mutex);
697 if (pnres.sk[res] == sk) {
698 rcu_assign_pointer(pnres.sk[res], NULL);
699 ret = 0;
700 }
701 mutex_unlock(&resource_mutex);
702
703 if (ret == 0) {
704 synchronize_rcu();
705 sock_put(sk);
706 }
707 return ret;
708}
709
710void pn_sock_unbind_all_res(struct sock *sk)
711{
712 unsigned res, match = 0;
713
714 mutex_lock(&resource_mutex);
715 for (res = 0; res < 256; res++) {
716 if (pnres.sk[res] == sk) {
717 rcu_assign_pointer(pnres.sk[res], NULL);
718 match++;
719 }
720 }
721 mutex_unlock(&resource_mutex);
722
Rémi Denis-Courmont4e3d16c2010-09-15 12:30:11 +0000723 while (match > 0) {
Rémi Denis-Courmont44f4d5a2011-04-12 23:27:36 +0000724 __sock_put(sk);
Rémi Denis-Courmont4e3d16c2010-09-15 12:30:11 +0000725 match--;
726 }
Rémi Denis-Courmont44f4d5a2011-04-12 23:27:36 +0000727 /* Caller is responsible for RCU sync before final sock_put() */
Rémi Denis-Courmont4e3d16c2010-09-15 12:30:11 +0000728}
Rémi Denis-Courmont507215f2010-09-15 12:30:14 +0000729
730#ifdef CONFIG_PROC_FS
731static struct sock **pn_res_get_idx(struct seq_file *seq, loff_t pos)
732{
733 struct net *net = seq_file_net(seq);
734 unsigned i;
735
736 if (!net_eq(net, &init_net))
737 return NULL;
738
739 for (i = 0; i < 256; i++) {
740 if (pnres.sk[i] == NULL)
741 continue;
742 if (!pos)
743 return pnres.sk + i;
744 pos--;
745 }
746 return NULL;
747}
748
749static struct sock **pn_res_get_next(struct seq_file *seq, struct sock **sk)
750{
751 struct net *net = seq_file_net(seq);
752 unsigned i;
753
754 BUG_ON(!net_eq(net, &init_net));
755
756 for (i = (sk - pnres.sk) + 1; i < 256; i++)
757 if (pnres.sk[i])
758 return pnres.sk + i;
759 return NULL;
760}
761
762static void *pn_res_seq_start(struct seq_file *seq, loff_t *pos)
763 __acquires(resource_mutex)
764{
765 mutex_lock(&resource_mutex);
766 return *pos ? pn_res_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
767}
768
769static void *pn_res_seq_next(struct seq_file *seq, void *v, loff_t *pos)
770{
771 struct sock **sk;
772
773 if (v == SEQ_START_TOKEN)
774 sk = pn_res_get_idx(seq, 0);
775 else
776 sk = pn_res_get_next(seq, v);
777 (*pos)++;
778 return sk;
779}
780
781static void pn_res_seq_stop(struct seq_file *seq, void *v)
782 __releases(resource_mutex)
783{
784 mutex_unlock(&resource_mutex);
785}
786
787static int pn_res_seq_show(struct seq_file *seq, void *v)
788{
789 int len;
790
791 if (v == SEQ_START_TOKEN)
792 seq_printf(seq, "%s%n", "rs uid inode", &len);
793 else {
794 struct sock **psk = v;
795 struct sock *sk = *psk;
796
797 seq_printf(seq, "%02X %5d %lu%n",
David S. Miller9e0064a2010-09-15 21:34:41 -0700798 (int) (psk - pnres.sk), sock_i_uid(sk),
799 sock_i_ino(sk), &len);
Rémi Denis-Courmont507215f2010-09-15 12:30:14 +0000800 }
801 seq_printf(seq, "%*s\n", 63 - len, "");
802 return 0;
803}
804
805static const struct seq_operations pn_res_seq_ops = {
806 .start = pn_res_seq_start,
807 .next = pn_res_seq_next,
808 .stop = pn_res_seq_stop,
809 .show = pn_res_seq_show,
810};
811
812static int pn_res_open(struct inode *inode, struct file *file)
813{
814 return seq_open_net(inode, file, &pn_res_seq_ops,
815 sizeof(struct seq_net_private));
816}
817
818const struct file_operations pn_res_seq_fops = {
819 .owner = THIS_MODULE,
820 .open = pn_res_open,
821 .read = seq_read,
822 .llseek = seq_lseek,
823 .release = seq_release_net,
824};
825#endif