net/ieee802154/nl-mac.c - LeafOS-Devices/android_kernel_samsung_gta4xl - Gitiles

 /*
  * Netlink inteface for IEEE 802.15.4 stack
  *
  * 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 <slapin@ossfans.org>
  * Dmitry Eremin-Solenikov <dbaryshkov@gmail.com>
  * Maxim Osipov <maxim.osipov@siemens.com>
  */

 #include <linux/gfp.h>
 #include <linux/kernel.h>
 #include <linux/if_arp.h>
 #include <linux/netdevice.h>
 #include <net/netlink.h>
 #include <net/genetlink.h>
 #include <net/sock.h>
 #include <linux/nl802154.h>
 #include <linux/export.h>
 #include <net/af_ieee802154.h>
 #include <net/nl802154.h>
 #include <net/ieee802154.h>
 #include <net/ieee802154_netdev.h>
 #include <net/wpan-phy.h>

 #include "ieee802154.h"

 int ieee802154_nl_assoc_indic(struct net_device *dev,
 		struct ieee802154_addr_sa *addr, u8 cap)
 {
 	struct sk_buff *msg;

 	pr_debug("%s\n", __func__);

 	if (addr->addr_type != IEEE802154_ADDR_LONG) {
 		pr_err("%s: received non-long source address!\n", __func__);
 		return -EINVAL;
 	}

 	msg = ieee802154_nl_create(0, IEEE802154_ASSOCIATE_INDIC);
 	if (!msg)
 		return -ENOBUFS;

 	if (nla_put_string(msg, IEEE802154_ATTR_DEV_NAME, dev->name) ||
 	    nla_put_u32(msg, IEEE802154_ATTR_DEV_INDEX, dev->ifindex) ||
 	    nla_put(msg, IEEE802154_ATTR_HW_ADDR, IEEE802154_ADDR_LEN,
 		    dev->dev_addr) ||
 	    nla_put(msg, IEEE802154_ATTR_SRC_HW_ADDR, IEEE802154_ADDR_LEN,
 		    addr->hwaddr) ||
 	    nla_put_u8(msg, IEEE802154_ATTR_CAPABILITY, cap))
 		goto nla_put_failure;

 	return ieee802154_nl_mcast(msg, IEEE802154_COORD_MCGRP);

 nla_put_failure:
 	nlmsg_free(msg);
 	return -ENOBUFS;
 }
 EXPORT_SYMBOL(ieee802154_nl_assoc_indic);

 int ieee802154_nl_assoc_confirm(struct net_device *dev, __le16 short_addr,
 		u8 status)
 {
 	struct sk_buff *msg;

 	pr_debug("%s\n", __func__);

 	msg = ieee802154_nl_create(0, IEEE802154_ASSOCIATE_CONF);
 	if (!msg)
 		return -ENOBUFS;

 	if (nla_put_string(msg, IEEE802154_ATTR_DEV_NAME, dev->name) ||
 	    nla_put_u32(msg, IEEE802154_ATTR_DEV_INDEX, dev->ifindex) ||
 	    nla_put(msg, IEEE802154_ATTR_HW_ADDR, IEEE802154_ADDR_LEN,
 		    dev->dev_addr) ||
 	    nla_put_u16(msg, IEEE802154_ATTR_SHORT_ADDR,
 			le16_to_cpu(short_addr)) ||
 	    nla_put_u8(msg, IEEE802154_ATTR_STATUS, status))
 		goto nla_put_failure;
 	return ieee802154_nl_mcast(msg, IEEE802154_COORD_MCGRP);

 nla_put_failure:
 	nlmsg_free(msg);
 	return -ENOBUFS;
 }
 EXPORT_SYMBOL(ieee802154_nl_assoc_confirm);

 int ieee802154_nl_disassoc_indic(struct net_device *dev,
 		struct ieee802154_addr_sa *addr, u8 reason)
 {
 	struct sk_buff *msg;

 	pr_debug("%s\n", __func__);

 	msg = ieee802154_nl_create(0, IEEE802154_DISASSOCIATE_INDIC);
 	if (!msg)
 		return -ENOBUFS;

 	if (nla_put_string(msg, IEEE802154_ATTR_DEV_NAME, dev->name) ||
 	    nla_put_u32(msg, IEEE802154_ATTR_DEV_INDEX, dev->ifindex) ||
 	    nla_put(msg, IEEE802154_ATTR_HW_ADDR, IEEE802154_ADDR_LEN,
 		    dev->dev_addr))
 		goto nla_put_failure;
 	if (addr->addr_type == IEEE802154_ADDR_LONG) {
 		if (nla_put(msg, IEEE802154_ATTR_SRC_HW_ADDR, IEEE802154_ADDR_LEN,
 			    addr->hwaddr))
 			goto nla_put_failure;
 	} else {
 		if (nla_put_u16(msg, IEEE802154_ATTR_SRC_SHORT_ADDR,
 				addr->short_addr))
 			goto nla_put_failure;
 	}
 	if (nla_put_u8(msg, IEEE802154_ATTR_REASON, reason))
 		goto nla_put_failure;
 	return ieee802154_nl_mcast(msg, IEEE802154_COORD_MCGRP);

 nla_put_failure:
 	nlmsg_free(msg);
 	return -ENOBUFS;
 }
 EXPORT_SYMBOL(ieee802154_nl_disassoc_indic);

 int ieee802154_nl_disassoc_confirm(struct net_device *dev, u8 status)
 {
 	struct sk_buff *msg;

 	pr_debug("%s\n", __func__);

 	msg = ieee802154_nl_create(0, IEEE802154_DISASSOCIATE_CONF);
 	if (!msg)
 		return -ENOBUFS;

 	if (nla_put_string(msg, IEEE802154_ATTR_DEV_NAME, dev->name) ||
 	    nla_put_u32(msg, IEEE802154_ATTR_DEV_INDEX, dev->ifindex) ||
 	    nla_put(msg, IEEE802154_ATTR_HW_ADDR, IEEE802154_ADDR_LEN,
 		    dev->dev_addr) ||
 	    nla_put_u8(msg, IEEE802154_ATTR_STATUS, status))
 		goto nla_put_failure;
 	return ieee802154_nl_mcast(msg, IEEE802154_COORD_MCGRP);

 nla_put_failure:
 	nlmsg_free(msg);
 	return -ENOBUFS;
 }
 EXPORT_SYMBOL(ieee802154_nl_disassoc_confirm);

 int ieee802154_nl_beacon_indic(struct net_device *dev, __le16 panid,
 			       __le16 coord_addr)
 {
 	struct sk_buff *msg;

 	pr_debug("%s\n", __func__);

 	msg = ieee802154_nl_create(0, IEEE802154_BEACON_NOTIFY_INDIC);
 	if (!msg)
 		return -ENOBUFS;

 	if (nla_put_string(msg, IEEE802154_ATTR_DEV_NAME, dev->name) ||
 	    nla_put_u32(msg, IEEE802154_ATTR_DEV_INDEX, dev->ifindex) ||
 	    nla_put(msg, IEEE802154_ATTR_HW_ADDR, IEEE802154_ADDR_LEN,
 		    dev->dev_addr) ||
 	    nla_put_u16(msg, IEEE802154_ATTR_COORD_SHORT_ADDR,
 			le16_to_cpu(coord_addr)) ||
 	    nla_put_u16(msg, IEEE802154_ATTR_COORD_PAN_ID,
 			le16_to_cpu(panid)))
 		goto nla_put_failure;
 	return ieee802154_nl_mcast(msg, IEEE802154_COORD_MCGRP);

 nla_put_failure:
 	nlmsg_free(msg);
 	return -ENOBUFS;
 }
 EXPORT_SYMBOL(ieee802154_nl_beacon_indic);

 int ieee802154_nl_scan_confirm(struct net_device *dev,
 		u8 status, u8 scan_type, u32 unscanned, u8 page,
 		u8 *edl/* , struct list_head *pan_desc_list */)
 {
 	struct sk_buff *msg;

 	pr_debug("%s\n", __func__);

 	msg = ieee802154_nl_create(0, IEEE802154_SCAN_CONF);
 	if (!msg)
 		return -ENOBUFS;

 	if (nla_put_string(msg, IEEE802154_ATTR_DEV_NAME, dev->name) ||
 	    nla_put_u32(msg, IEEE802154_ATTR_DEV_INDEX, dev->ifindex) ||
 	    nla_put(msg, IEEE802154_ATTR_HW_ADDR, IEEE802154_ADDR_LEN,
 		    dev->dev_addr) ||
 	    nla_put_u8(msg, IEEE802154_ATTR_STATUS, status) ||
 	    nla_put_u8(msg, IEEE802154_ATTR_SCAN_TYPE, scan_type) ||
 	    nla_put_u32(msg, IEEE802154_ATTR_CHANNELS, unscanned) ||
 	    nla_put_u8(msg, IEEE802154_ATTR_PAGE, page) ||
 	    (edl &&
 	     nla_put(msg, IEEE802154_ATTR_ED_LIST, 27, edl)))
 		goto nla_put_failure;
 	return ieee802154_nl_mcast(msg, IEEE802154_COORD_MCGRP);

 nla_put_failure:
 	nlmsg_free(msg);
 	return -ENOBUFS;
 }
 EXPORT_SYMBOL(ieee802154_nl_scan_confirm);

 int ieee802154_nl_start_confirm(struct net_device *dev, u8 status)
 {
 	struct sk_buff *msg;

 	pr_debug("%s\n", __func__);

 	msg = ieee802154_nl_create(0, IEEE802154_START_CONF);
 	if (!msg)
 		return -ENOBUFS;

 	if (nla_put_string(msg, IEEE802154_ATTR_DEV_NAME, dev->name) ||
 	    nla_put_u32(msg, IEEE802154_ATTR_DEV_INDEX, dev->ifindex) ||
 	    nla_put(msg, IEEE802154_ATTR_HW_ADDR, IEEE802154_ADDR_LEN,
 		    dev->dev_addr) ||
 	    nla_put_u8(msg, IEEE802154_ATTR_STATUS, status))
 		goto nla_put_failure;
 	return ieee802154_nl_mcast(msg, IEEE802154_COORD_MCGRP);

 nla_put_failure:
 	nlmsg_free(msg);
 	return -ENOBUFS;
 }
 EXPORT_SYMBOL(ieee802154_nl_start_confirm);

 static int ieee802154_nl_fill_iface(struct sk_buff *msg, u32 portid,
 	u32 seq, int flags, struct net_device *dev)
 {
 	void *hdr;
 	struct wpan_phy *phy;
 	u16 short_addr, pan_id;

 	pr_debug("%s\n", __func__);

 	hdr = genlmsg_put(msg, 0, seq, &nl802154_family, flags,
 		IEEE802154_LIST_IFACE);
 	if (!hdr)
 		goto out;

 	phy = ieee802154_mlme_ops(dev)->get_phy(dev);
 	BUG_ON(!phy);

 	short_addr = le16_to_cpu(ieee802154_mlme_ops(dev)->get_short_addr(dev));
 	pan_id = le16_to_cpu(ieee802154_mlme_ops(dev)->get_pan_id(dev));

 	if (nla_put_string(msg, IEEE802154_ATTR_DEV_NAME, dev->name) ||
 	    nla_put_string(msg, IEEE802154_ATTR_PHY_NAME, wpan_phy_name(phy)) ||
 	    nla_put_u32(msg, IEEE802154_ATTR_DEV_INDEX, dev->ifindex) ||
 	    nla_put(msg, IEEE802154_ATTR_HW_ADDR, IEEE802154_ADDR_LEN,
 		    dev->dev_addr) ||
 	    nla_put_u16(msg, IEEE802154_ATTR_SHORT_ADDR, short_addr) ||
 	    nla_put_u16(msg, IEEE802154_ATTR_PAN_ID, pan_id))
 		goto nla_put_failure;
 	wpan_phy_put(phy);
 	return genlmsg_end(msg, hdr);

 nla_put_failure:
 	wpan_phy_put(phy);
 	genlmsg_cancel(msg, hdr);
 out:
 	return -EMSGSIZE;
 }

 /* Requests from userspace */
 static struct net_device *ieee802154_nl_get_dev(struct genl_info *info)
 {
 	struct net_device *dev;

 	if (info->attrs[IEEE802154_ATTR_DEV_NAME]) {
 		char name[IFNAMSIZ + 1];
 		nla_strlcpy(name, info->attrs[IEEE802154_ATTR_DEV_NAME],
 				sizeof(name));
 		dev = dev_get_by_name(&init_net, name);
 	} else if (info->attrs[IEEE802154_ATTR_DEV_INDEX])
 		dev = dev_get_by_index(&init_net,
 			nla_get_u32(info->attrs[IEEE802154_ATTR_DEV_INDEX]));
 	else
 		return NULL;

 	if (!dev)
 		return NULL;

 	if (dev->type != ARPHRD_IEEE802154) {
 		dev_put(dev);
 		return NULL;
 	}

 	return dev;
 }

 int ieee802154_associate_req(struct sk_buff *skb, struct genl_info *info)
 {
 	struct net_device *dev;
 	struct ieee802154_addr_sa addr;
 	u8 page;
 	int ret = -EOPNOTSUPP;

 	if (!info->attrs[IEEE802154_ATTR_CHANNEL] ||
 	    !info->attrs[IEEE802154_ATTR_COORD_PAN_ID] ||
 	    (!info->attrs[IEEE802154_ATTR_COORD_HW_ADDR] &&
 		!info->attrs[IEEE802154_ATTR_COORD_SHORT_ADDR]) ||
 	    !info->attrs[IEEE802154_ATTR_CAPABILITY])
 		return -EINVAL;

 	dev = ieee802154_nl_get_dev(info);
 	if (!dev)
 		return -ENODEV;
 	if (!ieee802154_mlme_ops(dev)->assoc_req)
 		goto out;

 	if (info->attrs[IEEE802154_ATTR_COORD_HW_ADDR]) {
 		addr.addr_type = IEEE802154_ADDR_LONG;
 		nla_memcpy(addr.hwaddr,
 				info->attrs[IEEE802154_ATTR_COORD_HW_ADDR],
 				IEEE802154_ADDR_LEN);
 	} else {
 		addr.addr_type = IEEE802154_ADDR_SHORT;
 		addr.short_addr = nla_get_u16(
 				info->attrs[IEEE802154_ATTR_COORD_SHORT_ADDR]);
 	}
 	addr.pan_id = nla_get_u16(info->attrs[IEEE802154_ATTR_COORD_PAN_ID]);

 	if (info->attrs[IEEE802154_ATTR_PAGE])
 		page = nla_get_u8(info->attrs[IEEE802154_ATTR_PAGE]);
 	else
 		page = 0;

 	ret = ieee802154_mlme_ops(dev)->assoc_req(dev, &addr,
 			nla_get_u8(info->attrs[IEEE802154_ATTR_CHANNEL]),
 			page,
 			nla_get_u8(info->attrs[IEEE802154_ATTR_CAPABILITY]));

 out:
 	dev_put(dev);
 	return ret;
 }

 int ieee802154_associate_resp(struct sk_buff *skb, struct genl_info *info)
 {
 	struct net_device *dev;
 	struct ieee802154_addr_sa addr;
 	int ret = -EOPNOTSUPP;

 	if (!info->attrs[IEEE802154_ATTR_STATUS] ||
 	    !info->attrs[IEEE802154_ATTR_DEST_HW_ADDR] ||
 	    !info->attrs[IEEE802154_ATTR_DEST_SHORT_ADDR])
 		return -EINVAL;

 	dev = ieee802154_nl_get_dev(info);
 	if (!dev)
 		return -ENODEV;
 	if (!ieee802154_mlme_ops(dev)->assoc_resp)
 		goto out;

 	addr.addr_type = IEEE802154_ADDR_LONG;
 	nla_memcpy(addr.hwaddr, info->attrs[IEEE802154_ATTR_DEST_HW_ADDR],
 			IEEE802154_ADDR_LEN);
 	addr.pan_id = le16_to_cpu(ieee802154_mlme_ops(dev)->get_pan_id(dev));

 	ret = ieee802154_mlme_ops(dev)->assoc_resp(dev, &addr,
 		cpu_to_le16(nla_get_u16(info->attrs[IEEE802154_ATTR_DEST_SHORT_ADDR])),
 		nla_get_u8(info->attrs[IEEE802154_ATTR_STATUS]));

 out:
 	dev_put(dev);
 	return ret;
 }

 int ieee802154_disassociate_req(struct sk_buff *skb, struct genl_info *info)
 {
 	struct net_device *dev;
 	struct ieee802154_addr_sa addr;
 	int ret = -EOPNOTSUPP;

 	if ((!info->attrs[IEEE802154_ATTR_DEST_HW_ADDR] &&
 		!info->attrs[IEEE802154_ATTR_DEST_SHORT_ADDR]) ||
 	    !info->attrs[IEEE802154_ATTR_REASON])
 		return -EINVAL;

 	dev = ieee802154_nl_get_dev(info);
 	if (!dev)
 		return -ENODEV;
 	if (!ieee802154_mlme_ops(dev)->disassoc_req)
 		goto out;

 	if (info->attrs[IEEE802154_ATTR_DEST_HW_ADDR]) {
 		addr.addr_type = IEEE802154_ADDR_LONG;
 		nla_memcpy(addr.hwaddr,
 				info->attrs[IEEE802154_ATTR_DEST_HW_ADDR],
 				IEEE802154_ADDR_LEN);
 	} else {
 		addr.addr_type = IEEE802154_ADDR_SHORT;
 		addr.short_addr = nla_get_u16(
 				info->attrs[IEEE802154_ATTR_DEST_SHORT_ADDR]);
 	}
 	addr.pan_id = le16_to_cpu(ieee802154_mlme_ops(dev)->get_pan_id(dev));

 	ret = ieee802154_mlme_ops(dev)->disassoc_req(dev, &addr,
 			nla_get_u8(info->attrs[IEEE802154_ATTR_REASON]));

 out:
 	dev_put(dev);
 	return ret;
 }

 /*
  * PANid, channel, beacon_order = 15, superframe_order = 15,
  * PAN_coordinator, battery_life_extension = 0,
  * coord_realignment = 0, security_enable = 0
 */
 int ieee802154_start_req(struct sk_buff *skb, struct genl_info *info)
 {
 	struct net_device *dev;
 	struct ieee802154_addr_sa addr;

 	u8 channel, bcn_ord, sf_ord;
 	u8 page;
 	int pan_coord, blx, coord_realign;
 	int ret = -EOPNOTSUPP;

 	if (!info->attrs[IEEE802154_ATTR_COORD_PAN_ID] ||
 	    !info->attrs[IEEE802154_ATTR_COORD_SHORT_ADDR] ||
 	    !info->attrs[IEEE802154_ATTR_CHANNEL] ||
 	    !info->attrs[IEEE802154_ATTR_BCN_ORD] ||
 	    !info->attrs[IEEE802154_ATTR_SF_ORD] ||
 	    !info->attrs[IEEE802154_ATTR_PAN_COORD] ||
 	    !info->attrs[IEEE802154_ATTR_BAT_EXT] ||
 	    !info->attrs[IEEE802154_ATTR_COORD_REALIGN]
 	 )
 		return -EINVAL;

 	dev = ieee802154_nl_get_dev(info);
 	if (!dev)
 		return -ENODEV;
 	if (!ieee802154_mlme_ops(dev)->start_req)
 		goto out;

 	addr.addr_type = IEEE802154_ADDR_SHORT;
 	addr.short_addr = nla_get_u16(
 			info->attrs[IEEE802154_ATTR_COORD_SHORT_ADDR]);
 	addr.pan_id = nla_get_u16(info->attrs[IEEE802154_ATTR_COORD_PAN_ID]);

 	channel = nla_get_u8(info->attrs[IEEE802154_ATTR_CHANNEL]);
 	bcn_ord = nla_get_u8(info->attrs[IEEE802154_ATTR_BCN_ORD]);
 	sf_ord = nla_get_u8(info->attrs[IEEE802154_ATTR_SF_ORD]);
 	pan_coord = nla_get_u8(info->attrs[IEEE802154_ATTR_PAN_COORD]);
 	blx = nla_get_u8(info->attrs[IEEE802154_ATTR_BAT_EXT]);
 	coord_realign = nla_get_u8(info->attrs[IEEE802154_ATTR_COORD_REALIGN]);

 	if (info->attrs[IEEE802154_ATTR_PAGE])
 		page = nla_get_u8(info->attrs[IEEE802154_ATTR_PAGE]);
 	else
 		page = 0;


 	if (addr.short_addr == IEEE802154_ADDR_BROADCAST) {
 		ieee802154_nl_start_confirm(dev, IEEE802154_NO_SHORT_ADDRESS);
 		dev_put(dev);
 		return -EINVAL;
 	}

 	ret = ieee802154_mlme_ops(dev)->start_req(dev, &addr, channel, page,
 		bcn_ord, sf_ord, pan_coord, blx, coord_realign);

 out:
 	dev_put(dev);
 	return ret;
 }

 int ieee802154_scan_req(struct sk_buff *skb, struct genl_info *info)
 {
 	struct net_device *dev;
 	int ret = -EOPNOTSUPP;
 	u8 type;
 	u32 channels;
 	u8 duration;
 	u8 page;

 	if (!info->attrs[IEEE802154_ATTR_SCAN_TYPE] ||
 	    !info->attrs[IEEE802154_ATTR_CHANNELS] ||
 	    !info->attrs[IEEE802154_ATTR_DURATION])
 		return -EINVAL;

 	dev = ieee802154_nl_get_dev(info);
 	if (!dev)
 		return -ENODEV;
 	if (!ieee802154_mlme_ops(dev)->scan_req)
 		goto out;

 	type = nla_get_u8(info->attrs[IEEE802154_ATTR_SCAN_TYPE]);
 	channels = nla_get_u32(info->attrs[IEEE802154_ATTR_CHANNELS]);
 	duration = nla_get_u8(info->attrs[IEEE802154_ATTR_DURATION]);

 	if (info->attrs[IEEE802154_ATTR_PAGE])
 		page = nla_get_u8(info->attrs[IEEE802154_ATTR_PAGE]);
 	else
 		page = 0;


 	ret = ieee802154_mlme_ops(dev)->scan_req(dev, type, channels, page,
 			duration);

 out:
 	dev_put(dev);
 	return ret;
 }

 int ieee802154_list_iface(struct sk_buff *skb, struct genl_info *info)
 {
 	/* Request for interface name, index, type, IEEE address,
 	   PAN Id, short address */
 	struct sk_buff *msg;
 	struct net_device *dev = NULL;
 	int rc = -ENOBUFS;

 	pr_debug("%s\n", __func__);

 	dev = ieee802154_nl_get_dev(info);
 	if (!dev)
 		return -ENODEV;

 	msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
 	if (!msg)
 		goto out_dev;

 	rc = ieee802154_nl_fill_iface(msg, info->snd_portid, info->snd_seq,
 			0, dev);
 	if (rc < 0)
 		goto out_free;

 	dev_put(dev);

 	return genlmsg_reply(msg, info);
 out_free:
 	nlmsg_free(msg);
 out_dev:
 	dev_put(dev);
 	return rc;

 }

 int ieee802154_dump_iface(struct sk_buff *skb, struct netlink_callback *cb)
 {
 	struct net *net = sock_net(skb->sk);
 	struct net_device *dev;
 	int idx;
 	int s_idx = cb->args[0];

 	pr_debug("%s\n", __func__);

 	idx = 0;
 	for_each_netdev(net, dev) {
 		if (idx < s_idx || (dev->type != ARPHRD_IEEE802154))
 			goto cont;

 		if (ieee802154_nl_fill_iface(skb, NETLINK_CB(cb->skb).portid,
 			cb->nlh->nlmsg_seq, NLM_F_MULTI, dev) < 0)
 			break;
 cont:
 		idx++;
 	}
 	cb->args[0] = idx;

 	return skb->len;
 }
	/*
	* Netlink inteface for IEEE 802.15.4 stack
	*
	* 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 <slapin@ossfans.org>
	* Dmitry Eremin-Solenikov <dbaryshkov@gmail.com>
	* Maxim Osipov <maxim.osipov@siemens.com>
	*/

	#include <linux/gfp.h>
	#include <linux/kernel.h>
	#include <linux/if_arp.h>
	#include <linux/netdevice.h>
	#include <net/netlink.h>
	#include <net/genetlink.h>
	#include <net/sock.h>
	#include <linux/nl802154.h>
	#include <linux/export.h>
	#include <net/af_ieee802154.h>
	#include <net/nl802154.h>
	#include <net/ieee802154.h>
	#include <net/ieee802154_netdev.h>
	#include <net/wpan-phy.h>

	#include "ieee802154.h"

	int ieee802154_nl_assoc_indic(struct net_device *dev,
	struct ieee802154_addr_sa *addr, u8 cap)
	{
	struct sk_buff *msg;

	pr_debug("%s\n", __func__);

	if (addr->addr_type != IEEE802154_ADDR_LONG) {
	pr_err("%s: received non-long source address!\n", __func__);
	return -EINVAL;
	}

	msg = ieee802154_nl_create(0, IEEE802154_ASSOCIATE_INDIC);
	if (!msg)
	return -ENOBUFS;

	if (nla_put_string(msg, IEEE802154_ATTR_DEV_NAME, dev->name) \|\|
	nla_put_u32(msg, IEEE802154_ATTR_DEV_INDEX, dev->ifindex) \|\|
	nla_put(msg, IEEE802154_ATTR_HW_ADDR, IEEE802154_ADDR_LEN,
	dev->dev_addr) \|\|
	nla_put(msg, IEEE802154_ATTR_SRC_HW_ADDR, IEEE802154_ADDR_LEN,
	addr->hwaddr) \|\|
	nla_put_u8(msg, IEEE802154_ATTR_CAPABILITY, cap))
	goto nla_put_failure;

	return ieee802154_nl_mcast(msg, IEEE802154_COORD_MCGRP);

	nla_put_failure:
	nlmsg_free(msg);
	return -ENOBUFS;
	}
	EXPORT_SYMBOL(ieee802154_nl_assoc_indic);

	int ieee802154_nl_assoc_confirm(struct net_device *dev, __le16 short_addr,
	u8 status)
	{
	struct sk_buff *msg;

	pr_debug("%s\n", __func__);

	msg = ieee802154_nl_create(0, IEEE802154_ASSOCIATE_CONF);
	if (!msg)
	return -ENOBUFS;

	if (nla_put_string(msg, IEEE802154_ATTR_DEV_NAME, dev->name) \|\|
	nla_put_u32(msg, IEEE802154_ATTR_DEV_INDEX, dev->ifindex) \|\|
	nla_put(msg, IEEE802154_ATTR_HW_ADDR, IEEE802154_ADDR_LEN,
	dev->dev_addr) \|\|
	nla_put_u16(msg, IEEE802154_ATTR_SHORT_ADDR,
	le16_to_cpu(short_addr)) \|\|
	nla_put_u8(msg, IEEE802154_ATTR_STATUS, status))
	goto nla_put_failure;
	return ieee802154_nl_mcast(msg, IEEE802154_COORD_MCGRP);

	nla_put_failure:
	nlmsg_free(msg);
	return -ENOBUFS;
	}
	EXPORT_SYMBOL(ieee802154_nl_assoc_confirm);

	int ieee802154_nl_disassoc_indic(struct net_device *dev,
	struct ieee802154_addr_sa *addr, u8 reason)
	{
	struct sk_buff *msg;

	pr_debug("%s\n", __func__);

	msg = ieee802154_nl_create(0, IEEE802154_DISASSOCIATE_INDIC);
	if (!msg)
	return -ENOBUFS;

	if (nla_put_string(msg, IEEE802154_ATTR_DEV_NAME, dev->name) \|\|
	nla_put_u32(msg, IEEE802154_ATTR_DEV_INDEX, dev->ifindex) \|\|
	nla_put(msg, IEEE802154_ATTR_HW_ADDR, IEEE802154_ADDR_LEN,
	dev->dev_addr))
	goto nla_put_failure;
	if (addr->addr_type == IEEE802154_ADDR_LONG) {
	if (nla_put(msg, IEEE802154_ATTR_SRC_HW_ADDR, IEEE802154_ADDR_LEN,
	addr->hwaddr))
	goto nla_put_failure;
	} else {
	if (nla_put_u16(msg, IEEE802154_ATTR_SRC_SHORT_ADDR,
	addr->short_addr))
	goto nla_put_failure;
	}
	if (nla_put_u8(msg, IEEE802154_ATTR_REASON, reason))
	goto nla_put_failure;
	return ieee802154_nl_mcast(msg, IEEE802154_COORD_MCGRP);

	nla_put_failure:
	nlmsg_free(msg);
	return -ENOBUFS;
	}
	EXPORT_SYMBOL(ieee802154_nl_disassoc_indic);

	int ieee802154_nl_disassoc_confirm(struct net_device *dev, u8 status)
	{
	struct sk_buff *msg;

	pr_debug("%s\n", __func__);

	msg = ieee802154_nl_create(0, IEEE802154_DISASSOCIATE_CONF);
	if (!msg)
	return -ENOBUFS;

	if (nla_put_string(msg, IEEE802154_ATTR_DEV_NAME, dev->name) \|\|
	nla_put_u32(msg, IEEE802154_ATTR_DEV_INDEX, dev->ifindex) \|\|
	nla_put(msg, IEEE802154_ATTR_HW_ADDR, IEEE802154_ADDR_LEN,
	dev->dev_addr) \|\|
	nla_put_u8(msg, IEEE802154_ATTR_STATUS, status))
	goto nla_put_failure;
	return ieee802154_nl_mcast(msg, IEEE802154_COORD_MCGRP);

	nla_put_failure:
	nlmsg_free(msg);
	return -ENOBUFS;
	}
	EXPORT_SYMBOL(ieee802154_nl_disassoc_confirm);

	int ieee802154_nl_beacon_indic(struct net_device *dev, __le16 panid,
	__le16 coord_addr)
	{
	struct sk_buff *msg;

	pr_debug("%s\n", __func__);

	msg = ieee802154_nl_create(0, IEEE802154_BEACON_NOTIFY_INDIC);
	if (!msg)
	return -ENOBUFS;

	if (nla_put_string(msg, IEEE802154_ATTR_DEV_NAME, dev->name) \|\|
	nla_put_u32(msg, IEEE802154_ATTR_DEV_INDEX, dev->ifindex) \|\|
	nla_put(msg, IEEE802154_ATTR_HW_ADDR, IEEE802154_ADDR_LEN,
	dev->dev_addr) \|\|
	nla_put_u16(msg, IEEE802154_ATTR_COORD_SHORT_ADDR,
	le16_to_cpu(coord_addr)) \|\|
	nla_put_u16(msg, IEEE802154_ATTR_COORD_PAN_ID,
	le16_to_cpu(panid)))
	goto nla_put_failure;
	return ieee802154_nl_mcast(msg, IEEE802154_COORD_MCGRP);

	nla_put_failure:
	nlmsg_free(msg);
	return -ENOBUFS;
	}
	EXPORT_SYMBOL(ieee802154_nl_beacon_indic);

	int ieee802154_nl_scan_confirm(struct net_device *dev,
	u8 status, u8 scan_type, u32 unscanned, u8 page,
	u8 edl/ , struct list_head pan_desc_list /)
	{
	struct sk_buff *msg;

	pr_debug("%s\n", __func__);

	msg = ieee802154_nl_create(0, IEEE802154_SCAN_CONF);
	if (!msg)
	return -ENOBUFS;

	if (nla_put_string(msg, IEEE802154_ATTR_DEV_NAME, dev->name) \|\|
	nla_put_u32(msg, IEEE802154_ATTR_DEV_INDEX, dev->ifindex) \|\|
	nla_put(msg, IEEE802154_ATTR_HW_ADDR, IEEE802154_ADDR_LEN,
	dev->dev_addr) \|\|
	nla_put_u8(msg, IEEE802154_ATTR_STATUS, status) \|\|
	nla_put_u8(msg, IEEE802154_ATTR_SCAN_TYPE, scan_type) \|\|
	nla_put_u32(msg, IEEE802154_ATTR_CHANNELS, unscanned) \|\|
	nla_put_u8(msg, IEEE802154_ATTR_PAGE, page) \|\|
	(edl &&
	nla_put(msg, IEEE802154_ATTR_ED_LIST, 27, edl)))
	goto nla_put_failure;
	return ieee802154_nl_mcast(msg, IEEE802154_COORD_MCGRP);

	nla_put_failure:
	nlmsg_free(msg);
	return -ENOBUFS;
	}
	EXPORT_SYMBOL(ieee802154_nl_scan_confirm);

	int ieee802154_nl_start_confirm(struct net_device *dev, u8 status)
	{
	struct sk_buff *msg;

	pr_debug("%s\n", __func__);

	msg = ieee802154_nl_create(0, IEEE802154_START_CONF);
	if (!msg)
	return -ENOBUFS;

	if (nla_put_string(msg, IEEE802154_ATTR_DEV_NAME, dev->name) \|\|
	nla_put_u32(msg, IEEE802154_ATTR_DEV_INDEX, dev->ifindex) \|\|
	nla_put(msg, IEEE802154_ATTR_HW_ADDR, IEEE802154_ADDR_LEN,
	dev->dev_addr) \|\|
	nla_put_u8(msg, IEEE802154_ATTR_STATUS, status))
	goto nla_put_failure;
	return ieee802154_nl_mcast(msg, IEEE802154_COORD_MCGRP);

	nla_put_failure:
	nlmsg_free(msg);
	return -ENOBUFS;
	}
	EXPORT_SYMBOL(ieee802154_nl_start_confirm);

	static int ieee802154_nl_fill_iface(struct sk_buff *msg, u32 portid,
	u32 seq, int flags, struct net_device *dev)
	{
	void *hdr;
	struct wpan_phy *phy;
	u16 short_addr, pan_id;

	pr_debug("%s\n", __func__);

	hdr = genlmsg_put(msg, 0, seq, &nl802154_family, flags,
	IEEE802154_LIST_IFACE);
	if (!hdr)
	goto out;

	phy = ieee802154_mlme_ops(dev)->get_phy(dev);
	BUG_ON(!phy);

	short_addr = le16_to_cpu(ieee802154_mlme_ops(dev)->get_short_addr(dev));
	pan_id = le16_to_cpu(ieee802154_mlme_ops(dev)->get_pan_id(dev));

	if (nla_put_string(msg, IEEE802154_ATTR_DEV_NAME, dev->name) \|\|
	nla_put_string(msg, IEEE802154_ATTR_PHY_NAME, wpan_phy_name(phy)) \|\|
	nla_put_u32(msg, IEEE802154_ATTR_DEV_INDEX, dev->ifindex) \|\|
	nla_put(msg, IEEE802154_ATTR_HW_ADDR, IEEE802154_ADDR_LEN,
	dev->dev_addr) \|\|
	nla_put_u16(msg, IEEE802154_ATTR_SHORT_ADDR, short_addr) \|\|
	nla_put_u16(msg, IEEE802154_ATTR_PAN_ID, pan_id))
	goto nla_put_failure;
	wpan_phy_put(phy);
	return genlmsg_end(msg, hdr);

	nla_put_failure:
	wpan_phy_put(phy);
	genlmsg_cancel(msg, hdr);
	out:
	return -EMSGSIZE;
	}

	/* Requests from userspace */
	static struct net_device ieee802154_nl_get_dev(struct genl_info info)
	{
	struct net_device *dev;

	if (info->attrs[IEEE802154_ATTR_DEV_NAME]) {
	char name[IFNAMSIZ + 1];
	nla_strlcpy(name, info->attrs[IEEE802154_ATTR_DEV_NAME],
	sizeof(name));
	dev = dev_get_by_name(&init_net, name);
	} else if (info->attrs[IEEE802154_ATTR_DEV_INDEX])
	dev = dev_get_by_index(&init_net,
	nla_get_u32(info->attrs[IEEE802154_ATTR_DEV_INDEX]));
	else
	return NULL;

	if (!dev)
	return NULL;

	if (dev->type != ARPHRD_IEEE802154) {
	dev_put(dev);
	return NULL;
	}

	return dev;
	}

	int ieee802154_associate_req(struct sk_buff skb, struct genl_info info)
	{
	struct net_device *dev;
	struct ieee802154_addr_sa addr;
	u8 page;
	int ret = -EOPNOTSUPP;

	if (!info->attrs[IEEE802154_ATTR_CHANNEL] \|\|
	!info->attrs[IEEE802154_ATTR_COORD_PAN_ID] \|\|
	(!info->attrs[IEEE802154_ATTR_COORD_HW_ADDR] &&
	!info->attrs[IEEE802154_ATTR_COORD_SHORT_ADDR]) \|\|
	!info->attrs[IEEE802154_ATTR_CAPABILITY])
	return -EINVAL;

	dev = ieee802154_nl_get_dev(info);
	if (!dev)
	return -ENODEV;
	if (!ieee802154_mlme_ops(dev)->assoc_req)
	goto out;

	if (info->attrs[IEEE802154_ATTR_COORD_HW_ADDR]) {
	addr.addr_type = IEEE802154_ADDR_LONG;
	nla_memcpy(addr.hwaddr,
	info->attrs[IEEE802154_ATTR_COORD_HW_ADDR],
	IEEE802154_ADDR_LEN);
	} else {
	addr.addr_type = IEEE802154_ADDR_SHORT;
	addr.short_addr = nla_get_u16(
	info->attrs[IEEE802154_ATTR_COORD_SHORT_ADDR]);
	}
	addr.pan_id = nla_get_u16(info->attrs[IEEE802154_ATTR_COORD_PAN_ID]);

	if (info->attrs[IEEE802154_ATTR_PAGE])
	page = nla_get_u8(info->attrs[IEEE802154_ATTR_PAGE]);
	else
	page = 0;

	ret = ieee802154_mlme_ops(dev)->assoc_req(dev, &addr,
	nla_get_u8(info->attrs[IEEE802154_ATTR_CHANNEL]),
	page,
	nla_get_u8(info->attrs[IEEE802154_ATTR_CAPABILITY]));

	out:
	dev_put(dev);
	return ret;
	}

	int ieee802154_associate_resp(struct sk_buff skb, struct genl_info info)
	{
	struct net_device *dev;
	struct ieee802154_addr_sa addr;
	int ret = -EOPNOTSUPP;

	if (!info->attrs[IEEE802154_ATTR_STATUS] \|\|
	!info->attrs[IEEE802154_ATTR_DEST_HW_ADDR] \|\|
	!info->attrs[IEEE802154_ATTR_DEST_SHORT_ADDR])
	return -EINVAL;

	dev = ieee802154_nl_get_dev(info);
	if (!dev)
	return -ENODEV;
	if (!ieee802154_mlme_ops(dev)->assoc_resp)
	goto out;

	addr.addr_type = IEEE802154_ADDR_LONG;
	nla_memcpy(addr.hwaddr, info->attrs[IEEE802154_ATTR_DEST_HW_ADDR],
	IEEE802154_ADDR_LEN);
	addr.pan_id = le16_to_cpu(ieee802154_mlme_ops(dev)->get_pan_id(dev));

	ret = ieee802154_mlme_ops(dev)->assoc_resp(dev, &addr,
	cpu_to_le16(nla_get_u16(info->attrs[IEEE802154_ATTR_DEST_SHORT_ADDR])),
	nla_get_u8(info->attrs[IEEE802154_ATTR_STATUS]));

	out:
	dev_put(dev);
	return ret;
	}

	int ieee802154_disassociate_req(struct sk_buff skb, struct genl_info info)
	{
	struct net_device *dev;
	struct ieee802154_addr_sa addr;
	int ret = -EOPNOTSUPP;

	if ((!info->attrs[IEEE802154_ATTR_DEST_HW_ADDR] &&
	!info->attrs[IEEE802154_ATTR_DEST_SHORT_ADDR]) \|\|
	!info->attrs[IEEE802154_ATTR_REASON])
	return -EINVAL;

	dev = ieee802154_nl_get_dev(info);
	if (!dev)
	return -ENODEV;
	if (!ieee802154_mlme_ops(dev)->disassoc_req)
	goto out;

	if (info->attrs[IEEE802154_ATTR_DEST_HW_ADDR]) {
	addr.addr_type = IEEE802154_ADDR_LONG;
	nla_memcpy(addr.hwaddr,
	info->attrs[IEEE802154_ATTR_DEST_HW_ADDR],
	IEEE802154_ADDR_LEN);
	} else {
	addr.addr_type = IEEE802154_ADDR_SHORT;
	addr.short_addr = nla_get_u16(
	info->attrs[IEEE802154_ATTR_DEST_SHORT_ADDR]);
	}
	addr.pan_id = le16_to_cpu(ieee802154_mlme_ops(dev)->get_pan_id(dev));

	ret = ieee802154_mlme_ops(dev)->disassoc_req(dev, &addr,
	nla_get_u8(info->attrs[IEEE802154_ATTR_REASON]));

	out:
	dev_put(dev);
	return ret;
	}

	/*
	* PANid, channel, beacon_order = 15, superframe_order = 15,
	* PAN_coordinator, battery_life_extension = 0,
	* coord_realignment = 0, security_enable = 0
	*/
	int ieee802154_start_req(struct sk_buff skb, struct genl_info info)
	{
	struct net_device *dev;
	struct ieee802154_addr_sa addr;

	u8 channel, bcn_ord, sf_ord;
	u8 page;
	int pan_coord, blx, coord_realign;
	int ret = -EOPNOTSUPP;

	if (!info->attrs[IEEE802154_ATTR_COORD_PAN_ID] \|\|
	!info->attrs[IEEE802154_ATTR_COORD_SHORT_ADDR] \|\|
	!info->attrs[IEEE802154_ATTR_CHANNEL] \|\|
	!info->attrs[IEEE802154_ATTR_BCN_ORD] \|\|
	!info->attrs[IEEE802154_ATTR_SF_ORD] \|\|
	!info->attrs[IEEE802154_ATTR_PAN_COORD] \|\|
	!info->attrs[IEEE802154_ATTR_BAT_EXT] \|\|
	!info->attrs[IEEE802154_ATTR_COORD_REALIGN]
	)
	return -EINVAL;

	dev = ieee802154_nl_get_dev(info);
	if (!dev)
	return -ENODEV;
	if (!ieee802154_mlme_ops(dev)->start_req)
	goto out;

	addr.addr_type = IEEE802154_ADDR_SHORT;
	addr.short_addr = nla_get_u16(
	info->attrs[IEEE802154_ATTR_COORD_SHORT_ADDR]);
	addr.pan_id = nla_get_u16(info->attrs[IEEE802154_ATTR_COORD_PAN_ID]);

	channel = nla_get_u8(info->attrs[IEEE802154_ATTR_CHANNEL]);
	bcn_ord = nla_get_u8(info->attrs[IEEE802154_ATTR_BCN_ORD]);
	sf_ord = nla_get_u8(info->attrs[IEEE802154_ATTR_SF_ORD]);
	pan_coord = nla_get_u8(info->attrs[IEEE802154_ATTR_PAN_COORD]);
	blx = nla_get_u8(info->attrs[IEEE802154_ATTR_BAT_EXT]);
	coord_realign = nla_get_u8(info->attrs[IEEE802154_ATTR_COORD_REALIGN]);

	if (info->attrs[IEEE802154_ATTR_PAGE])
	page = nla_get_u8(info->attrs[IEEE802154_ATTR_PAGE]);
	else
	page = 0;


	if (addr.short_addr == IEEE802154_ADDR_BROADCAST) {
	ieee802154_nl_start_confirm(dev, IEEE802154_NO_SHORT_ADDRESS);
	dev_put(dev);
	return -EINVAL;
	}

	ret = ieee802154_mlme_ops(dev)->start_req(dev, &addr, channel, page,
	bcn_ord, sf_ord, pan_coord, blx, coord_realign);

	out:
	dev_put(dev);
	return ret;
	}

	int ieee802154_scan_req(struct sk_buff skb, struct genl_info info)
	{
	struct net_device *dev;
	int ret = -EOPNOTSUPP;
	u8 type;
	u32 channels;
	u8 duration;
	u8 page;

	if (!info->attrs[IEEE802154_ATTR_SCAN_TYPE] \|\|
	!info->attrs[IEEE802154_ATTR_CHANNELS] \|\|
	!info->attrs[IEEE802154_ATTR_DURATION])
	return -EINVAL;

	dev = ieee802154_nl_get_dev(info);
	if (!dev)
	return -ENODEV;
	if (!ieee802154_mlme_ops(dev)->scan_req)
	goto out;

	type = nla_get_u8(info->attrs[IEEE802154_ATTR_SCAN_TYPE]);
	channels = nla_get_u32(info->attrs[IEEE802154_ATTR_CHANNELS]);
	duration = nla_get_u8(info->attrs[IEEE802154_ATTR_DURATION]);

	if (info->attrs[IEEE802154_ATTR_PAGE])
	page = nla_get_u8(info->attrs[IEEE802154_ATTR_PAGE]);
	else
	page = 0;


	ret = ieee802154_mlme_ops(dev)->scan_req(dev, type, channels, page,
	duration);

	out:
	dev_put(dev);
	return ret;
	}

	int ieee802154_list_iface(struct sk_buff skb, struct genl_info info)
	{
	/* Request for interface name, index, type, IEEE address,
	PAN Id, short address */
	struct sk_buff *msg;
	struct net_device *dev = NULL;
	int rc = -ENOBUFS;

	pr_debug("%s\n", __func__);

	dev = ieee802154_nl_get_dev(info);
	if (!dev)
	return -ENODEV;

	msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
	if (!msg)
	goto out_dev;

	rc = ieee802154_nl_fill_iface(msg, info->snd_portid, info->snd_seq,
	0, dev);
	if (rc < 0)
	goto out_free;

	dev_put(dev);

	return genlmsg_reply(msg, info);
	out_free:
	nlmsg_free(msg);
	out_dev:
	dev_put(dev);
	return rc;

	}

	int ieee802154_dump_iface(struct sk_buff skb, struct netlink_callback cb)
	{
	struct net *net = sock_net(skb->sk);
	struct net_device *dev;
	int idx;
	int s_idx = cb->args[0];

	pr_debug("%s\n", __func__);

	idx = 0;
	for_each_netdev(net, dev) {
	if (idx < s_idx \|\| (dev->type != ARPHRD_IEEE802154))
	goto cont;

	if (ieee802154_nl_fill_iface(skb, NETLINK_CB(cb->skb).portid,
	cb->nlh->nlmsg_seq, NLM_F_MULTI, dev) < 0)
	break;
	cont:
	idx++;
	}
	cb->args[0] = idx;

	return skb->len;
	}