1 /*
2  * EAP peer method: EAP-AKA (RFC 4187) and EAP-AKA' (RFC 5448)
3  * Copyright (c) 2004-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 "includes.h"
10 
11 #include "common.h"
12 #include "pcsc_funcs.h"
13 #include "crypto/crypto.h"
14 #include "crypto/sha1.h"
15 #include "crypto/sha256.h"
16 #include "crypto/milenage.h"
17 #include "eap_common/eap_sim_common.h"
18 #include "eap_config.h"
19 #include "eap_i.h"
20 
21 
22 struct eap_aka_data {
23 	u8 ik[EAP_AKA_IK_LEN], ck[EAP_AKA_CK_LEN], res[EAP_AKA_RES_MAX_LEN];
24 	size_t res_len;
25 	u8 nonce_s[EAP_SIM_NONCE_S_LEN];
26 	u8 mk[EAP_SIM_MK_LEN];
27 	u8 k_aut[EAP_AKA_PRIME_K_AUT_LEN];
28 	u8 k_encr[EAP_SIM_K_ENCR_LEN];
29 	u8 k_re[EAP_AKA_PRIME_K_RE_LEN]; /* EAP-AKA' only */
30 	u8 msk[EAP_SIM_KEYING_DATA_LEN];
31 	u8 emsk[EAP_EMSK_LEN];
32 	u8 rand[EAP_AKA_RAND_LEN], autn[EAP_AKA_AUTN_LEN];
33 	u8 auts[EAP_AKA_AUTS_LEN];
34 
35 	int num_id_req, num_notification;
36 	u8 *pseudonym;
37 	size_t pseudonym_len;
38 	u8 *reauth_id;
39 	size_t reauth_id_len;
40 	int reauth;
41 	unsigned int counter, counter_too_small;
42 	u8 *last_eap_identity;
43 	size_t last_eap_identity_len;
44 	enum {
45 		CONTINUE, RESULT_SUCCESS, RESULT_FAILURE, SUCCESS, FAILURE
46 	} state;
47 
48 	struct wpabuf *id_msgs;
49 	int prev_id;
50 	int result_ind, use_result_ind;
51 	u8 eap_method;
52 	u8 *network_name;
53 	size_t network_name_len;
54 	u16 kdf;
55 	int kdf_negotiation;
56 };
57 
58 
59 #ifndef CONFIG_NO_STDOUT_DEBUG
60 static const char * eap_aka_state_txt(int state)
61 {
62 	switch (state) {
63 	case CONTINUE:
64 		return "CONTINUE";
65 	case RESULT_SUCCESS:
66 		return "RESULT_SUCCESS";
67 	case RESULT_FAILURE:
68 		return "RESULT_FAILURE";
69 	case SUCCESS:
70 		return "SUCCESS";
71 	case FAILURE:
72 		return "FAILURE";
73 	default:
74 		return "?";
75 	}
76 }
77 #endif /* CONFIG_NO_STDOUT_DEBUG */
78 
79 
80 static void eap_aka_state(struct eap_aka_data *data, int state)
81 {
82 	wpa_printf(MSG_DEBUG, "EAP-AKA: %s -> %s",
83 		   eap_aka_state_txt(data->state),
84 		   eap_aka_state_txt(state));
85 	data->state = state;
86 }
87 
88 
89 static void * eap_aka_init(struct eap_sm *sm)
90 {
91 	struct eap_aka_data *data;
92 	const char *phase1 = eap_get_config_phase1(sm);
93 	struct eap_peer_config *config = eap_get_config(sm);
94 
95 	data = os_zalloc(sizeof(*data));
96 	if (data == NULL)
97 		return NULL;
98 
99 	data->eap_method = EAP_TYPE_AKA;
100 
101 	eap_aka_state(data, CONTINUE);
102 	data->prev_id = -1;
103 
104 	data->result_ind = phase1 && os_strstr(phase1, "result_ind=1") != NULL;
105 
106 	if (config && config->anonymous_identity) {
107 		data->pseudonym = os_malloc(config->anonymous_identity_len);
108 		if (data->pseudonym) {
109 			os_memcpy(data->pseudonym, config->anonymous_identity,
110 				  config->anonymous_identity_len);
111 			data->pseudonym_len = config->anonymous_identity_len;
112 		}
113 	}
114 
115 	return data;
116 }
117 
118 
119 #ifdef EAP_AKA_PRIME
120 static void * eap_aka_prime_init(struct eap_sm *sm)
121 {
122 	struct eap_aka_data *data = eap_aka_init(sm);
123 	if (data == NULL)
124 		return NULL;
125 	data->eap_method = EAP_TYPE_AKA_PRIME;
126 	return data;
127 }
128 #endif /* EAP_AKA_PRIME */
129 
130 
131 static void eap_aka_deinit(struct eap_sm *sm, void *priv)
132 {
133 	struct eap_aka_data *data = priv;
134 	if (data) {
135 		os_free(data->pseudonym);
136 		os_free(data->reauth_id);
137 		os_free(data->last_eap_identity);
138 		wpabuf_free(data->id_msgs);
139 		os_free(data->network_name);
140 		os_free(data);
141 	}
142 }
143 
144 
145 static int eap_aka_ext_sim_req(struct eap_sm *sm, struct eap_aka_data *data)
146 {
147 	char req[200], *pos, *end;
148 
149 	wpa_printf(MSG_DEBUG, "EAP-AKA: Use external USIM processing");
150 	pos = req;
151 	end = pos + sizeof(req);
152 	pos += os_snprintf(pos, end - pos, "UMTS-AUTH");
153 	pos += os_snprintf(pos, end - pos, ":");
154 	pos += wpa_snprintf_hex(pos, end - pos, data->rand, EAP_AKA_RAND_LEN);
155 	pos += os_snprintf(pos, end - pos, ":");
156 	pos += wpa_snprintf_hex(pos, end - pos, data->autn, EAP_AKA_AUTN_LEN);
157 
158 	eap_sm_request_sim(sm, req);
159 	return 1;
160 }
161 
162 
163 static int eap_aka_ext_sim_result(struct eap_sm *sm, struct eap_aka_data *data,
164 				  struct eap_peer_config *conf)
165 {
166 	char *resp, *pos;
167 
168 	wpa_printf(MSG_DEBUG,
169 		   "EAP-AKA: Use result from external USIM processing");
170 
171 	resp = conf->external_sim_resp;
172 	conf->external_sim_resp = NULL;
173 
174 	if (os_strncmp(resp, "UMTS-AUTS:", 10) == 0) {
175 		pos = resp + 10;
176 		if (hexstr2bin(pos, data->auts, EAP_AKA_AUTS_LEN) < 0)
177 			goto invalid;
178 		wpa_hexdump_key(MSG_DEBUG, "EAP-AKA: AUTS", data->auts,
179 				EAP_AKA_AUTS_LEN);
180 		os_free(resp);
181 		return -2;
182 	}
183 
184 	if (os_strncmp(resp, "UMTS-AUTH:", 10) != 0) {
185 		wpa_printf(MSG_DEBUG, "EAP-AKA: Unrecognized external USIM processing response");
186 		os_free(resp);
187 		return -1;
188 	}
189 
190 	pos = resp + 10;
191 	wpa_hexdump(MSG_DEBUG, "EAP-AKA: RAND", data->rand, EAP_AKA_RAND_LEN);
192 
193 	if (hexstr2bin(pos, data->ik, EAP_AKA_IK_LEN) < 0)
194 		goto invalid;
195 	wpa_hexdump_key(MSG_DEBUG, "EAP-AKA: IK", data->ik, EAP_AKA_IK_LEN);
196 	pos += EAP_AKA_IK_LEN * 2;
197 	if (*pos != ':')
198 		goto invalid;
199 	pos++;
200 
201 	if (hexstr2bin(pos, data->ck, EAP_AKA_CK_LEN) < 0)
202 		goto invalid;
203 	wpa_hexdump_key(MSG_DEBUG, "EAP-AKA: CK", data->ck, EAP_AKA_CK_LEN);
204 	pos += EAP_AKA_CK_LEN * 2;
205 	if (*pos != ':')
206 		goto invalid;
207 	pos++;
208 
209 	data->res_len = os_strlen(pos) / 2;
210 	if (data->res_len > EAP_AKA_RES_MAX_LEN) {
211 		data->res_len = 0;
212 		goto invalid;
213 	}
214 	if (hexstr2bin(pos, data->res, data->res_len) < 0)
215 		goto invalid;
216 	wpa_hexdump_key(MSG_DEBUG, "EAP-AKA: RES", data->res, data->res_len);
217 
218 	os_free(resp);
219 	return 0;
220 
221 invalid:
222 	wpa_printf(MSG_DEBUG, "EAP-AKA: Invalid external USIM processing UMTS-AUTH response");
223 	os_free(resp);
224 	return -1;
225 }
226 
227 
228 static int eap_aka_umts_auth(struct eap_sm *sm, struct eap_aka_data *data)
229 {
230 	struct eap_peer_config *conf;
231 
232 	wpa_printf(MSG_DEBUG, "EAP-AKA: UMTS authentication algorithm");
233 
234 	conf = eap_get_config(sm);
235 	if (conf == NULL)
236 		return -1;
237 
238 	if (sm->external_sim) {
239 		if (conf->external_sim_resp)
240 			return eap_aka_ext_sim_result(sm, data, conf);
241 		else
242 			return eap_aka_ext_sim_req(sm, data);
243 	}
244 
245 	if (conf->pcsc) {
246 		return scard_umts_auth(sm->scard_ctx, data->rand,
247 				       data->autn, data->res, &data->res_len,
248 				       data->ik, data->ck, data->auts);
249 	}
250 
251 #ifdef CONFIG_USIM_SIMULATOR
252 	if (conf->password) {
253 		u8 opc[16], k[16], sqn[6];
254 		const char *pos;
255 		wpa_printf(MSG_DEBUG, "EAP-AKA: Use internal Milenage "
256 			   "implementation for UMTS authentication");
257 		if (conf->password_len < 78) {
258 			wpa_printf(MSG_DEBUG, "EAP-AKA: invalid Milenage "
259 				   "password");
260 			return -1;
261 		}
262 		pos = (const char *) conf->password;
263 		if (hexstr2bin(pos, k, 16))
264 			return -1;
265 		pos += 32;
266 		if (*pos != ':')
267 			return -1;
268 		pos++;
269 
270 		if (hexstr2bin(pos, opc, 16))
271 			return -1;
272 		pos += 32;
273 		if (*pos != ':')
274 			return -1;
275 		pos++;
276 
277 		if (hexstr2bin(pos, sqn, 6))
278 			return -1;
279 
280 		return milenage_check(opc, k, sqn, data->rand, data->autn,
281 				      data->ik, data->ck,
282 				      data->res, &data->res_len, data->auts);
283 	}
284 #endif /* CONFIG_USIM_SIMULATOR */
285 
286 #ifdef CONFIG_USIM_HARDCODED
287 	wpa_printf(MSG_DEBUG, "EAP-AKA: Use hardcoded Kc and SRES values for "
288 		   "testing");
289 
290 	/* These hardcoded Kc and SRES values are used for testing.
291 	 * Could consider making them configurable. */
292 	os_memset(data->res, '2', EAP_AKA_RES_MAX_LEN);
293 	data->res_len = EAP_AKA_RES_MAX_LEN;
294 	os_memset(data->ik, '3', EAP_AKA_IK_LEN);
295 	os_memset(data->ck, '4', EAP_AKA_CK_LEN);
296 	{
297 		u8 autn[EAP_AKA_AUTN_LEN];
298 		os_memset(autn, '1', EAP_AKA_AUTN_LEN);
299 		if (os_memcmp(autn, data->autn, EAP_AKA_AUTN_LEN) != 0) {
300 			wpa_printf(MSG_WARNING, "EAP-AKA: AUTN did not match "
301 				   "with expected value");
302 			return -1;
303 		}
304 	}
305 #if 0
306 	{
307 		static int test_resync = 1;
308 		if (test_resync) {
309 			/* Test Resynchronization */
310 			test_resync = 0;
311 			return -2;
312 		}
313 	}
314 #endif
315 	return 0;
316 
317 #else /* CONFIG_USIM_HARDCODED */
318 
319 	wpa_printf(MSG_DEBUG, "EAP-AKA: No UMTS authentication algorith "
320 		   "enabled");
321 	return -1;
322 
323 #endif /* CONFIG_USIM_HARDCODED */
324 }
325 
326 
327 #define CLEAR_PSEUDONYM	0x01
328 #define CLEAR_REAUTH_ID	0x02
329 #define CLEAR_EAP_ID	0x04
330 
331 static void eap_aka_clear_identities(struct eap_sm *sm,
332 				     struct eap_aka_data *data, int id)
333 {
334 	if ((id & CLEAR_PSEUDONYM) && data->pseudonym) {
335 		wpa_printf(MSG_DEBUG, "EAP-AKA: forgetting old pseudonym");
336 		os_free(data->pseudonym);
337 		data->pseudonym = NULL;
338 		data->pseudonym_len = 0;
339 		eap_set_anon_id(sm, NULL, 0);
340 	}
341 	if ((id & CLEAR_REAUTH_ID) && data->reauth_id) {
342 		wpa_printf(MSG_DEBUG, "EAP-AKA: forgetting old reauth_id");
343 		os_free(data->reauth_id);
344 		data->reauth_id = NULL;
345 		data->reauth_id_len = 0;
346 	}
347 	if ((id & CLEAR_EAP_ID) && data->last_eap_identity) {
348 		wpa_printf(MSG_DEBUG, "EAP-AKA: forgetting old eap_id");
349 		os_free(data->last_eap_identity);
350 		data->last_eap_identity = NULL;
351 		data->last_eap_identity_len = 0;
352 	}
353 }
354 
355 
356 static int eap_aka_learn_ids(struct eap_sm *sm, struct eap_aka_data *data,
357 			     struct eap_sim_attrs *attr)
358 {
359 	if (attr->next_pseudonym) {
360 		const u8 *identity = NULL;
361 		size_t identity_len = 0;
362 		const u8 *realm = NULL;
363 		size_t realm_len = 0;
364 
365 		wpa_hexdump_ascii(MSG_DEBUG,
366 				  "EAP-AKA: (encr) AT_NEXT_PSEUDONYM",
367 				  attr->next_pseudonym,
368 				  attr->next_pseudonym_len);
369 		os_free(data->pseudonym);
370 		/* Look for the realm of the permanent identity */
371 		identity = eap_get_config_identity(sm, &identity_len);
372 		if (identity) {
373 			for (realm = identity, realm_len = identity_len;
374 			     realm_len > 0; realm_len--, realm++) {
375 				if (*realm == '@')
376 					break;
377 			}
378 		}
379 		data->pseudonym = os_malloc(attr->next_pseudonym_len +
380 					    realm_len);
381 		if (data->pseudonym == NULL) {
382 			wpa_printf(MSG_INFO, "EAP-AKA: (encr) No memory for "
383 				   "next pseudonym");
384 			data->pseudonym_len = 0;
385 			return -1;
386 		}
387 		os_memcpy(data->pseudonym, attr->next_pseudonym,
388 			  attr->next_pseudonym_len);
389 		if (realm_len) {
390 			os_memcpy(data->pseudonym + attr->next_pseudonym_len,
391 				  realm, realm_len);
392 		}
393 		data->pseudonym_len = attr->next_pseudonym_len + realm_len;
394 		eap_set_anon_id(sm, data->pseudonym, data->pseudonym_len);
395 	}
396 
397 	if (attr->next_reauth_id) {
398 		os_free(data->reauth_id);
399 		data->reauth_id = os_malloc(attr->next_reauth_id_len);
400 		if (data->reauth_id == NULL) {
401 			wpa_printf(MSG_INFO, "EAP-AKA: (encr) No memory for "
402 				   "next reauth_id");
403 			data->reauth_id_len = 0;
404 			return -1;
405 		}
406 		os_memcpy(data->reauth_id, attr->next_reauth_id,
407 			  attr->next_reauth_id_len);
408 		data->reauth_id_len = attr->next_reauth_id_len;
409 		wpa_hexdump_ascii(MSG_DEBUG,
410 				  "EAP-AKA: (encr) AT_NEXT_REAUTH_ID",
411 				  data->reauth_id,
412 				  data->reauth_id_len);
413 	}
414 
415 	return 0;
416 }
417 
418 
419 static int eap_aka_add_id_msg(struct eap_aka_data *data,
420 			      const struct wpabuf *msg)
421 {
422 	if (msg == NULL)
423 		return -1;
424 
425 	if (data->id_msgs == NULL) {
426 		data->id_msgs = wpabuf_dup(msg);
427 		return data->id_msgs == NULL ? -1 : 0;
428 	}
429 
430 	if (wpabuf_resize(&data->id_msgs, wpabuf_len(msg)) < 0)
431 		return -1;
432 	wpabuf_put_buf(data->id_msgs, msg);
433 
434 	return 0;
435 }
436 
437 
438 static void eap_aka_add_checkcode(struct eap_aka_data *data,
439 				  struct eap_sim_msg *msg)
440 {
441 	const u8 *addr;
442 	size_t len;
443 	u8 hash[SHA256_MAC_LEN];
444 
445 	wpa_printf(MSG_DEBUG, "   AT_CHECKCODE");
446 
447 	if (data->id_msgs == NULL) {
448 		/*
449 		 * No EAP-AKA/Identity packets were exchanged - send empty
450 		 * checkcode.
451 		 */
452 		eap_sim_msg_add(msg, EAP_SIM_AT_CHECKCODE, 0, NULL, 0);
453 		return;
454 	}
455 
456 	/* Checkcode is SHA1/SHA256 hash over all EAP-AKA/Identity packets. */
457 	addr = wpabuf_head(data->id_msgs);
458 	len = wpabuf_len(data->id_msgs);
459 	wpa_hexdump(MSG_MSGDUMP, "EAP-AKA: AT_CHECKCODE data", addr, len);
460 #ifdef EAP_AKA_PRIME
461 	if (data->eap_method == EAP_TYPE_AKA_PRIME)
462 		sha256_vector(1, &addr, &len, hash);
463 	else
464 #endif /* EAP_AKA_PRIME */
465 		sha1_vector(1, &addr, &len, hash);
466 
467 	eap_sim_msg_add(msg, EAP_SIM_AT_CHECKCODE, 0, hash,
468 			data->eap_method == EAP_TYPE_AKA_PRIME ?
469 			EAP_AKA_PRIME_CHECKCODE_LEN : EAP_AKA_CHECKCODE_LEN);
470 }
471 
472 
473 static int eap_aka_verify_checkcode(struct eap_aka_data *data,
474 				    const u8 *checkcode, size_t checkcode_len)
475 {
476 	const u8 *addr;
477 	size_t len;
478 	u8 hash[SHA256_MAC_LEN];
479 	size_t hash_len;
480 
481 	if (checkcode == NULL)
482 		return -1;
483 
484 	if (data->id_msgs == NULL) {
485 		if (checkcode_len != 0) {
486 			wpa_printf(MSG_DEBUG, "EAP-AKA: Checkcode from server "
487 				   "indicates that AKA/Identity messages were "
488 				   "used, but they were not");
489 			return -1;
490 		}
491 		return 0;
492 	}
493 
494 	hash_len = data->eap_method == EAP_TYPE_AKA_PRIME ?
495 		EAP_AKA_PRIME_CHECKCODE_LEN : EAP_AKA_CHECKCODE_LEN;
496 
497 	if (checkcode_len != hash_len) {
498 		wpa_printf(MSG_DEBUG, "EAP-AKA: Checkcode from server "
499 			   "indicates that AKA/Identity message were not "
500 			   "used, but they were");
501 		return -1;
502 	}
503 
504 	/* Checkcode is SHA1/SHA256 hash over all EAP-AKA/Identity packets. */
505 	addr = wpabuf_head(data->id_msgs);
506 	len = wpabuf_len(data->id_msgs);
507 #ifdef EAP_AKA_PRIME
508 	if (data->eap_method == EAP_TYPE_AKA_PRIME)
509 		sha256_vector(1, &addr, &len, hash);
510 	else
511 #endif /* EAP_AKA_PRIME */
512 		sha1_vector(1, &addr, &len, hash);
513 
514 	if (os_memcmp(hash, checkcode, hash_len) != 0) {
515 		wpa_printf(MSG_DEBUG, "EAP-AKA: Mismatch in AT_CHECKCODE");
516 		return -1;
517 	}
518 
519 	return 0;
520 }
521 
522 
523 static struct wpabuf * eap_aka_client_error(struct eap_aka_data *data, u8 id,
524 					    int err)
525 {
526 	struct eap_sim_msg *msg;
527 
528 	eap_aka_state(data, FAILURE);
529 	data->num_id_req = 0;
530 	data->num_notification = 0;
531 
532 	wpa_printf(MSG_DEBUG, "EAP-AKA: Send Client-Error (error code %d)",
533 		   err);
534 	msg = eap_sim_msg_init(EAP_CODE_RESPONSE, id, data->eap_method,
535 			       EAP_AKA_SUBTYPE_CLIENT_ERROR);
536 	eap_sim_msg_add(msg, EAP_SIM_AT_CLIENT_ERROR_CODE, err, NULL, 0);
537 	return eap_sim_msg_finish(msg, NULL, NULL, 0);
538 }
539 
540 
541 static struct wpabuf * eap_aka_authentication_reject(struct eap_aka_data *data,
542 						     u8 id)
543 {
544 	struct eap_sim_msg *msg;
545 
546 	eap_aka_state(data, FAILURE);
547 	data->num_id_req = 0;
548 	data->num_notification = 0;
549 
550 	wpa_printf(MSG_DEBUG, "Generating EAP-AKA Authentication-Reject "
551 		   "(id=%d)", id);
552 	msg = eap_sim_msg_init(EAP_CODE_RESPONSE, id, data->eap_method,
553 			       EAP_AKA_SUBTYPE_AUTHENTICATION_REJECT);
554 	return eap_sim_msg_finish(msg, NULL, NULL, 0);
555 }
556 
557 
558 static struct wpabuf * eap_aka_synchronization_failure(
559 	struct eap_aka_data *data, u8 id)
560 {
561 	struct eap_sim_msg *msg;
562 
563 	data->num_id_req = 0;
564 	data->num_notification = 0;
565 
566 	wpa_printf(MSG_DEBUG, "Generating EAP-AKA Synchronization-Failure "
567 		   "(id=%d)", id);
568 	msg = eap_sim_msg_init(EAP_CODE_RESPONSE, id, data->eap_method,
569 			       EAP_AKA_SUBTYPE_SYNCHRONIZATION_FAILURE);
570 	wpa_printf(MSG_DEBUG, "   AT_AUTS");
571 	eap_sim_msg_add_full(msg, EAP_SIM_AT_AUTS, data->auts,
572 			     EAP_AKA_AUTS_LEN);
573 	return eap_sim_msg_finish(msg, NULL, NULL, 0);
574 }
575 
576 
577 static struct wpabuf * eap_aka_response_identity(struct eap_sm *sm,
578 						 struct eap_aka_data *data,
579 						 u8 id,
580 						 enum eap_sim_id_req id_req)
581 {
582 	const u8 *identity = NULL;
583 	size_t identity_len = 0;
584 	struct eap_sim_msg *msg;
585 
586 	data->reauth = 0;
587 	if (id_req == ANY_ID && data->reauth_id) {
588 		identity = data->reauth_id;
589 		identity_len = data->reauth_id_len;
590 		data->reauth = 1;
591 	} else if ((id_req == ANY_ID || id_req == FULLAUTH_ID) &&
592 		   data->pseudonym) {
593 		identity = data->pseudonym;
594 		identity_len = data->pseudonym_len;
595 		eap_aka_clear_identities(sm, data, CLEAR_REAUTH_ID);
596 	} else if (id_req != NO_ID_REQ) {
597 		identity = eap_get_config_identity(sm, &identity_len);
598 		if (identity) {
599 			eap_aka_clear_identities(sm, data, CLEAR_PSEUDONYM |
600 						 CLEAR_REAUTH_ID);
601 		}
602 	}
603 	if (id_req != NO_ID_REQ)
604 		eap_aka_clear_identities(sm, data, CLEAR_EAP_ID);
605 
606 	wpa_printf(MSG_DEBUG, "Generating EAP-AKA Identity (id=%d)", id);
607 	msg = eap_sim_msg_init(EAP_CODE_RESPONSE, id, data->eap_method,
608 			       EAP_AKA_SUBTYPE_IDENTITY);
609 
610 	if (identity) {
611 		wpa_hexdump_ascii(MSG_DEBUG, "   AT_IDENTITY",
612 				  identity, identity_len);
613 		eap_sim_msg_add(msg, EAP_SIM_AT_IDENTITY, identity_len,
614 				identity, identity_len);
615 	}
616 
617 	return eap_sim_msg_finish(msg, NULL, NULL, 0);
618 }
619 
620 
621 static struct wpabuf * eap_aka_response_challenge(struct eap_aka_data *data,
622 						  u8 id)
623 {
624 	struct eap_sim_msg *msg;
625 
626 	wpa_printf(MSG_DEBUG, "Generating EAP-AKA Challenge (id=%d)", id);
627 	msg = eap_sim_msg_init(EAP_CODE_RESPONSE, id, data->eap_method,
628 			       EAP_AKA_SUBTYPE_CHALLENGE);
629 	wpa_printf(MSG_DEBUG, "   AT_RES");
630 	eap_sim_msg_add(msg, EAP_SIM_AT_RES, data->res_len * 8,
631 			data->res, data->res_len);
632 	eap_aka_add_checkcode(data, msg);
633 	if (data->use_result_ind) {
634 		wpa_printf(MSG_DEBUG, "   AT_RESULT_IND");
635 		eap_sim_msg_add(msg, EAP_SIM_AT_RESULT_IND, 0, NULL, 0);
636 	}
637 	wpa_printf(MSG_DEBUG, "   AT_MAC");
638 	eap_sim_msg_add_mac(msg, EAP_SIM_AT_MAC);
639 	return eap_sim_msg_finish(msg, data->k_aut, (u8 *) "", 0);
640 }
641 
642 
643 static struct wpabuf * eap_aka_response_reauth(struct eap_aka_data *data,
644 					       u8 id, int counter_too_small,
645 					       const u8 *nonce_s)
646 {
647 	struct eap_sim_msg *msg;
648 	unsigned int counter;
649 
650 	wpa_printf(MSG_DEBUG, "Generating EAP-AKA Reauthentication (id=%d)",
651 		   id);
652 	msg = eap_sim_msg_init(EAP_CODE_RESPONSE, id, data->eap_method,
653 			       EAP_AKA_SUBTYPE_REAUTHENTICATION);
654 	wpa_printf(MSG_DEBUG, "   AT_IV");
655 	wpa_printf(MSG_DEBUG, "   AT_ENCR_DATA");
656 	eap_sim_msg_add_encr_start(msg, EAP_SIM_AT_IV, EAP_SIM_AT_ENCR_DATA);
657 
658 	if (counter_too_small) {
659 		wpa_printf(MSG_DEBUG, "   *AT_COUNTER_TOO_SMALL");
660 		eap_sim_msg_add(msg, EAP_SIM_AT_COUNTER_TOO_SMALL, 0, NULL, 0);
661 		counter = data->counter_too_small;
662 	} else
663 		counter = data->counter;
664 
665 	wpa_printf(MSG_DEBUG, "   *AT_COUNTER %d", counter);
666 	eap_sim_msg_add(msg, EAP_SIM_AT_COUNTER, counter, NULL, 0);
667 
668 	if (eap_sim_msg_add_encr_end(msg, data->k_encr, EAP_SIM_AT_PADDING)) {
669 		wpa_printf(MSG_WARNING, "EAP-AKA: Failed to encrypt "
670 			   "AT_ENCR_DATA");
671 		eap_sim_msg_free(msg);
672 		return NULL;
673 	}
674 	eap_aka_add_checkcode(data, msg);
675 	if (data->use_result_ind) {
676 		wpa_printf(MSG_DEBUG, "   AT_RESULT_IND");
677 		eap_sim_msg_add(msg, EAP_SIM_AT_RESULT_IND, 0, NULL, 0);
678 	}
679 	wpa_printf(MSG_DEBUG, "   AT_MAC");
680 	eap_sim_msg_add_mac(msg, EAP_SIM_AT_MAC);
681 	return eap_sim_msg_finish(msg, data->k_aut, nonce_s,
682 				  EAP_SIM_NONCE_S_LEN);
683 }
684 
685 
686 static struct wpabuf * eap_aka_response_notification(struct eap_aka_data *data,
687 						     u8 id, u16 notification)
688 {
689 	struct eap_sim_msg *msg;
690 	u8 *k_aut = (notification & 0x4000) == 0 ? data->k_aut : NULL;
691 
692 	wpa_printf(MSG_DEBUG, "Generating EAP-AKA Notification (id=%d)", id);
693 	msg = eap_sim_msg_init(EAP_CODE_RESPONSE, id, data->eap_method,
694 			       EAP_AKA_SUBTYPE_NOTIFICATION);
695 	if (k_aut && data->reauth) {
696 		wpa_printf(MSG_DEBUG, "   AT_IV");
697 		wpa_printf(MSG_DEBUG, "   AT_ENCR_DATA");
698 		eap_sim_msg_add_encr_start(msg, EAP_SIM_AT_IV,
699 					   EAP_SIM_AT_ENCR_DATA);
700 		wpa_printf(MSG_DEBUG, "   *AT_COUNTER %d", data->counter);
701 		eap_sim_msg_add(msg, EAP_SIM_AT_COUNTER, data->counter,
702 				NULL, 0);
703 		if (eap_sim_msg_add_encr_end(msg, data->k_encr,
704 					     EAP_SIM_AT_PADDING)) {
705 			wpa_printf(MSG_WARNING, "EAP-AKA: Failed to encrypt "
706 				   "AT_ENCR_DATA");
707 			eap_sim_msg_free(msg);
708 			return NULL;
709 		}
710 	}
711 	if (k_aut) {
712 		wpa_printf(MSG_DEBUG, "   AT_MAC");
713 		eap_sim_msg_add_mac(msg, EAP_SIM_AT_MAC);
714 	}
715 	return eap_sim_msg_finish(msg, k_aut, (u8 *) "", 0);
716 }
717 
718 
719 static struct wpabuf * eap_aka_process_identity(struct eap_sm *sm,
720 						struct eap_aka_data *data,
721 						u8 id,
722 						const struct wpabuf *reqData,
723 						struct eap_sim_attrs *attr)
724 {
725 	int id_error;
726 	struct wpabuf *buf;
727 
728 	wpa_printf(MSG_DEBUG, "EAP-AKA: subtype Identity");
729 
730 	id_error = 0;
731 	switch (attr->id_req) {
732 	case NO_ID_REQ:
733 		break;
734 	case ANY_ID:
735 		if (data->num_id_req > 0)
736 			id_error++;
737 		data->num_id_req++;
738 		break;
739 	case FULLAUTH_ID:
740 		if (data->num_id_req > 1)
741 			id_error++;
742 		data->num_id_req++;
743 		break;
744 	case PERMANENT_ID:
745 		if (data->num_id_req > 2)
746 			id_error++;
747 		data->num_id_req++;
748 		break;
749 	}
750 	if (id_error) {
751 		wpa_printf(MSG_INFO, "EAP-AKA: Too many ID requests "
752 			   "used within one authentication");
753 		return eap_aka_client_error(data, id,
754 					    EAP_AKA_UNABLE_TO_PROCESS_PACKET);
755 	}
756 
757 	buf = eap_aka_response_identity(sm, data, id, attr->id_req);
758 
759 	if (data->prev_id != id) {
760 		eap_aka_add_id_msg(data, reqData);
761 		eap_aka_add_id_msg(data, buf);
762 		data->prev_id = id;
763 	}
764 
765 	return buf;
766 }
767 
768 
769 static int eap_aka_verify_mac(struct eap_aka_data *data,
770 			      const struct wpabuf *req,
771 			      const u8 *mac, const u8 *extra,
772 			      size_t extra_len)
773 {
774 	if (data->eap_method == EAP_TYPE_AKA_PRIME)
775 		return eap_sim_verify_mac_sha256(data->k_aut, req, mac, extra,
776 						 extra_len);
777 	return eap_sim_verify_mac(data->k_aut, req, mac, extra, extra_len);
778 }
779 
780 
781 #ifdef EAP_AKA_PRIME
782 static struct wpabuf * eap_aka_prime_kdf_select(struct eap_aka_data *data,
783 						u8 id, u16 kdf)
784 {
785 	struct eap_sim_msg *msg;
786 
787 	data->kdf_negotiation = 1;
788 	data->kdf = kdf;
789 	wpa_printf(MSG_DEBUG, "Generating EAP-AKA Challenge (id=%d) (KDF "
790 		   "select)", id);
791 	msg = eap_sim_msg_init(EAP_CODE_RESPONSE, id, data->eap_method,
792 			       EAP_AKA_SUBTYPE_CHALLENGE);
793 	wpa_printf(MSG_DEBUG, "   AT_KDF");
794 	eap_sim_msg_add(msg, EAP_SIM_AT_KDF, kdf, NULL, 0);
795 	return eap_sim_msg_finish(msg, NULL, NULL, 0);
796 }
797 
798 
799 static struct wpabuf * eap_aka_prime_kdf_neg(struct eap_aka_data *data,
800 					     u8 id, struct eap_sim_attrs *attr)
801 {
802 	size_t i;
803 
804 	for (i = 0; i < attr->kdf_count; i++) {
805 		if (attr->kdf[i] == EAP_AKA_PRIME_KDF)
806 			return eap_aka_prime_kdf_select(data, id,
807 							EAP_AKA_PRIME_KDF);
808 	}
809 
810 	/* No matching KDF found - fail authentication as if AUTN had been
811 	 * incorrect */
812 	return eap_aka_authentication_reject(data, id);
813 }
814 
815 
816 static int eap_aka_prime_kdf_valid(struct eap_aka_data *data,
817 				   struct eap_sim_attrs *attr)
818 {
819 	size_t i, j;
820 
821 	if (attr->kdf_count == 0)
822 		return 0;
823 
824 	/* The only allowed (and required) duplication of a KDF is the addition
825 	 * of the selected KDF into the beginning of the list. */
826 
827 	if (data->kdf_negotiation) {
828 		if (attr->kdf[0] != data->kdf) {
829 			wpa_printf(MSG_WARNING, "EAP-AKA': The server did not "
830 				   "accept the selected KDF");
831 			return 0;
832 		}
833 
834 		for (i = 1; i < attr->kdf_count; i++) {
835 			if (attr->kdf[i] == data->kdf)
836 				break;
837 		}
838 		if (i == attr->kdf_count &&
839 		    attr->kdf_count < EAP_AKA_PRIME_KDF_MAX) {
840 			wpa_printf(MSG_WARNING, "EAP-AKA': The server did not "
841 				   "duplicate the selected KDF");
842 			return 0;
843 		}
844 
845 		/* TODO: should check that the list is identical to the one
846 		 * used in the previous Challenge message apart from the added
847 		 * entry in the beginning. */
848 	}
849 
850 	for (i = data->kdf ? 1 : 0; i < attr->kdf_count; i++) {
851 		for (j = i + 1; j < attr->kdf_count; j++) {
852 			if (attr->kdf[i] == attr->kdf[j]) {
853 				wpa_printf(MSG_WARNING, "EAP-AKA': The server "
854 					   "included a duplicated KDF");
855 				return 0;
856 			}
857 		}
858 	}
859 
860 	return 1;
861 }
862 #endif /* EAP_AKA_PRIME */
863 
864 
865 static struct wpabuf * eap_aka_process_challenge(struct eap_sm *sm,
866 						 struct eap_aka_data *data,
867 						 u8 id,
868 						 const struct wpabuf *reqData,
869 						 struct eap_sim_attrs *attr)
870 {
871 	const u8 *identity;
872 	size_t identity_len;
873 	int res;
874 	struct eap_sim_attrs eattr;
875 
876 	wpa_printf(MSG_DEBUG, "EAP-AKA: subtype Challenge");
877 
878 	if (attr->checkcode &&
879 	    eap_aka_verify_checkcode(data, attr->checkcode,
880 				     attr->checkcode_len)) {
881 		wpa_printf(MSG_WARNING, "EAP-AKA: Invalid AT_CHECKCODE in the "
882 			   "message");
883 		return eap_aka_client_error(data, id,
884 					    EAP_AKA_UNABLE_TO_PROCESS_PACKET);
885 	}
886 
887 #ifdef EAP_AKA_PRIME
888 	if (data->eap_method == EAP_TYPE_AKA_PRIME) {
889 		if (!attr->kdf_input || attr->kdf_input_len == 0) {
890 			wpa_printf(MSG_WARNING, "EAP-AKA': Challenge message "
891 				   "did not include non-empty AT_KDF_INPUT");
892 			/* Fail authentication as if AUTN had been incorrect */
893 			return eap_aka_authentication_reject(data, id);
894 		}
895 		os_free(data->network_name);
896 		data->network_name = os_malloc(attr->kdf_input_len);
897 		if (data->network_name == NULL) {
898 			wpa_printf(MSG_WARNING, "EAP-AKA': No memory for "
899 				   "storing Network Name");
900 			return eap_aka_authentication_reject(data, id);
901 		}
902 		os_memcpy(data->network_name, attr->kdf_input,
903 			  attr->kdf_input_len);
904 		data->network_name_len = attr->kdf_input_len;
905 		wpa_hexdump_ascii(MSG_DEBUG, "EAP-AKA': Network Name "
906 				  "(AT_KDF_INPUT)",
907 				  data->network_name, data->network_name_len);
908 		/* TODO: check Network Name per 3GPP.33.402 */
909 
910 		if (!eap_aka_prime_kdf_valid(data, attr))
911 			return eap_aka_authentication_reject(data, id);
912 
913 		if (attr->kdf[0] != EAP_AKA_PRIME_KDF)
914 			return eap_aka_prime_kdf_neg(data, id, attr);
915 
916 		data->kdf = EAP_AKA_PRIME_KDF;
917 		wpa_printf(MSG_DEBUG, "EAP-AKA': KDF %d selected", data->kdf);
918 	}
919 
920 	if (data->eap_method == EAP_TYPE_AKA && attr->bidding) {
921 		u16 flags = WPA_GET_BE16(attr->bidding);
922 		if ((flags & EAP_AKA_BIDDING_FLAG_D) &&
923 		    eap_allowed_method(sm, EAP_VENDOR_IETF,
924 				       EAP_TYPE_AKA_PRIME)) {
925 			wpa_printf(MSG_WARNING, "EAP-AKA: Bidding down from "
926 				   "AKA' to AKA detected");
927 			/* Fail authentication as if AUTN had been incorrect */
928 			return eap_aka_authentication_reject(data, id);
929 		}
930 	}
931 #endif /* EAP_AKA_PRIME */
932 
933 	data->reauth = 0;
934 	if (!attr->mac || !attr->rand || !attr->autn) {
935 		wpa_printf(MSG_WARNING, "EAP-AKA: Challenge message "
936 			   "did not include%s%s%s",
937 			   !attr->mac ? " AT_MAC" : "",
938 			   !attr->rand ? " AT_RAND" : "",
939 			   !attr->autn ? " AT_AUTN" : "");
940 		return eap_aka_client_error(data, id,
941 					    EAP_AKA_UNABLE_TO_PROCESS_PACKET);
942 	}
943 	os_memcpy(data->rand, attr->rand, EAP_AKA_RAND_LEN);
944 	os_memcpy(data->autn, attr->autn, EAP_AKA_AUTN_LEN);
945 
946 	res = eap_aka_umts_auth(sm, data);
947 	if (res == -1) {
948 		wpa_printf(MSG_WARNING, "EAP-AKA: UMTS authentication "
949 			   "failed (AUTN)");
950 		return eap_aka_authentication_reject(data, id);
951 	} else if (res == -2) {
952 		wpa_printf(MSG_WARNING, "EAP-AKA: UMTS authentication "
953 			   "failed (AUTN seq# -> AUTS)");
954 		return eap_aka_synchronization_failure(data, id);
955 	} else if (res > 0) {
956 		wpa_printf(MSG_DEBUG, "EAP-AKA: Wait for external USIM processing");
957 		return NULL;
958 	} else if (res) {
959 		wpa_printf(MSG_WARNING, "EAP-AKA: UMTS authentication failed");
960 		return eap_aka_client_error(data, id,
961 					    EAP_AKA_UNABLE_TO_PROCESS_PACKET);
962 	}
963 #ifdef EAP_AKA_PRIME
964 	if (data->eap_method == EAP_TYPE_AKA_PRIME) {
965 		/* Note: AUTN = (SQN ^ AK) || AMF || MAC which gives us the
966 		 * needed 6-octet SQN ^ AK for CK',IK' derivation */
967 		u16 amf = WPA_GET_BE16(data->autn + 6);
968 		if (!(amf & 0x8000)) {
969 			wpa_printf(MSG_WARNING, "EAP-AKA': AMF separation bit "
970 				   "not set (AMF=0x%4x)", amf);
971 			return eap_aka_authentication_reject(data, id);
972 		}
973 		eap_aka_prime_derive_ck_ik_prime(data->ck, data->ik,
974 						 data->autn,
975 						 data->network_name,
976 						 data->network_name_len);
977 	}
978 #endif /* EAP_AKA_PRIME */
979 	if (data->last_eap_identity) {
980 		identity = data->last_eap_identity;
981 		identity_len = data->last_eap_identity_len;
982 	} else if (data->pseudonym) {
983 		identity = data->pseudonym;
984 		identity_len = data->pseudonym_len;
985 	} else
986 		identity = eap_get_config_identity(sm, &identity_len);
987 	wpa_hexdump_ascii(MSG_DEBUG, "EAP-AKA: Selected identity for MK "
988 			  "derivation", identity, identity_len);
989 	if (data->eap_method == EAP_TYPE_AKA_PRIME) {
990 		eap_aka_prime_derive_keys(identity, identity_len, data->ik,
991 					  data->ck, data->k_encr, data->k_aut,
992 					  data->k_re, data->msk, data->emsk);
993 	} else {
994 		eap_aka_derive_mk(identity, identity_len, data->ik, data->ck,
995 				  data->mk);
996 		eap_sim_derive_keys(data->mk, data->k_encr, data->k_aut,
997 				    data->msk, data->emsk);
998 	}
999 	if (eap_aka_verify_mac(data, reqData, attr->mac, (u8 *) "", 0)) {
1000 		wpa_printf(MSG_WARNING, "EAP-AKA: Challenge message "
1001 			   "used invalid AT_MAC");
1002 		return eap_aka_client_error(data, id,
1003 					    EAP_AKA_UNABLE_TO_PROCESS_PACKET);
1004 	}
1005 
1006 	/* Old reauthentication identity must not be used anymore. In
1007 	 * other words, if no new identities are received, full
1008 	 * authentication will be used on next reauthentication (using
1009 	 * pseudonym identity or permanent identity). */
1010 	eap_aka_clear_identities(sm, data, CLEAR_REAUTH_ID | CLEAR_EAP_ID);
1011 
1012 	if (attr->encr_data) {
1013 		u8 *decrypted;
1014 		decrypted = eap_sim_parse_encr(data->k_encr, attr->encr_data,
1015 					       attr->encr_data_len, attr->iv,
1016 					       &eattr, 0);
1017 		if (decrypted == NULL) {
1018 			return eap_aka_client_error(
1019 				data, id, EAP_AKA_UNABLE_TO_PROCESS_PACKET);
1020 		}
1021 		eap_aka_learn_ids(sm, data, &eattr);
1022 		os_free(decrypted);
1023 	}
1024 
1025 	if (data->result_ind && attr->result_ind)
1026 		data->use_result_ind = 1;
1027 
1028 	if (data->state != FAILURE && data->state != RESULT_FAILURE) {
1029 		eap_aka_state(data, data->use_result_ind ?
1030 			      RESULT_SUCCESS : SUCCESS);
1031 	}
1032 
1033 	data->num_id_req = 0;
1034 	data->num_notification = 0;
1035 	/* RFC 4187 specifies that counter is initialized to one after
1036 	 * fullauth, but initializing it to zero makes it easier to implement
1037 	 * reauth verification. */
1038 	data->counter = 0;
1039 	return eap_aka_response_challenge(data, id);
1040 }
1041 
1042 
1043 static int eap_aka_process_notification_reauth(struct eap_aka_data *data,
1044 					       struct eap_sim_attrs *attr)
1045 {
1046 	struct eap_sim_attrs eattr;
1047 	u8 *decrypted;
1048 
1049 	if (attr->encr_data == NULL || attr->iv == NULL) {
1050 		wpa_printf(MSG_WARNING, "EAP-AKA: Notification message after "
1051 			   "reauth did not include encrypted data");
1052 		return -1;
1053 	}
1054 
1055 	decrypted = eap_sim_parse_encr(data->k_encr, attr->encr_data,
1056 				       attr->encr_data_len, attr->iv, &eattr,
1057 				       0);
1058 	if (decrypted == NULL) {
1059 		wpa_printf(MSG_WARNING, "EAP-AKA: Failed to parse encrypted "
1060 			   "data from notification message");
1061 		return -1;
1062 	}
1063 
1064 	if (eattr.counter < 0 || (size_t) eattr.counter != data->counter) {
1065 		wpa_printf(MSG_WARNING, "EAP-AKA: Counter in notification "
1066 			   "message does not match with counter in reauth "
1067 			   "message");
1068 		os_free(decrypted);
1069 		return -1;
1070 	}
1071 
1072 	os_free(decrypted);
1073 	return 0;
1074 }
1075 
1076 
1077 static int eap_aka_process_notification_auth(struct eap_aka_data *data,
1078 					     const struct wpabuf *reqData,
1079 					     struct eap_sim_attrs *attr)
1080 {
1081 	if (attr->mac == NULL) {
1082 		wpa_printf(MSG_INFO, "EAP-AKA: no AT_MAC in after_auth "
1083 			   "Notification message");
1084 		return -1;
1085 	}
1086 
1087 	if (eap_aka_verify_mac(data, reqData, attr->mac, (u8 *) "", 0)) {
1088 		wpa_printf(MSG_WARNING, "EAP-AKA: Notification message "
1089 			   "used invalid AT_MAC");
1090 		return -1;
1091 	}
1092 
1093 	if (data->reauth &&
1094 	    eap_aka_process_notification_reauth(data, attr)) {
1095 		wpa_printf(MSG_WARNING, "EAP-AKA: Invalid notification "
1096 			   "message after reauth");
1097 		return -1;
1098 	}
1099 
1100 	return 0;
1101 }
1102 
1103 
1104 static struct wpabuf * eap_aka_process_notification(
1105 	struct eap_sm *sm, struct eap_aka_data *data, u8 id,
1106 	const struct wpabuf *reqData, struct eap_sim_attrs *attr)
1107 {
1108 	wpa_printf(MSG_DEBUG, "EAP-AKA: subtype Notification");
1109 	if (data->num_notification > 0) {
1110 		wpa_printf(MSG_INFO, "EAP-AKA: too many notification "
1111 			   "rounds (only one allowed)");
1112 		return eap_aka_client_error(data, id,
1113 					    EAP_AKA_UNABLE_TO_PROCESS_PACKET);
1114 	}
1115 	data->num_notification++;
1116 	if (attr->notification == -1) {
1117 		wpa_printf(MSG_INFO, "EAP-AKA: no AT_NOTIFICATION in "
1118 			   "Notification message");
1119 		return eap_aka_client_error(data, id,
1120 					    EAP_AKA_UNABLE_TO_PROCESS_PACKET);
1121 	}
1122 
1123 	if ((attr->notification & 0x4000) == 0 &&
1124 	    eap_aka_process_notification_auth(data, reqData, attr)) {
1125 		return eap_aka_client_error(data, id,
1126 					    EAP_AKA_UNABLE_TO_PROCESS_PACKET);
1127 	}
1128 
1129 	eap_sim_report_notification(sm->msg_ctx, attr->notification, 1);
1130 	if (attr->notification >= 0 && attr->notification < 32768) {
1131 		eap_aka_state(data, FAILURE);
1132 	} else if (attr->notification == EAP_SIM_SUCCESS &&
1133 		   data->state == RESULT_SUCCESS)
1134 		eap_aka_state(data, SUCCESS);
1135 	return eap_aka_response_notification(data, id, attr->notification);
1136 }
1137 
1138 
1139 static struct wpabuf * eap_aka_process_reauthentication(
1140 	struct eap_sm *sm, struct eap_aka_data *data, u8 id,
1141 	const struct wpabuf *reqData, struct eap_sim_attrs *attr)
1142 {
1143 	struct eap_sim_attrs eattr;
1144 	u8 *decrypted;
1145 
1146 	wpa_printf(MSG_DEBUG, "EAP-AKA: subtype Reauthentication");
1147 
1148 	if (attr->checkcode &&
1149 	    eap_aka_verify_checkcode(data, attr->checkcode,
1150 				     attr->checkcode_len)) {
1151 		wpa_printf(MSG_WARNING, "EAP-AKA: Invalid AT_CHECKCODE in the "
1152 			   "message");
1153 		return eap_aka_client_error(data, id,
1154 					    EAP_AKA_UNABLE_TO_PROCESS_PACKET);
1155 	}
1156 
1157 	if (data->reauth_id == NULL) {
1158 		wpa_printf(MSG_WARNING, "EAP-AKA: Server is trying "
1159 			   "reauthentication, but no reauth_id available");
1160 		return eap_aka_client_error(data, id,
1161 					    EAP_AKA_UNABLE_TO_PROCESS_PACKET);
1162 	}
1163 
1164 	data->reauth = 1;
1165 	if (eap_aka_verify_mac(data, reqData, attr->mac, (u8 *) "", 0)) {
1166 		wpa_printf(MSG_WARNING, "EAP-AKA: Reauthentication "
1167 			   "did not have valid AT_MAC");
1168 		return eap_aka_client_error(data, id,
1169 					    EAP_AKA_UNABLE_TO_PROCESS_PACKET);
1170 	}
1171 
1172 	if (attr->encr_data == NULL || attr->iv == NULL) {
1173 		wpa_printf(MSG_WARNING, "EAP-AKA: Reauthentication "
1174 			   "message did not include encrypted data");
1175 		return eap_aka_client_error(data, id,
1176 					    EAP_AKA_UNABLE_TO_PROCESS_PACKET);
1177 	}
1178 
1179 	decrypted = eap_sim_parse_encr(data->k_encr, attr->encr_data,
1180 				       attr->encr_data_len, attr->iv, &eattr,
1181 				       0);
1182 	if (decrypted == NULL) {
1183 		wpa_printf(MSG_WARNING, "EAP-AKA: Failed to parse encrypted "
1184 			   "data from reauthentication message");
1185 		return eap_aka_client_error(data, id,
1186 					    EAP_AKA_UNABLE_TO_PROCESS_PACKET);
1187 	}
1188 
1189 	if (eattr.nonce_s == NULL || eattr.counter < 0) {
1190 		wpa_printf(MSG_INFO, "EAP-AKA: (encr) No%s%s in reauth packet",
1191 			   !eattr.nonce_s ? " AT_NONCE_S" : "",
1192 			   eattr.counter < 0 ? " AT_COUNTER" : "");
1193 		os_free(decrypted);
1194 		return eap_aka_client_error(data, id,
1195 					    EAP_AKA_UNABLE_TO_PROCESS_PACKET);
1196 	}
1197 
1198 	if (eattr.counter < 0 || (size_t) eattr.counter <= data->counter) {
1199 		struct wpabuf *res;
1200 		wpa_printf(MSG_INFO, "EAP-AKA: (encr) Invalid counter "
1201 			   "(%d <= %d)", eattr.counter, data->counter);
1202 		data->counter_too_small = eattr.counter;
1203 
1204 		/* Reply using Re-auth w/ AT_COUNTER_TOO_SMALL. The current
1205 		 * reauth_id must not be used to start a new reauthentication.
1206 		 * However, since it was used in the last EAP-Response-Identity
1207 		 * packet, it has to saved for the following fullauth to be
1208 		 * used in MK derivation. */
1209 		os_free(data->last_eap_identity);
1210 		data->last_eap_identity = data->reauth_id;
1211 		data->last_eap_identity_len = data->reauth_id_len;
1212 		data->reauth_id = NULL;
1213 		data->reauth_id_len = 0;
1214 
1215 		res = eap_aka_response_reauth(data, id, 1, eattr.nonce_s);
1216 		os_free(decrypted);
1217 
1218 		return res;
1219 	}
1220 	data->counter = eattr.counter;
1221 
1222 	os_memcpy(data->nonce_s, eattr.nonce_s, EAP_SIM_NONCE_S_LEN);
1223 	wpa_hexdump(MSG_DEBUG, "EAP-AKA: (encr) AT_NONCE_S",
1224 		    data->nonce_s, EAP_SIM_NONCE_S_LEN);
1225 
1226 	if (data->eap_method == EAP_TYPE_AKA_PRIME) {
1227 		eap_aka_prime_derive_keys_reauth(data->k_re, data->counter,
1228 						 data->reauth_id,
1229 						 data->reauth_id_len,
1230 						 data->nonce_s,
1231 						 data->msk, data->emsk);
1232 	} else {
1233 		eap_sim_derive_keys_reauth(data->counter, data->reauth_id,
1234 					   data->reauth_id_len,
1235 					   data->nonce_s, data->mk,
1236 					   data->msk, data->emsk);
1237 	}
1238 	eap_aka_clear_identities(sm, data, CLEAR_REAUTH_ID | CLEAR_EAP_ID);
1239 	eap_aka_learn_ids(sm, data, &eattr);
1240 
1241 	if (data->result_ind && attr->result_ind)
1242 		data->use_result_ind = 1;
1243 
1244 	if (data->state != FAILURE && data->state != RESULT_FAILURE) {
1245 		eap_aka_state(data, data->use_result_ind ?
1246 			      RESULT_SUCCESS : SUCCESS);
1247 	}
1248 
1249 	data->num_id_req = 0;
1250 	data->num_notification = 0;
1251 	if (data->counter > EAP_AKA_MAX_FAST_REAUTHS) {
1252 		wpa_printf(MSG_DEBUG, "EAP-AKA: Maximum number of "
1253 			   "fast reauths performed - force fullauth");
1254 		eap_aka_clear_identities(sm, data,
1255 					 CLEAR_REAUTH_ID | CLEAR_EAP_ID);
1256 	}
1257 	os_free(decrypted);
1258 	return eap_aka_response_reauth(data, id, 0, data->nonce_s);
1259 }
1260 
1261 
1262 static struct wpabuf * eap_aka_process(struct eap_sm *sm, void *priv,
1263 				       struct eap_method_ret *ret,
1264 				       const struct wpabuf *reqData)
1265 {
1266 	struct eap_aka_data *data = priv;
1267 	const struct eap_hdr *req;
1268 	u8 subtype, id;
1269 	struct wpabuf *res;
1270 	const u8 *pos;
1271 	struct eap_sim_attrs attr;
1272 	size_t len;
1273 
1274 	wpa_hexdump_buf(MSG_DEBUG, "EAP-AKA: EAP data", reqData);
1275 	if (eap_get_config_identity(sm, &len) == NULL) {
1276 		wpa_printf(MSG_INFO, "EAP-AKA: Identity not configured");
1277 		eap_sm_request_identity(sm);
1278 		ret->ignore = TRUE;
1279 		return NULL;
1280 	}
1281 
1282 	pos = eap_hdr_validate(EAP_VENDOR_IETF, data->eap_method, reqData,
1283 			       &len);
1284 	if (pos == NULL || len < 1) {
1285 		ret->ignore = TRUE;
1286 		return NULL;
1287 	}
1288 	req = wpabuf_head(reqData);
1289 	id = req->identifier;
1290 	len = be_to_host16(req->length);
1291 
1292 	ret->ignore = FALSE;
1293 	ret->methodState = METHOD_MAY_CONT;
1294 	ret->decision = DECISION_FAIL;
1295 	ret->allowNotifications = TRUE;
1296 
1297 	subtype = *pos++;
1298 	wpa_printf(MSG_DEBUG, "EAP-AKA: Subtype=%d", subtype);
1299 	pos += 2; /* Reserved */
1300 
1301 	if (eap_sim_parse_attr(pos, wpabuf_head_u8(reqData) + len, &attr,
1302 			       data->eap_method == EAP_TYPE_AKA_PRIME ? 2 : 1,
1303 			       0)) {
1304 		res = eap_aka_client_error(data, id,
1305 					   EAP_AKA_UNABLE_TO_PROCESS_PACKET);
1306 		goto done;
1307 	}
1308 
1309 	switch (subtype) {
1310 	case EAP_AKA_SUBTYPE_IDENTITY:
1311 		res = eap_aka_process_identity(sm, data, id, reqData, &attr);
1312 		break;
1313 	case EAP_AKA_SUBTYPE_CHALLENGE:
1314 		res = eap_aka_process_challenge(sm, data, id, reqData, &attr);
1315 		break;
1316 	case EAP_AKA_SUBTYPE_NOTIFICATION:
1317 		res = eap_aka_process_notification(sm, data, id, reqData,
1318 						   &attr);
1319 		break;
1320 	case EAP_AKA_SUBTYPE_REAUTHENTICATION:
1321 		res = eap_aka_process_reauthentication(sm, data, id, reqData,
1322 						       &attr);
1323 		break;
1324 	case EAP_AKA_SUBTYPE_CLIENT_ERROR:
1325 		wpa_printf(MSG_DEBUG, "EAP-AKA: subtype Client-Error");
1326 		res = eap_aka_client_error(data, id,
1327 					   EAP_AKA_UNABLE_TO_PROCESS_PACKET);
1328 		break;
1329 	default:
1330 		wpa_printf(MSG_DEBUG, "EAP-AKA: Unknown subtype=%d", subtype);
1331 		res = eap_aka_client_error(data, id,
1332 					   EAP_AKA_UNABLE_TO_PROCESS_PACKET);
1333 		break;
1334 	}
1335 
1336 done:
1337 	if (data->state == FAILURE) {
1338 		ret->decision = DECISION_FAIL;
1339 		ret->methodState = METHOD_DONE;
1340 	} else if (data->state == SUCCESS) {
1341 		ret->decision = data->use_result_ind ?
1342 			DECISION_UNCOND_SUCC : DECISION_COND_SUCC;
1343 		/*
1344 		 * It is possible for the server to reply with AKA
1345 		 * Notification, so we must allow the method to continue and
1346 		 * not only accept EAP-Success at this point.
1347 		 */
1348 		ret->methodState = data->use_result_ind ?
1349 			METHOD_DONE : METHOD_MAY_CONT;
1350 	} else if (data->state == RESULT_FAILURE)
1351 		ret->methodState = METHOD_CONT;
1352 	else if (data->state == RESULT_SUCCESS)
1353 		ret->methodState = METHOD_CONT;
1354 
1355 	if (ret->methodState == METHOD_DONE) {
1356 		ret->allowNotifications = FALSE;
1357 	}
1358 
1359 	return res;
1360 }
1361 
1362 
1363 static Boolean eap_aka_has_reauth_data(struct eap_sm *sm, void *priv)
1364 {
1365 	struct eap_aka_data *data = priv;
1366 	return data->pseudonym || data->reauth_id;
1367 }
1368 
1369 
1370 static void eap_aka_deinit_for_reauth(struct eap_sm *sm, void *priv)
1371 {
1372 	struct eap_aka_data *data = priv;
1373 	eap_aka_clear_identities(sm, data, CLEAR_EAP_ID);
1374 	data->prev_id = -1;
1375 	wpabuf_free(data->id_msgs);
1376 	data->id_msgs = NULL;
1377 	data->use_result_ind = 0;
1378 	data->kdf_negotiation = 0;
1379 }
1380 
1381 
1382 static void * eap_aka_init_for_reauth(struct eap_sm *sm, void *priv)
1383 {
1384 	struct eap_aka_data *data = priv;
1385 	data->num_id_req = 0;
1386 	data->num_notification = 0;
1387 	eap_aka_state(data, CONTINUE);
1388 	return priv;
1389 }
1390 
1391 
1392 static const u8 * eap_aka_get_identity(struct eap_sm *sm, void *priv,
1393 				       size_t *len)
1394 {
1395 	struct eap_aka_data *data = priv;
1396 
1397 	if (data->reauth_id) {
1398 		*len = data->reauth_id_len;
1399 		return data->reauth_id;
1400 	}
1401 
1402 	if (data->pseudonym) {
1403 		*len = data->pseudonym_len;
1404 		return data->pseudonym;
1405 	}
1406 
1407 	return NULL;
1408 }
1409 
1410 
1411 static Boolean eap_aka_isKeyAvailable(struct eap_sm *sm, void *priv)
1412 {
1413 	struct eap_aka_data *data = priv;
1414 	return data->state == SUCCESS;
1415 }
1416 
1417 
1418 static u8 * eap_aka_getKey(struct eap_sm *sm, void *priv, size_t *len)
1419 {
1420 	struct eap_aka_data *data = priv;
1421 	u8 *key;
1422 
1423 	if (data->state != SUCCESS)
1424 		return NULL;
1425 
1426 	key = os_malloc(EAP_SIM_KEYING_DATA_LEN);
1427 	if (key == NULL)
1428 		return NULL;
1429 
1430 	*len = EAP_SIM_KEYING_DATA_LEN;
1431 	os_memcpy(key, data->msk, EAP_SIM_KEYING_DATA_LEN);
1432 
1433 	return key;
1434 }
1435 
1436 
1437 static u8 * eap_aka_get_session_id(struct eap_sm *sm, void *priv, size_t *len)
1438 {
1439 	struct eap_aka_data *data = priv;
1440 	u8 *id;
1441 
1442 	if (data->state != SUCCESS)
1443 		return NULL;
1444 
1445 	*len = 1 + EAP_AKA_RAND_LEN + EAP_AKA_AUTN_LEN;
1446 	id = os_malloc(*len);
1447 	if (id == NULL)
1448 		return NULL;
1449 
1450 	id[0] = data->eap_method;
1451 	os_memcpy(id + 1, data->rand, EAP_AKA_RAND_LEN);
1452 	os_memcpy(id + 1 + EAP_AKA_RAND_LEN, data->autn, EAP_AKA_AUTN_LEN);
1453 	wpa_hexdump(MSG_DEBUG, "EAP-AKA: Derived Session-Id", id, *len);
1454 
1455 	return id;
1456 }
1457 
1458 
1459 static u8 * eap_aka_get_emsk(struct eap_sm *sm, void *priv, size_t *len)
1460 {
1461 	struct eap_aka_data *data = priv;
1462 	u8 *key;
1463 
1464 	if (data->state != SUCCESS)
1465 		return NULL;
1466 
1467 	key = os_malloc(EAP_EMSK_LEN);
1468 	if (key == NULL)
1469 		return NULL;
1470 
1471 	*len = EAP_EMSK_LEN;
1472 	os_memcpy(key, data->emsk, EAP_EMSK_LEN);
1473 
1474 	return key;
1475 }
1476 
1477 
1478 int eap_peer_aka_register(void)
1479 {
1480 	struct eap_method *eap;
1481 	int ret;
1482 
1483 	eap = eap_peer_method_alloc(EAP_PEER_METHOD_INTERFACE_VERSION,
1484 				    EAP_VENDOR_IETF, EAP_TYPE_AKA, "AKA");
1485 	if (eap == NULL)
1486 		return -1;
1487 
1488 	eap->init = eap_aka_init;
1489 	eap->deinit = eap_aka_deinit;
1490 	eap->process = eap_aka_process;
1491 	eap->isKeyAvailable = eap_aka_isKeyAvailable;
1492 	eap->getKey = eap_aka_getKey;
1493 	eap->getSessionId = eap_aka_get_session_id;
1494 	eap->has_reauth_data = eap_aka_has_reauth_data;
1495 	eap->deinit_for_reauth = eap_aka_deinit_for_reauth;
1496 	eap->init_for_reauth = eap_aka_init_for_reauth;
1497 	eap->get_identity = eap_aka_get_identity;
1498 	eap->get_emsk = eap_aka_get_emsk;
1499 
1500 	ret = eap_peer_method_register(eap);
1501 	if (ret)
1502 		eap_peer_method_free(eap);
1503 	return ret;
1504 }
1505 
1506 
1507 #ifdef EAP_AKA_PRIME
1508 int eap_peer_aka_prime_register(void)
1509 {
1510 	struct eap_method *eap;
1511 	int ret;
1512 
1513 	eap = eap_peer_method_alloc(EAP_PEER_METHOD_INTERFACE_VERSION,
1514 				    EAP_VENDOR_IETF, EAP_TYPE_AKA_PRIME,
1515 				    "AKA'");
1516 	if (eap == NULL)
1517 		return -1;
1518 
1519 	eap->init = eap_aka_prime_init;
1520 	eap->deinit = eap_aka_deinit;
1521 	eap->process = eap_aka_process;
1522 	eap->isKeyAvailable = eap_aka_isKeyAvailable;
1523 	eap->getKey = eap_aka_getKey;
1524 	eap->getSessionId = eap_aka_get_session_id;
1525 	eap->has_reauth_data = eap_aka_has_reauth_data;
1526 	eap->deinit_for_reauth = eap_aka_deinit_for_reauth;
1527 	eap->init_for_reauth = eap_aka_init_for_reauth;
1528 	eap->get_identity = eap_aka_get_identity;
1529 	eap->get_emsk = eap_aka_get_emsk;
1530 
1531 	ret = eap_peer_method_register(eap);
1532 	if (ret)
1533 		eap_peer_method_free(eap);
1534 
1535 	return ret;
1536 }
1537 #endif /* EAP_AKA_PRIME */
1538