connect.c

   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
  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
  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
  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         conn_argc = 3 + argc;
  98         conn_argv = calloc(conn_argc, sizeof(*conn_argv));
  99         if (!conn_argv)
 100                 return -ENOMEM;
 101
 102         err = __prepare_listen_events(state);
 103         if (err)
 104                 return err;
 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
 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");