view toys/other/rmmod.c @ 674:7e846e281e38

New build infrastructure to generate FLAG_ macros and TT alias, #define FOR_commandname before #including toys.h to trigger it. Rename DEFINE_GLOBALS() to just GLOBALS() (because I could never remember if it was DECLARE_GLOBALS). Convert existing commands to use new infrastructure, and replace optflag constants with FLAG_ macros where appropriate.
author Rob Landley <rob@landley.net>
date Mon, 08 Oct 2012 00:02:30 -0500
parents 6df4ccc0acbe
children 786841fdb1e0
line wrap: on
line source

/* vi: set sw=4 ts=4:
 *
 * rmmod.c - Remove a module from the Linux kernel.
 *
 * Copyright 2012 Elie De Brauwer <eliedebrauwer@gmail.com>

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.

*/

#define FOR_rmmod
#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 & FLAG_f) flags |= O_TRUNC;
	if (toys.optflags & FLAG_w) flags &= ~O_NONBLOCK;

	if (delete_module(mod_name, flags))
		perror_exit("failed to unload %s", mod_name);
}