Track global connection state
[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 struct wifi_data {
45         GSList *list;
46         gchar *identifier;
47         gboolean connected;
48 };
49
50 static int network_probe(struct connman_element *element)
51 {
52         DBG("element %p name %s", element, element->name);
53
54         return 0;
55 }
56
57 static void network_remove(struct connman_element *element)
58 {
59         DBG("element %p name %s", element, element->name);
60 }
61
62 static int network_enable(struct connman_element *element)
63 {
64         char *identifier, *security = NULL, *passphrase = NULL;
65         unsigned char *ssid;
66         int ssid_len;
67
68         DBG("element %p name %s", element, element->name);
69
70         if (connman_element_get_static_property(element,
71                                         "WiFi.Name", &identifier) == FALSE)
72                 return -EIO;
73
74         if (connman_element_get_static_array_property(element,
75                                 "WiFi.SSID", &ssid, &ssid_len) == FALSE)
76                 return -EIO;
77
78         if (element->parent != NULL) {
79                 struct wifi_data *data = connman_element_get_data(element->parent);
80
81                 if (data->connected == TRUE)
82                         return -EBUSY;
83
84                 if (data != NULL) {
85                         g_free(data->identifier);
86                         data->identifier = g_strdup(identifier);
87                 }
88         }
89
90         connman_element_get_value(element,
91                         CONNMAN_PROPERTY_ID_WIFI_SECURITY, &security);
92
93         connman_element_get_value(element,
94                         CONNMAN_PROPERTY_ID_WIFI_PASSPHRASE, &passphrase);
95
96         DBG("identifier %s security %s passhprase %s",
97                                         identifier, security, passphrase);
98
99         if (__supplicant_connect(element, ssid, ssid_len,
100                                                 security, passphrase) < 0)
101                 connman_error("Failed to initiate connect");
102
103         return 0;
104 }
105
106 static int network_disable(struct connman_element *element)
107 {
108         DBG("element %p name %s", element, element->name);
109
110         connman_element_unregister_children(element);
111
112         __supplicant_disconnect(element);
113
114         return 0;
115 }
116
117 static struct connman_driver network_driver = {
118         .name           = "wifi-network",
119         .type           = CONNMAN_ELEMENT_TYPE_NETWORK,
120         .subtype        = CONNMAN_ELEMENT_SUBTYPE_WIFI,
121         .probe          = network_probe,
122         .remove         = network_remove,
123         .enable         = network_enable,
124         .disable        = network_disable,
125 };
126
127 static struct connman_element *find_element(struct wifi_data *data,
128                                                 const char *identifier)
129 {
130         GSList *list;
131
132         for (list = data->list; list; list = list->next) {
133                 struct connman_element *element = list->data;
134
135                 if (connman_element_match_static_property(element,
136                                         "WiFi.Name", &identifier) == TRUE)
137                         return element;
138         }
139
140         return NULL;
141 }
142
143 static void state_change(struct connman_element *parent,
144                                                 enum supplicant_state state)
145 {
146         struct wifi_data *data = connman_element_get_data(parent);
147         struct connman_element *element;
148
149         DBG("state %d", state);
150
151         if (data->identifier == NULL)
152                 return;
153
154         element = find_element(data, data->identifier);
155         if (element == NULL)
156                 return;
157
158         if (state == STATE_COMPLETED) {
159                 struct connman_element *dhcp;
160
161                 data->connected = TRUE;
162
163                 dhcp = connman_element_create(NULL);
164
165                 dhcp->type = CONNMAN_ELEMENT_TYPE_DHCP;
166                 dhcp->index = element->index;
167
168                 connman_element_register(dhcp, element);
169         } else if (state == STATE_DISCONNECTED || state == STATE_INACTIVE)
170                 data->connected = FALSE;
171 }
172
173 static void scan_result(struct connman_element *parent,
174                                         struct supplicant_network *network)
175 {
176         struct wifi_data *data = connman_element_get_data(parent);
177         struct connman_element *element;
178         gchar *temp;
179         int i;
180
181         DBG("network %p identifier %s", network, network->identifier);
182
183         if (data == NULL)
184                 return;
185
186         if (network->identifier == NULL)
187                 return;
188
189         if (network->identifier[0] == '\0')
190                 return;
191
192         temp = g_strdup(network->identifier);
193
194         for (i = 0; i < strlen(temp); i++) {
195                 if (temp[i] == ' ' || temp[i] == '.')
196                         temp[i] = '_';
197                 else if (temp[i] == '-' || temp[i] == '+')
198                         temp[i] = '_';
199                 else if (temp[i] == '!' || temp[i] == '?')
200                         temp[i] = '_';
201                 else if (temp[i] == '(' || temp[i] == ')')
202                         temp[i] = '_';
203                 else if (g_ascii_isprint(temp[i]) == FALSE)
204                         temp[i] = '_';
205                 temp[i] = g_ascii_tolower(temp[i]);
206         }
207
208         element = find_element(data, network->identifier);
209         if (element == NULL) {
210                 guint8 strength;
211
212                 element = connman_element_create(temp);
213
214                 element->type = CONNMAN_ELEMENT_TYPE_NETWORK;
215                 element->index = parent->index;
216
217                 data->list = g_slist_append(data->list, element);
218
219                 connman_element_add_static_property(element, "WiFi.Name",
220                                 DBUS_TYPE_STRING, &network->identifier);
221
222                 connman_element_add_static_array_property(element, "WiFi.SSID",
223                         DBUS_TYPE_BYTE, &network->ssid, network->ssid_len);
224
225                 if (element->wifi.security == NULL) {
226                         const char *security;
227
228                         if (network->has_rsn == TRUE)
229                                 security = "wpa2";
230                         else if (network->has_wpa == TRUE)
231                                 security = "wpa";
232                         else if (network->has_wep == TRUE)
233                                 security = "wep";
234                         else
235                                 security = "none";
236
237                         element->wifi.security = g_strdup(security);
238                 }
239
240                 strength = network->quality;
241
242                 //connman_element_add_static_property(element, "WiFi.Strength",
243                 //                              DBUS_TYPE_BYTE, &strength);
244
245                 //connman_element_add_static_property(element, "WiFi.Noise",
246                 //                      DBUS_TYPE_INT32, &network->noise);
247
248                 DBG("%s (%s) strength %d", network->identifier,
249                                         element->wifi.security, strength);
250
251                 connman_element_register(element, parent);
252         }
253
254         g_free(temp);
255 }
256
257 static struct supplicant_callback wifi_callback = {
258         .state_change   = state_change,
259         .scan_result    = scan_result,
260 };
261
262 static int wifi_probe(struct connman_element *element)
263 {
264         struct wifi_data *data;
265
266         DBG("element %p name %s", element, element->name);
267
268         data = g_try_new0(struct wifi_data, 1);
269         if (data == NULL)
270                 return -ENOMEM;
271
272         data->connected = FALSE;
273
274         connman_element_set_data(element, data);
275
276         return 0;
277 }
278
279 static void wifi_remove(struct connman_element *element)
280 {
281         struct wifi_data *data = connman_element_get_data(element);
282
283         DBG("element %p name %s", element, element->name);
284
285         connman_element_set_data(element, NULL);
286
287         g_free(data->identifier);
288         g_free(data);
289 }
290
291 static int wifi_update(struct connman_element *element)
292 {
293         DBG("element %p name %s", element, element->name);
294
295         __supplicant_scan(element);
296
297         return 0;
298 }
299
300 static int wifi_enable(struct connman_element *element)
301 {
302         int err;
303
304         DBG("element %p name %s", element, element->name);
305
306         err = __supplicant_start(element, &wifi_callback);
307         if (err < 0)
308                 return err;
309
310         __supplicant_scan(element);
311
312         return 0;
313 }
314
315 static int wifi_disable(struct connman_element *element)
316 {
317         struct wifi_data *data = connman_element_get_data(element);
318         GSList *list;
319
320         DBG("element %p name %s", element, element->name);
321
322         __supplicant_disconnect(element);
323
324         for (list = data->list; list; list = list->next) {
325                 struct connman_element *network = list->data;
326
327                 connman_element_unref(network);
328         }
329
330         g_slist_free(data->list);
331         data->list = NULL;
332
333         connman_element_unregister_children(element);
334
335         return 0;
336 }
337
338 static struct connman_driver wifi_driver = {
339         .name           = "wifi-device",
340         .type           = CONNMAN_ELEMENT_TYPE_DEVICE,
341         .subtype        = CONNMAN_ELEMENT_SUBTYPE_WIFI,
342         .probe          = wifi_probe,
343         .remove         = wifi_remove,
344         .update         = wifi_update,
345         .enable         = wifi_enable,
346         .disable        = wifi_disable,
347 };
348
349 static GSList *device_list = NULL;
350
351 static void wifi_newlink(unsigned short type, int index,
352                                         unsigned flags, unsigned change)
353 {
354         struct connman_element *device;
355         GSList *list;
356         gboolean exists = FALSE;
357         gchar *name;
358         struct iwreq iwr;
359         int sk;
360
361         DBG("index %d", index);
362
363         if (type != ARPHRD_ETHER)
364                 return;
365
366         name = inet_index2name(index);
367
368         memset(&iwr, 0, sizeof(iwr));
369         strncpy(iwr.ifr_ifrn.ifrn_name, name, IFNAMSIZ);
370
371         sk = socket(PF_INET, SOCK_DGRAM, 0);
372
373         if (ioctl(sk, SIOCGIWNAME, &iwr) < 0) {
374                 g_free(name);
375                 close(sk);
376                 return;
377         }
378
379         close(sk);
380
381         for (list = device_list; list; list = list->next) {
382                 struct connman_element *device = list->data;
383
384                 if (device->index == index) {
385                         exists = TRUE;
386                         break;
387                 }
388         }
389
390         if (exists == TRUE) {
391                 g_free(name);
392                 return;
393         }
394
395         device = connman_element_create(NULL);
396         device->type = CONNMAN_ELEMENT_TYPE_DEVICE;
397         device->subtype = CONNMAN_ELEMENT_SUBTYPE_WIFI;
398
399         device->index = index;
400         device->name = name;
401
402         connman_element_register(device, NULL);
403         device_list = g_slist_append(device_list, device);
404 }
405
406 static void wifi_dellink(unsigned short type, int index,
407                                         unsigned flags, unsigned change)
408 {
409         GSList *list;
410
411         DBG("index %d", index);
412
413         for (list = device_list; list; list = list->next) {
414                 struct connman_element *device = list->data;
415
416                 if (device->index == index) {
417                         device_list = g_slist_remove(device_list, device);
418                         connman_element_unregister(device);
419                         connman_element_unref(device);
420                         break;
421                 }
422         }
423 }
424
425 static struct connman_rtnl wifi_rtnl = {
426         .name           = "wifi",
427         .newlink        = wifi_newlink,
428         .dellink        = wifi_dellink,
429 };
430
431 static void supplicant_connect(DBusConnection *connection, void *user_data)
432 {
433         DBG("connection %p", connection);
434
435         __supplicant_init(connection);
436
437         if (connman_rtnl_register(&wifi_rtnl) < 0)
438                 return;
439
440         connman_rtnl_send_getlink();
441 }
442
443 static void supplicant_disconnect(DBusConnection *connection, void *user_data)
444 {
445         GSList *list;
446
447         DBG("connection %p", connection);
448
449         connman_rtnl_unregister(&wifi_rtnl);
450
451         for (list = device_list; list; list = list->next) {
452                 struct connman_element *device = list->data;
453
454                 connman_element_unregister(device);
455                 connman_element_unref(device);
456         }
457
458         g_slist_free(device_list);
459         device_list = NULL;
460
461         __supplicant_exit();
462 }
463
464 static DBusConnection *connection;
465 static guint watch;
466
467 static int wifi_init(void)
468 {
469         int err;
470
471         connection = dbus_bus_get(DBUS_BUS_SYSTEM, NULL);
472         if (connection == NULL)
473                 return -EIO;
474
475         err = connman_driver_register(&network_driver);
476         if (err < 0) {
477                 dbus_connection_unref(connection);
478                 return err;
479         }
480
481         err = connman_driver_register(&wifi_driver);
482         if (err < 0) {
483                 connman_driver_unregister(&network_driver);
484                 dbus_connection_unref(connection);
485                 return err;
486         }
487
488         watch = g_dbus_add_service_watch(connection, SUPPLICANT_NAME,
489                         supplicant_connect, supplicant_disconnect, NULL, NULL);
490
491         if (g_dbus_check_service(connection, SUPPLICANT_NAME) == TRUE)
492                 supplicant_connect(connection, NULL);
493
494         return 0;
495 }
496
497 static void wifi_exit(void)
498 {
499         connman_driver_unregister(&network_driver);
500         connman_driver_unregister(&wifi_driver);
501
502         if (watch > 0)
503                 g_dbus_remove_watch(connection, watch);
504
505         supplicant_disconnect(connection, NULL);
506
507         dbus_connection_unref(connection);
508 }
509
510 CONNMAN_PLUGIN_DEFINE("wifi", "WiFi interface plugin", VERSION,
511                                                         wifi_init, wifi_exit)