blob: 25f746d20c1f2dcbf42826cf33f07a35e44ace1c [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];
Remi Denis-Courmontba113a92008-09-22 20:05:19 -070055 spinlock_t 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);
64 spin_lock_init(&pnsocks.lock);
65}
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
85 spin_lock_bh(&pnsocks.lock);
86
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +000087 sk_for_each(sknode, node, hlist) {
Remi Denis-Courmontba113a92008-09-22 20:05:19 -070088 struct pn_sock *pn = pn_sk(sknode);
89 BUG_ON(!pn->sobject); /* unbound socket */
90
Rémi Denis-Courmont52404882008-12-03 15:42:56 -080091 if (!net_eq(sock_net(sknode), net))
92 continue;
Remi Denis-Courmontba113a92008-09-22 20:05:19 -070093 if (pn_port(obj)) {
94 /* Look up socket by port */
95 if (pn_port(pn->sobject) != pn_port(obj))
96 continue;
97 } else {
98 /* If port is zero, look up by resource */
99 if (pn->resource != res)
100 continue;
101 }
Joe Perchesf64f9e72009-11-29 16:55:45 -0800102 if (pn_addr(pn->sobject) &&
103 pn_addr(pn->sobject) != pn_addr(obj))
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700104 continue;
105
106 rval = sknode;
107 sock_hold(sknode);
108 break;
109 }
110
111 spin_unlock_bh(&pnsocks.lock);
112
113 return rval;
Rémi Denis-Courmontf14001f2009-10-14 00:48:27 +0000114}
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700115
Rémi Denis-Courmontf14001f2009-10-14 00:48:27 +0000116/* Deliver a broadcast packet (only in bottom-half) */
117void pn_deliver_sock_broadcast(struct net *net, struct sk_buff *skb)
118{
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +0000119 struct hlist_head *hlist = pnsocks.hlist;
120 unsigned h;
Rémi Denis-Courmontf14001f2009-10-14 00:48:27 +0000121
122 spin_lock(&pnsocks.lock);
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +0000123 for (h = 0; h < PN_HASHSIZE; h++) {
124 struct hlist_node *node;
125 struct sock *sknode;
Rémi Denis-Courmontf14001f2009-10-14 00:48:27 +0000126
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +0000127 sk_for_each(sknode, node, hlist) {
128 struct sk_buff *clone;
Rémi Denis-Courmontf14001f2009-10-14 00:48:27 +0000129
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +0000130 if (!net_eq(sock_net(sknode), net))
131 continue;
132 if (!sock_flag(sknode, SOCK_BROADCAST))
133 continue;
134
135 clone = skb_clone(skb, GFP_ATOMIC);
136 if (clone) {
137 sock_hold(sknode);
138 sk_receive_skb(sknode, clone, 0);
139 }
Rémi Denis-Courmont21912d12009-10-15 12:29:14 -0700140 }
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +0000141 hlist++;
Rémi Denis-Courmontf14001f2009-10-14 00:48:27 +0000142 }
143 spin_unlock(&pnsocks.lock);
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700144}
145
146void pn_sock_hash(struct sock *sk)
147{
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +0000148 struct hlist_head *hlist = pn_hash_list(pn_sk(sk)->sobject);
149
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700150 spin_lock_bh(&pnsocks.lock);
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +0000151 sk_add_node(sk, hlist);
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700152 spin_unlock_bh(&pnsocks.lock);
153}
154EXPORT_SYMBOL(pn_sock_hash);
155
156void pn_sock_unhash(struct sock *sk)
157{
158 spin_lock_bh(&pnsocks.lock);
159 sk_del_node_init(sk);
160 spin_unlock_bh(&pnsocks.lock);
Rémi Denis-Courmont7417fa82010-09-15 12:30:12 +0000161 pn_sock_unbind_all_res(sk);
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700162}
163EXPORT_SYMBOL(pn_sock_unhash);
164
Rémi Denis-Courmont582b0b62009-09-23 03:17:10 +0000165static DEFINE_MUTEX(port_mutex);
166
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700167static int pn_socket_bind(struct socket *sock, struct sockaddr *addr, int len)
168{
169 struct sock *sk = sock->sk;
170 struct pn_sock *pn = pn_sk(sk);
171 struct sockaddr_pn *spn = (struct sockaddr_pn *)addr;
172 int err;
173 u16 handle;
174 u8 saddr;
175
176 if (sk->sk_prot->bind)
177 return sk->sk_prot->bind(sk, addr, len);
178
179 if (len < sizeof(struct sockaddr_pn))
180 return -EINVAL;
181 if (spn->spn_family != AF_PHONET)
182 return -EAFNOSUPPORT;
183
184 handle = pn_sockaddr_get_object((struct sockaddr_pn *)addr);
185 saddr = pn_addr(handle);
Rémi Denis-Courmont52404882008-12-03 15:42:56 -0800186 if (saddr && phonet_address_lookup(sock_net(sk), saddr))
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700187 return -EADDRNOTAVAIL;
188
189 lock_sock(sk);
190 if (sk->sk_state != TCP_CLOSE || pn_port(pn->sobject)) {
191 err = -EINVAL; /* attempt to rebind */
192 goto out;
193 }
Rémi Denis-Courmont582b0b62009-09-23 03:17:10 +0000194 WARN_ON(sk_hashed(sk));
195 mutex_lock(&port_mutex);
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700196 err = sk->sk_prot->get_port(sk, pn_port(handle));
197 if (err)
Rémi Denis-Courmont582b0b62009-09-23 03:17:10 +0000198 goto out_port;
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700199
200 /* get_port() sets the port, bind() sets the address if applicable */
201 pn->sobject = pn_object(saddr, pn_port(pn->sobject));
202 pn->resource = spn->spn_resource;
203
204 /* Enable RX on the socket */
205 sk->sk_prot->hash(sk);
Rémi Denis-Courmont582b0b62009-09-23 03:17:10 +0000206out_port:
207 mutex_unlock(&port_mutex);
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700208out:
209 release_sock(sk);
210 return err;
211}
212
213static int pn_socket_autobind(struct socket *sock)
214{
215 struct sockaddr_pn sa;
216 int err;
217
218 memset(&sa, 0, sizeof(sa));
219 sa.spn_family = AF_PHONET;
220 err = pn_socket_bind(sock, (struct sockaddr *)&sa,
221 sizeof(struct sockaddr_pn));
222 if (err != -EINVAL)
223 return err;
224 BUG_ON(!pn_port(pn_sk(sock->sk)->sobject));
225 return 0; /* socket was already bound */
226}
227
Kumar Sanghvib3d62552010-10-12 20:14:43 +0000228#ifdef CONFIG_PHONET_PIPECTRLR
229static int pn_socket_connect(struct socket *sock, struct sockaddr *addr,
230 int len, int flags)
231{
232 struct sock *sk = sock->sk;
233 struct sockaddr_pn *spn = (struct sockaddr_pn *)addr;
234 long timeo;
235 int err;
236
237 if (len < sizeof(struct sockaddr_pn))
238 return -EINVAL;
239 if (spn->spn_family != AF_PHONET)
240 return -EAFNOSUPPORT;
241
242 lock_sock(sk);
243
244 switch (sock->state) {
245 case SS_UNCONNECTED:
246 sk->sk_state = TCP_CLOSE;
247 break;
248 case SS_CONNECTING:
249 switch (sk->sk_state) {
250 case TCP_SYN_RECV:
251 sock->state = SS_CONNECTED;
252 err = -EISCONN;
253 goto out;
254 case TCP_CLOSE:
255 err = -EALREADY;
256 if (flags & O_NONBLOCK)
257 goto out;
258 goto wait_connect;
259 }
260 break;
261 case SS_CONNECTED:
262 switch (sk->sk_state) {
263 case TCP_SYN_RECV:
264 err = -EISCONN;
265 goto out;
266 case TCP_CLOSE:
267 sock->state = SS_UNCONNECTED;
268 break;
269 }
270 break;
271 case SS_DISCONNECTING:
272 case SS_FREE:
273 break;
274 }
275 sk->sk_state = TCP_CLOSE;
276 sk_stream_kill_queues(sk);
277
278 sock->state = SS_CONNECTING;
279 err = sk->sk_prot->connect(sk, addr, len);
280 if (err < 0) {
281 sock->state = SS_UNCONNECTED;
282 sk->sk_state = TCP_CLOSE;
283 goto out;
284 }
285
286 err = -EINPROGRESS;
287wait_connect:
288 if (sk->sk_state != TCP_SYN_RECV && (flags & O_NONBLOCK))
289 goto out;
290
291 timeo = sock_sndtimeo(sk, flags & O_NONBLOCK);
292 release_sock(sk);
293
294 err = -ERESTARTSYS;
295 timeo = wait_event_interruptible_timeout(*sk_sleep(sk),
296 sk->sk_state != TCP_CLOSE,
297 timeo);
298
299 lock_sock(sk);
300 if (timeo < 0)
301 goto out; /* -ERESTARTSYS */
302
303 err = -ETIMEDOUT;
304 if (timeo == 0 && sk->sk_state != TCP_SYN_RECV)
305 goto out;
306
307 if (sk->sk_state != TCP_SYN_RECV) {
308 sock->state = SS_UNCONNECTED;
309 err = sock_error(sk);
310 if (!err)
311 err = -ECONNREFUSED;
312 goto out;
313 }
314 sock->state = SS_CONNECTED;
315 err = 0;
316
317out:
318 release_sock(sk);
319 return err;
320}
321#endif
322
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700323static int pn_socket_accept(struct socket *sock, struct socket *newsock,
324 int flags)
325{
326 struct sock *sk = sock->sk;
327 struct sock *newsk;
328 int err;
329
330 newsk = sk->sk_prot->accept(sk, flags, &err);
331 if (!newsk)
332 return err;
333
334 lock_sock(newsk);
335 sock_graft(newsk, newsock);
336 newsock->state = SS_CONNECTED;
337 release_sock(newsk);
338 return 0;
339}
340
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700341static int pn_socket_getname(struct socket *sock, struct sockaddr *addr,
342 int *sockaddr_len, int peer)
343{
344 struct sock *sk = sock->sk;
345 struct pn_sock *pn = pn_sk(sk);
346
347 memset(addr, 0, sizeof(struct sockaddr_pn));
348 addr->sa_family = AF_PHONET;
349 if (!peer) /* Race with bind() here is userland's problem. */
350 pn_sockaddr_set_object((struct sockaddr_pn *)addr,
351 pn->sobject);
352
353 *sockaddr_len = sizeof(struct sockaddr_pn);
354 return 0;
355}
356
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700357static unsigned int pn_socket_poll(struct file *file, struct socket *sock,
358 poll_table *wait)
359{
360 struct sock *sk = sock->sk;
361 struct pep_sock *pn = pep_sk(sk);
362 unsigned int mask = 0;
363
Eric Dumazet43815482010-04-29 11:01:49 +0000364 poll_wait(file, sk_sleep(sk), wait);
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700365
366 switch (sk->sk_state) {
367 case TCP_LISTEN:
368 return hlist_empty(&pn->ackq) ? 0 : POLLIN;
369 case TCP_CLOSE:
370 return POLLERR;
371 }
372
373 if (!skb_queue_empty(&sk->sk_receive_queue))
374 mask |= POLLIN | POLLRDNORM;
Rémi Denis-Courmontc41bd97f82008-10-05 11:15:43 -0700375 if (!skb_queue_empty(&pn->ctrlreq_queue))
376 mask |= POLLPRI;
377 if (!mask && sk->sk_state == TCP_CLOSE_WAIT)
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700378 return POLLHUP;
379
Rémi Denis-Courmont01b38602010-08-30 12:57:05 +0000380 if (sk->sk_state == TCP_ESTABLISHED &&
381 atomic_read(&sk->sk_wmem_alloc) < sk->sk_sndbuf &&
382 atomic_read(&pn->tx_credits))
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700383 mask |= POLLOUT | POLLWRNORM | POLLWRBAND;
384
385 return mask;
386}
387
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700388static int pn_socket_ioctl(struct socket *sock, unsigned int cmd,
389 unsigned long arg)
390{
391 struct sock *sk = sock->sk;
392 struct pn_sock *pn = pn_sk(sk);
393
394 if (cmd == SIOCPNGETOBJECT) {
395 struct net_device *dev;
396 u16 handle;
397 u8 saddr;
398
399 if (get_user(handle, (__u16 __user *)arg))
400 return -EFAULT;
401
402 lock_sock(sk);
403 if (sk->sk_bound_dev_if)
404 dev = dev_get_by_index(sock_net(sk),
405 sk->sk_bound_dev_if);
406 else
407 dev = phonet_device_get(sock_net(sk));
408 if (dev && (dev->flags & IFF_UP))
409 saddr = phonet_address_get(dev, pn_addr(handle));
410 else
411 saddr = PN_NO_ADDR;
412 release_sock(sk);
413
414 if (dev)
415 dev_put(dev);
416 if (saddr == PN_NO_ADDR)
417 return -EHOSTUNREACH;
418
419 handle = pn_object(saddr, pn_port(pn->sobject));
420 return put_user(handle, (__u16 __user *)arg);
421 }
422
423 return sk->sk_prot->ioctl(sk, cmd, arg);
424}
425
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700426static int pn_socket_listen(struct socket *sock, int backlog)
427{
428 struct sock *sk = sock->sk;
429 int err = 0;
430
431 if (sock->state != SS_UNCONNECTED)
432 return -EINVAL;
433 if (pn_socket_autobind(sock))
434 return -ENOBUFS;
435
436 lock_sock(sk);
437 if (sk->sk_state != TCP_CLOSE) {
438 err = -EINVAL;
439 goto out;
440 }
441
442 sk->sk_state = TCP_LISTEN;
443 sk->sk_ack_backlog = 0;
444 sk->sk_max_ack_backlog = backlog;
445out:
446 release_sock(sk);
447 return err;
448}
449
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700450static int pn_socket_sendmsg(struct kiocb *iocb, struct socket *sock,
451 struct msghdr *m, size_t total_len)
452{
453 struct sock *sk = sock->sk;
454
455 if (pn_socket_autobind(sock))
456 return -EAGAIN;
457
458 return sk->sk_prot->sendmsg(iocb, sk, m, total_len);
459}
460
461const struct proto_ops phonet_dgram_ops = {
462 .family = AF_PHONET,
463 .owner = THIS_MODULE,
464 .release = pn_socket_release,
465 .bind = pn_socket_bind,
466 .connect = sock_no_connect,
467 .socketpair = sock_no_socketpair,
468 .accept = sock_no_accept,
469 .getname = pn_socket_getname,
470 .poll = datagram_poll,
471 .ioctl = pn_socket_ioctl,
472 .listen = sock_no_listen,
473 .shutdown = sock_no_shutdown,
474 .setsockopt = sock_no_setsockopt,
475 .getsockopt = sock_no_getsockopt,
476#ifdef CONFIG_COMPAT
477 .compat_setsockopt = sock_no_setsockopt,
478 .compat_getsockopt = sock_no_getsockopt,
479#endif
480 .sendmsg = pn_socket_sendmsg,
481 .recvmsg = sock_common_recvmsg,
482 .mmap = sock_no_mmap,
483 .sendpage = sock_no_sendpage,
484};
485
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700486const struct proto_ops phonet_stream_ops = {
487 .family = AF_PHONET,
488 .owner = THIS_MODULE,
489 .release = pn_socket_release,
490 .bind = pn_socket_bind,
Kumar Sanghvib3d62552010-10-12 20:14:43 +0000491#ifdef CONFIG_PHONET_PIPECTRLR
492 .connect = pn_socket_connect,
493#else
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700494 .connect = sock_no_connect,
Kumar Sanghvib3d62552010-10-12 20:14:43 +0000495#endif
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700496 .socketpair = sock_no_socketpair,
497 .accept = pn_socket_accept,
498 .getname = pn_socket_getname,
499 .poll = pn_socket_poll,
500 .ioctl = pn_socket_ioctl,
501 .listen = pn_socket_listen,
502 .shutdown = sock_no_shutdown,
Rémi Denis-Courmont02a47612008-10-05 11:16:16 -0700503 .setsockopt = sock_common_setsockopt,
504 .getsockopt = sock_common_getsockopt,
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700505#ifdef CONFIG_COMPAT
Rémi Denis-Courmont02a47612008-10-05 11:16:16 -0700506 .compat_setsockopt = compat_sock_common_setsockopt,
507 .compat_getsockopt = compat_sock_common_getsockopt,
Rémi Denis-Courmont9995a322008-10-05 11:14:48 -0700508#endif
509 .sendmsg = pn_socket_sendmsg,
510 .recvmsg = sock_common_recvmsg,
511 .mmap = sock_no_mmap,
512 .sendpage = sock_no_sendpage,
513};
514EXPORT_SYMBOL(phonet_stream_ops);
515
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700516/* allocate port for a socket */
517int pn_sock_get_port(struct sock *sk, unsigned short sport)
518{
519 static int port_cur;
Rémi Denis-Courmont52404882008-12-03 15:42:56 -0800520 struct net *net = sock_net(sk);
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700521 struct pn_sock *pn = pn_sk(sk);
522 struct sockaddr_pn try_sa;
523 struct sock *tmpsk;
524
525 memset(&try_sa, 0, sizeof(struct sockaddr_pn));
526 try_sa.spn_family = AF_PHONET;
Rémi Denis-Courmont582b0b62009-09-23 03:17:10 +0000527 WARN_ON(!mutex_is_locked(&port_mutex));
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700528 if (!sport) {
529 /* search free port */
Remi Denis-Courmont87ab4e22008-09-22 20:08:39 -0700530 int port, pmin, pmax;
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700531
Remi Denis-Courmont87ab4e22008-09-22 20:08:39 -0700532 phonet_get_local_port_range(&pmin, &pmax);
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700533 for (port = pmin; port <= pmax; port++) {
534 port_cur++;
535 if (port_cur < pmin || port_cur > pmax)
536 port_cur = pmin;
537
538 pn_sockaddr_set_port(&try_sa, port_cur);
Rémi Denis-Courmont52404882008-12-03 15:42:56 -0800539 tmpsk = pn_find_sock_by_sa(net, &try_sa);
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700540 if (tmpsk == NULL) {
541 sport = port_cur;
542 goto found;
543 } else
544 sock_put(tmpsk);
545 }
546 } else {
547 /* try to find specific port */
548 pn_sockaddr_set_port(&try_sa, sport);
Rémi Denis-Courmont52404882008-12-03 15:42:56 -0800549 tmpsk = pn_find_sock_by_sa(net, &try_sa);
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700550 if (tmpsk == NULL)
551 /* No sock there! We can use that port... */
552 goto found;
553 else
554 sock_put(tmpsk);
555 }
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700556 /* the port must be in use already */
557 return -EADDRINUSE;
558
559found:
Remi Denis-Courmontba113a92008-09-22 20:05:19 -0700560 pn->sobject = pn_object(pn_addr(pn->sobject), sport);
561 return 0;
562}
563EXPORT_SYMBOL(pn_sock_get_port);
Rémi Denis-Courmontc1dc13e2009-07-21 01:57:57 +0000564
Randy Dunlapae6e2ae2009-08-17 10:35:49 +0000565#ifdef CONFIG_PROC_FS
Rémi Denis-Courmontc1dc13e2009-07-21 01:57:57 +0000566static struct sock *pn_sock_get_idx(struct seq_file *seq, loff_t pos)
567{
568 struct net *net = seq_file_net(seq);
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +0000569 struct hlist_head *hlist = pnsocks.hlist;
Rémi Denis-Courmontc1dc13e2009-07-21 01:57:57 +0000570 struct hlist_node *node;
571 struct sock *sknode;
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +0000572 unsigned h;
Rémi Denis-Courmontc1dc13e2009-07-21 01:57:57 +0000573
Rémi Denis-Courmont6b0d07b2009-11-09 02:17:01 +0000574 for (h = 0; h < PN_HASHSIZE; h++) {
575 sk_for_each(sknode, node, hlist) {
576 if (!net_eq(net, sock_net(sknode)))
577 continue;
578 if (!pos)
579 return sknode;
580 pos--;
581 }
582 hlist++;
Rémi Denis-Courmontc1dc13e2009-07-21 01:57:57 +0000583 }
584 return NULL;
585}
586
587static struct sock *pn_sock_get_next(struct seq_file *seq, struct sock *sk)
588{
589 struct net *net = seq_file_net(seq);
590
591 do
592 sk = sk_next(sk);
593 while (sk && !net_eq(net, sock_net(sk)));
594
595 return sk;
596}
597
598static void *pn_sock_seq_start(struct seq_file *seq, loff_t *pos)
599 __acquires(pnsocks.lock)
600{
601 spin_lock_bh(&pnsocks.lock);
602 return *pos ? pn_sock_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
603}
604
605static void *pn_sock_seq_next(struct seq_file *seq, void *v, loff_t *pos)
606{
607 struct sock *sk;
608
609 if (v == SEQ_START_TOKEN)
610 sk = pn_sock_get_idx(seq, 0);
611 else
612 sk = pn_sock_get_next(seq, v);
613 (*pos)++;
614 return sk;
615}
616
617static void pn_sock_seq_stop(struct seq_file *seq, void *v)
618 __releases(pnsocks.lock)
619{
620 spin_unlock_bh(&pnsocks.lock);
621}
622
623static int pn_sock_seq_show(struct seq_file *seq, void *v)
624{
625 int len;
626
627 if (v == SEQ_START_TOKEN)
628 seq_printf(seq, "%s%n", "pt loc rem rs st tx_queue rx_queue "
629 " uid inode ref pointer drops", &len);
630 else {
631 struct sock *sk = v;
632 struct pn_sock *pn = pn_sk(sk);
633
634 seq_printf(seq, "%2d %04X:%04X:%02X %02X %08X:%08X %5d %lu "
635 "%d %p %d%n",
636 sk->sk_protocol, pn->sobject, 0, pn->resource,
637 sk->sk_state,
638 sk_wmem_alloc_get(sk), sk_rmem_alloc_get(sk),
639 sock_i_uid(sk), sock_i_ino(sk),
640 atomic_read(&sk->sk_refcnt), sk,
641 atomic_read(&sk->sk_drops), &len);
642 }
643 seq_printf(seq, "%*s\n", 127 - len, "");
644 return 0;
645}
646
647static const struct seq_operations pn_sock_seq_ops = {
648 .start = pn_sock_seq_start,
649 .next = pn_sock_seq_next,
650 .stop = pn_sock_seq_stop,
651 .show = pn_sock_seq_show,
652};
653
654static int pn_sock_open(struct inode *inode, struct file *file)
655{
Rémi Denis-Courmontcb7d9e72009-08-11 03:12:07 +0000656 return seq_open_net(inode, file, &pn_sock_seq_ops,
657 sizeof(struct seq_net_private));
Rémi Denis-Courmontc1dc13e2009-07-21 01:57:57 +0000658}
659
660const struct file_operations pn_sock_seq_fops = {
661 .owner = THIS_MODULE,
662 .open = pn_sock_open,
663 .read = seq_read,
664 .llseek = seq_lseek,
Rémi Denis-Courmontcb7d9e72009-08-11 03:12:07 +0000665 .release = seq_release_net,
Rémi Denis-Courmontc1dc13e2009-07-21 01:57:57 +0000666};
Randy Dunlapae6e2ae2009-08-17 10:35:49 +0000667#endif
Rémi Denis-Courmont4e3d16c2010-09-15 12:30:11 +0000668
669static struct {
670 struct sock *sk[256];
671} pnres;
672
673/*
674 * Find and hold socket based on resource.
675 */
676struct sock *pn_find_sock_by_res(struct net *net, u8 res)
677{
678 struct sock *sk;
679
680 if (!net_eq(net, &init_net))
681 return NULL;
682
683 rcu_read_lock();
684 sk = rcu_dereference(pnres.sk[res]);
685 if (sk)
686 sock_hold(sk);
687 rcu_read_unlock();
688 return sk;
689}
690
691static DEFINE_MUTEX(resource_mutex);
692
693int pn_sock_bind_res(struct sock *sk, u8 res)
694{
695 int ret = -EADDRINUSE;
696
697 if (!net_eq(sock_net(sk), &init_net))
698 return -ENOIOCTLCMD;
699 if (!capable(CAP_SYS_ADMIN))
700 return -EPERM;
701 if (pn_socket_autobind(sk->sk_socket))
702 return -EAGAIN;
703
704 mutex_lock(&resource_mutex);
705 if (pnres.sk[res] == NULL) {
706 sock_hold(sk);
707 rcu_assign_pointer(pnres.sk[res], sk);
708 ret = 0;
709 }
710 mutex_unlock(&resource_mutex);
711 return ret;
712}
713
714int pn_sock_unbind_res(struct sock *sk, u8 res)
715{
716 int ret = -ENOENT;
717
718 if (!capable(CAP_SYS_ADMIN))
719 return -EPERM;
720
721 mutex_lock(&resource_mutex);
722 if (pnres.sk[res] == sk) {
723 rcu_assign_pointer(pnres.sk[res], NULL);
724 ret = 0;
725 }
726 mutex_unlock(&resource_mutex);
727
728 if (ret == 0) {
729 synchronize_rcu();
730 sock_put(sk);
731 }
732 return ret;
733}
734
735void pn_sock_unbind_all_res(struct sock *sk)
736{
737 unsigned res, match = 0;
738
739 mutex_lock(&resource_mutex);
740 for (res = 0; res < 256; res++) {
741 if (pnres.sk[res] == sk) {
742 rcu_assign_pointer(pnres.sk[res], NULL);
743 match++;
744 }
745 }
746 mutex_unlock(&resource_mutex);
747
748 if (match == 0)
749 return;
750 synchronize_rcu();
751 while (match > 0) {
752 sock_put(sk);
753 match--;
754 }
755}
Rémi Denis-Courmont507215f2010-09-15 12:30:14 +0000756
757#ifdef CONFIG_PROC_FS
758static struct sock **pn_res_get_idx(struct seq_file *seq, loff_t pos)
759{
760 struct net *net = seq_file_net(seq);
761 unsigned i;
762
763 if (!net_eq(net, &init_net))
764 return NULL;
765
766 for (i = 0; i < 256; i++) {
767 if (pnres.sk[i] == NULL)
768 continue;
769 if (!pos)
770 return pnres.sk + i;
771 pos--;
772 }
773 return NULL;
774}
775
776static struct sock **pn_res_get_next(struct seq_file *seq, struct sock **sk)
777{
778 struct net *net = seq_file_net(seq);
779 unsigned i;
780
781 BUG_ON(!net_eq(net, &init_net));
782
783 for (i = (sk - pnres.sk) + 1; i < 256; i++)
784 if (pnres.sk[i])
785 return pnres.sk + i;
786 return NULL;
787}
788
789static void *pn_res_seq_start(struct seq_file *seq, loff_t *pos)
790 __acquires(resource_mutex)
791{
792 mutex_lock(&resource_mutex);
793 return *pos ? pn_res_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
794}
795
796static void *pn_res_seq_next(struct seq_file *seq, void *v, loff_t *pos)
797{
798 struct sock **sk;
799
800 if (v == SEQ_START_TOKEN)
801 sk = pn_res_get_idx(seq, 0);
802 else
803 sk = pn_res_get_next(seq, v);
804 (*pos)++;
805 return sk;
806}
807
808static void pn_res_seq_stop(struct seq_file *seq, void *v)
809 __releases(resource_mutex)
810{
811 mutex_unlock(&resource_mutex);
812}
813
814static int pn_res_seq_show(struct seq_file *seq, void *v)
815{
816 int len;
817
818 if (v == SEQ_START_TOKEN)
819 seq_printf(seq, "%s%n", "rs uid inode", &len);
820 else {
821 struct sock **psk = v;
822 struct sock *sk = *psk;
823
824 seq_printf(seq, "%02X %5d %lu%n",
David S. Miller9e0064a2010-09-15 21:34:41 -0700825 (int) (psk - pnres.sk), sock_i_uid(sk),
826 sock_i_ino(sk), &len);
Rémi Denis-Courmont507215f2010-09-15 12:30:14 +0000827 }
828 seq_printf(seq, "%*s\n", 63 - len, "");
829 return 0;
830}
831
832static const struct seq_operations pn_res_seq_ops = {
833 .start = pn_res_seq_start,
834 .next = pn_res_seq_next,
835 .stop = pn_res_seq_stop,
836 .show = pn_res_seq_show,
837};
838
839static int pn_res_open(struct inode *inode, struct file *file)
840{
841 return seq_open_net(inode, file, &pn_res_seq_ops,
842 sizeof(struct seq_net_private));
843}
844
845const struct file_operations pn_res_seq_fops = {
846 .owner = THIS_MODULE,
847 .open = pn_res_open,
848 .read = seq_read,
849 .llseek = seq_lseek,
850 .release = seq_release_net,
851};
852#endif