udp: Merge udp_send and udp_output
[dragonfly.git] / contrib / wpa_supplicant / wpa_supplicant / ctrl_iface_udp.c
1 /*
2  * WPA Supplicant / UDP socket -based control interface
3  * Copyright (c) 2004-2005, Jouni Malinen <j@w1.fi>
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License version 2 as
7  * published by the Free Software Foundation.
8  *
9  * Alternatively, this software may be distributed under the terms of BSD
10  * license.
11  *
12  * See README and COPYING for more details.
13  */
14
15 #include "includes.h"
16
17 #include "common.h"
18 #include "eloop.h"
19 #include "config.h"
20 #include "eapol_supp/eapol_supp_sm.h"
21 #include "wpa_supplicant_i.h"
22 #include "ctrl_iface.h"
23 #include "wpa_ctrl.h"
24
25
26 #define COOKIE_LEN 8
27
28 /* Per-interface ctrl_iface */
29
30 /**
31  * struct wpa_ctrl_dst - Internal data structure of control interface monitors
32  *
33  * This structure is used to store information about registered control
34  * interface monitors into struct wpa_supplicant. This data is private to
35  * ctrl_iface_udp.c and should not be touched directly from other files.
36  */
37 struct wpa_ctrl_dst {
38         struct wpa_ctrl_dst *next;
39         struct sockaddr_in addr;
40         socklen_t addrlen;
41         int debug_level;
42         int errors;
43 };
44
45
46 struct ctrl_iface_priv {
47         struct wpa_supplicant *wpa_s;
48         int sock;
49         struct wpa_ctrl_dst *ctrl_dst;
50         u8 cookie[COOKIE_LEN];
51 };
52
53
54 static void wpa_supplicant_ctrl_iface_send(struct ctrl_iface_priv *priv,
55                                            int level, const char *buf,
56                                            size_t len);
57
58
59 static int wpa_supplicant_ctrl_iface_attach(struct ctrl_iface_priv *priv,
60                                             struct sockaddr_in *from,
61                                             socklen_t fromlen)
62 {
63         struct wpa_ctrl_dst *dst;
64
65         dst = os_zalloc(sizeof(*dst));
66         if (dst == NULL)
67                 return -1;
68         os_memcpy(&dst->addr, from, sizeof(struct sockaddr_in));
69         dst->addrlen = fromlen;
70         dst->debug_level = MSG_INFO;
71         dst->next = priv->ctrl_dst;
72         priv->ctrl_dst = dst;
73         wpa_printf(MSG_DEBUG, "CTRL_IFACE monitor attached %s:%d",
74                    inet_ntoa(from->sin_addr), ntohs(from->sin_port));
75         return 0;
76 }
77
78
79 static int wpa_supplicant_ctrl_iface_detach(struct ctrl_iface_priv *priv,
80                                             struct sockaddr_in *from,
81                                             socklen_t fromlen)
82 {
83         struct wpa_ctrl_dst *dst, *prev = NULL;
84
85         dst = priv->ctrl_dst;
86         while (dst) {
87                 if (from->sin_addr.s_addr == dst->addr.sin_addr.s_addr &&
88                     from->sin_port == dst->addr.sin_port) {
89                         if (prev == NULL)
90                                 priv->ctrl_dst = dst->next;
91                         else
92                                 prev->next = dst->next;
93                         os_free(dst);
94                         wpa_printf(MSG_DEBUG, "CTRL_IFACE monitor detached "
95                                    "%s:%d", inet_ntoa(from->sin_addr),
96                                    ntohs(from->sin_port));
97                         return 0;
98                 }
99                 prev = dst;
100                 dst = dst->next;
101         }
102         return -1;
103 }
104
105
106 static int wpa_supplicant_ctrl_iface_level(struct ctrl_iface_priv *priv,
107                                            struct sockaddr_in *from,
108                                            socklen_t fromlen,
109                                            char *level)
110 {
111         struct wpa_ctrl_dst *dst;
112
113         wpa_printf(MSG_DEBUG, "CTRL_IFACE LEVEL %s", level);
114
115         dst = priv->ctrl_dst;
116         while (dst) {
117                 if (from->sin_addr.s_addr == dst->addr.sin_addr.s_addr &&
118                     from->sin_port == dst->addr.sin_port) {
119                         wpa_printf(MSG_DEBUG, "CTRL_IFACE changed monitor "
120                                    "level %s:%d", inet_ntoa(from->sin_addr),
121                                    ntohs(from->sin_port));
122                         dst->debug_level = atoi(level);
123                         return 0;
124                 }
125                 dst = dst->next;
126         }
127
128         return -1;
129 }
130
131
132 static char *
133 wpa_supplicant_ctrl_iface_get_cookie(struct ctrl_iface_priv *priv,
134                                      size_t *reply_len)
135 {
136         char *reply;
137         reply = os_malloc(7 + 2 * COOKIE_LEN + 1);
138         if (reply == NULL) {
139                 *reply_len = 1;
140                 return NULL;
141         }
142
143         os_memcpy(reply, "COOKIE=", 7);
144         wpa_snprintf_hex(reply + 7, 2 * COOKIE_LEN + 1,
145                          priv->cookie, COOKIE_LEN);
146
147         *reply_len = 7 + 2 * COOKIE_LEN;
148         return reply;
149 }
150
151
152 static void wpa_supplicant_ctrl_iface_receive(int sock, void *eloop_ctx,
153                                               void *sock_ctx)
154 {
155         struct wpa_supplicant *wpa_s = eloop_ctx;
156         struct ctrl_iface_priv *priv = sock_ctx;
157         char buf[256], *pos;
158         int res;
159         struct sockaddr_in from;
160         socklen_t fromlen = sizeof(from);
161         char *reply = NULL;
162         size_t reply_len = 0;
163         int new_attached = 0;
164         u8 cookie[COOKIE_LEN];
165
166         res = recvfrom(sock, buf, sizeof(buf) - 1, 0,
167                        (struct sockaddr *) &from, &fromlen);
168         if (res < 0) {
169                 perror("recvfrom(ctrl_iface)");
170                 return;
171         }
172         if (from.sin_addr.s_addr != htonl((127 << 24) | 1)) {
173                 /*
174                  * The OS networking stack is expected to drop this kind of
175                  * frames since the socket is bound to only localhost address.
176                  * Just in case, drop the frame if it is coming from any other
177                  * address.
178                  */
179                 wpa_printf(MSG_DEBUG, "CTRL: Drop packet from unexpected "
180                            "source %s", inet_ntoa(from.sin_addr));
181                 return;
182         }
183         buf[res] = '\0';
184
185         if (os_strcmp(buf, "GET_COOKIE") == 0) {
186                 reply = wpa_supplicant_ctrl_iface_get_cookie(priv, &reply_len);
187                 goto done;
188         }
189
190         /*
191          * Require that the client includes a prefix with the 'cookie' value
192          * fetched with GET_COOKIE command. This is used to verify that the
193          * client has access to a bidirectional link over UDP in order to
194          * avoid attacks using forged localhost IP address even if the OS does
195          * not block such frames from remote destinations.
196          */
197         if (os_strncmp(buf, "COOKIE=", 7) != 0) {
198                 wpa_printf(MSG_DEBUG, "CTLR: No cookie in the request - "
199                            "drop request");
200                 return;
201         }
202
203         if (hexstr2bin(buf + 7, cookie, COOKIE_LEN) < 0) {
204                 wpa_printf(MSG_DEBUG, "CTLR: Invalid cookie format in the "
205                            "request - drop request");
206                 return;
207         }
208
209         if (os_memcmp(cookie, priv->cookie, COOKIE_LEN) != 0) {
210                 wpa_printf(MSG_DEBUG, "CTLR: Invalid cookie in the request - "
211                            "drop request");
212                 return;
213         }
214
215         pos = buf + 7 + 2 * COOKIE_LEN;
216         while (*pos == ' ')
217                 pos++;
218
219         if (os_strcmp(pos, "ATTACH") == 0) {
220                 if (wpa_supplicant_ctrl_iface_attach(priv, &from, fromlen))
221                         reply_len = 1;
222                 else {
223                         new_attached = 1;
224                         reply_len = 2;
225                 }
226         } else if (os_strcmp(pos, "DETACH") == 0) {
227                 if (wpa_supplicant_ctrl_iface_detach(priv, &from, fromlen))
228                         reply_len = 1;
229                 else
230                         reply_len = 2;
231         } else if (os_strncmp(pos, "LEVEL ", 6) == 0) {
232                 if (wpa_supplicant_ctrl_iface_level(priv, &from, fromlen,
233                                                     pos + 6))
234                         reply_len = 1;
235                 else
236                         reply_len = 2;
237         } else {
238                 reply = wpa_supplicant_ctrl_iface_process(wpa_s, pos,
239                                                           &reply_len);
240         }
241
242  done:
243         if (reply) {
244                 sendto(sock, reply, reply_len, 0, (struct sockaddr *) &from,
245                        fromlen);
246                 os_free(reply);
247         } else if (reply_len == 1) {
248                 sendto(sock, "FAIL\n", 5, 0, (struct sockaddr *) &from,
249                        fromlen);
250         } else if (reply_len == 2) {
251                 sendto(sock, "OK\n", 3, 0, (struct sockaddr *) &from,
252                        fromlen);
253         }
254
255         if (new_attached)
256                 eapol_sm_notify_ctrl_attached(wpa_s->eapol);
257 }
258
259
260 static void wpa_supplicant_ctrl_iface_msg_cb(void *ctx, int level,
261                                              const char *txt, size_t len)
262 {
263         struct wpa_supplicant *wpa_s = ctx;
264         if (wpa_s == NULL || wpa_s->ctrl_iface == NULL)
265                 return;
266         wpa_supplicant_ctrl_iface_send(wpa_s->ctrl_iface, level, txt, len);
267 }
268
269
270 struct ctrl_iface_priv *
271 wpa_supplicant_ctrl_iface_init(struct wpa_supplicant *wpa_s)
272 {
273         struct ctrl_iface_priv *priv;
274         struct sockaddr_in addr;
275
276         priv = os_zalloc(sizeof(*priv));
277         if (priv == NULL)
278                 return NULL;
279         priv->wpa_s = wpa_s;
280         priv->sock = -1;
281         os_get_random(priv->cookie, COOKIE_LEN);
282
283         if (wpa_s->conf->ctrl_interface == NULL)
284                 return priv;
285
286         priv->sock = socket(PF_INET, SOCK_DGRAM, 0);
287         if (priv->sock < 0) {
288                 perror("socket(PF_INET)");
289                 goto fail;
290         }
291
292         os_memset(&addr, 0, sizeof(addr));
293         addr.sin_family = AF_INET;
294         addr.sin_addr.s_addr = htonl((127 << 24) | 1);
295         addr.sin_port = htons(WPA_CTRL_IFACE_PORT);
296         if (bind(priv->sock, (struct sockaddr *) &addr, sizeof(addr)) < 0) {
297                 perror("bind(AF_INET)");
298                 goto fail;
299         }
300
301         eloop_register_read_sock(priv->sock, wpa_supplicant_ctrl_iface_receive,
302                                  wpa_s, priv);
303         wpa_msg_register_cb(wpa_supplicant_ctrl_iface_msg_cb);
304
305         return priv;
306
307 fail:
308         if (priv->sock >= 0)
309                 close(priv->sock);
310         os_free(priv);
311         return NULL;
312 }
313
314
315 void wpa_supplicant_ctrl_iface_deinit(struct ctrl_iface_priv *priv)
316 {
317         struct wpa_ctrl_dst *dst, *prev;
318
319         if (priv->sock > -1) {
320                 eloop_unregister_read_sock(priv->sock);
321                 if (priv->ctrl_dst) {
322                         /*
323                          * Wait a second before closing the control socket if
324                          * there are any attached monitors in order to allow
325                          * them to receive any pending messages.
326                          */
327                         wpa_printf(MSG_DEBUG, "CTRL_IFACE wait for attached "
328                                    "monitors to receive messages");
329                         os_sleep(1, 0);
330                 }
331                 close(priv->sock);
332                 priv->sock = -1;
333         }
334
335         dst = priv->ctrl_dst;
336         while (dst) {
337                 prev = dst;
338                 dst = dst->next;
339                 os_free(prev);
340         }
341         os_free(priv);
342 }
343
344
345 static void wpa_supplicant_ctrl_iface_send(struct ctrl_iface_priv *priv,
346                                            int level, const char *buf,
347                                            size_t len)
348 {
349         struct wpa_ctrl_dst *dst, *next;
350         char levelstr[10];
351         int idx;
352         char *sbuf;
353         int llen;
354
355         dst = priv->ctrl_dst;
356         if (priv->sock < 0 || dst == NULL)
357                 return;
358
359         os_snprintf(levelstr, sizeof(levelstr), "<%d>", level);
360
361         llen = os_strlen(levelstr);
362         sbuf = os_malloc(llen + len);
363         if (sbuf == NULL)
364                 return;
365
366         os_memcpy(sbuf, levelstr, llen);
367         os_memcpy(sbuf + llen, buf, len);
368
369         idx = 0;
370         while (dst) {
371                 next = dst->next;
372                 if (level >= dst->debug_level) {
373                         wpa_printf(MSG_DEBUG, "CTRL_IFACE monitor send %s:%d",
374                                    inet_ntoa(dst->addr.sin_addr),
375                                    ntohs(dst->addr.sin_port));
376                         if (sendto(priv->sock, sbuf, llen + len, 0,
377                                    (struct sockaddr *) &dst->addr,
378                                    sizeof(dst->addr)) < 0) {
379                                 perror("sendto(CTRL_IFACE monitor)");
380                                 dst->errors++;
381                                 if (dst->errors > 10) {
382                                         wpa_supplicant_ctrl_iface_detach(
383                                                 priv, &dst->addr,
384                                                 dst->addrlen);
385                                 }
386                         } else
387                                 dst->errors = 0;
388                 }
389                 idx++;
390                 dst = next;
391         }
392         os_free(sbuf);
393 }
394
395
396 void wpa_supplicant_ctrl_iface_wait(struct ctrl_iface_priv *priv)
397 {
398         wpa_printf(MSG_DEBUG, "CTRL_IFACE - %s - wait for monitor",
399                    priv->wpa_s->ifname);
400         eloop_wait_for_read_sock(priv->sock);
401 }
402
403
404 /* Global ctrl_iface */
405
406 struct ctrl_iface_global_priv {
407         int sock;
408         u8 cookie[COOKIE_LEN];
409 };
410
411
412 static char *
413 wpa_supplicant_global_get_cookie(struct ctrl_iface_global_priv *priv,
414                                  size_t *reply_len)
415 {
416         char *reply;
417         reply = os_malloc(7 + 2 * COOKIE_LEN + 1);
418         if (reply == NULL) {
419                 *reply_len = 1;
420                 return NULL;
421         }
422
423         os_memcpy(reply, "COOKIE=", 7);
424         wpa_snprintf_hex(reply + 7, 2 * COOKIE_LEN + 1,
425                          priv->cookie, COOKIE_LEN);
426
427         *reply_len = 7 + 2 * COOKIE_LEN;
428         return reply;
429 }
430
431
432 static void wpa_supplicant_global_ctrl_iface_receive(int sock, void *eloop_ctx,
433                                                      void *sock_ctx)
434 {
435         struct wpa_global *global = eloop_ctx;
436         struct ctrl_iface_global_priv *priv = sock_ctx;
437         char buf[256], *pos;
438         int res;
439         struct sockaddr_in from;
440         socklen_t fromlen = sizeof(from);
441         char *reply;
442         size_t reply_len;
443         u8 cookie[COOKIE_LEN];
444
445         res = recvfrom(sock, buf, sizeof(buf) - 1, 0,
446                        (struct sockaddr *) &from, &fromlen);
447         if (res < 0) {
448                 perror("recvfrom(ctrl_iface)");
449                 return;
450         }
451         if (from.sin_addr.s_addr != htonl((127 << 24) | 1)) {
452                 /*
453                  * The OS networking stack is expected to drop this kind of
454                  * frames since the socket is bound to only localhost address.
455                  * Just in case, drop the frame if it is coming from any other
456                  * address.
457                  */
458                 wpa_printf(MSG_DEBUG, "CTRL: Drop packet from unexpected "
459                            "source %s", inet_ntoa(from.sin_addr));
460                 return;
461         }
462         buf[res] = '\0';
463
464         if (os_strcmp(buf, "GET_COOKIE") == 0) {
465                 reply = wpa_supplicant_global_get_cookie(priv, &reply_len);
466                 goto done;
467         }
468
469         if (os_strncmp(buf, "COOKIE=", 7) != 0) {
470                 wpa_printf(MSG_DEBUG, "CTLR: No cookie in the request - "
471                            "drop request");
472                 return;
473         }
474
475         if (hexstr2bin(buf + 7, cookie, COOKIE_LEN) < 0) {
476                 wpa_printf(MSG_DEBUG, "CTLR: Invalid cookie format in the "
477                            "request - drop request");
478                 return;
479         }
480
481         if (os_memcmp(cookie, priv->cookie, COOKIE_LEN) != 0) {
482                 wpa_printf(MSG_DEBUG, "CTLR: Invalid cookie in the request - "
483                            "drop request");
484                 return;
485         }
486
487         pos = buf + 7 + 2 * COOKIE_LEN;
488         while (*pos == ' ')
489                 pos++;
490
491         reply = wpa_supplicant_global_ctrl_iface_process(global, pos,
492                                                          &reply_len);
493
494  done:
495         if (reply) {
496                 sendto(sock, reply, reply_len, 0, (struct sockaddr *) &from,
497                        fromlen);
498                 os_free(reply);
499         } else if (reply_len) {
500                 sendto(sock, "FAIL\n", 5, 0, (struct sockaddr *) &from,
501                        fromlen);
502         }
503 }
504
505
506 struct ctrl_iface_global_priv *
507 wpa_supplicant_global_ctrl_iface_init(struct wpa_global *global)
508 {
509         struct ctrl_iface_global_priv *priv;
510         struct sockaddr_in addr;
511
512         priv = os_zalloc(sizeof(*priv));
513         if (priv == NULL)
514                 return NULL;
515         priv->sock = -1;
516         os_get_random(priv->cookie, COOKIE_LEN);
517
518         if (global->params.ctrl_interface == NULL)
519                 return priv;
520
521         wpa_printf(MSG_DEBUG, "Global control interface '%s'",
522                    global->params.ctrl_interface);
523
524         priv->sock = socket(PF_INET, SOCK_DGRAM, 0);
525         if (priv->sock < 0) {
526                 perror("socket(PF_INET)");
527                 goto fail;
528         }
529
530         os_memset(&addr, 0, sizeof(addr));
531         addr.sin_family = AF_INET;
532         addr.sin_addr.s_addr = htonl((127 << 24) | 1);
533         addr.sin_port = htons(WPA_GLOBAL_CTRL_IFACE_PORT);
534         if (bind(priv->sock, (struct sockaddr *) &addr, sizeof(addr)) < 0) {
535                 perror("bind(AF_INET)");
536                 goto fail;
537         }
538
539         eloop_register_read_sock(priv->sock,
540                                  wpa_supplicant_global_ctrl_iface_receive,
541                                  global, priv);
542
543         return priv;
544
545 fail:
546         if (priv->sock >= 0)
547                 close(priv->sock);
548         os_free(priv);
549         return NULL;
550 }
551
552
553 void
554 wpa_supplicant_global_ctrl_iface_deinit(struct ctrl_iface_global_priv *priv)
555 {
556         if (priv->sock >= 0) {
557                 eloop_unregister_read_sock(priv->sock);
558                 close(priv->sock);
559         }
560         os_free(priv);
561 }