busybox/modutils/modprobe.c

693 lines
14 KiB
C
Raw Normal View History

/* vi: set sw=4 ts=4: */
/*
* Modprobe written from scratch for BusyBox
*
* Copyright (c) 2002 by Robert Griebl, griebl@gmx.de
* Copyright (c) 2003 by Andrew Dennison, andrew.dennison@motec.com.au
* Copyright (c) 2005 by Jim Bauer, jfbauer@nfr.com
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*
*/
#include <sys/utsname.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <getopt.h>
#include <stdlib.h>
#include <unistd.h>
#include <syslog.h>
#include <string.h>
#include <ctype.h>
#include <fcntl.h>
#include "busybox.h"
struct dep_t {
char * m_name;
char * m_path;
char * m_options;
2003-12-25 02:00:45 +05:30
int m_isalias : 1;
int m_reserved : 15;
2003-12-25 02:00:45 +05:30
2004-01-10 16:59:31 +05:30
int m_depcnt : 16;
char ** m_deparr;
2003-12-25 02:00:45 +05:30
struct dep_t * m_next;
};
struct mod_list_t {
char * m_name;
char * m_path;
char * m_options;
2003-12-25 02:00:45 +05:30
struct mod_list_t * m_prev;
struct mod_list_t * m_next;
};
static struct dep_t *depend;
static int autoclean, show_only, quiet, do_syslog, verbose;
static int k_version;
static int parse_tag_value ( char *buffer, char **ptag, char **pvalue )
{
char *tag, *value;
while ( isspace ( *buffer ))
2004-01-10 16:59:31 +05:30
buffer++;
tag = value = buffer;
while ( !isspace ( *value ))
if (!*value) return 0;
else value++;
*value++ = 0;
while ( isspace ( *value ))
value++;
if (!*value) return 0;
*ptag = tag;
*pvalue = value;
2003-12-25 02:00:45 +05:30
return 1;
}
/* Jump through hoops to simulate how fgets() grabs just one line at a
* time... Don't use any stdio since modprobe gets called from a kernel
2004-01-10 16:59:31 +05:30
* thread and stdio junk can overflow the limited stack...
*/
static char *reads ( int fd, char *buffer, size_t len )
{
int n = read ( fd, buffer, len );
2003-12-25 02:00:45 +05:30
if ( n > 0 ) {
char *p;
2003-12-25 02:00:45 +05:30
buffer [len-1] = 0;
p = strchr ( buffer, '\n' );
2003-12-25 02:00:45 +05:30
if ( p ) {
off_t offset;
2003-12-25 02:00:45 +05:30
2004-01-10 16:59:31 +05:30
offset = lseek ( fd, 0L, SEEK_CUR ); // Get the current file descriptor offset
lseek ( fd, offset-n + (p-buffer) + 1, SEEK_SET ); // Set the file descriptor offset to right after the \n
p[1] = 0;
}
return buffer;
}
2003-12-25 02:00:45 +05:30
else
return 0;
}
static struct dep_t *build_dep ( void )
{
int fd;
struct utsname un;
struct dep_t *first = 0;
struct dep_t *current = 0;
char buffer[2048];
char *filename = buffer;
int continuation_line = 0;
2003-12-25 02:00:45 +05:30
k_version = 0;
if ( uname ( &un ))
return 0;
2003-12-25 02:00:45 +05:30
// check for buffer overflow in following code
2003-03-19 14:43:01 +05:30
if ( bb_strlen ( un.release ) > ( sizeof( buffer ) - 64 )) {
return 0;
}
if (un.release[0] == '2') {
k_version = un.release[2] - '0';
}
2003-12-25 02:00:45 +05:30
strcpy ( filename, "/lib/modules/" );
strcat ( filename, un.release );
strcat ( filename, "/modules.dep" );
if (( fd = open ( filename, O_RDONLY )) < 0 ) {
/* Ok, that didn't work. Fall back to looking in /lib/modules */
if (( fd = open ( "/lib/modules/modules.dep", O_RDONLY )) < 0 ) {
return 0;
}
}
while ( reads ( fd, buffer, sizeof( buffer ))) {
2003-03-19 14:43:01 +05:30
int l = bb_strlen ( buffer );
char *p = 0;
2003-12-25 02:00:45 +05:30
while ( isspace ( buffer [l-1] )) {
buffer [l-1] = 0;
l--;
}
2003-12-25 02:00:45 +05:30
if ( l == 0 ) {
continuation_line = 0;
continue;
}
2003-12-25 02:00:45 +05:30
2004-01-10 16:59:31 +05:30
if ( !continuation_line ) {
char *col = strchr ( buffer, ':' );
char *dot = col;
2003-12-25 02:00:45 +05:30
if ( col ) {
char *mods;
char *modpath;
char *mod;
2003-12-25 02:00:45 +05:30
*col = 0;
mods = strrchr ( buffer, '/' );
2003-12-25 02:00:45 +05:30
if ( !mods )
mods = buffer;
else
mods++;
2003-12-25 02:00:45 +05:30
modpath = strchr ( buffer, '/' );
if ( !modpath )
modpath = buffer;
#if defined(CONFIG_FEATURE_2_6_MODULES)
if ((k_version > 4) && ( *(col-3) == '.' ) &&
2003-12-25 02:00:45 +05:30
( *(col-2) == 'k' ) && ( *(col-1) == 'o' ))
dot = col - 3;
else
#endif
2003-12-25 02:00:45 +05:30
if (( *(col-2) == '.' ) && ( *(col-1) == 'o' ))
dot = col - 2;
2003-12-25 02:00:45 +05:30
mod = bb_xstrndup ( mods, dot - mods );
2003-12-25 02:00:45 +05:30
if ( !current ) {
first = current = (struct dep_t *) xmalloc ( sizeof ( struct dep_t ));
}
else {
current-> m_next = (struct dep_t *) xmalloc ( sizeof ( struct dep_t ));
current = current-> m_next;
}
current-> m_name = mod;
current-> m_path = bb_xstrdup(modpath);
current-> m_options = 0;
current-> m_isalias = 0;
current-> m_depcnt = 0;
current-> m_deparr = 0;
current-> m_next = 0;
2003-12-25 02:00:45 +05:30
//printf ( "%s:\n", mod );
2004-01-10 16:59:31 +05:30
p = col + 1;
}
else
p = 0;
}
else
p = buffer;
2003-12-25 02:00:45 +05:30
while ( p && *p && isblank(*p))
p++;
if ( p && *p ) {
char *end = &buffer [l-1];
char *deps;
char *dep;
char *next;
int ext = 0;
2003-12-25 02:00:45 +05:30
while ( isblank ( *end ) || ( *end == '\\' ))
end--;
2003-12-25 02:00:45 +05:30
do
{
next = strchr (p, ' ' );
if (next)
{
*next = 0;
next--;
}
else
next = end;
2004-01-10 16:59:31 +05:30
deps = strrchr ( p, '/' );
2003-12-25 02:00:45 +05:30
2004-01-10 16:59:31 +05:30
if ( !deps || ( deps < p )) {
deps = p;
2003-12-25 02:00:45 +05:30
2004-01-10 16:59:31 +05:30
while ( isblank ( *deps ))
deps++;
}
else
deps++;
2003-12-25 02:00:45 +05:30
#if defined(CONFIG_FEATURE_2_6_MODULES)
if ((k_version > 4) && ( *(next-2) == '.' ) && *(next-1) == 'k' &&
( *next == 'o' ))
2004-01-10 16:59:31 +05:30
ext = 3;
else
#endif
if (( *(next-1) == '.' ) && ( *next == 'o' ))
2004-01-10 16:59:31 +05:30
ext = 2;
2004-01-10 16:59:31 +05:30
/* Cope with blank lines */
if ((next-deps-ext+1) <= 0)
2004-01-10 16:59:31 +05:30
continue;
dep = bb_xstrndup ( deps, next - deps - ext + 1 );
2003-12-25 02:00:45 +05:30
2004-01-10 16:59:31 +05:30
current-> m_depcnt++;
current-> m_deparr = (char **) xrealloc ( current-> m_deparr,
sizeof ( char *) * current-> m_depcnt );
current-> m_deparr [current-> m_depcnt - 1] = dep;
2003-12-25 02:00:45 +05:30
2004-01-10 16:59:31 +05:30
//printf ( " %d) %s\n", current-> m_depcnt, current-> m_deparr [current-> m_depcnt -1] );
p = next + 2;
} while (next < end);
}
2003-12-25 02:00:45 +05:30
if ( buffer [l-1] == '\\' )
continuation_line = 1;
else
continuation_line = 0;
}
close ( fd );
// alias parsing is not 100% correct (no correct handling of continuation lines within an alias) !
#if defined(CONFIG_FEATURE_2_6_MODULES)
if (( fd = open ( "/etc/modprobe.conf", O_RDONLY )) < 0 )
#endif
if (( fd = open ( "/etc/modules.conf", O_RDONLY )) < 0 )
if (( fd = open ( "/etc/conf.modules", O_RDONLY )) < 0 )
return first;
2003-12-25 02:00:45 +05:30
continuation_line = 0;
while ( reads ( fd, buffer, sizeof( buffer ))) {
int l;
char *p;
2003-12-25 02:00:45 +05:30
2004-01-10 16:59:31 +05:30
p = strchr ( buffer, '#' );
if ( p )
*p = 0;
2003-12-25 02:00:45 +05:30
2003-03-19 14:43:01 +05:30
l = bb_strlen ( buffer );
2003-12-25 02:00:45 +05:30
while ( l && isspace ( buffer [l-1] )) {
buffer [l-1] = 0;
l--;
}
2003-12-25 02:00:45 +05:30
if ( l == 0 ) {
continuation_line = 0;
continue;
}
2003-12-25 02:00:45 +05:30
2004-01-10 16:59:31 +05:30
if ( !continuation_line ) {
if (( strncmp ( buffer, "alias", 5 ) == 0 ) && isspace ( buffer [5] )) {
char *alias, *mod;
if ( parse_tag_value ( buffer + 6, &alias, &mod )) {
// fprintf ( stderr, "ALIAS: '%s' -> '%s'\n", alias, mod );
2003-12-25 02:00:45 +05:30
if ( !current ) {
first = current = (struct dep_t *) xcalloc ( 1, sizeof ( struct dep_t ));
}
else {
current-> m_next = (struct dep_t *) xcalloc ( 1, sizeof ( struct dep_t ));
current = current-> m_next;
}
current-> m_name = bb_xstrdup ( alias );
current-> m_isalias = 1;
2003-12-25 02:00:45 +05:30
if (( strcmp ( mod, "off" ) == 0 ) || ( strcmp ( mod, "null" ) == 0 )) {
current-> m_depcnt = 0;
current-> m_deparr = 0;
}
else {
current-> m_depcnt = 1;
current-> m_deparr = xmalloc ( 1 * sizeof( char * ));
2003-03-19 14:43:01 +05:30
current-> m_deparr[0] = bb_xstrdup ( mod );
}
2004-01-10 16:59:31 +05:30
current-> m_next = 0;
}
2004-01-10 16:59:31 +05:30
}
else if (( strncmp ( buffer, "options", 7 ) == 0 ) && isspace ( buffer [7] )) {
char *mod, *opt;
2003-12-25 02:00:45 +05:30
if ( parse_tag_value ( buffer + 8, &mod, &opt )) {
struct dep_t *dt;
2003-12-25 02:00:45 +05:30
for ( dt = first; dt; dt = dt-> m_next ) {
if ( strcmp ( dt-> m_name, mod ) == 0 )
break;
}
if ( dt ) {
2003-03-19 14:43:01 +05:30
dt-> m_options = xrealloc ( dt-> m_options, bb_strlen( opt ) + 1 );
strcpy ( dt-> m_options, opt );
2003-12-25 02:00:45 +05:30
// fprintf ( stderr, "OPTION: '%s' -> '%s'\n", dt-> m_name, dt-> m_options );
}
}
}
}
}
close ( fd );
2003-12-25 02:00:45 +05:30
return first;
}
/* return 1 = loaded, 0 = not loaded, -1 = can't tell */
static int already_loaded (const char *name)
{
int fd;
char buffer[4096];
fd = open ("/proc/modules", O_RDONLY);
if (fd < 0)
return -1;
while ( reads ( fd, buffer, sizeof( buffer ))) {
char *p;
p = strchr (buffer, ' ');
if (p) {
*p = 0;
if (strcmp (name, buffer) == 0) {
close (fd);
return 1;
}
}
}
close (fd);
return 0;
}
static int mod_process ( struct mod_list_t *list, int do_insert )
{
int rc = 0;
char *argv[10];
int argc;
while ( list ) {
argc = 0;
if ( do_insert ) {
if (already_loaded (list->m_name) != 1) {
argv[argc++] = "insmod";
if (do_syslog)
argv[argc++] = "-s";
if (autoclean)
argv[argc++] = "-k";
if (quiet)
argv[argc++] = "-q";
argv[argc++] = list-> m_path;
if (list-> m_options)
argv[argc++] = list-> m_options;
}
} else {
/* modutils uses short name for removal */
if (already_loaded (list->m_name) != 0) {
argv[argc++] = "rmmod";
if (do_syslog)
argv[argc++] = "-s";
argv[argc++] = list->m_name;
}
}
argv[argc] = NULL;
2003-12-25 02:00:45 +05:30
if (argc) {
if (verbose) {
int i;
for (i=0; i<argc; i++)
printf("%s ", argv[i]);
printf("\n");
}
if (!show_only) {
int rc2 = 0;
int status;
switch (fork()) {
case -1:
rc2 = 1;
break;
case 0: //child
execvp(argv[0], argv);
bb_perror_msg_and_die("exec of %s", argv[0]);
/* NOTREACHED */
default:
if (wait(&status) == -1) {
rc2 = 1;
break;
}
if (WIFEXITED(status))
rc2 = WEXITSTATUS(status);
if (WIFSIGNALED(status))
rc2 = WTERMSIG(status);
break;
}
if (do_insert) {
rc = rc2; /* only last module matters */
}
else if (!rc2) {
rc = 0; /* success if remove any mod */
}
}
}
list = do_insert ? list-> m_prev : list-> m_next;
}
return (show_only) ? 0 : rc;
}
static void check_dep ( char *mod, struct mod_list_t **head, struct mod_list_t **tail )
{
struct mod_list_t *find;
struct dep_t *dt;
char *opt = 0;
char *path = 0;
// check dependencies
for ( dt = depend; dt; dt = dt-> m_next ) {
if ( strcmp ( dt-> m_name, mod ) == 0) {
mod = dt-> m_name;
path = dt-> m_path;
opt = dt-> m_options;
break;
}
}
2003-12-25 02:00:45 +05:30
// resolve alias names
while ( dt && dt-> m_isalias ) {
if ( dt-> m_depcnt == 1 ) {
struct dep_t *adt;
2003-12-25 02:00:45 +05:30
for ( adt = depend; adt; adt = adt-> m_next ) {
if ( strcmp ( adt-> m_name, dt-> m_deparr [0] ) == 0 )
break;
}
if ( adt ) {
2004-01-10 16:59:31 +05:30
dt = adt;
mod = dt-> m_name;
path = dt-> m_path;
if ( !opt )
opt = dt-> m_options;
}
else
return;
}
else
2004-01-10 16:59:31 +05:30
return;
}
2003-12-25 02:00:45 +05:30
if ( !path ) {
bb_error_msg ("module %s not found.", mod);
return;
}
// search for duplicates
for ( find = *head; find; find = find-> m_next ) {
if ( !strcmp ( mod, find-> m_name )) {
// found -> dequeue it
if ( find-> m_prev )
find-> m_prev-> m_next = find-> m_next;
else
*head = find-> m_next;
2003-12-25 02:00:45 +05:30
if ( find-> m_next )
find-> m_next-> m_prev = find-> m_prev;
else
*tail = find-> m_prev;
2003-12-25 02:00:45 +05:30
break; // there can be only one duplicate
2004-01-10 16:59:31 +05:30
}
}
if ( !find ) { // did not find a duplicate
2004-01-10 16:59:31 +05:30
find = (struct mod_list_t *) xmalloc ( sizeof(struct mod_list_t));
find-> m_name = mod;
find-> m_path = path;
find-> m_options = opt;
}
2004-01-10 16:59:31 +05:30
// enqueue at tail
if ( *tail )
(*tail)-> m_next = find;
find-> m_prev = *tail;
find-> m_next = 0;
if ( !*head )
*head = find;
*tail = find;
2003-12-25 02:00:45 +05:30
2004-01-10 16:59:31 +05:30
if ( dt ) {
int i;
2003-12-25 02:00:45 +05:30
for ( i = 0; i < dt-> m_depcnt; i++ )
check_dep ( dt-> m_deparr [i], head, tail );
}
}
static int mod_insert ( char *mod, int argc, char **argv )
{
struct mod_list_t *tail = 0;
2004-01-10 16:59:31 +05:30
struct mod_list_t *head = 0;
int rc;
2003-12-25 02:00:45 +05:30
// get dep list for module mod
check_dep ( mod, &head, &tail );
2003-12-25 02:00:45 +05:30
if ( head && tail ) {
#if defined(CONFIG_FEATURE_2_6_MODULES)
if ( argc ) {
2004-01-10 16:59:31 +05:30
int i;
int l = 0;
2003-12-25 02:00:45 +05:30
// append module args
2004-01-10 16:59:31 +05:30
for ( i = 0; i < argc; i++ )
2003-03-19 14:43:01 +05:30
l += ( bb_strlen ( argv [i] ) + 1 );
2003-12-25 02:00:45 +05:30
head-> m_options = xrealloc ( head-> m_options, l + 1 );
head-> m_options [0] = 0;
2003-12-25 02:00:45 +05:30
for ( i = 0; i < argc; i++ ) {
strcat ( head-> m_options, argv [i] );
strcat ( head-> m_options, " " );
}
}
#endif
2003-12-25 02:00:45 +05:30
// process tail ---> head
rc = mod_process ( tail, 1 );
}
else
rc = 1;
2003-12-25 02:00:45 +05:30
return rc;
}
static int mod_remove ( char *mod )
{
int rc;
2004-01-10 16:59:31 +05:30
static struct mod_list_t rm_a_dummy = { "-a", 0, 0 };
2003-12-25 02:00:45 +05:30
struct mod_list_t *head = 0;
struct mod_list_t *tail = 0;
2003-12-25 02:00:45 +05:30
if ( mod )
check_dep ( mod, &head, &tail );
else // autoclean
head = tail = &rm_a_dummy;
2003-12-25 02:00:45 +05:30
if ( head && tail )
rc = mod_process ( head, 0 ); // process head ---> tail
else
rc = 1;
return rc;
2003-12-25 02:00:45 +05:30
}
extern int modprobe_main(int argc, char** argv)
{
int opt;
int remove_opt = 0;
autoclean = show_only = quiet = do_syslog = verbose = 0;
while ((opt = getopt(argc, argv, "acdklnqrst:vVC:")) != -1) {
switch(opt) {
2003-12-25 02:00:45 +05:30
case 'c': // no config used
case 'l': // no pattern matching
return EXIT_SUCCESS;
break;
case 'C': // no config used
case 't': // no pattern matching
bb_error_msg_and_die("-t and -C not supported");
case 'a': // ignore
case 'd': // ignore
break;
case 'k':
autoclean++;
break;
case 'n':
show_only++;
break;
case 'q':
quiet++;
break;
case 'r':
remove_opt++;
break;
case 's':
do_syslog++;
break;
case 'v':
verbose++;
break;
case 'V':
default:
bb_show_usage();
break;
}
}
2003-12-25 02:00:45 +05:30
2004-01-10 16:59:31 +05:30
depend = build_dep ( );
2004-01-10 16:59:31 +05:30
if ( !depend )
2003-03-19 14:43:01 +05:30
bb_error_msg_and_die ( "could not parse modules.dep\n" );
2003-12-25 02:00:45 +05:30
if (remove_opt) {
int rc = EXIT_SUCCESS;
do {
if (mod_remove ( optind < argc ?
2003-12-25 02:00:45 +05:30
bb_xstrdup (argv [optind]) : NULL )) {
bb_error_msg ("failed to remove module %s",
2003-12-25 02:00:45 +05:30
argv [optind] );
rc = EXIT_FAILURE;
}
} while ( ++optind < argc );
2003-12-25 02:00:45 +05:30
return rc;
}
2004-01-10 16:59:31 +05:30
if (optind >= argc)
2003-03-19 14:43:01 +05:30
bb_error_msg_and_die ( "No module or pattern provided\n" );
2003-12-25 02:00:45 +05:30
2004-01-10 16:59:31 +05:30
if ( mod_insert ( bb_xstrdup ( argv [optind] ), argc - optind - 1, argv + optind + 1 ))
2003-03-19 14:43:01 +05:30
bb_error_msg_and_die ( "failed to load module %s", argv [optind] );
2003-12-25 02:00:45 +05:30
return EXIT_SUCCESS;
}