Kernel/iwd
3
0
Fork 0
mirror of https://git.kernel.org/pub/scm/network/wireless/iwd.git synced 2024-12-02 16:39:25 +01:00
Code Releases Activity

unit: update use of handshake_state with ref/unref

Browse Source
This commit is contained in:
James Prestwood 2024-11-22 07:15:38 -08:00 committed by Denis Kenzior
parent b9c3feb198
commit f58cad8cd9
3 changed files with 35 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
unit/test-eapol.c
View File

@ -94,6 +94,7 @@ static struct handshake_state *test_handshake_state_new(uint32_t ifindex)
ths->super.ifindex = ifindex;
ths->super.free = test_handshake_state_free;
ths->super.refcount = 1;
return &ths->super;
}
@ -2199,7 +2200,7 @@ static void eapol_sm_test_ptk(const void *data)
assert(verify_step4_called);
eapol_sm_free(sm);
handshake_state_free(hs);
handshake_state_unref(hs);
eapol_exit();
}
@ -2263,7 +2264,7 @@ static void eapol_sm_test_igtk(const void *data)
assert(verify_step4_called);
eapol_sm_free(sm);
handshake_state_free(hs);
handshake_state_unref(hs);
eapol_exit();
}
@ -2332,7 +2333,7 @@ static void eapol_sm_test_wpa2_ptk_gtk(const void *data)
assert(verify_gtk_step2_called);
eapol_sm_free(sm);
handshake_state_free(hs);
handshake_state_unref(hs);
eapol_exit();
}
@ -2399,7 +2400,7 @@ static void eapol_sm_test_wpa_ptk_gtk(const void *data)
assert(verify_gtk_step2_called);
eapol_sm_free(sm);
handshake_state_free(hs);
handshake_state_unref(hs);
eapol_exit();
}
@ -2467,7 +2468,7 @@ static void eapol_sm_test_wpa_ptk_gtk_2(const void *data)
assert(verify_gtk_step2_called);
eapol_sm_free(sm);
handshake_state_free(hs);
handshake_state_unref(hs);
eapol_exit();
}
@ -2721,7 +2722,7 @@ static void eapol_sm_wpa2_retransmit_test(const void *data)
l_free(ptk);
eapol_sm_free(sm);
handshake_state_free(hs);
handshake_state_unref(hs);
eapol_exit();
}
@ -3141,7 +3142,7 @@ done:
if (sm)
eapol_sm_free(sm);
handshake_state_free(hs);
handshake_state_unref(hs);
eapol_exit();
eap_exit();
}
@ -3424,7 +3425,7 @@ static void eapol_sm_test_eap_nak(const void *data)
sizeof(eap_failure), false);
assert(ths->handshake_failed);
handshake_state_free(hs);
handshake_state_unref(hs);
eapol_exit();
eap_exit();
}
@ -3510,7 +3511,7 @@ static void eapol_ft_handshake_test(const void *data)
assert(verify_step4_called);
eapol_sm_free(sm);
handshake_state_free(hs);
handshake_state_unref(hs);
eapol_exit();
}
@ -3609,6 +3610,7 @@ static struct handshake_state *test_ap_sta_hs_new(struct test_ap_sta_data *s,
ths->super.ifindex = ifindex;
ths->super.free = (void (*)(struct handshake_state *s)) l_free;
ths->super.refcount = 1;
ths->s = s;
return &ths->super;
@ -3692,8 +3694,10 @@ static void eapol_ap_sta_handshake_test(const void *data)
test_ap_sta_run(&s);
handshake_state_free(s.ap_hs);
handshake_state_free(s.sta_hs);
l_info("freeing in eapol_ap_sta_handshake_test()");
handshake_state_unref(s.ap_hs);
handshake_state_unref(s.sta_hs);
__handshake_set_install_tk_func(NULL);
assert(s.ap_success && s.sta_success);
@ -3753,8 +3757,8 @@ static void eapol_ap_sta_handshake_bad_psk_test(const void *data)
test_ap_sta_run(&s);
handshake_state_free(s.ap_hs);
handshake_state_free(s.sta_hs);
handshake_state_unref(s.ap_hs);
handshake_state_unref(s.sta_hs);
__handshake_set_install_tk_func(NULL);
/*
@ -3825,8 +3829,8 @@ static void eapol_ap_sta_handshake_ip_alloc_ok_test(const void *data)
assert(s.sta_hs->subnet_mask == s.ap_hs->subnet_mask);
assert(s.sta_hs->go_ip_addr == s.ap_hs->go_ip_addr);
handshake_state_free(s.ap_hs);
handshake_state_free(s.sta_hs);
handshake_state_unref(s.ap_hs);
handshake_state_unref(s.sta_hs);
__handshake_set_install_tk_func(NULL);
assert(s.ap_success && s.sta_success);
@ -3889,8 +3893,8 @@ static void eapol_ap_sta_handshake_ip_alloc_no_req_test(const void *data)
assert(!s.ap_hs->support_ip_allocation);
assert(!s.sta_hs->support_ip_allocation);
handshake_state_free(s.ap_hs);
handshake_state_free(s.sta_hs);
handshake_state_unref(s.ap_hs);
handshake_state_unref(s.sta_hs);
__handshake_set_install_tk_func(NULL);
assert(s.ap_success && s.sta_success);

15
unit/test-sae.c
View File

@ -101,6 +101,7 @@ static struct handshake_state *test_handshake_state_new(uint32_t ifindex)
ths->super.ifindex = ifindex;
ths->super.free = test_handshake_state_free;
ths->super.refcount = 1;
return &ths->super;
}
@ -183,7 +184,7 @@ static struct auth_proto *test_initialize(struct test_data *td)
static void test_destruct(struct test_data *td)
{
handshake_state_free(td->handshake);
handshake_state_unref(td->handshake);
l_free(td);
}
@ -459,8 +460,8 @@ static void test_bad_confirm(const void *arg)
assert(td1->tx_assoc_called);
assert(td2->status != 0);
handshake_state_free(hs1);
handshake_state_free(hs2);
handshake_state_unref(hs1);
handshake_state_unref(hs2);
ap1->free(ap1);
ap2->free(ap2);
@ -544,8 +545,8 @@ static void test_confirm_after_accept(const void *arg)
assert(auth_proto_rx_associate(ap1, (uint8_t *)assoc, frame_len) == 0);
assert(auth_proto_rx_associate(ap2, (uint8_t *)assoc, frame_len) == 0);
handshake_state_free(hs1);
handshake_state_free(hs2);
handshake_state_unref(hs1);
handshake_state_unref(hs2);
auth_proto_free(ap1);
auth_proto_free(ap2);
@ -621,8 +622,8 @@ static void test_end_to_end(const void *arg)
assert(auth_proto_rx_associate(ap1, (uint8_t *)assoc, frame_len) == 0);
assert(auth_proto_rx_associate(ap2, (uint8_t *)assoc, frame_len) == 0);
handshake_state_free(hs1);
handshake_state_free(hs2);
handshake_state_unref(hs1);
handshake_state_unref(hs2);
auth_proto_free(ap1);
auth_proto_free(ap2);

10
unit/test-wsc.c
View File

@ -61,6 +61,7 @@ static struct handshake_state *test_handshake_state_new(uint32_t ifindex)
ths->super.ifindex = ifindex;
ths->super.free = test_handshake_state_free;
ths->super.refcount = 1;
return &ths->super;
}
@ -2093,7 +2094,7 @@ static void wsc_test_pbc_handshake(const void *data)
eap_fail, sizeof(eap_fail), false);
assert(verify.eapol_failed);
handshake_state_free(hs);
handshake_state_unref(hs);
eapol_exit();
eap_exit();
}
@ -2216,7 +2217,7 @@ static void wsc_test_retransmission_no_fragmentation(const void *data)
sizeof(eap_fail), false);
assert(verify.eapol_failed);
handshake_state_free(hs);
handshake_state_unref(hs);
eapol_exit();
eap_exit();
}
@ -2335,6 +2336,7 @@ static struct handshake_state *test_ap_sta_hs_new(struct test_ap_sta_data *s,
ths->super.ifindex = ifindex;
ths->super.free = (void (*)(struct handshake_state *s)) l_free;
ths->super.refcount = 1;
ths->s = s;
return &ths->super;
@ -2535,8 +2537,8 @@ static void wsc_r_test_pbc_handshake(const void *data)
test_ap_sta_run(&s);
handshake_state_free(s.ap_hs);
handshake_state_free(s.sta_hs);
handshake_state_unref(s.ap_hs);
handshake_state_unref(s.sta_hs);
__handshake_set_install_tk_func(NULL);
l_settings_free(ap_8021x_settings);
l_settings_free(sta_8021x_settings);