Some cleanups, some size reductions and some buffer overflow checks
Most of it based on ideas from vodz
This commit is contained in:
parent
94a6a956f0
commit
236abbfd71
@ -37,7 +37,8 @@ struct mod_list_t {
|
||||
};
|
||||
|
||||
|
||||
static struct dep_t *depend = 0;
|
||||
static struct dep_t *depend;
|
||||
static int autoclean, show_only, quiet, do_syslog, verbose;
|
||||
|
||||
|
||||
static struct dep_t *build_dep ( void )
|
||||
@ -52,6 +53,11 @@ static struct dep_t *build_dep ( void )
|
||||
|
||||
if ( uname ( &un ))
|
||||
return 0;
|
||||
|
||||
// check for buffer overflow in following code
|
||||
if ( xstrlen ( un. release ) > ( sizeof( buffer ) - 64 ))
|
||||
return 0;
|
||||
|
||||
strcpy ( filename, "/lib/modules/" );
|
||||
strcat ( filename, un. release );
|
||||
strcat ( filename, "/modules.dep" );
|
||||
@ -160,7 +166,7 @@ static struct dep_t *build_dep ( void )
|
||||
}
|
||||
|
||||
|
||||
static int mod_process ( struct mod_list_t *list, int do_insert, int autoclean, int quiet, int do_syslog, int show_only, int verbose )
|
||||
static int mod_process ( struct mod_list_t *list, int do_insert )
|
||||
{
|
||||
char lcmd [256];
|
||||
int rc = 0;
|
||||
@ -196,8 +202,6 @@ static void check_dep ( char *mod, struct mod_list_t **head, struct mod_list_t *
|
||||
if (( mod [lm-2] == '.' ) && ( mod [lm-1] == 'o' ))
|
||||
mod [lm-2] = 0;
|
||||
|
||||
//printf ( "check_dep: %s\n", mod );
|
||||
|
||||
// search for duplicates
|
||||
for ( find = *head; find; find = find-> m_next ) {
|
||||
if ( !strcmp ( mod, find-> m_module )) {
|
||||
@ -213,8 +217,6 @@ static void check_dep ( char *mod, struct mod_list_t **head, struct mod_list_t *
|
||||
else
|
||||
*tail = find-> m_prev;
|
||||
|
||||
//printf ( "DUP\n" );
|
||||
|
||||
break; // there can be only one duplicate
|
||||
}
|
||||
}
|
||||
@ -247,7 +249,7 @@ static void check_dep ( char *mod, struct mod_list_t **head, struct mod_list_t *
|
||||
|
||||
|
||||
|
||||
static int mod_insert ( char *mod, int argc, char **argv, int autoclean, int quiet, int do_syslog, int show_only, int verbose )
|
||||
static int mod_insert ( char *mod, int argc, char **argv )
|
||||
{
|
||||
struct mod_list_t *tail = 0;
|
||||
struct mod_list_t *head = 0;
|
||||
@ -258,10 +260,14 @@ static int mod_insert ( char *mod, int argc, char **argv, int autoclean, int qui
|
||||
|
||||
if ( head && tail ) {
|
||||
int i;
|
||||
int l = 0;
|
||||
|
||||
// append module args
|
||||
head-> m_module = xmalloc ( 256 );
|
||||
strcpy ( head-> m_module, mod );
|
||||
l = xstrlen ( mod );
|
||||
for ( i = 0; i < argc; i++ )
|
||||
l += ( xstrlen ( argv [i] ) + 1 );
|
||||
|
||||
head-> m_module = xstrndup ( mod, l );
|
||||
|
||||
for ( i = 0; i < argc; i++ ) {
|
||||
strcat ( head-> m_module, " " );
|
||||
@ -269,7 +275,7 @@ static int mod_insert ( char *mod, int argc, char **argv, int autoclean, int qui
|
||||
}
|
||||
|
||||
// process tail ---> head
|
||||
rc |= mod_process ( tail, 1, autoclean, quiet, do_syslog, show_only, verbose );
|
||||
rc |= mod_process ( tail, 1 );
|
||||
}
|
||||
else
|
||||
rc = 1;
|
||||
@ -277,7 +283,7 @@ static int mod_insert ( char *mod, int argc, char **argv, int autoclean, int qui
|
||||
return rc;
|
||||
}
|
||||
|
||||
static void mod_remove ( char *mod, int do_syslog, int show_only, int verbose )
|
||||
static void mod_remove ( char *mod )
|
||||
{
|
||||
static struct mod_list_t rm_a_dummy = { "-a", 0, 0 };
|
||||
|
||||
@ -289,38 +295,35 @@ static void mod_remove ( char *mod, int do_syslog, int show_only, int verbose )
|
||||
else // autoclean
|
||||
head = tail = &rm_a_dummy;
|
||||
|
||||
if ( head && tail ) {
|
||||
// process head ---> tail
|
||||
mod_process ( head, 0, 0, 0, do_syslog, show_only, verbose );
|
||||
}
|
||||
if ( head && tail )
|
||||
mod_process ( head, 0 ); // process head ---> tail
|
||||
}
|
||||
|
||||
|
||||
|
||||
extern int modprobe_main(int argc, char** argv)
|
||||
{
|
||||
int ch, rc = 0;
|
||||
int loadall = 0, showconfig = 0, debug = 0, autoclean = 0, list = 0;
|
||||
int show_only = 0, quiet = 0, remove_opt = 0, do_syslog = 0, verbose = 0;
|
||||
char *load_type = NULL, *config = NULL;
|
||||
int opt;
|
||||
int remove_opt = 0;
|
||||
|
||||
while ((ch = getopt(argc, argv, "acdklnqrst:vVC:")) != -1)
|
||||
switch(ch) {
|
||||
case 'a':
|
||||
loadall++;
|
||||
autoclean = show_only = quiet = do_syslog = verbose = 0;
|
||||
|
||||
while ((opt = getopt(argc, argv, "acdklnqrst:vVC:")) != -1) {
|
||||
switch(opt) {
|
||||
case 'c': // no config used
|
||||
case 'l': // no pattern matching
|
||||
return EXIT_SUCCESS;
|
||||
break;
|
||||
case 'c':
|
||||
showconfig++;
|
||||
break;
|
||||
case 'd':
|
||||
debug++;
|
||||
case 'C': // no config used
|
||||
case 't': // no pattern matching
|
||||
error_msg_and_die("-t and -C not supported");
|
||||
|
||||
case 'a': // ignore
|
||||
case 'd': // ignore
|
||||
break;
|
||||
case 'k':
|
||||
autoclean++;
|
||||
break;
|
||||
case 'l':
|
||||
list++;
|
||||
break;
|
||||
case 'n':
|
||||
show_only++;
|
||||
break;
|
||||
@ -333,55 +336,34 @@ extern int modprobe_main(int argc, char** argv)
|
||||
case 's':
|
||||
do_syslog++;
|
||||
break;
|
||||
case 't':
|
||||
load_type = optarg;
|
||||
break;
|
||||
case 'v':
|
||||
verbose++;
|
||||
break;
|
||||
case 'C':
|
||||
config = optarg;
|
||||
break;
|
||||
case 'V':
|
||||
default:
|
||||
show_usage();
|
||||
break;
|
||||
}
|
||||
|
||||
if (load_type || config) {
|
||||
fprintf(stderr, "-t and -C not supported\n");
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
if (showconfig)
|
||||
exit(EXIT_SUCCESS);
|
||||
|
||||
if (list)
|
||||
exit(EXIT_SUCCESS);
|
||||
|
||||
depend = build_dep ( );
|
||||
|
||||
if ( !depend ) {
|
||||
fprintf (stderr, "could not parse modules.dep\n" );
|
||||
exit (EXIT_FAILURE);
|
||||
}
|
||||
if ( !depend )
|
||||
error_msg_and_die ( "could not parse modules.dep\n" );
|
||||
|
||||
if (remove_opt) {
|
||||
do {
|
||||
mod_remove ( optind < argc ? argv [optind] : 0, do_syslog, show_only, verbose );
|
||||
mod_remove ( optind < argc ? argv [optind] : 0 );
|
||||
} while ( ++optind < argc );
|
||||
|
||||
exit(EXIT_SUCCESS);
|
||||
return EXIT_SUCCESS;
|
||||
}
|
||||
|
||||
if (optind >= argc) {
|
||||
fprintf(stderr, "No module or pattern provided\n");
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
if (optind >= argc)
|
||||
error_msg_and_die ( "No module or pattern provided\n" );
|
||||
|
||||
rc = mod_insert ( argv [optind], argc - optind - 1, argv + optind + 1, autoclean, quiet, do_syslog, show_only, verbose );
|
||||
|
||||
exit(rc ? EXIT_FAILURE : EXIT_SUCCESS);
|
||||
return mod_insert ( argv [optind], argc - optind - 1, argv + optind + 1 ) ? \
|
||||
EXIT_FAILURE : EXIT_SUCCESS;
|
||||
}
|
||||
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user