src/rsn_supp/tdls.c

   1 /*
   2  * wpa_supplicant - TDLS
   3  * Copyright (c) 2010-2011, Atheros Communications
   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 "utils/includes.h"
  16
  17 #include "utils/common.h"
  18 #include "utils/eloop.h"
  19 #include "utils/os.h"
  20 #include "common/ieee802_11_defs.h"
  21 #include "crypto/sha256.h"
  22 #include "crypto/crypto.h"
  23 #include "crypto/aes_wrap.h"
  24 #include "rsn_supp/wpa.h"
  25 #include "rsn_supp/wpa_ie.h"
  26 #include "rsn_supp/wpa_i.h"
  27 #include "drivers/driver.h"
  28 #include "l2_packet/l2_packet.h"
  29
  30 #ifdef CONFIG_TDLS_TESTING
  31 #define TDLS_TESTING_LONG_FRAME BIT(0)
  32 #define TDLS_TESTING_ALT_RSN_IE BIT(1)
  33 #define TDLS_TESTING_DIFF_BSSID BIT(2)
  34 #define TDLS_TESTING_SHORT_LIFETIME BIT(3)
  35 #define TDLS_TESTING_WRONG_LIFETIME_RESP BIT(4)
  36 #define TDLS_TESTING_WRONG_LIFETIME_CONF BIT(5)
  37 #define TDLS_TESTING_LONG_LIFETIME BIT(6)
  38 #define TDLS_TESTING_CONCURRENT_INIT BIT(7)
  39 #define TDLS_TESTING_NO_TPK_EXPIRATION BIT(8)
  40 #define TDLS_TESTING_DECLINE_RESP BIT(9)
  41 #define TDLS_TESTING_IGNORE_AP_PROHIBIT BIT(10)
  42 unsigned int tdls_testing = 0;
  43 #endif /* CONFIG_TDLS_TESTING */
  44
  45 #define TPK_LIFETIME 43200 /* 12 hours */
  46 #define TPK_RETRY_COUNT 3
  47 #define TPK_TIMEOUT 5000 /* in milliseconds */
  48
  49 #define TDLS_MIC_LEN            16
  50
  51 #define TDLS_TIMEOUT_LEN        4
  52
  53 struct wpa_tdls_ftie {
  54         u8 ie_type; /* FTIE */
  55         u8 ie_len;
  56         u8 mic_ctrl[2];
  57         u8 mic[TDLS_MIC_LEN];
  58         u8 Anonce[WPA_NONCE_LEN]; /* Responder Nonce in TDLS */
  59         u8 Snonce[WPA_NONCE_LEN]; /* Initiator Nonce in TDLS */
  60         /* followed by optional elements */
  61 } STRUCT_PACKED;
  62
  63 struct wpa_tdls_timeoutie {
  64         u8 ie_type; /* Timeout IE */
  65         u8 ie_len;
  66         u8 interval_type;
  67         u8 value[TDLS_TIMEOUT_LEN];
  68 } STRUCT_PACKED;
  69
  70 struct wpa_tdls_lnkid {
  71         u8 ie_type; /* Link Identifier IE */
  72         u8 ie_len;
  73         u8 bssid[ETH_ALEN];
  74         u8 init_sta[ETH_ALEN];
  75         u8 resp_sta[ETH_ALEN];
  76 } STRUCT_PACKED;
  77
  78 /* TDLS frame headers as per IEEE Std 802.11z-2010 */
  79 struct wpa_tdls_frame {
  80         u8 payloadtype; /* IEEE80211_TDLS_RFTYPE */
  81         u8 category; /* Category */
  82         u8 action; /* Action (enum tdls_frame_type) */
  83 } STRUCT_PACKED;
  84
  85 static u8 * wpa_add_tdls_timeoutie(u8 *pos, u8 *ie, size_t ie_len, u32 tsecs);
  86 static void wpa_tdls_tpk_retry_timeout(void *eloop_ctx, void *timeout_ctx);
  87 static void wpa_tdls_peer_free(struct wpa_sm *sm, struct wpa_tdls_peer *peer);
  88
  89
  90 #define TDLS_MAX_IE_LEN 80
  91 struct wpa_tdls_peer {
  92         struct wpa_tdls_peer *next;
  93         int initiator; /* whether this end was initiator for TDLS setup */
  94         u8 addr[ETH_ALEN]; /* other end MAC address */
  95         u8 inonce[WPA_NONCE_LEN]; /* Initiator Nonce */
  96         u8 rnonce[WPA_NONCE_LEN]; /* Responder Nonce */
  97         u8 rsnie_i[TDLS_MAX_IE_LEN]; /* Initiator RSN IE */
  98         size_t rsnie_i_len;
  99         u8 rsnie_p[TDLS_MAX_IE_LEN]; /* Peer RSN IE */
 100         size_t rsnie_p_len;
 101         u32 lifetime;
 102         int cipher; /* Selected cipher (WPA_CIPHER_*) */
 103         u8 dtoken;
 104
 105         struct tpk {
 106                 u8 kck[16]; /* TPK-KCK */
 107                 u8 tk[16]; /* TPK-TK; assuming only CCMP will be used */
 108         } tpk;
 109         int tpk_set;
 110         int tpk_success;
 111
 112         struct tpk_timer {
 113                 u8 dest[ETH_ALEN];
 114                 int count;      /* Retry Count */
 115                 int timer;      /* Timeout in milliseconds */
 116                 u8 action_code; /* TDLS frame type */
 117                 u8 dialog_token;
 118                 u16 status_code;
 119                 int buf_len;    /* length of TPK message for retransmission */
 120                 u8 *buf;        /* buffer for TPK message */
 121         } sm_tmr;
 122 };
 123
 124
 125 static int wpa_tdls_get_privacy(struct wpa_sm *sm)
 126 {
 127         /*
 128          * Get info needed from supplicant to check if the current BSS supports
 129          * security. Other than OPEN mode, rest are considered secured
 130          * WEP/WPA/WPA2 hence TDLS frames are processed for TPK handshake.
 131          */
 132         return sm->pairwise_cipher != WPA_CIPHER_NONE;
 133 }
 134
 135
 136 static u8 * wpa_add_ie(u8 *pos, const u8 *ie, size_t ie_len)
 137 {
 138         os_memcpy(pos, ie, ie_len);
 139         return pos + ie_len;
 140 }
 141
 142
 143 static int wpa_tdls_del_key(struct wpa_sm *sm, struct wpa_tdls_peer *peer)
 144 {
 145         if (wpa_sm_set_key(sm, WPA_ALG_NONE, peer->addr,
 146                            0, 0, NULL, 0, NULL, 0) < 0) {
 147                 wpa_printf(MSG_WARNING, "TDLS: Failed to delete TPK-TK from "
 148                            "the driver");
 149                 return -1;
 150         }
 151
 152         return 0;
 153 }
 154
 155
 156 static int wpa_tdls_set_key(struct wpa_sm *sm, struct wpa_tdls_peer *peer)
 157 {
 158         u8 key_len;
 159         u8 rsc[6];
 160         enum wpa_alg alg;
 161
 162         os_memset(rsc, 0, 6);
 163
 164         switch (peer->cipher) {
 165         case WPA_CIPHER_CCMP:
 166                 alg = WPA_ALG_CCMP;
 167                 key_len = 16;
 168                 break;
 169         case WPA_CIPHER_NONE:
 170                 wpa_printf(MSG_DEBUG, "TDLS: Pairwise Cipher Suite: "
 171                            "NONE - do not use pairwise keys");
 172                 return -1;
 173         default:
 174                 wpa_printf(MSG_WARNING, "TDLS: Unsupported pairwise cipher %d",
 175                            sm->pairwise_cipher);
 176                 return -1;
 177         }
 178
 179         if (wpa_sm_set_key(sm, alg, peer->addr, -1, 1,
 180                            rsc, sizeof(rsc), peer->tpk.tk, key_len) < 0) {
 181                 wpa_printf(MSG_WARNING, "TDLS: Failed to set TPK to the "
 182                            "driver");
 183                 return -1;
 184         }
 185         return 0;
 186 }
 187
 188
 189 static int wpa_tdls_send_tpk_msg(struct wpa_sm *sm, const u8 *dst,
 190                                  u8 action_code, u8 dialog_token,
 191                                  u16 status_code, const u8 *buf, size_t len)
 192 {
 193         return wpa_sm_send_tdls_mgmt(sm, dst, action_code, dialog_token,
 194                                      status_code, buf, len);
 195 }
 196
 197
 198 static int wpa_tdls_tpk_send(struct wpa_sm *sm, const u8 *dest, u8 action_code,
 199                              u8 dialog_token, u16 status_code,
 200                              const u8 *msg, size_t msg_len)
 201 {
 202         struct wpa_tdls_peer *peer;
 203
 204         wpa_printf(MSG_DEBUG, "TDLS: TPK send dest=" MACSTR " action_code=%u "
 205                    "dialog_token=%u status_code=%u msg_len=%u",
 206                    MAC2STR(dest), action_code, dialog_token, status_code,
 207                    (unsigned int) msg_len);
 208
 209         if (wpa_tdls_send_tpk_msg(sm, dest, action_code, dialog_token,
 210                                   status_code, msg, msg_len)) {
 211                 wpa_printf(MSG_INFO, "TDLS: Failed to send message "
 212                            "(action_code=%u)", action_code);
 213                 return -1;
 214         }
 215
 216         if (action_code == WLAN_TDLS_SETUP_CONFIRM ||
 217             action_code == WLAN_TDLS_TEARDOWN ||
 218             action_code == WLAN_TDLS_DISCOVERY_REQUEST ||
 219             action_code == WLAN_TDLS_DISCOVERY_RESPONSE)
 220                 return 0; /* No retries */
 221
 222         for (peer = sm->tdls; peer; peer = peer->next) {
 223                 if (os_memcmp(peer->addr, dest, ETH_ALEN) == 0)
 224                         break;
 225         }
 226
 227         if (peer == NULL) {
 228                 wpa_printf(MSG_INFO, "TDLS: No matching entry found for "
 229                            "retry " MACSTR, MAC2STR(dest));
 230                 return 0;
 231         }
 232
 233         eloop_cancel_timeout(wpa_tdls_tpk_retry_timeout, sm, peer);
 234
 235         peer->sm_tmr.count = TPK_RETRY_COUNT;
 236         peer->sm_tmr.timer = TPK_TIMEOUT;
 237
 238         /* Copy message to resend on timeout */
 239         os_memcpy(peer->sm_tmr.dest, dest, ETH_ALEN);
 240         peer->sm_tmr.action_code = action_code;
 241         peer->sm_tmr.dialog_token = dialog_token;
 242         peer->sm_tmr.status_code = status_code;
 243         peer->sm_tmr.buf_len = msg_len;
 244         os_free(peer->sm_tmr.buf);
 245         peer->sm_tmr.buf = os_malloc(msg_len);
 246         if (peer->sm_tmr.buf == NULL)
 247                 return -1;
 248         os_memcpy(peer->sm_tmr.buf, msg, msg_len);
 249
 250         wpa_printf(MSG_DEBUG, "TDLS: Retry timeout registered "
 251                    "(action_code=%u)", action_code);
 252         eloop_register_timeout(peer->sm_tmr.timer / 1000, 0,
 253                                wpa_tdls_tpk_retry_timeout, sm, peer);
 254         return 0;
 255 }
 256
 257
 258 static void wpa_tdls_tpk_retry_timeout(void *eloop_ctx, void *timeout_ctx)
 259 {
 260
 261         struct wpa_sm *sm = eloop_ctx;
 262         struct wpa_tdls_peer *peer = timeout_ctx;
 263
 264         if (peer->sm_tmr.count) {
 265                 peer->sm_tmr.count--;
 266                 peer->sm_tmr.timer = TPK_TIMEOUT;
 267
 268                 wpa_printf(MSG_INFO, "TDLS: Retrying sending of message "
 269                            "(action_code=%u)",
 270                            peer->sm_tmr.action_code);
 271
 272                 if (peer->sm_tmr.buf == NULL) {
 273                         wpa_printf(MSG_INFO, "TDLS: No retry buffer available "
 274                                    "for action_code=%u",
 275                                    peer->sm_tmr.action_code);
 276                         eloop_cancel_timeout(wpa_tdls_tpk_retry_timeout, sm,
 277                                              peer);
 278                         return;
 279                 }
 280
 281                 /* resend TPK Handshake Message to Peer */
 282                 if (wpa_tdls_send_tpk_msg(sm, peer->sm_tmr.dest,
 283                                           peer->sm_tmr.action_code,
 284                                           peer->sm_tmr.dialog_token,
 285                                           peer->sm_tmr.status_code,
 286                                           peer->sm_tmr.buf,
 287                                           peer->sm_tmr.buf_len)) {
 288                         wpa_printf(MSG_INFO, "TDLS: Failed to retry "
 289                                    "transmission");
 290                 }
 291
 292                 eloop_cancel_timeout(wpa_tdls_tpk_retry_timeout, sm, peer);
 293                 eloop_register_timeout(peer->sm_tmr.timer / 1000, 0,
 294                                        wpa_tdls_tpk_retry_timeout, sm, peer);
 295         } else {
 296                 wpa_printf(MSG_INFO, "Sending Tear_Down Request");
 297                 wpa_sm_tdls_oper(sm, TDLS_TEARDOWN, peer->addr);
 298
 299                 wpa_printf(MSG_INFO, "Clearing SM: Peerkey(" MACSTR ")",
 300                            MAC2STR(peer->addr));
 301                 eloop_cancel_timeout(wpa_tdls_tpk_retry_timeout, sm, peer);
 302
 303                 /* clear the Peerkey statemachine */
 304                 wpa_tdls_peer_free(sm, peer);
 305         }
 306 }
 307
 308
 309 static void wpa_tdls_tpk_retry_timeout_cancel(struct wpa_sm *sm,
 310                                               struct wpa_tdls_peer *peer,
 311                                               u8 action_code)
 312 {
 313         if (action_code == peer->sm_tmr.action_code) {
 314                 wpa_printf(MSG_DEBUG, "TDLS: Retry timeout cancelled for "
 315                            "action_code=%u", action_code);
 316
 317                 /* Cancel Timeout registered */
 318                 eloop_cancel_timeout(wpa_tdls_tpk_retry_timeout, sm, peer);
 319
 320                 /* free all resources meant for retry */
 321                 os_free(peer->sm_tmr.buf);
 322                 peer->sm_tmr.buf = NULL;
 323
 324                 peer->sm_tmr.count = 0;
 325                 peer->sm_tmr.timer = 0;
 326                 peer->sm_tmr.buf_len = 0;
 327                 peer->sm_tmr.action_code = 0xff;
 328         } else {
 329                 wpa_printf(MSG_INFO, "TDLS: Error in cancelling retry timeout "
 330                            "(Unknown action_code=%u)", action_code);
 331         }
 332 }
 333
 334
 335 static void wpa_tdls_generate_tpk(struct wpa_tdls_peer *peer,
 336                                   const u8 *own_addr, const u8 *bssid)
 337 {
 338         u8 key_input[SHA256_MAC_LEN];
 339         const u8 *nonce[2];
 340         size_t len[2];
 341         u8 data[3 * ETH_ALEN];
 342
 343         /* IEEE Std 802.11z-2010 8.5.9.1:
 344          * TPK-Key-Input = SHA-256(min(SNonce, ANonce) || max(SNonce, ANonce))
 345          */
 346         len[0] = WPA_NONCE_LEN;
 347         len[1] = WPA_NONCE_LEN;
 348         if (os_memcmp(peer->inonce, peer->rnonce, WPA_NONCE_LEN) < 0) {
 349                 nonce[0] = peer->inonce;
 350                 nonce[1] = peer->rnonce;
 351         } else {
 352                 nonce[0] = peer->rnonce;
 353                 nonce[1] = peer->inonce;
 354         }
 355         wpa_hexdump(MSG_DEBUG, "TDLS: min(Nonce)", nonce[0], WPA_NONCE_LEN);
 356         wpa_hexdump(MSG_DEBUG, "TDLS: max(Nonce)", nonce[1], WPA_NONCE_LEN);
 357         sha256_vector(2, nonce, len, key_input);
 358         wpa_hexdump_key(MSG_DEBUG, "TDLS: TPK-Key-Input",
 359                         key_input, SHA256_MAC_LEN);
 360
 361         /*
 362          * TPK-Key-Data = KDF-N_KEY(TPK-Key-Input, "TDLS PMK",
 363          *      min(MAC_I, MAC_R) || max(MAC_I, MAC_R) || BSSID || N_KEY)
 364          * TODO: is N_KEY really included in KDF Context and if so, in which
 365          * presentation format (little endian 16-bit?) is it used? It gets
 366          * added by the KDF anyway..
 367          */
 368
 369         if (os_memcmp(own_addr, peer->addr, ETH_ALEN) < 0) {
 370                 os_memcpy(data, own_addr, ETH_ALEN);
 371                 os_memcpy(data + ETH_ALEN, peer->addr, ETH_ALEN);
 372         } else {
 373                 os_memcpy(data, peer->addr, ETH_ALEN);
 374                 os_memcpy(data + ETH_ALEN, own_addr, ETH_ALEN);
 375         }
 376         os_memcpy(data + 2 * ETH_ALEN, bssid, ETH_ALEN);
 377         wpa_hexdump(MSG_DEBUG, "TDLS: KDF Context", data, sizeof(data));
 378
 379         sha256_prf(key_input, SHA256_MAC_LEN, "TDLS PMK", data, sizeof(data),
 380                    (u8 *) &peer->tpk, sizeof(peer->tpk));
 381         wpa_hexdump_key(MSG_DEBUG, "TDLS: TPK-KCK",
 382                         peer->tpk.kck, sizeof(peer->tpk.kck));
 383         wpa_hexdump_key(MSG_DEBUG, "TDLS: TPK-TK",
 384                         peer->tpk.tk, sizeof(peer->tpk.tk));
 385         peer->tpk_set = 1;
 386 }
 387
 388
 389 /**
 390  * wpa_tdls_ftie_mic - Calculate TDLS FTIE MIC
 391  * @kck: TPK-KCK
 392  * @lnkid: Pointer to the beginning of Link Identifier IE
 393  * @rsnie: Pointer to the beginning of RSN IE used for handshake
 394  * @timeoutie: Pointer to the beginning of Timeout IE used for handshake
 395  * @ftie: Pointer to the beginning of FT IE
 396  * @mic: Pointer for writing MIC
 397  *
 398  * Calculate MIC for TDLS frame.
 399  */
 400 static int wpa_tdls_ftie_mic(const u8 *kck, u8 trans_seq, const u8 *lnkid,
 401                              const u8 *rsnie, const u8 *timeoutie,
 402                              const u8 *ftie, u8 *mic)
 403 {
 404         u8 *buf, *pos;
 405         struct wpa_tdls_ftie *_ftie;
 406         const struct wpa_tdls_lnkid *_lnkid;
 407         int ret;
 408         int len = 2 * ETH_ALEN + 1 + 2 + lnkid[1] + 2 + rsnie[1] +
 409                 2 + timeoutie[1] + 2 + ftie[1];
 410         buf = os_zalloc(len);
 411         if (!buf) {
 412                 wpa_printf(MSG_WARNING, "TDLS: No memory for MIC calculation");
 413                 return -1;
 414         }
 415
 416         pos = buf;
 417         _lnkid = (const struct wpa_tdls_lnkid *) lnkid;
 418         /* 1) TDLS initiator STA MAC address */
 419         os_memcpy(pos, _lnkid->init_sta, ETH_ALEN);
 420         pos += ETH_ALEN;
 421         /* 2) TDLS responder STA MAC address */
 422         os_memcpy(pos, _lnkid->resp_sta, ETH_ALEN);
 423         pos += ETH_ALEN;
 424         /* 3) Transaction Sequence number */
 425         *pos++ = trans_seq;
 426         /* 4) Link Identifier IE */
 427         os_memcpy(pos, lnkid, 2 + lnkid[1]);
 428         pos += 2 + lnkid[1];
 429         /* 5) RSN IE */
 430         os_memcpy(pos, rsnie, 2 + rsnie[1]);
 431         pos += 2 + rsnie[1];
 432         /* 6) Timeout Interval IE */
 433         os_memcpy(pos, timeoutie, 2 + timeoutie[1]);
 434         pos += 2 + timeoutie[1];
 435         /* 7) FTIE, with the MIC field of the FTIE set to 0 */
 436         os_memcpy(pos, ftie, 2 + ftie[1]);
 437         _ftie = (struct wpa_tdls_ftie *) pos;
 438         os_memset(_ftie->mic, 0, TDLS_MIC_LEN);
 439         pos += 2 + ftie[1];
 440
 441         wpa_hexdump(MSG_DEBUG, "TDLS: Data for FTIE MIC", buf, pos - buf);
 442         wpa_hexdump_key(MSG_DEBUG, "TDLS: KCK", kck, 16);
 443         ret = omac1_aes_128(kck, buf, pos - buf, mic);
 444         os_free(buf);
 445         wpa_hexdump(MSG_DEBUG, "TDLS: FTIE MIC", mic, 16);
 446         return ret;
 447 }
 448
 449
 450 /**
 451  * wpa_tdls_key_mic_teardown - Calculate TDLS FTIE MIC for Teardown frame
 452  * @kck: TPK-KCK
 453  * @trans_seq: Transaction Sequence Number (4 - Teardown)
 454  * @rcode: Reason code for Teardown
 455  * @dtoken: Dialog Token used for that particular link
 456  * @lnkid: Pointer to the beginning of Link Identifier IE
 457  * @ftie: Pointer to the beginning of FT IE
 458  * @mic: Pointer for writing MIC
 459  *
 460  * Calculate MIC for TDLS frame.
 461  */
 462 static int wpa_tdls_key_mic_teardown(const u8 *kck, u8 trans_seq, u16 rcode,
 463                                      u8 dtoken, const u8 *lnkid,
 464                                      const u8 *ftie, u8 *mic)
 465 {
 466         u8 *buf, *pos;
 467         struct wpa_tdls_ftie *_ftie;
 468         int ret;
 469         int len;
 470
 471         if (lnkid == NULL)
 472                 return -1;
 473
 474         len = 2 + lnkid[1] + sizeof(rcode) + sizeof(dtoken) +
 475                 sizeof(trans_seq) + 2 + ftie[1];
 476
 477         buf = os_zalloc(len);
 478         if (!buf) {
 479                 wpa_printf(MSG_WARNING, "TDLS: No memory for MIC calculation");
 480                 return -1;
 481         }
 482
 483         pos = buf;
 484         /* 1) Link Identifier IE */
 485         os_memcpy(pos, lnkid, 2 + lnkid[1]);
 486         pos += 2 + lnkid[1];
 487         /* 2) Reason Code */
 488         WPA_PUT_LE16(pos, rcode);
 489         pos += sizeof(rcode);
 490         /* 3) Dialog token */
 491         *pos++ = dtoken;
 492         /* 4) Transaction Sequence number */
 493         *pos++ = trans_seq;
 494         /* 7) FTIE, with the MIC field of the FTIE set to 0 */
 495         os_memcpy(pos, ftie, 2 + ftie[1]);
 496         _ftie = (struct wpa_tdls_ftie *) pos;
 497         os_memset(_ftie->mic, 0, TDLS_MIC_LEN);
 498         pos += 2 + ftie[1];
 499
 500         wpa_hexdump(MSG_DEBUG, "TDLS: Data for FTIE MIC", buf, pos - buf);
 501         wpa_hexdump_key(MSG_DEBUG, "TDLS: KCK", kck, 16);
 502         ret = omac1_aes_128(kck, buf, pos - buf, mic);
 503         os_free(buf);
 504         wpa_hexdump(MSG_DEBUG, "TDLS: FTIE MIC", mic, 16);
 505         return ret;
 506 }
 507
 508
 509 static int wpa_supplicant_verify_tdls_mic(u8 trans_seq,
 510                                           struct wpa_tdls_peer *peer,
 511                                           const u8 *lnkid, const u8 *timeoutie,
 512                                           const struct wpa_tdls_ftie *ftie)
 513 {
 514         u8 mic[16];
 515
 516         if (peer->tpk_set) {
 517                 wpa_tdls_ftie_mic(peer->tpk.kck, trans_seq, lnkid,
 518                                   peer->rsnie_p, timeoutie, (u8 *) ftie,
 519                                   mic);
 520                 if (os_memcmp(mic, ftie->mic, 16) != 0) {
 521                         wpa_printf(MSG_INFO, "TDLS: Invalid MIC in FTIE - "
 522                                    "dropping packet");
 523                         wpa_hexdump(MSG_DEBUG, "TDLS: Received MIC",
 524                                     ftie->mic, 16);
 525                         wpa_hexdump(MSG_DEBUG, "TDLS: Calculated MIC",
 526                                     mic, 16);
 527                         return -1;
 528                 }
 529         } else {
 530                 wpa_printf(MSG_WARNING, "TDLS: Could not verify TDLS MIC, "
 531                            "TPK not set - dropping packet");
 532                 return -1;
 533         }
 534         return 0;
 535 }
 536
 537
 538 static int wpa_supplicant_verify_tdls_mic_teardown(
 539         u8 trans_seq, u16 rcode, u8 dtoken, struct wpa_tdls_peer *peer,
 540         const u8 *lnkid, const struct wpa_tdls_ftie *ftie)
 541 {
 542         u8 mic[16];
 543
 544         if (peer->tpk_set) {
 545                 wpa_tdls_key_mic_teardown(peer->tpk.kck, trans_seq, rcode,
 546                                           dtoken, lnkid, (u8 *) ftie, mic);
 547                 if (os_memcmp(mic, ftie->mic, 16) != 0) {
 548                         wpa_printf(MSG_INFO, "TDLS: Invalid MIC in Teardown - "
 549                                    "dropping packet");
 550                         return -1;
 551                 }
 552         } else {
 553                 wpa_printf(MSG_INFO, "TDLS: Could not verify TDLS Teardown "
 554                            "MIC, TPK not set - dropping packet");
 555                 return -1;
 556         }
 557         return 0;
 558 }
 559
 560
 561 static void wpa_tdls_tpk_timeout(void *eloop_ctx, void *timeout_ctx)
 562 {
 563         struct wpa_sm *sm = eloop_ctx;
 564         struct wpa_tdls_peer *peer = timeout_ctx;
 565
 566         /*
 567          * On TPK lifetime expiration, we have an option of either tearing down
 568          * the direct link or trying to re-initiate it. The selection of what
 569          * to do is not strictly speaking controlled by our role in the expired
 570          * link, but for now, use that to select whether to renew or tear down
 571          * the link.
 572          */
 573
 574         if (peer->initiator) {
 575                 wpa_printf(MSG_DEBUG, "TDLS: TPK lifetime expired for " MACSTR
 576                            " - try to renew", MAC2STR(peer->addr));
 577                 wpa_tdls_start(sm, peer->addr);
 578         } else {
 579                 wpa_printf(MSG_DEBUG, "TDLS: TPK lifetime expired for " MACSTR
 580                            " - tear down", MAC2STR(peer->addr));
 581                 wpa_sm_tdls_oper(sm, TDLS_TEARDOWN, peer->addr);
 582         }
 583 }
 584
 585
 586 static void wpa_tdls_peer_free(struct wpa_sm *sm, struct wpa_tdls_peer *peer)
 587 {
 588         wpa_printf(MSG_DEBUG, "TDLS: Clear state for peer " MACSTR,
 589                    MAC2STR(peer->addr));
 590         eloop_cancel_timeout(wpa_tdls_tpk_timeout, sm, peer);
 591         eloop_cancel_timeout(wpa_tdls_tpk_retry_timeout, sm, peer);
 592         peer->initiator = 0;
 593         os_free(peer->sm_tmr.buf);
 594         peer->sm_tmr.buf = NULL;
 595         peer->rsnie_i_len = peer->rsnie_p_len = 0;
 596         peer->cipher = 0;
 597         peer->tpk_set = peer->tpk_success = 0;
 598         os_memset(&peer->tpk, 0, sizeof(peer->tpk));
 599         os_memset(peer->inonce, 0, WPA_NONCE_LEN);
 600         os_memset(peer->rnonce, 0, WPA_NONCE_LEN);
 601 }
 602
 603
 604 static void wpa_tdls_linkid(struct wpa_sm *sm, struct wpa_tdls_peer *peer,
 605                             struct wpa_tdls_lnkid *lnkid)
 606 {
 607         lnkid->ie_type = WLAN_EID_LINK_ID;
 608         lnkid->ie_len = 3 * ETH_ALEN;
 609         os_memcpy(lnkid->bssid, sm->bssid, ETH_ALEN);
 610         if (peer->initiator) {
 611                 os_memcpy(lnkid->init_sta, sm->own_addr, ETH_ALEN);
 612                 os_memcpy(lnkid->resp_sta, peer->addr, ETH_ALEN);
 613         } else {
 614                 os_memcpy(lnkid->init_sta, peer->addr, ETH_ALEN);
 615                 os_memcpy(lnkid->resp_sta, sm->own_addr, ETH_ALEN);
 616         }
 617 }
 618
 619
 620 int wpa_tdls_recv_teardown_notify(struct wpa_sm *sm, const u8 *addr,
 621                                   u16 reason_code)
 622 {
 623         struct wpa_tdls_peer *peer;
 624         struct wpa_tdls_ftie *ftie;
 625         struct wpa_tdls_lnkid lnkid;
 626         u8 dialog_token;
 627         u8 *rbuf, *pos;
 628         int ielen;
 629
 630         if (sm->tdls_disabled || !sm->tdls_supported)
 631                 return -1;
 632
 633         /* Find the node and free from the list */
 634         for (peer = sm->tdls; peer; peer = peer->next) {
 635                 if (os_memcmp(peer->addr, addr, ETH_ALEN) == 0)
 636                         break;
 637         }
 638
 639         if (peer == NULL) {
 640                 wpa_printf(MSG_INFO, "TDLS: No matching entry found for "
 641                            "Teardown " MACSTR, MAC2STR(addr));
 642                 return 0;
 643         }
 644
 645         dialog_token = peer->dtoken;
 646
 647         wpa_printf(MSG_DEBUG, "TDLS: TDLS Teardown for " MACSTR,
 648                    MAC2STR(addr));
 649
 650         ielen = 0;
 651         if (wpa_tdls_get_privacy(sm) && peer->tpk_set && peer->tpk_success) {
 652                 /* To add FTIE for Teardown request and compute MIC */
 653                 ielen += sizeof(*ftie);
 654 #ifdef CONFIG_TDLS_TESTING
 655                 if (tdls_testing & TDLS_TESTING_LONG_FRAME)
 656                         ielen += 170;
 657 #endif /* CONFIG_TDLS_TESTING */
 658         }
 659
 660         rbuf = os_zalloc(ielen + 1);
 661         if (rbuf == NULL)
 662                 return -1;
 663         pos = rbuf;
 664
 665         if (!wpa_tdls_get_privacy(sm) || !peer->tpk_set || !peer->tpk_success)
 666                 goto skip_ies;
 667
 668         ftie = (struct wpa_tdls_ftie *) pos;
 669         ftie->ie_type = WLAN_EID_FAST_BSS_TRANSITION;
 670         /* Using the recent nonce which should be for CONFIRM frame */
 671         os_memcpy(ftie->Anonce, peer->rnonce, WPA_NONCE_LEN);
 672         os_memcpy(ftie->Snonce, peer->inonce, WPA_NONCE_LEN);
 673         ftie->ie_len = sizeof(struct wpa_tdls_ftie) - 2;
 674         pos = (u8 *) (ftie + 1);
 675 #ifdef CONFIG_TDLS_TESTING
 676         if (tdls_testing & TDLS_TESTING_LONG_FRAME) {
 677                 wpa_printf(MSG_DEBUG, "TDLS: Testing - add extra subelem to "
 678                            "FTIE");
 679                 ftie->ie_len += 170;
 680                 *pos++ = 255; /* FTIE subelem */
 681                 *pos++ = 168; /* FTIE subelem length */
 682         }
 683 #endif /* CONFIG_TDLS_TESTING */
 684         wpa_hexdump(MSG_DEBUG, "TDLS: FTIE for TDLS Teardown handshake",
 685                     (u8 *) ftie, sizeof(*ftie));
 686
 687         /* compute MIC before sending */
 688         wpa_tdls_linkid(sm, peer, &lnkid);
 689         wpa_tdls_key_mic_teardown(peer->tpk.kck, 4, reason_code,
 690                                   dialog_token, (u8 *) &lnkid, (u8 *) ftie,
 691                                   ftie->mic);
 692
 693 skip_ies:
 694         /* TODO: register for a Timeout handler, if Teardown is not received at
 695          * the other end, then try again another time */
 696
 697         /* request driver to send Teardown using this FTIE */
 698         wpa_tdls_tpk_send(sm, addr, WLAN_TDLS_TEARDOWN, 0,
 699                           WLAN_REASON_TDLS_TEARDOWN_UNSPECIFIED, rbuf,
 700                           pos - rbuf);
 701         os_free(rbuf);
 702
 703         /* clear the Peerkey statemachine */
 704         wpa_tdls_peer_free(sm, peer);
 705
 706         return 0;
 707 }
 708
 709
 710 static int wpa_tdls_recv_teardown(struct wpa_sm *sm, const u8 *src_addr,
 711                                   const u8 *buf, size_t len)
 712 {
 713         struct wpa_tdls_peer *peer = NULL;
 714         struct wpa_tdls_ftie *ftie;
 715         struct wpa_tdls_lnkid *lnkid;
 716         struct wpa_eapol_ie_parse kde;
 717         u16 reason_code;
 718         const u8 *pos;
 719         int ielen;
 720
 721         /* Find the node and free from the list */
 722         for (peer = sm->tdls; peer; peer = peer->next) {
 723                 if (os_memcmp(peer->addr, src_addr, ETH_ALEN) == 0)
 724                         break;
 725         }
 726
 727         if (peer == NULL) {
 728                 wpa_printf(MSG_INFO, "TDLS: No matching entry found for "
 729                            "Teardown " MACSTR, MAC2STR(src_addr));
 730                 return 0;
 731         }
 732
 733         pos = buf;
 734         pos += 1 /* pkt_type */ + 1 /* Category */ + 1 /* Action */;
 735
 736         reason_code = WPA_GET_LE16(pos);
 737         pos += 2;
 738
 739         wpa_printf(MSG_DEBUG, "TDLS: TDLS Teardown Request from " MACSTR
 740                    " (reason code %u)", MAC2STR(src_addr), reason_code);
 741
 742         ielen = len - (pos - buf); /* start of IE in buf */
 743         if (wpa_supplicant_parse_ies((const u8 *) pos, ielen, &kde) < 0) {
 744                 wpa_printf(MSG_INFO, "TDLS: Failed to parse IEs in Teardown");
 745                 return -1;
 746         }
 747
 748         if (kde.lnkid == NULL || kde.lnkid_len < 3 * ETH_ALEN) {
 749                 wpa_printf(MSG_INFO, "TDLS: No Link Identifier IE in TDLS "
 750                            "Teardown");
 751                 return -1;
 752         }
 753         lnkid = (struct wpa_tdls_lnkid *) kde.lnkid;
 754
 755         if (!wpa_tdls_get_privacy(sm) || !peer->tpk_set || !peer->tpk_success)
 756                 goto skip_ftie;
 757
 758         if (kde.ftie == NULL || kde.ftie_len < sizeof(*ftie)) {
 759                 wpa_printf(MSG_INFO, "TDLS: No FTIE in TDLS Teardown");
 760                 return -1;
 761         }
 762
 763         ftie = (struct wpa_tdls_ftie *) kde.ftie;
 764
 765         /* Process MIC check to see if TDLS Teardown is right */
 766         if (wpa_supplicant_verify_tdls_mic_teardown(4, reason_code,
 767                                                     peer->dtoken, peer,
 768                                                     (u8 *) lnkid, ftie) < 0) {
 769                 wpa_printf(MSG_DEBUG, "TDLS: MIC failure for TDLS "
 770                            "Teardown Request from " MACSTR, MAC2STR(src_addr));
 771                 return -1;
 772         }
 773
 774 skip_ftie:
 775         /*
 776          * Request the driver to disable the direct link and clear associated
 777          * keys.
 778          */
 779         wpa_sm_tdls_oper(sm, TDLS_DISABLE_LINK, src_addr);
 780
 781         /* clear the Peerkey statemachine */
 782         wpa_tdls_peer_free(sm, peer);
 783
 784         return 0;
 785 }
 786
 787
 788 /**
 789  * wpa_tdls_send_error - To send suitable TDLS status response with
 790  *      appropriate status code mentioning reason for error/failure.
 791  * @dst         - MAC addr of Peer station
 792  * @tdls_action - TDLS frame type for which error code is sent
 793  * @status      - status code mentioning reason
 794  */
 795
 796 static int wpa_tdls_send_error(struct wpa_sm *sm, const u8 *dst,
 797                                u8 tdls_action, u8 dialog_token, u16 status)
 798 {
 799         wpa_printf(MSG_DEBUG, "TDLS: Sending error to " MACSTR
 800                    " (action=%u status=%u)",
 801                    MAC2STR(dst), tdls_action, status);
 802         return wpa_tdls_tpk_send(sm, dst, tdls_action, dialog_token, status,
 803                                  NULL, 0);
 804 }
 805
 806
 807 static struct wpa_tdls_peer *
 808 wpa_tdls_add_peer(struct wpa_sm *sm, const u8 *addr)
 809 {
 810         struct wpa_tdls_peer *peer;
 811
 812         wpa_printf(MSG_INFO, "TDLS: Creating peer entry for " MACSTR,
 813                    MAC2STR(addr));
 814
 815         peer = os_zalloc(sizeof(*peer));
 816         if (peer == NULL)
 817                 return NULL;
 818
 819         os_memcpy(peer->addr, addr, ETH_ALEN);
 820         peer->next = sm->tdls;
 821         sm->tdls = peer;
 822
 823         return peer;
 824 }
 825
 826
 827 static int wpa_tdls_send_tpk_m1(struct wpa_sm *sm,
 828                                 struct wpa_tdls_peer *peer)
 829 {
 830         size_t buf_len;
 831         struct wpa_tdls_timeoutie timeoutie;
 832         u16 rsn_capab;
 833         struct wpa_tdls_ftie *ftie;
 834         u8 *rbuf, *pos, *count_pos;
 835         u16 count;
 836         struct rsn_ie_hdr *hdr;
 837
 838         if (!wpa_tdls_get_privacy(sm)) {
 839                 wpa_printf(MSG_DEBUG, "TDLS: No security used on the link");
 840                 peer->rsnie_i_len = 0;
 841                 goto skip_rsnie;
 842         }
 843
 844         /*
 845          * TPK Handshake Message 1:
 846          * FTIE: ANonce=0, SNonce=initiator nonce MIC=0, DataKDs=(RSNIE_I,
 847          * Timeout Interval IE))
 848          */
 849
 850         /* Filling RSN IE */
 851         hdr = (struct rsn_ie_hdr *) peer->rsnie_i;
 852         hdr->elem_id = WLAN_EID_RSN;
 853         WPA_PUT_LE16(hdr->version, RSN_VERSION);
 854
 855         pos = (u8 *) (hdr + 1);
 856         RSN_SELECTOR_PUT(pos, RSN_CIPHER_SUITE_NO_GROUP_ADDRESSED);
 857         pos += RSN_SELECTOR_LEN;
 858         count_pos = pos;
 859         pos += 2;
 860
 861         count = 0;
 862
 863         /*
 864          * AES-CCMP is the default Encryption preferred for TDLS, so
 865          * RSN IE is filled only with CCMP CIPHER
 866          * Note: TKIP is not used to encrypt TDLS link.
 867          *
 868          * Regardless of the cipher used on the AP connection, select CCMP
 869          * here.
 870          */
 871         RSN_SELECTOR_PUT(pos, RSN_CIPHER_SUITE_CCMP);
 872         pos += RSN_SELECTOR_LEN;
 873         count++;
 874
 875         WPA_PUT_LE16(count_pos, count);
 876
 877         WPA_PUT_LE16(pos, 1);
 878         pos += 2;
 879         RSN_SELECTOR_PUT(pos, RSN_AUTH_KEY_MGMT_TPK_HANDSHAKE);
 880         pos += RSN_SELECTOR_LEN;
 881
 882         rsn_capab = WPA_CAPABILITY_PEERKEY_ENABLED;
 883         rsn_capab |= RSN_NUM_REPLAY_COUNTERS_16 << 2;
 884 #ifdef CONFIG_TDLS_TESTING
 885         if (tdls_testing & TDLS_TESTING_ALT_RSN_IE) {
 886                 wpa_printf(MSG_DEBUG, "TDLS: Use alternative RSN IE for "
 887                            "testing");
 888                 rsn_capab = WPA_CAPABILITY_PEERKEY_ENABLED;
 889         }
 890 #endif /* CONFIG_TDLS_TESTING */
 891         WPA_PUT_LE16(pos, rsn_capab);
 892         pos += 2;
 893 #ifdef CONFIG_TDLS_TESTING
 894         if (tdls_testing & TDLS_TESTING_ALT_RSN_IE) {
 895                 /* Number of PMKIDs */
 896                 *pos++ = 0x00;
 897                 *pos++ = 0x00;
 898         }
 899 #endif /* CONFIG_TDLS_TESTING */
 900
 901         hdr->len = (pos - peer->rsnie_i) - 2;
 902         peer->rsnie_i_len = pos - peer->rsnie_i;
 903         wpa_hexdump(MSG_DEBUG, "TDLS: RSN IE for TPK handshake",
 904                     peer->rsnie_i, peer->rsnie_i_len);
 905
 906 skip_rsnie:
 907         buf_len = 0;
 908         if (wpa_tdls_get_privacy(sm))
 909                 buf_len += peer->rsnie_i_len + sizeof(struct wpa_tdls_ftie) +
 910                         sizeof(struct wpa_tdls_timeoutie);
 911 #ifdef CONFIG_TDLS_TESTING
 912         if (wpa_tdls_get_privacy(sm) &&
 913             (tdls_testing & TDLS_TESTING_LONG_FRAME))
 914                 buf_len += 170;
 915         if (tdls_testing & TDLS_TESTING_DIFF_BSSID)
 916                 buf_len += sizeof(struct wpa_tdls_lnkid);
 917 #endif /* CONFIG_TDLS_TESTING */
 918         rbuf = os_zalloc(buf_len + 1);
 919         if (rbuf == NULL) {
 920                 wpa_tdls_peer_free(sm, peer);
 921                 return -1;
 922         }
 923         pos = rbuf;
 924
 925         if (!wpa_tdls_get_privacy(sm))
 926                 goto skip_ies;
 927
 928         /* Initiator RSN IE */
 929         pos = wpa_add_ie(pos, peer->rsnie_i, peer->rsnie_i_len);
 930
 931         ftie = (struct wpa_tdls_ftie *) pos;
 932         ftie->ie_type = WLAN_EID_FAST_BSS_TRANSITION;
 933         ftie->ie_len = sizeof(struct wpa_tdls_ftie) - 2;
 934
 935         if (os_get_random(peer->inonce, WPA_NONCE_LEN)) {
 936                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
 937                         "TDLS: Failed to get random data for initiator Nonce");
 938                 os_free(rbuf);
 939                 wpa_tdls_peer_free(sm, peer);
 940                 return -1;
 941         }
 942         wpa_hexdump(MSG_DEBUG, "TDLS: Initiator Nonce for TPK handshake",
 943                     peer->inonce, WPA_NONCE_LEN);
 944         os_memcpy(ftie->Snonce, peer->inonce, WPA_NONCE_LEN);
 945
 946         wpa_hexdump(MSG_DEBUG, "TDLS: FTIE for TPK Handshake M1",
 947                     (u8 *) ftie, sizeof(struct wpa_tdls_ftie));
 948
 949         pos = (u8 *) (ftie + 1);
 950
 951 #ifdef CONFIG_TDLS_TESTING
 952         if (tdls_testing & TDLS_TESTING_LONG_FRAME) {
 953                 wpa_printf(MSG_DEBUG, "TDLS: Testing - add extra subelem to "
 954                            "FTIE");
 955                 ftie->ie_len += 170;
 956                 *pos++ = 255; /* FTIE subelem */
 957                 *pos++ = 168; /* FTIE subelem length */
 958                 pos += 168;
 959         }
 960 #endif /* CONFIG_TDLS_TESTING */
 961
 962         /* Lifetime */
 963         peer->lifetime = TPK_LIFETIME;
 964 #ifdef CONFIG_TDLS_TESTING
 965         if (tdls_testing & TDLS_TESTING_SHORT_LIFETIME) {
 966                 wpa_printf(MSG_DEBUG, "TDLS: Testing - use short TPK "
 967                            "lifetime");
 968                 peer->lifetime = 301;
 969         }
 970         if (tdls_testing & TDLS_TESTING_LONG_LIFETIME) {
 971                 wpa_printf(MSG_DEBUG, "TDLS: Testing - use long TPK "
 972                            "lifetime");
 973                 peer->lifetime = 0xffffffff;
 974         }
 975 #endif /* CONFIG_TDLS_TESTING */
 976         pos = wpa_add_tdls_timeoutie(pos, (u8 *) &timeoutie,
 977                                      sizeof(timeoutie), peer->lifetime);
 978         wpa_printf(MSG_DEBUG, "TDLS: TPK lifetime %u seconds", peer->lifetime);
 979
 980 skip_ies:
 981
 982 #ifdef CONFIG_TDLS_TESTING
 983         if (tdls_testing & TDLS_TESTING_DIFF_BSSID) {
 984                 wpa_printf(MSG_DEBUG, "TDLS: Testing - use incorrect BSSID in "
 985                            "Link Identifier");
 986                 struct wpa_tdls_lnkid *l = (struct wpa_tdls_lnkid *) pos;
 987                 wpa_tdls_linkid(sm, peer, l);
 988                 l->bssid[5] ^= 0x01;
 989                 pos += sizeof(*l);
 990         }
 991 #endif /* CONFIG_TDLS_TESTING */
 992
 993         wpa_printf(MSG_DEBUG, "TDLS: Sending TDLS Setup Request / TPK "
 994                    "Handshake Message 1 (peer " MACSTR ")",
 995                    MAC2STR(peer->addr));
 996
 997         wpa_tdls_tpk_send(sm, peer->addr, WLAN_TDLS_SETUP_REQUEST, 1, 0,
 998                           rbuf, pos - rbuf);
 999         os_free(rbuf);
1000
1001         return 0;
1002 }
1003
1004
1005 static int wpa_tdls_send_tpk_m2(struct wpa_sm *sm,
1006                                 const unsigned char *src_addr, u8 dtoken,
1007                                 struct wpa_tdls_lnkid *lnkid,
1008                                 const struct wpa_tdls_peer *peer)
1009 {
1010         u8 *rbuf, *pos;
1011         size_t buf_len;
1012         u32 lifetime;
1013         struct wpa_tdls_timeoutie timeoutie;
1014         struct wpa_tdls_ftie *ftie;
1015
1016         buf_len = 0;
1017         if (wpa_tdls_get_privacy(sm)) {
1018                 /* Peer RSN IE, FTIE(Initiator Nonce, Responder Nonce),
1019                  * Lifetime */
1020                 buf_len += peer->rsnie_i_len + sizeof(struct wpa_tdls_ftie) +
1021                         sizeof(struct wpa_tdls_timeoutie);
1022 #ifdef CONFIG_TDLS_TESTING
1023                 if (tdls_testing & TDLS_TESTING_LONG_FRAME)
1024                         buf_len += 170;
1025 #endif /* CONFIG_TDLS_TESTING */
1026         }
1027
1028         rbuf = os_zalloc(buf_len + 1);
1029         if (rbuf == NULL)
1030                 return -1;
1031         pos = rbuf;
1032
1033         if (!wpa_tdls_get_privacy(sm))
1034                 goto skip_ies;
1035
1036         /* Peer RSN IE */
1037         pos = wpa_add_ie(pos, peer->rsnie_p, peer->rsnie_p_len);
1038
1039         ftie = (struct wpa_tdls_ftie *) pos;
1040         ftie->ie_type = WLAN_EID_FAST_BSS_TRANSITION;
1041         /* TODO: ftie->mic_control to set 2-RESPONSE */
1042         os_memcpy(ftie->Anonce, peer->rnonce, WPA_NONCE_LEN);
1043         os_memcpy(ftie->Snonce, peer->inonce, WPA_NONCE_LEN);
1044         ftie->ie_len = sizeof(struct wpa_tdls_ftie) - 2;
1045         wpa_hexdump(MSG_DEBUG, "TDLS: FTIE for TPK M2",
1046                     (u8 *) ftie, sizeof(*ftie));
1047
1048         pos = (u8 *) (ftie + 1);
1049
1050 #ifdef CONFIG_TDLS_TESTING
1051         if (tdls_testing & TDLS_TESTING_LONG_FRAME) {
1052                 wpa_printf(MSG_DEBUG, "TDLS: Testing - add extra subelem to "
1053                            "FTIE");
1054                 ftie->ie_len += 170;
1055                 *pos++ = 255; /* FTIE subelem */
1056                 *pos++ = 168; /* FTIE subelem length */
1057                 pos += 168;
1058         }
1059 #endif /* CONFIG_TDLS_TESTING */
1060
1061         /* Lifetime */
1062         lifetime = peer->lifetime;
1063 #ifdef CONFIG_TDLS_TESTING
1064         if (tdls_testing & TDLS_TESTING_WRONG_LIFETIME_RESP) {
1065                 wpa_printf(MSG_DEBUG, "TDLS: Testing - use wrong TPK "
1066                            "lifetime in response");
1067                 lifetime++;
1068         }
1069 #endif /* CONFIG_TDLS_TESTING */
1070         pos = wpa_add_tdls_timeoutie(pos, (u8 *) &timeoutie,
1071                                      sizeof(timeoutie), lifetime);
1072         wpa_printf(MSG_DEBUG, "TDLS: TPK lifetime %u seconds from initiator",
1073                    lifetime);
1074
1075         /* compute MIC before sending */
1076         wpa_tdls_ftie_mic(peer->tpk.kck, 2, (u8 *) lnkid, peer->rsnie_p,
1077                           (u8 *) &timeoutie, (u8 *) ftie, ftie->mic);
1078
1079 skip_ies:
1080         wpa_tdls_tpk_send(sm, src_addr, WLAN_TDLS_SETUP_RESPONSE, dtoken, 0,
1081                           rbuf, pos - rbuf);
1082         os_free(rbuf);
1083
1084         return 0;
1085 }
1086
1087
1088 static int wpa_tdls_send_tpk_m3(struct wpa_sm *sm,
1089                                 const unsigned char *src_addr, u8 dtoken,
1090                                 struct wpa_tdls_lnkid *lnkid,
1091                                 const struct wpa_tdls_peer *peer)
1092 {
1093         u8 *rbuf, *pos;
1094         size_t buf_len;
1095         struct wpa_tdls_ftie *ftie;
1096         struct wpa_tdls_timeoutie timeoutie;
1097         u32 lifetime;
1098
1099         buf_len = 0;
1100         if (wpa_tdls_get_privacy(sm)) {
1101                 /* Peer RSN IE, FTIE(Initiator Nonce, Responder Nonce),
1102                  * Lifetime */
1103                 buf_len += peer->rsnie_i_len + sizeof(struct wpa_tdls_ftie) +
1104                         sizeof(struct wpa_tdls_timeoutie);
1105 #ifdef CONFIG_TDLS_TESTING
1106                 if (tdls_testing & TDLS_TESTING_LONG_FRAME)
1107                         buf_len += 170;
1108 #endif /* CONFIG_TDLS_TESTING */
1109         }
1110
1111         rbuf = os_zalloc(buf_len + 1);
1112         if (rbuf == NULL)
1113                 return -1;
1114         pos = rbuf;
1115
1116         if (!wpa_tdls_get_privacy(sm))
1117                 goto skip_ies;
1118
1119         /* Peer RSN IE */
1120         pos = wpa_add_ie(pos, peer->rsnie_p, peer->rsnie_p_len);
1121
1122         ftie = (struct wpa_tdls_ftie *) pos;
1123         ftie->ie_type = WLAN_EID_FAST_BSS_TRANSITION;
1124         /*TODO: ftie->mic_control to set 3-CONFIRM */
1125         os_memcpy(ftie->Anonce, peer->rnonce, WPA_NONCE_LEN);
1126         os_memcpy(ftie->Snonce, peer->inonce, WPA_NONCE_LEN);
1127         ftie->ie_len = sizeof(struct wpa_tdls_ftie) - 2;
1128
1129         pos = (u8 *) (ftie + 1);
1130
1131 #ifdef CONFIG_TDLS_TESTING
1132         if (tdls_testing & TDLS_TESTING_LONG_FRAME) {
1133                 wpa_printf(MSG_DEBUG, "TDLS: Testing - add extra subelem to "
1134                            "FTIE");
1135                 ftie->ie_len += 170;
1136                 *pos++ = 255; /* FTIE subelem */
1137                 *pos++ = 168; /* FTIE subelem length */
1138                 pos += 168;
1139         }
1140 #endif /* CONFIG_TDLS_TESTING */
1141
1142         /* Lifetime */
1143         lifetime = peer->lifetime;
1144 #ifdef CONFIG_TDLS_TESTING
1145         if (tdls_testing & TDLS_TESTING_WRONG_LIFETIME_CONF) {
1146                 wpa_printf(MSG_DEBUG, "TDLS: Testing - use wrong TPK "
1147                            "lifetime in confirm");
1148                 lifetime++;
1149         }
1150 #endif /* CONFIG_TDLS_TESTING */
1151         pos = wpa_add_tdls_timeoutie(pos, (u8 *) &timeoutie,
1152                                      sizeof(timeoutie), lifetime);
1153         wpa_printf(MSG_DEBUG, "TDLS: TPK lifetime %u seconds",
1154                    lifetime);
1155
1156         /* compute MIC before sending */
1157         wpa_tdls_ftie_mic(peer->tpk.kck, 3, (u8 *) lnkid, peer->rsnie_p,
1158                           (u8 *) &timeoutie, (u8 *) ftie, ftie->mic);
1159
1160 skip_ies:
1161         wpa_tdls_tpk_send(sm, src_addr, WLAN_TDLS_SETUP_CONFIRM, dtoken, 0,
1162                           rbuf, pos - rbuf);
1163         os_free(rbuf);
1164
1165         return 0;
1166 }
1167
1168
1169 static int wpa_tdls_send_discovery_response(struct wpa_sm *sm,
1170                                             struct wpa_tdls_peer *peer,
1171                                             u8 dialog_token)
1172 {
1173         wpa_printf(MSG_DEBUG, "TDLS: Sending TDLS Discovery Response "
1174                    "(peer " MACSTR ")", MAC2STR(peer->addr));
1175
1176         return wpa_tdls_tpk_send(sm, peer->addr, WLAN_TDLS_DISCOVERY_RESPONSE,
1177                                  dialog_token, 0, NULL, 0);
1178 }
1179
1180
1181 static int
1182 wpa_tdls_process_discovery_request(struct wpa_sm *sm, const u8 *addr,
1183                                    const u8 *buf, size_t len)
1184 {
1185         struct wpa_eapol_ie_parse kde;
1186         const struct wpa_tdls_lnkid *lnkid;
1187         struct wpa_tdls_peer *peer;
1188         size_t min_req_len = sizeof(struct wpa_tdls_frame) +
1189                 1 /* dialog token */ + sizeof(struct wpa_tdls_lnkid);
1190         u8 dialog_token;
1191
1192         wpa_printf(MSG_DEBUG, "TDLS: Discovery Request from " MACSTR,
1193                    MAC2STR(addr));
1194
1195         if (len < min_req_len) {
1196                 wpa_printf(MSG_DEBUG, "TDLS Discovery Request is too short: "
1197                            "%d", (int) len);
1198                 return -1;
1199         }
1200
1201         dialog_token = buf[sizeof(struct wpa_tdls_frame)];
1202
1203         if (wpa_supplicant_parse_ies(buf + sizeof(struct wpa_tdls_frame) + 1,
1204                                      len - (sizeof(struct wpa_tdls_frame) + 1),
1205                                      &kde) < 0)
1206                 return -1;
1207
1208         if (!kde.lnkid) {
1209                 wpa_printf(MSG_DEBUG, "TDLS: Link ID not found in Discovery "
1210                            "Request");
1211                 return -1;
1212         }
1213
1214         lnkid = (const struct wpa_tdls_lnkid *) kde.lnkid;
1215
1216         if (os_memcmp(sm->bssid, lnkid->bssid, ETH_ALEN) != 0) {
1217                 wpa_printf(MSG_DEBUG, "TDLS: Discovery Request from different "
1218                            " BSS " MACSTR, MAC2STR(lnkid->bssid));
1219                 return -1;
1220         }
1221
1222         peer = wpa_tdls_add_peer(sm, addr);
1223         if (peer == NULL)
1224                 return -1;
1225
1226         return wpa_tdls_send_discovery_response(sm, peer, dialog_token);
1227 }
1228
1229
1230 static int wpa_tdls_process_tpk_m1(struct wpa_sm *sm, const u8 *src_addr,
1231                                    const u8 *buf, size_t len)
1232 {
1233         struct wpa_tdls_peer *peer;
1234         struct wpa_eapol_ie_parse kde;
1235         struct wpa_ie_data ie;
1236         int cipher;
1237         const u8 *cpos;
1238         struct wpa_tdls_ftie *ftie = NULL;
1239         struct wpa_tdls_timeoutie *timeoutie;
1240         struct wpa_tdls_lnkid *lnkid;
1241         u32 lifetime = 0;
1242 #if 0
1243         struct rsn_ie_hdr *hdr;
1244         u8 *pos;
1245         u16 rsn_capab;
1246         u16 rsn_ver;
1247 #endif
1248         u8 dtoken;
1249         u16 ielen;
1250         u16 status = WLAN_STATUS_UNSPECIFIED_FAILURE;
1251         int tdls_prohibited = sm->tdls_prohibited;
1252
1253         if (len < 3 + 3)
1254                 return -1;
1255
1256         cpos = buf;
1257         cpos += 1 /* pkt_type */ + 1 /* Category */ + 1 /* Action */;
1258
1259         /* driver had already verified the frame format */
1260         dtoken = *cpos++; /* dialog token */
1261
1262         wpa_printf(MSG_INFO, "TDLS: Dialog Token in TPK M1 %d", dtoken);
1263
1264         cpos += 2; /* capability information */
1265
1266         ielen = len - (cpos - buf); /* start of IE in buf */
1267         if (wpa_supplicant_parse_ies(cpos, ielen, &kde) < 0) {
1268                 wpa_printf(MSG_INFO, "TDLS: Failed to parse IEs in TPK M1");
1269                 goto error;
1270         }
1271
1272         if (kde.lnkid == NULL || kde.lnkid_len < 3 * ETH_ALEN) {
1273                 wpa_printf(MSG_INFO, "TDLS: No valid Link Identifier IE in "
1274                            "TPK M1");
1275                 goto error;
1276         }
1277         wpa_hexdump(MSG_DEBUG, "TDLS: Link ID Received from TPK M1",
1278                     kde.lnkid, kde.lnkid_len);
1279         lnkid = (struct wpa_tdls_lnkid *) kde.lnkid;
1280         if (os_memcmp(sm->bssid, lnkid->bssid, ETH_ALEN) != 0) {
1281                 wpa_printf(MSG_INFO, "TDLS: TPK M1 from diff BSS");
1282                 status = WLAN_STATUS_NOT_IN_SAME_BSS;
1283                 goto error;
1284         }
1285
1286         wpa_printf(MSG_DEBUG, "TDLS: TPK M1 - TPK initiator " MACSTR,
1287                    MAC2STR(src_addr));
1288
1289 #ifdef CONFIG_TDLS_TESTING
1290         if (tdls_testing & TDLS_TESTING_CONCURRENT_INIT) {
1291                 for (peer = sm->tdls; peer; peer = peer->next) {
1292                         if (os_memcmp(peer->addr, src_addr, ETH_ALEN) == 0)
1293                                 break;
1294                 }
1295                 if (peer == NULL) {
1296                         peer = os_zalloc(sizeof(*peer));
1297                         if (peer == NULL)
1298                                 goto error;
1299                         os_memcpy(peer->addr, src_addr, ETH_ALEN);
1300                         peer->next = sm->tdls;
1301                         sm->tdls = peer;
1302                 }
1303                 wpa_printf(MSG_DEBUG, "TDLS: Testing concurrent initiation of "
1304                            "TDLS setup - send own request");
1305                 peer->initiator = 1;
1306                 wpa_tdls_send_tpk_m1(sm, peer);
1307         }
1308
1309         if ((tdls_testing & TDLS_TESTING_IGNORE_AP_PROHIBIT) &&
1310             tdls_prohibited) {
1311                 wpa_printf(MSG_DEBUG, "TDLS: Testing - ignore AP prohibition "
1312                            "on TDLS");
1313                 tdls_prohibited = 0;
1314         }
1315 #endif /* CONFIG_TDLS_TESTING */
1316
1317         if (tdls_prohibited) {
1318                 wpa_printf(MSG_INFO, "TDLS: TDLS prohibited in this BSS");
1319                 status = WLAN_STATUS_REQUEST_DECLINED;
1320                 goto error;
1321         }
1322
1323         if (!wpa_tdls_get_privacy(sm)) {
1324                 if (kde.rsn_ie) {
1325                         wpa_printf(MSG_INFO, "TDLS: RSN IE in TPK M1 while "
1326                                    "security is disabled");
1327                         status = WLAN_STATUS_SECURITY_DISABLED;
1328                         goto error;
1329                 }
1330                 goto skip_rsn;
1331         }
1332
1333         if (kde.ftie == NULL || kde.ftie_len < sizeof(*ftie) ||
1334             kde.rsn_ie == NULL) {
1335                 wpa_printf(MSG_INFO, "TDLS: No FTIE or RSN IE in TPK M1");
1336                 status = WLAN_STATUS_INVALID_PARAMETERS;
1337                 goto error;
1338         }
1339
1340         if (kde.rsn_ie_len > TDLS_MAX_IE_LEN) {
1341                 wpa_printf(MSG_INFO, "TDLS: Too long Initiator RSN IE in "
1342                            "TPK M1");
1343                 status = WLAN_STATUS_INVALID_RSNIE;
1344                 goto error;
1345         }
1346
1347         if (wpa_parse_wpa_ie_rsn(kde.rsn_ie, kde.rsn_ie_len, &ie) < 0) {
1348                 wpa_printf(MSG_INFO, "TDLS: Failed to parse RSN IE in TPK M1");
1349                 status = WLAN_STATUS_INVALID_RSNIE;
1350                 goto error;
1351         }
1352
1353         cipher = ie.pairwise_cipher;
1354         if (cipher & WPA_CIPHER_CCMP) {
1355                 wpa_printf(MSG_DEBUG, "TDLS: Using CCMP for direct link");
1356                 cipher = WPA_CIPHER_CCMP;
1357         } else {
1358                 wpa_printf(MSG_INFO, "TDLS: No acceptable cipher in TPK M1");
1359                 status = WLAN_STATUS_PAIRWISE_CIPHER_NOT_VALID;
1360                 goto error;
1361         }
1362
1363         if ((ie.capabilities &
1364              (WPA_CAPABILITY_NO_PAIRWISE | WPA_CAPABILITY_PEERKEY_ENABLED)) !=
1365             WPA_CAPABILITY_PEERKEY_ENABLED) {
1366                 wpa_printf(MSG_INFO, "TDLS: Invalid RSN Capabilities in "
1367                            "TPK M1");
1368                 status = WLAN_STATUS_INVALID_RSN_IE_CAPAB;
1369                 goto error;
1370         }
1371
1372         /* Lifetime */
1373         if (kde.key_lifetime == NULL) {
1374                 wpa_printf(MSG_INFO, "TDLS: No Key Lifetime IE in TPK M1");
1375                 status = WLAN_STATUS_UNACCEPTABLE_LIFETIME;
1376                 goto error;
1377         }
1378         timeoutie = (struct wpa_tdls_timeoutie *) kde.key_lifetime;
1379         lifetime = WPA_GET_LE32(timeoutie->value);
1380         wpa_printf(MSG_DEBUG, "TDLS: TPK lifetime %u seconds", lifetime);
1381         if (lifetime < 300) {
1382                 wpa_printf(MSG_INFO, "TDLS: Too short TPK lifetime");
1383                 status = WLAN_STATUS_UNACCEPTABLE_LIFETIME;
1384                 goto error;
1385         }
1386
1387 skip_rsn:
1388         /* Find existing entry and if found, use that instead of adding
1389          * a new one; how to handle the case where both ends initiate at the
1390          * same time? */
1391         for (peer = sm->tdls; peer; peer = peer->next) {
1392                 if (os_memcmp(peer->addr, src_addr, ETH_ALEN) == 0)
1393                         break;
1394         }
1395
1396         if (peer == NULL) {
1397                 wpa_printf(MSG_INFO, "TDLS: No matching entry found for "
1398                            "peer, creating one for " MACSTR,
1399                            MAC2STR(src_addr));
1400                 peer = os_malloc(sizeof(*peer));
1401                 if (peer == NULL)
1402                         goto error;
1403                 os_memset(peer, 0, sizeof(*peer));
1404                 os_memcpy(peer->addr, src_addr, ETH_ALEN);
1405                 peer->next = sm->tdls;
1406                 sm->tdls = peer;
1407         } else {
1408                 if (peer->tpk_success) {
1409                         wpa_printf(MSG_DEBUG, "TDLS: TDLS Setup Request while "
1410                                    "direct link is enabled - tear down the "
1411                                    "old link first");
1412 #if 0
1413                         /* TODO: Disabling the link would be more proper
1414                          * operation here, but it seems to trigger a race with
1415                          * some drivers handling the new request frame. */
1416                         wpa_sm_tdls_oper(sm, TDLS_DISABLE_LINK, src_addr);
1417 #else
1418                         wpa_tdls_del_key(sm, peer);
1419 #endif
1420                         wpa_tdls_peer_free(sm, peer);
1421                 }
1422
1423                 /*
1424                  * An entry is already present, so check if we already sent a
1425                  * TDLS Setup Request. If so, compare MAC addresses and let the
1426                  * STA with the lower MAC address continue as the initiator.
1427                  * The other negotiation is terminated.
1428                  */
1429                 if (peer->initiator) {
1430                         if (os_memcmp(sm->own_addr, src_addr, ETH_ALEN) < 0) {
1431                                 wpa_printf(MSG_DEBUG, "TDLS: Discard request "
1432                                            "from peer with higher address "
1433                                            MACSTR, MAC2STR(src_addr));
1434                                 return -1;
1435                         } else {
1436                                 wpa_printf(MSG_DEBUG, "TDLS: Accept request "
1437                                            "from peer with lower address "
1438                                            MACSTR " (terminate previously "
1439                                            "initiated negotiation",
1440                                            MAC2STR(src_addr));
1441                                 wpa_tdls_peer_free(sm, peer);
1442                         }
1443                 }
1444         }
1445
1446         peer->initiator = 0; /* Need to check */
1447         peer->dtoken = dtoken;
1448
1449         if (!wpa_tdls_get_privacy(sm)) {
1450                 peer->rsnie_i_len = 0;
1451                 peer->rsnie_p_len = 0;
1452                 peer->cipher = WPA_CIPHER_NONE;
1453                 goto skip_rsn_check;
1454         }
1455
1456         ftie = (struct wpa_tdls_ftie *) kde.ftie;
1457         os_memcpy(peer->inonce, ftie->Snonce, WPA_NONCE_LEN);
1458         os_memcpy(peer->rsnie_i, kde.rsn_ie, kde.rsn_ie_len);
1459         peer->rsnie_i_len = kde.rsn_ie_len;
1460         peer->cipher = cipher;
1461
1462         if (os_get_random(peer->rnonce, WPA_NONCE_LEN)) {
1463                 wpa_msg(sm->ctx->ctx, MSG_WARNING,
1464                         "TDLS: Failed to get random data for responder nonce");
1465                 wpa_tdls_peer_free(sm, peer);
1466                 goto error;
1467         }
1468
1469 #if 0
1470         /* get version info from RSNIE received from Peer */
1471         hdr = (struct rsn_ie_hdr *) kde.rsn_ie;
1472         rsn_ver = WPA_GET_LE16(hdr->version);
1473
1474         /* use min(peer's version, out version) */
1475         if (rsn_ver > RSN_VERSION)
1476                 rsn_ver = RSN_VERSION;
1477
1478         hdr = (struct rsn_ie_hdr *) peer->rsnie_p;
1479
1480         hdr->elem_id = WLAN_EID_RSN;
1481         WPA_PUT_LE16(hdr->version, rsn_ver);
1482         pos = (u8 *) (hdr + 1);
1483
1484         RSN_SELECTOR_PUT(pos, RSN_CIPHER_SUITE_NO_GROUP_ADDRESSED);
1485         pos += RSN_SELECTOR_LEN;
1486         /* Include only the selected cipher in pairwise cipher suite */
1487         WPA_PUT_LE16(pos, 1);
1488         pos += 2;
1489         if (cipher == WPA_CIPHER_CCMP)
1490                 RSN_SELECTOR_PUT(pos, RSN_CIPHER_SUITE_CCMP);
1491         pos += RSN_SELECTOR_LEN;
1492
1493         WPA_PUT_LE16(pos, 1);
1494         pos += 2;
1495         RSN_SELECTOR_PUT(pos, RSN_AUTH_KEY_MGMT_TPK_HANDSHAKE);
1496         pos += RSN_SELECTOR_LEN;
1497
1498         rsn_capab = WPA_CAPABILITY_PEERKEY_ENABLED;
1499         rsn_capab |= RSN_NUM_REPLAY_COUNTERS_16 << 2;
1500         WPA_PUT_LE16(pos, rsn_capab);
1501         pos += 2;
1502
1503         hdr->len = (pos - peer->rsnie_p) - 2;
1504         peer->rsnie_p_len = pos - peer->rsnie_p;
1505 #endif
1506
1507         /* temp fix: validation of RSNIE later */
1508         os_memcpy(peer->rsnie_p, peer->rsnie_i, peer->rsnie_i_len);
1509         peer->rsnie_p_len = peer->rsnie_i_len;
1510
1511         wpa_hexdump(MSG_DEBUG, "TDLS: RSN IE for TPK handshake",
1512                     peer->rsnie_p, peer->rsnie_p_len);
1513
1514         peer->lifetime = lifetime;
1515
1516         wpa_tdls_generate_tpk(peer, sm->own_addr, sm->bssid);
1517
1518 skip_rsn_check:
1519         wpa_printf(MSG_DEBUG, "TDLS: Sending TDLS Setup Response / TPK M2");
1520         wpa_tdls_send_tpk_m2(sm, src_addr, dtoken, lnkid, peer);
1521
1522         return 0;
1523
1524 error:
1525         wpa_tdls_send_error(sm, src_addr, WLAN_TDLS_SETUP_RESPONSE, dtoken,
1526                             status);
1527         return -1;
1528 }
1529
1530
1531 static void wpa_tdls_enable_link(struct wpa_sm *sm, struct wpa_tdls_peer *peer)
1532 {
1533         peer->tpk_success = 1;
1534         eloop_cancel_timeout(wpa_tdls_tpk_timeout, sm, peer);
1535         if (wpa_tdls_get_privacy(sm)) {
1536                 u32 lifetime = peer->lifetime;
1537                 /*
1538                  * Start the initiator process a bit earlier to avoid race
1539                  * condition with the responder sending teardown request.
1540                  */
1541                 if (lifetime > 3 && peer->initiator)
1542                         lifetime -= 3;
1543                 eloop_register_timeout(lifetime, 0, wpa_tdls_tpk_timeout,
1544                                        sm, peer);
1545 #ifdef CONFIG_TDLS_TESTING
1546         if (tdls_testing & TDLS_TESTING_NO_TPK_EXPIRATION) {
1547                 wpa_printf(MSG_DEBUG, "TDLS: Testing - disable TPK "
1548                            "expiration");
1549                 eloop_cancel_timeout(wpa_tdls_tpk_timeout, sm, peer);
1550         }
1551 #endif /* CONFIG_TDLS_TESTING */
1552         }
1553         wpa_sm_tdls_oper(sm, TDLS_ENABLE_LINK, peer->addr);
1554 }
1555
1556
1557 static int wpa_tdls_process_tpk_m2(struct wpa_sm *sm, const u8 *src_addr,
1558                                    const u8 *buf, size_t len)
1559 {
1560         struct wpa_tdls_peer *peer;
1561         struct wpa_eapol_ie_parse kde;
1562         struct wpa_ie_data ie;
1563         int cipher;
1564         struct wpa_tdls_ftie *ftie;
1565         struct wpa_tdls_timeoutie *timeoutie;
1566         struct wpa_tdls_lnkid *lnkid;
1567         u32 lifetime;
1568         u8 dtoken;
1569         int ielen;
1570         u16 status;
1571         const u8 *pos;
1572
1573         wpa_printf(MSG_DEBUG, "TDLS: Received TDLS Setup Response / TPK M2 "
1574                    "(Peer " MACSTR ")", MAC2STR(src_addr));
1575         for (peer = sm->tdls; peer; peer = peer->next) {
1576                 if (os_memcmp(peer->addr, src_addr, ETH_ALEN) == 0)
1577                         break;
1578         }
1579         if (peer == NULL) {
1580                 wpa_printf(MSG_INFO, "TDLS: No matching peer found for "
1581                            "TPK M2: " MACSTR, MAC2STR(src_addr));
1582                 return -1;
1583         }
1584         wpa_tdls_tpk_retry_timeout_cancel(sm, peer, WLAN_TDLS_SETUP_REQUEST);
1585
1586         if (len < 3 + 2 + 1)
1587                 return -1;
1588         pos = buf;
1589         pos += 1 /* pkt_type */ + 1 /* Category */ + 1 /* Action */;
1590         status = WPA_GET_LE16(pos);
1591         pos += 2 /* status code */;
1592
1593         if (status != WLAN_STATUS_SUCCESS) {
1594                 wpa_printf(MSG_INFO, "TDLS: Status code in TPK M2: %u",
1595                            status);
1596                 return -1;
1597         }
1598
1599         status = WLAN_STATUS_UNSPECIFIED_FAILURE;
1600
1601         /* TODO: need to verify dialog token matches here or in kernel */
1602         dtoken = *pos++; /* dialog token */
1603
1604         wpa_printf(MSG_DEBUG, "TDLS: Dialog Token in TPK M2 %d", dtoken);
1605
1606         if (len < 3 + 2 + 1 + 2)
1607                 return -1;
1608         pos += 2; /* capability information */
1609
1610         ielen = len - (pos - buf); /* start of IE in buf */
1611         if (wpa_supplicant_parse_ies(pos, ielen, &kde) < 0) {
1612                 wpa_printf(MSG_INFO, "TDLS: Failed to parse IEs in TPK M2");
1613                 goto error;
1614         }
1615
1616 #ifdef CONFIG_TDLS_TESTING
1617         if (tdls_testing & TDLS_TESTING_DECLINE_RESP) {
1618                 wpa_printf(MSG_DEBUG, "TDLS: Testing - decline response");
1619                 status = WLAN_STATUS_REQUEST_DECLINED;
1620                 goto error;
1621         }
1622 #endif /* CONFIG_TDLS_TESTING */
1623
1624         if (kde.lnkid == NULL || kde.lnkid_len < 3 * ETH_ALEN) {
1625                 wpa_printf(MSG_INFO, "TDLS: No valid Link Identifier IE in "
1626                            "TPK M2");
1627                 goto error;
1628         }
1629         wpa_hexdump(MSG_DEBUG, "TDLS: Link ID Received from TPK M2",
1630                     kde.lnkid, kde.lnkid_len);
1631         lnkid = (struct wpa_tdls_lnkid *) kde.lnkid;
1632
1633         if (os_memcmp(sm->bssid, lnkid->bssid, ETH_ALEN) != 0) {
1634                 wpa_printf(MSG_INFO, "TDLS: TPK M2 from different BSS");
1635                 status = WLAN_STATUS_NOT_IN_SAME_BSS;
1636                 goto error;
1637         }
1638
1639         if (!wpa_tdls_get_privacy(sm)) {
1640                 peer->rsnie_p_len = 0;
1641                 peer->cipher = WPA_CIPHER_NONE;
1642                 goto skip_rsn;
1643         }
1644
1645         if (kde.ftie == NULL || kde.ftie_len < sizeof(*ftie) ||
1646             kde.rsn_ie == NULL) {
1647                 wpa_printf(MSG_INFO, "TDLS: No FTIE or RSN IE in TPK M2");
1648                 status = WLAN_STATUS_INVALID_PARAMETERS;
1649                 goto error;
1650         }
1651         wpa_hexdump(MSG_DEBUG, "TDLS: RSN IE Received from TPK M2",
1652                     kde.rsn_ie, kde.rsn_ie_len);
1653
1654         /*
1655          * FIX: bitwise comparison of RSN IE is not the correct way of
1656          * validation this. It can be different, but certain fields must
1657          * match. Since we list only a single pairwise cipher in TPK M1, the
1658          * memcmp is likely to work in most cases, though.
1659          */
1660         if (kde.rsn_ie_len != peer->rsnie_i_len ||
1661             os_memcmp(peer->rsnie_i, kde.rsn_ie, peer->rsnie_i_len) != 0) {
1662                 wpa_printf(MSG_INFO, "TDLS: RSN IE in TPK M2 does "
1663                            "not match with RSN IE used in TPK M1");
1664                 wpa_hexdump(MSG_DEBUG, "TDLS: RSN IE Sent in TPK M1",
1665                             peer->rsnie_i, peer->rsnie_i_len);
1666                 wpa_hexdump(MSG_DEBUG, "TDLS: RSN IE Received from TPK M2",
1667                             kde.rsn_ie, kde.rsn_ie_len);
1668                 status = WLAN_STATUS_INVALID_RSNIE;
1669                 goto error;
1670         }
1671
1672         if (wpa_parse_wpa_ie_rsn(kde.rsn_ie, kde.rsn_ie_len, &ie) < 0) {
1673                 wpa_printf(MSG_INFO, "TDLS: Failed to parse RSN IE in TPK M2");
1674                 status = WLAN_STATUS_INVALID_RSNIE;
1675                 goto error;
1676         }
1677
1678         cipher = ie.pairwise_cipher;
1679         if (cipher == WPA_CIPHER_CCMP) {
1680                 wpa_printf(MSG_DEBUG, "TDLS: Using CCMP for direct link");
1681                 cipher = WPA_CIPHER_CCMP;
1682         } else {
1683                 wpa_printf(MSG_INFO, "TDLS: No acceptable cipher in TPK M2");
1684                 status = WLAN_STATUS_PAIRWISE_CIPHER_NOT_VALID;
1685                 goto error;
1686         }
1687
1688         wpa_hexdump(MSG_DEBUG, "TDLS: FTIE Received from TPK M2",
1689                     kde.ftie, sizeof(*ftie));
1690         ftie = (struct wpa_tdls_ftie *) kde.ftie;
1691
1692         if (!os_memcmp(peer->inonce, ftie->Snonce, WPA_NONCE_LEN) == 0) {
1693                 wpa_printf(MSG_INFO, "TDLS: FTIE SNonce in TPK M2 does "
1694                            "not match with FTIE SNonce used in TPK M1");
1695                 /* Silently discard the frame */
1696                 return -1;
1697         }
1698
1699         /* Responder Nonce and RSN IE */
1700         os_memcpy(peer->rnonce, ftie->Anonce, WPA_NONCE_LEN);
1701         os_memcpy(peer->rsnie_p, kde.rsn_ie, kde.rsn_ie_len);
1702         peer->rsnie_p_len = kde.rsn_ie_len;
1703         peer->cipher = cipher;
1704
1705         /* Lifetime */
1706         if (kde.key_lifetime == NULL) {
1707                 wpa_printf(MSG_INFO, "TDLS: No Key Lifetime IE in TPK M2");
1708                 status = WLAN_STATUS_UNACCEPTABLE_LIFETIME;
1709                 goto error;
1710         }
1711         timeoutie = (struct wpa_tdls_timeoutie *) kde.key_lifetime;
1712         lifetime = WPA_GET_LE32(timeoutie->value);
1713         wpa_printf(MSG_DEBUG, "TDLS: TPK lifetime %u seconds in TPK M2",
1714                    lifetime);
1715         if (lifetime != peer->lifetime) {
1716                 wpa_printf(MSG_INFO, "TDLS: Unexpected TPK lifetime %u in "
1717                            "TPK M2 (expected %u)", lifetime, peer->lifetime);
1718                 status = WLAN_STATUS_UNACCEPTABLE_LIFETIME;
1719                 goto error;
1720         }
1721
1722         wpa_tdls_generate_tpk(peer, sm->own_addr, sm->bssid);
1723
1724         /* Process MIC check to see if TPK M2 is right */
1725         if (wpa_supplicant_verify_tdls_mic(2, peer, (u8 *) lnkid,
1726                                            (u8 *) timeoutie, ftie) < 0) {
1727                 /* Discard the frame */
1728                 wpa_tdls_del_key(sm, peer);
1729                 wpa_tdls_peer_free(sm, peer);
1730                 return -1;
1731         }
1732
1733         wpa_tdls_set_key(sm, peer);
1734
1735 skip_rsn:
1736         peer->dtoken = dtoken;
1737
1738         wpa_printf(MSG_DEBUG, "TDLS: Sending TDLS Setup Confirm / "
1739                    "TPK Handshake Message 3");
1740         wpa_tdls_send_tpk_m3(sm, src_addr, dtoken, lnkid, peer);
1741
1742         wpa_tdls_enable_link(sm, peer);
1743
1744         return 0;
1745
1746 error:
1747         wpa_tdls_send_error(sm, src_addr, WLAN_TDLS_SETUP_CONFIRM, dtoken,
1748                             status);
1749         return -1;
1750 }
1751
1752
1753 static int wpa_tdls_process_tpk_m3(struct wpa_sm *sm, const u8 *src_addr,
1754                                    const u8 *buf, size_t len)
1755 {
1756         struct wpa_tdls_peer *peer;
1757         struct wpa_eapol_ie_parse kde;
1758         struct wpa_tdls_ftie *ftie;
1759         struct wpa_tdls_timeoutie *timeoutie;
1760         struct wpa_tdls_lnkid *lnkid;
1761         int ielen;
1762         u16 status;
1763         const u8 *pos;
1764         u32 lifetime;
1765
1766         wpa_printf(MSG_DEBUG, "TDLS: Received TDLS Setup Confirm / TPK M3 "
1767                    "(Peer " MACSTR ")", MAC2STR(src_addr));
1768         for (peer = sm->tdls; peer; peer = peer->next) {
1769                 if (os_memcmp(peer->addr, src_addr, ETH_ALEN) == 0)
1770                         break;
1771         }
1772         if (peer == NULL) {
1773                 wpa_printf(MSG_INFO, "TDLS: No matching peer found for "
1774                            "TPK M3: " MACSTR, MAC2STR(src_addr));
1775                 return -1;
1776         }
1777         wpa_tdls_tpk_retry_timeout_cancel(sm, peer, WLAN_TDLS_SETUP_RESPONSE);
1778
1779         if (len < 3 + 3)
1780                 return -1;
1781         pos = buf;
1782         pos += 1 /* pkt_type */ + 1 /* Category */ + 1 /* Action */;
1783
1784         status = WPA_GET_LE16(pos);
1785
1786         if (status != 0) {
1787                 wpa_printf(MSG_INFO, "TDLS: Status code in TPK M3: %u",
1788                            status);
1789                 return -1;
1790         }
1791         pos += 2 /* status code */ + 1 /* dialog token */;
1792
1793         ielen = len - (pos - buf); /* start of IE in buf */
1794         if (wpa_supplicant_parse_ies((const u8 *) pos, ielen, &kde) < 0) {
1795                 wpa_printf(MSG_INFO, "TDLS: Failed to parse KDEs in TPK M3");
1796                 return -1;
1797         }
1798
1799         if (kde.lnkid == NULL || kde.lnkid_len < 3 * ETH_ALEN) {
1800                 wpa_printf(MSG_INFO, "TDLS: No Link Identifier IE in TPK M3");
1801                 return -1;
1802         }
1803         wpa_hexdump(MSG_DEBUG, "TDLS: Link ID Received from TPK M3",
1804                     (u8 *) kde.lnkid, kde.lnkid_len);
1805         lnkid = (struct wpa_tdls_lnkid *) kde.lnkid;
1806
1807         if (os_memcmp(sm->bssid, lnkid->bssid, ETH_ALEN) != 0) {
1808                 wpa_printf(MSG_INFO, "TDLS: TPK M3 from diff BSS");
1809                 return -1;
1810         }
1811
1812         if (!wpa_tdls_get_privacy(sm))
1813                 goto skip_rsn;
1814
1815         if (kde.ftie == NULL || kde.ftie_len < sizeof(*ftie)) {
1816                 wpa_printf(MSG_INFO, "TDLS: No FTIE in TPK M3");
1817                 return -1;
1818         }
1819         wpa_hexdump(MSG_DEBUG, "TDLS: FTIE Received from TPK M3",
1820                     kde.ftie, sizeof(*ftie));
1821         ftie = (struct wpa_tdls_ftie *) kde.ftie;
1822
1823         if (kde.rsn_ie == NULL) {
1824                 wpa_printf(MSG_INFO, "TDLS: No RSN IE in TPK M3");
1825                 return -1;
1826         }
1827         wpa_hexdump(MSG_DEBUG, "TDLS: RSN IE Received from TPK M3",
1828                     kde.rsn_ie, kde.rsn_ie_len);
1829         if (kde.rsn_ie_len != peer->rsnie_p_len ||
1830             os_memcmp(kde.rsn_ie, peer->rsnie_p, peer->rsnie_p_len) != 0) {
1831                 wpa_printf(MSG_INFO, "TDLS: RSN IE in TPK M3 does not match "
1832                            "with the one sent in TPK M2");
1833                 return -1;
1834         }
1835
1836         if (!os_memcmp(peer->rnonce, ftie->Anonce, WPA_NONCE_LEN) == 0) {
1837                 wpa_printf(MSG_INFO, "TDLS: FTIE ANonce in TPK M3 does "
1838                            "not match with FTIE ANonce used in TPK M2");
1839                 return -1;
1840         }
1841
1842         if (!os_memcmp(peer->inonce, ftie->Snonce, WPA_NONCE_LEN) == 0) {
1843                 wpa_printf(MSG_INFO, "TDLS: FTIE SNonce in TPK M3 does not "
1844                            "match with FTIE SNonce used in TPK M1");
1845                 return -1;
1846         }
1847
1848         if (kde.key_lifetime == NULL) {
1849                 wpa_printf(MSG_INFO, "TDLS: No Key Lifetime IE in TPK M3");
1850                 return -1;
1851         }
1852         timeoutie = (struct wpa_tdls_timeoutie *) kde.key_lifetime;
1853         wpa_hexdump(MSG_DEBUG, "TDLS: Timeout IE Received from TPK M3",
1854                     (u8 *) timeoutie, sizeof(*timeoutie));
1855         lifetime = WPA_GET_LE32(timeoutie->value);
1856         wpa_printf(MSG_DEBUG, "TDLS: TPK lifetime %u seconds in TPK M3",
1857                    lifetime);
1858         if (lifetime != peer->lifetime) {
1859                 wpa_printf(MSG_INFO, "TDLS: Unexpected TPK lifetime %u in "
1860                            "TPK M3 (expected %u)", lifetime, peer->lifetime);
1861                 return -1;
1862         }
1863
1864         if (wpa_supplicant_verify_tdls_mic(3, peer, (u8 *) lnkid,
1865                                            (u8 *) timeoutie, ftie) < 0) {
1866                 wpa_tdls_del_key(sm, peer);
1867                 wpa_tdls_peer_free(sm, peer);
1868                 return -1;
1869         }
1870
1871         if (wpa_tdls_set_key(sm, peer) < 0)
1872                 return -1;
1873
1874 skip_rsn:
1875         wpa_tdls_enable_link(sm, peer);
1876
1877         return 0;
1878 }
1879
1880
1881 static u8 * wpa_add_tdls_timeoutie(u8 *pos, u8 *ie, size_t ie_len, u32 tsecs)
1882 {
1883         struct wpa_tdls_timeoutie *lifetime = (struct wpa_tdls_timeoutie *) ie;
1884
1885         os_memset(lifetime, 0, ie_len);
1886         lifetime->ie_type = WLAN_EID_TIMEOUT_INTERVAL;
1887         lifetime->ie_len = sizeof(struct wpa_tdls_timeoutie) - 2;
1888         lifetime->interval_type = WLAN_TIMEOUT_KEY_LIFETIME;
1889         WPA_PUT_LE32(lifetime->value, tsecs);
1890         os_memcpy(pos, ie, ie_len);
1891         return pos + ie_len;
1892 }
1893
1894
1895 /**
1896  * wpa_tdls_start - Initiate TDLS handshake (send TPK Handshake Message 1)
1897  * @sm: Pointer to WPA state machine data from wpa_sm_init()
1898  * @peer: MAC address of the peer STA
1899  * Returns: 0 on success, or -1 on failure
1900  *
1901  * Send TPK Handshake Message 1 info to driver to start TDLS
1902  * handshake with the peer.
1903  */
1904 int wpa_tdls_start(struct wpa_sm *sm, const u8 *addr)
1905 {
1906         struct wpa_tdls_peer *peer;
1907         int tdls_prohibited = sm->tdls_prohibited;
1908
1909         if (sm->tdls_disabled || !sm->tdls_supported)
1910                 return -1;
1911
1912 #ifdef CONFIG_TDLS_TESTING
1913         if ((tdls_testing & TDLS_TESTING_IGNORE_AP_PROHIBIT) &&
1914             tdls_prohibited) {
1915                 wpa_printf(MSG_DEBUG, "TDLS: Testing - ignore AP prohibition "
1916                            "on TDLS");
1917                 tdls_prohibited = 0;
1918         }
1919 #endif /* CONFIG_TDLS_TESTING */
1920
1921         if (tdls_prohibited) {
1922                 wpa_printf(MSG_DEBUG, "TDLS: TDLS is prohibited in this BSS - "
1923                            "reject request to start setup");
1924                 return -1;
1925         }
1926
1927         /* Find existing entry and if found, use that instead of adding
1928          * a new one */
1929         for (peer = sm->tdls; peer; peer = peer->next) {
1930                 if (os_memcmp(peer->addr, addr, ETH_ALEN) == 0)
1931                         break;
1932         }
1933
1934         if (peer == NULL) {
1935                 wpa_printf(MSG_INFO, "TDLS: No matching entry found for "
1936                            "peer, creating one for " MACSTR, MAC2STR(addr));
1937                 peer = os_malloc(sizeof(*peer));
1938                 if (peer == NULL)
1939                         return -1;
1940                 os_memset(peer, 0, sizeof(*peer));
1941                 os_memcpy(peer->addr, addr, ETH_ALEN);
1942                 peer->next = sm->tdls;
1943                 sm->tdls = peer;
1944         }
1945
1946         peer->initiator = 1;
1947
1948         return wpa_tdls_send_tpk_m1(sm, peer);
1949 }
1950
1951
1952 int wpa_tdls_reneg(struct wpa_sm *sm, const u8 *addr)
1953 {
1954         struct wpa_tdls_peer *peer;
1955
1956         if (sm->tdls_disabled || !sm->tdls_supported)
1957                 return -1;
1958
1959         for (peer = sm->tdls; peer; peer = peer->next) {
1960                 if (os_memcmp(peer->addr, addr, ETH_ALEN) == 0)
1961                         break;
1962         }
1963
1964         if (peer == NULL || !peer->tpk_success)
1965                 return -1;
1966
1967         return wpa_tdls_start(sm, addr);
1968 }
1969
1970
1971 /**
1972  * wpa_supplicant_rx_tdls - Receive TDLS data frame
1973  *
1974  * This function is called to receive TDLS (ethertype = 0x890d) data frames.
1975  */
1976 static void wpa_supplicant_rx_tdls(void *ctx, const u8 *src_addr,
1977                                    const u8 *buf, size_t len)
1978 {
1979         struct wpa_sm *sm = ctx;
1980         struct wpa_tdls_frame *tf;
1981
1982         wpa_hexdump(MSG_DEBUG, "TDLS: Received Data frame encapsulation",
1983                     buf, len);
1984
1985         if (sm->tdls_disabled || !sm->tdls_supported) {
1986                 wpa_printf(MSG_DEBUG, "TDLS: Discard message - TDLS disabled "
1987                            "or unsupported by driver");
1988                 return;
1989         }
1990
1991         if (os_memcmp(src_addr, sm->own_addr, ETH_ALEN) == 0) {
1992                 wpa_printf(MSG_DEBUG, "TDLS: Discard copy of own message");
1993                 return;
1994         }
1995
1996         if (len < sizeof(*tf)) {
1997                 wpa_printf(MSG_INFO, "TDLS: Drop too short frame");
1998                 return;
1999         }
2000
2001         /* Check to make sure its a valid encapsulated TDLS frame */
2002         tf = (struct wpa_tdls_frame *) buf;
2003         if (tf->payloadtype != 2 /* TDLS_RFTYPE */ ||
2004             tf->category != WLAN_ACTION_TDLS) {
2005                 wpa_printf(MSG_INFO, "TDLS: Invalid frame - payloadtype=%u "
2006                            "category=%u action=%u",
2007                            tf->payloadtype, tf->category, tf->action);
2008                 return;
2009         }
2010
2011         switch (tf->action) {
2012         case WLAN_TDLS_SETUP_REQUEST:
2013                 wpa_tdls_process_tpk_m1(sm, src_addr, buf, len);
2014                 break;
2015         case WLAN_TDLS_SETUP_RESPONSE:
2016                 wpa_tdls_process_tpk_m2(sm, src_addr, buf, len);
2017                 break;
2018         case WLAN_TDLS_SETUP_CONFIRM:
2019                 wpa_tdls_process_tpk_m3(sm, src_addr, buf, len);
2020                 break;
2021         case WLAN_TDLS_TEARDOWN:
2022                 wpa_tdls_recv_teardown(sm, src_addr, buf, len);
2023                 break;
2024         case WLAN_TDLS_DISCOVERY_REQUEST:
2025                 wpa_tdls_process_discovery_request(sm, src_addr, buf, len);
2026                 break;
2027         default:
2028                 /* Kernel code will process remaining frames */
2029                 wpa_printf(MSG_DEBUG, "TDLS: Ignore TDLS frame action code %u",
2030                            tf->action);
2031                 break;
2032         }
2033 }
2034
2035
2036 /**
2037  * wpa_tdls_init - Initialize driver interface parameters for TDLS
2038  * @wpa_s: Pointer to wpa_supplicant data
2039  * Returns: 0 on success, -1 on failure
2040  *
2041  * This function is called to initialize driver interface parameters for TDLS.
2042  * wpa_drv_init() must have been called before this function to initialize the
2043  * driver interface.
2044  */
2045 int wpa_tdls_init(struct wpa_sm *sm)
2046 {
2047         if (sm == NULL)
2048                 return -1;
2049
2050         sm->l2_tdls = l2_packet_init(sm->ifname, sm->own_addr,
2051                                      ETH_P_80211_ENCAP, wpa_supplicant_rx_tdls,
2052                                      sm, 0);
2053         if (sm->l2_tdls == NULL) {
2054                 wpa_printf(MSG_ERROR, "TDLS: Failed to open l2_packet "
2055                            "connection");
2056                 return -1;
2057         }
2058
2059         /*
2060          * Drivers that support TDLS but don't implement the get_capa callback
2061          * are assumed to perform everything internally
2062          */
2063         if (wpa_sm_tdls_get_capa(sm, &sm->tdls_supported,
2064                                  &sm->tdls_external_setup) < 0) {
2065                 sm->tdls_supported = 1;
2066                 sm->tdls_external_setup = 0;
2067         }
2068
2069         wpa_printf(MSG_DEBUG, "TDLS: TDLS operation%s supported by "
2070                    "driver", sm->tdls_supported ? "" : " not");
2071         wpa_printf(MSG_DEBUG, "TDLS: Driver uses %s link setup",
2072                    sm->tdls_external_setup ? "external" : "internal");
2073
2074         return 0;
2075 }
2076
2077
2078 static void wpa_tdls_remove_peers(struct wpa_sm *sm)
2079 {
2080         struct wpa_tdls_peer *peer, *tmp;
2081
2082         peer = sm->tdls;
2083         sm->tdls = NULL;
2084
2085         while (peer) {
2086                 int res;
2087                 tmp = peer->next;
2088                 res = wpa_sm_tdls_oper(sm, TDLS_DISABLE_LINK, peer->addr);
2089                 wpa_printf(MSG_DEBUG, "TDLS: Remove peer " MACSTR " (res=%d)",
2090                            MAC2STR(peer->addr), res);
2091                 wpa_tdls_peer_free(sm, peer);
2092                 os_free(peer);
2093                 peer = tmp;
2094         }
2095 }
2096
2097
2098 /**
2099  * wpa_tdls_deinit - Deinitialize driver interface parameters for TDLS
2100  *
2101  * This function is called to recover driver interface parameters for TDLS
2102  * and frees resources allocated for it.
2103  */
2104 void wpa_tdls_deinit(struct wpa_sm *sm)
2105 {
2106         if (sm == NULL)
2107                 return;
2108
2109         if (sm->l2_tdls)
2110                 l2_packet_deinit(sm->l2_tdls);
2111         sm->l2_tdls = NULL;
2112
2113         wpa_tdls_remove_peers(sm);
2114 }
2115
2116
2117 void wpa_tdls_assoc(struct wpa_sm *sm)
2118 {
2119         wpa_printf(MSG_DEBUG, "TDLS: Remove peers on association");
2120         wpa_tdls_remove_peers(sm);
2121 }
2122
2123
2124 void wpa_tdls_disassoc(struct wpa_sm *sm)
2125 {
2126         wpa_printf(MSG_DEBUG, "TDLS: Remove peers on disassociation");
2127         wpa_tdls_remove_peers(sm);
2128 }
2129
2130
2131 static int wpa_tdls_prohibited(const u8 *ies, size_t len)
2132 {
2133         struct wpa_eapol_ie_parse elems;
2134
2135         if (ies == NULL)
2136                 return 0;
2137
2138         if (wpa_supplicant_parse_ies(ies, len, &elems) < 0)
2139                 return 0;
2140
2141         if (elems.ext_capab == NULL || elems.ext_capab_len < 2 + 5)
2142                 return 0;
2143
2144          /* bit 38 - TDLS Prohibited */
2145         return !!(elems.ext_capab[2 + 4] & 0x40);
2146 }
2147
2148
2149 void wpa_tdls_ap_ies(struct wpa_sm *sm, const u8 *ies, size_t len)
2150 {
2151         sm->tdls_prohibited = wpa_tdls_prohibited(ies, len);
2152         wpa_printf(MSG_DEBUG, "TDLS: TDLS is %s in the target BSS",
2153                    sm->tdls_prohibited ? "prohibited" : "allowed");
2154 }
2155
2156
2157 void wpa_tdls_assoc_resp_ies(struct wpa_sm *sm, const u8 *ies, size_t len)
2158 {
2159         if (!sm->tdls_prohibited && wpa_tdls_prohibited(ies, len)) {
2160                 wpa_printf(MSG_DEBUG, "TDLS: TDLS prohibited based on "
2161                            "(Re)Association Response IEs");
2162                 sm->tdls_prohibited = 1;
2163         }
2164 }
2165
2166
2167 void wpa_tdls_enable(struct wpa_sm *sm, int enabled)
2168 {
2169         wpa_printf(MSG_DEBUG, "TDLS: %s", enabled ? "enabled" : "disabled");
2170         sm->tdls_disabled = !enabled;
2171 }