Trigger update process after scan results
[connman] / plugins / wifi.c
1 /*
2  *
3  *  Connection Manager
4  *
5  *  Copyright (C) 2007-2008  Intel Corporation. All rights reserved.
6  *
7  *  This program is free software; you can redistribute it and/or modify
8  *  it under the terms of the GNU General Public License version 2 as
9  *  published by the Free Software Foundation.
10  *
11  *  This program is distributed in the hope that it will be useful,
12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  *  GNU General Public License for more details.
15  *
16  *  You should have received a copy of the GNU General Public License
17  *  along with this program; if not, write to the Free Software
18  *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
19  *
20  */
21
22 #ifdef HAVE_CONFIG_H
23 #include <config.h>
24 #endif
25
26 #include <stdio.h>
27 #include <unistd.h>
28 #include <string.h>
29 #include <sys/ioctl.h>
30 #include <sys/socket.h>
31 #include <linux/if_arp.h>
32 #include <linux/wireless.h>
33
34 #include <gdbus.h>
35
36 #include <connman/plugin.h>
37 #include <connman/driver.h>
38 #include <connman/rtnl.h>
39 #include <connman/log.h>
40
41 #include "inet.h"
42 #include "supplicant.h"
43
44 #define CLEANUP_TIMEOUT   8     /* in seconds */
45 #define INACTIVE_TIMEOUT  12    /* in seconds */
46
47 struct wifi_data {
48         GSList *current;
49         GSList *pending;
50         guint cleanup_timer;
51         guint inactive_timer;
52         gchar *identifier;
53         gboolean connected;
54 };
55
56 static int network_probe(struct connman_element *element)
57 {
58         DBG("element %p name %s", element, element->name);
59
60         return 0;
61 }
62
63 static void network_remove(struct connman_element *element)
64 {
65         DBG("element %p name %s", element, element->name);
66 }
67
68 static int network_enable(struct connman_element *element)
69 {
70         struct connman_element *device = element->parent;
71         char *name, *security = NULL, *passphrase = NULL;
72         unsigned char *ssid;
73         int ssid_len;
74
75         DBG("element %p name %s", element, element->name);
76
77         if (connman_element_get_static_property(element,
78                                                 "Name", &name) == FALSE)
79                 return -EIO;
80
81         if (connman_element_get_static_array_property(element,
82                                 "WiFi.SSID", &ssid, &ssid_len) == FALSE)
83                 return -EIO;
84
85         if (device != NULL) {
86                 struct wifi_data *data = connman_element_get_data(device);
87
88                 if (data != NULL) {
89                         if (data->connected == TRUE)
90                                 return -EBUSY;
91
92                         g_free(data->identifier);
93                         data->identifier = g_strdup(name);
94                 }
95         }
96
97         connman_element_get_value(element,
98                         CONNMAN_PROPERTY_ID_WIFI_SECURITY, &security);
99
100         connman_element_get_value(element,
101                         CONNMAN_PROPERTY_ID_WIFI_PASSPHRASE, &passphrase);
102
103         DBG("name %s security %s passhprase %s",
104                                         name, security, passphrase);
105
106         if (__supplicant_connect(element, ssid, ssid_len,
107                                                 security, passphrase) < 0)
108                 connman_error("Failed to initiate connect");
109
110         return 0;
111 }
112
113 static int network_disable(struct connman_element *element)
114 {
115         DBG("element %p name %s", element, element->name);
116
117         connman_element_unregister_children(element);
118
119         __supplicant_disconnect(element);
120
121         return 0;
122 }
123
124 static struct connman_driver network_driver = {
125         .name           = "wifi-network",
126         .type           = CONNMAN_ELEMENT_TYPE_NETWORK,
127         .subtype        = CONNMAN_ELEMENT_SUBTYPE_WIFI,
128         .probe          = network_probe,
129         .remove         = network_remove,
130         .enable         = network_enable,
131         .disable        = network_disable,
132 };
133
134 static struct connman_element *find_current_element(struct wifi_data *data,
135                                                         const char *identifier)
136 {
137         GSList *list;
138
139         for (list = data->current; list; list = list->next) {
140                 struct connman_element *element = list->data;
141
142                 if (connman_element_match_static_property(element,
143                                         "Name", &identifier) == TRUE)
144                         return element;
145         }
146
147         return NULL;
148 }
149
150 static struct connman_element *find_pending_element(struct wifi_data *data,
151                                                         const char *identifier)
152 {
153         GSList *list;
154
155         for (list = data->pending; list; list = list->next) {
156                 struct connman_element *element = list->data;
157
158                 if (connman_element_match_static_property(element,
159                                         "Name", &identifier) == TRUE)
160                         return element;
161         }
162
163         return NULL;
164 }
165
166 static gboolean inactive_scan(gpointer user_data)
167 {
168         struct connman_element *device = user_data;
169         struct wifi_data *data = connman_element_get_data(device);
170
171         DBG("");
172
173         if (data->cleanup_timer > 0) {
174                 g_source_remove(data->cleanup_timer);
175                 data->cleanup_timer = 0;
176         }
177
178         __supplicant_scan(device);
179
180         data->inactive_timer = 0;
181
182         return FALSE;
183 }
184
185 static void connect_known_networks(struct connman_element *device)
186 {
187         struct wifi_data *data = connman_element_get_data(device);
188         GSList *list;
189
190         DBG("");
191
192         if (data->inactive_timer > 0) {
193                 g_source_remove(data->inactive_timer);
194                 data->inactive_timer = 0;
195         }
196
197         for (list = data->current; list; list = list->next) {
198                 struct connman_element *element = list->data;
199
200                 if (element->policy == CONNMAN_ELEMENT_POLICY_AUTO &&
201                                                 element->remember == TRUE &&
202                                                 element->available == TRUE) {
203                         if (network_enable(element) == 0)
204                                 return;
205                 }
206         }
207
208         data->inactive_timer = g_timeout_add_seconds(INACTIVE_TIMEOUT,
209                                                         inactive_scan, device);
210 }
211
212 static void state_change(struct connman_element *device,
213                                                 enum supplicant_state state)
214 {
215         struct wifi_data *data = connman_element_get_data(device);
216         struct connman_element *element;
217
218         DBG("state %d", state);
219
220         if (data == NULL)
221                 return;
222
223         if (data->identifier == NULL)
224                 goto reconnect;
225
226         element = find_current_element(data, data->identifier);
227         if (element == NULL)
228                 goto reconnect;
229
230         if (state == STATE_COMPLETED) {
231                 struct connman_element *dhcp;
232
233                 data->connected = TRUE;
234                 connman_element_set_enabled(element, TRUE);
235
236                 dhcp = connman_element_create(NULL);
237
238                 dhcp->type = CONNMAN_ELEMENT_TYPE_DHCP;
239                 dhcp->index = element->index;
240
241                 connman_element_register(dhcp, element);
242         } else if (state == STATE_INACTIVE || state == STATE_DISCONNECTED) {
243                 data->connected = FALSE;
244                 connman_element_set_enabled(element, FALSE);
245
246                 connman_element_unregister_children(element);
247         }
248
249 reconnect:
250         if (state == STATE_INACTIVE || state == STATE_DISCONNECTED) {
251                 data->connected = FALSE;
252                 connect_known_networks(device);
253         }
254 }
255
256 static gboolean cleanup_pending(gpointer user_data)
257 {
258         struct wifi_data *data = user_data;
259         GSList *list;
260
261         DBG("");
262
263         for (list = data->pending; list; list = list->next) {
264                 struct connman_element *element = list->data;
265
266                 DBG("element %p name %s", element, element->name);
267
268                 connman_element_unregister(element);
269                 connman_element_unref(element);
270         }
271
272         g_slist_free(data->pending);
273         data->pending = NULL;
274
275         data->cleanup_timer = 0;
276
277         return FALSE;
278 }
279
280 static void clear_results(struct connman_element *device)
281 {
282         struct wifi_data *data = connman_element_get_data(device);
283
284         DBG("pending %d", g_slist_length(data->pending));
285         DBG("current %d", g_slist_length(data->current));
286
287         data->pending = data->current;
288         data->current = NULL;
289
290         if (data->cleanup_timer > 0)
291                 return;
292
293         data->cleanup_timer = g_timeout_add_seconds(CLEANUP_TIMEOUT,
294                                                         cleanup_pending, data);
295 }
296
297 static void scan_result(struct connman_element *device,
298                                         struct supplicant_network *network)
299 {
300         struct wifi_data *data = connman_element_get_data(device);
301         struct connman_element *element;
302         gchar *temp;
303         int i;
304
305         DBG("network %p identifier %s", network, network->identifier);
306
307         if (data == NULL)
308                 return;
309
310         if (network->identifier == NULL)
311                 return;
312
313         if (network->identifier[0] == '\0')
314                 return;
315
316         temp = g_strdup(network->identifier);
317
318         for (i = 0; i < strlen(temp); i++) {
319                 gchar tmp = g_ascii_tolower(temp[i]);
320
321                 if (tmp < 'a' || tmp > 'z')
322                         temp[i] = '_';
323         }
324
325         element = find_pending_element(data, network->identifier);
326         if (element == NULL) {
327                 element = connman_element_create(temp);
328
329                 element->type = CONNMAN_ELEMENT_TYPE_NETWORK;
330                 element->index = device->index;
331
332                 connman_element_add_static_property(element, "Name",
333                                 DBUS_TYPE_STRING, &network->identifier);
334
335                 connman_element_add_static_array_property(element, "WiFi.SSID",
336                         DBUS_TYPE_BYTE, &network->ssid, network->ssid_len);
337
338                 if (element->wifi.security == NULL) {
339                         const char *security;
340
341                         if (network->has_rsn == TRUE)
342                                 security = "wpa2";
343                         else if (network->has_wpa == TRUE)
344                                 security = "wpa";
345                         else if (network->has_wep == TRUE)
346                                 security = "wep";
347                         else
348                                 security = "none";
349
350                         element->wifi.security = g_strdup(security);
351                 }
352
353                 element->strength = network->quality;
354
355                 connman_element_add_static_property(element, "Strength",
356                                         DBUS_TYPE_BYTE, &element->strength);
357
358                 DBG("%s (%s) strength %d", network->identifier,
359                                 element->wifi.security, element->strength);
360
361                 connman_element_register(element, device);
362         } else {
363                 data->pending = g_slist_remove(data->pending, element);
364
365                 element->strength = network->quality;
366
367                 connman_element_set_static_property(element, "Strength",
368                                         DBUS_TYPE_BYTE, &element->strength);
369
370                 connman_element_update(element);
371         }
372
373         data->current = g_slist_append(data->current, element);
374
375         element->available = TRUE;
376
377         g_free(temp);
378 }
379
380 static struct supplicant_callback wifi_callback = {
381         .state_change   = state_change,
382         .clear_results  = clear_results,
383         .scan_result    = scan_result,
384 };
385
386 static int wifi_probe(struct connman_element *element)
387 {
388         struct wifi_data *data;
389
390         DBG("element %p name %s", element, element->name);
391
392         data = g_try_new0(struct wifi_data, 1);
393         if (data == NULL)
394                 return -ENOMEM;
395
396         data->connected = FALSE;
397
398         connman_element_set_data(element, data);
399
400         return 0;
401 }
402
403 static void wifi_remove(struct connman_element *element)
404 {
405         struct wifi_data *data = connman_element_get_data(element);
406
407         DBG("element %p name %s", element, element->name);
408
409         connman_element_set_data(element, NULL);
410
411         g_free(data->identifier);
412         g_free(data);
413 }
414
415 static int wifi_update(struct connman_element *element)
416 {
417         DBG("element %p name %s", element, element->name);
418
419         __supplicant_scan(element);
420
421         return 0;
422 }
423
424 static int wifi_enable(struct connman_element *element)
425 {
426         int err;
427
428         DBG("element %p name %s", element, element->name);
429
430         err = __supplicant_start(element, &wifi_callback);
431         if (err < 0)
432                 return err;
433
434         __supplicant_scan(element);
435
436         return 0;
437 }
438
439 static int wifi_disable(struct connman_element *element)
440 {
441         struct wifi_data *data = connman_element_get_data(element);
442         GSList *list;
443
444         DBG("element %p name %s", element, element->name);
445
446         if (data->cleanup_timer > 0) {
447                 g_source_remove(data->cleanup_timer);
448                 data->cleanup_timer = 0;
449         }
450
451         if (data->inactive_timer > 0) {
452                 g_source_remove(data->inactive_timer);
453                 data->inactive_timer = 0;
454         }
455
456         __supplicant_disconnect(element);
457
458         for (list = data->pending; list; list = list->next) {
459                 struct connman_element *network = list->data;
460
461                 connman_element_unref(network);
462         }
463
464         g_slist_free(data->pending);
465         data->pending = NULL;
466
467         for (list = data->current; list; list = list->next) {
468                 struct connman_element *network = list->data;
469
470                 connman_element_unref(network);
471         }
472
473         g_slist_free(data->current);
474         data->current = NULL;
475
476         connman_element_unregister_children(element);
477
478         __supplicant_stop(element);
479
480         return 0;
481 }
482
483 static struct connman_driver wifi_driver = {
484         .name           = "wifi-device",
485         .type           = CONNMAN_ELEMENT_TYPE_DEVICE,
486         .subtype        = CONNMAN_ELEMENT_SUBTYPE_WIFI,
487         .probe          = wifi_probe,
488         .remove         = wifi_remove,
489         .update         = wifi_update,
490         .enable         = wifi_enable,
491         .disable        = wifi_disable,
492 };
493
494 static GSList *device_list = NULL;
495
496 static void wifi_newlink(unsigned short type, int index,
497                                         unsigned flags, unsigned change)
498 {
499         struct connman_element *device;
500         GSList *list;
501         gboolean exists = FALSE;
502         gchar *name, *devname;
503         struct iwreq iwr;
504         int sk;
505
506         DBG("index %d", index);
507
508         if (type != ARPHRD_ETHER)
509                 return;
510
511         name = inet_index2ident(index, "dev_");
512         devname = inet_index2name(index);
513
514         memset(&iwr, 0, sizeof(iwr));
515         strncpy(iwr.ifr_ifrn.ifrn_name, devname, IFNAMSIZ);
516
517         sk = socket(PF_INET, SOCK_DGRAM, 0);
518
519         if (ioctl(sk, SIOCGIWNAME, &iwr) < 0) {
520                 g_free(name);
521                 close(sk);
522                 return;
523         }
524
525         close(sk);
526
527         for (list = device_list; list; list = list->next) {
528                 struct connman_element *device = list->data;
529
530                 if (device->index == index) {
531                         exists = TRUE;
532                         break;
533                 }
534         }
535
536         if (exists == TRUE) {
537                 g_free(name);
538                 return;
539         }
540
541         device = connman_element_create(NULL);
542         device->type = CONNMAN_ELEMENT_TYPE_DEVICE;
543         device->subtype = CONNMAN_ELEMENT_SUBTYPE_WIFI;
544
545         device->index = index;
546         device->name = name;
547         device->devname = devname;
548
549         connman_element_register(device, NULL);
550         device_list = g_slist_append(device_list, device);
551 }
552
553 static void wifi_dellink(unsigned short type, int index,
554                                         unsigned flags, unsigned change)
555 {
556         GSList *list;
557
558         DBG("index %d", index);
559
560         for (list = device_list; list; list = list->next) {
561                 struct connman_element *device = list->data;
562
563                 if (device->index == index) {
564                         device_list = g_slist_remove(device_list, device);
565                         connman_element_unregister(device);
566                         connman_element_unref(device);
567                         break;
568                 }
569         }
570 }
571
572 static struct connman_rtnl wifi_rtnl = {
573         .name           = "wifi",
574         .newlink        = wifi_newlink,
575         .dellink        = wifi_dellink,
576 };
577
578 static void supplicant_connect(DBusConnection *connection, void *user_data)
579 {
580         DBG("connection %p", connection);
581
582         __supplicant_init(connection);
583
584         if (connman_rtnl_register(&wifi_rtnl) < 0)
585                 return;
586
587         connman_rtnl_send_getlink();
588 }
589
590 static void supplicant_disconnect(DBusConnection *connection, void *user_data)
591 {
592         GSList *list;
593
594         DBG("connection %p", connection);
595
596         connman_rtnl_unregister(&wifi_rtnl);
597
598         for (list = device_list; list; list = list->next) {
599                 struct connman_element *device = list->data;
600
601                 connman_element_unregister(device);
602                 connman_element_unref(device);
603         }
604
605         g_slist_free(device_list);
606         device_list = NULL;
607
608         __supplicant_exit();
609 }
610
611 static DBusConnection *connection;
612 static guint watch;
613
614 static int wifi_init(void)
615 {
616         int err;
617
618         connection = dbus_bus_get(DBUS_BUS_SYSTEM, NULL);
619         if (connection == NULL)
620                 return -EIO;
621
622         err = connman_driver_register(&network_driver);
623         if (err < 0) {
624                 dbus_connection_unref(connection);
625                 return err;
626         }
627
628         err = connman_driver_register(&wifi_driver);
629         if (err < 0) {
630                 connman_driver_unregister(&network_driver);
631                 dbus_connection_unref(connection);
632                 return err;
633         }
634
635         watch = g_dbus_add_service_watch(connection, SUPPLICANT_NAME,
636                         supplicant_connect, supplicant_disconnect, NULL, NULL);
637
638         if (g_dbus_check_service(connection, SUPPLICANT_NAME) == TRUE)
639                 supplicant_connect(connection, NULL);
640
641         return 0;
642 }
643
644 static void wifi_exit(void)
645 {
646         connman_driver_unregister(&network_driver);
647         connman_driver_unregister(&wifi_driver);
648
649         if (watch > 0)
650                 g_dbus_remove_watch(connection, watch);
651
652         supplicant_disconnect(connection, NULL);
653
654         dbus_connection_unref(connection);
655 }
656
657 CONNMAN_PLUGIN_DEFINE(wifi, "WiFi interface plugin", VERSION,
658                                                         wifi_init, wifi_exit)