Merge branch 'vendor/XZ'
[dragonfly.git] / contrib / hostapd / src / eap_peer / eap_psk.c
1 /*
2  * EAP peer method: EAP-PSK (RFC 4764)
3  * Copyright (c) 2004-2008, 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  * Note: EAP-PSK is an EAP authentication method and as such, completely
9  * different from WPA-PSK. This file is not needed for WPA-PSK functionality.
10  */
11
12 #include "includes.h"
13
14 #include "common.h"
15 #include "crypto/aes_wrap.h"
16 #include "crypto/random.h"
17 #include "eap_common/eap_psk_common.h"
18 #include "eap_i.h"
19
20
21 struct eap_psk_data {
22         enum { PSK_INIT, PSK_MAC_SENT, PSK_DONE } state;
23         u8 rand_p[EAP_PSK_RAND_LEN];
24         u8 rand_s[EAP_PSK_RAND_LEN];
25         u8 ak[EAP_PSK_AK_LEN], kdk[EAP_PSK_KDK_LEN], tek[EAP_PSK_TEK_LEN];
26         u8 *id_s, *id_p;
27         size_t id_s_len, id_p_len;
28         u8 msk[EAP_MSK_LEN];
29         u8 emsk[EAP_EMSK_LEN];
30 };
31
32
33 static void * eap_psk_init(struct eap_sm *sm)
34 {
35         struct eap_psk_data *data;
36         const u8 *identity, *password;
37         size_t identity_len, password_len;
38
39         password = eap_get_config_password(sm, &password_len);
40         if (!password || password_len != 16) {
41                 wpa_printf(MSG_INFO, "EAP-PSK: 16-octet pre-shared key not "
42                            "configured");
43                 return NULL;
44         }
45
46         data = os_zalloc(sizeof(*data));
47         if (data == NULL)
48                 return NULL;
49         if (eap_psk_key_setup(password, data->ak, data->kdk)) {
50                 os_free(data);
51                 return NULL;
52         }
53         wpa_hexdump_key(MSG_DEBUG, "EAP-PSK: AK", data->ak, EAP_PSK_AK_LEN);
54         wpa_hexdump_key(MSG_DEBUG, "EAP-PSK: KDK", data->kdk, EAP_PSK_KDK_LEN);
55         data->state = PSK_INIT;
56
57         identity = eap_get_config_identity(sm, &identity_len);
58         if (identity) {
59                 data->id_p = os_malloc(identity_len);
60                 if (data->id_p)
61                         os_memcpy(data->id_p, identity, identity_len);
62                 data->id_p_len = identity_len;
63         }
64         if (data->id_p == NULL) {
65                 wpa_printf(MSG_INFO, "EAP-PSK: could not get own identity");
66                 os_free(data);
67                 return NULL;
68         }
69
70         return data;
71 }
72
73
74 static void eap_psk_deinit(struct eap_sm *sm, void *priv)
75 {
76         struct eap_psk_data *data = priv;
77         os_free(data->id_s);
78         os_free(data->id_p);
79         os_free(data);
80 }
81
82
83 static struct wpabuf * eap_psk_process_1(struct eap_psk_data *data,
84                                          struct eap_method_ret *ret,
85                                          const struct wpabuf *reqData)
86 {
87         const struct eap_psk_hdr_1 *hdr1;
88         struct eap_psk_hdr_2 *hdr2;
89         struct wpabuf *resp;
90         u8 *buf, *pos;
91         size_t buflen, len;
92         const u8 *cpos;
93
94         wpa_printf(MSG_DEBUG, "EAP-PSK: in INIT state");
95
96         cpos = eap_hdr_validate(EAP_VENDOR_IETF, EAP_TYPE_PSK, reqData, &len);
97         hdr1 = (const struct eap_psk_hdr_1 *) cpos;
98         if (cpos == NULL || len < sizeof(*hdr1)) {
99                 wpa_printf(MSG_INFO, "EAP-PSK: Invalid first message "
100                            "length (%lu; expected %lu or more)",
101                            (unsigned long) len,
102                            (unsigned long) sizeof(*hdr1));
103                 ret->ignore = TRUE;
104                 return NULL;
105         }
106         wpa_printf(MSG_DEBUG, "EAP-PSK: Flags=0x%x", hdr1->flags);
107         if (EAP_PSK_FLAGS_GET_T(hdr1->flags) != 0) {
108                 wpa_printf(MSG_INFO, "EAP-PSK: Unexpected T=%d (expected 0)",
109                            EAP_PSK_FLAGS_GET_T(hdr1->flags));
110                 ret->methodState = METHOD_DONE;
111                 ret->decision = DECISION_FAIL;
112                 return NULL;
113         }
114         wpa_hexdump(MSG_DEBUG, "EAP-PSK: RAND_S", hdr1->rand_s,
115                     EAP_PSK_RAND_LEN);
116         os_memcpy(data->rand_s, hdr1->rand_s, EAP_PSK_RAND_LEN);
117         os_free(data->id_s);
118         data->id_s_len = len - sizeof(*hdr1);
119         data->id_s = os_malloc(data->id_s_len);
120         if (data->id_s == NULL) {
121                 wpa_printf(MSG_ERROR, "EAP-PSK: Failed to allocate memory for "
122                            "ID_S (len=%lu)", (unsigned long) data->id_s_len);
123                 ret->ignore = TRUE;
124                 return NULL;
125         }
126         os_memcpy(data->id_s, (u8 *) (hdr1 + 1), data->id_s_len);
127         wpa_hexdump_ascii(MSG_DEBUG, "EAP-PSK: ID_S",
128                           data->id_s, data->id_s_len);
129
130         if (random_get_bytes(data->rand_p, EAP_PSK_RAND_LEN)) {
131                 wpa_printf(MSG_ERROR, "EAP-PSK: Failed to get random data");
132                 ret->ignore = TRUE;
133                 return NULL;
134         }
135
136         resp = eap_msg_alloc(EAP_VENDOR_IETF, EAP_TYPE_PSK,
137                              sizeof(*hdr2) + data->id_p_len, EAP_CODE_RESPONSE,
138                              eap_get_id(reqData));
139         if (resp == NULL)
140                 return NULL;
141         hdr2 = wpabuf_put(resp, sizeof(*hdr2));
142         hdr2->flags = EAP_PSK_FLAGS_SET_T(1); /* T=1 */
143         os_memcpy(hdr2->rand_s, hdr1->rand_s, EAP_PSK_RAND_LEN);
144         os_memcpy(hdr2->rand_p, data->rand_p, EAP_PSK_RAND_LEN);
145         wpabuf_put_data(resp, data->id_p, data->id_p_len);
146         /* MAC_P = OMAC1-AES-128(AK, ID_P||ID_S||RAND_S||RAND_P) */
147         buflen = data->id_p_len + data->id_s_len + 2 * EAP_PSK_RAND_LEN;
148         buf = os_malloc(buflen);
149         if (buf == NULL) {
150                 wpabuf_free(resp);
151                 return NULL;
152         }
153         os_memcpy(buf, data->id_p, data->id_p_len);
154         pos = buf + data->id_p_len;
155         os_memcpy(pos, data->id_s, data->id_s_len);
156         pos += data->id_s_len;
157         os_memcpy(pos, hdr1->rand_s, EAP_PSK_RAND_LEN);
158         pos += EAP_PSK_RAND_LEN;
159         os_memcpy(pos, data->rand_p, EAP_PSK_RAND_LEN);
160         if (omac1_aes_128(data->ak, buf, buflen, hdr2->mac_p)) {
161                 os_free(buf);
162                 wpabuf_free(resp);
163                 return NULL;
164         }
165         os_free(buf);
166         wpa_hexdump(MSG_DEBUG, "EAP-PSK: RAND_P", hdr2->rand_p,
167                     EAP_PSK_RAND_LEN);
168         wpa_hexdump(MSG_DEBUG, "EAP-PSK: MAC_P", hdr2->mac_p, EAP_PSK_MAC_LEN);
169         wpa_hexdump_ascii(MSG_DEBUG, "EAP-PSK: ID_P",
170                           data->id_p, data->id_p_len);
171
172         data->state = PSK_MAC_SENT;
173
174         return resp;
175 }
176
177
178 static struct wpabuf * eap_psk_process_3(struct eap_psk_data *data,
179                                          struct eap_method_ret *ret,
180                                          const struct wpabuf *reqData)
181 {
182         const struct eap_psk_hdr_3 *hdr3;
183         struct eap_psk_hdr_4 *hdr4;
184         struct wpabuf *resp;
185         u8 *buf, *rpchannel, nonce[16], *decrypted;
186         const u8 *pchannel, *tag, *msg;
187         u8 mac[EAP_PSK_MAC_LEN];
188         size_t buflen, left, data_len, len, plen;
189         int failed = 0;
190         const u8 *pos;
191
192         wpa_printf(MSG_DEBUG, "EAP-PSK: in MAC_SENT state");
193
194         pos = eap_hdr_validate(EAP_VENDOR_IETF, EAP_TYPE_PSK,
195                                reqData, &len);
196         hdr3 = (const struct eap_psk_hdr_3 *) pos;
197         if (pos == NULL || len < sizeof(*hdr3)) {
198                 wpa_printf(MSG_INFO, "EAP-PSK: Invalid third message "
199                            "length (%lu; expected %lu or more)",
200                            (unsigned long) len,
201                            (unsigned long) sizeof(*hdr3));
202                 ret->ignore = TRUE;
203                 return NULL;
204         }
205         left = len - sizeof(*hdr3);
206         pchannel = (const u8 *) (hdr3 + 1);
207         wpa_printf(MSG_DEBUG, "EAP-PSK: Flags=0x%x", hdr3->flags);
208         if (EAP_PSK_FLAGS_GET_T(hdr3->flags) != 2) {
209                 wpa_printf(MSG_INFO, "EAP-PSK: Unexpected T=%d (expected 2)",
210                            EAP_PSK_FLAGS_GET_T(hdr3->flags));
211                 ret->methodState = METHOD_DONE;
212                 ret->decision = DECISION_FAIL;
213                 return NULL;
214         }
215         wpa_hexdump(MSG_DEBUG, "EAP-PSK: RAND_S", hdr3->rand_s,
216                     EAP_PSK_RAND_LEN);
217         wpa_hexdump(MSG_DEBUG, "EAP-PSK: MAC_S", hdr3->mac_s, EAP_PSK_MAC_LEN);
218         wpa_hexdump(MSG_DEBUG, "EAP-PSK: PCHANNEL", pchannel, left);
219
220         if (left < 4 + 16 + 1) {
221                 wpa_printf(MSG_INFO, "EAP-PSK: Too short PCHANNEL data in "
222                            "third message (len=%lu, expected 21)",
223                            (unsigned long) left);
224                 ret->ignore = TRUE;
225                 return NULL;
226         }
227
228         /* MAC_S = OMAC1-AES-128(AK, ID_S||RAND_P) */
229         buflen = data->id_s_len + EAP_PSK_RAND_LEN;
230         buf = os_malloc(buflen);
231         if (buf == NULL)
232                 return NULL;
233         os_memcpy(buf, data->id_s, data->id_s_len);
234         os_memcpy(buf + data->id_s_len, data->rand_p, EAP_PSK_RAND_LEN);
235         if (omac1_aes_128(data->ak, buf, buflen, mac)) {
236                 os_free(buf);
237                 return NULL;
238         }
239         os_free(buf);
240         if (os_memcmp(mac, hdr3->mac_s, EAP_PSK_MAC_LEN) != 0) {
241                 wpa_printf(MSG_WARNING, "EAP-PSK: Invalid MAC_S in third "
242                            "message");
243                 ret->methodState = METHOD_DONE;
244                 ret->decision = DECISION_FAIL;
245                 return NULL;
246         }
247         wpa_printf(MSG_DEBUG, "EAP-PSK: MAC_S verified successfully");
248
249         if (eap_psk_derive_keys(data->kdk, data->rand_p, data->tek,
250                                 data->msk, data->emsk)) {
251                 ret->methodState = METHOD_DONE;
252                 ret->decision = DECISION_FAIL;
253                 return NULL;
254         }
255         wpa_hexdump_key(MSG_DEBUG, "EAP-PSK: TEK", data->tek, EAP_PSK_TEK_LEN);
256         wpa_hexdump_key(MSG_DEBUG, "EAP-PSK: MSK", data->msk, EAP_MSK_LEN);
257         wpa_hexdump_key(MSG_DEBUG, "EAP-PSK: EMSK", data->emsk, EAP_EMSK_LEN);
258
259         os_memset(nonce, 0, 12);
260         os_memcpy(nonce + 12, pchannel, 4);
261         pchannel += 4;
262         left -= 4;
263
264         tag = pchannel;
265         pchannel += 16;
266         left -= 16;
267
268         msg = pchannel;
269
270         wpa_hexdump(MSG_MSGDUMP, "EAP-PSK: PCHANNEL - nonce",
271                     nonce, sizeof(nonce));
272         wpa_hexdump(MSG_MSGDUMP, "EAP-PSK: PCHANNEL - hdr",
273                     wpabuf_head(reqData), 5);
274         wpa_hexdump(MSG_MSGDUMP, "EAP-PSK: PCHANNEL - cipher msg", msg, left);
275
276         decrypted = os_malloc(left);
277         if (decrypted == NULL) {
278                 ret->methodState = METHOD_DONE;
279                 ret->decision = DECISION_FAIL;
280                 return NULL;
281         }
282         os_memcpy(decrypted, msg, left);
283
284         if (aes_128_eax_decrypt(data->tek, nonce, sizeof(nonce),
285                                 wpabuf_head(reqData),
286                                 sizeof(struct eap_hdr) + 1 +
287                                 sizeof(*hdr3) - EAP_PSK_MAC_LEN, decrypted,
288                                 left, tag)) {
289                 wpa_printf(MSG_WARNING, "EAP-PSK: PCHANNEL decryption failed");
290                 os_free(decrypted);
291                 return NULL;
292         }
293         wpa_hexdump(MSG_DEBUG, "EAP-PSK: Decrypted PCHANNEL message",
294                     decrypted, left);
295
296         /* Verify R flag */
297         switch (decrypted[0] >> 6) {
298         case EAP_PSK_R_FLAG_CONT:
299                 wpa_printf(MSG_DEBUG, "EAP-PSK: R flag - CONT - unsupported");
300                 failed = 1;
301                 break;
302         case EAP_PSK_R_FLAG_DONE_SUCCESS:
303                 wpa_printf(MSG_DEBUG, "EAP-PSK: R flag - DONE_SUCCESS");
304                 break;
305         case EAP_PSK_R_FLAG_DONE_FAILURE:
306                 wpa_printf(MSG_DEBUG, "EAP-PSK: R flag - DONE_FAILURE");
307                 wpa_printf(MSG_INFO, "EAP-PSK: Authentication server rejected "
308                            "authentication");
309                 failed = 1;
310                 break;
311         }
312
313         data_len = 1;
314         if ((decrypted[0] & EAP_PSK_E_FLAG) && left > 1)
315                 data_len++;
316         plen = sizeof(*hdr4) + 4 + 16 + data_len;
317         resp = eap_msg_alloc(EAP_VENDOR_IETF, EAP_TYPE_PSK, plen,
318                              EAP_CODE_RESPONSE, eap_get_id(reqData));
319         if (resp == NULL) {
320                 os_free(decrypted);
321                 return NULL;
322         }
323         hdr4 = wpabuf_put(resp, sizeof(*hdr4));
324         hdr4->flags = EAP_PSK_FLAGS_SET_T(3); /* T=3 */
325         os_memcpy(hdr4->rand_s, hdr3->rand_s, EAP_PSK_RAND_LEN);
326         rpchannel = wpabuf_put(resp, 4 + 16 + data_len);
327
328         /* nonce++ */
329         inc_byte_array(nonce, sizeof(nonce));
330         os_memcpy(rpchannel, nonce + 12, 4);
331
332         if (decrypted[0] & EAP_PSK_E_FLAG) {
333                 wpa_printf(MSG_DEBUG, "EAP-PSK: Unsupported E (Ext) flag");
334                 failed = 1;
335                 rpchannel[4 + 16] = (EAP_PSK_R_FLAG_DONE_FAILURE << 6) |
336                         EAP_PSK_E_FLAG;
337                 if (left > 1) {
338                         /* Add empty EXT_Payload with same EXT_Type */
339                         rpchannel[4 + 16 + 1] = decrypted[1];
340                 }
341         } else if (failed)
342                 rpchannel[4 + 16] = EAP_PSK_R_FLAG_DONE_FAILURE << 6;
343         else
344                 rpchannel[4 + 16] = EAP_PSK_R_FLAG_DONE_SUCCESS << 6;
345
346         wpa_hexdump(MSG_DEBUG, "EAP-PSK: reply message (plaintext)",
347                     rpchannel + 4 + 16, data_len);
348         if (aes_128_eax_encrypt(data->tek, nonce, sizeof(nonce),
349                                 wpabuf_head(resp),
350                                 sizeof(struct eap_hdr) + 1 + sizeof(*hdr4),
351                                 rpchannel + 4 + 16, data_len, rpchannel + 4)) {
352                 os_free(decrypted);
353                 wpabuf_free(resp);
354                 return NULL;
355         }
356         wpa_hexdump(MSG_DEBUG, "EAP-PSK: reply message (PCHANNEL)",
357                     rpchannel, 4 + 16 + data_len);
358
359         wpa_printf(MSG_DEBUG, "EAP-PSK: Completed %ssuccessfully",
360                    failed ? "un" : "");
361         data->state = PSK_DONE;
362         ret->methodState = METHOD_DONE;
363         ret->decision = failed ? DECISION_FAIL : DECISION_UNCOND_SUCC;
364
365         os_free(decrypted);
366
367         return resp;
368 }
369
370
371 static struct wpabuf * eap_psk_process(struct eap_sm *sm, void *priv,
372                                        struct eap_method_ret *ret,
373                                        const struct wpabuf *reqData)
374 {
375         struct eap_psk_data *data = priv;
376         const u8 *pos;
377         struct wpabuf *resp = NULL;
378         size_t len;
379
380         pos = eap_hdr_validate(EAP_VENDOR_IETF, EAP_TYPE_PSK, reqData, &len);
381         if (pos == NULL) {
382                 ret->ignore = TRUE;
383                 return NULL;
384         }
385
386         ret->ignore = FALSE;
387         ret->methodState = METHOD_MAY_CONT;
388         ret->decision = DECISION_FAIL;
389         ret->allowNotifications = TRUE;
390
391         switch (data->state) {
392         case PSK_INIT:
393                 resp = eap_psk_process_1(data, ret, reqData);
394                 break;
395         case PSK_MAC_SENT:
396                 resp = eap_psk_process_3(data, ret, reqData);
397                 break;
398         case PSK_DONE:
399                 wpa_printf(MSG_DEBUG, "EAP-PSK: in DONE state - ignore "
400                            "unexpected message");
401                 ret->ignore = TRUE;
402                 return NULL;
403         }
404
405         if (ret->methodState == METHOD_DONE) {
406                 ret->allowNotifications = FALSE;
407         }
408
409         return resp;
410 }
411
412
413 static Boolean eap_psk_isKeyAvailable(struct eap_sm *sm, void *priv)
414 {
415         struct eap_psk_data *data = priv;
416         return data->state == PSK_DONE;
417 }
418
419
420 static u8 * eap_psk_getKey(struct eap_sm *sm, void *priv, size_t *len)
421 {
422         struct eap_psk_data *data = priv;
423         u8 *key;
424
425         if (data->state != PSK_DONE)
426                 return NULL;
427
428         key = os_malloc(EAP_MSK_LEN);
429         if (key == NULL)
430                 return NULL;
431
432         *len = EAP_MSK_LEN;
433         os_memcpy(key, data->msk, EAP_MSK_LEN);
434
435         return key;
436 }
437
438
439 static u8 * eap_psk_get_session_id(struct eap_sm *sm, void *priv, size_t *len)
440 {
441         struct eap_psk_data *data = priv;
442         u8 *id;
443
444         if (data->state != PSK_DONE)
445                 return NULL;
446
447         *len = 1 + 2 * EAP_PSK_RAND_LEN;
448         id = os_malloc(*len);
449         if (id == NULL)
450                 return NULL;
451
452         id[0] = EAP_TYPE_PSK;
453         os_memcpy(id + 1, data->rand_p, EAP_PSK_RAND_LEN);
454         os_memcpy(id + 1 + EAP_PSK_RAND_LEN, data->rand_s, EAP_PSK_RAND_LEN);
455         wpa_hexdump(MSG_DEBUG, "EAP-PSK: Derived Session-Id", id, *len);
456
457         return id;
458 }
459
460
461 static u8 * eap_psk_get_emsk(struct eap_sm *sm, void *priv, size_t *len)
462 {
463         struct eap_psk_data *data = priv;
464         u8 *key;
465
466         if (data->state != PSK_DONE)
467                 return NULL;
468
469         key = os_malloc(EAP_EMSK_LEN);
470         if (key == NULL)
471                 return NULL;
472
473         *len = EAP_EMSK_LEN;
474         os_memcpy(key, data->emsk, EAP_EMSK_LEN);
475
476         return key;
477 }
478
479
480 int eap_peer_psk_register(void)
481 {
482         struct eap_method *eap;
483         int ret;
484
485         eap = eap_peer_method_alloc(EAP_PEER_METHOD_INTERFACE_VERSION,
486                                     EAP_VENDOR_IETF, EAP_TYPE_PSK, "PSK");
487         if (eap == NULL)
488                 return -1;
489
490         eap->init = eap_psk_init;
491         eap->deinit = eap_psk_deinit;
492         eap->process = eap_psk_process;
493         eap->isKeyAvailable = eap_psk_isKeyAvailable;
494         eap->getKey = eap_psk_getKey;
495         eap->getSessionId = eap_psk_get_session_id;
496         eap->get_emsk = eap_psk_get_emsk;
497
498         ret = eap_peer_method_register(eap);
499         if (ret)
500                 eap_peer_method_free(eap);
501         return ret;
502 }