[thirdparty/iw.git] / vendor.c

#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"

SECTION(vendor);

static int print_vendor_response(struct nl_msg *msg, void *arg)
{
	struct nlattr *attr;
	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
	bool print_ascii = (bool) arg;
	uint8_t *data;
	int len;

	attr = nla_find(genlmsg_attrdata(gnlh, 0),
			genlmsg_attrlen(gnlh, 0),
			NL80211_ATTR_VENDOR_DATA);
	if (!attr) {
		fprintf(stderr, "vendor data attribute missing!\n");
		return NL_SKIP;
	}

	data = (uint8_t *) nla_data(attr);
	len = nla_len(attr);

	if (print_ascii)
		iw_hexdump("vendor response", data, len);
	else
		fwrite(data, 1, len, stdout);

	return NL_OK;
}

static int read_file(FILE *file, char *buf, size_t size)
{
	size_t count = 0;
	int data;

	while ((data = fgetc(file)) != EOF) {
		if (count >= size)
			return -EINVAL;
		buf[count] = data;
		count++;
	}

	return count;
}

static int read_hex(unsigned int argc, char **argv, char *buf, size_t size)
{
	unsigned int i, data;
	int res;

	if (argc > size)
		return -EINVAL;

	for (i = 0; i < argc; i++) {
		res = sscanf(argv[i], "0x%x", &data);
		if (res != 1 || data > 0xff)
			return -EINVAL;
		buf[i] = data;
	}

	return argc;
}

static int handle_vendor(struct nl80211_state *state,
			 struct nl_msg *msg, int argc, char **argv,
			 enum id_input id)
{
	unsigned int oui;
	unsigned int subcmd;
	char buf[2048] = {};
	int res, count = 0;
	FILE *file = NULL;

	if (argc < 3)
		return 1;

	res = sscanf(argv[0], "0x%x", &oui);
	if (res != 1) {
		printf("Vendor command must start with 0x\n");
		return 2;
	}

	res = sscanf(argv[1], "0x%x", &subcmd);
	if (res != 1) {
		printf("Sub command must start with 0x\n");
		return 2;
	}

	if (!strcmp(argv[2], "-"))
		file = stdin;
	else
		file = fopen(argv[2], "r");

	NLA_PUT_U32(msg, NL80211_ATTR_VENDOR_ID, oui);
	NLA_PUT_U32(msg, NL80211_ATTR_VENDOR_SUBCMD, subcmd);

	if (file) {
		count = read_file(file, buf, sizeof(buf));
		if (file != stdin)
			fclose(file);
	} else
		count = read_hex(argc - 2, &argv[2], buf, sizeof(buf));

	if (count < 0)
		return -EINVAL;

	if (count > 0)
		NLA_PUT(msg, NL80211_ATTR_VENDOR_DATA, count, buf);

	return 0;

nla_put_failure:
	if (file && file != stdin)
		fclose(file);
	return -ENOBUFS;
}

static int handle_vendor_recv(struct nl80211_state *state,
			      struct nl_msg *msg, int argc,
			      char **argv, enum id_input id)
{
	register_handler(print_vendor_response, (void *) true);
	return handle_vendor(state, msg, argc, argv, id);
}

static int handle_vendor_recv_bin(struct nl80211_state *state,
				  struct nl_msg *msg, int argc,
				  char **argv, enum id_input id)
{
	register_handler(print_vendor_response, (void *) false);
	return handle_vendor(state, msg, argc, argv, id);
}

COMMAND(vendor, send, "<oui> <subcmd> <filename|-|hex data>", NL80211_CMD_VENDOR, 0, CIB_NETDEV, handle_vendor, "");
COMMAND(vendor, recv, "<oui> <subcmd> <filename|-|hex data>", NL80211_CMD_VENDOR, 0, CIB_NETDEV, handle_vendor_recv, "");
COMMAND(vendor, recvbin, "<oui> <subcmd> <filename|-|hex data>", NL80211_CMD_VENDOR, 0, CIB_NETDEV, handle_vendor_recv_bin, "");
Commit	Line	Data
894d2e2b JD	1	#include <errno.h>
	2	#include <string.h>
	3
	4	#include <netlink/genl/genl.h>
	5	#include <netlink/genl/family.h>
	6	#include <netlink/genl/ctrl.h>
	7	#include <netlink/msg.h>
	8	#include <netlink/attr.h>
	9
	10	#include "nl80211.h"
	11	#include "iw.h"
	12
	13	SECTION(vendor);
	14
e7c09230 ES	15	static int print_vendor_response(struct nl_msg msg, void arg)
	16	{
	17	struct nlattr *attr;
	18	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
	19	bool print_ascii = (bool) arg;
	20	uint8_t *data;
	21	int len;
	22
	23	attr = nla_find(genlmsg_attrdata(gnlh, 0),
	24	genlmsg_attrlen(gnlh, 0),
	25	NL80211_ATTR_VENDOR_DATA);
	26	if (!attr) {
	27	fprintf(stderr, "vendor data attribute missing!\n");
	28	return NL_SKIP;
	29	}
	30
	31	data = (uint8_t *) nla_data(attr);
	32	len = nla_len(attr);
	33
	34	if (print_ascii)
	35	iw_hexdump("vendor response", data, len);
	36	else
	37	fwrite(data, 1, len, stdout);
	38
	39	return NL_OK;
	40	}
	41
894d2e2b JD	42	static int read_file(FILE file, char buf, size_t size)
894d2e2b JD	43	{
0ee571d5 JB	44	size_t count = 0;
0ee571d5 JB	45	int data;
894d2e2b JD	46
	47	while ((data = fgetc(file)) != EOF) {
	48	if (count >= size)
	49	return -EINVAL;
	50	buf[count] = data;
	51	count++;
	52	}
	53
	54	return count;
	55	}
	56
0ee571d5	57	static int read_hex(unsigned int argc, char *argv, char buf, size_t size)
894d2e2b	58	{
0ee571d5 JB	59	unsigned int i, data;
0ee571d5 JB	60	int res;
894d2e2b JD	61
	62	if (argc > size)
	63	return -EINVAL;
	64
	65	for (i = 0; i < argc; i++) {
	66	res = sscanf(argv[i], "0x%x", &data);
	67	if (res != 1 \|\| data > 0xff)
	68	return -EINVAL;
	69	buf[i] = data;
	70	}
	71
	72	return argc;
	73	}
	74
34b23014	75	static int handle_vendor(struct nl80211_state *state,
894d2e2b JD	76	struct nl_msg msg, int argc, char *argv,
	77	enum id_input id)
	78	{
	79	unsigned int oui;
	80	unsigned int subcmd;
	81	char buf[2048] = {};
	82	int res, count = 0;
	83	FILE *file = NULL;
	84
	85	if (argc < 3)
6cd36342	86	return 1;
894d2e2b JD	87
894d2e2b JD	88	res = sscanf(argv[0], "0x%x", &oui);
6cd36342 OO	89	if (res != 1) {
	90	printf("Vendor command must start with 0x\n");
	91	return 2;
	92	}
894d2e2b JD	93
894d2e2b JD	94	res = sscanf(argv[1], "0x%x", &subcmd);
6cd36342 OO	95	if (res != 1) {
	96	printf("Sub command must start with 0x\n");
	97	return 2;
	98	}
894d2e2b JD	99
	100	if (!strcmp(argv[2], "-"))
	101	file = stdin;
	102	else
	103	file = fopen(argv[2], "r");
	104
	105	NLA_PUT_U32(msg, NL80211_ATTR_VENDOR_ID, oui);
	106	NLA_PUT_U32(msg, NL80211_ATTR_VENDOR_SUBCMD, subcmd);
	107
	108	if (file) {
	109	count = read_file(file, buf, sizeof(buf));
fc38d3e8 JC	110	if (file != stdin)
fc38d3e8 JC	111	fclose(file);
894d2e2b JD	112	} else
	113	count = read_hex(argc - 2, &argv[2], buf, sizeof(buf));
	114
	115	if (count < 0)
	116	return -EINVAL;
	117
	118	if (count > 0)
	119	NLA_PUT(msg, NL80211_ATTR_VENDOR_DATA, count, buf);
	120
	121	return 0;
	122
	123	nla_put_failure:
fc38d3e8 JC	124	if (file && file != stdin)
fc38d3e8 JC	125	fclose(file);
894d2e2b JD	126	return -ENOBUFS;
	127	}
	128
e7c09230 ES	129	static int handle_vendor_recv(struct nl80211_state *state,
	130	struct nl_msg *msg, int argc,
	131	char **argv, enum id_input id)
	132	{
	133	register_handler(print_vendor_response, (void *) true);
	134	return handle_vendor(state, msg, argc, argv, id);
	135	}
	136
	137	static int handle_vendor_recv_bin(struct nl80211_state *state,
	138	struct nl_msg *msg, int argc,
	139	char **argv, enum id_input id)
	140	{
	141	register_handler(print_vendor_response, (void *) false);
	142	return handle_vendor(state, msg, argc, argv, id);
	143	}
	144
894d2e2b	145	COMMAND(vendor, send, "<oui> <subcmd> <filename\|-\|hex data>", NL80211_CMD_VENDOR, 0, CIB_NETDEV, handle_vendor, "");
e7c09230 ES	146	COMMAND(vendor, recv, "<oui> <subcmd> <filename\|-\|hex data>", NL80211_CMD_VENDOR, 0, CIB_NETDEV, handle_vendor_recv, "");
e7c09230 ES	147	COMMAND(vendor, recvbin, "<oui> <subcmd> <filename\|-\|hex data>", NL80211_CMD_VENDOR, 0, CIB_NETDEV, handle_vendor_recv_bin, "");