[thirdparty/iw.git] / connect.c

#include <errno.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"

static int iw_conn(struct nl80211_state *state,
		   struct nl_msg *msg, int argc, char **argv,
		   enum id_input id)
{
	char *end;
	unsigned char bssid[6];
	int freq;

	if (argc < 1)
		return 1;

	/* SSID */
	NLA_PUT(msg, NL80211_ATTR_SSID, strlen(argv[0]), argv[0]);
	argv++;
	argc--;

	/* freq */
	if (argc) {
		freq = strtoul(argv[0], &end, 10);
		if (*end == '\0') {
			NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, freq);
			argv++;
			argc--;
		}
	}

	/* bssid */
	if (argc) {
		if (mac_addr_a2n(bssid, argv[0]) == 0) {
			NLA_PUT(msg, NL80211_ATTR_MAC, 6, bssid);
			argv++;
			argc--;
		}
	}

	if (!argc)
		return 0;

	if (strcmp(*argv, "key") != 0 && strcmp(*argv, "keys") != 0)
		return 1;

	argv++;
	argc--;

	return parse_keys(msg, argv, argc);
 nla_put_failure:
	return -ENOSPC;
}

static int disconnect(struct nl80211_state *state,
		      struct nl_msg *msg,
		      int argc, char **argv,
		      enum id_input id)
{
	return 0;
}
TOPLEVEL(disconnect, NULL,
	NL80211_CMD_DISCONNECT, 0, CIB_NETDEV, disconnect,
	"Disconnect from the current network.");

static int iw_connect(struct nl80211_state *state,
		      struct nl_msg *msg, int argc, char **argv,
		      enum id_input id)
{
	char **conn_argv, *dev = argv[0];
	static const __u32 cmds[] = {
		NL80211_CMD_CONNECT,
	};
	struct print_event_args printargs = { };
	int conn_argc, err;
	bool wait = false;
	int i;

	/* strip "wlan0 connect" */
	argc -= 2;
	argv += 2;

	/* check -w */
	if (argc && strcmp(argv[0], "-w") == 0) {
		wait = true;
		argc--;
		argv++;
	}

	err = __prepare_listen_events(state);
	if (err)
		return err;

	conn_argc = 3 + argc;
	conn_argv = calloc(conn_argc, sizeof(*conn_argv));
	if (!conn_argv)
		return -ENOMEM;

	conn_argv[0] = dev;
	conn_argv[1] = "connect";
	conn_argv[2] = "establish";
	for (i = 0; i < argc; i++)
		conn_argv[i + 3] = argv[i];
	err = handle_cmd(state, id, conn_argc, conn_argv);
	free(conn_argv);
	if (err)
		return err;

	if (!wait)
		return 0;

	/*
	 * WARNING: DO NOT COPY THIS CODE INTO YOUR APPLICATION
	 *
	 * This code has a bug:
	 *
	 * It is possible for a connect result message from another
	 * connect attempt to be processed here first, because we
	 * start listening to the multicast group before starting
	 * our own connect request, which may succeed but we get a
	 * fail message from a previous attempt that raced with us,
	 * or similar.
	 *
	 * The only proper way to fix this would be to listen to events
	 * before sending the command, and for the kernel to send the
	 * connect request or a cookie along with the event, so that you
	 * can match up whether the connect _you_ requested was finished
	 * or aborted.
	 *
	 * Alas, the kernel doesn't do that (yet).
	 */

	__do_listen_events(state, ARRAY_SIZE(cmds), cmds, &printargs);
	return 0;
}
TOPLEVEL(connect, "[-w] <SSID> [<freq in MHz>] [<bssid>] [key 0:abcde d:1:6162636465]",
	0, 0, CIB_NETDEV, iw_connect,
	"Join the network with the given SSID (and frequency, BSSID).\n"
	"With -w, wait for the connect to finish or fail.");
HIDDEN(connect, establish, "", NL80211_CMD_CONNECT, 0, CIB_NETDEV, iw_conn);

static int iw_auth(struct nl80211_state *state,
		   struct nl_msg *msg, int argc, char **argv,
		   enum id_input id)
{
	char *end;
	unsigned char bssid[6];
	int freq;
	bool need_key = false;

	if (argc < 4)
		return 1;

	/* SSID */
	NLA_PUT(msg, NL80211_ATTR_SSID, strlen(argv[0]), argv[0]);
	argv++;
	argc--;

	/* bssid */
	if (mac_addr_a2n(bssid, argv[0]) == 0) {
		NLA_PUT(msg, NL80211_ATTR_MAC, 6, bssid);
		argv++;
		argc--;
	} else {
		return 1;
	}

	/* FIXME */
	if (strcmp(argv[0], "open") == 0) {
		NLA_PUT_U32(msg, NL80211_ATTR_AUTH_TYPE,
			    NL80211_AUTHTYPE_OPEN_SYSTEM);
	} else if (strcmp(argv[0], "shared") == 0) {
		NLA_PUT_U32(msg, NL80211_ATTR_AUTH_TYPE,
			    NL80211_AUTHTYPE_SHARED_KEY);
		need_key = true;
	} else {
		return 1;
	}
	argv++;
	argc--;

	freq = strtoul(argv[0], &end, 10);
	if (*end == '\0') {
		NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, freq);
		argv++;
		argc--;
	} else {
		return 1;
	}

	if (!argc && need_key)
		return 1;
	if (argc && !need_key)
		return 1;
	if (!argc)
		return 0;

	if (strcmp(*argv, "key") != 0 && strcmp(*argv, "keys") != 0)
		return 1;

	argv++;
	argc--;

	return parse_keys(msg, argv, argc);
 nla_put_failure:
	return -ENOSPC;
}

TOPLEVEL(auth, "<SSID> <bssid> <type:open|shared> <freq in MHz> [key 0:abcde d:1:6162636465]",
	 NL80211_CMD_AUTHENTICATE, 0, CIB_NETDEV, iw_auth,
	 "Authenticate with the given network.\n");
Commit	Line	Data
99dde7aa JB	1	#include <errno.h>
	2
	3	#include <netlink/genl/genl.h>
	4	#include <netlink/genl/family.h>
	5	#include <netlink/genl/ctrl.h>
	6	#include <netlink/msg.h>
	7	#include <netlink/attr.h>
	8
	9	#include "nl80211.h"
	10	#include "iw.h"
	11
34b23014	12	static int iw_conn(struct nl80211_state *state,
05514f95 JB	13	struct nl_msg msg, int argc, char *argv,
05514f95 JB	14	enum id_input id)
99dde7aa JB	15	{
99dde7aa JB	16	char *end;
51e9bd80	17	unsigned char bssid[6];
1e03690e	18	int freq;
99dde7aa JB	19
	20	if (argc < 1)
	21	return 1;
	22
	23	/* SSID */
	24	NLA_PUT(msg, NL80211_ATTR_SSID, strlen(argv[0]), argv[0]);
	25	argv++;
	26	argc--;
	27
	28	/* freq */
	29	if (argc) {
	30	freq = strtoul(argv[0], &end, 10);
	31	if (*end == '\0') {
	32	NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, freq);
	33	argv++;
	34	argc--;
	35	}
	36	}
	37
	38	/* bssid */
	39	if (argc) {
51e9bd80 JB	40	if (mac_addr_a2n(bssid, argv[0]) == 0) {
	41	NLA_PUT(msg, NL80211_ATTR_MAC, 6, bssid);
	42	argv++;
	43	argc--;
	44	}
99dde7aa JB	45	}
99dde7aa JB	46
1e03690e JB	47	if (!argc)
1e03690e JB	48	return 0;
99dde7aa	49
1e03690e JB	50	if (strcmp(argv, "key") != 0 && strcmp(argv, "keys") != 0)
1e03690e JB	51	return 1;
ded1f078	52
1e03690e JB	53	argv++;
1e03690e JB	54	argc--;
51e9bd80	55
1e03690e	56	return parse_keys(msg, argv, argc);
99dde7aa JB	57	nla_put_failure:
	58	return -ENOSPC;
	59	}
	60
	61	static int disconnect(struct nl80211_state *state,
99dde7aa	62	struct nl_msg *msg,
05514f95 JB	63	int argc, char **argv,
05514f95 JB	64	enum id_input id)
99dde7aa JB	65	{
	66	return 0;
	67	}
	68	TOPLEVEL(disconnect, NULL,
	69	NL80211_CMD_DISCONNECT, 0, CIB_NETDEV, disconnect,
	70	"Disconnect from the current network.");
7fabd346	71
34b23014	72	static int iw_connect(struct nl80211_state *state,
05514f95 JB	73	struct nl_msg msg, int argc, char *argv,
05514f95 JB	74	enum id_input id)
7fabd346 JB	75	{
	76	char *conn_argv, dev = argv[0];
	77	static const __u32 cmds[] = {
	78	NL80211_CMD_CONNECT,
	79	};
	80	struct print_event_args printargs = { };
	81	int conn_argc, err;
	82	bool wait = false;
	83	int i;
	84
	85	/* strip "wlan0 connect" */
	86	argc -= 2;
	87	argv += 2;
	88
	89	/* check -w */
	90	if (argc && strcmp(argv[0], "-w") == 0) {
	91	wait = true;
	92	argc--;
	93	argv++;
	94	}
	95
d8d294c4 JL	96	err = __prepare_listen_events(state);
	97	if (err)
	98	return err;
	99
7fabd346 JB	100	conn_argc = 3 + argc;
	101	conn_argv = calloc(conn_argc, sizeof(*conn_argv));
	102	if (!conn_argv)
	103	return -ENOMEM;
ee374e4d	104
7fabd346	105	conn_argv[0] = dev;
4698bfc2	106	conn_argv[1] = "connect";
7fabd346 JB	107	conn_argv[2] = "establish";
	108	for (i = 0; i < argc; i++)
	109	conn_argv[i + 3] = argv[i];
75f4204c	110	err = handle_cmd(state, id, conn_argc, conn_argv);
7fabd346 JB	111	free(conn_argv);
	112	if (err)
	113	return err;
	114
	115	if (!wait)
	116	return 0;
	117
	118	/*
	119	* WARNING: DO NOT COPY THIS CODE INTO YOUR APPLICATION
	120	*
ee374e4d	121	* This code has a bug:
7fabd346	122	*
ee374e4d JB	123	* It is possible for a connect result message from another
	124	* connect attempt to be processed here first, because we
	125	* start listening to the multicast group before starting
	126	* our own connect request, which may succeed but we get a
	127	* fail message from a previous attempt that raced with us,
	128	* or similar.
7fabd346 JB	129	*
	130	* The only proper way to fix this would be to listen to events
	131	* before sending the command, and for the kernel to send the
ee374e4d JB	132	* connect request or a cookie along with the event, so that you
	133	* can match up whether the connect _you_ requested was finished
	134	* or aborted.
7fabd346 JB	135	*
	136	* Alas, the kernel doesn't do that (yet).
	137	*/
	138
ee374e4d	139	__do_listen_events(state, ARRAY_SIZE(cmds), cmds, &printargs);
7fabd346 JB	140	return 0;
	141	}
	142	TOPLEVEL(connect, "[-w] <SSID> [<freq in MHz>] [<bssid>] [key 0:abcde d:1:6162636465]",
	143	0, 0, CIB_NETDEV, iw_connect,
	144	"Join the network with the given SSID (and frequency, BSSID).\n"
	145	"With -w, wait for the connect to finish or fail.");
4698bfc2	146	HIDDEN(connect, establish, "", NL80211_CMD_CONNECT, 0, CIB_NETDEV, iw_conn);
5727fe03	147
34b23014	148	static int iw_auth(struct nl80211_state *state,
5727fe03 JB	149	struct nl_msg msg, int argc, char *argv,
	150	enum id_input id)
	151	{
	152	char *end;
	153	unsigned char bssid[6];
	154	int freq;
	155	bool need_key = false;
	156
	157	if (argc < 4)
	158	return 1;
	159
	160	/* SSID */
	161	NLA_PUT(msg, NL80211_ATTR_SSID, strlen(argv[0]), argv[0]);
	162	argv++;
	163	argc--;
	164
	165	/* bssid */
	166	if (mac_addr_a2n(bssid, argv[0]) == 0) {
	167	NLA_PUT(msg, NL80211_ATTR_MAC, 6, bssid);
	168	argv++;
	169	argc--;
	170	} else {
	171	return 1;
	172	}
	173
	174	/* FIXME */
	175	if (strcmp(argv[0], "open") == 0) {
	176	NLA_PUT_U32(msg, NL80211_ATTR_AUTH_TYPE,
	177	NL80211_AUTHTYPE_OPEN_SYSTEM);
	178	} else if (strcmp(argv[0], "shared") == 0) {
	179	NLA_PUT_U32(msg, NL80211_ATTR_AUTH_TYPE,
	180	NL80211_AUTHTYPE_SHARED_KEY);
	181	need_key = true;
	182	} else {
	183	return 1;
	184	}
	185	argv++;
	186	argc--;
	187
	188	freq = strtoul(argv[0], &end, 10);
	189	if (*end == '\0') {
	190	NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, freq);
	191	argv++;
	192	argc--;
	193	} else {
	194	return 1;
	195	}
	196
	197	if (!argc && need_key)
	198	return 1;
	199	if (argc && !need_key)
	200	return 1;
	201	if (!argc)
	202	return 0;
	203
	204	if (strcmp(argv, "key") != 0 && strcmp(argv, "keys") != 0)
	205	return 1;
	206
	207	argv++;
	208	argc--;
	209
	210	return parse_keys(msg, argv, argc);
	211	nla_put_failure:
	212	return -ENOSPC;
213	}
214
215	TOPLEVEL(auth, "<SSID> <bssid> <type:open\|shared> <freq in MHz> [key 0:abcde d:1:6162636465]",
216	NL80211_CMD_AUTHENTICATE, 0, CIB_NETDEV, iw_auth,
217	"Authenticate with the given network.\n");