net/mac80211/wpa.c

   1 /*
   2  * Copyright 2002-2004, Instant802 Networks, Inc.
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License version 2 as
   6  * published by the Free Software Foundation.
   7  */
   8
   9 #include <linux/netdevice.h>
  10 #include <linux/types.h>
  11 #include <linux/slab.h>
  12 #include <linux/skbuff.h>
  13 #include <linux/compiler.h>
  14 #include <net/mac80211.h>
  15
  16 #include "ieee80211_i.h"
  17 #include "michael.h"
  18 #include "tkip.h"
  19 #include "aes_ccm.h"
  20 #include "wpa.h"
  21
  22 static int ieee80211_get_hdr_info(const struct sk_buff *skb, u8 **sa, u8 **da,
  23                                   u8 *qos_tid, u8 **data, size_t *data_len)
  24 {
  25         struct ieee80211_hdr *hdr;
  26         size_t hdrlen;
  27         __le16 fc;
  28
  29         hdr = (struct ieee80211_hdr *)skb->data;
  30         fc = hdr->frame_control;
  31
  32         hdrlen = ieee80211_hdrlen(fc);
  33
  34         *sa = ieee80211_get_SA(hdr);
  35         *da = ieee80211_get_DA(hdr);
  36
  37         *data = skb->data + hdrlen;
  38         *data_len = skb->len - hdrlen;
  39
  40         if (ieee80211_is_data_qos(fc))
  41                 *qos_tid = (*ieee80211_get_qos_ctl(hdr) & 0x0f) | 0x80;
  42         else
  43                 *qos_tid = 0;
  44
  45         return skb->len < hdrlen ? -1 : 0;
  46 }
  47
  48
  49 ieee80211_tx_result
  50 ieee80211_tx_h_michael_mic_add(struct ieee80211_tx_data *tx)
  51 {
  52         u8 *data, *sa, *da, *key, *mic, qos_tid;
  53         size_t data_len;
  54         u16 fc;
  55         struct sk_buff *skb = tx->skb;
  56         int authenticator;
  57         int wpa_test = 0;
  58         int tail;
  59
  60         fc = tx->fc;
  61
  62         if (!tx->key || tx->key->conf.alg != ALG_TKIP || skb->len < 24 ||
  63             !WLAN_FC_DATA_PRESENT(fc))
  64                 return TX_CONTINUE;
  65
  66         if (ieee80211_get_hdr_info(skb, &sa, &da, &qos_tid, &data, &data_len))
  67                 return TX_DROP;
  68
  69         if ((tx->key->flags & KEY_FLAG_UPLOADED_TO_HARDWARE) &&
  70             !(tx->flags & IEEE80211_TX_FRAGMENTED) &&
  71             !(tx->key->conf.flags & IEEE80211_KEY_FLAG_GENERATE_MMIC) &&
  72             !wpa_test) {
  73                 /* hwaccel - with no need for preallocated room for Michael MIC
  74                  */
  75                 return TX_CONTINUE;
  76         }
  77
  78         tail = MICHAEL_MIC_LEN;
  79         if (!(tx->key->flags & KEY_FLAG_UPLOADED_TO_HARDWARE))
  80                 tail += TKIP_ICV_LEN;
  81
  82         if (WARN_ON(skb_tailroom(skb) < tail ||
  83                     skb_headroom(skb) < TKIP_IV_LEN))
  84                 return TX_DROP;
  85
  86 #if 0
  87         authenticator = fc & IEEE80211_FCTL_FROMDS; /* FIX */
  88 #else
  89         authenticator = 1;
  90 #endif
  91         key = &tx->key->conf.key[authenticator ? ALG_TKIP_TEMP_AUTH_TX_MIC_KEY :
  92                                  ALG_TKIP_TEMP_AUTH_RX_MIC_KEY];
  93         mic = skb_put(skb, MICHAEL_MIC_LEN);
  94         michael_mic(key, da, sa, qos_tid & 0x0f, data, data_len, mic);
  95
  96         return TX_CONTINUE;
  97 }
  98
  99
 100 ieee80211_rx_result
 101 ieee80211_rx_h_michael_mic_verify(struct ieee80211_rx_data *rx)
 102 {
 103         u8 *data, *sa, *da, *key = NULL, qos_tid;
 104         size_t data_len;
 105         u16 fc;
 106         u8 mic[MICHAEL_MIC_LEN];
 107         struct sk_buff *skb = rx->skb;
 108         int authenticator = 1, wpa_test = 0;
 109         DECLARE_MAC_BUF(mac);
 110
 111         fc = rx->fc;
 112
 113         /*
 114          * No way to verify the MIC if the hardware stripped it
 115          */
 116         if (rx->status->flag & RX_FLAG_MMIC_STRIPPED)
 117                 return RX_CONTINUE;
 118
 119         if (!rx->key || rx->key->conf.alg != ALG_TKIP ||
 120             !(rx->fc & IEEE80211_FCTL_PROTECTED) || !WLAN_FC_DATA_PRESENT(fc))
 121                 return RX_CONTINUE;
 122
 123         if (ieee80211_get_hdr_info(skb, &sa, &da, &qos_tid, &data, &data_len)
 124             || data_len < MICHAEL_MIC_LEN)
 125                 return RX_DROP_UNUSABLE;
 126
 127         data_len -= MICHAEL_MIC_LEN;
 128
 129 #if 0
 130         authenticator = fc & IEEE80211_FCTL_TODS; /* FIX */
 131 #else
 132         authenticator = 1;
 133 #endif
 134         key = &rx->key->conf.key[authenticator ? ALG_TKIP_TEMP_AUTH_RX_MIC_KEY :
 135                                  ALG_TKIP_TEMP_AUTH_TX_MIC_KEY];
 136         michael_mic(key, da, sa, qos_tid & 0x0f, data, data_len, mic);
 137         if (memcmp(mic, data + data_len, MICHAEL_MIC_LEN) != 0 || wpa_test) {
 138                 if (!(rx->flags & IEEE80211_RX_RA_MATCH))
 139                         return RX_DROP_UNUSABLE;
 140
 141                 printk(KERN_DEBUG "%s: invalid Michael MIC in data frame from "
 142                        "%s\n", rx->dev->name, print_mac(mac, sa));
 143
 144                 mac80211_ev_michael_mic_failure(rx->dev, rx->key->conf.keyidx,
 145                                                 (void *) skb->data);
 146                 return RX_DROP_UNUSABLE;
 147         }
 148
 149         /* remove Michael MIC from payload */
 150         skb_trim(skb, skb->len - MICHAEL_MIC_LEN);
 151
 152         /* update IV in key information to be able to detect replays */
 153         rx->key->u.tkip.rx[rx->queue].iv32 = rx->tkip_iv32;
 154         rx->key->u.tkip.rx[rx->queue].iv16 = rx->tkip_iv16;
 155
 156         return RX_CONTINUE;
 157 }
 158
 159
 160 static int tkip_encrypt_skb(struct ieee80211_tx_data *tx, struct sk_buff *skb)
 161 {
 162         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
 163         struct ieee80211_key *key = tx->key;
 164         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
 165         unsigned int hdrlen;
 166         int len, tail;
 167         u8 *pos;
 168
 169         info->control.icv_len = TKIP_ICV_LEN;
 170         info->control.iv_len = TKIP_IV_LEN;
 171
 172         if ((tx->key->flags & KEY_FLAG_UPLOADED_TO_HARDWARE) &&
 173             !(tx->key->conf.flags & IEEE80211_KEY_FLAG_GENERATE_IV)) {
 174                 /* hwaccel - with no need for preallocated room for IV/ICV */
 175                 info->control.hw_key = &tx->key->conf;
 176                 return 0;
 177         }
 178
 179         hdrlen = ieee80211_hdrlen(hdr->frame_control);
 180         len = skb->len - hdrlen;
 181
 182         if (tx->key->flags & KEY_FLAG_UPLOADED_TO_HARDWARE)
 183                 tail = 0;
 184         else
 185                 tail = TKIP_ICV_LEN;
 186
 187         if (WARN_ON(skb_tailroom(skb) < tail ||
 188                     skb_headroom(skb) < TKIP_IV_LEN))
 189                 return -1;
 190
 191         pos = skb_push(skb, TKIP_IV_LEN);
 192         memmove(pos, pos + TKIP_IV_LEN, hdrlen);
 193         pos += hdrlen;
 194
 195         /* Increase IV for the frame */
 196         key->u.tkip.tx.iv16++;
 197         if (key->u.tkip.tx.iv16 == 0)
 198                 key->u.tkip.tx.iv32++;
 199
 200         if (tx->key->flags & KEY_FLAG_UPLOADED_TO_HARDWARE) {
 201                 hdr = (struct ieee80211_hdr *)skb->data;
 202
 203                 /* hwaccel - with preallocated room for IV */
 204                 ieee80211_tkip_add_iv(pos, key,
 205                                       (u8) (key->u.tkip.tx.iv16 >> 8),
 206                                       (u8) (((key->u.tkip.tx.iv16 >> 8) | 0x20) &
 207                                             0x7f),
 208                                       (u8) key->u.tkip.tx.iv16);
 209
 210                 info->control.hw_key = &tx->key->conf;
 211                 return 0;
 212         }
 213
 214         /* Add room for ICV */
 215         skb_put(skb, TKIP_ICV_LEN);
 216
 217         hdr = (struct ieee80211_hdr *) skb->data;
 218         ieee80211_tkip_encrypt_data(tx->local->wep_tx_tfm,
 219                                     key, pos, len, hdr->addr2);
 220         return 0;
 221 }
 222
 223
 224 ieee80211_tx_result
 225 ieee80211_crypto_tkip_encrypt(struct ieee80211_tx_data *tx)
 226 {
 227         struct sk_buff *skb = tx->skb;
 228
 229         ieee80211_tx_set_protected(tx);
 230
 231         if (tkip_encrypt_skb(tx, skb) < 0)
 232                 return TX_DROP;
 233
 234         if (tx->extra_frag) {
 235                 int i;
 236                 for (i = 0; i < tx->num_extra_frag; i++) {
 237                         if (tkip_encrypt_skb(tx, tx->extra_frag[i]) < 0)
 238                                 return TX_DROP;
 239                 }
 240         }
 241
 242         return TX_CONTINUE;
 243 }
 244
 245
 246 ieee80211_rx_result
 247 ieee80211_crypto_tkip_decrypt(struct ieee80211_rx_data *rx)
 248 {
 249         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) rx->skb->data;
 250         int hdrlen, res, hwaccel = 0, wpa_test = 0;
 251         struct ieee80211_key *key = rx->key;
 252         struct sk_buff *skb = rx->skb;
 253         DECLARE_MAC_BUF(mac);
 254
 255         hdrlen = ieee80211_hdrlen(hdr->frame_control);
 256
 257         if ((rx->fc & IEEE80211_FCTL_FTYPE) != IEEE80211_FTYPE_DATA)
 258                 return RX_CONTINUE;
 259
 260         if (!rx->sta || skb->len - hdrlen < 12)
 261                 return RX_DROP_UNUSABLE;
 262
 263         if (rx->status->flag & RX_FLAG_DECRYPTED) {
 264                 if (rx->status->flag & RX_FLAG_IV_STRIPPED) {
 265                         /*
 266                          * Hardware took care of all processing, including
 267                          * replay protection, and stripped the ICV/IV so
 268                          * we cannot do any checks here.
 269                          */
 270                         return RX_CONTINUE;
 271                 }
 272
 273                 /* let TKIP code verify IV, but skip decryption */
 274                 hwaccel = 1;
 275         }
 276
 277         res = ieee80211_tkip_decrypt_data(rx->local->wep_rx_tfm,
 278                                           key, skb->data + hdrlen,
 279                                           skb->len - hdrlen, rx->sta->addr,
 280                                           hdr->addr1, hwaccel, rx->queue,
 281                                           &rx->tkip_iv32,
 282                                           &rx->tkip_iv16);
 283         if (res != TKIP_DECRYPT_OK || wpa_test) {
 284 #ifdef CONFIG_MAC80211_DEBUG
 285                 if (net_ratelimit())
 286                         printk(KERN_DEBUG "%s: TKIP decrypt failed for RX "
 287                                "frame from %s (res=%d)\n", rx->dev->name,
 288                                print_mac(mac, rx->sta->addr), res);
 289 #endif /* CONFIG_MAC80211_DEBUG */
 290                 return RX_DROP_UNUSABLE;
 291         }
 292
 293         /* Trim ICV */
 294         skb_trim(skb, skb->len - TKIP_ICV_LEN);
 295
 296         /* Remove IV */
 297         memmove(skb->data + TKIP_IV_LEN, skb->data, hdrlen);
 298         skb_pull(skb, TKIP_IV_LEN);
 299
 300         return RX_CONTINUE;
 301 }
 302
 303
 304 static void ccmp_special_blocks(struct sk_buff *skb, u8 *pn, u8 *b_0, u8 *aad,
 305                                 int encrypted)
 306 {
 307         u16 fc;
 308         int a4_included, qos_included;
 309         u8 qos_tid, *fc_pos, *data, *sa, *da;
 310         int len_a;
 311         size_t data_len;
 312         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
 313
 314         fc_pos = (u8 *) &hdr->frame_control;
 315         fc = fc_pos[0] ^ (fc_pos[1] << 8);
 316         a4_included = (fc & (IEEE80211_FCTL_TODS | IEEE80211_FCTL_FROMDS)) ==
 317                 (IEEE80211_FCTL_TODS | IEEE80211_FCTL_FROMDS);
 318
 319         ieee80211_get_hdr_info(skb, &sa, &da, &qos_tid, &data, &data_len);
 320         data_len -= CCMP_HDR_LEN + (encrypted ? CCMP_MIC_LEN : 0);
 321         if (qos_tid & 0x80) {
 322                 qos_included = 1;
 323                 qos_tid &= 0x0f;
 324         } else
 325                 qos_included = 0;
 326         /* First block, b_0 */
 327
 328         b_0[0] = 0x59; /* flags: Adata: 1, M: 011, L: 001 */
 329         /* Nonce: QoS Priority | A2 | PN */
 330         b_0[1] = qos_tid;
 331         memcpy(&b_0[2], hdr->addr2, 6);
 332         memcpy(&b_0[8], pn, CCMP_PN_LEN);
 333         /* l(m) */
 334         b_0[14] = (data_len >> 8) & 0xff;
 335         b_0[15] = data_len & 0xff;
 336
 337
 338         /* AAD (extra authenticate-only data) / masked 802.11 header
 339          * FC | A1 | A2 | A3 | SC | [A4] | [QC] */
 340
 341         len_a = a4_included ? 28 : 22;
 342         if (qos_included)
 343                 len_a += 2;
 344
 345         aad[0] = 0; /* (len_a >> 8) & 0xff; */
 346         aad[1] = len_a & 0xff;
 347         /* Mask FC: zero subtype b4 b5 b6 */
 348         aad[2] = fc_pos[0] & ~(BIT(4) | BIT(5) | BIT(6));
 349         /* Retry, PwrMgt, MoreData; set Protected */
 350         aad[3] = (fc_pos[1] & ~(BIT(3) | BIT(4) | BIT(5))) | BIT(6);
 351         memcpy(&aad[4], &hdr->addr1, 18);
 352
 353         /* Mask Seq#, leave Frag# */
 354         aad[22] = *((u8 *) &hdr->seq_ctrl) & 0x0f;
 355         aad[23] = 0;
 356         if (a4_included) {
 357                 memcpy(&aad[24], hdr->addr4, 6);
 358                 aad[30] = 0;
 359                 aad[31] = 0;
 360         } else
 361                 memset(&aad[24], 0, 8);
 362         if (qos_included) {
 363                 u8 *dpos = &aad[a4_included ? 30 : 24];
 364
 365                 /* Mask QoS Control field */
 366                 dpos[0] = qos_tid;
 367                 dpos[1] = 0;
 368         }
 369 }
 370
 371
 372 static inline void ccmp_pn2hdr(u8 *hdr, u8 *pn, int key_id)
 373 {
 374         hdr[0] = pn[5];
 375         hdr[1] = pn[4];
 376         hdr[2] = 0;
 377         hdr[3] = 0x20 | (key_id << 6);
 378         hdr[4] = pn[3];
 379         hdr[5] = pn[2];
 380         hdr[6] = pn[1];
 381         hdr[7] = pn[0];
 382 }
 383
 384
 385 static inline int ccmp_hdr2pn(u8 *pn, u8 *hdr)
 386 {
 387         pn[0] = hdr[7];
 388         pn[1] = hdr[6];
 389         pn[2] = hdr[5];
 390         pn[3] = hdr[4];
 391         pn[4] = hdr[1];
 392         pn[5] = hdr[0];
 393         return (hdr[3] >> 6) & 0x03;
 394 }
 395
 396
 397 static int ccmp_encrypt_skb(struct ieee80211_tx_data *tx, struct sk_buff *skb)
 398 {
 399         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
 400         struct ieee80211_key *key = tx->key;
 401         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
 402         int hdrlen, len, tail;
 403         u8 *pos, *pn, *b_0, *aad, *scratch;
 404         int i;
 405
 406         info->control.icv_len = CCMP_MIC_LEN;
 407         info->control.iv_len = CCMP_HDR_LEN;
 408
 409         if ((tx->key->flags & KEY_FLAG_UPLOADED_TO_HARDWARE) &&
 410             !(tx->key->conf.flags & IEEE80211_KEY_FLAG_GENERATE_IV)) {
 411                 /* hwaccel - with no need for preallocated room for CCMP "
 412                  * header or MIC fields */
 413                 info->control.hw_key = &tx->key->conf;
 414                 return 0;
 415         }
 416
 417         scratch = key->u.ccmp.tx_crypto_buf;
 418         b_0 = scratch + 3 * AES_BLOCK_LEN;
 419         aad = scratch + 4 * AES_BLOCK_LEN;
 420
 421         hdrlen = ieee80211_hdrlen(hdr->frame_control);
 422         len = skb->len - hdrlen;
 423
 424         if (key->flags & KEY_FLAG_UPLOADED_TO_HARDWARE)
 425                 tail = 0;
 426         else
 427                 tail = CCMP_MIC_LEN;
 428
 429         if (WARN_ON(skb_tailroom(skb) < tail ||
 430                     skb_headroom(skb) < CCMP_HDR_LEN))
 431                 return -1;
 432
 433         pos = skb_push(skb, CCMP_HDR_LEN);
 434         memmove(pos, pos + CCMP_HDR_LEN, hdrlen);
 435         hdr = (struct ieee80211_hdr *) pos;
 436         pos += hdrlen;
 437
 438         /* PN = PN + 1 */
 439         pn = key->u.ccmp.tx_pn;
 440
 441         for (i = CCMP_PN_LEN - 1; i >= 0; i--) {
 442                 pn[i]++;
 443                 if (pn[i])
 444                         break;
 445         }
 446
 447         ccmp_pn2hdr(pos, pn, key->conf.keyidx);
 448
 449         if (key->flags & KEY_FLAG_UPLOADED_TO_HARDWARE) {
 450                 /* hwaccel - with preallocated room for CCMP header */
 451                 info->control.hw_key = &tx->key->conf;
 452                 return 0;
 453         }
 454
 455         pos += CCMP_HDR_LEN;
 456         ccmp_special_blocks(skb, pn, b_0, aad, 0);
 457         ieee80211_aes_ccm_encrypt(key->u.ccmp.tfm, scratch, b_0, aad, pos, len,
 458                                   pos, skb_put(skb, CCMP_MIC_LEN));
 459
 460         return 0;
 461 }
 462
 463
 464 ieee80211_tx_result
 465 ieee80211_crypto_ccmp_encrypt(struct ieee80211_tx_data *tx)
 466 {
 467         struct sk_buff *skb = tx->skb;
 468
 469         ieee80211_tx_set_protected(tx);
 470
 471         if (ccmp_encrypt_skb(tx, skb) < 0)
 472                 return TX_DROP;
 473
 474         if (tx->extra_frag) {
 475                 int i;
 476                 for (i = 0; i < tx->num_extra_frag; i++) {
 477                         if (ccmp_encrypt_skb(tx, tx->extra_frag[i]) < 0)
 478                                 return TX_DROP;
 479                 }
 480         }
 481
 482         return TX_CONTINUE;
 483 }
 484
 485
 486 ieee80211_rx_result
 487 ieee80211_crypto_ccmp_decrypt(struct ieee80211_rx_data *rx)
 488 {
 489         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) rx->skb->data;
 490         int hdrlen;
 491         struct ieee80211_key *key = rx->key;
 492         struct sk_buff *skb = rx->skb;
 493         u8 pn[CCMP_PN_LEN];
 494         int data_len;
 495         DECLARE_MAC_BUF(mac);
 496
 497         hdrlen = ieee80211_hdrlen(hdr->frame_control);
 498
 499         if ((rx->fc & IEEE80211_FCTL_FTYPE) != IEEE80211_FTYPE_DATA)
 500                 return RX_CONTINUE;
 501
 502         data_len = skb->len - hdrlen - CCMP_HDR_LEN - CCMP_MIC_LEN;
 503         if (!rx->sta || data_len < 0)
 504                 return RX_DROP_UNUSABLE;
 505
 506         if ((rx->status->flag & RX_FLAG_DECRYPTED) &&
 507             (rx->status->flag & RX_FLAG_IV_STRIPPED))
 508                 return RX_CONTINUE;
 509
 510         (void) ccmp_hdr2pn(pn, skb->data + hdrlen);
 511
 512         if (memcmp(pn, key->u.ccmp.rx_pn[rx->queue], CCMP_PN_LEN) <= 0) {
 513 #ifdef CONFIG_MAC80211_DEBUG
 514                 u8 *ppn = key->u.ccmp.rx_pn[rx->queue];
 515
 516                 printk(KERN_DEBUG "%s: CCMP replay detected for RX frame from "
 517                        "%s (RX PN %02x%02x%02x%02x%02x%02x <= prev. PN "
 518                        "%02x%02x%02x%02x%02x%02x)\n", rx->dev->name,
 519                        print_mac(mac, rx->sta->addr),
 520                        pn[0], pn[1], pn[2], pn[3], pn[4], pn[5],
 521                        ppn[0], ppn[1], ppn[2], ppn[3], ppn[4], ppn[5]);
 522 #endif /* CONFIG_MAC80211_DEBUG */
 523                 key->u.ccmp.replays++;
 524                 return RX_DROP_UNUSABLE;
 525         }
 526
 527         if (!(rx->status->flag & RX_FLAG_DECRYPTED)) {
 528                 /* hardware didn't decrypt/verify MIC */
 529                 u8 *scratch, *b_0, *aad;
 530
 531                 scratch = key->u.ccmp.rx_crypto_buf;
 532                 b_0 = scratch + 3 * AES_BLOCK_LEN;
 533                 aad = scratch + 4 * AES_BLOCK_LEN;
 534
 535                 ccmp_special_blocks(skb, pn, b_0, aad, 1);
 536
 537                 if (ieee80211_aes_ccm_decrypt(
 538                             key->u.ccmp.tfm, scratch, b_0, aad,
 539                             skb->data + hdrlen + CCMP_HDR_LEN, data_len,
 540                             skb->data + skb->len - CCMP_MIC_LEN,
 541                             skb->data + hdrlen + CCMP_HDR_LEN)) {
 542 #ifdef CONFIG_MAC80211_DEBUG
 543                         if (net_ratelimit())
 544                                 printk(KERN_DEBUG "%s: CCMP decrypt failed "
 545                                        "for RX frame from %s\n", rx->dev->name,
 546                                        print_mac(mac, rx->sta->addr));
 547 #endif /* CONFIG_MAC80211_DEBUG */
 548                         return RX_DROP_UNUSABLE;
 549                 }
 550         }
 551
 552         memcpy(key->u.ccmp.rx_pn[rx->queue], pn, CCMP_PN_LEN);
 553
 554         /* Remove CCMP header and MIC */
 555         skb_trim(skb, skb->len - CCMP_MIC_LEN);
 556         memmove(skb->data + CCMP_HDR_LEN, skb->data, hdrlen);
 557         skb_pull(skb, CCMP_HDR_LEN);
 558
 559         return RX_CONTINUE;
 560 }