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