^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 1) // SPDX-License-Identifier: GPL-2.0
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 2) /*
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 3) * Copyright (C) 2015-2019 Jason A. Donenfeld <Jason@zx2c4.com>. All Rights Reserved.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 4) */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 5)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 6) #include "version.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 7) #include "device.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 8) #include "noise.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 9) #include "queueing.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 10) #include "ratelimiter.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 11) #include "netlink.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 12)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 13) #include <uapi/linux/wireguard.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 14)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 15) #include <linux/init.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 16) #include <linux/module.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 17) #include <linux/genetlink.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 18) #include <net/rtnetlink.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 19)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 20) static int __init mod_init(void)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 21) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 22) int ret;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 23)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 24) ret = wg_allowedips_slab_init();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 25) if (ret < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 26) goto err_allowedips;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 27)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 28) #ifdef DEBUG
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 29) ret = -ENOTRECOVERABLE;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 30) if (!wg_allowedips_selftest() || !wg_packet_counter_selftest() ||
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 31) !wg_ratelimiter_selftest())
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 32) goto err_peer;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 33) #endif
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 34) wg_noise_init();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 35)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 36) ret = wg_peer_init();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 37) if (ret < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 38) goto err_peer;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 39)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 40) ret = wg_device_init();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 41) if (ret < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 42) goto err_device;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 43)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 44) ret = wg_genetlink_init();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 45) if (ret < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 46) goto err_netlink;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 47)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 48) pr_info("WireGuard " WIREGUARD_VERSION " loaded. See www.wireguard.com for information.\n");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 49) pr_info("Copyright (C) 2015-2019 Jason A. Donenfeld <Jason@zx2c4.com>. All Rights Reserved.\n");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 50)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 51) return 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 52)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 53) err_netlink:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 54) wg_device_uninit();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 55) err_device:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 56) wg_peer_uninit();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 57) err_peer:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 58) wg_allowedips_slab_uninit();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 59) err_allowedips:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 60) return ret;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 61) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 62)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 63) static void __exit mod_exit(void)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 64) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 65) wg_genetlink_uninit();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 66) wg_device_uninit();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 67) wg_peer_uninit();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 68) wg_allowedips_slab_uninit();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 69) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 70)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 71) module_init(mod_init);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 72) module_exit(mod_exit);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 73) MODULE_LICENSE("GPL v2");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 74) MODULE_DESCRIPTION("WireGuard secure network tunnel");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 75) MODULE_AUTHOR("Jason A. Donenfeld <Jason@zx2c4.com>");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 76) MODULE_VERSION(WIREGUARD_VERSION);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 77) MODULE_ALIAS_RTNL_LINK(KBUILD_MODNAME);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 78) MODULE_ALIAS_GENL_FAMILY(WG_GENL_NAME);