[linux.git] / net / ieee802154 / raw.c

/*
 * Raw IEEE 802.15.4 sockets
 *
 * Copyright 2007, 2008 Siemens AG
 *
 * 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 Street, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Written by:
 * Sergey Lapin <[email protected]>
 * Dmitry Eremin-Solenikov <[email protected]>
 */

#include <linux/net.h>
#include <linux/module.h>
#include <linux/if_arp.h>
#include <linux/list.h>
#include <linux/slab.h>
#include <net/sock.h>
#include <net/af_ieee802154.h>
#include <net/ieee802154_netdev.h>

#include "af802154.h"

static HLIST_HEAD(raw_head);
static DEFINE_RWLOCK(raw_lock);

static void raw_hash(struct sock *sk)
{
	write_lock_bh(&raw_lock);
	sk_add_node(sk, &raw_head);
	sock_prot_inuse_add(sock_net(sk), sk->sk_prot, 1);
	write_unlock_bh(&raw_lock);
}

static void raw_unhash(struct sock *sk)
{
	write_lock_bh(&raw_lock);
	if (sk_del_node_init(sk))
		sock_prot_inuse_add(sock_net(sk), sk->sk_prot, -1);
	write_unlock_bh(&raw_lock);
}

static void raw_close(struct sock *sk, long timeout)
{
	sk_common_release(sk);
}

static int raw_bind(struct sock *sk, struct sockaddr *_uaddr, int len)
{
	struct ieee802154_addr addr;
	struct sockaddr_ieee802154 *uaddr = (struct sockaddr_ieee802154 *)_uaddr;
	int err = 0;
	struct net_device *dev = NULL;

	if (len < sizeof(*uaddr))
		return -EINVAL;

	uaddr = (struct sockaddr_ieee802154 *)_uaddr;
	if (uaddr->family != AF_IEEE802154)
		return -EINVAL;

	lock_sock(sk);

	ieee802154_addr_from_sa(&addr, &uaddr->addr);
	dev = ieee802154_get_dev(sock_net(sk), &addr);
	if (!dev) {
		err = -ENODEV;
		goto out;
	}

	if (dev->type != ARPHRD_IEEE802154) {
		err = -ENODEV;
		goto out_put;
	}

	sk->sk_bound_dev_if = dev->ifindex;
	sk_dst_reset(sk);

out_put:
	dev_put(dev);
out:
	release_sock(sk);

	return err;
}

static int raw_connect(struct sock *sk, struct sockaddr *uaddr,
		       int addr_len)
{
	return -ENOTSUPP;
}

static int raw_disconnect(struct sock *sk, int flags)
{
	return 0;
}

static int raw_sendmsg(struct kiocb *iocb, struct sock *sk,
		       struct msghdr *msg, size_t size)
{
	struct net_device *dev;
	unsigned int mtu;
	struct sk_buff *skb;
	int hlen, tlen;
	int err;

	if (msg->msg_flags & MSG_OOB) {
		pr_debug("msg->msg_flags = 0x%x\n", msg->msg_flags);
		return -EOPNOTSUPP;
	}

	lock_sock(sk);
	if (!sk->sk_bound_dev_if)
		dev = dev_getfirstbyhwtype(sock_net(sk), ARPHRD_IEEE802154);
	else
		dev = dev_get_by_index(sock_net(sk), sk->sk_bound_dev_if);
	release_sock(sk);

	if (!dev) {
		pr_debug("no dev\n");
		err = -ENXIO;
		goto out;
	}

	mtu = dev->mtu;
	pr_debug("name = %s, mtu = %u\n", dev->name, mtu);

	if (size > mtu) {
		pr_debug("size = %Zu, mtu = %u\n", size, mtu);
		err = -EINVAL;
		goto out_dev;
	}

	hlen = LL_RESERVED_SPACE(dev);
	tlen = dev->needed_tailroom;
	skb = sock_alloc_send_skb(sk, hlen + tlen + size,
				  msg->msg_flags & MSG_DONTWAIT, &err);
	if (!skb)
		goto out_dev;

	skb_reserve(skb, hlen);

	skb_reset_mac_header(skb);
	skb_reset_network_header(skb);

	err = memcpy_fromiovec(skb_put(skb, size), msg->msg_iov, size);
	if (err < 0)
		goto out_skb;

	skb->dev = dev;
	skb->sk  = sk;
	skb->protocol = htons(ETH_P_IEEE802154);

	dev_put(dev);

	err = dev_queue_xmit(skb);
	if (err > 0)
		err = net_xmit_errno(err);

	return err ?: size;

out_skb:
	kfree_skb(skb);
out_dev:
	dev_put(dev);
out:
	return err;
}

static int raw_recvmsg(struct kiocb *iocb, struct sock *sk, struct msghdr *msg,
		       size_t len, int noblock, int flags, int *addr_len)
{
	size_t copied = 0;
	int err = -EOPNOTSUPP;
	struct sk_buff *skb;

	skb = skb_recv_datagram(sk, flags, noblock, &err);
	if (!skb)
		goto out;

	copied = skb->len;
	if (len < copied) {
		msg->msg_flags |= MSG_TRUNC;
		copied = len;
	}

	err = skb_copy_datagram_iovec(skb, 0, msg->msg_iov, copied);
	if (err)
		goto done;

	sock_recv_ts_and_drops(msg, sk, skb);

	if (flags & MSG_TRUNC)
		copied = skb->len;
done:
	skb_free_datagram(sk, skb);
out:
	if (err)
		return err;
	return copied;
}

static int raw_rcv_skb(struct sock *sk, struct sk_buff *skb)
{
	skb = skb_share_check(skb, GFP_ATOMIC);
	if (!skb)
		return NET_RX_DROP;

	if (sock_queue_rcv_skb(sk, skb) < 0) {
		kfree_skb(skb);
		return NET_RX_DROP;
	}

	return NET_RX_SUCCESS;
}


void ieee802154_raw_deliver(struct net_device *dev, struct sk_buff *skb)
{
	struct sock *sk;

	read_lock(&raw_lock);
	sk_for_each(sk, &raw_head) {
		bh_lock_sock(sk);
		if (!sk->sk_bound_dev_if ||
		    sk->sk_bound_dev_if == dev->ifindex) {
			struct sk_buff *clone;

			clone = skb_clone(skb, GFP_ATOMIC);
			if (clone)
				raw_rcv_skb(sk, clone);
		}
		bh_unlock_sock(sk);
	}
	read_unlock(&raw_lock);
}

static int raw_getsockopt(struct sock *sk, int level, int optname,
			  char __user *optval, int __user *optlen)
{
	return -EOPNOTSUPP;
}

static int raw_setsockopt(struct sock *sk, int level, int optname,
			  char __user *optval, unsigned int optlen)
{
	return -EOPNOTSUPP;
}

struct proto ieee802154_raw_prot = {
	.name		= "IEEE-802.15.4-RAW",
	.owner		= THIS_MODULE,
	.obj_size	= sizeof(struct sock),
	.close		= raw_close,
	.bind		= raw_bind,
	.sendmsg	= raw_sendmsg,
	.recvmsg	= raw_recvmsg,
	.hash		= raw_hash,
	.unhash		= raw_unhash,
	.connect	= raw_connect,
	.disconnect	= raw_disconnect,
	.getsockopt	= raw_getsockopt,
	.setsockopt	= raw_setsockopt,
};
Commit	Line	Data
9ec76716 SL	1	/*
	2	* Raw IEEE 802.15.4 sockets
	3	*
	4	* Copyright 2007, 2008 Siemens AG
	5	*
	6	* This program is free software; you can redistribute it and/or modify
	7	* it under the terms of the GNU General Public License version 2
	8	* as published by the Free Software Foundation.
	9	*
	10	* This program is distributed in the hope that it will be useful,
	11	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	12	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	13	* GNU General Public License for more details.
	14	*
	15	* You should have received a copy of the GNU General Public License along
	16	* with this program; if not, write to the Free Software Foundation, Inc.,
	17	* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
	18	*
	19	* Written by:
	20	* Sergey Lapin <[email protected]>
	21	* Dmitry Eremin-Solenikov <[email protected]>
	22	*/
	23
	24	#include <linux/net.h>
	25	#include <linux/module.h>
	26	#include <linux/if_arp.h>
	27	#include <linux/list.h>
5a0e3ad6	28	#include <linux/slab.h>
9ec76716	29	#include <net/sock.h>
f0166e5e	30	#include <net/af_ieee802154.h>
e6278d92	31	#include <net/ieee802154_netdev.h>
9ec76716 SL	32
	33	#include "af802154.h"
	34
	35	static HLIST_HEAD(raw_head);
	36	static DEFINE_RWLOCK(raw_lock);
	37
	38	static void raw_hash(struct sock *sk)
	39	{
	40	write_lock_bh(&raw_lock);
	41	sk_add_node(sk, &raw_head);
	42	sock_prot_inuse_add(sock_net(sk), sk->sk_prot, 1);
	43	write_unlock_bh(&raw_lock);
	44	}
	45
	46	static void raw_unhash(struct sock *sk)
	47	{
	48	write_lock_bh(&raw_lock);
	49	if (sk_del_node_init(sk))
	50	sock_prot_inuse_add(sock_net(sk), sk->sk_prot, -1);
	51	write_unlock_bh(&raw_lock);
	52	}
	53
	54	static void raw_close(struct sock *sk, long timeout)
	55	{
	56	sk_common_release(sk);
	57	}
	58
e6278d92	59	static int raw_bind(struct sock sk, struct sockaddr _uaddr, int len)
9ec76716	60	{
e6278d92 PB	61	struct ieee802154_addr addr;
e6278d92 PB	62	struct sockaddr_ieee802154 uaddr = (struct sockaddr_ieee802154 )_uaddr;
9ec76716 SL	63	int err = 0;
	64	struct net_device *dev = NULL;
	65
e6278d92	66	if (len < sizeof(*uaddr))
9ec76716 SL	67	return -EINVAL;
9ec76716 SL	68
e6278d92 PB	69	uaddr = (struct sockaddr_ieee802154 *)_uaddr;
e6278d92 PB	70	if (uaddr->family != AF_IEEE802154)
9ec76716 SL	71	return -EINVAL;
	72
	73	lock_sock(sk);
	74
e6278d92 PB	75	ieee802154_addr_from_sa(&addr, &uaddr->addr);
e6278d92 PB	76	dev = ieee802154_get_dev(sock_net(sk), &addr);
9ec76716 SL	77	if (!dev) {
	78	err = -ENODEV;
	79	goto out;
	80	}
	81
929122cd	82	if (dev->type != ARPHRD_IEEE802154) {
9ec76716 SL	83	err = -ENODEV;
	84	goto out_put;
	85	}
	86
	87	sk->sk_bound_dev_if = dev->ifindex;
	88	sk_dst_reset(sk);
	89
	90	out_put:
	91	dev_put(dev);
	92	out:
	93	release_sock(sk);
	94
	95	return err;
	96	}
	97
	98	static int raw_connect(struct sock sk, struct sockaddr uaddr,
4710d806	99	int addr_len)
9ec76716 SL	100	{
	101	return -ENOTSUPP;
	102	}
	103
	104	static int raw_disconnect(struct sock *sk, int flags)
	105	{
	106	return 0;
	107	}
	108
4710d806 VB	109	static int raw_sendmsg(struct kiocb iocb, struct sock sk,
4710d806 VB	110	struct msghdr *msg, size_t size)
9ec76716 SL	111	{
9ec76716 SL	112	struct net_device *dev;
95c96174	113	unsigned int mtu;
9ec76716	114	struct sk_buff *skb;
ae641949	115	int hlen, tlen;
9ec76716 SL	116	int err;
	117
	118	if (msg->msg_flags & MSG_OOB) {
	119	pr_debug("msg->msg_flags = 0x%x\n", msg->msg_flags);
	120	return -EOPNOTSUPP;
	121	}
	122
	123	lock_sock(sk);
	124	if (!sk->sk_bound_dev_if)
	125	dev = dev_getfirstbyhwtype(sock_net(sk), ARPHRD_IEEE802154);
	126	else
	127	dev = dev_get_by_index(sock_net(sk), sk->sk_bound_dev_if);
	128	release_sock(sk);
	129
	130	if (!dev) {
	131	pr_debug("no dev\n");
	132	err = -ENXIO;
	133	goto out;
	134	}
	135
	136	mtu = dev->mtu;
	137	pr_debug("name = %s, mtu = %u\n", dev->name, mtu);
	138
	139	if (size > mtu) {
e5241c44	140	pr_debug("size = %Zu, mtu = %u\n", size, mtu);
9ec76716 SL	141	err = -EINVAL;
	142	goto out_dev;
	143	}
	144
ae641949 HX	145	hlen = LL_RESERVED_SPACE(dev);
	146	tlen = dev->needed_tailroom;
	147	skb = sock_alloc_send_skb(sk, hlen + tlen + size,
4710d806	148	msg->msg_flags & MSG_DONTWAIT, &err);
9ec76716 SL	149	if (!skb)
	150	goto out_dev;
	151
ae641949	152	skb_reserve(skb, hlen);
9ec76716 SL	153
	154	skb_reset_mac_header(skb);
	155	skb_reset_network_header(skb);
	156
	157	err = memcpy_fromiovec(skb_put(skb, size), msg->msg_iov, size);
	158	if (err < 0)
	159	goto out_skb;
	160
	161	skb->dev = dev;
	162	skb->sk = sk;
	163	skb->protocol = htons(ETH_P_IEEE802154);
	164
	165	dev_put(dev);
	166
	167	err = dev_queue_xmit(skb);
	168	if (err > 0)
	169	err = net_xmit_errno(err);
	170
	171	return err ?: size;
	172
	173	out_skb:
	174	kfree_skb(skb);
	175	out_dev:
	176	dev_put(dev);
	177	out:
	178	return err;
	179	}
	180
	181	static int raw_recvmsg(struct kiocb iocb, struct sock sk, struct msghdr *msg,
	182	size_t len, int noblock, int flags, int *addr_len)
	183	{
	184	size_t copied = 0;
	185	int err = -EOPNOTSUPP;
	186	struct sk_buff *skb;
	187
	188	skb = skb_recv_datagram(sk, flags, noblock, &err);
	189	if (!skb)
	190	goto out;
	191
	192	copied = skb->len;
	193	if (len < copied) {
	194	msg->msg_flags \|= MSG_TRUNC;
	195	copied = len;
	196	}
	197
	198	err = skb_copy_datagram_iovec(skb, 0, msg->msg_iov, copied);
	199	if (err)
	200	goto done;
	201
3b885787	202	sock_recv_ts_and_drops(msg, sk, skb);
9ec76716 SL	203
	204	if (flags & MSG_TRUNC)
	205	copied = skb->len;
	206	done:
	207	skb_free_datagram(sk, skb);
	208	out:
	209	if (err)
	210	return err;
	211	return copied;
	212	}
	213
	214	static int raw_rcv_skb(struct sock sk, struct sk_buff skb)
	215	{
8cfad496 PB	216	skb = skb_share_check(skb, GFP_ATOMIC);
	217	if (!skb)
	218	return NET_RX_DROP;
	219
9ec76716	220	if (sock_queue_rcv_skb(sk, skb) < 0) {
9ec76716 SL	221	kfree_skb(skb);
	222	return NET_RX_DROP;
	223	}
	224
	225	return NET_RX_SUCCESS;
	226	}
	227
	228
	229	void ieee802154_raw_deliver(struct net_device dev, struct sk_buff skb)
	230	{
	231	struct sock *sk;
9ec76716 SL	232
9ec76716 SL	233	read_lock(&raw_lock);
b67bfe0d	234	sk_for_each(sk, &raw_head) {
9ec76716 SL	235	bh_lock_sock(sk);
	236	if (!sk->sk_bound_dev_if \|\|
	237	sk->sk_bound_dev_if == dev->ifindex) {
9ec76716 SL	238	struct sk_buff *clone;
	239
	240	clone = skb_clone(skb, GFP_ATOMIC);
	241	if (clone)
	242	raw_rcv_skb(sk, clone);
	243	}
	244	bh_unlock_sock(sk);
	245	}
	246	read_unlock(&raw_lock);
	247	}
	248
a9dfac33	249	static int raw_getsockopt(struct sock *sk, int level, int optname,
4710d806	250	char __user optval, int __user optlen)
a9dfac33 DB	251	{
	252	return -EOPNOTSUPP;
	253	}
	254
	255	static int raw_setsockopt(struct sock *sk, int level, int optname,
4710d806	256	char __user *optval, unsigned int optlen)
a9dfac33 DB	257	{
	258	return -EOPNOTSUPP;
	259	}
	260
9ec76716 SL	261	struct proto ieee802154_raw_prot = {
	262	.name = "IEEE-802.15.4-RAW",
	263	.owner = THIS_MODULE,
	264	.obj_size = sizeof(struct sock),
	265	.close = raw_close,
	266	.bind = raw_bind,
	267	.sendmsg = raw_sendmsg,
	268	.recvmsg = raw_recvmsg,
	269	.hash = raw_hash,
	270	.unhash = raw_unhash,
	271	.connect = raw_connect,
	272	.disconnect = raw_disconnect,
a9dfac33 DB	273	.getsockopt = raw_getsockopt,
a9dfac33 DB	274	.setsockopt = raw_setsockopt,
9ec76716	275	};