Linux Audio

Check our new training course

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