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