Add handling of scanning and stations
authorMarcel Holtmann <marcel@holtmann.org>
Tue, 1 Jan 2008 06:20:26 +0000 (07:20 +0100)
committerMarcel Holtmann <marcel@holtmann.org>
Tue, 1 Jan 2008 06:20:26 +0000 (07:20 +0100)
plugins/80211.c

index 5d52ed9..d4c63dc 100644 (file)
 #endif
 
 #include <stdio.h>
+#include <errno.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/ioctl.h>
+#include <sys/socket.h>
 #include <arpa/inet.h>
+#include <net/if.h>
+#include <net/ethernet.h>
+#include <linux/wireless.h>
+
+#include <glib.h>
 
 #include <connman/plugin.h>
 #include <connman/iface.h>
 
-#include "net.h"
 #include "supplicant.h"
 
+struct station_data {
+       char *address;
+       char *name;
+       int mode;
+       int qual;
+       int noise;
+       int level;
+
+       unsigned char wpa_ie[40];
+       int wpa_ie_len;
+       unsigned char rsn_ie[40];
+       int rsn_ie_len;
+};
+
+struct iface_data {
+       char ifname[IFNAMSIZ];
+       GSList *stations;
+};
+
+static struct station_data *create_station(struct iface_data *iface,
+                                                       const char *address)
+{
+       struct station_data *station;
+       GSList *list;
+
+       for (list = iface->stations; list; list = list->next) {
+               station = list->data;
+
+               if (g_ascii_strcasecmp(station->address, address) == 0)
+                       return station;
+       }
+
+       station = g_try_new0(struct station_data, 1);
+       if (station == NULL)
+               return NULL;
+
+       station->address = g_strdup(address);
+       if (station->address == NULL) {
+               g_free(station);
+               return NULL;
+       }
+
+       iface->stations = g_slist_append(iface->stations, station);
+
+       return station;
+}
+
+static void load_stations(struct iface_data *iface)
+{
+       GKeyFile *keyfile;
+       gchar **groups, **group;
+       gsize length;
+
+       keyfile = g_key_file_new();
+
+       if (g_key_file_load_from_file(keyfile, "/tmp/stations.list",
+                               G_KEY_FILE_KEEP_COMMENTS, NULL) == FALSE)
+               goto done;
+
+       groups = g_key_file_get_groups(keyfile, &length);
+
+       for (group = groups; *group; group++) {
+               struct station_data *station;
+
+               station = create_station(iface, *group);
+               if (station == NULL)
+                       continue;
+
+               station->name = g_key_file_get_string(keyfile,
+                                               *group, "Name", NULL);
+       
+               station->mode = g_key_file_get_integer(keyfile,
+                                               *group, "Mode", NULL);
+       }
+
+       g_strfreev(groups);
+
+done:
+       g_key_file_free(keyfile);
+
+       printf("[802.11] loaded %d stations\n",
+                               g_slist_length(iface->stations));
+}
+
+static void print_stations(struct iface_data *iface)
+{
+       GKeyFile *keyfile;
+       gchar *data;
+       gsize length;
+       GSList *list;
+
+       keyfile = g_key_file_new();
+
+       for (list = iface->stations; list; list = list->next) {
+               struct station_data *station = list->data;
+
+               //printf("Address:%s Mode:%d ESSID:\"%s\" Quality:%d/100\n",
+               //                      station->address, station->mode,
+               //                              station->name, station->qual);
+
+               g_key_file_set_string(keyfile, station->address,
+                                               "Name", station->name);
+
+               g_key_file_set_integer(keyfile, station->address,
+                                               "Mode", station->mode);
+       }
+
+       data = g_key_file_to_data(keyfile, &length, NULL);
+
+       g_file_set_contents("/tmp/stations.list", data, length, NULL);
+
+       g_key_file_free(keyfile);
+}
+
 static int iface_probe(struct connman_iface *iface)
 {
-       printf("[802.11] probe interface index %d\n", iface->index);
+       struct iface_data *data;
+       struct ifreq ifr;
+       int sk, err;
+
+       sk = socket(PF_INET, SOCK_DGRAM, 0);
+       if (sk < 0)
+               return -EIO;
+
+       memset(&ifr, 0, sizeof(ifr));
+       ifr.ifr_ifindex = iface->index;
+
+       err = ioctl(sk, SIOCGIFNAME, &ifr);
+
+       close(sk);
+
+       if (err < 0)
+               return -EIO;
+
+       printf("[802.11] probe %s\n", ifr.ifr_name);
+
+       data = malloc(sizeof(*data));
+       if (data == NULL)
+               return -ENOMEM;
+
+       memset(data, 0, sizeof(*data));
+
+       memcpy(data->ifname, ifr.ifr_name, IFNAMSIZ);
 
        iface->type = CONNMAN_IFACE_TYPE_80211;
 
        iface->flags = CONNMAN_IFACE_FLAG_RTNL |
                                CONNMAN_IFACE_FLAG_IPV4;
 
+       connman_iface_set_data(iface, data);
+
+       load_stations(data);
+
        return 0;
 }
 
 static void iface_remove(struct connman_iface *iface)
 {
-       printf("[802.11] remove interface index %d\n", iface->index);
+       struct iface_data *data = connman_iface_get_data(iface);
 
-       __net_clear(iface->index);
+       printf("[802.11] remove %s\n", data->ifname);
 
        __supplicant_stop(iface);
+
+       connman_iface_set_data(iface, NULL);
+
+       free(data);
 }
 
 static int iface_activate(struct connman_iface *iface)
 {
-       printf("[802.11] activate interface index %d\n", iface->index);
+       struct iface_data *data = connman_iface_get_data(iface);
 
-       __supplicant_start(iface);
+       printf("[802.11] activate %s\n", data->ifname);
 
        connman_iface_update(iface, CONNMAN_IFACE_STATE_ACTIVE);
 
        return 0;
 }
 
-static int iface_get_ipv4(struct connman_iface *iface,
-                                       struct connman_ipv4 *ipv4)
+static int iface_scan(struct connman_iface *iface)
 {
-       if (__net_ifaddr(iface->index, &ipv4->address) < 0)
-               return -1;
+       struct iface_data *data = connman_iface_get_data(iface);
+       struct iwreq iwr;
+       struct iw_scan_req iws;
+       int sk, err;
 
-       printf("[802.11] get address %s\n", inet_ntoa(ipv4->address));
+       printf("[802.11] scanning %s\n", data->ifname);
 
-       return 0;
-}
+       sk = socket(PF_INET, SOCK_DGRAM, 0);
+       if (sk < 0)
+               return -EIO;
 
-static int iface_set_ipv4(struct connman_iface *iface,
-                                       struct connman_ipv4 *ipv4)
-{
-       printf("[802.11] set address %s\n", inet_ntoa(ipv4->address));
-       printf("[802.11] set netmask %s\n", inet_ntoa(ipv4->netmask));
-       printf("[802.11] set gateway %s\n", inet_ntoa(ipv4->gateway));
+       memset(&iws, 0, sizeof(iws));
+       iws.scan_type = IW_SCAN_TYPE_PASSIVE;
+       //iws.scan_type = IW_SCAN_TYPE_ACTIVE;
 
-       __net_set(iface->index, &ipv4->address, &ipv4->netmask,
-                               &ipv4->gateway, &ipv4->broadcast,
-                                               &ipv4->nameserver);
+       memset(&iwr, 0, sizeof(iwr));
+       strncpy(iwr.ifr_name, data->ifname, IFNAMSIZ);
 
-       return 0;
-}
+       iwr.u.data.pointer = (caddr_t ) &iws;
+       iwr.u.data.length = sizeof(iws);
+       iwr.u.data.flags = IW_SCAN_DEFAULT;
 
-static int iface_scan(struct connman_iface *iface)
-{
-       printf("[802.11] scanning interface index %d\n", iface->index);
+       err = ioctl(sk, SIOCSIWSCAN, &iwr);
 
-       return 0;
+       close(sk);
+
+       if (err < 0)
+               printf("[802.11] scan initiate error %d\n", errno);
+
+       return err;
 }
 
 static int iface_connect(struct connman_iface *iface,
@@ -101,21 +261,227 @@ static int iface_connect(struct connman_iface *iface,
 {
        printf("[802.11] connect interface index %d\n", iface->index);
 
+       __supplicant_start(iface);
+
        __supplicant_connect(iface);
 
        return 0;
 }
 
+static void iface_carrier(struct connman_iface *iface, int carrier)
+{
+       printf("[802.11] carrier %s\n", carrier ? "on" : "off");
+
+       connman_iface_indicate_carrier(iface, carrier);
+}
+
+static void parse_genie(struct station_data *station,
+                                       unsigned char *data, int len)
+{
+       int offset = 0;
+
+       while (offset <= len - 2) {
+               //int i;
+
+               switch (data[offset]) {
+               case 0xdd:      /* WPA1 (and other) */
+                       break;
+               case 0x30:      /* WPA2 (RSN) */
+                       break;
+               default:
+                       break;
+               }
+
+               //for (i = 0; i < len; i++)
+               //      printf(" %02x", data[i]);
+               //printf("\n");
+
+               offset += data[offset + 1] + 2;
+       }
+}
+
+static void parse_scan_results(struct connman_iface *iface,
+                                       unsigned char *data, int len)
+{
+       unsigned char *ptr = data;
+       struct station_data *station = NULL;
+       struct ether_addr *eth;
+       char addr[18];
+       int num = 0;
+
+       while (len > IW_EV_LCP_PK_LEN) {
+               struct iw_event *event = (void *) ptr;
+
+               switch (event->cmd) {
+               case SIOCGIWAP:
+                       eth = (void *) &event->u.ap_addr.sa_data;
+                       sprintf(addr, "%02X:%02X:%02X:%02X:%02X:%02X",
+                                               eth->ether_addr_octet[0],
+                                               eth->ether_addr_octet[1],
+                                               eth->ether_addr_octet[2],
+                                               eth->ether_addr_octet[3],
+                                               eth->ether_addr_octet[4],
+                                               eth->ether_addr_octet[5]);
+                       station = create_station(connman_iface_get_data(iface),
+                                                                       addr);
+                       num++;
+                       break;
+               case SIOCGIWESSID:
+                       if (station != NULL) {
+                               station->name = malloc(event->len - 7);
+                               if (station->name != NULL) {
+                                       memset(station->name, 0, event->len - 7);
+                                       memcpy(station->name, ptr + 8,
+                                                               event->len - 8);
+                               }
+                       }
+                       break;
+               case SIOCGIWNAME:
+                       break;
+               case SIOCGIWMODE:
+                       if (station != NULL)
+                               station->mode = event->u.mode;
+                       break;
+               case SIOCGIWFREQ:
+                       break;
+               case SIOCGIWENCODE:
+                       if (station != NULL) {
+                               if (!(event->u.data.flags & IW_ENCODE_DISABLED)) {
+                                       /* privacy */
+                               }
+                       }
+                       break;
+               case SIOCGIWRATE:
+                       break;
+               case IWEVQUAL:
+                       if (station != NULL) {
+                               station->qual = event->u.qual.qual;
+                               station->noise = event->u.qual.noise;
+                               station->level = event->u.qual.level;
+                       }
+                       break;
+               case IWEVGENIE:
+                       if (station != NULL)
+                               parse_genie(station, ptr + 8, event->len - 8);
+                       break;
+               case IWEVCUSTOM:
+                       break;
+               default:
+                       printf("[802.11] scan element 0x%04x (len %d)\n",
+                                               event->cmd, event->len);
+                       break;
+               }
+
+               ptr += event->len;
+               len -= event->len;
+       }
+
+       printf("[802.11] found %d networks\n", num);
+}
+
+static void iface_scan_results(struct connman_iface *iface)
+{
+       struct iface_data *data = connman_iface_get_data(iface);
+       struct iwreq iwr;
+       unsigned char *buf;
+       int sk, err, size = 1024;
+
+       sk = socket(PF_INET, SOCK_DGRAM, 0);
+       if (sk < 0)
+               return;
+
+retrieve:
+       buf = malloc(size);
+       if (buf == NULL) {
+               close(sk);
+               return;
+       }
+
+       memset(&iwr, 0, sizeof(iwr));
+       strncpy(iwr.ifr_name, data->ifname, IFNAMSIZ);
+       iwr.u.data.pointer = buf;
+       iwr.u.data.length = size;
+       iwr.u.data.flags = 0;
+
+       err = ioctl(sk, SIOCGIWSCAN, &iwr);
+       if (err < 0) {
+               if (errno == E2BIG) {
+                       free(buf);
+                       size *= 2;
+                       goto retrieve;
+               }
+       } else
+               parse_scan_results(iface, iwr.u.data.pointer,
+                                               iwr.u.data.length);
+
+       close(sk);
+
+       free(buf);
+
+       print_stations(data);
+}
+
+static void iface_wireless(struct connman_iface *iface,
+                                       void *data, unsigned short len)
+{
+       struct iw_event *event = data;
+       struct iw_point point;
+       struct ether_addr *eth;
+       char addr[18];
+
+       switch (event->cmd) {
+       case SIOCSIWFREQ:
+               printf("[802.11] Set Frequency (flags %d)\n",
+                                                       event->u.freq.flags);
+               break;
+       case SIOCSIWMODE:
+               printf("[802.11] Set Mode (mode %d)\n", event->u.mode);
+               break;
+       case SIOCSIWESSID:
+               memcpy(&point, data + IW_EV_LCP_LEN -
+                                       IW_EV_POINT_OFF, sizeof(point));
+               point.pointer = data + IW_EV_LCP_LEN +
+                                       sizeof(point) - IW_EV_POINT_OFF;
+               printf("[802.11] Set ESSID (length %d flags %d) \"%s\"\n",
+                                       point.length, point.flags,
+                                               (char *) point.pointer);
+               break;
+       case SIOCSIWENCODE:
+               printf("[802.11] Set Encryption key (flags %d)\n",
+                                                       event->u.data.flags);
+               break;
+
+       case SIOCGIWAP:
+               eth = (void *) &event->u.ap_addr.sa_data;
+               sprintf(addr, "%02X:%02X:%02X:%02X:%02X:%02X",
+                                               eth->ether_addr_octet[0],
+                                               eth->ether_addr_octet[1],
+                                               eth->ether_addr_octet[2],
+                                               eth->ether_addr_octet[3],
+                                               eth->ether_addr_octet[4],
+                                               eth->ether_addr_octet[5]);
+               printf("[802.11] New Access Point %s\n", addr);
+               break;
+       case SIOCGIWSCAN:
+               iface_scan_results(iface);
+               break;
+       default:
+               printf("[802.11] Wireless event (cmd 0x%04x len %d)\n",
+                                               event->cmd, event->len);
+               break;
+       }
+}
+
 static struct connman_iface_driver iface_driver = {
        .name           = "80211",
        .capability     = "net.80211",
        .probe          = iface_probe,
        .remove         = iface_remove,
        .activate       = iface_activate,
-       .get_ipv4       = iface_get_ipv4,
-       .set_ipv4       = iface_set_ipv4,
        .scan           = iface_scan,
        .connect        = iface_connect,
+       .rtnl_carrier   = iface_carrier,
+       .rtnl_wireless  = iface_wireless,
 };
 
 static int plugin_init(void)