[net-next-2.6.git] / net / phonet / socket.c

/*
 * File: socket.c
 *
 * Phonet sockets
 *
 * Copyright (C) 2008 Nokia Corporation.
 *
 * Contact: Remi Denis-Courmont <remi.denis-courmont@nokia.com>
 * Original author: Sakari Ailus <sakari.ailus@nokia.com>
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * version 2 as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA
 */

#include <linux/kernel.h>
#include <linux/net.h>
#include <linux/poll.h>
#include <net/sock.h>
#include <net/tcp_states.h>

#include <linux/phonet.h>
#include <net/phonet/phonet.h>
#include <net/phonet/pep.h>
#include <net/phonet/pn_dev.h>

static int pn_socket_release(struct socket *sock)
{
	struct sock *sk = sock->sk;

	if (sk) {
		sock->sk = NULL;
		sk->sk_prot->close(sk, 0);
	}
	return 0;
}

#define PN_HASHSIZE	16
#define PN_HASHMASK	(PN_HASHSIZE-1)


static struct  {
	struct hlist_head hlist[PN_HASHSIZE];
	spinlock_t lock;
} pnsocks;

void __init pn_sock_init(void)
{
	unsigned i;

	for (i = 0; i < PN_HASHSIZE; i++)
		INIT_HLIST_HEAD(pnsocks.hlist + i);
	spin_lock_init(&pnsocks.lock);
}

static struct hlist_head *pn_hash_list(u16 obj)
{
	return pnsocks.hlist + (obj & PN_HASHMASK);
}

/*
 * Find address based on socket address, match only certain fields.
 * Also grab sock if it was found. Remember to sock_put it later.
 */
struct sock *pn_find_sock_by_sa(struct net *net, const struct sockaddr_pn *spn)
{
	struct hlist_node *node;
	struct sock *sknode;
	struct sock *rval = NULL;
	u16 obj = pn_sockaddr_get_object(spn);
	u8 res = spn->spn_resource;
	struct hlist_head *hlist = pn_hash_list(obj);

	spin_lock_bh(&pnsocks.lock);

	sk_for_each(sknode, node, hlist) {
		struct pn_sock *pn = pn_sk(sknode);
		BUG_ON(!pn->sobject); /* unbound socket */

		if (!net_eq(sock_net(sknode), net))
			continue;
		if (pn_port(obj)) {
			/* Look up socket by port */
			if (pn_port(pn->sobject) != pn_port(obj))
				continue;
		} else {
			/* If port is zero, look up by resource */
			if (pn->resource != res)
				continue;
		}
		if (pn_addr(pn->sobject) &&
		    pn_addr(pn->sobject) != pn_addr(obj))
			continue;

		rval = sknode;
		sock_hold(sknode);
		break;
	}

	spin_unlock_bh(&pnsocks.lock);

	return rval;
}

/* Deliver a broadcast packet (only in bottom-half) */
void pn_deliver_sock_broadcast(struct net *net, struct sk_buff *skb)
{
	struct hlist_head *hlist = pnsocks.hlist;
	unsigned h;

	spin_lock(&pnsocks.lock);
	for (h = 0; h < PN_HASHSIZE; h++) {
		struct hlist_node *node;
		struct sock *sknode;

		sk_for_each(sknode, node, hlist) {
			struct sk_buff *clone;

			if (!net_eq(sock_net(sknode), net))
				continue;
			if (!sock_flag(sknode, SOCK_BROADCAST))
				continue;

			clone = skb_clone(skb, GFP_ATOMIC);
			if (clone) {
				sock_hold(sknode);
				sk_receive_skb(sknode, clone, 0);
			}
		}
		hlist++;
	}
	spin_unlock(&pnsocks.lock);
}

void pn_sock_hash(struct sock *sk)
{
	struct hlist_head *hlist = pn_hash_list(pn_sk(sk)->sobject);

	spin_lock_bh(&pnsocks.lock);
	sk_add_node(sk, hlist);
	spin_unlock_bh(&pnsocks.lock);
}
EXPORT_SYMBOL(pn_sock_hash);

void pn_sock_unhash(struct sock *sk)
{
	spin_lock_bh(&pnsocks.lock);
	sk_del_node_init(sk);
	spin_unlock_bh(&pnsocks.lock);
}
EXPORT_SYMBOL(pn_sock_unhash);

static DEFINE_MUTEX(port_mutex);

static int pn_socket_bind(struct socket *sock, struct sockaddr *addr, int len)
{
	struct sock *sk = sock->sk;
	struct pn_sock *pn = pn_sk(sk);
	struct sockaddr_pn *spn = (struct sockaddr_pn *)addr;
	int err;
	u16 handle;
	u8 saddr;

	if (sk->sk_prot->bind)
		return sk->sk_prot->bind(sk, addr, len);

	if (len < sizeof(struct sockaddr_pn))
		return -EINVAL;
	if (spn->spn_family != AF_PHONET)
		return -EAFNOSUPPORT;

	handle = pn_sockaddr_get_object((struct sockaddr_pn *)addr);
	saddr = pn_addr(handle);
	if (saddr && phonet_address_lookup(sock_net(sk), saddr))
		return -EADDRNOTAVAIL;

	lock_sock(sk);
	if (sk->sk_state != TCP_CLOSE || pn_port(pn->sobject)) {
		err = -EINVAL; /* attempt to rebind */
		goto out;
	}
	WARN_ON(sk_hashed(sk));
	mutex_lock(&port_mutex);
	err = sk->sk_prot->get_port(sk, pn_port(handle));
	if (err)
		goto out_port;

	/* get_port() sets the port, bind() sets the address if applicable */
	pn->sobject = pn_object(saddr, pn_port(pn->sobject));
	pn->resource = spn->spn_resource;

	/* Enable RX on the socket */
	sk->sk_prot->hash(sk);
out_port:
	mutex_unlock(&port_mutex);
out:
	release_sock(sk);
	return err;
}

static int pn_socket_autobind(struct socket *sock)
{
	struct sockaddr_pn sa;
	int err;

	memset(&sa, 0, sizeof(sa));
	sa.spn_family = AF_PHONET;
	err = pn_socket_bind(sock, (struct sockaddr *)&sa,
				sizeof(struct sockaddr_pn));
	if (err != -EINVAL)
		return err;
	BUG_ON(!pn_port(pn_sk(sock->sk)->sobject));
	return 0; /* socket was already bound */
}

static int pn_socket_accept(struct socket *sock, struct socket *newsock,
				int flags)
{
	struct sock *sk = sock->sk;
	struct sock *newsk;
	int err;

	newsk = sk->sk_prot->accept(sk, flags, &err);
	if (!newsk)
		return err;

	lock_sock(newsk);
	sock_graft(newsk, newsock);
	newsock->state = SS_CONNECTED;
	release_sock(newsk);
	return 0;
}

static int pn_socket_getname(struct socket *sock, struct sockaddr *addr,
				int *sockaddr_len, int peer)
{
	struct sock *sk = sock->sk;
	struct pn_sock *pn = pn_sk(sk);

	memset(addr, 0, sizeof(struct sockaddr_pn));
	addr->sa_family = AF_PHONET;
	if (!peer) /* Race with bind() here is userland's problem. */
		pn_sockaddr_set_object((struct sockaddr_pn *)addr,
					pn->sobject);

	*sockaddr_len = sizeof(struct sockaddr_pn);
	return 0;
}

static unsigned int pn_socket_poll(struct file *file, struct socket *sock,
					poll_table *wait)
{
	struct sock *sk = sock->sk;
	struct pep_sock *pn = pep_sk(sk);
	unsigned int mask = 0;

	poll_wait(file, &sock->wait, wait);

	switch (sk->sk_state) {
	case TCP_LISTEN:
		return hlist_empty(&pn->ackq) ? 0 : POLLIN;
	case TCP_CLOSE:
		return POLLERR;
	}

	if (!skb_queue_empty(&sk->sk_receive_queue))
		mask |= POLLIN | POLLRDNORM;
	if (!skb_queue_empty(&pn->ctrlreq_queue))
		mask |= POLLPRI;
	if (!mask && sk->sk_state == TCP_CLOSE_WAIT)
		return POLLHUP;

	if (sk->sk_state == TCP_ESTABLISHED && atomic_read(&pn->tx_credits))
		mask |= POLLOUT | POLLWRNORM | POLLWRBAND;

	return mask;
}

static int pn_socket_ioctl(struct socket *sock, unsigned int cmd,
				unsigned long arg)
{
	struct sock *sk = sock->sk;
	struct pn_sock *pn = pn_sk(sk);

	if (cmd == SIOCPNGETOBJECT) {
		struct net_device *dev;
		u16 handle;
		u8 saddr;

		if (get_user(handle, (__u16 __user *)arg))
			return -EFAULT;

		lock_sock(sk);
		if (sk->sk_bound_dev_if)
			dev = dev_get_by_index(sock_net(sk),
						sk->sk_bound_dev_if);
		else
			dev = phonet_device_get(sock_net(sk));
		if (dev && (dev->flags & IFF_UP))
			saddr = phonet_address_get(dev, pn_addr(handle));
		else
			saddr = PN_NO_ADDR;
		release_sock(sk);

		if (dev)
			dev_put(dev);
		if (saddr == PN_NO_ADDR)
			return -EHOSTUNREACH;

		handle = pn_object(saddr, pn_port(pn->sobject));
		return put_user(handle, (__u16 __user *)arg);
	}

	return sk->sk_prot->ioctl(sk, cmd, arg);
}

static int pn_socket_listen(struct socket *sock, int backlog)
{
	struct sock *sk = sock->sk;
	int err = 0;

	if (sock->state != SS_UNCONNECTED)
		return -EINVAL;
	if (pn_socket_autobind(sock))
		return -ENOBUFS;

	lock_sock(sk);
	if (sk->sk_state != TCP_CLOSE) {
		err = -EINVAL;
		goto out;
	}

	sk->sk_state = TCP_LISTEN;
	sk->sk_ack_backlog = 0;
	sk->sk_max_ack_backlog = backlog;
out:
	release_sock(sk);
	return err;
}

static int pn_socket_sendmsg(struct kiocb *iocb, struct socket *sock,
				struct msghdr *m, size_t total_len)
{
	struct sock *sk = sock->sk;

	if (pn_socket_autobind(sock))
		return -EAGAIN;

	return sk->sk_prot->sendmsg(iocb, sk, m, total_len);
}

const struct proto_ops phonet_dgram_ops = {
	.family		= AF_PHONET,
	.owner		= THIS_MODULE,
	.release	= pn_socket_release,
	.bind		= pn_socket_bind,
	.connect	= sock_no_connect,
	.socketpair	= sock_no_socketpair,
	.accept		= sock_no_accept,
	.getname	= pn_socket_getname,
	.poll		= datagram_poll,
	.ioctl		= pn_socket_ioctl,
	.listen		= sock_no_listen,
	.shutdown	= sock_no_shutdown,
	.setsockopt	= sock_no_setsockopt,
	.getsockopt	= sock_no_getsockopt,
#ifdef CONFIG_COMPAT
	.compat_setsockopt = sock_no_setsockopt,
	.compat_getsockopt = sock_no_getsockopt,
#endif
	.sendmsg	= pn_socket_sendmsg,
	.recvmsg	= sock_common_recvmsg,
	.mmap		= sock_no_mmap,
	.sendpage	= sock_no_sendpage,
};

const struct proto_ops phonet_stream_ops = {
	.family		= AF_PHONET,
	.owner		= THIS_MODULE,
	.release	= pn_socket_release,
	.bind		= pn_socket_bind,
	.connect	= sock_no_connect,
	.socketpair	= sock_no_socketpair,
	.accept		= pn_socket_accept,
	.getname	= pn_socket_getname,
	.poll		= pn_socket_poll,
	.ioctl		= pn_socket_ioctl,
	.listen		= pn_socket_listen,
	.shutdown	= sock_no_shutdown,
	.setsockopt	= sock_common_setsockopt,
	.getsockopt	= sock_common_getsockopt,
#ifdef CONFIG_COMPAT
	.compat_setsockopt = compat_sock_common_setsockopt,
	.compat_getsockopt = compat_sock_common_getsockopt,
#endif
	.sendmsg	= pn_socket_sendmsg,
	.recvmsg	= sock_common_recvmsg,
	.mmap		= sock_no_mmap,
	.sendpage	= sock_no_sendpage,
};
EXPORT_SYMBOL(phonet_stream_ops);

/* allocate port for a socket */
int pn_sock_get_port(struct sock *sk, unsigned short sport)
{
	static int port_cur;
	struct net *net = sock_net(sk);
	struct pn_sock *pn = pn_sk(sk);
	struct sockaddr_pn try_sa;
	struct sock *tmpsk;

	memset(&try_sa, 0, sizeof(struct sockaddr_pn));
	try_sa.spn_family = AF_PHONET;
	WARN_ON(!mutex_is_locked(&port_mutex));
	if (!sport) {
		/* search free port */
		int port, pmin, pmax;

		phonet_get_local_port_range(&pmin, &pmax);
		for (port = pmin; port <= pmax; port++) {
			port_cur++;
			if (port_cur < pmin || port_cur > pmax)
				port_cur = pmin;

			pn_sockaddr_set_port(&try_sa, port_cur);
			tmpsk = pn_find_sock_by_sa(net, &try_sa);
			if (tmpsk == NULL) {
				sport = port_cur;
				goto found;
			} else
				sock_put(tmpsk);
		}
	} else {
		/* try to find specific port */
		pn_sockaddr_set_port(&try_sa, sport);
		tmpsk = pn_find_sock_by_sa(net, &try_sa);
		if (tmpsk == NULL)
			/* No sock there! We can use that port... */
			goto found;
		else
			sock_put(tmpsk);
	}
	/* the port must be in use already */
	return -EADDRINUSE;

found:
	pn->sobject = pn_object(pn_addr(pn->sobject), sport);
	return 0;
}
EXPORT_SYMBOL(pn_sock_get_port);

#ifdef CONFIG_PROC_FS
static struct sock *pn_sock_get_idx(struct seq_file *seq, loff_t pos)
{
	struct net *net = seq_file_net(seq);
	struct hlist_head *hlist = pnsocks.hlist;
	struct hlist_node *node;
	struct sock *sknode;
	unsigned h;

	for (h = 0; h < PN_HASHSIZE; h++) {
		sk_for_each(sknode, node, hlist) {
			if (!net_eq(net, sock_net(sknode)))
				continue;
			if (!pos)
				return sknode;
			pos--;
		}
		hlist++;
	}
	return NULL;
}

static struct sock *pn_sock_get_next(struct seq_file *seq, struct sock *sk)
{
	struct net *net = seq_file_net(seq);

	do
		sk = sk_next(sk);
	while (sk && !net_eq(net, sock_net(sk)));

	return sk;
}

static void *pn_sock_seq_start(struct seq_file *seq, loff_t *pos)
	__acquires(pnsocks.lock)
{
	spin_lock_bh(&pnsocks.lock);
	return *pos ? pn_sock_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
}

static void *pn_sock_seq_next(struct seq_file *seq, void *v, loff_t *pos)
{
	struct sock *sk;

	if (v == SEQ_START_TOKEN)
		sk = pn_sock_get_idx(seq, 0);
	else
		sk = pn_sock_get_next(seq, v);
	(*pos)++;
	return sk;
}

static void pn_sock_seq_stop(struct seq_file *seq, void *v)
	__releases(pnsocks.lock)
{
	spin_unlock_bh(&pnsocks.lock);
}

static int pn_sock_seq_show(struct seq_file *seq, void *v)
{
	int len;

	if (v == SEQ_START_TOKEN)
		seq_printf(seq, "%s%n", "pt  loc  rem rs st tx_queue rx_queue "
			"  uid inode ref pointer drops", &len);
	else {
		struct sock *sk = v;
		struct pn_sock *pn = pn_sk(sk);

		seq_printf(seq, "%2d %04X:%04X:%02X %02X %08X:%08X %5d %lu "
			"%d %p %d%n",
			sk->sk_protocol, pn->sobject, 0, pn->resource,
			sk->sk_state,
			sk_wmem_alloc_get(sk), sk_rmem_alloc_get(sk),
			sock_i_uid(sk), sock_i_ino(sk),
			atomic_read(&sk->sk_refcnt), sk,
			atomic_read(&sk->sk_drops), &len);
	}
	seq_printf(seq, "%*s\n", 127 - len, "");
	return 0;
}

static const struct seq_operations pn_sock_seq_ops = {
	.start = pn_sock_seq_start,
	.next = pn_sock_seq_next,
	.stop = pn_sock_seq_stop,
	.show = pn_sock_seq_show,
};

static int pn_sock_open(struct inode *inode, struct file *file)
{
	return seq_open_net(inode, file, &pn_sock_seq_ops,
				sizeof(struct seq_net_private));
}

const struct file_operations pn_sock_seq_fops = {
	.owner = THIS_MODULE,
	.open = pn_sock_open,
	.read = seq_read,
	.llseek = seq_lseek,
	.release = seq_release_net,
};
#endif
Commit	Line	Data
ba113a94 RDC	1	/*
	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
	26	#include <linux/kernel.h>
	27	#include <linux/net.h>
9995a32b	28	#include <linux/poll.h>
ba113a94 RDC	29	#include <net/sock.h>
	30	#include <net/tcp_states.h>
	31
	32	#include <linux/phonet.h>
	33	#include <net/phonet/phonet.h>
9995a32b	34	#include <net/phonet/pep.h>
ba113a94 RDC	35	#include <net/phonet/pn_dev.h>
	36
	37	static int pn_socket_release(struct socket *sock)
	38	{
	39	struct sock *sk = sock->sk;
	40
	41	if (sk) {
	42	sock->sk = NULL;
	43	sk->sk_prot->close(sk, 0);
	44	}
	45	return 0;
	46	}
	47
6b0d07ba RDC	48	#define PN_HASHSIZE 16
	49	#define PN_HASHMASK (PN_HASHSIZE-1)
	50
	51
ba113a94	52	static struct {
6b0d07ba	53	struct hlist_head hlist[PN_HASHSIZE];
ba113a94	54	spinlock_t lock;
6b0d07ba RDC	55	} pnsocks;
	56
	57	void __init pn_sock_init(void)
	58	{
	59	unsigned i;
	60
	61	for (i = 0; i < PN_HASHSIZE; i++)
	62	INIT_HLIST_HEAD(pnsocks.hlist + i);
	63	spin_lock_init(&pnsocks.lock);
	64	}
	65
	66	static struct hlist_head *pn_hash_list(u16 obj)
	67	{
	68	return pnsocks.hlist + (obj & PN_HASHMASK);
	69	}
ba113a94 RDC	70
	71	/*
	72	* Find address based on socket address, match only certain fields.
	73	* Also grab sock if it was found. Remember to sock_put it later.
	74	*/
52404881	75	struct sock pn_find_sock_by_sa(struct net net, const struct sockaddr_pn *spn)
ba113a94 RDC	76	{
	77	struct hlist_node *node;
	78	struct sock *sknode;
	79	struct sock *rval = NULL;
	80	u16 obj = pn_sockaddr_get_object(spn);
	81	u8 res = spn->spn_resource;
6b0d07ba	82	struct hlist_head *hlist = pn_hash_list(obj);
ba113a94 RDC	83
	84	spin_lock_bh(&pnsocks.lock);
	85
6b0d07ba	86	sk_for_each(sknode, node, hlist) {
ba113a94 RDC	87	struct pn_sock *pn = pn_sk(sknode);
	88	BUG_ON(!pn->sobject); /* unbound socket */
	89
52404881 RDC	90	if (!net_eq(sock_net(sknode), net))
52404881 RDC	91	continue;
ba113a94 RDC	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	}
f64f9e71 JP	101	if (pn_addr(pn->sobject) &&
f64f9e71 JP	102	pn_addr(pn->sobject) != pn_addr(obj))
ba113a94 RDC	103	continue;
	104
	105	rval = sknode;
	106	sock_hold(sknode);
	107	break;
	108	}
	109
	110	spin_unlock_bh(&pnsocks.lock);
	111
	112	return rval;
f14001fc RDC	113	}
	114
	115	/* Deliver a broadcast packet (only in bottom-half) */
	116	void pn_deliver_sock_broadcast(struct net net, struct sk_buff skb)
	117	{
6b0d07ba RDC	118	struct hlist_head *hlist = pnsocks.hlist;
6b0d07ba RDC	119	unsigned h;
f14001fc RDC	120
f14001fc RDC	121	spin_lock(&pnsocks.lock);
6b0d07ba RDC	122	for (h = 0; h < PN_HASHSIZE; h++) {
	123	struct hlist_node *node;
	124	struct sock *sknode;
f14001fc	125
6b0d07ba RDC	126	sk_for_each(sknode, node, hlist) {
6b0d07ba RDC	127	struct sk_buff *clone;
ba113a94	128
6b0d07ba RDC	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	}
21912d1c	139	}
6b0d07ba	140	hlist++;
f14001fc RDC	141	}
f14001fc RDC	142	spin_unlock(&pnsocks.lock);
ba113a94 RDC	143	}
	144
	145	void pn_sock_hash(struct sock *sk)
	146	{
6b0d07ba RDC	147	struct hlist_head *hlist = pn_hash_list(pn_sk(sk)->sobject);
6b0d07ba RDC	148
ba113a94	149	spin_lock_bh(&pnsocks.lock);
6b0d07ba	150	sk_add_node(sk, hlist);
ba113a94 RDC	151	spin_unlock_bh(&pnsocks.lock);
	152	}
	153	EXPORT_SYMBOL(pn_sock_hash);
	154
	155	void pn_sock_unhash(struct sock *sk)
	156	{
	157	spin_lock_bh(&pnsocks.lock);
	158	sk_del_node_init(sk);
	159	spin_unlock_bh(&pnsocks.lock);
	160	}
	161	EXPORT_SYMBOL(pn_sock_unhash);
	162
582b0b61 RDC	163	static DEFINE_MUTEX(port_mutex);
582b0b61 RDC	164
ba113a94 RDC	165	static int pn_socket_bind(struct socket sock, struct sockaddr addr, int len)
	166	{
	167	struct sock *sk = sock->sk;
	168	struct pn_sock *pn = pn_sk(sk);
	169	struct sockaddr_pn spn = (struct sockaddr_pn )addr;
	170	int err;
	171	u16 handle;
	172	u8 saddr;
	173
	174	if (sk->sk_prot->bind)
	175	return sk->sk_prot->bind(sk, addr, len);
	176
	177	if (len < sizeof(struct sockaddr_pn))
	178	return -EINVAL;
	179	if (spn->spn_family != AF_PHONET)
	180	return -EAFNOSUPPORT;
	181
	182	handle = pn_sockaddr_get_object((struct sockaddr_pn *)addr);
	183	saddr = pn_addr(handle);
52404881	184	if (saddr && phonet_address_lookup(sock_net(sk), saddr))
ba113a94 RDC	185	return -EADDRNOTAVAIL;
	186
	187	lock_sock(sk);
	188	if (sk->sk_state != TCP_CLOSE \|\| pn_port(pn->sobject)) {
	189	err = -EINVAL; /* attempt to rebind */
	190	goto out;
	191	}
582b0b61 RDC	192	WARN_ON(sk_hashed(sk));
582b0b61 RDC	193	mutex_lock(&port_mutex);
ba113a94 RDC	194	err = sk->sk_prot->get_port(sk, pn_port(handle));
ba113a94 RDC	195	if (err)
582b0b61	196	goto out_port;
ba113a94 RDC	197
	198	/* get_port() sets the port, bind() sets the address if applicable */
	199	pn->sobject = pn_object(saddr, pn_port(pn->sobject));
	200	pn->resource = spn->spn_resource;
	201
	202	/* Enable RX on the socket */
	203	sk->sk_prot->hash(sk);
582b0b61 RDC	204	out_port:
582b0b61 RDC	205	mutex_unlock(&port_mutex);
ba113a94 RDC	206	out:
	207	release_sock(sk);
	208	return err;
	209	}
	210
	211	static int pn_socket_autobind(struct socket *sock)
	212	{
	213	struct sockaddr_pn sa;
	214	int err;
	215
	216	memset(&sa, 0, sizeof(sa));
	217	sa.spn_family = AF_PHONET;
	218	err = pn_socket_bind(sock, (struct sockaddr *)&sa,
	219	sizeof(struct sockaddr_pn));
	220	if (err != -EINVAL)
	221	return err;
	222	BUG_ON(!pn_port(pn_sk(sock->sk)->sobject));
	223	return 0; /* socket was already bound */
	224	}
	225
9995a32b RDC	226	static int pn_socket_accept(struct socket sock, struct socket newsock,
	227	int flags)
	228	{
	229	struct sock *sk = sock->sk;
	230	struct sock *newsk;
	231	int err;
	232
	233	newsk = sk->sk_prot->accept(sk, flags, &err);
	234	if (!newsk)
	235	return err;
	236
	237	lock_sock(newsk);
	238	sock_graft(newsk, newsock);
	239	newsock->state = SS_CONNECTED;
	240	release_sock(newsk);
	241	return 0;
	242	}
	243
ba113a94 RDC	244	static int pn_socket_getname(struct socket sock, struct sockaddr addr,
	245	int *sockaddr_len, int peer)
	246	{
	247	struct sock *sk = sock->sk;
	248	struct pn_sock *pn = pn_sk(sk);
	249
	250	memset(addr, 0, sizeof(struct sockaddr_pn));
	251	addr->sa_family = AF_PHONET;
	252	if (!peer) /* Race with bind() here is userland's problem. */
	253	pn_sockaddr_set_object((struct sockaddr_pn *)addr,
	254	pn->sobject);
	255
	256	*sockaddr_len = sizeof(struct sockaddr_pn);
	257	return 0;
	258	}
	259
9995a32b RDC	260	static unsigned int pn_socket_poll(struct file file, struct socket sock,
	261	poll_table *wait)
	262	{
	263	struct sock *sk = sock->sk;
	264	struct pep_sock *pn = pep_sk(sk);
	265	unsigned int mask = 0;
	266
	267	poll_wait(file, &sock->wait, wait);
	268
	269	switch (sk->sk_state) {
	270	case TCP_LISTEN:
	271	return hlist_empty(&pn->ackq) ? 0 : POLLIN;
	272	case TCP_CLOSE:
	273	return POLLERR;
	274	}
	275
	276	if (!skb_queue_empty(&sk->sk_receive_queue))
	277	mask \|= POLLIN \| POLLRDNORM;
c41bd97f RDC	278	if (!skb_queue_empty(&pn->ctrlreq_queue))
	279	mask \|= POLLPRI;
	280	if (!mask && sk->sk_state == TCP_CLOSE_WAIT)
9995a32b RDC	281	return POLLHUP;
9995a32b RDC	282
be677730	283	if (sk->sk_state == TCP_ESTABLISHED && atomic_read(&pn->tx_credits))
9995a32b RDC	284	mask \|= POLLOUT \| POLLWRNORM \| POLLWRBAND;
	285
	286	return mask;
	287	}
	288
ba113a94 RDC	289	static int pn_socket_ioctl(struct socket *sock, unsigned int cmd,
	290	unsigned long arg)
	291	{
	292	struct sock *sk = sock->sk;
	293	struct pn_sock *pn = pn_sk(sk);
	294
	295	if (cmd == SIOCPNGETOBJECT) {
	296	struct net_device *dev;
	297	u16 handle;
	298	u8 saddr;
	299
	300	if (get_user(handle, (__u16 __user *)arg))
	301	return -EFAULT;
	302
	303	lock_sock(sk);
	304	if (sk->sk_bound_dev_if)
	305	dev = dev_get_by_index(sock_net(sk),
	306	sk->sk_bound_dev_if);
	307	else
	308	dev = phonet_device_get(sock_net(sk));
	309	if (dev && (dev->flags & IFF_UP))
	310	saddr = phonet_address_get(dev, pn_addr(handle));
	311	else
	312	saddr = PN_NO_ADDR;
	313	release_sock(sk);
	314
	315	if (dev)
	316	dev_put(dev);
	317	if (saddr == PN_NO_ADDR)
	318	return -EHOSTUNREACH;
	319
	320	handle = pn_object(saddr, pn_port(pn->sobject));
	321	return put_user(handle, (__u16 __user *)arg);
	322	}
	323
	324	return sk->sk_prot->ioctl(sk, cmd, arg);
	325	}
	326
9995a32b RDC	327	static int pn_socket_listen(struct socket *sock, int backlog)
	328	{
	329	struct sock *sk = sock->sk;
	330	int err = 0;
	331
	332	if (sock->state != SS_UNCONNECTED)
	333	return -EINVAL;
	334	if (pn_socket_autobind(sock))
	335	return -ENOBUFS;
	336
	337	lock_sock(sk);
	338	if (sk->sk_state != TCP_CLOSE) {
	339	err = -EINVAL;
	340	goto out;
	341	}
	342
	343	sk->sk_state = TCP_LISTEN;
	344	sk->sk_ack_backlog = 0;
	345	sk->sk_max_ack_backlog = backlog;
	346	out:
	347	release_sock(sk);
	348	return err;
	349	}
	350
ba113a94 RDC	351	static int pn_socket_sendmsg(struct kiocb iocb, struct socket sock,
	352	struct msghdr *m, size_t total_len)
	353	{
	354	struct sock *sk = sock->sk;
	355
	356	if (pn_socket_autobind(sock))
	357	return -EAGAIN;
	358
	359	return sk->sk_prot->sendmsg(iocb, sk, m, total_len);
	360	}
	361
	362	const struct proto_ops phonet_dgram_ops = {
	363	.family = AF_PHONET,
	364	.owner = THIS_MODULE,
	365	.release = pn_socket_release,
	366	.bind = pn_socket_bind,
	367	.connect = sock_no_connect,
	368	.socketpair = sock_no_socketpair,
	369	.accept = sock_no_accept,
	370	.getname = pn_socket_getname,
	371	.poll = datagram_poll,
	372	.ioctl = pn_socket_ioctl,
	373	.listen = sock_no_listen,
	374	.shutdown = sock_no_shutdown,
	375	.setsockopt = sock_no_setsockopt,
	376	.getsockopt = sock_no_getsockopt,
	377	#ifdef CONFIG_COMPAT
	378	.compat_setsockopt = sock_no_setsockopt,
	379	.compat_getsockopt = sock_no_getsockopt,
	380	#endif
	381	.sendmsg = pn_socket_sendmsg,
	382	.recvmsg = sock_common_recvmsg,
	383	.mmap = sock_no_mmap,
	384	.sendpage = sock_no_sendpage,
	385	};
	386
9995a32b RDC	387	const struct proto_ops phonet_stream_ops = {
	388	.family = AF_PHONET,
	389	.owner = THIS_MODULE,
	390	.release = pn_socket_release,
	391	.bind = pn_socket_bind,
	392	.connect = sock_no_connect,
	393	.socketpair = sock_no_socketpair,
	394	.accept = pn_socket_accept,
	395	.getname = pn_socket_getname,
	396	.poll = pn_socket_poll,
	397	.ioctl = pn_socket_ioctl,
	398	.listen = pn_socket_listen,
	399	.shutdown = sock_no_shutdown,
02a47617 RDC	400	.setsockopt = sock_common_setsockopt,
02a47617 RDC	401	.getsockopt = sock_common_getsockopt,
9995a32b	402	#ifdef CONFIG_COMPAT
02a47617 RDC	403	.compat_setsockopt = compat_sock_common_setsockopt,
02a47617 RDC	404	.compat_getsockopt = compat_sock_common_getsockopt,
9995a32b RDC	405	#endif
	406	.sendmsg = pn_socket_sendmsg,
	407	.recvmsg = sock_common_recvmsg,
	408	.mmap = sock_no_mmap,
	409	.sendpage = sock_no_sendpage,
	410	};
	411	EXPORT_SYMBOL(phonet_stream_ops);
	412
ba113a94 RDC	413	/* allocate port for a socket */
	414	int pn_sock_get_port(struct sock *sk, unsigned short sport)
	415	{
	416	static int port_cur;
52404881	417	struct net *net = sock_net(sk);
ba113a94 RDC	418	struct pn_sock *pn = pn_sk(sk);
	419	struct sockaddr_pn try_sa;
	420	struct sock *tmpsk;
	421
	422	memset(&try_sa, 0, sizeof(struct sockaddr_pn));
	423	try_sa.spn_family = AF_PHONET;
582b0b61	424	WARN_ON(!mutex_is_locked(&port_mutex));
ba113a94 RDC	425	if (!sport) {
ba113a94 RDC	426	/* search free port */
87ab4e20	427	int port, pmin, pmax;
ba113a94	428
87ab4e20	429	phonet_get_local_port_range(&pmin, &pmax);
ba113a94 RDC	430	for (port = pmin; port <= pmax; port++) {
	431	port_cur++;
	432	if (port_cur < pmin \|\| port_cur > pmax)
	433	port_cur = pmin;
	434
	435	pn_sockaddr_set_port(&try_sa, port_cur);
52404881	436	tmpsk = pn_find_sock_by_sa(net, &try_sa);
ba113a94 RDC	437	if (tmpsk == NULL) {
	438	sport = port_cur;
	439	goto found;
	440	} else
	441	sock_put(tmpsk);
	442	}
	443	} else {
	444	/* try to find specific port */
	445	pn_sockaddr_set_port(&try_sa, sport);
52404881	446	tmpsk = pn_find_sock_by_sa(net, &try_sa);
ba113a94 RDC	447	if (tmpsk == NULL)
	448	/* No sock there! We can use that port... */
	449	goto found;
	450	else
	451	sock_put(tmpsk);
	452	}
ba113a94 RDC	453	/* the port must be in use already */
	454	return -EADDRINUSE;
	455
	456	found:
ba113a94 RDC	457	pn->sobject = pn_object(pn_addr(pn->sobject), sport);
	458	return 0;
	459	}
	460	EXPORT_SYMBOL(pn_sock_get_port);
c1dc13e9	461
ae6e2aef	462	#ifdef CONFIG_PROC_FS
c1dc13e9 RDC	463	static struct sock pn_sock_get_idx(struct seq_file seq, loff_t pos)
	464	{
	465	struct net *net = seq_file_net(seq);
6b0d07ba	466	struct hlist_head *hlist = pnsocks.hlist;
c1dc13e9 RDC	467	struct hlist_node *node;
c1dc13e9 RDC	468	struct sock *sknode;
6b0d07ba	469	unsigned h;
c1dc13e9	470
6b0d07ba RDC	471	for (h = 0; h < PN_HASHSIZE; h++) {
	472	sk_for_each(sknode, node, hlist) {
	473	if (!net_eq(net, sock_net(sknode)))
	474	continue;
	475	if (!pos)
	476	return sknode;
	477	pos--;
	478	}
	479	hlist++;
c1dc13e9 RDC	480	}
	481	return NULL;
	482	}
	483
	484	static struct sock pn_sock_get_next(struct seq_file seq, struct sock *sk)
	485	{
	486	struct net *net = seq_file_net(seq);
	487
	488	do
	489	sk = sk_next(sk);
	490	while (sk && !net_eq(net, sock_net(sk)));
	491
	492	return sk;
	493	}
	494
	495	static void pn_sock_seq_start(struct seq_file seq, loff_t *pos)
	496	__acquires(pnsocks.lock)
	497	{
	498	spin_lock_bh(&pnsocks.lock);
	499	return pos ? pn_sock_get_idx(seq, pos - 1) : SEQ_START_TOKEN;
	500	}
	501
	502	static void pn_sock_seq_next(struct seq_file seq, void v, loff_t pos)
	503	{
	504	struct sock *sk;
	505
	506	if (v == SEQ_START_TOKEN)
	507	sk = pn_sock_get_idx(seq, 0);
	508	else
	509	sk = pn_sock_get_next(seq, v);
	510	(*pos)++;
	511	return sk;
	512	}
	513
	514	static void pn_sock_seq_stop(struct seq_file seq, void v)
	515	__releases(pnsocks.lock)
	516	{
	517	spin_unlock_bh(&pnsocks.lock);
	518	}
	519
	520	static int pn_sock_seq_show(struct seq_file seq, void v)
	521	{
	522	int len;
	523
	524	if (v == SEQ_START_TOKEN)
	525	seq_printf(seq, "%s%n", "pt loc rem rs st tx_queue rx_queue "
	526	" uid inode ref pointer drops", &len);
	527	else {
	528	struct sock *sk = v;
	529	struct pn_sock *pn = pn_sk(sk);
	530
	531	seq_printf(seq, "%2d %04X:%04X:%02X %02X %08X:%08X %5d %lu "
	532	"%d %p %d%n",
	533	sk->sk_protocol, pn->sobject, 0, pn->resource,
	534	sk->sk_state,
	535	sk_wmem_alloc_get(sk), sk_rmem_alloc_get(sk),
	536	sock_i_uid(sk), sock_i_ino(sk),
	537	atomic_read(&sk->sk_refcnt), sk,
	538	atomic_read(&sk->sk_drops), &len);
	539	}
	540	seq_printf(seq, "%*s\n", 127 - len, "");
	541	return 0;
	542	}
	543
544	static const struct seq_operations pn_sock_seq_ops = {
545	.start = pn_sock_seq_start,
546	.next = pn_sock_seq_next,
547	.stop = pn_sock_seq_stop,
548	.show = pn_sock_seq_show,
549	};
550
551	static int pn_sock_open(struct inode inode, struct file file)
552	{
cb7d9e7f RDC	553	return seq_open_net(inode, file, &pn_sock_seq_ops,
cb7d9e7f RDC	554	sizeof(struct seq_net_private));
c1dc13e9 RDC	555	}
	556
	557	const struct file_operations pn_sock_seq_fops = {
	558	.owner = THIS_MODULE,
	559	.open = pn_sock_open,
	560	.read = seq_read,
	561	.llseek = seq_lseek,
cb7d9e7f	562	.release = seq_release_net,
c1dc13e9	563	};
ae6e2aef	564	#endif