mk-rootfs.sh

hc

2023-12-11 072de836f53be56a70cecf70b43ae43b7ce17376

 kernel/drivers/net/wireguard/main.c
..
..
@@ -21,12 +21,21 @@
21
21
 {
22
22
 	int ret;
23
23
 
24
+	ret = wg_allowedips_slab_init();
25
+	if (ret < 0)
26
+		goto err_allowedips;
27
+
24
28
 #ifdef DEBUG
29
+	ret = -ENOTRECOVERABLE;
25
30
 	if (!wg_allowedips_selftest() || !wg_packet_counter_selftest() ||
26
31
 	    !wg_ratelimiter_selftest())
27
-		return -ENOTRECOVERABLE;
32
+		goto err_peer;
28
33
 #endif
29
34
 	wg_noise_init();
35
+
36
+	ret = wg_peer_init();
37
+	if (ret < 0)
38
+		goto err_peer;
30
39
 
31
40
 	ret = wg_device_init();
32
41
 	if (ret < 0)
..
..
@@ -44,6 +53,10 @@
44
53
 err_netlink:
45
54
 	wg_device_uninit();
46
55
 err_device:
56
+	wg_peer_uninit();
57
+err_peer:
58
+	wg_allowedips_slab_uninit();
59
+err_allowedips:
47
60
 	return ret;
48
61
 }
49
62
 
..
..
@@ -51,6 +64,8 @@
51
64
 {
52
65
 	wg_genetlink_uninit();
53
66
 	wg_device_uninit();
67
+	wg_peer_uninit();
68
+	wg_allowedips_slab_uninit();
54
69
 }
55
70
 
56
71
 module_init(mod_init);