Adding insmod and rmmod

2 files changed, 98 insertions, 0 deletions

diff --git a/toys/insmod.c b/toys/insmod.c
new file mode 100644
index 00000000..e794828c
--- /dev/null
+++ b/toys/insmod.c
@@ -0,0 +1,47 @@
+/* vi: set sw=4 ts=4:
+ *
+ * insmod.c - Load a module into the Linux kernel.
+ *
+ * Copyright 2012 Elie De Brauwer <eliedebrauwer@gmail.com>
+ *
+ * Not in SUSv4.
+
+USE_INSMOD(NEWTOY(insmod, "<1", TOYFLAG_BIN|TOYFLAG_NEEDROOT))
+
+config INSMOD
+	bool "insmod"
+	default y
+	help
+	  usage: insmod MODULE [MODULE_OPTIONS]
+
+	  Load the module named MODULE passing options if given.
+*/
+
+#include "toys.h"
+
+#include <sys/syscall.h>
+#define init_module(mod, len, opts) syscall(__NR_init_module, mod, len, opts)
+
+void insmod_main(void)
+{
+	char * buf = NULL;
+	int len, res, i;
+	int fd = xopen(toys.optargs[0], O_RDONLY);
+
+	len = fdlength(fd);
+	buf = xmalloc(len);
+	xreadall(fd, buf, len);
+
+	i = 1;
+	while(toys.optargs[i] &&
+		strlen(toybuf) + strlen(toys.optargs[i]) + 2 < sizeof(toybuf)) {
+		strcat(toybuf, toys.optargs[i++]);
+		strcat(toybuf, " ");
+	}
+
+	res = init_module(buf, len, toybuf);
+	if (CFG_TOYBOX_FREE && buf != toybuf) free(buf);
+
+	if (res)
+		perror_exit("failed to load %s", toys.optargs[0]);
+}
diff --git a/toys/rmmod.c b/toys/rmmod.c
new file mode 100644
index 00000000..d730b45d
--- /dev/null
+++ b/toys/rmmod.c
@@ -0,0 +1,51 @@
+/* vi: set sw=4 ts=4:
+ *
+ * rmmod.c - Remove a module from the Linux kernel.
+ *
+ * Copyright 2012 Elie De Brauwer <eliedebrauwer@gmail.com>
+ *
+ * Not in SUSv4.
+
+USE_RMMOD(NEWTOY(rmmod, "<1wf", TOYFLAG_BIN|TOYFLAG_NEEDROOT))
+
+config RMMOD
+	bool "rmmod"
+	default y
+	help
+	  usage: rmmod [-wf] [MODULE]
+
+	  Unload the module named MODULE from the Linux kernel.
+	  -f   Force unload of a module
+	  -w   Wait until the module is no longer used.
+
+*/
+
+#include "toys.h"
+
+#include <sys/syscall.h>
+#define delete_module(mod, flags) syscall(__NR_delete_module, mod, flags)
+
+void rmmod_main(void)
+{
+	unsigned int flags = O_NONBLOCK|O_EXCL;
+	char * mod_name;
+	int len;
+
+	// Basename
+	mod_name = strrchr(toys.optargs[0],'/');
+	if (mod_name)
+		mod_name++;
+	else
+		mod_name = toys.optargs[0];
+
+	// Remove .ko if present
+	len = strlen(mod_name);
+	if (len > 3 && !strcmp(&mod_name[len-3], ".ko" ))
+		mod_name[len-3] = 0;
+
+	if (toys.optflags & 1) flags |= O_TRUNC;
+	if (toys.optflags & 2) flags &= ~O_NONBLOCK;
+
+	if (delete_module(mod_name, flags))
+		perror_exit("failed to unload %s", mod_name);
+}