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