[thirdparty/iw.git] / link.c

#include <net/if.h>
#include <errno.h>
#include <string.h>
#include <ctype.h>
#include <stdbool.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"

struct link_result {
	uint8_t bssid[8];
	bool link_found;
	bool anything_found;
};

static struct link_result lr = { .link_found = false };

static int link_bss_handler(struct nl_msg *msg, void *arg)
{
	struct nlattr *tb[NL80211_ATTR_MAX + 1];
	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
	struct nlattr *bss[NL80211_BSS_MAX + 1];
	static struct nla_policy bss_policy[NL80211_BSS_MAX + 1] = {
		[NL80211_BSS_TSF] = { .type = NLA_U64 },
		[NL80211_BSS_FREQUENCY] = { .type = NLA_U32 },
		[NL80211_BSS_BSSID] = { },
		[NL80211_BSS_BEACON_INTERVAL] = { .type = NLA_U16 },
		[NL80211_BSS_CAPABILITY] = { .type = NLA_U16 },
		[NL80211_BSS_INFORMATION_ELEMENTS] = { },
		[NL80211_BSS_SIGNAL_MBM] = { .type = NLA_U32 },
		[NL80211_BSS_SIGNAL_UNSPEC] = { .type = NLA_U8 },
		[NL80211_BSS_STATUS] = { .type = NLA_U32 },
	};
	struct link_result *result = arg;
	char mac_addr[20], dev[20];

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

	if (!tb[NL80211_ATTR_BSS]) {
		fprintf(stderr, "bss info missing!\n");
		return NL_SKIP;
	}
	if (nla_parse_nested(bss, NL80211_BSS_MAX,
			     tb[NL80211_ATTR_BSS],
			     bss_policy)) {
		fprintf(stderr, "failed to parse nested attributes!\n");
		return NL_SKIP;
	}

	if (!bss[NL80211_BSS_BSSID])
		return NL_SKIP;

	if (!bss[NL80211_BSS_STATUS])
		return NL_SKIP;

	mac_addr_n2a(mac_addr, nla_data(bss[NL80211_BSS_BSSID]));
	if_indextoname(nla_get_u32(tb[NL80211_ATTR_IFINDEX]), dev);

	switch (nla_get_u32(bss[NL80211_BSS_STATUS])) {
	case NL80211_BSS_STATUS_ASSOCIATED:
		printf("Connected to %s (on %s)\n", mac_addr, dev);
		break;
	case NL80211_BSS_STATUS_AUTHENTICATED:
		printf("Authenticated with %s (on %s)\n", mac_addr, dev);
		return NL_SKIP;
	case NL80211_BSS_STATUS_IBSS_JOINED:
		printf("Joined IBSS %s (on %s)\n", mac_addr, dev);
		break;
	default:
		return NL_SKIP;
	}

	result->anything_found = true;

	if (bss[NL80211_BSS_INFORMATION_ELEMENTS])
		print_ies(nla_data(bss[NL80211_BSS_INFORMATION_ELEMENTS]),
			  nla_len(bss[NL80211_BSS_INFORMATION_ELEMENTS]),
			  false, PRINT_LINK);

	if (bss[NL80211_BSS_FREQUENCY])
		printf("\tfreq: %d\n",
			nla_get_u32(bss[NL80211_BSS_FREQUENCY]));

	if (nla_get_u32(bss[NL80211_BSS_STATUS]) != NL80211_BSS_STATUS_ASSOCIATED)
		return NL_SKIP;

	/* only in the assoc case do we want more info from station get */
	result->link_found = true;
	memcpy(result->bssid, nla_data(bss[NL80211_BSS_BSSID]), 6);
	return NL_SKIP;
}

static int handle_scan_for_link(struct nl80211_state *state,
				struct nl_cb *cb,
				struct nl_msg *msg,
				int argc, char **argv)
{
	if (argc > 0)
		return 1;

	nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, link_bss_handler, &lr);
	return 0;
}

static int print_link_sta(struct nl_msg *msg, void *arg)
{
	struct nlattr *tb[NL80211_ATTR_MAX + 1];
	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
	struct nlattr *sinfo[NL80211_STA_INFO_MAX + 1];
	struct nlattr *rinfo[NL80211_RATE_INFO_MAX + 1];
	static struct nla_policy stats_policy[NL80211_STA_INFO_MAX + 1] = {
		[NL80211_STA_INFO_INACTIVE_TIME] = { .type = NLA_U32 },
		[NL80211_STA_INFO_RX_BYTES] = { .type = NLA_U32 },
		[NL80211_STA_INFO_TX_BYTES] = { .type = NLA_U32 },
		[NL80211_STA_INFO_RX_PACKETS] = { .type = NLA_U32 },
		[NL80211_STA_INFO_TX_PACKETS] = { .type = NLA_U32 },
		[NL80211_STA_INFO_SIGNAL] = { .type = NLA_U8 },
		[NL80211_STA_INFO_TX_BITRATE] = { .type = NLA_NESTED },
		[NL80211_STA_INFO_LLID] = { .type = NLA_U16 },
		[NL80211_STA_INFO_PLID] = { .type = NLA_U16 },
		[NL80211_STA_INFO_PLINK_STATE] = { .type = NLA_U8 },
	};

	static struct nla_policy rate_policy[NL80211_RATE_INFO_MAX + 1] = {
		[NL80211_RATE_INFO_BITRATE] = { .type = NLA_U16 },
		[NL80211_RATE_INFO_MCS] = { .type = NLA_U8 },
		[NL80211_RATE_INFO_40_MHZ_WIDTH] = { .type = NLA_FLAG },
		[NL80211_RATE_INFO_SHORT_GI] = { .type = NLA_FLAG },
	};

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

	if (!tb[NL80211_ATTR_STA_INFO]) {
		fprintf(stderr, "sta stats missing!\n");
		return NL_SKIP;
	}
	if (nla_parse_nested(sinfo, NL80211_STA_INFO_MAX,
			     tb[NL80211_ATTR_STA_INFO],
			     stats_policy)) {
		fprintf(stderr, "failed to parse nested attributes!\n");
		return NL_SKIP;
	}

	if (sinfo[NL80211_STA_INFO_RX_BYTES] && sinfo[NL80211_STA_INFO_RX_PACKETS])
		printf("\tRX: %u bytes (%u packets)\n",
			nla_get_u32(sinfo[NL80211_STA_INFO_RX_BYTES]),
			nla_get_u32(sinfo[NL80211_STA_INFO_RX_PACKETS]));
	if (sinfo[NL80211_STA_INFO_TX_BYTES] && sinfo[NL80211_STA_INFO_TX_PACKETS])
		printf("\tTX: %u bytes (%u packets)\n",
			nla_get_u32(sinfo[NL80211_STA_INFO_TX_BYTES]),
			nla_get_u32(sinfo[NL80211_STA_INFO_TX_PACKETS]));
	if (sinfo[NL80211_STA_INFO_SIGNAL])
		printf("\tsignal: %d dBm\n",
			(int8_t)nla_get_u8(sinfo[NL80211_STA_INFO_SIGNAL]));

	if (sinfo[NL80211_STA_INFO_TX_BITRATE]) {
		if (nla_parse_nested(rinfo, NL80211_RATE_INFO_MAX,
				     sinfo[NL80211_STA_INFO_TX_BITRATE], rate_policy)) {
			fprintf(stderr, "failed to parse nested rate attributes!\n");
		} else {
			printf("\ttx bitrate: ");
			if (rinfo[NL80211_RATE_INFO_BITRATE]) {
				int rate = nla_get_u16(rinfo[NL80211_RATE_INFO_BITRATE]);
				printf("%d.%d MBit/s", rate / 10, rate % 10);
			}

			if (rinfo[NL80211_RATE_INFO_MCS])
				printf(" MCS %d", nla_get_u8(rinfo[NL80211_RATE_INFO_MCS]));
			if (rinfo[NL80211_RATE_INFO_40_MHZ_WIDTH])
				printf(" 40Mhz");
			if (rinfo[NL80211_RATE_INFO_SHORT_GI])
				printf(" short GI");
			printf("\n");
		}
	}

	return NL_SKIP;
}

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

	if (argc < 1)
		return 1;

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

	argc--;
	argv++;

	if (argc)
		return 1;

	NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, mac_addr);

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

	return 0;
 nla_put_failure:
	return -ENOBUFS;
}

static int handle_link(struct nl80211_state *state, struct nl_cb *cb,
		       struct nl_msg *msg, int argc, char **argv)
{
	char *link_argv[] = {
		NULL,
		"link",
		"get_bss",
		NULL,
	};
	char *station_argv[] = {
		NULL,
		"link",
		"get_sta",
		NULL,
		NULL,
	};
	char bssid_buf[3*6];
	int err;

	link_argv[0] = argv[0];
	err = handle_cmd(state, II_NETDEV, 3, link_argv);
	if (err)
		return err;

	if (!lr.link_found) {
		if (!lr.anything_found)
			printf("Not connected.\n");
		return 0;
	}

	mac_addr_n2a(bssid_buf, lr.bssid);
	bssid_buf[17] = '\0';

	station_argv[0] = argv[0];
	station_argv[3] = bssid_buf;
	return handle_cmd(state, II_NETDEV, 4, station_argv);
}
TOPLEVEL(link, NULL, 0, 0, CIB_NETDEV, handle_link,
	 "Print information about the current link, if any.");
HIDDEN(link, get_sta, "", NL80211_CMD_GET_STATION, 0,
	CIB_NETDEV, handle_link_sta);
HIDDEN(link, get_bss, NULL, NL80211_CMD_GET_SCAN, NLM_F_DUMP,
	CIB_NETDEV, handle_scan_for_link);
Commit	Line	Data
febeb0c0 JB	1	#include <net/if.h>
	2	#include <errno.h>
	3	#include <string.h>
	4	#include <ctype.h>
	5	#include <stdbool.h>
	6
	7	#include <netlink/genl/genl.h>
	8	#include <netlink/genl/family.h>
	9	#include <netlink/genl/ctrl.h>
	10	#include <netlink/msg.h>
	11	#include <netlink/attr.h>
	12
	13	#include "nl80211.h"
	14	#include "iw.h"
	15
	16	struct link_result {
	17	uint8_t bssid[8];
	18	bool link_found;
	19	bool anything_found;
	20	};
	21
	22	static struct link_result lr = { .link_found = false };
	23
	24	static int link_bss_handler(struct nl_msg msg, void arg)
	25	{
	26	struct nlattr *tb[NL80211_ATTR_MAX + 1];
	27	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
	28	struct nlattr *bss[NL80211_BSS_MAX + 1];
	29	static struct nla_policy bss_policy[NL80211_BSS_MAX + 1] = {
	30	[NL80211_BSS_TSF] = { .type = NLA_U64 },
	31	[NL80211_BSS_FREQUENCY] = { .type = NLA_U32 },
	32	[NL80211_BSS_BSSID] = { },
	33	[NL80211_BSS_BEACON_INTERVAL] = { .type = NLA_U16 },
	34	[NL80211_BSS_CAPABILITY] = { .type = NLA_U16 },
	35	[NL80211_BSS_INFORMATION_ELEMENTS] = { },
	36	[NL80211_BSS_SIGNAL_MBM] = { .type = NLA_U32 },
	37	[NL80211_BSS_SIGNAL_UNSPEC] = { .type = NLA_U8 },
	38	[NL80211_BSS_STATUS] = { .type = NLA_U32 },
	39	};
	40	struct link_result *result = arg;
	41	char mac_addr[20], dev[20];
	42
	43	nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
	44	genlmsg_attrlen(gnlh, 0), NULL);
	45
	46	if (!tb[NL80211_ATTR_BSS]) {
5fe70c0e	47	fprintf(stderr, "bss info missing!\n");
febeb0c0 JB	48	return NL_SKIP;
	49	}
	50	if (nla_parse_nested(bss, NL80211_BSS_MAX,
	51	tb[NL80211_ATTR_BSS],
	52	bss_policy)) {
5fe70c0e	53	fprintf(stderr, "failed to parse nested attributes!\n");
febeb0c0 JB	54	return NL_SKIP;
	55	}
	56
	57	if (!bss[NL80211_BSS_BSSID])
	58	return NL_SKIP;
	59
	60	if (!bss[NL80211_BSS_STATUS])
	61	return NL_SKIP;
	62
	63	mac_addr_n2a(mac_addr, nla_data(bss[NL80211_BSS_BSSID]));
	64	if_indextoname(nla_get_u32(tb[NL80211_ATTR_IFINDEX]), dev);
	65
	66	switch (nla_get_u32(bss[NL80211_BSS_STATUS])) {
	67	case NL80211_BSS_STATUS_ASSOCIATED:
	68	printf("Connected to %s (on %s)\n", mac_addr, dev);
	69	break;
	70	case NL80211_BSS_STATUS_AUTHENTICATED:
	71	printf("Authenticated with %s (on %s)\n", mac_addr, dev);
	72	return NL_SKIP;
28b1efb7 JB	73	case NL80211_BSS_STATUS_IBSS_JOINED:
	74	printf("Joined IBSS %s (on %s)\n", mac_addr, dev);
	75	break;
febeb0c0 JB	76	default:
	77	return NL_SKIP;
	78	}
	79
	80	result->anything_found = true;
	81
	82	if (bss[NL80211_BSS_INFORMATION_ELEMENTS])
	83	print_ies(nla_data(bss[NL80211_BSS_INFORMATION_ELEMENTS]),
	84	nla_len(bss[NL80211_BSS_INFORMATION_ELEMENTS]),
	85	false, PRINT_LINK);
	86
	87	if (bss[NL80211_BSS_FREQUENCY])
	88	printf("\tfreq: %d\n",
	89	nla_get_u32(bss[NL80211_BSS_FREQUENCY]));
	90
	91	if (nla_get_u32(bss[NL80211_BSS_STATUS]) != NL80211_BSS_STATUS_ASSOCIATED)
	92	return NL_SKIP;
	93
	94	/* only in the assoc case do we want more info from station get */
	95	result->link_found = true;
	96	memcpy(result->bssid, nla_data(bss[NL80211_BSS_BSSID]), 6);
	97	return NL_SKIP;
	98	}
	99
	100	static int handle_scan_for_link(struct nl80211_state *state,
	101	struct nl_cb *cb,
	102	struct nl_msg *msg,
	103	int argc, char **argv)
	104	{
	105	if (argc > 0)
	106	return 1;
	107
	108	nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, link_bss_handler, &lr);
	109	return 0;
	110	}
febeb0c0 JB	111
	112	static int print_link_sta(struct nl_msg msg, void arg)
	113	{
	114	struct nlattr *tb[NL80211_ATTR_MAX + 1];
	115	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
	116	struct nlattr *sinfo[NL80211_STA_INFO_MAX + 1];
	117	struct nlattr *rinfo[NL80211_RATE_INFO_MAX + 1];
	118	static struct nla_policy stats_policy[NL80211_STA_INFO_MAX + 1] = {
	119	[NL80211_STA_INFO_INACTIVE_TIME] = { .type = NLA_U32 },
	120	[NL80211_STA_INFO_RX_BYTES] = { .type = NLA_U32 },
	121	[NL80211_STA_INFO_TX_BYTES] = { .type = NLA_U32 },
	122	[NL80211_STA_INFO_RX_PACKETS] = { .type = NLA_U32 },
	123	[NL80211_STA_INFO_TX_PACKETS] = { .type = NLA_U32 },
	124	[NL80211_STA_INFO_SIGNAL] = { .type = NLA_U8 },
	125	[NL80211_STA_INFO_TX_BITRATE] = { .type = NLA_NESTED },
	126	[NL80211_STA_INFO_LLID] = { .type = NLA_U16 },
	127	[NL80211_STA_INFO_PLID] = { .type = NLA_U16 },
	128	[NL80211_STA_INFO_PLINK_STATE] = { .type = NLA_U8 },
	129	};
	130
	131	static struct nla_policy rate_policy[NL80211_RATE_INFO_MAX + 1] = {
	132	[NL80211_RATE_INFO_BITRATE] = { .type = NLA_U16 },
	133	[NL80211_RATE_INFO_MCS] = { .type = NLA_U8 },
	134	[NL80211_RATE_INFO_40_MHZ_WIDTH] = { .type = NLA_FLAG },
	135	[NL80211_RATE_INFO_SHORT_GI] = { .type = NLA_FLAG },
	136	};
	137
	138	nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
	139	genlmsg_attrlen(gnlh, 0), NULL);
	140
	141	if (!tb[NL80211_ATTR_STA_INFO]) {
5fe70c0e	142	fprintf(stderr, "sta stats missing!\n");
febeb0c0 JB	143	return NL_SKIP;
	144	}
	145	if (nla_parse_nested(sinfo, NL80211_STA_INFO_MAX,
	146	tb[NL80211_ATTR_STA_INFO],
	147	stats_policy)) {
5fe70c0e	148	fprintf(stderr, "failed to parse nested attributes!\n");
febeb0c0 JB	149	return NL_SKIP;
	150	}
	151
	152	if (sinfo[NL80211_STA_INFO_RX_BYTES] && sinfo[NL80211_STA_INFO_RX_PACKETS])
26181415	153	printf("\tRX: %u bytes (%u packets)\n",
febeb0c0 JB	154	nla_get_u32(sinfo[NL80211_STA_INFO_RX_BYTES]),
	155	nla_get_u32(sinfo[NL80211_STA_INFO_RX_PACKETS]));
	156	if (sinfo[NL80211_STA_INFO_TX_BYTES] && sinfo[NL80211_STA_INFO_TX_PACKETS])
26181415	157	printf("\tTX: %u bytes (%u packets)\n",
febeb0c0 JB	158	nla_get_u32(sinfo[NL80211_STA_INFO_TX_BYTES]),
	159	nla_get_u32(sinfo[NL80211_STA_INFO_TX_PACKETS]));
	160	if (sinfo[NL80211_STA_INFO_SIGNAL])
	161	printf("\tsignal: %d dBm\n",
	162	(int8_t)nla_get_u8(sinfo[NL80211_STA_INFO_SIGNAL]));
	163
	164	if (sinfo[NL80211_STA_INFO_TX_BITRATE]) {
	165	if (nla_parse_nested(rinfo, NL80211_RATE_INFO_MAX,
	166	sinfo[NL80211_STA_INFO_TX_BITRATE], rate_policy)) {
5fe70c0e	167	fprintf(stderr, "failed to parse nested rate attributes!\n");
febeb0c0 JB	168	} else {
	169	printf("\ttx bitrate: ");
	170	if (rinfo[NL80211_RATE_INFO_BITRATE]) {
	171	int rate = nla_get_u16(rinfo[NL80211_RATE_INFO_BITRATE]);
	172	printf("%d.%d MBit/s", rate / 10, rate % 10);
	173	}
	174
	175	if (rinfo[NL80211_RATE_INFO_MCS])
	176	printf(" MCS %d", nla_get_u8(rinfo[NL80211_RATE_INFO_MCS]));
	177	if (rinfo[NL80211_RATE_INFO_40_MHZ_WIDTH])
	178	printf(" 40Mhz");
	179	if (rinfo[NL80211_RATE_INFO_SHORT_GI])
	180	printf(" short GI");
	181	printf("\n");
	182	}
	183	}
	184
	185	return NL_SKIP;
	186	}
	187
	188	static int handle_link_sta(struct nl80211_state *state,
	189	struct nl_cb *cb,
	190	struct nl_msg *msg,
	191	int argc, char **argv)
	192	{
	193	unsigned char mac_addr[ETH_ALEN];
	194
	195	if (argc < 1)
	196	return 1;
	197
	198	if (mac_addr_a2n(mac_addr, argv[0])) {
	199	fprintf(stderr, "invalid mac address\n");
	200	return 2;
	201	}
	202
	203	argc--;
	204	argv++;
	205
	206	if (argc)
	207	return 1;
	208
	209	NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, mac_addr);
	210
	211	nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, print_link_sta, NULL);
	212
	213	return 0;
	214	nla_put_failure:
	215	return -ENOBUFS;
	216	}
febeb0c0 JB	217
	218	static int handle_link(struct nl80211_state state, struct nl_cb cb,
	219	struct nl_msg msg, int argc, char *argv)
	220	{
	221	char *link_argv[] = {
	222	NULL,
	223	"link",
	224	"get_bss",
	225	NULL,
	226	};
	227	char *station_argv[] = {
	228	NULL,
	229	"link",
	230	"get_sta",
	231	NULL,
	232	NULL,
	233	};
	234	char bssid_buf[3*6];
	235	int err;
	236
	237	link_argv[0] = argv[0];
	238	err = handle_cmd(state, II_NETDEV, 3, link_argv);
	239	if (err)
	240	return err;
	241
	242	if (!lr.link_found) {
	243	if (!lr.anything_found)
bb68dbd4	244	printf("Not connected.\n");
febeb0c0 JB	245	return 0;
	246	}
	247
	248	mac_addr_n2a(bssid_buf, lr.bssid);
	249	bssid_buf[17] = '\0';
	250
	251	station_argv[0] = argv[0];
	252	station_argv[3] = bssid_buf;
	253	return handle_cmd(state, II_NETDEV, 4, station_argv);
	254	}
	255	TOPLEVEL(link, NULL, 0, 0, CIB_NETDEV, handle_link,
	256	"Print information about the current link, if any.");
4698bfc2 JB	257	HIDDEN(link, get_sta, "", NL80211_CMD_GET_STATION, 0,
	258	CIB_NETDEV, handle_link_sta);
	259	HIDDEN(link, get_bss, NULL, NL80211_CMD_GET_SCAN, NLM_F_DUMP,
	260	CIB_NETDEV, handle_scan_for_link);