Kernel/iwd
3
0
Fork 0
mirror of https://git.kernel.org/pub/scm/network/wireless/iwd.git synced 2024-11-22 06:29:23 +01:00
Code Releases Activity

unit: updated unit test with eapol deauth removal

Browse Source
This commit is contained in:
James Prestwood 2018-06-27 14:45:00 -07:00 committed by Denis Kenzior
parent 92f1ceb3ce
commit 5d31fc25d7
2 changed files with 36 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
unit/test-eapol.c
View File

@ -2857,11 +2857,16 @@ static void eapol_sm_test_tls_test_disconnected(enum l_tls_alert_desc reason,
assert(false); assert(false);
} }
static void verify_deauthenticate(uint32_t ifindex, const uint8_t *aa, static void verify_handshake(struct handshake_state *hs,
const uint8_t *spa, enum handshake_event event, void *event_data, void *user_data)
uint16_t reason_code, void *user_data)
{ {
assert(false); switch (event) {
case HANDSHAKE_EVENT_FAILED:
assert(false);
break;
default:
break;
}
} }
static void eapol_sm_test_tls(struct eapol_8021x_tls_test_state *s, static void eapol_sm_test_tls(struct eapol_8021x_tls_test_state *s,
@ -2892,7 +2897,6 @@ static void eapol_sm_test_tls(struct eapol_8021x_tls_test_state *s,
eap_init(0); eap_init(0);
eapol_init(); eapol_init();
__handshake_set_get_nonce_func(test_nonce); __handshake_set_get_nonce_func(test_nonce);
__eapol_set_deauthenticate_func(verify_deauthenticate);
hs = test_handshake_state_new(1); hs = test_handshake_state_new(1);
sm = eapol_sm_new(hs); sm = eapol_sm_new(hs);
@ -2900,6 +2904,7 @@ static void eapol_sm_test_tls(struct eapol_8021x_tls_test_state *s,
handshake_state_set_authenticator_address(hs, ap_address); handshake_state_set_authenticator_address(hs, ap_address);
handshake_state_set_supplicant_address(hs, sta_address); handshake_state_set_supplicant_address(hs, sta_address);
handshake_state_set_event_func(hs, verify_handshake, NULL);
__eapol_set_tx_user_data(s); __eapol_set_tx_user_data(s);
r = handshake_state_set_own_wpa(hs, r = handshake_state_set_own_wpa(hs,
@ -3190,15 +3195,18 @@ static void eapol_sm_test_eap_ttls_md5(const void *data)
eapol_sm_test_tls(&s.tls, eapol_8021x_config); eapol_sm_test_tls(&s.tls, eapol_8021x_config);
} }
static bool eap_nak_verify_deauthenticate_called; static bool eap_nak_verify_handshake_failed;
static void eap_nak_verify_deauthenticate(uint32_t ifindex, const uint8_t *aa, static void verify_handshake_event(struct handshake_state *hs,
const uint8_t *spa, enum handshake_event event, void *event_data, void *user_data)
uint16_t reason_code,
void *user_data)
{ {
assert(!eap_nak_verify_deauthenticate_called); switch (event) {
eap_nak_verify_deauthenticate_called = true; case HANDSHAKE_EVENT_FAILED:
eap_nak_verify_handshake_failed = true;
break;
default:
break;
}
} }
static const uint8_t eap_ttls_start_req[] = { static const uint8_t eap_ttls_start_req[] = {
@ -3258,7 +3266,6 @@ static void eapol_sm_test_eap_nak(const void *data)
eap_init(0); eap_init(0);
eapol_init(); eapol_init();
__handshake_set_get_nonce_func(test_nonce); __handshake_set_get_nonce_func(test_nonce);
__eapol_set_deauthenticate_func(verify_deauthenticate);
hs = test_handshake_state_new(1); hs = test_handshake_state_new(1);
sm = eapol_sm_new(hs); sm = eapol_sm_new(hs);
@ -3266,6 +3273,7 @@ static void eapol_sm_test_eap_nak(const void *data)
handshake_state_set_authenticator_address(hs, ap_address); handshake_state_set_authenticator_address(hs, ap_address);
handshake_state_set_supplicant_address(hs, sta_address); handshake_state_set_supplicant_address(hs, sta_address);
handshake_state_set_event_func(hs, verify_handshake_event, NULL);
__eapol_set_tx_user_data(&s); __eapol_set_tx_user_data(&s);
r = handshake_state_set_own_wpa(hs, r = handshake_state_set_own_wpa(hs,
@ -3294,11 +3302,10 @@ static void eapol_sm_test_eap_nak(const void *data)
sizeof(eap_ttls_start_req), false); sizeof(eap_ttls_start_req), false);
assert(!s.pending_req); assert(!s.pending_req);
eap_nak_verify_deauthenticate_called = false; eap_nak_verify_handshake_failed = false;
__eapol_set_deauthenticate_func(eap_nak_verify_deauthenticate);
__eapol_rx_packet(1, ap_address, ETH_P_PAE, eap_failure, __eapol_rx_packet(1, ap_address, ETH_P_PAE, eap_failure,
sizeof(eap_failure), false); sizeof(eap_failure), false);
assert(eap_nak_verify_deauthenticate_called); assert(eap_nak_verify_handshake_failed);
handshake_state_free(hs); handshake_state_free(hs);
eapol_exit(); eapol_exit();

20
unit/test-wsc.c
View File

@ -1931,14 +1931,20 @@ struct verify_data {
verify.expected = e; \ verify.expected = e; \
verify.expected_len = sizeof(e) verify.expected_len = sizeof(e)
static void verify_deauthenticate(uint32_t ifindex, const uint8_t *aa, static void verify_handshake_event(struct handshake_state *hs,
const uint8_t *spa, uint16_t reason_code, enum handshake_event event, void *event_data, void *user_data)
void *user_data)
{ {
struct verify_data *data = user_data; struct verify_data *data = user_data;
assert(reason_code == MMPDU_REASON_CODE_IEEE8021X_FAILED); switch (event) {
data->eapol_failed = true; case HANDSHAKE_EVENT_FAILED:
assert(l_get_u16(event_data) ==
MMPDU_REASON_CODE_IEEE8021X_FAILED);
data->eapol_failed = true;
break;
default:
break;
}
} }
static int verify_8021x(uint32_t ifindex, static int verify_8021x(uint32_t ifindex,
@ -1993,10 +1999,10 @@ static void wsc_test_pbc_handshake(const void *data)
handshake_state_set_authenticator_address(hs, ap_address); handshake_state_set_authenticator_address(hs, ap_address);
handshake_state_set_supplicant_address(hs, sta_address); handshake_state_set_supplicant_address(hs, sta_address);
handshake_state_set_event_func(hs, verify_handshake_event, &verify);
__eapol_set_tx_packet_func(verify_8021x); __eapol_set_tx_packet_func(verify_8021x);
__eapol_set_tx_user_data(&verify); __eapol_set_tx_user_data(&verify);
__eapol_set_deauthenticate_func(verify_deauthenticate);
eapol_sm_set_user_data(sm, &verify); eapol_sm_set_user_data(sm, &verify);
eapol_sm_set_event_func(sm, verify_credential); eapol_sm_set_event_func(sm, verify_credential);
@ -2101,10 +2107,10 @@ static void wsc_test_retransmission_no_fragmentation(const void *data)
handshake_state_set_authenticator_address(hs, ap_address); handshake_state_set_authenticator_address(hs, ap_address);
handshake_state_set_supplicant_address(hs, sta_address); handshake_state_set_supplicant_address(hs, sta_address);
handshake_state_set_event_func(hs, verify_handshake_event, &verify);
__eapol_set_tx_packet_func(verify_8021x); __eapol_set_tx_packet_func(verify_8021x);
__eapol_set_tx_user_data(&verify); __eapol_set_tx_user_data(&verify);
__eapol_set_deauthenticate_func(verify_deauthenticate);
eapol_sm_set_user_data(sm, &verify); eapol_sm_set_user_data(sm, &verify);
eapol_sm_set_event_func(sm, verify_credential); eapol_sm_set_event_func(sm, verify_credential);