1
0
mirror of git://git.suckless.org/ubase synced 2024-12-20 22:00:51 +00:00
ubase/rmmod.c
Hiltjo Posthuma dfc2f7edac rmmod: fix segfault if '.' is not found
Signed-off-by: Hiltjo Posthuma <hiltjo@codemadness.org>
2014-04-06 12:51:31 +01:00

49 lines
763 B
C

/* See LICENSE file for copyright and license details. */
#include <sys/syscall.h>
#include <unistd.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <libgen.h>
#include "util.h"
static void
usage(void)
{
eprintf("usage: %s [-fw] module...\n", argv0);
}
int
main(int argc, char *argv[])
{
char *mod, *p;
int i;
int flags = O_NONBLOCK;
ARGBEGIN {
case 'f':
flags |= O_TRUNC;
break;
case 'w':
flags &= ~O_NONBLOCK;
break;
default:
usage();
} ARGEND;
if (argc < 1)
usage();
for (i = 0; i < argc; i++) {
mod = argv[i];
p = strrchr(mod, '.');
if (p && !strcmp(p, ".ko"))
*p = '\0';
if (syscall(__NR_delete_module, mod, flags) < 0)
eprintf("delete_module:");
}
return EXIT_SUCCESS;
}