kernel: enumerate internal kexts during boot

2023-04-08 09:27:21 +01:00
parent a1f54fd156
commit 9b75ca8b8c
6 changed files with 151 additions and 4 deletions
--- a/include/socks/kext.h
+++ b/include/socks/kext.h
@@ -0,0 +1,81 @@
+#ifndef SOCKS_KEXT_H_
+#define SOCKS_KEXT_H_
+
+#include <socks/status.h>
+#include <socks/compiler.h>
+
+#define KERNEL_KEXT_ID "net.doorstuck.socks-kernel"
+#define KEXT_IDENT_MAX 80
+
+#define __KEXT_INFO_VARNAME_2(a, b) a ## b
+#define __KEXT_INFO_VARNAME_1(a, b) __KEXT_INFO_VARNAME_2(a, b)
+
+#ifdef SOCKS_INTERNAL
+#define __KEXT_INFO_LINKAGE static
+#define __KEXT_INFO_VARNAME() __KEXT_INFO_VARNAME_1(__kext_info, __LINE__)
+#define __KEXT_INFO_DEPNAME() __KEXT_INFO_VARNAME_1(__kext_deps, __LINE__)
+#define __KEXT_INFO_FLAGS KEXT_INTERNAL
+#else
+#define __KEXT_INFO_LINKAGE
+#define __KEXT_INFO_VARNAME() __kext_info
+#define __KEXT_INFO_DEPNAME() __kext_deps
+#define __KEXT_INFO_FLAGS KEXT_NONE
+#endif
+
+#ifdef __cplusplus
+#define DEFINE_KEXT(ident, online, offline, ...)                                        \
+	static const char *__KEXT_INFO_DEPNAME()[] = {          \
+		__VA_ARGS__                                                             \
+	};                                                                              \
+	static struct kext_info __section(".kextinfo") __used __KEXT_INFO_VARNAME() = { \
+		__KEXT_INFO_FLAGS,                                                      \
+		ident,                                                                  \
+		online,                                                                 \
+		offline,                                                                \
+		__KEXT_INFO_DEPNAME(),	                                                \
+	}
+#else
+#define DEFINE_KEXT(ident, online, offline, ...)                              \
+	static const char *__kext_deps[] = {          \
+		__VA_ARGS__                                                   \
+	};                                                                    \
+	static struct kext_info __section(".kextinfo") __used __kext_info = { \
+		.k_flags = __KEXT_INFO_FLAGS,                                 \
+		.k_ident = ident,                                             \
+		.k_online = online,                                           \
+		.k_offline = offline,                                         \
+		.k_dependencies = __kext_deps,                                \
+	}
+#endif
+
+struct kext;
+
+enum kext_flags {
+	KEXT_NONE = 0x00u,
+	KEXT_INTERNAL = 0x01u,
+	KEXT_ONLINE = 0x02u,
+};
+
+struct kext_info {
+	enum kext_flags k_flags;
+	char k_ident[KEXT_IDENT_MAX];
+	kern_status_t(*k_online)(struct kext *);
+	kern_status_t(*k_offline)(struct kext *);
+	const char **k_dependencies;
+};
+
+struct kext {
+	enum kext_flags k_flags;
+	char k_ident[KEXT_IDENT_MAX];
+
+	kern_status_t(*k_online)(struct kext *);
+	kern_status_t(*k_offline)(struct kext *);
+
+	unsigned int k_nr_dependencies;
+	struct kext **k_dependencies;
+};
+
+extern kern_status_t scan_internal_kexts(void);
+extern kern_status_t bring_internal_kexts_online(void);
+
+#endif