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

   1 /*
   2  * hostapd - PMKSA cache for IEEE 802.11i RSN
   3  * Copyright (c) 2004-2008, 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 "utils/eloop.h"
  19 #include "eapol_auth/eapol_auth_sm.h"
  20 #include "eapol_auth/eapol_auth_sm_i.h"
  21 #include "sta_info.h"
  22 #include "ap_config.h"
  23 #include "pmksa_cache_auth.h"
  24
  25
  26 static const int pmksa_cache_max_entries = 1024;
  27 static const int dot11RSNAConfigPMKLifetime = 43200;
  28
  29 struct rsn_pmksa_cache {
  30 #define PMKID_HASH_SIZE 128
  31 #define PMKID_HASH(pmkid) (unsigned int) ((pmkid)[0] & 0x7f)
  32         struct rsn_pmksa_cache_entry *pmkid[PMKID_HASH_SIZE];
  33         struct rsn_pmksa_cache_entry *pmksa;
  34         int pmksa_count;
  35
  36         void (*free_cb)(struct rsn_pmksa_cache_entry *entry, void *ctx);
  37         void *ctx;
  38 };
  39
  40
  41 static void pmksa_cache_set_expiration(struct rsn_pmksa_cache *pmksa);
  42
  43
  44 static void _pmksa_cache_free_entry(struct rsn_pmksa_cache_entry *entry)
  45 {
  46         if (entry == NULL)
  47                 return;
  48         os_free(entry->identity);
  49 #ifndef CONFIG_NO_RADIUS
  50         radius_free_class(&entry->radius_class);
  51 #endif /* CONFIG_NO_RADIUS */
  52         os_free(entry);
  53 }
  54
  55
  56 static void pmksa_cache_free_entry(struct rsn_pmksa_cache *pmksa,
  57                                    struct rsn_pmksa_cache_entry *entry)
  58 {
  59         struct rsn_pmksa_cache_entry *pos, *prev;
  60
  61         pmksa->pmksa_count--;
  62         pmksa->free_cb(entry, pmksa->ctx);
  63         pos = pmksa->pmkid[PMKID_HASH(entry->pmkid)];
  64         prev = NULL;
  65         while (pos) {
  66                 if (pos == entry) {
  67                         if (prev != NULL) {
  68                                 prev->hnext = pos->hnext;
  69                         } else {
  70                                 pmksa->pmkid[PMKID_HASH(entry->pmkid)] =
  71                                         pos->hnext;
  72                         }
  73                         break;
  74                 }
  75                 prev = pos;
  76                 pos = pos->hnext;
  77         }
  78
  79         pos = pmksa->pmksa;
  80         prev = NULL;
  81         while (pos) {
  82                 if (pos == entry) {
  83                         if (prev != NULL)
  84                                 prev->next = pos->next;
  85                         else
  86                                 pmksa->pmksa = pos->next;
  87                         break;
  88                 }
  89                 prev = pos;
  90                 pos = pos->next;
  91         }
  92         _pmksa_cache_free_entry(entry);
  93 }
  94
  95
  96 static void pmksa_cache_expire(void *eloop_ctx, void *timeout_ctx)
  97 {
  98         struct rsn_pmksa_cache *pmksa = eloop_ctx;
  99         struct os_time now;
 100
 101         os_get_time(&now);
 102         while (pmksa->pmksa && pmksa->pmksa->expiration <= now.sec) {
 103                 struct rsn_pmksa_cache_entry *entry = pmksa->pmksa;
 104                 pmksa->pmksa = entry->next;
 105                 wpa_printf(MSG_DEBUG, "RSN: expired PMKSA cache entry for "
 106                            MACSTR, MAC2STR(entry->spa));
 107                 pmksa_cache_free_entry(pmksa, entry);
 108         }
 109
 110         pmksa_cache_set_expiration(pmksa);
 111 }
 112
 113
 114 static void pmksa_cache_set_expiration(struct rsn_pmksa_cache *pmksa)
 115 {
 116         int sec;
 117         struct os_time now;
 118
 119         eloop_cancel_timeout(pmksa_cache_expire, pmksa, NULL);
 120         if (pmksa->pmksa == NULL)
 121                 return;
 122         os_get_time(&now);
 123         sec = pmksa->pmksa->expiration - now.sec;
 124         if (sec < 0)
 125                 sec = 0;
 126         eloop_register_timeout(sec + 1, 0, pmksa_cache_expire, pmksa, NULL);
 127 }
 128
 129
 130 static void pmksa_cache_from_eapol_data(struct rsn_pmksa_cache_entry *entry,
 131                                         struct eapol_state_machine *eapol)
 132 {
 133         if (eapol == NULL)
 134                 return;
 135
 136         if (eapol->identity) {
 137                 entry->identity = os_malloc(eapol->identity_len);
 138                 if (entry->identity) {
 139                         entry->identity_len = eapol->identity_len;
 140                         os_memcpy(entry->identity, eapol->identity,
 141                                   eapol->identity_len);
 142                 }
 143         }
 144
 145 #ifndef CONFIG_NO_RADIUS
 146         radius_copy_class(&entry->radius_class, &eapol->radius_class);
 147 #endif /* CONFIG_NO_RADIUS */
 148
 149         entry->eap_type_authsrv = eapol->eap_type_authsrv;
 150         entry->vlan_id = ((struct sta_info *) eapol->sta)->vlan_id;
 151 }
 152
 153
 154 void pmksa_cache_to_eapol_data(struct rsn_pmksa_cache_entry *entry,
 155                                struct eapol_state_machine *eapol)
 156 {
 157         if (entry == NULL || eapol == NULL)
 158                 return;
 159
 160         if (entry->identity) {
 161                 os_free(eapol->identity);
 162                 eapol->identity = os_malloc(entry->identity_len);
 163                 if (eapol->identity) {
 164                         eapol->identity_len = entry->identity_len;
 165                         os_memcpy(eapol->identity, entry->identity,
 166                                   entry->identity_len);
 167                 }
 168                 wpa_hexdump_ascii(MSG_DEBUG, "STA identity from PMKSA",
 169                                   eapol->identity, eapol->identity_len);
 170         }
 171
 172 #ifndef CONFIG_NO_RADIUS
 173         radius_free_class(&eapol->radius_class);
 174         radius_copy_class(&eapol->radius_class, &entry->radius_class);
 175 #endif /* CONFIG_NO_RADIUS */
 176         if (eapol->radius_class.attr) {
 177                 wpa_printf(MSG_DEBUG, "Copied %lu Class attribute(s) from "
 178                            "PMKSA", (unsigned long) eapol->radius_class.count);
 179         }
 180
 181         eapol->eap_type_authsrv = entry->eap_type_authsrv;
 182         ((struct sta_info *) eapol->sta)->vlan_id = entry->vlan_id;
 183 }
 184
 185
 186 static void pmksa_cache_link_entry(struct rsn_pmksa_cache *pmksa,
 187                                    struct rsn_pmksa_cache_entry *entry)
 188 {
 189         struct rsn_pmksa_cache_entry *pos, *prev;
 190
 191         /* Add the new entry; order by expiration time */
 192         pos = pmksa->pmksa;
 193         prev = NULL;
 194         while (pos) {
 195                 if (pos->expiration > entry->expiration)
 196                         break;
 197                 prev = pos;
 198                 pos = pos->next;
 199         }
 200         if (prev == NULL) {
 201                 entry->next = pmksa->pmksa;
 202                 pmksa->pmksa = entry;
 203         } else {
 204                 entry->next = prev->next;
 205                 prev->next = entry;
 206         }
 207         entry->hnext = pmksa->pmkid[PMKID_HASH(entry->pmkid)];
 208         pmksa->pmkid[PMKID_HASH(entry->pmkid)] = entry;
 209
 210         pmksa->pmksa_count++;
 211         wpa_printf(MSG_DEBUG, "RSN: added PMKSA cache entry for " MACSTR,
 212                    MAC2STR(entry->spa));
 213         wpa_hexdump(MSG_DEBUG, "RSN: added PMKID", entry->pmkid, PMKID_LEN);
 214 }
 215
 216
 217 /**
 218  * pmksa_cache_auth_add - Add a PMKSA cache entry
 219  * @pmksa: Pointer to PMKSA cache data from pmksa_cache_auth_init()
 220  * @pmk: The new pairwise master key
 221  * @pmk_len: PMK length in bytes, usually PMK_LEN (32)
 222  * @aa: Authenticator address
 223  * @spa: Supplicant address
 224  * @session_timeout: Session timeout
 225  * @eapol: Pointer to EAPOL state machine data
 226  * @akmp: WPA_KEY_MGMT_* used in key derivation
 227  * Returns: Pointer to the added PMKSA cache entry or %NULL on error
 228  *
 229  * This function create a PMKSA entry for a new PMK and adds it to the PMKSA
 230  * cache. If an old entry is already in the cache for the same Supplicant,
 231  * this entry will be replaced with the new entry. PMKID will be calculated
 232  * based on the PMK.
 233  */
 234 struct rsn_pmksa_cache_entry *
 235 pmksa_cache_auth_add(struct rsn_pmksa_cache *pmksa,
 236                      const u8 *pmk, size_t pmk_len,
 237                 const u8 *aa, const u8 *spa, int session_timeout,
 238                 struct eapol_state_machine *eapol, int akmp)
 239 {
 240         struct rsn_pmksa_cache_entry *entry, *pos;
 241         struct os_time now;
 242
 243         if (pmk_len > PMK_LEN)
 244                 return NULL;
 245
 246         entry = os_zalloc(sizeof(*entry));
 247         if (entry == NULL)
 248                 return NULL;
 249         os_memcpy(entry->pmk, pmk, pmk_len);
 250         entry->pmk_len = pmk_len;
 251         rsn_pmkid(pmk, pmk_len, aa, spa, entry->pmkid,
 252                   wpa_key_mgmt_sha256(akmp));
 253         os_get_time(&now);
 254         entry->expiration = now.sec;
 255         if (session_timeout > 0)
 256                 entry->expiration += session_timeout;
 257         else
 258                 entry->expiration += dot11RSNAConfigPMKLifetime;
 259         entry->akmp = akmp;
 260         os_memcpy(entry->spa, spa, ETH_ALEN);
 261         pmksa_cache_from_eapol_data(entry, eapol);
 262
 263         /* Replace an old entry for the same STA (if found) with the new entry
 264          */
 265         pos = pmksa_cache_auth_get(pmksa, spa, NULL);
 266         if (pos)
 267                 pmksa_cache_free_entry(pmksa, pos);
 268
 269         if (pmksa->pmksa_count >= pmksa_cache_max_entries && pmksa->pmksa) {
 270                 /* Remove the oldest entry to make room for the new entry */
 271                 wpa_printf(MSG_DEBUG, "RSN: removed the oldest PMKSA cache "
 272                            "entry (for " MACSTR ") to make room for new one",
 273                            MAC2STR(pmksa->pmksa->spa));
 274                 pmksa_cache_free_entry(pmksa, pmksa->pmksa);
 275         }
 276
 277         pmksa_cache_link_entry(pmksa, entry);
 278
 279         return entry;
 280 }
 281
 282
 283 struct rsn_pmksa_cache_entry *
 284 pmksa_cache_add_okc(struct rsn_pmksa_cache *pmksa,
 285                     const struct rsn_pmksa_cache_entry *old_entry,
 286                     const u8 *aa, const u8 *pmkid)
 287 {
 288         struct rsn_pmksa_cache_entry *entry;
 289
 290         entry = os_zalloc(sizeof(*entry));
 291         if (entry == NULL)
 292                 return NULL;
 293         os_memcpy(entry->pmkid, pmkid, PMKID_LEN);
 294         os_memcpy(entry->pmk, old_entry->pmk, old_entry->pmk_len);
 295         entry->pmk_len = old_entry->pmk_len;
 296         entry->expiration = old_entry->expiration;
 297         entry->akmp = old_entry->akmp;
 298         os_memcpy(entry->spa, old_entry->spa, ETH_ALEN);
 299         entry->opportunistic = 1;
 300         if (old_entry->identity) {
 301                 entry->identity = os_malloc(old_entry->identity_len);
 302                 if (entry->identity) {
 303                         entry->identity_len = old_entry->identity_len;
 304                         os_memcpy(entry->identity, old_entry->identity,
 305                                   old_entry->identity_len);
 306                 }
 307         }
 308 #ifndef CONFIG_NO_RADIUS
 309         radius_copy_class(&entry->radius_class, &old_entry->radius_class);
 310 #endif /* CONFIG_NO_RADIUS */
 311         entry->eap_type_authsrv = old_entry->eap_type_authsrv;
 312         entry->vlan_id = old_entry->vlan_id;
 313         entry->opportunistic = 1;
 314
 315         pmksa_cache_link_entry(pmksa, entry);
 316
 317         return entry;
 318 }
 319
 320
 321 /**
 322  * pmksa_cache_auth_deinit - Free all entries in PMKSA cache
 323  * @pmksa: Pointer to PMKSA cache data from pmksa_cache_auth_init()
 324  */
 325 void pmksa_cache_auth_deinit(struct rsn_pmksa_cache *pmksa)
 326 {
 327         struct rsn_pmksa_cache_entry *entry, *prev;
 328         int i;
 329
 330         if (pmksa == NULL)
 331                 return;
 332
 333         entry = pmksa->pmksa;
 334         while (entry) {
 335                 prev = entry;
 336                 entry = entry->next;
 337                 _pmksa_cache_free_entry(prev);
 338         }
 339         eloop_cancel_timeout(pmksa_cache_expire, pmksa, NULL);
 340         for (i = 0; i < PMKID_HASH_SIZE; i++)
 341                 pmksa->pmkid[i] = NULL;
 342         os_free(pmksa);
 343 }
 344
 345
 346 /**
 347  * pmksa_cache_auth_get - Fetch a PMKSA cache entry
 348  * @pmksa: Pointer to PMKSA cache data from pmksa_cache_auth_init()
 349  * @spa: Supplicant address or %NULL to match any
 350  * @pmkid: PMKID or %NULL to match any
 351  * Returns: Pointer to PMKSA cache entry or %NULL if no match was found
 352  */
 353 struct rsn_pmksa_cache_entry *
 354 pmksa_cache_auth_get(struct rsn_pmksa_cache *pmksa,
 355                      const u8 *spa, const u8 *pmkid)
 356 {
 357         struct rsn_pmksa_cache_entry *entry;
 358
 359         if (pmkid)
 360                 entry = pmksa->pmkid[PMKID_HASH(pmkid)];
 361         else
 362                 entry = pmksa->pmksa;
 363         while (entry) {
 364                 if ((spa == NULL ||
 365                      os_memcmp(entry->spa, spa, ETH_ALEN) == 0) &&
 366                     (pmkid == NULL ||
 367                      os_memcmp(entry->pmkid, pmkid, PMKID_LEN) == 0))
 368                         return entry;
 369                 entry = pmkid ? entry->hnext : entry->next;
 370         }
 371         return NULL;
 372 }
 373
 374
 375 /**
 376  * pmksa_cache_get_okc - Fetch a PMKSA cache entry using OKC
 377  * @pmksa: Pointer to PMKSA cache data from pmksa_cache_auth_init()
 378  * @aa: Authenticator address
 379  * @spa: Supplicant address
 380  * @pmkid: PMKID
 381  * Returns: Pointer to PMKSA cache entry or %NULL if no match was found
 382  *
 383  * Use opportunistic key caching (OKC) to find a PMK for a supplicant.
 384  */
 385 struct rsn_pmksa_cache_entry * pmksa_cache_get_okc(
 386         struct rsn_pmksa_cache *pmksa, const u8 *aa, const u8 *spa,
 387         const u8 *pmkid)
 388 {
 389         struct rsn_pmksa_cache_entry *entry;
 390         u8 new_pmkid[PMKID_LEN];
 391
 392         entry = pmksa->pmksa;
 393         while (entry) {
 394                 if (os_memcmp(entry->spa, spa, ETH_ALEN) != 0)
 395                         continue;
 396                 rsn_pmkid(entry->pmk, entry->pmk_len, aa, spa, new_pmkid,
 397                           wpa_key_mgmt_sha256(entry->akmp));
 398                 if (os_memcmp(new_pmkid, pmkid, PMKID_LEN) == 0)
 399                         return entry;
 400                 entry = entry->next;
 401         }
 402         return NULL;
 403 }
 404
 405
 406 /**
 407  * pmksa_cache_auth_init - Initialize PMKSA cache
 408  * @free_cb: Callback function to be called when a PMKSA cache entry is freed
 409  * @ctx: Context pointer for free_cb function
 410  * Returns: Pointer to PMKSA cache data or %NULL on failure
 411  */
 412 struct rsn_pmksa_cache *
 413 pmksa_cache_auth_init(void (*free_cb)(struct rsn_pmksa_cache_entry *entry,
 414                                       void *ctx), void *ctx)
 415 {
 416         struct rsn_pmksa_cache *pmksa;
 417
 418         pmksa = os_zalloc(sizeof(*pmksa));
 419         if (pmksa) {
 420                 pmksa->free_cb = free_cb;
 421                 pmksa->ctx = ctx;
 422         }
 423
 424         return pmksa;
 425 }