Copy Chargeable-User-Identity in PMKSA caching case
[mech_eap.git] / src / ap / pmksa_cache_auth.c
1 /*
2  * hostapd - PMKSA cache for IEEE 802.11i RSN
3  * Copyright (c) 2004-2008, 2012, Jouni Malinen <j@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 }