1 files changed, 86 insertions, 0 deletions
diff --git a/libexec/flua/libfreebsd/sys/linker/linker.c b/libexec/flua/libfreebsd/sys/linker/linker.c
new file mode 100644
index 000000000000..c78fbb2b39d2
--- /dev/null
+++ b/libexec/flua/libfreebsd/sys/linker/linker.c
@@ -0,0 +1,86 @@
+/*-
+ * SPDX-License-Identifier: BSD-2-Clause
+ *
+ * Copyright (c) 2024, Baptiste Daroussin <bapt@FreeBSD.org>
+ */
+
+#include <sys/param.h>
+#include <sys/linker.h>
+
+#include <errno.h>
+#include <stdio.h>
+#include <string.h>
+
+#include <lua.h>
+#include <lualib.h>
+#include <lauxlib.h>
+
+#include "bootstrap.h"
+
+int luaopen_freebsd_sys_linker(lua_State *L);
+
+static int
+lua_kldload(lua_State *L)
+{
+	const char *name;
+	int ret;
+
+	name = luaL_checkstring(L, 1);
+	ret = kldload(name);
+	if (ret == -1) {
+		lua_pushnil(L);
+		lua_pushstring(L, strerror(errno));
+		lua_pushinteger(L, errno);
+		return (3);
+	}
+	lua_pushinteger(L, ret);
+	return (1);
+}
+
+static int
+lua_kldunload(lua_State *L)
+{
+	const char *name;
+	int ret, fileid;
+
+	if (lua_isinteger(L, 1)) {
+		fileid = lua_tointeger(L, 1);
+	} else {
+		name = luaL_checkstring(L, 1);
+		fileid = kldfind(name);
+	}
+	if (fileid == -1) {
+		lua_pushnil(L);
+		lua_pushstring(L, strerror(errno));
+		lua_pushinteger(L, errno);
+		return (3);
+	}
+	ret = kldunload(fileid);
+	lua_pushinteger(L, ret);
+	if (ret == -1) {
+		lua_pushnil(L);
+		lua_pushstring(L, strerror(errno));
+		lua_pushinteger(L, errno);
+		return (3);
+	}
+	lua_pushinteger(L, 0);
+	return (1);
+}
+
+#define REG_SIMPLE(n)	{ #n, lua_ ## n }
+static const struct luaL_Reg freebsd_sys_linker[] = {
+	REG_SIMPLE(kldload),
+	REG_SIMPLE(kldunload),
+	{ NULL, NULL },
+};
+#undef REG_SIMPLE
+
+int
+luaopen_freebsd_sys_linker(lua_State *L)
+{
+	luaL_newlib(L, freebsd_sys_linker);
+
+	return (1);
+}
+
+FLUA_MODULE_NAMED(freebsd_sys_linker, "freebsd.sys.linker");