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