6 * Copyright (C) 2008 Nokia Corporation.
8 * Contact: Remi Denis-Courmont <remi.denis-courmont@nokia.com>
9 * Original author: Sakari Ailus <sakari.ailus@nokia.com>
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.
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.
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
26 #include <linux/gfp.h>
27 #include <linux/kernel.h>
28 #include <linux/net.h>
29 #include <linux/poll.h>
31 #include <net/tcp_states.h>
33 #include <linux/phonet.h>
34 #include <net/phonet/phonet.h>
35 #include <net/phonet/pep.h>
36 #include <net/phonet/pn_dev.h>
38 static int pn_socket_release(struct socket *sock)
40 struct sock *sk = sock->sk;
44 sk->sk_prot->close(sk, 0);
49 #define PN_HASHSIZE 16
50 #define PN_HASHMASK (PN_HASHSIZE-1)
54 struct hlist_head hlist[PN_HASHSIZE];
58 void __init pn_sock_init(void)
62 for (i = 0; i < PN_HASHSIZE; i++)
63 INIT_HLIST_HEAD(pnsocks.hlist + i);
64 spin_lock_init(&pnsocks.lock);
67 static struct hlist_head *pn_hash_list(u16 obj)
69 return pnsocks.hlist + (obj & PN_HASHMASK);
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.
76 struct sock *pn_find_sock_by_sa(struct net *net, const struct sockaddr_pn *spn)
78 struct hlist_node *node;
80 struct sock *rval = NULL;
81 u16 obj = pn_sockaddr_get_object(spn);
82 u8 res = spn->spn_resource;
83 struct hlist_head *hlist = pn_hash_list(obj);
85 spin_lock_bh(&pnsocks.lock);
87 sk_for_each(sknode, node, hlist) {
88 struct pn_sock *pn = pn_sk(sknode);
89 BUG_ON(!pn->sobject); /* unbound socket */
91 if (!net_eq(sock_net(sknode), net))
94 /* Look up socket by port */
95 if (pn_port(pn->sobject) != pn_port(obj))
98 /* If port is zero, look up by resource */
99 if (pn->resource != res)
102 if (pn_addr(pn->sobject) &&
103 pn_addr(pn->sobject) != pn_addr(obj))
111 spin_unlock_bh(&pnsocks.lock);
116 /* Deliver a broadcast packet (only in bottom-half) */
117 void pn_deliver_sock_broadcast(struct net *net, struct sk_buff *skb)
119 struct hlist_head *hlist = pnsocks.hlist;
122 spin_lock(&pnsocks.lock);
123 for (h = 0; h < PN_HASHSIZE; h++) {
124 struct hlist_node *node;
127 sk_for_each(sknode, node, hlist) {
128 struct sk_buff *clone;
130 if (!net_eq(sock_net(sknode), net))
132 if (!sock_flag(sknode, SOCK_BROADCAST))
135 clone = skb_clone(skb, GFP_ATOMIC);
138 sk_receive_skb(sknode, clone, 0);
143 spin_unlock(&pnsocks.lock);
146 void pn_sock_hash(struct sock *sk)
148 struct hlist_head *hlist = pn_hash_list(pn_sk(sk)->sobject);
150 spin_lock_bh(&pnsocks.lock);
151 sk_add_node(sk, hlist);
152 spin_unlock_bh(&pnsocks.lock);
154 EXPORT_SYMBOL(pn_sock_hash);
156 void pn_sock_unhash(struct sock *sk)
158 spin_lock_bh(&pnsocks.lock);
159 sk_del_node_init(sk);
160 spin_unlock_bh(&pnsocks.lock);
161 pn_sock_unbind_all_res(sk);
163 EXPORT_SYMBOL(pn_sock_unhash);
165 static DEFINE_MUTEX(port_mutex);
167 static int pn_socket_bind(struct socket *sock, struct sockaddr *addr, int len)
169 struct sock *sk = sock->sk;
170 struct pn_sock *pn = pn_sk(sk);
171 struct sockaddr_pn *spn = (struct sockaddr_pn *)addr;
176 if (sk->sk_prot->bind)
177 return sk->sk_prot->bind(sk, addr, len);
179 if (len < sizeof(struct sockaddr_pn))
181 if (spn->spn_family != AF_PHONET)
182 return -EAFNOSUPPORT;
184 handle = pn_sockaddr_get_object((struct sockaddr_pn *)addr);
185 saddr = pn_addr(handle);
186 if (saddr && phonet_address_lookup(sock_net(sk), saddr))
187 return -EADDRNOTAVAIL;
190 if (sk->sk_state != TCP_CLOSE || pn_port(pn->sobject)) {
191 err = -EINVAL; /* attempt to rebind */
194 WARN_ON(sk_hashed(sk));
195 mutex_lock(&port_mutex);
196 err = sk->sk_prot->get_port(sk, pn_port(handle));
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;
204 /* Enable RX on the socket */
205 sk->sk_prot->hash(sk);
207 mutex_unlock(&port_mutex);
213 static int pn_socket_autobind(struct socket *sock)
215 struct sockaddr_pn sa;
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));
224 BUG_ON(!pn_port(pn_sk(sock->sk)->sobject));
225 return 0; /* socket was already bound */
228 static int pn_socket_accept(struct socket *sock, struct socket *newsock,
231 struct sock *sk = sock->sk;
235 newsk = sk->sk_prot->accept(sk, flags, &err);
240 sock_graft(newsk, newsock);
241 newsock->state = SS_CONNECTED;
246 static int pn_socket_getname(struct socket *sock, struct sockaddr *addr,
247 int *sockaddr_len, int peer)
249 struct sock *sk = sock->sk;
250 struct pn_sock *pn = pn_sk(sk);
252 memset(addr, 0, sizeof(struct sockaddr_pn));
253 addr->sa_family = AF_PHONET;
254 if (!peer) /* Race with bind() here is userland's problem. */
255 pn_sockaddr_set_object((struct sockaddr_pn *)addr,
258 *sockaddr_len = sizeof(struct sockaddr_pn);
262 static unsigned int pn_socket_poll(struct file *file, struct socket *sock,
265 struct sock *sk = sock->sk;
266 struct pep_sock *pn = pep_sk(sk);
267 unsigned int mask = 0;
269 poll_wait(file, sk_sleep(sk), wait);
271 switch (sk->sk_state) {
273 return hlist_empty(&pn->ackq) ? 0 : POLLIN;
278 if (!skb_queue_empty(&sk->sk_receive_queue))
279 mask |= POLLIN | POLLRDNORM;
280 if (!skb_queue_empty(&pn->ctrlreq_queue))
282 if (!mask && sk->sk_state == TCP_CLOSE_WAIT)
285 if (sk->sk_state == TCP_ESTABLISHED &&
286 atomic_read(&sk->sk_wmem_alloc) < sk->sk_sndbuf &&
287 atomic_read(&pn->tx_credits))
288 mask |= POLLOUT | POLLWRNORM | POLLWRBAND;
293 static int pn_socket_ioctl(struct socket *sock, unsigned int cmd,
296 struct sock *sk = sock->sk;
297 struct pn_sock *pn = pn_sk(sk);
299 if (cmd == SIOCPNGETOBJECT) {
300 struct net_device *dev;
304 if (get_user(handle, (__u16 __user *)arg))
308 if (sk->sk_bound_dev_if)
309 dev = dev_get_by_index(sock_net(sk),
310 sk->sk_bound_dev_if);
312 dev = phonet_device_get(sock_net(sk));
313 if (dev && (dev->flags & IFF_UP))
314 saddr = phonet_address_get(dev, pn_addr(handle));
321 if (saddr == PN_NO_ADDR)
322 return -EHOSTUNREACH;
324 handle = pn_object(saddr, pn_port(pn->sobject));
325 return put_user(handle, (__u16 __user *)arg);
328 return sk->sk_prot->ioctl(sk, cmd, arg);
331 static int pn_socket_listen(struct socket *sock, int backlog)
333 struct sock *sk = sock->sk;
336 if (sock->state != SS_UNCONNECTED)
338 if (pn_socket_autobind(sock))
342 if (sk->sk_state != TCP_CLOSE) {
347 sk->sk_state = TCP_LISTEN;
348 sk->sk_ack_backlog = 0;
349 sk->sk_max_ack_backlog = backlog;
355 static int pn_socket_sendmsg(struct kiocb *iocb, struct socket *sock,
356 struct msghdr *m, size_t total_len)
358 struct sock *sk = sock->sk;
360 if (pn_socket_autobind(sock))
363 return sk->sk_prot->sendmsg(iocb, sk, m, total_len);
366 const struct proto_ops phonet_dgram_ops = {
368 .owner = THIS_MODULE,
369 .release = pn_socket_release,
370 .bind = pn_socket_bind,
371 .connect = sock_no_connect,
372 .socketpair = sock_no_socketpair,
373 .accept = sock_no_accept,
374 .getname = pn_socket_getname,
375 .poll = datagram_poll,
376 .ioctl = pn_socket_ioctl,
377 .listen = sock_no_listen,
378 .shutdown = sock_no_shutdown,
379 .setsockopt = sock_no_setsockopt,
380 .getsockopt = sock_no_getsockopt,
382 .compat_setsockopt = sock_no_setsockopt,
383 .compat_getsockopt = sock_no_getsockopt,
385 .sendmsg = pn_socket_sendmsg,
386 .recvmsg = sock_common_recvmsg,
387 .mmap = sock_no_mmap,
388 .sendpage = sock_no_sendpage,
391 const struct proto_ops phonet_stream_ops = {
393 .owner = THIS_MODULE,
394 .release = pn_socket_release,
395 .bind = pn_socket_bind,
396 .connect = sock_no_connect,
397 .socketpair = sock_no_socketpair,
398 .accept = pn_socket_accept,
399 .getname = pn_socket_getname,
400 .poll = pn_socket_poll,
401 .ioctl = pn_socket_ioctl,
402 .listen = pn_socket_listen,
403 .shutdown = sock_no_shutdown,
404 .setsockopt = sock_common_setsockopt,
405 .getsockopt = sock_common_getsockopt,
407 .compat_setsockopt = compat_sock_common_setsockopt,
408 .compat_getsockopt = compat_sock_common_getsockopt,
410 .sendmsg = pn_socket_sendmsg,
411 .recvmsg = sock_common_recvmsg,
412 .mmap = sock_no_mmap,
413 .sendpage = sock_no_sendpage,
415 EXPORT_SYMBOL(phonet_stream_ops);
417 /* allocate port for a socket */
418 int pn_sock_get_port(struct sock *sk, unsigned short sport)
421 struct net *net = sock_net(sk);
422 struct pn_sock *pn = pn_sk(sk);
423 struct sockaddr_pn try_sa;
426 memset(&try_sa, 0, sizeof(struct sockaddr_pn));
427 try_sa.spn_family = AF_PHONET;
428 WARN_ON(!mutex_is_locked(&port_mutex));
430 /* search free port */
431 int port, pmin, pmax;
433 phonet_get_local_port_range(&pmin, &pmax);
434 for (port = pmin; port <= pmax; port++) {
436 if (port_cur < pmin || port_cur > pmax)
439 pn_sockaddr_set_port(&try_sa, port_cur);
440 tmpsk = pn_find_sock_by_sa(net, &try_sa);
448 /* try to find specific port */
449 pn_sockaddr_set_port(&try_sa, sport);
450 tmpsk = pn_find_sock_by_sa(net, &try_sa);
452 /* No sock there! We can use that port... */
457 /* the port must be in use already */
461 pn->sobject = pn_object(pn_addr(pn->sobject), sport);
464 EXPORT_SYMBOL(pn_sock_get_port);
466 #ifdef CONFIG_PROC_FS
467 static struct sock *pn_sock_get_idx(struct seq_file *seq, loff_t pos)
469 struct net *net = seq_file_net(seq);
470 struct hlist_head *hlist = pnsocks.hlist;
471 struct hlist_node *node;
475 for (h = 0; h < PN_HASHSIZE; h++) {
476 sk_for_each(sknode, node, hlist) {
477 if (!net_eq(net, sock_net(sknode)))
488 static struct sock *pn_sock_get_next(struct seq_file *seq, struct sock *sk)
490 struct net *net = seq_file_net(seq);
494 while (sk && !net_eq(net, sock_net(sk)));
499 static void *pn_sock_seq_start(struct seq_file *seq, loff_t *pos)
500 __acquires(pnsocks.lock)
502 spin_lock_bh(&pnsocks.lock);
503 return *pos ? pn_sock_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
506 static void *pn_sock_seq_next(struct seq_file *seq, void *v, loff_t *pos)
510 if (v == SEQ_START_TOKEN)
511 sk = pn_sock_get_idx(seq, 0);
513 sk = pn_sock_get_next(seq, v);
518 static void pn_sock_seq_stop(struct seq_file *seq, void *v)
519 __releases(pnsocks.lock)
521 spin_unlock_bh(&pnsocks.lock);
524 static int pn_sock_seq_show(struct seq_file *seq, void *v)
528 if (v == SEQ_START_TOKEN)
529 seq_printf(seq, "%s%n", "pt loc rem rs st tx_queue rx_queue "
530 " uid inode ref pointer drops", &len);
533 struct pn_sock *pn = pn_sk(sk);
535 seq_printf(seq, "%2d %04X:%04X:%02X %02X %08X:%08X %5d %lu "
537 sk->sk_protocol, pn->sobject, 0, pn->resource,
539 sk_wmem_alloc_get(sk), sk_rmem_alloc_get(sk),
540 sock_i_uid(sk), sock_i_ino(sk),
541 atomic_read(&sk->sk_refcnt), sk,
542 atomic_read(&sk->sk_drops), &len);
544 seq_printf(seq, "%*s\n", 127 - len, "");
548 static const struct seq_operations pn_sock_seq_ops = {
549 .start = pn_sock_seq_start,
550 .next = pn_sock_seq_next,
551 .stop = pn_sock_seq_stop,
552 .show = pn_sock_seq_show,
555 static int pn_sock_open(struct inode *inode, struct file *file)
557 return seq_open_net(inode, file, &pn_sock_seq_ops,
558 sizeof(struct seq_net_private));
561 const struct file_operations pn_sock_seq_fops = {
562 .owner = THIS_MODULE,
563 .open = pn_sock_open,
566 .release = seq_release_net,
571 struct sock *sk[256];
575 * Find and hold socket based on resource.
577 struct sock *pn_find_sock_by_res(struct net *net, u8 res)
581 if (!net_eq(net, &init_net))
585 sk = rcu_dereference(pnres.sk[res]);
592 static DEFINE_MUTEX(resource_mutex);
594 int pn_sock_bind_res(struct sock *sk, u8 res)
596 int ret = -EADDRINUSE;
598 if (!net_eq(sock_net(sk), &init_net))
600 if (!capable(CAP_SYS_ADMIN))
602 if (pn_socket_autobind(sk->sk_socket))
605 mutex_lock(&resource_mutex);
606 if (pnres.sk[res] == NULL) {
608 rcu_assign_pointer(pnres.sk[res], sk);
611 mutex_unlock(&resource_mutex);
615 int pn_sock_unbind_res(struct sock *sk, u8 res)
619 if (!capable(CAP_SYS_ADMIN))
622 mutex_lock(&resource_mutex);
623 if (pnres.sk[res] == sk) {
624 rcu_assign_pointer(pnres.sk[res], NULL);
627 mutex_unlock(&resource_mutex);
636 void pn_sock_unbind_all_res(struct sock *sk)
638 unsigned res, match = 0;
640 mutex_lock(&resource_mutex);
641 for (res = 0; res < 256; res++) {
642 if (pnres.sk[res] == sk) {
643 rcu_assign_pointer(pnres.sk[res], NULL);
647 mutex_unlock(&resource_mutex);
658 #ifdef CONFIG_PROC_FS
659 static struct sock **pn_res_get_idx(struct seq_file *seq, loff_t pos)
661 struct net *net = seq_file_net(seq);
664 if (!net_eq(net, &init_net))
667 for (i = 0; i < 256; i++) {
668 if (pnres.sk[i] == NULL)
677 static struct sock **pn_res_get_next(struct seq_file *seq, struct sock **sk)
679 struct net *net = seq_file_net(seq);
682 BUG_ON(!net_eq(net, &init_net));
684 for (i = (sk - pnres.sk) + 1; i < 256; i++)
690 static void *pn_res_seq_start(struct seq_file *seq, loff_t *pos)
691 __acquires(resource_mutex)
693 mutex_lock(&resource_mutex);
694 return *pos ? pn_res_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
697 static void *pn_res_seq_next(struct seq_file *seq, void *v, loff_t *pos)
701 if (v == SEQ_START_TOKEN)
702 sk = pn_res_get_idx(seq, 0);
704 sk = pn_res_get_next(seq, v);
709 static void pn_res_seq_stop(struct seq_file *seq, void *v)
710 __releases(resource_mutex)
712 mutex_unlock(&resource_mutex);
715 static int pn_res_seq_show(struct seq_file *seq, void *v)
719 if (v == SEQ_START_TOKEN)
720 seq_printf(seq, "%s%n", "rs uid inode", &len);
722 struct sock **psk = v;
723 struct sock *sk = *psk;
725 seq_printf(seq, "%02X %5d %lu%n",
726 psk - pnres.sk, sock_i_uid(sk), sock_i_ino(sk), &len);
728 seq_printf(seq, "%*s\n", 63 - len, "");
732 static const struct seq_operations pn_res_seq_ops = {
733 .start = pn_res_seq_start,
734 .next = pn_res_seq_next,
735 .stop = pn_res_seq_stop,
736 .show = pn_res_seq_show,
739 static int pn_res_open(struct inode *inode, struct file *file)
741 return seq_open_net(inode, file, &pn_res_seq_ops,
742 sizeof(struct seq_net_private));
745 const struct file_operations pn_res_seq_fops = {
746 .owner = THIS_MODULE,
750 .release = seq_release_net,