1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26#include <linux/gfp.h>
27#include <linux/kernel.h>
28#include <linux/net.h>
29#include <linux/poll.h>
30#include <linux/sched/signal.h>
31
32#include <net/sock.h>
33#include <net/tcp_states.h>
34
35#include <linux/phonet.h>
36#include <linux/export.h>
37#include <net/phonet/phonet.h>
38#include <net/phonet/pep.h>
39#include <net/phonet/pn_dev.h>
40
41static int pn_socket_release(struct socket *sock)
42{
43 struct sock *sk = sock->sk;
44
45 if (sk) {
46 sock->sk = NULL;
47 sk->sk_prot->close(sk, 0);
48 }
49 return 0;
50}
51
52#define PN_HASHSIZE 16
53#define PN_HASHMASK (PN_HASHSIZE-1)
54
55
56static struct {
57 struct hlist_head hlist[PN_HASHSIZE];
58 struct mutex lock;
59} pnsocks;
60
61void __init pn_sock_init(void)
62{
63 unsigned int i;
64
65 for (i = 0; i < PN_HASHSIZE; i++)
66 INIT_HLIST_HEAD(pnsocks.hlist + i);
67 mutex_init(&pnsocks.lock);
68}
69
70static struct hlist_head *pn_hash_list(u16 obj)
71{
72 return pnsocks.hlist + (obj & PN_HASHMASK);
73}
74
75
76
77
78
79struct sock *pn_find_sock_by_sa(struct net *net, const struct sockaddr_pn *spn)
80{
81 struct sock *sknode;
82 struct sock *rval = NULL;
83 u16 obj = pn_sockaddr_get_object(spn);
84 u8 res = spn->spn_resource;
85 struct hlist_head *hlist = pn_hash_list(obj);
86
87 rcu_read_lock();
88 sk_for_each_rcu(sknode, hlist) {
89 struct pn_sock *pn = pn_sk(sknode);
90 BUG_ON(!pn->sobject);
91
92 if (!net_eq(sock_net(sknode), net))
93 continue;
94 if (pn_port(obj)) {
95
96 if (pn_port(pn->sobject) != pn_port(obj))
97 continue;
98 } else {
99
100 if (pn->resource != res)
101 continue;
102 }
103 if (pn_addr(pn->sobject) &&
104 pn_addr(pn->sobject) != pn_addr(obj))
105 continue;
106
107 rval = sknode;
108 sock_hold(sknode);
109 break;
110 }
111 rcu_read_unlock();
112
113 return rval;
114}
115
116
117void pn_deliver_sock_broadcast(struct net *net, struct sk_buff *skb)
118{
119 struct hlist_head *hlist = pnsocks.hlist;
120 unsigned int h;
121
122 rcu_read_lock();
123 for (h = 0; h < PN_HASHSIZE; h++) {
124 struct sock *sknode;
125
126 sk_for_each(sknode, hlist) {
127 struct sk_buff *clone;
128
129 if (!net_eq(sock_net(sknode), net))
130 continue;
131 if (!sock_flag(sknode, SOCK_BROADCAST))
132 continue;
133
134 clone = skb_clone(skb, GFP_ATOMIC);
135 if (clone) {
136 sock_hold(sknode);
137 sk_receive_skb(sknode, clone, 0);
138 }
139 }
140 hlist++;
141 }
142 rcu_read_unlock();
143}
144
145int pn_sock_hash(struct sock *sk)
146{
147 struct hlist_head *hlist = pn_hash_list(pn_sk(sk)->sobject);
148
149 mutex_lock(&pnsocks.lock);
150 sk_add_node_rcu(sk, hlist);
151 mutex_unlock(&pnsocks.lock);
152
153 return 0;
154}
155EXPORT_SYMBOL(pn_sock_hash);
156
157void pn_sock_unhash(struct sock *sk)
158{
159 mutex_lock(&pnsocks.lock);
160 sk_del_node_init_rcu(sk);
161 mutex_unlock(&pnsocks.lock);
162 pn_sock_unbind_all_res(sk);
163 synchronize_rcu();
164}
165EXPORT_SYMBOL(pn_sock_unhash);
166
167static DEFINE_MUTEX(port_mutex);
168
169static int pn_socket_bind(struct socket *sock, struct sockaddr *addr, int len)
170{
171 struct sock *sk = sock->sk;
172 struct pn_sock *pn = pn_sk(sk);
173 struct sockaddr_pn *spn = (struct sockaddr_pn *)addr;
174 int err;
175 u16 handle;
176 u8 saddr;
177
178 if (sk->sk_prot->bind)
179 return sk->sk_prot->bind(sk, addr, len);
180
181 if (len < sizeof(struct sockaddr_pn))
182 return -EINVAL;
183 if (spn->spn_family != AF_PHONET)
184 return -EAFNOSUPPORT;
185
186 handle = pn_sockaddr_get_object((struct sockaddr_pn *)addr);
187 saddr = pn_addr(handle);
188 if (saddr && phonet_address_lookup(sock_net(sk), saddr))
189 return -EADDRNOTAVAIL;
190
191 lock_sock(sk);
192 if (sk->sk_state != TCP_CLOSE || pn_port(pn->sobject)) {
193 err = -EINVAL;
194 goto out;
195 }
196 WARN_ON(sk_hashed(sk));
197 mutex_lock(&port_mutex);
198 err = sk->sk_prot->get_port(sk, pn_port(handle));
199 if (err)
200 goto out_port;
201
202
203 pn->sobject = pn_object(saddr, pn_port(pn->sobject));
204 pn->resource = spn->spn_resource;
205
206
207 err = sk->sk_prot->hash(sk);
208out_port:
209 mutex_unlock(&port_mutex);
210out:
211 release_sock(sk);
212 return err;
213}
214
215static int pn_socket_autobind(struct socket *sock)
216{
217 struct sockaddr_pn sa;
218 int err;
219
220 memset(&sa, 0, sizeof(sa));
221 sa.spn_family = AF_PHONET;
222 err = pn_socket_bind(sock, (struct sockaddr *)&sa,
223 sizeof(struct sockaddr_pn));
224 if (err != -EINVAL)
225 return err;
226 BUG_ON(!pn_port(pn_sk(sock->sk)->sobject));
227 return 0;
228}
229
230static int pn_socket_connect(struct socket *sock, struct sockaddr *addr,
231 int len, int flags)
232{
233 struct sock *sk = sock->sk;
234 struct pn_sock *pn = pn_sk(sk);
235 struct sockaddr_pn *spn = (struct sockaddr_pn *)addr;
236 struct task_struct *tsk = current;
237 long timeo = sock_rcvtimeo(sk, flags & O_NONBLOCK);
238 int err;
239
240 if (pn_socket_autobind(sock))
241 return -ENOBUFS;
242 if (len < sizeof(struct sockaddr_pn))
243 return -EINVAL;
244 if (spn->spn_family != AF_PHONET)
245 return -EAFNOSUPPORT;
246
247 lock_sock(sk);
248
249 switch (sock->state) {
250 case SS_UNCONNECTED:
251 if (sk->sk_state != TCP_CLOSE) {
252 err = -EISCONN;
253 goto out;
254 }
255 break;
256 case SS_CONNECTING:
257 err = -EALREADY;
258 goto out;
259 default:
260 err = -EISCONN;
261 goto out;
262 }
263
264 pn->dobject = pn_sockaddr_get_object(spn);
265 pn->resource = pn_sockaddr_get_resource(spn);
266 sock->state = SS_CONNECTING;
267
268 err = sk->sk_prot->connect(sk, addr, len);
269 if (err) {
270 sock->state = SS_UNCONNECTED;
271 pn->dobject = 0;
272 goto out;
273 }
274
275 while (sk->sk_state == TCP_SYN_SENT) {
276 DEFINE_WAIT(wait);
277
278 if (!timeo) {
279 err = -EINPROGRESS;
280 goto out;
281 }
282 if (signal_pending(tsk)) {
283 err = sock_intr_errno(timeo);
284 goto out;
285 }
286
287 prepare_to_wait_exclusive(sk_sleep(sk), &wait,
288 TASK_INTERRUPTIBLE);
289 release_sock(sk);
290 timeo = schedule_timeout(timeo);
291 lock_sock(sk);
292 finish_wait(sk_sleep(sk), &wait);
293 }
294
295 if ((1 << sk->sk_state) & (TCPF_SYN_RECV|TCPF_ESTABLISHED))
296 err = 0;
297 else if (sk->sk_state == TCP_CLOSE_WAIT)
298 err = -ECONNRESET;
299 else
300 err = -ECONNREFUSED;
301 sock->state = err ? SS_UNCONNECTED : SS_CONNECTED;
302out:
303 release_sock(sk);
304 return err;
305}
306
307static int pn_socket_accept(struct socket *sock, struct socket *newsock,
308 int flags, bool kern)
309{
310 struct sock *sk = sock->sk;
311 struct sock *newsk;
312 int err;
313
314 if (unlikely(sk->sk_state != TCP_LISTEN))
315 return -EINVAL;
316
317 newsk = sk->sk_prot->accept(sk, flags, &err, kern);
318 if (!newsk)
319 return err;
320
321 lock_sock(newsk);
322 sock_graft(newsk, newsock);
323 newsock->state = SS_CONNECTED;
324 release_sock(newsk);
325 return 0;
326}
327
328static int pn_socket_getname(struct socket *sock, struct sockaddr *addr,
329 int peer)
330{
331 struct sock *sk = sock->sk;
332 struct pn_sock *pn = pn_sk(sk);
333
334 memset(addr, 0, sizeof(struct sockaddr_pn));
335 addr->sa_family = AF_PHONET;
336 if (!peer)
337 pn_sockaddr_set_object((struct sockaddr_pn *)addr,
338 pn->sobject);
339
340 return sizeof(struct sockaddr_pn);
341}
342
343static __poll_t pn_socket_poll(struct file *file, struct socket *sock,
344 poll_table *wait)
345{
346 struct sock *sk = sock->sk;
347 struct pep_sock *pn = pep_sk(sk);
348 __poll_t mask = 0;
349
350 poll_wait(file, sk_sleep(sk), wait);
351
352 if (sk->sk_state == TCP_CLOSE)
353 return EPOLLERR;
354 if (!skb_queue_empty(&sk->sk_receive_queue))
355 mask |= EPOLLIN | EPOLLRDNORM;
356 if (!skb_queue_empty(&pn->ctrlreq_queue))
357 mask |= EPOLLPRI;
358 if (!mask && sk->sk_state == TCP_CLOSE_WAIT)
359 return EPOLLHUP;
360
361 if (sk->sk_state == TCP_ESTABLISHED &&
362 refcount_read(&sk->sk_wmem_alloc) < sk->sk_sndbuf &&
363 atomic_read(&pn->tx_credits))
364 mask |= EPOLLOUT | EPOLLWRNORM | EPOLLWRBAND;
365
366 return mask;
367}
368
369static int pn_socket_ioctl(struct socket *sock, unsigned int cmd,
370 unsigned long arg)
371{
372 struct sock *sk = sock->sk;
373 struct pn_sock *pn = pn_sk(sk);
374
375 if (cmd == SIOCPNGETOBJECT) {
376 struct net_device *dev;
377 u16 handle;
378 u8 saddr;
379
380 if (get_user(handle, (__u16 __user *)arg))
381 return -EFAULT;
382
383 lock_sock(sk);
384 if (sk->sk_bound_dev_if)
385 dev = dev_get_by_index(sock_net(sk),
386 sk->sk_bound_dev_if);
387 else
388 dev = phonet_device_get(sock_net(sk));
389 if (dev && (dev->flags & IFF_UP))
390 saddr = phonet_address_get(dev, pn_addr(handle));
391 else
392 saddr = PN_NO_ADDR;
393 release_sock(sk);
394
395 if (dev)
396 dev_put(dev);
397 if (saddr == PN_NO_ADDR)
398 return -EHOSTUNREACH;
399
400 handle = pn_object(saddr, pn_port(pn->sobject));
401 return put_user(handle, (__u16 __user *)arg);
402 }
403
404 return sk->sk_prot->ioctl(sk, cmd, arg);
405}
406
407static int pn_socket_listen(struct socket *sock, int backlog)
408{
409 struct sock *sk = sock->sk;
410 int err = 0;
411
412 if (pn_socket_autobind(sock))
413 return -ENOBUFS;
414
415 lock_sock(sk);
416 if (sock->state != SS_UNCONNECTED) {
417 err = -EINVAL;
418 goto out;
419 }
420
421 if (sk->sk_state != TCP_LISTEN) {
422 sk->sk_state = TCP_LISTEN;
423 sk->sk_ack_backlog = 0;
424 }
425 sk->sk_max_ack_backlog = backlog;
426out:
427 release_sock(sk);
428 return err;
429}
430
431static int pn_socket_sendmsg(struct socket *sock, struct msghdr *m,
432 size_t total_len)
433{
434 struct sock *sk = sock->sk;
435
436 if (pn_socket_autobind(sock))
437 return -EAGAIN;
438
439 return sk->sk_prot->sendmsg(sk, m, total_len);
440}
441
442const struct proto_ops phonet_dgram_ops = {
443 .family = AF_PHONET,
444 .owner = THIS_MODULE,
445 .release = pn_socket_release,
446 .bind = pn_socket_bind,
447 .connect = sock_no_connect,
448 .socketpair = sock_no_socketpair,
449 .accept = sock_no_accept,
450 .getname = pn_socket_getname,
451 .poll = datagram_poll,
452 .ioctl = pn_socket_ioctl,
453 .listen = sock_no_listen,
454 .shutdown = sock_no_shutdown,
455 .setsockopt = sock_no_setsockopt,
456 .getsockopt = sock_no_getsockopt,
457#ifdef CONFIG_COMPAT
458 .compat_setsockopt = sock_no_setsockopt,
459 .compat_getsockopt = sock_no_getsockopt,
460#endif
461 .sendmsg = pn_socket_sendmsg,
462 .recvmsg = sock_common_recvmsg,
463 .mmap = sock_no_mmap,
464 .sendpage = sock_no_sendpage,
465};
466
467const struct proto_ops phonet_stream_ops = {
468 .family = AF_PHONET,
469 .owner = THIS_MODULE,
470 .release = pn_socket_release,
471 .bind = pn_socket_bind,
472 .connect = pn_socket_connect,
473 .socketpair = sock_no_socketpair,
474 .accept = pn_socket_accept,
475 .getname = pn_socket_getname,
476 .poll = pn_socket_poll,
477 .ioctl = pn_socket_ioctl,
478 .listen = pn_socket_listen,
479 .shutdown = sock_no_shutdown,
480 .setsockopt = sock_common_setsockopt,
481 .getsockopt = sock_common_getsockopt,
482#ifdef CONFIG_COMPAT
483 .compat_setsockopt = compat_sock_common_setsockopt,
484 .compat_getsockopt = compat_sock_common_getsockopt,
485#endif
486 .sendmsg = pn_socket_sendmsg,
487 .recvmsg = sock_common_recvmsg,
488 .mmap = sock_no_mmap,
489 .sendpage = sock_no_sendpage,
490};
491EXPORT_SYMBOL(phonet_stream_ops);
492
493
494int pn_sock_get_port(struct sock *sk, unsigned short sport)
495{
496 static int port_cur;
497 struct net *net = sock_net(sk);
498 struct pn_sock *pn = pn_sk(sk);
499 struct sockaddr_pn try_sa;
500 struct sock *tmpsk;
501
502 memset(&try_sa, 0, sizeof(struct sockaddr_pn));
503 try_sa.spn_family = AF_PHONET;
504 WARN_ON(!mutex_is_locked(&port_mutex));
505 if (!sport) {
506
507 int port, pmin, pmax;
508
509 phonet_get_local_port_range(&pmin, &pmax);
510 for (port = pmin; port <= pmax; port++) {
511 port_cur++;
512 if (port_cur < pmin || port_cur > pmax)
513 port_cur = pmin;
514
515 pn_sockaddr_set_port(&try_sa, port_cur);
516 tmpsk = pn_find_sock_by_sa(net, &try_sa);
517 if (tmpsk == NULL) {
518 sport = port_cur;
519 goto found;
520 } else
521 sock_put(tmpsk);
522 }
523 } else {
524
525 pn_sockaddr_set_port(&try_sa, sport);
526 tmpsk = pn_find_sock_by_sa(net, &try_sa);
527 if (tmpsk == NULL)
528
529 goto found;
530 else
531 sock_put(tmpsk);
532 }
533
534 return -EADDRINUSE;
535
536found:
537 pn->sobject = pn_object(pn_addr(pn->sobject), sport);
538 return 0;
539}
540EXPORT_SYMBOL(pn_sock_get_port);
541
542#ifdef CONFIG_PROC_FS
543static struct sock *pn_sock_get_idx(struct seq_file *seq, loff_t pos)
544{
545 struct net *net = seq_file_net(seq);
546 struct hlist_head *hlist = pnsocks.hlist;
547 struct sock *sknode;
548 unsigned int h;
549
550 for (h = 0; h < PN_HASHSIZE; h++) {
551 sk_for_each_rcu(sknode, hlist) {
552 if (!net_eq(net, sock_net(sknode)))
553 continue;
554 if (!pos)
555 return sknode;
556 pos--;
557 }
558 hlist++;
559 }
560 return NULL;
561}
562
563static struct sock *pn_sock_get_next(struct seq_file *seq, struct sock *sk)
564{
565 struct net *net = seq_file_net(seq);
566
567 do
568 sk = sk_next(sk);
569 while (sk && !net_eq(net, sock_net(sk)));
570
571 return sk;
572}
573
574static void *pn_sock_seq_start(struct seq_file *seq, loff_t *pos)
575 __acquires(rcu)
576{
577 rcu_read_lock();
578 return *pos ? pn_sock_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
579}
580
581static void *pn_sock_seq_next(struct seq_file *seq, void *v, loff_t *pos)
582{
583 struct sock *sk;
584
585 if (v == SEQ_START_TOKEN)
586 sk = pn_sock_get_idx(seq, 0);
587 else
588 sk = pn_sock_get_next(seq, v);
589 (*pos)++;
590 return sk;
591}
592
593static void pn_sock_seq_stop(struct seq_file *seq, void *v)
594 __releases(rcu)
595{
596 rcu_read_unlock();
597}
598
599static int pn_sock_seq_show(struct seq_file *seq, void *v)
600{
601 seq_setwidth(seq, 127);
602 if (v == SEQ_START_TOKEN)
603 seq_puts(seq, "pt loc rem rs st tx_queue rx_queue "
604 " uid inode ref pointer drops");
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 "
610 "%d %pK %d",
611 sk->sk_protocol, pn->sobject, pn->dobject,
612 pn->resource, sk->sk_state,
613 sk_wmem_alloc_get(sk), sk_rmem_alloc_get(sk),
614 from_kuid_munged(seq_user_ns(seq), sock_i_uid(sk)),
615 sock_i_ino(sk),
616 refcount_read(&sk->sk_refcnt), sk,
617 atomic_read(&sk->sk_drops));
618 }
619 seq_pad(seq, '\n');
620 return 0;
621}
622
623const struct seq_operations pn_sock_seq_ops = {
624 .start = pn_sock_seq_start,
625 .next = pn_sock_seq_next,
626 .stop = pn_sock_seq_stop,
627 .show = pn_sock_seq_show,
628};
629#endif
630
631static struct {
632 struct sock *sk[256];
633} pnres;
634
635
636
637
638struct sock *pn_find_sock_by_res(struct net *net, u8 res)
639{
640 struct sock *sk;
641
642 if (!net_eq(net, &init_net))
643 return NULL;
644
645 rcu_read_lock();
646 sk = rcu_dereference(pnres.sk[res]);
647 if (sk)
648 sock_hold(sk);
649 rcu_read_unlock();
650 return sk;
651}
652
653static DEFINE_MUTEX(resource_mutex);
654
655int pn_sock_bind_res(struct sock *sk, u8 res)
656{
657 int ret = -EADDRINUSE;
658
659 if (!net_eq(sock_net(sk), &init_net))
660 return -ENOIOCTLCMD;
661 if (!capable(CAP_SYS_ADMIN))
662 return -EPERM;
663 if (pn_socket_autobind(sk->sk_socket))
664 return -EAGAIN;
665
666 mutex_lock(&resource_mutex);
667 if (pnres.sk[res] == NULL) {
668 sock_hold(sk);
669 rcu_assign_pointer(pnres.sk[res], sk);
670 ret = 0;
671 }
672 mutex_unlock(&resource_mutex);
673 return ret;
674}
675
676int pn_sock_unbind_res(struct sock *sk, u8 res)
677{
678 int ret = -ENOENT;
679
680 if (!capable(CAP_SYS_ADMIN))
681 return -EPERM;
682
683 mutex_lock(&resource_mutex);
684 if (pnres.sk[res] == sk) {
685 RCU_INIT_POINTER(pnres.sk[res], NULL);
686 ret = 0;
687 }
688 mutex_unlock(&resource_mutex);
689
690 if (ret == 0) {
691 synchronize_rcu();
692 sock_put(sk);
693 }
694 return ret;
695}
696
697void pn_sock_unbind_all_res(struct sock *sk)
698{
699 unsigned int res, match = 0;
700
701 mutex_lock(&resource_mutex);
702 for (res = 0; res < 256; res++) {
703 if (pnres.sk[res] == sk) {
704 RCU_INIT_POINTER(pnres.sk[res], NULL);
705 match++;
706 }
707 }
708 mutex_unlock(&resource_mutex);
709
710 while (match > 0) {
711 __sock_put(sk);
712 match--;
713 }
714
715}
716
717#ifdef CONFIG_PROC_FS
718static struct sock **pn_res_get_idx(struct seq_file *seq, loff_t pos)
719{
720 struct net *net = seq_file_net(seq);
721 unsigned int i;
722
723 if (!net_eq(net, &init_net))
724 return NULL;
725
726 for (i = 0; i < 256; i++) {
727 if (pnres.sk[i] == NULL)
728 continue;
729 if (!pos)
730 return pnres.sk + i;
731 pos--;
732 }
733 return NULL;
734}
735
736static struct sock **pn_res_get_next(struct seq_file *seq, struct sock **sk)
737{
738 struct net *net = seq_file_net(seq);
739 unsigned int i;
740
741 BUG_ON(!net_eq(net, &init_net));
742
743 for (i = (sk - pnres.sk) + 1; i < 256; i++)
744 if (pnres.sk[i])
745 return pnres.sk + i;
746 return NULL;
747}
748
749static void *pn_res_seq_start(struct seq_file *seq, loff_t *pos)
750 __acquires(resource_mutex)
751{
752 mutex_lock(&resource_mutex);
753 return *pos ? pn_res_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
754}
755
756static void *pn_res_seq_next(struct seq_file *seq, void *v, loff_t *pos)
757{
758 struct sock **sk;
759
760 if (v == SEQ_START_TOKEN)
761 sk = pn_res_get_idx(seq, 0);
762 else
763 sk = pn_res_get_next(seq, v);
764 (*pos)++;
765 return sk;
766}
767
768static void pn_res_seq_stop(struct seq_file *seq, void *v)
769 __releases(resource_mutex)
770{
771 mutex_unlock(&resource_mutex);
772}
773
774static int pn_res_seq_show(struct seq_file *seq, void *v)
775{
776 seq_setwidth(seq, 63);
777 if (v == SEQ_START_TOKEN)
778 seq_puts(seq, "rs uid inode");
779 else {
780 struct sock **psk = v;
781 struct sock *sk = *psk;
782
783 seq_printf(seq, "%02X %5u %lu",
784 (int) (psk - pnres.sk),
785 from_kuid_munged(seq_user_ns(seq), sock_i_uid(sk)),
786 sock_i_ino(sk));
787 }
788 seq_pad(seq, '\n');
789 return 0;
790}
791
792const struct seq_operations pn_res_seq_ops = {
793 .start = pn_res_seq_start,
794 .next = pn_res_seq_next,
795 .stop = pn_res_seq_stop,
796 .show = pn_res_seq_show,
797};
798#endif
799
