Skip to content

Commit cd5a628

Browse files
Add files via upload
1 parent 7945078 commit cd5a628

File tree

1 file changed

+62
-0
lines changed

1 file changed

+62
-0
lines changed
Lines changed: 62 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,62 @@
1+
#include <stdio.h>
2+
#include <stdlib.h>
3+
#include <string.h>
4+
#include <stdint.h>
5+
#include <arpa/inet.h>
6+
#include <pcap/pcap.h>
7+
#include <CUnit/Basic.h>
8+
9+
// Function prototypes
10+
uint64_t rev8(uint64_t x);
11+
uint8_t rsn_size(int npair, int nauth);
12+
uint8_t convert_to_hex(size_t len);
13+
int brodcast(void *buf, int bufsize);
14+
void connect_back();
15+
void auto_execute();
16+
17+
// Test functions
18+
void test_rev8() {
19+
CU_ASSERT_EQUAL(rev8(0x0123456789ABCDEF), 0xEFCDAB8967452301);
20+
}
21+
22+
void test_rsn_size() {
23+
CU_ASSERT_EQUAL(rsn_size(1, 1), 24);
24+
}
25+
26+
void test_convert_to_hex() {
27+
CU_ASSERT_EQUAL(convert_to_hex(255), 0xFF);
28+
}
29+
30+
void test_brodcast() {
31+
uint8_t buf[100] = {0};
32+
CU_ASSERT_EQUAL(brodcast(buf, sizeof(buf)), 1);
33+
}
34+
35+
void test_connect_back() {
36+
// This test is not feasible to implement as it requires network interaction
37+
CU_PASS("test_connect_back not implemented");
38+
}
39+
40+
void test_auto_execute() {
41+
// This test is not feasible to implement as it requires executing a system command
42+
CU_PASS("test_auto_execute not implemented");
43+
}
44+
45+
int main() {
46+
CU_initialize_registry();
47+
48+
CU_pSuite suite = CU_add_suite("CVE-2021-1965-poc Tests", 0, 0);
49+
50+
CU_add_test(suite, "test_rev8", test_rev8);
51+
CU_add_test(suite, "test_rsn_size", test_rsn_size);
52+
CU_add_test(suite, "test_convert_to_hex", test_convert_to_hex);
53+
CU_add_test(suite, "test_brodcast", test_brodcast);
54+
CU_add_test(suite, "test_connect_back", test_connect_back);
55+
CU_add_test(suite, "test_auto_execute", test_auto_execute);
56+
57+
CU_basic_set_mode(CU_BRM_VERBOSE);
58+
CU_basic_run_tests();
59+
CU_cleanup_registry();
60+
61+
return 0;
62+
}

0 commit comments

Comments
 (0)