www.project-moonshot.org Git - mech_eap.orig/blob - src/ap/wpa_auth_ft.c

   1 /*
   2  * hostapd - IEEE 802.11r - Fast BSS Transition
   3  * Copyright (c) 2004-2009, Jouni Malinen <j@w1.fi>
   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 "common/ieee802_11_defs.h"
  19 #include "common/ieee802_11_common.h"
  20 #include "crypto/aes_wrap.h"
  21 #include "ap_config.h"
  22 #include "ieee802_11.h"
  23 #include "wmm.h"
  24 #include "wpa_auth.h"
  25 #include "wpa_auth_i.h"
  26 #include "wpa_auth_ie.h"
  27
  28
  29 #ifdef CONFIG_IEEE80211R
  30
  31 struct wpa_ft_ies {
  32         const u8 *mdie;
  33         size_t mdie_len;
  34         const u8 *ftie;
  35         size_t ftie_len;
  36         const u8 *r1kh_id;
  37         const u8 *gtk;
  38         size_t gtk_len;
  39         const u8 *r0kh_id;
  40         size_t r0kh_id_len;
  41         const u8 *rsn;
  42         size_t rsn_len;
  43         const u8 *rsn_pmkid;
  44         const u8 *ric;
  45         size_t ric_len;
  46 };
  47
  48
  49 static int wpa_ft_parse_ies(const u8 *ies, size_t ies_len,
  50                             struct wpa_ft_ies *parse);
  51
  52
  53 static int wpa_ft_rrb_send(struct wpa_authenticator *wpa_auth, const u8 *dst,
  54                            const u8 *data, size_t data_len)
  55 {
  56         if (wpa_auth->cb.send_ether == NULL)
  57                 return -1;
  58         return wpa_auth->cb.send_ether(wpa_auth->cb.ctx, dst, ETH_P_RRB,
  59                                        data, data_len);
  60 }
  61
  62
  63 static int wpa_ft_action_send(struct wpa_authenticator *wpa_auth,
  64                               const u8 *dst, const u8 *data, size_t data_len)
  65 {
  66         if (wpa_auth->cb.send_ft_action == NULL)
  67                 return -1;
  68         return wpa_auth->cb.send_ft_action(wpa_auth->cb.ctx, dst,
  69                                            data, data_len);
  70 }
  71
  72
  73 static struct wpa_state_machine *
  74 wpa_ft_add_sta(struct wpa_authenticator *wpa_auth, const u8 *sta_addr)
  75 {
  76         if (wpa_auth->cb.add_sta == NULL)
  77                 return NULL;
  78         return wpa_auth->cb.add_sta(wpa_auth->cb.ctx, sta_addr);
  79 }
  80
  81
  82 int wpa_write_mdie(struct wpa_auth_config *conf, u8 *buf, size_t len)
  83 {
  84         u8 *pos = buf;
  85         u8 capab;
  86         if (len < 2 + sizeof(struct rsn_mdie))
  87                 return -1;
  88
  89         *pos++ = WLAN_EID_MOBILITY_DOMAIN;
  90         *pos++ = MOBILITY_DOMAIN_ID_LEN + 1;
  91         os_memcpy(pos, conf->mobility_domain, MOBILITY_DOMAIN_ID_LEN);
  92         pos += MOBILITY_DOMAIN_ID_LEN;
  93         capab = RSN_FT_CAPAB_FT_OVER_DS;
  94         *pos++ = capab;
  95
  96         return pos - buf;
  97 }
  98
  99
 100 int wpa_write_ftie(struct wpa_auth_config *conf, const u8 *r0kh_id,
 101                    size_t r0kh_id_len,
 102                    const u8 *anonce, const u8 *snonce,
 103                    u8 *buf, size_t len, const u8 *subelem,
 104                    size_t subelem_len)
 105 {
 106         u8 *pos = buf, *ielen;
 107         struct rsn_ftie *hdr;
 108
 109         if (len < 2 + sizeof(*hdr) + 2 + FT_R1KH_ID_LEN + 2 + r0kh_id_len +
 110             subelem_len)
 111                 return -1;
 112
 113         *pos++ = WLAN_EID_FAST_BSS_TRANSITION;
 114         ielen = pos++;
 115
 116         hdr = (struct rsn_ftie *) pos;
 117         os_memset(hdr, 0, sizeof(*hdr));
 118         pos += sizeof(*hdr);
 119         WPA_PUT_LE16(hdr->mic_control, 0);
 120         if (anonce)
 121                 os_memcpy(hdr->anonce, anonce, WPA_NONCE_LEN);
 122         if (snonce)
 123                 os_memcpy(hdr->snonce, snonce, WPA_NONCE_LEN);
 124
 125         /* Optional Parameters */
 126         *pos++ = FTIE_SUBELEM_R1KH_ID;
 127         *pos++ = FT_R1KH_ID_LEN;
 128         os_memcpy(pos, conf->r1_key_holder, FT_R1KH_ID_LEN);
 129         pos += FT_R1KH_ID_LEN;
 130
 131         if (r0kh_id) {
 132                 *pos++ = FTIE_SUBELEM_R0KH_ID;
 133                 *pos++ = r0kh_id_len;
 134                 os_memcpy(pos, r0kh_id, r0kh_id_len);
 135                 pos += r0kh_id_len;
 136         }
 137
 138         if (subelem) {
 139                 os_memcpy(pos, subelem, subelem_len);
 140                 pos += subelem_len;
 141         }
 142
 143         *ielen = pos - buf - 2;
 144
 145         return pos - buf;
 146 }
 147
 148
 149 struct wpa_ft_pmk_r0_sa {
 150         struct wpa_ft_pmk_r0_sa *next;
 151         u8 pmk_r0[PMK_LEN];
 152         u8 pmk_r0_name[WPA_PMK_NAME_LEN];
 153         u8 spa[ETH_ALEN];
 154         int pairwise; /* Pairwise cipher suite, WPA_CIPHER_* */
 155         /* TODO: expiration, identity, radius_class, EAP type, VLAN ID */
 156         int pmk_r1_pushed;
 157 };
 158
 159 struct wpa_ft_pmk_r1_sa {
 160         struct wpa_ft_pmk_r1_sa *next;
 161         u8 pmk_r1[PMK_LEN];
 162         u8 pmk_r1_name[WPA_PMK_NAME_LEN];
 163         u8 spa[ETH_ALEN];
 164         int pairwise; /* Pairwise cipher suite, WPA_CIPHER_* */
 165         /* TODO: expiration, identity, radius_class, EAP type, VLAN ID */
 166 };
 167
 168 struct wpa_ft_pmk_cache {
 169         struct wpa_ft_pmk_r0_sa *pmk_r0;
 170         struct wpa_ft_pmk_r1_sa *pmk_r1;
 171 };
 172
 173 struct wpa_ft_pmk_cache * wpa_ft_pmk_cache_init(void)
 174 {
 175         struct wpa_ft_pmk_cache *cache;
 176
 177         cache = os_zalloc(sizeof(*cache));
 178
 179         return cache;
 180 }
 181
 182
 183 void wpa_ft_pmk_cache_deinit(struct wpa_ft_pmk_cache *cache)
 184 {
 185         struct wpa_ft_pmk_r0_sa *r0, *r0prev;
 186         struct wpa_ft_pmk_r1_sa *r1, *r1prev;
 187
 188         r0 = cache->pmk_r0;
 189         while (r0) {
 190                 r0prev = r0;
 191                 r0 = r0->next;
 192                 os_memset(r0prev->pmk_r0, 0, PMK_LEN);
 193                 os_free(r0prev);
 194         }
 195
 196         r1 = cache->pmk_r1;
 197         while (r1) {
 198                 r1prev = r1;
 199                 r1 = r1->next;
 200                 os_memset(r1prev->pmk_r1, 0, PMK_LEN);
 201                 os_free(r1prev);
 202         }
 203
 204         os_free(cache);
 205 }
 206
 207
 208 static int wpa_ft_store_pmk_r0(struct wpa_authenticator *wpa_auth,
 209                                const u8 *spa, const u8 *pmk_r0,
 210                                const u8 *pmk_r0_name, int pairwise)
 211 {
 212         struct wpa_ft_pmk_cache *cache = wpa_auth->ft_pmk_cache;
 213         struct wpa_ft_pmk_r0_sa *r0;
 214
 215         /* TODO: add expiration and limit on number of entries in cache */
 216
 217         r0 = os_zalloc(sizeof(*r0));
 218         if (r0 == NULL)
 219                 return -1;
 220
 221         os_memcpy(r0->pmk_r0, pmk_r0, PMK_LEN);
 222         os_memcpy(r0->pmk_r0_name, pmk_r0_name, WPA_PMK_NAME_LEN);
 223         os_memcpy(r0->spa, spa, ETH_ALEN);
 224         r0->pairwise = pairwise;
 225
 226         r0->next = cache->pmk_r0;
 227         cache->pmk_r0 = r0;
 228
 229         return 0;
 230 }
 231
 232
 233 static int wpa_ft_fetch_pmk_r0(struct wpa_authenticator *wpa_auth,
 234                                const u8 *spa, const u8 *pmk_r0_name,
 235                                u8 *pmk_r0, int *pairwise)
 236 {
 237         struct wpa_ft_pmk_cache *cache = wpa_auth->ft_pmk_cache;
 238         struct wpa_ft_pmk_r0_sa *r0;
 239
 240         r0 = cache->pmk_r0;
 241         while (r0) {
 242                 if (os_memcmp(r0->spa, spa, ETH_ALEN) == 0 &&
 243                     os_memcmp(r0->pmk_r0_name, pmk_r0_name, WPA_PMK_NAME_LEN)
 244                     == 0) {
 245                         os_memcpy(pmk_r0, r0->pmk_r0, PMK_LEN);
 246                         if (pairwise)
 247                                 *pairwise = r0->pairwise;
 248                         return 0;
 249                 }
 250
 251                 r0 = r0->next;
 252         }
 253
 254         return -1;
 255 }
 256
 257
 258 static int wpa_ft_store_pmk_r1(struct wpa_authenticator *wpa_auth,
 259                                const u8 *spa, const u8 *pmk_r1,
 260                                const u8 *pmk_r1_name, int pairwise)
 261 {
 262         struct wpa_ft_pmk_cache *cache = wpa_auth->ft_pmk_cache;
 263         struct wpa_ft_pmk_r1_sa *r1;
 264
 265         /* TODO: add expiration and limit on number of entries in cache */
 266
 267         r1 = os_zalloc(sizeof(*r1));
 268         if (r1 == NULL)
 269                 return -1;
 270
 271         os_memcpy(r1->pmk_r1, pmk_r1, PMK_LEN);
 272         os_memcpy(r1->pmk_r1_name, pmk_r1_name, WPA_PMK_NAME_LEN);
 273         os_memcpy(r1->spa, spa, ETH_ALEN);
 274         r1->pairwise = pairwise;
 275
 276         r1->next = cache->pmk_r1;
 277         cache->pmk_r1 = r1;
 278
 279         return 0;
 280 }
 281
 282
 283 static int wpa_ft_fetch_pmk_r1(struct wpa_authenticator *wpa_auth,
 284                                const u8 *spa, const u8 *pmk_r1_name,
 285                                u8 *pmk_r1, int *pairwise)
 286 {
 287         struct wpa_ft_pmk_cache *cache = wpa_auth->ft_pmk_cache;
 288         struct wpa_ft_pmk_r1_sa *r1;
 289
 290         r1 = cache->pmk_r1;
 291         while (r1) {
 292                 if (os_memcmp(r1->spa, spa, ETH_ALEN) == 0 &&
 293                     os_memcmp(r1->pmk_r1_name, pmk_r1_name, WPA_PMK_NAME_LEN)
 294                     == 0) {
 295                         os_memcpy(pmk_r1, r1->pmk_r1, PMK_LEN);
 296                         if (pairwise)
 297                                 *pairwise = r1->pairwise;
 298                         return 0;
 299                 }
 300
 301                 r1 = r1->next;
 302         }
 303
 304         return -1;
 305 }
 306
 307
 308 static int wpa_ft_pull_pmk_r1(struct wpa_authenticator *wpa_auth,
 309                               const u8 *s1kh_id, const u8 *r0kh_id,
 310                               size_t r0kh_id_len, const u8 *pmk_r0_name)
 311 {
 312         struct ft_remote_r0kh *r0kh;
 313         struct ft_r0kh_r1kh_pull_frame frame, f;
 314
 315         r0kh = wpa_auth->conf.r0kh_list;
 316         while (r0kh) {
 317                 if (r0kh->id_len == r0kh_id_len &&
 318                     os_memcmp(r0kh->id, r0kh_id, r0kh_id_len) == 0)
 319                         break;
 320                 r0kh = r0kh->next;
 321         }
 322         if (r0kh == NULL)
 323                 return -1;
 324
 325         wpa_printf(MSG_DEBUG, "FT: Send PMK-R1 pull request to remote R0KH "
 326                    "address " MACSTR, MAC2STR(r0kh->addr));
 327
 328         os_memset(&frame, 0, sizeof(frame));
 329         frame.frame_type = RSN_REMOTE_FRAME_TYPE_FT_RRB;
 330         frame.packet_type = FT_PACKET_R0KH_R1KH_PULL;
 331         frame.data_length = host_to_le16(FT_R0KH_R1KH_PULL_DATA_LEN);
 332         os_memcpy(frame.ap_address, wpa_auth->addr, ETH_ALEN);
 333
 334         /* aes_wrap() does not support inplace encryption, so use a temporary
 335          * buffer for the data. */
 336         if (os_get_random(f.nonce, sizeof(f.nonce))) {
 337                 wpa_printf(MSG_DEBUG, "FT: Failed to get random data for "
 338                            "nonce");
 339                 return -1;
 340         }
 341         os_memcpy(f.pmk_r0_name, pmk_r0_name, WPA_PMK_NAME_LEN);
 342         os_memcpy(f.r1kh_id, wpa_auth->conf.r1_key_holder, FT_R1KH_ID_LEN);
 343         os_memcpy(f.s1kh_id, s1kh_id, ETH_ALEN);
 344
 345         if (aes_wrap(r0kh->key, (FT_R0KH_R1KH_PULL_DATA_LEN + 7) / 8,
 346                      f.nonce, frame.nonce) < 0)
 347                 return -1;
 348
 349         wpa_ft_rrb_send(wpa_auth, r0kh->addr, (u8 *) &frame, sizeof(frame));
 350
 351         return 0;
 352 }
 353
 354
 355 int wpa_auth_derive_ptk_ft(struct wpa_state_machine *sm, const u8 *pmk,
 356                            struct wpa_ptk *ptk, size_t ptk_len)
 357 {
 358         u8 pmk_r0[PMK_LEN], pmk_r0_name[WPA_PMK_NAME_LEN];
 359         u8 pmk_r1[PMK_LEN];
 360         u8 ptk_name[WPA_PMK_NAME_LEN];
 361         const u8 *mdid = sm->wpa_auth->conf.mobility_domain;
 362         const u8 *r0kh = sm->wpa_auth->conf.r0_key_holder;
 363         size_t r0kh_len = sm->wpa_auth->conf.r0_key_holder_len;
 364         const u8 *r1kh = sm->wpa_auth->conf.r1_key_holder;
 365         const u8 *ssid = sm->wpa_auth->conf.ssid;
 366         size_t ssid_len = sm->wpa_auth->conf.ssid_len;
 367
 368
 369         if (sm->xxkey_len == 0) {
 370                 wpa_printf(MSG_DEBUG, "FT: XXKey not available for key "
 371                            "derivation");
 372                 return -1;
 373         }
 374
 375         wpa_derive_pmk_r0(sm->xxkey, sm->xxkey_len, ssid, ssid_len, mdid,
 376                           r0kh, r0kh_len, sm->addr, pmk_r0, pmk_r0_name);
 377         wpa_hexdump_key(MSG_DEBUG, "FT: PMK-R0", pmk_r0, PMK_LEN);
 378         wpa_hexdump(MSG_DEBUG, "FT: PMKR0Name", pmk_r0_name, WPA_PMK_NAME_LEN);
 379         wpa_ft_store_pmk_r0(sm->wpa_auth, sm->addr, pmk_r0, pmk_r0_name,
 380                             sm->pairwise);
 381
 382         wpa_derive_pmk_r1(pmk_r0, pmk_r0_name, r1kh, sm->addr,
 383                           pmk_r1, sm->pmk_r1_name);
 384         wpa_hexdump_key(MSG_DEBUG, "FT: PMK-R1", pmk_r1, PMK_LEN);
 385         wpa_hexdump(MSG_DEBUG, "FT: PMKR1Name", sm->pmk_r1_name,
 386                     WPA_PMK_NAME_LEN);
 387         wpa_ft_store_pmk_r1(sm->wpa_auth, sm->addr, pmk_r1, sm->pmk_r1_name,
 388                             sm->pairwise);
 389
 390         wpa_pmk_r1_to_ptk(pmk_r1, sm->SNonce, sm->ANonce, sm->addr,
 391                           sm->wpa_auth->addr, sm->pmk_r1_name,
 392                           (u8 *) ptk, ptk_len, ptk_name);
 393         wpa_hexdump_key(MSG_DEBUG, "FT: PTK", (u8 *) ptk, ptk_len);
 394         wpa_hexdump(MSG_DEBUG, "FT: PTKName", ptk_name, WPA_PMK_NAME_LEN);
 395
 396         return 0;
 397 }
 398
 399
 400 static inline int wpa_auth_get_seqnum(struct wpa_authenticator *wpa_auth,
 401                                       const u8 *addr, int idx, u8 *seq)
 402 {
 403         if (wpa_auth->cb.get_seqnum == NULL)
 404                 return -1;
 405         return wpa_auth->cb.get_seqnum(wpa_auth->cb.ctx, addr, idx, seq);
 406 }
 407
 408
 409 static u8 * wpa_ft_gtk_subelem(struct wpa_state_machine *sm, size_t *len)
 410 {
 411         u8 *subelem;
 412         struct wpa_group *gsm = sm->group;
 413         size_t subelem_len, pad_len;
 414         const u8 *key;
 415         size_t key_len;
 416         u8 keybuf[32];
 417
 418         key_len = gsm->GTK_len;
 419         if (key_len > sizeof(keybuf))
 420                 return NULL;
 421
 422         /*
 423          * Pad key for AES Key Wrap if it is not multiple of 8 bytes or is less
 424          * than 16 bytes.
 425          */
 426         pad_len = key_len % 8;
 427         if (pad_len)
 428                 pad_len = 8 - pad_len;
 429         if (key_len + pad_len < 16)
 430                 pad_len += 8;
 431         if (pad_len) {
 432                 os_memcpy(keybuf, gsm->GTK[gsm->GN - 1], key_len);
 433                 os_memset(keybuf + key_len, 0, pad_len);
 434                 keybuf[key_len] = 0xdd;
 435                 key_len += pad_len;
 436                 key = keybuf;
 437         } else
 438                 key = gsm->GTK[gsm->GN - 1];
 439
 440         /*
 441          * Sub-elem ID[1] | Length[1] | Key Info[2] | Key Length[1] | RSC[8] |
 442          * Key[5..32].
 443          */
 444         subelem_len = 13 + key_len + 8;
 445         subelem = os_zalloc(subelem_len);
 446         if (subelem == NULL)
 447                 return NULL;
 448
 449         subelem[0] = FTIE_SUBELEM_GTK;
 450         subelem[1] = 11 + key_len + 8;
 451         /* Key ID in B0-B1 of Key Info */
 452         WPA_PUT_LE16(&subelem[2], gsm->GN & 0x03);
 453         subelem[4] = gsm->GTK_len;
 454         wpa_auth_get_seqnum(sm->wpa_auth, NULL, gsm->GN, subelem + 5);
 455         if (aes_wrap(sm->PTK.kek, key_len / 8, key, subelem + 13)) {
 456                 os_free(subelem);
 457                 return NULL;
 458         }
 459
 460         *len = subelem_len;
 461         return subelem;
 462 }
 463
 464
 465 #ifdef CONFIG_IEEE80211W
 466 static u8 * wpa_ft_igtk_subelem(struct wpa_state_machine *sm, size_t *len)
 467 {
 468         u8 *subelem, *pos;
 469         struct wpa_group *gsm = sm->group;
 470         size_t subelem_len;
 471
 472         /* Sub-elem ID[1] | Length[1] | KeyID[2] | IPN[6] | Key Length[1] |
 473          * Key[16+8] */
 474         subelem_len = 1 + 1 + 2 + 6 + 1 + WPA_IGTK_LEN + 8;
 475         subelem = os_zalloc(subelem_len);
 476         if (subelem == NULL)
 477                 return NULL;
 478
 479         pos = subelem;
 480         *pos++ = FTIE_SUBELEM_IGTK;
 481         *pos++ = subelem_len - 2;
 482         WPA_PUT_LE16(pos, gsm->GN_igtk);
 483         pos += 2;
 484         wpa_auth_get_seqnum(sm->wpa_auth, NULL, gsm->GN_igtk, pos);
 485         pos += 6;
 486         *pos++ = WPA_IGTK_LEN;
 487         if (aes_wrap(sm->PTK.kek, WPA_IGTK_LEN / 8,
 488                      gsm->IGTK[gsm->GN_igtk - 4], pos)) {
 489                 os_free(subelem);
 490                 return NULL;
 491         }
 492
 493         *len = subelem_len;
 494         return subelem;
 495 }
 496 #endif /* CONFIG_IEEE80211W */
 497
 498
 499 static u8 * wpa_ft_process_rdie(u8 *pos, u8 *end, u8 id, u8 descr_count,
 500                                 const u8 *ies, size_t ies_len)
 501 {
 502         struct ieee802_11_elems parse;
 503         struct rsn_rdie *rdie;
 504
 505         wpa_printf(MSG_DEBUG, "FT: Resource Request: id=%d descr_count=%d",
 506                    id, descr_count);
 507         wpa_hexdump(MSG_MSGDUMP, "FT: Resource descriptor IE(s)",
 508                     ies, ies_len);
 509
 510         if (end - pos < (int) sizeof(*rdie)) {
 511                 wpa_printf(MSG_ERROR, "FT: Not enough room for response RDIE");
 512                 return pos;
 513         }
 514
 515         *pos++ = WLAN_EID_RIC_DATA;
 516         *pos++ = sizeof(*rdie);
 517         rdie = (struct rsn_rdie *) pos;
 518         rdie->id = id;
 519         rdie->descr_count = 0;
 520         rdie->status_code = host_to_le16(WLAN_STATUS_SUCCESS);
 521         pos += sizeof(*rdie);
 522
 523         if (ieee802_11_parse_elems((u8 *) ies, ies_len, &parse, 1) ==
 524             ParseFailed) {
 525                 wpa_printf(MSG_DEBUG, "FT: Failed to parse request IEs");
 526                 rdie->status_code =
 527                         host_to_le16(WLAN_STATUS_UNSPECIFIED_FAILURE);
 528                 return pos;
 529         }
 530
 531 #ifdef NEED_AP_MLME
 532         if (parse.wmm_tspec) {
 533                 struct wmm_tspec_element *tspec;
 534                 int res;
 535
 536                 if (parse.wmm_tspec_len + 2 < (int) sizeof(*tspec)) {
 537                         wpa_printf(MSG_DEBUG, "FT: Too short WMM TSPEC IE "
 538                                    "(%d)", (int) parse.wmm_tspec_len);
 539                         rdie->status_code =
 540                                 host_to_le16(WLAN_STATUS_UNSPECIFIED_FAILURE);
 541                         return pos;
 542                 }
 543                 if (end - pos < (int) sizeof(*tspec)) {
 544                         wpa_printf(MSG_ERROR, "FT: Not enough room for "
 545                                    "response TSPEC");
 546                         rdie->status_code =
 547                                 host_to_le16(WLAN_STATUS_UNSPECIFIED_FAILURE);
 548                         return pos;
 549                 }
 550                 tspec = (struct wmm_tspec_element *) pos;
 551                 os_memcpy(tspec, parse.wmm_tspec - 2, sizeof(*tspec));
 552                 res = wmm_process_tspec(tspec);
 553                 wpa_printf(MSG_DEBUG, "FT: ADDTS processing result: %d", res);
 554                 if (res == WMM_ADDTS_STATUS_INVALID_PARAMETERS)
 555                         rdie->status_code =
 556                                 host_to_le16(WLAN_STATUS_INVALID_PARAMETERS);
 557                 else if (res == WMM_ADDTS_STATUS_REFUSED)
 558                         rdie->status_code =
 559                                 host_to_le16(WLAN_STATUS_REQUEST_DECLINED);
 560                 else {
 561                         /* TSPEC accepted; include updated TSPEC in response */
 562                         rdie->descr_count = 1;
 563                         pos += sizeof(*tspec);
 564                 }
 565                 return pos;
 566         }
 567 #endif /* NEED_AP_MLME */
 568
 569         wpa_printf(MSG_DEBUG, "FT: No supported resource requested");
 570         rdie->status_code = host_to_le16(WLAN_STATUS_UNSPECIFIED_FAILURE);
 571         return pos;
 572 }
 573
 574
 575 static u8 * wpa_ft_process_ric(u8 *pos, u8 *end, const u8 *ric, size_t ric_len)
 576 {
 577         const u8 *rpos, *start;
 578         const struct rsn_rdie *rdie;
 579
 580         wpa_hexdump(MSG_MSGDUMP, "FT: RIC Request", ric, ric_len);
 581
 582         rpos = ric;
 583         while (rpos + sizeof(*rdie) < ric + ric_len) {
 584                 if (rpos[0] != WLAN_EID_RIC_DATA || rpos[1] < sizeof(*rdie) ||
 585                     rpos + 2 + rpos[1] > ric + ric_len)
 586                         break;
 587                 rdie = (const struct rsn_rdie *) (rpos + 2);
 588                 rpos += 2 + rpos[1];
 589                 start = rpos;
 590
 591                 while (rpos + 2 <= ric + ric_len &&
 592                        rpos + 2 + rpos[1] <= ric + ric_len) {
 593                         if (rpos[0] == WLAN_EID_RIC_DATA)
 594                                 break;
 595                         rpos += 2 + rpos[1];
 596                 }
 597                 pos = wpa_ft_process_rdie(pos, end, rdie->id,
 598                                           rdie->descr_count,
 599                                           start, rpos - start);
 600         }
 601
 602         return pos;
 603 }
 604
 605
 606 u8 * wpa_sm_write_assoc_resp_ies(struct wpa_state_machine *sm, u8 *pos,
 607                                  size_t max_len, int auth_alg,
 608                                  const u8 *req_ies, size_t req_ies_len)
 609 {
 610         u8 *end, *mdie, *ftie, *rsnie = NULL, *r0kh_id, *subelem = NULL;
 611         size_t mdie_len, ftie_len, rsnie_len = 0, r0kh_id_len, subelem_len = 0;
 612         int res;
 613         struct wpa_auth_config *conf;
 614         struct rsn_ftie *_ftie;
 615         struct wpa_ft_ies parse;
 616         u8 *ric_start;
 617         u8 *anonce, *snonce;
 618
 619         if (sm == NULL)
 620                 return pos;
 621
 622         conf = &sm->wpa_auth->conf;
 623
 624         if (sm->wpa_key_mgmt != WPA_KEY_MGMT_FT_IEEE8021X &&
 625             sm->wpa_key_mgmt != WPA_KEY_MGMT_FT_PSK)
 626                 return pos;
 627
 628         end = pos + max_len;
 629
 630         if (auth_alg == WLAN_AUTH_FT) {
 631                 /*
 632                  * RSN (only present if this is a Reassociation Response and
 633                  * part of a fast BSS transition)
 634                  */
 635                 res = wpa_write_rsn_ie(conf, pos, end - pos, sm->pmk_r1_name);
 636                 if (res < 0)
 637                         return pos;
 638                 rsnie = pos;
 639                 rsnie_len = res;
 640                 pos += res;
 641         }
 642
 643         /* Mobility Domain Information */
 644         res = wpa_write_mdie(conf, pos, end - pos);
 645         if (res < 0)
 646                 return pos;
 647         mdie = pos;
 648         mdie_len = res;
 649         pos += res;
 650
 651         /* Fast BSS Transition Information */
 652         if (auth_alg == WLAN_AUTH_FT) {
 653                 subelem = wpa_ft_gtk_subelem(sm, &subelem_len);
 654                 r0kh_id = sm->r0kh_id;
 655                 r0kh_id_len = sm->r0kh_id_len;
 656                 anonce = sm->ANonce;
 657                 snonce = sm->SNonce;
 658 #ifdef CONFIG_IEEE80211W
 659                 if (sm->mgmt_frame_prot) {
 660                         u8 *igtk;
 661                         size_t igtk_len;
 662                         u8 *nbuf;
 663                         igtk = wpa_ft_igtk_subelem(sm, &igtk_len);
 664                         if (igtk == NULL) {
 665                                 os_free(subelem);
 666                                 return pos;
 667                         }
 668                         nbuf = os_realloc(subelem, subelem_len + igtk_len);
 669                         if (nbuf == NULL) {
 670                                 os_free(subelem);
 671                                 os_free(igtk);
 672                                 return pos;
 673                         }
 674                         subelem = nbuf;
 675                         os_memcpy(subelem + subelem_len, igtk, igtk_len);
 676                         subelem_len += igtk_len;
 677                         os_free(igtk);
 678                 }
 679 #endif /* CONFIG_IEEE80211W */
 680         } else {
 681                 r0kh_id = conf->r0_key_holder;
 682                 r0kh_id_len = conf->r0_key_holder_len;
 683                 anonce = NULL;
 684                 snonce = NULL;
 685         }
 686         res = wpa_write_ftie(conf, r0kh_id, r0kh_id_len, anonce, snonce, pos,
 687                              end - pos, subelem, subelem_len);
 688         os_free(subelem);
 689         if (res < 0)
 690                 return pos;
 691         ftie = pos;
 692         ftie_len = res;
 693         pos += res;
 694
 695         os_free(sm->assoc_resp_ftie);
 696         sm->assoc_resp_ftie = os_malloc(ftie_len);
 697         if (sm->assoc_resp_ftie)
 698                 os_memcpy(sm->assoc_resp_ftie, ftie, ftie_len);
 699
 700         _ftie = (struct rsn_ftie *) (ftie + 2);
 701         if (auth_alg == WLAN_AUTH_FT)
 702                 _ftie->mic_control[1] = 3; /* Information element count */
 703
 704         ric_start = pos;
 705         if (wpa_ft_parse_ies(req_ies, req_ies_len, &parse) == 0 && parse.ric) {
 706                 pos = wpa_ft_process_ric(pos, end, parse.ric, parse.ric_len);
 707                 if (auth_alg == WLAN_AUTH_FT)
 708                         _ftie->mic_control[1] +=
 709                                 ieee802_11_ie_count(ric_start,
 710                                                     pos - ric_start);
 711         }
 712         if (ric_start == pos)
 713                 ric_start = NULL;
 714
 715         if (auth_alg == WLAN_AUTH_FT &&
 716             wpa_ft_mic(sm->PTK.kck, sm->addr, sm->wpa_auth->addr, 6,
 717                        mdie, mdie_len, ftie, ftie_len,
 718                        rsnie, rsnie_len,
 719                        ric_start, ric_start ? pos - ric_start : 0,
 720                        _ftie->mic) < 0)
 721                 wpa_printf(MSG_DEBUG, "FT: Failed to calculate MIC");
 722
 723         return pos;
 724 }
 725
 726
 727 static int wpa_ft_parse_ftie(const u8 *ie, size_t ie_len,
 728                              struct wpa_ft_ies *parse)
 729 {
 730         const u8 *end, *pos;
 731
 732         parse->ftie = ie;
 733         parse->ftie_len = ie_len;
 734
 735         pos = ie + sizeof(struct rsn_ftie);
 736         end = ie + ie_len;
 737
 738         while (pos + 2 <= end && pos + 2 + pos[1] <= end) {
 739                 switch (pos[0]) {
 740                 case FTIE_SUBELEM_R1KH_ID:
 741                         if (pos[1] != FT_R1KH_ID_LEN) {
 742                                 wpa_printf(MSG_DEBUG, "FT: Invalid R1KH-ID "
 743                                            "length in FTIE: %d", pos[1]);
 744                                 return -1;
 745                         }
 746                         parse->r1kh_id = pos + 2;
 747                         break;
 748                 case FTIE_SUBELEM_GTK:
 749                         parse->gtk = pos + 2;
 750                         parse->gtk_len = pos[1];
 751                         break;
 752                 case FTIE_SUBELEM_R0KH_ID:
 753                         if (pos[1] < 1 || pos[1] > FT_R0KH_ID_MAX_LEN) {
 754                                 wpa_printf(MSG_DEBUG, "FT: Invalid R0KH-ID "
 755                                            "length in FTIE: %d", pos[1]);
 756                                 return -1;
 757                         }
 758                         parse->r0kh_id = pos + 2;
 759                         parse->r0kh_id_len = pos[1];
 760                         break;
 761                 }
 762
 763                 pos += 2 + pos[1];
 764         }
 765
 766         return 0;
 767 }
 768
 769
 770 static int wpa_ft_parse_ies(const u8 *ies, size_t ies_len,
 771                             struct wpa_ft_ies *parse)
 772 {
 773         const u8 *end, *pos;
 774         struct wpa_ie_data data;
 775         int ret;
 776         const struct rsn_ftie *ftie;
 777         int prot_ie_count = 0;
 778
 779         os_memset(parse, 0, sizeof(*parse));
 780         if (ies == NULL)
 781                 return 0;
 782
 783         pos = ies;
 784         end = ies + ies_len;
 785         while (pos + 2 <= end && pos + 2 + pos[1] <= end) {
 786                 switch (pos[0]) {
 787                 case WLAN_EID_RSN:
 788                         parse->rsn = pos + 2;
 789                         parse->rsn_len = pos[1];
 790                         ret = wpa_parse_wpa_ie_rsn(parse->rsn - 2,
 791                                                    parse->rsn_len + 2,
 792                                                    &data);
 793                         if (ret < 0) {
 794                                 wpa_printf(MSG_DEBUG, "FT: Failed to parse "
 795                                            "RSN IE: %d", ret);
 796                                 return -1;
 797                         }
 798                         if (data.num_pmkid == 1 && data.pmkid)
 799                                 parse->rsn_pmkid = data.pmkid;
 800                         break;
 801                 case WLAN_EID_MOBILITY_DOMAIN:
 802                         parse->mdie = pos + 2;
 803                         parse->mdie_len = pos[1];
 804                         break;
 805                 case WLAN_EID_FAST_BSS_TRANSITION:
 806                         if (pos[1] < sizeof(*ftie))
 807                                 return -1;
 808                         ftie = (const struct rsn_ftie *) (pos + 2);
 809                         prot_ie_count = ftie->mic_control[1];
 810                         if (wpa_ft_parse_ftie(pos + 2, pos[1], parse) < 0)
 811                                 return -1;
 812                         break;
 813                 case WLAN_EID_RIC_DATA:
 814                         if (parse->ric == NULL)
 815                                 parse->ric = pos;
 816                 }
 817
 818                 pos += 2 + pos[1];
 819         }
 820
 821         if (prot_ie_count == 0)
 822                 return 0; /* no MIC */
 823
 824         /*
 825          * Check that the protected IE count matches with IEs included in the
 826          * frame.
 827          */
 828         if (parse->rsn)
 829                 prot_ie_count--;
 830         if (parse->mdie)
 831                 prot_ie_count--;
 832         if (parse->ftie)
 833                 prot_ie_count--;
 834         if (prot_ie_count < 0) {
 835                 wpa_printf(MSG_DEBUG, "FT: Some required IEs not included in "
 836                            "the protected IE count");
 837                 return -1;
 838         }
 839
 840         if (prot_ie_count == 0 && parse->ric) {
 841                 wpa_printf(MSG_DEBUG, "FT: RIC IE(s) in the frame, but not "
 842                            "included in protected IE count");
 843                 return -1;
 844         }
 845
 846         /* Determine the end of the RIC IE(s) */
 847         pos = parse->ric;
 848         while (pos && pos + 2 <= end && pos + 2 + pos[1] <= end &&
 849                prot_ie_count) {
 850                 prot_ie_count--;
 851                 pos += 2 + pos[1];
 852         }
 853         parse->ric_len = pos - parse->ric;
 854         if (prot_ie_count) {
 855                 wpa_printf(MSG_DEBUG, "FT: %d protected IEs missing from "
 856                            "frame", (int) prot_ie_count);
 857                 return -1;
 858         }
 859
 860         return 0;
 861 }
 862
 863
 864 static inline int wpa_auth_set_key(struct wpa_authenticator *wpa_auth,
 865                                    int vlan_id,
 866                                    enum wpa_alg alg, const u8 *addr, int idx,
 867                                    u8 *key, size_t key_len)
 868 {
 869         if (wpa_auth->cb.set_key == NULL)
 870                 return -1;
 871         return wpa_auth->cb.set_key(wpa_auth->cb.ctx, vlan_id, alg, addr, idx,
 872                                     key, key_len);
 873 }
 874
 875
 876 void wpa_ft_install_ptk(struct wpa_state_machine *sm)
 877 {
 878         enum wpa_alg alg;
 879         int klen;
 880
 881         /* MLME-SETKEYS.request(PTK) */
 882         if (sm->pairwise == WPA_CIPHER_TKIP) {
 883                 alg = WPA_ALG_TKIP;
 884                 klen = 32;
 885         } else if (sm->pairwise == WPA_CIPHER_CCMP) {
 886                 alg = WPA_ALG_CCMP;
 887                 klen = 16;
 888         } else {
 889                 wpa_printf(MSG_DEBUG, "FT: Unknown pairwise alg 0x%x - skip "
 890                            "PTK configuration", sm->pairwise);
 891                 return;
 892         }
 893
 894         /* FIX: add STA entry to kernel/driver here? The set_key will fail
 895          * most likely without this.. At the moment, STA entry is added only
 896          * after association has been completed. This function will be called
 897          * again after association to get the PTK configured, but that could be
 898          * optimized by adding the STA entry earlier.
 899          */
 900         if (wpa_auth_set_key(sm->wpa_auth, 0, alg, sm->addr, 0,
 901                              sm->PTK.tk1, klen))
 902                 return;
 903
 904         /* FIX: MLME-SetProtection.Request(TA, Tx_Rx) */
 905         sm->pairwise_set = TRUE;
 906 }
 907
 908
 909 static u16 wpa_ft_process_auth_req(struct wpa_state_machine *sm,
 910                                    const u8 *ies, size_t ies_len,
 911                                    u8 **resp_ies, size_t *resp_ies_len)
 912 {
 913         struct rsn_mdie *mdie;
 914         struct rsn_ftie *ftie;
 915         u8 pmk_r1[PMK_LEN], pmk_r1_name[WPA_PMK_NAME_LEN];
 916         u8 ptk_name[WPA_PMK_NAME_LEN];
 917         struct wpa_auth_config *conf;
 918         struct wpa_ft_ies parse;
 919         size_t buflen, ptk_len;
 920         int ret;
 921         u8 *pos, *end;
 922         int pairwise;
 923
 924         *resp_ies = NULL;
 925         *resp_ies_len = 0;
 926
 927         sm->pmk_r1_name_valid = 0;
 928         conf = &sm->wpa_auth->conf;
 929
 930         wpa_hexdump(MSG_DEBUG, "FT: Received authentication frame IEs",
 931                     ies, ies_len);
 932
 933         if (wpa_ft_parse_ies(ies, ies_len, &parse) < 0) {
 934                 wpa_printf(MSG_DEBUG, "FT: Failed to parse FT IEs");
 935                 return WLAN_STATUS_UNSPECIFIED_FAILURE;
 936         }
 937
 938         mdie = (struct rsn_mdie *) parse.mdie;
 939         if (mdie == NULL || parse.mdie_len < sizeof(*mdie) ||
 940             os_memcmp(mdie->mobility_domain,
 941                       sm->wpa_auth->conf.mobility_domain,
 942                       MOBILITY_DOMAIN_ID_LEN) != 0) {
 943                 wpa_printf(MSG_DEBUG, "FT: Invalid MDIE");
 944                 return WLAN_STATUS_INVALID_MDIE;
 945         }
 946
 947         ftie = (struct rsn_ftie *) parse.ftie;
 948         if (ftie == NULL || parse.ftie_len < sizeof(*ftie)) {
 949                 wpa_printf(MSG_DEBUG, "FT: Invalid FTIE");
 950                 return WLAN_STATUS_INVALID_FTIE;
 951         }
 952
 953         os_memcpy(sm->SNonce, ftie->snonce, WPA_NONCE_LEN);
 954
 955         if (parse.r0kh_id == NULL) {
 956                 wpa_printf(MSG_DEBUG, "FT: Invalid FTIE - no R0KH-ID");
 957                 return WLAN_STATUS_INVALID_FTIE;
 958         }
 959
 960         wpa_hexdump(MSG_DEBUG, "FT: STA R0KH-ID",
 961                     parse.r0kh_id, parse.r0kh_id_len);
 962         os_memcpy(sm->r0kh_id, parse.r0kh_id, parse.r0kh_id_len);
 963         sm->r0kh_id_len = parse.r0kh_id_len;
 964
 965         if (parse.rsn_pmkid == NULL) {
 966                 wpa_printf(MSG_DEBUG, "FT: No PMKID in RSNIE");
 967                 return WLAN_STATUS_INVALID_PMKID;
 968         }
 969
 970         wpa_hexdump(MSG_DEBUG, "FT: Requested PMKR0Name",
 971                     parse.rsn_pmkid, WPA_PMK_NAME_LEN);
 972         wpa_derive_pmk_r1_name(parse.rsn_pmkid,
 973                                sm->wpa_auth->conf.r1_key_holder, sm->addr,
 974                                pmk_r1_name);
 975         wpa_hexdump(MSG_DEBUG, "FT: Derived requested PMKR1Name",
 976                     pmk_r1_name, WPA_PMK_NAME_LEN);
 977
 978         if (wpa_ft_fetch_pmk_r1(sm->wpa_auth, sm->addr, pmk_r1_name, pmk_r1,
 979                     &pairwise) < 0) {
 980                 if (wpa_ft_pull_pmk_r1(sm->wpa_auth, sm->addr, sm->r0kh_id,
 981                                        sm->r0kh_id_len, parse.rsn_pmkid) < 0) {
 982                         wpa_printf(MSG_DEBUG, "FT: Did not have matching "
 983                                    "PMK-R1 and unknown R0KH-ID");
 984                         return WLAN_STATUS_INVALID_PMKID;
 985                 }
 986
 987                 /*
 988                  * TODO: Should return "status pending" (and the caller should
 989                  * not send out response now). The real response will be sent
 990                  * once the response from R0KH is received.
 991                  */
 992                 return WLAN_STATUS_INVALID_PMKID;
 993         }
 994
 995         wpa_hexdump_key(MSG_DEBUG, "FT: Selected PMK-R1", pmk_r1, PMK_LEN);
 996         sm->pmk_r1_name_valid = 1;
 997         os_memcpy(sm->pmk_r1_name, pmk_r1_name, WPA_PMK_NAME_LEN);
 998
 999         if (os_get_random(sm->ANonce, WPA_NONCE_LEN)) {
1000                 wpa_printf(MSG_DEBUG, "FT: Failed to get random data for "
1001                            "ANonce");
1002                 return WLAN_STATUS_UNSPECIFIED_FAILURE;
1003         }
1004
1005         wpa_hexdump(MSG_DEBUG, "FT: Received SNonce",
1006                     sm->SNonce, WPA_NONCE_LEN);
1007         wpa_hexdump(MSG_DEBUG, "FT: Generated ANonce",
1008                     sm->ANonce, WPA_NONCE_LEN);
1009
1010         ptk_len = pairwise != WPA_CIPHER_CCMP ? 64 : 48;
1011         wpa_pmk_r1_to_ptk(pmk_r1, sm->SNonce, sm->ANonce, sm->addr,
1012                           sm->wpa_auth->addr, pmk_r1_name,
1013                           (u8 *) &sm->PTK, ptk_len, ptk_name);
1014         wpa_hexdump_key(MSG_DEBUG, "FT: PTK",
1015                         (u8 *) &sm->PTK, ptk_len);
1016         wpa_hexdump(MSG_DEBUG, "FT: PTKName", ptk_name, WPA_PMK_NAME_LEN);
1017
1018         sm->pairwise = pairwise;
1019         wpa_ft_install_ptk(sm);
1020
1021         buflen = 2 + sizeof(struct rsn_mdie) + 2 + sizeof(struct rsn_ftie) +
1022                 2 + FT_R1KH_ID_LEN + 200;
1023         *resp_ies = os_zalloc(buflen);
1024         if (*resp_ies == NULL) {
1025                 return WLAN_STATUS_UNSPECIFIED_FAILURE;
1026         }
1027
1028         pos = *resp_ies;
1029         end = *resp_ies + buflen;
1030
1031         ret = wpa_write_rsn_ie(conf, pos, end - pos, parse.rsn_pmkid);
1032         if (ret < 0) {
1033                 os_free(*resp_ies);
1034                 *resp_ies = NULL;
1035                 return WLAN_STATUS_UNSPECIFIED_FAILURE;
1036         }
1037         pos += ret;
1038
1039         ret = wpa_write_mdie(conf, pos, end - pos);
1040         if (ret < 0) {
1041                 os_free(*resp_ies);
1042                 *resp_ies = NULL;
1043                 return WLAN_STATUS_UNSPECIFIED_FAILURE;
1044         }
1045         pos += ret;
1046
1047         ret = wpa_write_ftie(conf, parse.r0kh_id, parse.r0kh_id_len,
1048                              sm->ANonce, sm->SNonce, pos, end - pos, NULL, 0);
1049         if (ret < 0) {
1050                 os_free(*resp_ies);
1051                 *resp_ies = NULL;
1052                 return WLAN_STATUS_UNSPECIFIED_FAILURE;
1053         }
1054         pos += ret;
1055
1056         *resp_ies_len = pos - *resp_ies;
1057
1058         return WLAN_STATUS_SUCCESS;
1059 }
1060
1061
1062 void wpa_ft_process_auth(struct wpa_state_machine *sm, const u8 *bssid,
1063                          u16 auth_transaction, const u8 *ies, size_t ies_len,
1064                          void (*cb)(void *ctx, const u8 *dst, const u8 *bssid,
1065                                     u16 auth_transaction, u16 status,
1066                                     const u8 *ies, size_t ies_len),
1067                          void *ctx)
1068 {
1069         u16 status;
1070         u8 *resp_ies;
1071         size_t resp_ies_len;
1072
1073         if (sm == NULL) {
1074                 wpa_printf(MSG_DEBUG, "FT: Received authentication frame, but "
1075                            "WPA SM not available");
1076                 return;
1077         }
1078
1079         wpa_printf(MSG_DEBUG, "FT: Received authentication frame: STA=" MACSTR
1080                    " BSSID=" MACSTR " transaction=%d",
1081                    MAC2STR(sm->addr), MAC2STR(bssid), auth_transaction);
1082         status = wpa_ft_process_auth_req(sm, ies, ies_len, &resp_ies,
1083                                          &resp_ies_len);
1084
1085         wpa_printf(MSG_DEBUG, "FT: FT authentication response: dst=" MACSTR
1086                    " auth_transaction=%d status=%d",
1087                    MAC2STR(sm->addr), auth_transaction + 1, status);
1088         wpa_hexdump(MSG_DEBUG, "FT: Response IEs", resp_ies, resp_ies_len);
1089         cb(ctx, sm->addr, bssid, auth_transaction + 1, status,
1090            resp_ies, resp_ies_len);
1091         os_free(resp_ies);
1092 }
1093
1094
1095 u16 wpa_ft_validate_reassoc(struct wpa_state_machine *sm, const u8 *ies,
1096                             size_t ies_len)
1097 {
1098         struct wpa_ft_ies parse;
1099         struct rsn_mdie *mdie;
1100         struct rsn_ftie *ftie;
1101         u8 mic[16];
1102         unsigned int count;
1103
1104         if (sm == NULL)
1105                 return WLAN_STATUS_UNSPECIFIED_FAILURE;
1106
1107         wpa_hexdump(MSG_DEBUG, "FT: Reassoc Req IEs", ies, ies_len);
1108
1109         if (wpa_ft_parse_ies(ies, ies_len, &parse) < 0) {
1110                 wpa_printf(MSG_DEBUG, "FT: Failed to parse FT IEs");
1111                 return WLAN_STATUS_UNSPECIFIED_FAILURE;
1112         }
1113
1114         if (parse.rsn == NULL) {
1115                 wpa_printf(MSG_DEBUG, "FT: No RSNIE in Reassoc Req");
1116                 return WLAN_STATUS_UNSPECIFIED_FAILURE;
1117         }
1118
1119         if (parse.rsn_pmkid == NULL) {
1120                 wpa_printf(MSG_DEBUG, "FT: No PMKID in RSNIE");
1121                 return WLAN_STATUS_INVALID_PMKID;
1122         }
1123
1124         if (os_memcmp(parse.rsn_pmkid, sm->pmk_r1_name, WPA_PMK_NAME_LEN) != 0)
1125         {
1126                 wpa_printf(MSG_DEBUG, "FT: PMKID in Reassoc Req did not match "
1127                            "with the PMKR1Name derived from auth request");
1128                 return WLAN_STATUS_INVALID_PMKID;
1129         }
1130
1131         mdie = (struct rsn_mdie *) parse.mdie;
1132         if (mdie == NULL || parse.mdie_len < sizeof(*mdie) ||
1133             os_memcmp(mdie->mobility_domain,
1134                       sm->wpa_auth->conf.mobility_domain,
1135                       MOBILITY_DOMAIN_ID_LEN) != 0) {
1136                 wpa_printf(MSG_DEBUG, "FT: Invalid MDIE");
1137                 return WLAN_STATUS_INVALID_MDIE;
1138         }
1139
1140         ftie = (struct rsn_ftie *) parse.ftie;
1141         if (ftie == NULL || parse.ftie_len < sizeof(*ftie)) {
1142                 wpa_printf(MSG_DEBUG, "FT: Invalid FTIE");
1143                 return WLAN_STATUS_INVALID_FTIE;
1144         }
1145
1146         if (os_memcmp(ftie->snonce, sm->SNonce, WPA_NONCE_LEN) != 0) {
1147                 wpa_printf(MSG_DEBUG, "FT: SNonce mismatch in FTIE");
1148                 wpa_hexdump(MSG_DEBUG, "FT: Received SNonce",
1149                             ftie->snonce, WPA_NONCE_LEN);
1150                 wpa_hexdump(MSG_DEBUG, "FT: Expected SNonce",
1151                             sm->SNonce, WPA_NONCE_LEN);
1152                 return -1;
1153         }
1154
1155         if (os_memcmp(ftie->anonce, sm->ANonce, WPA_NONCE_LEN) != 0) {
1156                 wpa_printf(MSG_DEBUG, "FT: ANonce mismatch in FTIE");
1157                 wpa_hexdump(MSG_DEBUG, "FT: Received ANonce",
1158                             ftie->anonce, WPA_NONCE_LEN);
1159                 wpa_hexdump(MSG_DEBUG, "FT: Expected ANonce",
1160                             sm->ANonce, WPA_NONCE_LEN);
1161                 return -1;
1162         }
1163
1164
1165         if (parse.r0kh_id == NULL) {
1166                 wpa_printf(MSG_DEBUG, "FT: No R0KH-ID subelem in FTIE");
1167                 return -1;
1168         }
1169
1170         if (parse.r0kh_id_len != sm->r0kh_id_len ||
1171             os_memcmp(parse.r0kh_id, sm->r0kh_id, parse.r0kh_id_len) != 0) {
1172                 wpa_printf(MSG_DEBUG, "FT: R0KH-ID in FTIE did not match with "
1173                            "the current R0KH-ID");
1174                 wpa_hexdump(MSG_DEBUG, "FT: R0KH-ID in FTIE",
1175                             parse.r0kh_id, parse.r0kh_id_len);
1176                 wpa_hexdump(MSG_DEBUG, "FT: The current R0KH-ID",
1177                             sm->r0kh_id, sm->r0kh_id_len);
1178                 return -1;
1179         }
1180
1181         if (parse.r1kh_id == NULL) {
1182                 wpa_printf(MSG_DEBUG, "FT: No R1KH-ID subelem in FTIE");
1183                 return -1;
1184         }
1185
1186         if (os_memcmp(parse.r1kh_id, sm->wpa_auth->conf.r1_key_holder,
1187                       FT_R1KH_ID_LEN) != 0) {
1188                 wpa_printf(MSG_DEBUG, "FT: Unknown R1KH-ID used in "
1189                            "ReassocReq");
1190                 wpa_hexdump(MSG_DEBUG, "FT: R1KH-ID in FTIE",
1191                             parse.r1kh_id, FT_R1KH_ID_LEN);
1192                 wpa_hexdump(MSG_DEBUG, "FT: Expected R1KH-ID",
1193                             sm->wpa_auth->conf.r1_key_holder, FT_R1KH_ID_LEN);
1194                 return -1;
1195         }
1196
1197         if (parse.rsn_pmkid == NULL ||
1198             os_memcmp(parse.rsn_pmkid, sm->pmk_r1_name, WPA_PMK_NAME_LEN)) {
1199                 wpa_printf(MSG_DEBUG, "FT: No matching PMKR1Name (PMKID) in "
1200                            "RSNIE (pmkid=%d)", !!parse.rsn_pmkid);
1201                 return -1;
1202         }
1203
1204         count = 3;
1205         if (parse.ric)
1206                 count++;
1207         if (ftie->mic_control[1] != count) {
1208                 wpa_printf(MSG_DEBUG, "FT: Unexpected IE count in MIC "
1209                            "Control: received %u expected %u",
1210                            ftie->mic_control[1], count);
1211                 return -1;
1212         }
1213
1214         if (wpa_ft_mic(sm->PTK.kck, sm->addr, sm->wpa_auth->addr, 5,
1215                        parse.mdie - 2, parse.mdie_len + 2,
1216                        parse.ftie - 2, parse.ftie_len + 2,
1217                        parse.rsn - 2, parse.rsn_len + 2,
1218                        parse.ric, parse.ric_len,
1219                        mic) < 0) {
1220                 wpa_printf(MSG_DEBUG, "FT: Failed to calculate MIC");
1221                 return WLAN_STATUS_UNSPECIFIED_FAILURE;
1222         }
1223
1224         if (os_memcmp(mic, ftie->mic, 16) != 0) {
1225                 wpa_printf(MSG_DEBUG, "FT: Invalid MIC in FTIE");
1226                 wpa_hexdump(MSG_MSGDUMP, "FT: Received MIC", ftie->mic, 16);
1227                 wpa_hexdump(MSG_MSGDUMP, "FT: Calculated MIC", mic, 16);
1228                 return WLAN_STATUS_INVALID_FTIE;
1229         }
1230
1231         return WLAN_STATUS_SUCCESS;
1232 }
1233
1234
1235 int wpa_ft_action_rx(struct wpa_state_machine *sm, const u8 *data, size_t len)
1236 {
1237         const u8 *sta_addr, *target_ap;
1238         const u8 *ies;
1239         size_t ies_len;
1240         u8 action;
1241         struct ft_rrb_frame *frame;
1242
1243         if (sm == NULL)
1244                 return -1;
1245
1246         /*
1247          * data: Category[1] Action[1] STA_Address[6] Target_AP_Address[6]
1248          * FT Request action frame body[variable]
1249          */
1250
1251         if (len < 14) {
1252                 wpa_printf(MSG_DEBUG, "FT: Too short FT Action frame "
1253                            "(len=%lu)", (unsigned long) len);
1254                 return -1;
1255         }
1256
1257         action = data[1];
1258         sta_addr = data + 2;
1259         target_ap = data + 8;
1260         ies = data + 14;
1261         ies_len = len - 14;
1262
1263         wpa_printf(MSG_DEBUG, "FT: Received FT Action frame (STA=" MACSTR
1264                    " Target AP=" MACSTR " Action=%d)",
1265                    MAC2STR(sta_addr), MAC2STR(target_ap), action);
1266
1267         if (os_memcmp(sta_addr, sm->addr, ETH_ALEN) != 0) {
1268                 wpa_printf(MSG_DEBUG, "FT: Mismatch in FT Action STA address: "
1269                            "STA=" MACSTR " STA-Address=" MACSTR,
1270                            MAC2STR(sm->addr), MAC2STR(sta_addr));
1271                 return -1;
1272         }
1273
1274         /*
1275          * Do some sanity checking on the target AP address (not own and not
1276          * broadcast. This could be extended to filter based on a list of known
1277          * APs in the MD (if such a list were configured).
1278          */
1279         if ((target_ap[0] & 0x01) ||
1280             os_memcmp(target_ap, sm->wpa_auth->addr, ETH_ALEN) == 0) {
1281                 wpa_printf(MSG_DEBUG, "FT: Invalid Target AP in FT Action "
1282                            "frame");
1283                 return -1;
1284         }
1285
1286         wpa_hexdump(MSG_MSGDUMP, "FT: Action frame body", ies, ies_len);
1287
1288         /* RRB - Forward action frame to the target AP */
1289         frame = os_malloc(sizeof(*frame) + len);
1290         frame->frame_type = RSN_REMOTE_FRAME_TYPE_FT_RRB;
1291         frame->packet_type = FT_PACKET_REQUEST;
1292         frame->action_length = host_to_le16(len);
1293         os_memcpy(frame->ap_address, sm->wpa_auth->addr, ETH_ALEN);
1294         os_memcpy(frame + 1, data, len);
1295
1296         wpa_ft_rrb_send(sm->wpa_auth, target_ap, (u8 *) frame,
1297                         sizeof(*frame) + len);
1298         os_free(frame);
1299
1300         return 0;
1301 }
1302
1303
1304 static int wpa_ft_rrb_rx_request(struct wpa_authenticator *wpa_auth,
1305                                  const u8 *current_ap, const u8 *sta_addr,
1306                                  const u8 *body, size_t len)
1307 {
1308         struct wpa_state_machine *sm;
1309         u16 status;
1310         u8 *resp_ies, *pos;
1311         size_t resp_ies_len, rlen;
1312         struct ft_rrb_frame *frame;
1313
1314         sm = wpa_ft_add_sta(wpa_auth, sta_addr);
1315         if (sm == NULL) {
1316                 wpa_printf(MSG_DEBUG, "FT: Failed to add new STA based on "
1317                            "RRB Request");
1318                 return -1;
1319         }
1320
1321         wpa_hexdump(MSG_MSGDUMP, "FT: RRB Request Frame body", body, len);
1322
1323         status = wpa_ft_process_auth_req(sm, body, len, &resp_ies,
1324                                          &resp_ies_len);
1325
1326         wpa_printf(MSG_DEBUG, "FT: RRB authentication response: STA=" MACSTR
1327                    " CurrentAP=" MACSTR " status=%d",
1328                    MAC2STR(sm->addr), MAC2STR(current_ap), status);
1329         wpa_hexdump(MSG_DEBUG, "FT: Response IEs", resp_ies, resp_ies_len);
1330
1331         /* RRB - Forward action frame response to the Current AP */
1332
1333         /*
1334          * data: Category[1] Action[1] STA_Address[6] Target_AP_Address[6]
1335          * Status_Code[2] FT Request action frame body[variable]
1336          */
1337         rlen = 2 + 2 * ETH_ALEN + 2 + resp_ies_len;
1338
1339         frame = os_malloc(sizeof(*frame) + rlen);
1340         frame->frame_type = RSN_REMOTE_FRAME_TYPE_FT_RRB;
1341         frame->packet_type = FT_PACKET_RESPONSE;
1342         frame->action_length = host_to_le16(rlen);
1343         os_memcpy(frame->ap_address, wpa_auth->addr, ETH_ALEN);
1344         pos = (u8 *) (frame + 1);
1345         *pos++ = WLAN_ACTION_FT;
1346         *pos++ = 2; /* Action: Response */
1347         os_memcpy(pos, sta_addr, ETH_ALEN);
1348         pos += ETH_ALEN;
1349         os_memcpy(pos, wpa_auth->addr, ETH_ALEN);
1350         pos += ETH_ALEN;
1351         WPA_PUT_LE16(pos, status);
1352         pos += 2;
1353         if (resp_ies) {
1354                 os_memcpy(pos, resp_ies, resp_ies_len);
1355                 os_free(resp_ies);
1356         }
1357
1358         wpa_ft_rrb_send(wpa_auth, current_ap, (u8 *) frame,
1359                         sizeof(*frame) + rlen);
1360         os_free(frame);
1361
1362         return 0;
1363 }
1364
1365
1366 static int wpa_ft_rrb_rx_pull(struct wpa_authenticator *wpa_auth,
1367                               const u8 *src_addr,
1368                               const u8 *data, size_t data_len)
1369 {
1370         struct ft_r0kh_r1kh_pull_frame *frame, f;
1371         struct ft_remote_r1kh *r1kh;
1372         struct ft_r0kh_r1kh_resp_frame resp, r;
1373         u8 pmk_r0[PMK_LEN];
1374         int pairwise;
1375
1376         wpa_printf(MSG_DEBUG, "FT: Received PMK-R1 pull");
1377
1378         if (data_len < sizeof(*frame))
1379                 return -1;
1380
1381         r1kh = wpa_auth->conf.r1kh_list;
1382         while (r1kh) {
1383                 if (os_memcmp(r1kh->addr, src_addr, ETH_ALEN) == 0)
1384                         break;
1385                 r1kh = r1kh->next;
1386         }
1387         if (r1kh == NULL) {
1388                 wpa_printf(MSG_DEBUG, "FT: No matching R1KH address found for "
1389                            "PMK-R1 pull source address " MACSTR,
1390                            MAC2STR(src_addr));
1391                 return -1;
1392         }
1393
1394         frame = (struct ft_r0kh_r1kh_pull_frame *) data;
1395         /* aes_unwrap() does not support inplace decryption, so use a temporary
1396          * buffer for the data. */
1397         if (aes_unwrap(r1kh->key, (FT_R0KH_R1KH_PULL_DATA_LEN + 7) / 8,
1398                        frame->nonce, f.nonce) < 0) {
1399                 wpa_printf(MSG_DEBUG, "FT: Failed to decrypt PMK-R1 pull "
1400                            "request from " MACSTR, MAC2STR(src_addr));
1401                 return -1;
1402         }
1403
1404         wpa_hexdump(MSG_DEBUG, "FT: PMK-R1 pull - nonce",
1405                     f.nonce, sizeof(f.nonce));
1406         wpa_hexdump(MSG_DEBUG, "FT: PMK-R1 pull - PMKR0Name",
1407                     f.pmk_r0_name, WPA_PMK_NAME_LEN);
1408         wpa_printf(MSG_DEBUG, "FT: PMK-R1 pull - R1KH-ID=" MACSTR "S1KH-ID="
1409                    MACSTR, MAC2STR(f.r1kh_id), MAC2STR(f.s1kh_id));
1410
1411         os_memset(&resp, 0, sizeof(resp));
1412         resp.frame_type = RSN_REMOTE_FRAME_TYPE_FT_RRB;
1413         resp.packet_type = FT_PACKET_R0KH_R1KH_RESP;
1414         resp.data_length = host_to_le16(FT_R0KH_R1KH_RESP_DATA_LEN);
1415         os_memcpy(resp.ap_address, wpa_auth->addr, ETH_ALEN);
1416
1417         /* aes_wrap() does not support inplace encryption, so use a temporary
1418          * buffer for the data. */
1419         os_memcpy(r.nonce, f.nonce, sizeof(f.nonce));
1420         os_memcpy(r.r1kh_id, f.r1kh_id, FT_R1KH_ID_LEN);
1421         os_memcpy(r.s1kh_id, f.s1kh_id, ETH_ALEN);
1422         if (wpa_ft_fetch_pmk_r0(wpa_auth, f.s1kh_id, f.pmk_r0_name, pmk_r0,
1423                                 &pairwise) < 0) {
1424                 wpa_printf(MSG_DEBUG, "FT: No matching PMKR0Name found for "
1425                            "PMK-R1 pull");
1426                 return -1;
1427         }
1428
1429         wpa_derive_pmk_r1(pmk_r0, f.pmk_r0_name, f.r1kh_id, f.s1kh_id,
1430                           r.pmk_r1, r.pmk_r1_name);
1431         wpa_hexdump_key(MSG_DEBUG, "FT: PMK-R1", r.pmk_r1, PMK_LEN);
1432         wpa_hexdump(MSG_DEBUG, "FT: PMKR1Name", r.pmk_r1_name,
1433                     WPA_PMK_NAME_LEN);
1434         r.pairwise = pairwise;
1435
1436         if (aes_wrap(r1kh->key, (FT_R0KH_R1KH_RESP_DATA_LEN + 7) / 8,
1437                      r.nonce, resp.nonce) < 0) {
1438                 os_memset(pmk_r0, 0, PMK_LEN);
1439                 return -1;
1440         }
1441
1442         os_memset(pmk_r0, 0, PMK_LEN);
1443
1444         wpa_ft_rrb_send(wpa_auth, src_addr, (u8 *) &resp, sizeof(resp));
1445
1446         return 0;
1447 }
1448
1449
1450 static int wpa_ft_rrb_rx_resp(struct wpa_authenticator *wpa_auth,
1451                               const u8 *src_addr,
1452                               const u8 *data, size_t data_len)
1453 {
1454         struct ft_r0kh_r1kh_resp_frame *frame, f;
1455         struct ft_remote_r0kh *r0kh;
1456         int pairwise;
1457
1458         wpa_printf(MSG_DEBUG, "FT: Received PMK-R1 pull response");
1459
1460         if (data_len < sizeof(*frame))
1461                 return -1;
1462
1463         r0kh = wpa_auth->conf.r0kh_list;
1464         while (r0kh) {
1465                 if (os_memcmp(r0kh->addr, src_addr, ETH_ALEN) == 0)
1466                         break;
1467                 r0kh = r0kh->next;
1468         }
1469         if (r0kh == NULL) {
1470                 wpa_printf(MSG_DEBUG, "FT: No matching R0KH address found for "
1471                            "PMK-R0 pull response source address " MACSTR,
1472                            MAC2STR(src_addr));
1473                 return -1;
1474         }
1475
1476         frame = (struct ft_r0kh_r1kh_resp_frame *) data;
1477         /* aes_unwrap() does not support inplace decryption, so use a temporary
1478          * buffer for the data. */
1479         if (aes_unwrap(r0kh->key, (FT_R0KH_R1KH_RESP_DATA_LEN + 7) / 8,
1480                        frame->nonce, f.nonce) < 0) {
1481                 wpa_printf(MSG_DEBUG, "FT: Failed to decrypt PMK-R1 pull "
1482                            "response from " MACSTR, MAC2STR(src_addr));
1483                 return -1;
1484         }
1485
1486         if (os_memcmp(f.r1kh_id, wpa_auth->conf.r1_key_holder, FT_R1KH_ID_LEN)
1487             != 0) {
1488                 wpa_printf(MSG_DEBUG, "FT: PMK-R1 pull response did not use a "
1489                            "matching R1KH-ID");
1490                 return -1;
1491         }
1492
1493         /* TODO: verify that <nonce,s1kh_id> matches with a pending request
1494          * and call this requests callback function to finish request
1495          * processing */
1496
1497         pairwise = le_to_host16(f.pairwise);
1498         wpa_hexdump(MSG_DEBUG, "FT: PMK-R1 pull - nonce",
1499                     f.nonce, sizeof(f.nonce));
1500         wpa_printf(MSG_DEBUG, "FT: PMK-R1 pull - R1KH-ID=" MACSTR "S1KH-ID="
1501                    MACSTR " pairwise=0x%x",
1502                    MAC2STR(f.r1kh_id), MAC2STR(f.s1kh_id), pairwise);
1503         wpa_hexdump_key(MSG_DEBUG, "FT: PMK-R1 pull - PMK-R1",
1504                         f.pmk_r1, PMK_LEN);
1505         wpa_hexdump(MSG_DEBUG, "FT: PMK-R1 pull - PMKR1Name",
1506                         f.pmk_r1_name, WPA_PMK_NAME_LEN);
1507
1508         wpa_ft_store_pmk_r1(wpa_auth, f.s1kh_id, f.pmk_r1, f.pmk_r1_name,
1509                             pairwise);
1510         os_memset(f.pmk_r1, 0, PMK_LEN);
1511
1512         return 0;
1513 }
1514
1515
1516 static int wpa_ft_rrb_rx_push(struct wpa_authenticator *wpa_auth,
1517                               const u8 *src_addr,
1518                               const u8 *data, size_t data_len)
1519 {
1520         struct ft_r0kh_r1kh_push_frame *frame, f;
1521         struct ft_remote_r0kh *r0kh;
1522         struct os_time now;
1523         os_time_t tsend;
1524         int pairwise;
1525
1526         wpa_printf(MSG_DEBUG, "FT: Received PMK-R1 push");
1527
1528         if (data_len < sizeof(*frame))
1529                 return -1;
1530
1531         r0kh = wpa_auth->conf.r0kh_list;
1532         while (r0kh) {
1533                 if (os_memcmp(r0kh->addr, src_addr, ETH_ALEN) == 0)
1534                         break;
1535                 r0kh = r0kh->next;
1536         }
1537         if (r0kh == NULL) {
1538                 wpa_printf(MSG_DEBUG, "FT: No matching R0KH address found for "
1539                            "PMK-R0 push source address " MACSTR,
1540                            MAC2STR(src_addr));
1541                 return -1;
1542         }
1543
1544         frame = (struct ft_r0kh_r1kh_push_frame *) data;
1545         /* aes_unwrap() does not support inplace decryption, so use a temporary
1546          * buffer for the data. */
1547         if (aes_unwrap(r0kh->key, (FT_R0KH_R1KH_PUSH_DATA_LEN + 7) / 8,
1548                        frame->timestamp, f.timestamp) < 0) {
1549                 wpa_printf(MSG_DEBUG, "FT: Failed to decrypt PMK-R1 push from "
1550                            MACSTR, MAC2STR(src_addr));
1551                 return -1;
1552         }
1553
1554         os_get_time(&now);
1555         tsend = WPA_GET_LE32(f.timestamp);
1556         if ((now.sec > tsend && now.sec - tsend > 60) ||
1557             (now.sec < tsend && tsend - now.sec > 60)) {
1558                 wpa_printf(MSG_DEBUG, "FT: PMK-R1 push did not have a valid "
1559                            "timestamp: sender time %d own time %d\n",
1560                            (int) tsend, (int) now.sec);
1561                 return -1;
1562         }
1563
1564         if (os_memcmp(f.r1kh_id, wpa_auth->conf.r1_key_holder, FT_R1KH_ID_LEN)
1565             != 0) {
1566                 wpa_printf(MSG_DEBUG, "FT: PMK-R1 push did not use a matching "
1567                            "R1KH-ID (received " MACSTR " own " MACSTR ")",
1568                            MAC2STR(f.r1kh_id),
1569                            MAC2STR(wpa_auth->conf.r1_key_holder));
1570                 return -1;
1571         }
1572
1573         pairwise = le_to_host16(f.pairwise);
1574         wpa_printf(MSG_DEBUG, "FT: PMK-R1 push - R1KH-ID=" MACSTR " S1KH-ID="
1575                    MACSTR " pairwise=0x%x",
1576                    MAC2STR(f.r1kh_id), MAC2STR(f.s1kh_id), pairwise);
1577         wpa_hexdump_key(MSG_DEBUG, "FT: PMK-R1 push - PMK-R1",
1578                         f.pmk_r1, PMK_LEN);
1579         wpa_hexdump(MSG_DEBUG, "FT: PMK-R1 push - PMKR1Name",
1580                         f.pmk_r1_name, WPA_PMK_NAME_LEN);
1581
1582         wpa_ft_store_pmk_r1(wpa_auth, f.s1kh_id, f.pmk_r1, f.pmk_r1_name,
1583                             pairwise);
1584         os_memset(f.pmk_r1, 0, PMK_LEN);
1585
1586         return 0;
1587 }
1588
1589
1590 int wpa_ft_rrb_rx(struct wpa_authenticator *wpa_auth, const u8 *src_addr,
1591                   const u8 *data, size_t data_len)
1592 {
1593         struct ft_rrb_frame *frame;
1594         u16 alen;
1595         const u8 *pos, *end, *start;
1596         u8 action;
1597         const u8 *sta_addr, *target_ap_addr;
1598
1599         wpa_printf(MSG_DEBUG, "FT: RRB received frame from remote AP " MACSTR,
1600                    MAC2STR(src_addr));
1601
1602         if (data_len < sizeof(*frame)) {
1603                 wpa_printf(MSG_DEBUG, "FT: Too short RRB frame (data_len=%lu)",
1604                            (unsigned long) data_len);
1605                 return -1;
1606         }
1607
1608         pos = data;
1609         frame = (struct ft_rrb_frame *) pos;
1610         pos += sizeof(*frame);
1611
1612         alen = le_to_host16(frame->action_length);
1613         wpa_printf(MSG_DEBUG, "FT: RRB frame - frame_type=%d packet_type=%d "
1614                    "action_length=%d ap_address=" MACSTR,
1615                    frame->frame_type, frame->packet_type, alen,
1616                    MAC2STR(frame->ap_address));
1617
1618         if (frame->frame_type != RSN_REMOTE_FRAME_TYPE_FT_RRB) {
1619                 /* Discard frame per IEEE Std 802.11r-2008, 11A.10.3 */
1620                 wpa_printf(MSG_DEBUG, "FT: RRB discarded frame with "
1621                            "unrecognized type %d", frame->frame_type);
1622                 return -1;
1623         }
1624
1625         if (alen > data_len - sizeof(*frame)) {
1626                 wpa_printf(MSG_DEBUG, "FT: RRB frame too short for action "
1627                            "frame");
1628                 return -1;
1629         }
1630
1631         if (frame->packet_type == FT_PACKET_R0KH_R1KH_PULL)
1632                 return wpa_ft_rrb_rx_pull(wpa_auth, src_addr, data, data_len);
1633         if (frame->packet_type == FT_PACKET_R0KH_R1KH_RESP)
1634                 return wpa_ft_rrb_rx_resp(wpa_auth, src_addr, data, data_len);
1635         if (frame->packet_type == FT_PACKET_R0KH_R1KH_PUSH)
1636                 return wpa_ft_rrb_rx_push(wpa_auth, src_addr, data, data_len);
1637
1638         wpa_hexdump(MSG_MSGDUMP, "FT: RRB - FT Action frame", pos, alen);
1639
1640         if (alen < 1 + 1 + 2 * ETH_ALEN) {
1641                 wpa_printf(MSG_DEBUG, "FT: Too short RRB frame (not enough "
1642                            "room for Action Frame body); alen=%lu",
1643                            (unsigned long) alen);
1644                 return -1;
1645         }
1646         start = pos;
1647         end = pos + alen;
1648
1649         if (*pos != WLAN_ACTION_FT) {
1650                 wpa_printf(MSG_DEBUG, "FT: Unexpected Action frame category "
1651                            "%d", *pos);
1652                 return -1;
1653         }
1654
1655         pos++;
1656         action = *pos++;
1657         sta_addr = pos;
1658         pos += ETH_ALEN;
1659         target_ap_addr = pos;
1660         pos += ETH_ALEN;
1661         wpa_printf(MSG_DEBUG, "FT: RRB Action Frame: action=%d sta_addr="
1662                    MACSTR " target_ap_addr=" MACSTR,
1663                    action, MAC2STR(sta_addr), MAC2STR(target_ap_addr));
1664
1665         if (frame->packet_type == FT_PACKET_REQUEST) {
1666                 wpa_printf(MSG_DEBUG, "FT: FT Packet Type - Request");
1667
1668                 if (action != 1) {
1669                         wpa_printf(MSG_DEBUG, "FT: Unexpected Action %d in "
1670                                    "RRB Request", action);
1671                         return -1;
1672                 }
1673
1674                 if (os_memcmp(target_ap_addr, wpa_auth->addr, ETH_ALEN) != 0) {
1675                         wpa_printf(MSG_DEBUG, "FT: Target AP address in the "
1676                                    "RRB Request does not match with own "
1677                                    "address");
1678                         return -1;
1679                 }
1680
1681                 if (wpa_ft_rrb_rx_request(wpa_auth, frame->ap_address,
1682                                           sta_addr, pos, end - pos) < 0)
1683                         return -1;
1684         } else if (frame->packet_type == FT_PACKET_RESPONSE) {
1685                 u16 status_code;
1686
1687                 if (end - pos < 2) {
1688                         wpa_printf(MSG_DEBUG, "FT: Not enough room for status "
1689                                    "code in RRB Response");
1690                         return -1;
1691                 }
1692                 status_code = WPA_GET_LE16(pos);
1693                 pos += 2;
1694
1695                 wpa_printf(MSG_DEBUG, "FT: FT Packet Type - Response "
1696                            "(status_code=%d)", status_code);
1697
1698                 if (wpa_ft_action_send(wpa_auth, sta_addr, start, alen) < 0)
1699                         return -1;
1700         } else {
1701                 wpa_printf(MSG_DEBUG, "FT: RRB discarded frame with unknown "
1702                            "packet_type %d", frame->packet_type);
1703                 return -1;
1704         }
1705
1706         return 0;
1707 }
1708
1709
1710 static void wpa_ft_generate_pmk_r1(struct wpa_authenticator *wpa_auth,
1711                                    struct wpa_ft_pmk_r0_sa *pmk_r0,
1712                                    struct ft_remote_r1kh *r1kh,
1713                                    const u8 *s1kh_id, int pairwise)
1714 {
1715         struct ft_r0kh_r1kh_push_frame frame, f;
1716         struct os_time now;
1717
1718         os_memset(&frame, 0, sizeof(frame));
1719         frame.frame_type = RSN_REMOTE_FRAME_TYPE_FT_RRB;
1720         frame.packet_type = FT_PACKET_R0KH_R1KH_PUSH;
1721         frame.data_length = host_to_le16(FT_R0KH_R1KH_PUSH_DATA_LEN);
1722         os_memcpy(frame.ap_address, wpa_auth->addr, ETH_ALEN);
1723
1724         /* aes_wrap() does not support inplace encryption, so use a temporary
1725          * buffer for the data. */
1726         os_memcpy(f.r1kh_id, r1kh->id, FT_R1KH_ID_LEN);
1727         os_memcpy(f.s1kh_id, s1kh_id, ETH_ALEN);
1728         os_memcpy(f.pmk_r0_name, pmk_r0->pmk_r0_name, WPA_PMK_NAME_LEN);
1729         wpa_derive_pmk_r1(pmk_r0->pmk_r0, pmk_r0->pmk_r0_name, r1kh->id,
1730                           s1kh_id, f.pmk_r1, f.pmk_r1_name);
1731         wpa_printf(MSG_DEBUG, "FT: R1KH-ID " MACSTR, MAC2STR(r1kh->id));
1732         wpa_hexdump_key(MSG_DEBUG, "FT: PMK-R1", f.pmk_r1, PMK_LEN);
1733         wpa_hexdump(MSG_DEBUG, "FT: PMKR1Name", f.pmk_r1_name,
1734                     WPA_PMK_NAME_LEN);
1735         os_get_time(&now);
1736         WPA_PUT_LE32(f.timestamp, now.sec);
1737         f.pairwise = pairwise;
1738         if (aes_wrap(r1kh->key, (FT_R0KH_R1KH_PUSH_DATA_LEN + 7) / 8,
1739                      f.timestamp, frame.timestamp) < 0)
1740                 return;
1741
1742         wpa_ft_rrb_send(wpa_auth, r1kh->addr, (u8 *) &frame, sizeof(frame));
1743 }
1744
1745
1746 void wpa_ft_push_pmk_r1(struct wpa_authenticator *wpa_auth, const u8 *addr)
1747 {
1748         struct wpa_ft_pmk_r0_sa *r0;
1749         struct ft_remote_r1kh *r1kh;
1750
1751         if (!wpa_auth->conf.pmk_r1_push)
1752                 return;
1753
1754         r0 = wpa_auth->ft_pmk_cache->pmk_r0;
1755         while (r0) {
1756                 if (os_memcmp(r0->spa, addr, ETH_ALEN) == 0)
1757                         break;
1758                 r0 = r0->next;
1759         }
1760
1761         if (r0 == NULL || r0->pmk_r1_pushed)
1762                 return;
1763         r0->pmk_r1_pushed = 1;
1764
1765         wpa_printf(MSG_DEBUG, "FT: Deriving and pushing PMK-R1 keys to R1KHs "
1766                    "for STA " MACSTR, MAC2STR(addr));
1767
1768         r1kh = wpa_auth->conf.r1kh_list;
1769         while (r1kh) {
1770                 wpa_ft_generate_pmk_r1(wpa_auth, r0, r1kh, addr, r0->pairwise);
1771                 r1kh = r1kh->next;
1772         }
1773 }
1774
1775 #endif /* CONFIG_IEEE80211R */