[thirdparty/iw.git] / mpath.c

#include <net/if.h>
#include <errno.h>
#include <string.h>

#include <netlink/genl/genl.h>
#include <netlink/genl/family.h>
#include <netlink/genl/ctrl.h>
#include <netlink/msg.h>
#include <netlink/attr.h>

#include "nl80211.h"
#include "iw.h"

enum plink_state {
	LISTEN,
	OPN_SNT,
	OPN_RCVD,
	CNF_RCVD,
	ESTAB,
	HOLDING,
	BLOCKED
};

enum plink_actions {
	PLINK_ACTION_UNDEFINED,
	PLINK_ACTION_OPEN,
	PLINK_ACTION_BLOCK,
};


static int print_mpath_handler(struct nl_msg *msg, void *arg)
{
	struct nlattr *tb[NL80211_ATTR_MAX + 1];
	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
	struct nlattr *pinfo[NL80211_MPATH_INFO_MAX + 1];
	char dst[20], next_hop[20], dev[20];
	static struct nla_policy mpath_policy[NL80211_MPATH_INFO_MAX + 1] = {
		[NL80211_MPATH_INFO_FRAME_QLEN] = { .type = NLA_U32 },
		[NL80211_MPATH_INFO_DSN] = { .type = NLA_U32 },
		[NL80211_MPATH_INFO_METRIC] = { .type = NLA_U32 },
		[NL80211_MPATH_INFO_EXPTIME] = { .type = NLA_U32 },
		[NL80211_MPATH_INFO_DISCOVERY_TIMEOUT] = { .type = NLA_U32 },
		[NL80211_MPATH_INFO_DISCOVERY_RETRIES] = { .type = NLA_U8 },
		[NL80211_MPATH_INFO_FLAGS] = { .type = NLA_U8 },
	};

	nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
		  genlmsg_attrlen(gnlh, 0), NULL);

	/*
	 * TODO: validate the interface and mac address!
	 * Otherwise, there's a race condition as soon as
	 * the kernel starts sending mpath notifications.
	 */

	if (!tb[NL80211_ATTR_MPATH_INFO]) {
		fprintf(stderr, "mpath info missing!");
		return NL_SKIP;
	}
	if (nla_parse_nested(pinfo, NL80211_MPATH_INFO_MAX,
			     tb[NL80211_ATTR_MPATH_INFO],
			     mpath_policy)) {
		fprintf(stderr, "failed to parse nested attributes!");
		return NL_SKIP;
	}

	mac_addr_n2a(dst, nla_data(tb[NL80211_ATTR_MAC]));
	mac_addr_n2a(next_hop, nla_data(tb[NL80211_ATTR_MPATH_NEXT_HOP]));
	if_indextoname(nla_get_u32(tb[NL80211_ATTR_IFINDEX]), dev);
	printf("%s %s %s", dst, next_hop, dev);
	if (pinfo[NL80211_MPATH_INFO_DSN])
		printf("\t%u",
			nla_get_u32(pinfo[NL80211_MPATH_INFO_DSN]));
	if (pinfo[NL80211_MPATH_INFO_METRIC])
		printf("\t%u",
			nla_get_u32(pinfo[NL80211_MPATH_INFO_METRIC]));
	if (pinfo[NL80211_MPATH_INFO_FRAME_QLEN])
		printf("\t%u",
			nla_get_u32(pinfo[NL80211_MPATH_INFO_FRAME_QLEN]));
	if (pinfo[NL80211_MPATH_INFO_EXPTIME])
		printf("\t%u",
			nla_get_u32(pinfo[NL80211_MPATH_INFO_EXPTIME]));
	if (pinfo[NL80211_MPATH_INFO_DISCOVERY_TIMEOUT])
		printf("\t%u",
		nla_get_u32(pinfo[NL80211_MPATH_INFO_DISCOVERY_TIMEOUT]));
	if (pinfo[NL80211_MPATH_INFO_DISCOVERY_RETRIES])
		printf("\t%u",
		nla_get_u8(pinfo[NL80211_MPATH_INFO_DISCOVERY_RETRIES]));
	if (pinfo[NL80211_MPATH_INFO_FLAGS])
		printf("\t0x%x",
			nla_get_u8(pinfo[NL80211_MPATH_INFO_FLAGS]));

	printf("\n");
	return NL_SKIP;
}

static int handle_mpath_get(struct nl80211_state *state,
			    struct nl_cb *cb,
			    struct nl_msg *msg,
			    int argc, char **argv)
{
	unsigned char dst[ETH_ALEN];

	if (argc < 1)
		return 1;

	if (mac_addr_a2n(dst, argv[0])) {
		fprintf(stderr, "invalid mac address\n");
		return 2;
	}
	argc--;
	argv++;

	if (argc)
		return 1;

	NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, dst);

	nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, print_mpath_handler, NULL);

	return 0;
 nla_put_failure:
	return -ENOBUFS;
}
COMMAND(mpath, get, "<MAC address>",
	NL80211_CMD_GET_MPATH, 0, CIB_NETDEV, handle_mpath_get);
COMMAND(mpath, del, "<MAC address>",
	NL80211_CMD_DEL_MPATH, 0, CIB_NETDEV, handle_mpath_get);

static int handle_mpath_set(struct nl80211_state *state,
			    struct nl_cb *cb,
			    struct nl_msg *msg,
			    int argc, char **argv)
{
	unsigned char dst[ETH_ALEN];
	unsigned char next_hop[ETH_ALEN];

	if (argc < 3)
		return 1;

	if (mac_addr_a2n(dst, argv[0])) {
		fprintf(stderr, "invalid destination mac address\n");
		return 2;
	}
	argc--;
	argv++;

	if (strcmp("next_hop", argv[0]) != 0)
		return 1;
	argc--;
	argv++;

	if (mac_addr_a2n(next_hop, argv[0])) {
		fprintf(stderr, "invalid next hop mac address\n");
		return 2;
	}
	argc--;
	argv++;

	if (argc)
		return 1;

	NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, dst);
	NLA_PUT(msg, NL80211_ATTR_MPATH_NEXT_HOP, ETH_ALEN, next_hop);

	nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, print_mpath_handler, NULL);
	return 0;
 nla_put_failure:
	return -ENOBUFS;
}
COMMAND(mpath, new, "<destination MAC address> next_hop <next hop MAC address>",
	NL80211_CMD_NEW_MPATH, 0, CIB_NETDEV, handle_mpath_set);
COMMAND(mpath, set, "<destination MAC address> next_hop <next hop MAC address>",
	NL80211_CMD_SET_MPATH, 0, CIB_NETDEV, handle_mpath_set);

static int handle_mpath_dump(struct nl80211_state *state,
			     struct nl_cb *cb,
			     struct nl_msg *msg,
			     int argc, char **argv)
{
	nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, print_mpath_handler, NULL);
	return 0;
}
COMMAND(mpath, dump, NULL,
	NL80211_CMD_GET_MPATH, NLM_F_DUMP, CIB_NETDEV, handle_mpath_dump);
Commit	Line	Data
2ef1be68	1	#include <net/if.h>
3d1e8704	2	#include <errno.h>
d5ac8ad3	3	#include <string.h>
2ef1be68	4
3d1e8704 LCC	5	#include <netlink/genl/genl.h>
	6	#include <netlink/genl/family.h>
	7	#include <netlink/genl/ctrl.h>
	8	#include <netlink/msg.h>
	9	#include <netlink/attr.h>
3d1e8704	10
f408e01b	11	#include "nl80211.h"
3d1e8704 LCC	12	#include "iw.h"
	13
	14	enum plink_state {
	15	LISTEN,
	16	OPN_SNT,
	17	OPN_RCVD,
	18	CNF_RCVD,
	19	ESTAB,
	20	HOLDING,
	21	BLOCKED
	22	};
	23
	24	enum plink_actions {
	25	PLINK_ACTION_UNDEFINED,
	26	PLINK_ACTION_OPEN,
	27	PLINK_ACTION_BLOCK,
	28	};
	29
	30
3d1e8704 LCC	31	static int print_mpath_handler(struct nl_msg msg, void arg)
	32	{
	33	struct nlattr *tb[NL80211_ATTR_MAX + 1];
	34	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
	35	struct nlattr *pinfo[NL80211_MPATH_INFO_MAX + 1];
	36	char dst[20], next_hop[20], dev[20];
	37	static struct nla_policy mpath_policy[NL80211_MPATH_INFO_MAX + 1] = {
	38	[NL80211_MPATH_INFO_FRAME_QLEN] = { .type = NLA_U32 },
	39	[NL80211_MPATH_INFO_DSN] = { .type = NLA_U32 },
	40	[NL80211_MPATH_INFO_METRIC] = { .type = NLA_U32 },
	41	[NL80211_MPATH_INFO_EXPTIME] = { .type = NLA_U32 },
	42	[NL80211_MPATH_INFO_DISCOVERY_TIMEOUT] = { .type = NLA_U32 },
	43	[NL80211_MPATH_INFO_DISCOVERY_RETRIES] = { .type = NLA_U8 },
	44	[NL80211_MPATH_INFO_FLAGS] = { .type = NLA_U8 },
	45	};
	46
	47	nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
	48	genlmsg_attrlen(gnlh, 0), NULL);
	49
	50	/*
	51	* TODO: validate the interface and mac address!
	52	* Otherwise, there's a race condition as soon as
	53	* the kernel starts sending mpath notifications.
	54	*/
	55
	56	if (!tb[NL80211_ATTR_MPATH_INFO]) {
	57	fprintf(stderr, "mpath info missing!");
	58	return NL_SKIP;
	59	}
	60	if (nla_parse_nested(pinfo, NL80211_MPATH_INFO_MAX,
	61	tb[NL80211_ATTR_MPATH_INFO],
	62	mpath_policy)) {
	63	fprintf(stderr, "failed to parse nested attributes!");
	64	return NL_SKIP;
	65	}
	66
	67	mac_addr_n2a(dst, nla_data(tb[NL80211_ATTR_MAC]));
	68	mac_addr_n2a(next_hop, nla_data(tb[NL80211_ATTR_MPATH_NEXT_HOP]));
	69	if_indextoname(nla_get_u32(tb[NL80211_ATTR_IFINDEX]), dev);
	70	printf("%s %s %s", dst, next_hop, dev);
	71	if (pinfo[NL80211_MPATH_INFO_DSN])
	72	printf("\t%u",
	73	nla_get_u32(pinfo[NL80211_MPATH_INFO_DSN]));
	74	if (pinfo[NL80211_MPATH_INFO_METRIC])
	75	printf("\t%u",
	76	nla_get_u32(pinfo[NL80211_MPATH_INFO_METRIC]));
	77	if (pinfo[NL80211_MPATH_INFO_FRAME_QLEN])
	78	printf("\t%u",
	79	nla_get_u32(pinfo[NL80211_MPATH_INFO_FRAME_QLEN]));
	80	if (pinfo[NL80211_MPATH_INFO_EXPTIME])
	81	printf("\t%u",
	82	nla_get_u32(pinfo[NL80211_MPATH_INFO_EXPTIME]));
	83	if (pinfo[NL80211_MPATH_INFO_DISCOVERY_TIMEOUT])
	84	printf("\t%u",
	85	nla_get_u32(pinfo[NL80211_MPATH_INFO_DISCOVERY_TIMEOUT]));
	86	if (pinfo[NL80211_MPATH_INFO_DISCOVERY_RETRIES])
	87	printf("\t%u",
	88	nla_get_u8(pinfo[NL80211_MPATH_INFO_DISCOVERY_RETRIES]));
	89	if (pinfo[NL80211_MPATH_INFO_FLAGS])
	90	printf("\t0x%x",
	91	nla_get_u8(pinfo[NL80211_MPATH_INFO_FLAGS]));
	92
	93	printf("\n");
	94	return NL_SKIP;
95	}
96
7c37a24d JB	97	static int handle_mpath_get(struct nl80211_state *state,
7c37a24d JB	98	struct nl_cb *cb,
f5f937fb JB	99	struct nl_msg *msg,
f5f937fb JB	100	int argc, char **argv)
3d1e8704	101	{
3d1e8704 LCC	102	unsigned char dst[ETH_ALEN];
3d1e8704 LCC	103
f5f937fb	104	if (argc < 1)
5e75fd04	105	return 1;
3d1e8704 LCC	106
	107	if (mac_addr_a2n(dst, argv[0])) {
	108	fprintf(stderr, "invalid mac address\n");
5e75fd04	109	return 2;
3d1e8704 LCC	110	}
	111	argc--;
	112	argv++;
	113
f5f937fb	114	if (argc)
5e75fd04	115	return 1;
3d1e8704 LCC	116
3d1e8704 LCC	117	NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, dst);
3d1e8704	118
3d1e8704	119	nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, print_mpath_handler, NULL);
3d1e8704	120
70391ccf	121	return 0;
3d1e8704	122	nla_put_failure:
70391ccf	123	return -ENOBUFS;
3d1e8704	124	}
f5f937fb JB	125	COMMAND(mpath, get, "<MAC address>",
	126	NL80211_CMD_GET_MPATH, 0, CIB_NETDEV, handle_mpath_get);
	127	COMMAND(mpath, del, "<MAC address>",
	128	NL80211_CMD_DEL_MPATH, 0, CIB_NETDEV, handle_mpath_get);
	129
7c37a24d JB	130	static int handle_mpath_set(struct nl80211_state *state,
7c37a24d JB	131	struct nl_cb *cb,
f5f937fb JB	132	struct nl_msg *msg,
f5f937fb JB	133	int argc, char **argv)
3d1e8704	134	{
3d1e8704 LCC	135	unsigned char dst[ETH_ALEN];
	136	unsigned char next_hop[ETH_ALEN];
	137
f5f937fb	138	if (argc < 3)
5e75fd04	139	return 1;
3d1e8704 LCC	140
	141	if (mac_addr_a2n(dst, argv[0])) {
	142	fprintf(stderr, "invalid destination mac address\n");
5e75fd04	143	return 2;
3d1e8704 LCC	144	}
	145	argc--;
	146	argv++;
	147
f5f937fb	148	if (strcmp("next_hop", argv[0]) != 0)
5e75fd04	149	return 1;
3d1e8704 LCC	150	argc--;
	151	argv++;
	152
	153	if (mac_addr_a2n(next_hop, argv[0])) {
	154	fprintf(stderr, "invalid next hop mac address\n");
5e75fd04	155	return 2;
3d1e8704 LCC	156	}
	157	argc--;
	158	argv++;
	159
f5f937fb	160	if (argc)
5e75fd04	161	return 1;
3d1e8704	162
3d1e8704	163	NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, dst);
3d1e8704 LCC	164	NLA_PUT(msg, NL80211_ATTR_MPATH_NEXT_HOP, ETH_ALEN, next_hop);
3d1e8704 LCC	165
3d1e8704	166	nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, print_mpath_handler, NULL);
70391ccf	167	return 0;
3d1e8704	168	nla_put_failure:
70391ccf	169	return -ENOBUFS;
3d1e8704	170	}
f5f937fb JB	171	COMMAND(mpath, new, "<destination MAC address> next_hop <next hop MAC address>",
	172	NL80211_CMD_NEW_MPATH, 0, CIB_NETDEV, handle_mpath_set);
	173	COMMAND(mpath, set, "<destination MAC address> next_hop <next hop MAC address>",
	174	NL80211_CMD_SET_MPATH, 0, CIB_NETDEV, handle_mpath_set);
3d1e8704	175
7c37a24d JB	176	static int handle_mpath_dump(struct nl80211_state *state,
7c37a24d JB	177	struct nl_cb *cb,
f5f937fb JB	178	struct nl_msg *msg,
f5f937fb JB	179	int argc, char **argv)
3d1e8704	180	{
3d1e8704	181	nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, print_mpath_handler, NULL);
70391ccf	182	return 0;
3d1e8704	183	}
f5f937fb JB	184	COMMAND(mpath, dump, NULL,
f5f937fb JB	185	NL80211_CMD_GET_MPATH, NLM_F_DUMP, CIB_NETDEV, handle_mpath_dump);