.. | .. |
---|
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 | 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 | 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); |
---|