modutils/modprobe.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Modprobe written from scratch for BusyBox
   4  *
   5  * Copyright (c) 2002 by Robert Griebl, griebl@gmx.de
   6  * Copyright (c) 2003 by Andrew Dennison, andrew.dennison@motec.com.au
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16  * General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with this program; if not, write to the Free Software
  20  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  21  *
  22 */
  23
  24 #include <sys/utsname.h>
  25 #include <getopt.h>
  26 #include <stdlib.h>
  27 #include <unistd.h>
  28 #include <syslog.h>
  29 #include <string.h>
  30 #include <ctype.h>
  31 #include <fcntl.h>
  32 #include "busybox.h"
  33
  34
  35
  36 struct dep_t {
  37         char *  m_module;
  38         char *  m_options;
  39
  40         int     m_isalias  : 1;
  41         int     m_reserved : 15;
  42
  43         int     m_depcnt   : 16;
  44         char ** m_deparr;
  45
  46         struct dep_t * m_next;
  47 };
  48
  49 struct mod_list_t {
  50         char *  m_module;
  51         char *  m_options;
  52
  53         struct mod_list_t * m_prev;
  54         struct mod_list_t * m_next;
  55 };
  56
  57
  58 static struct dep_t *depend;
  59 static int autoclean, show_only, quiet, do_syslog, verbose;
  60
  61 int parse_tag_value ( char *buffer, char **ptag, char **pvalue )
  62 {
  63         char *tag, *value;
  64
  65         while ( isspace ( *buffer ))
  66                 buffer++;
  67         tag = value = buffer;
  68         while ( !isspace ( *value ))
  69                 value++;
  70         *value++ = 0;
  71         while ( isspace ( *value ))
  72                 value++;
  73
  74         *ptag = tag;
  75         *pvalue = value;
  76
  77         return bb_strlen( tag ) && bb_strlen( value );
  78 }
  79
  80 /* Jump through hoops to simulate how fgets() grabs just one line at a
  81  * time... Don't use any stdio since modprobe gets called from a kernel
  82  * thread and stdio junk can overflow the limited stack...
  83  */
  84 static char *reads ( int fd, char *buffer, size_t len )
  85 {
  86         int n = read ( fd, buffer, len );
  87
  88         if ( n > 0 ) {
  89                 char *p;
  90
  91                 buffer [len-1] = 0;
  92                 p = strchr ( buffer, '\n' );
  93
  94                 if ( p ) {
  95                         off_t offset;
  96
  97                         offset = lseek ( fd, 0L, SEEK_CUR );               // Get the current file descriptor offset
  98                         lseek ( fd, offset-n + (p-buffer) + 1, SEEK_SET ); // Set the file descriptor offset to right after the \n
  99
 100                         p[1] = 0;
 101                 }
 102                 return buffer;
 103         }
 104
 105         else
 106                 return 0;
 107 }
 108
 109 static struct dep_t *build_dep ( void )
 110 {
 111         int fd;
 112         struct utsname un;
 113         struct dep_t *first = 0;
 114         struct dep_t *current = 0;
 115         char buffer[256];
 116         char *filename = buffer;
 117         int continuation_line = 0;
 118
 119         if ( uname ( &un ))
 120                 return 0;
 121
 122         // check for buffer overflow in following code
 123         if ( bb_strlen ( un.release ) > ( sizeof( buffer ) - 64 )) {
 124                 return 0;
 125         }
 126
 127         strcpy ( filename, "/lib/modules/" );
 128         strcat ( filename, un.release );
 129         strcat ( filename, "/modules.dep" );
 130
 131         if (( fd = open ( filename, O_RDONLY )) < 0 ) {
 132
 133                 /* Ok, that didn't work.  Fall back to looking in /lib/modules */
 134                 if (( fd = open ( "/lib/modules/modules.dep", O_RDONLY )) < 0 ) {
 135                         return 0;
 136                 }
 137         }
 138
 139         while ( reads ( fd, buffer, sizeof( buffer ))) {
 140                 int l = bb_strlen ( buffer );
 141                 char *p = 0;
 142
 143                 while ( isspace ( buffer [l-1] )) {
 144                         buffer [l-1] = 0;
 145                         l--;
 146                 }
 147
 148                 if ( l == 0 ) {
 149                         continuation_line = 0;
 150                         continue;
 151                 }
 152
 153                 if ( !continuation_line ) {
 154                         char *col = strchr ( buffer, ':' );
 155
 156                         if ( col ) {
 157                                 char *mods;
 158                                 char *mod;
 159                                 int ext = 0;
 160
 161                                 *col = 0;
 162                                 mods = strrchr ( buffer, '/' );
 163
 164                                 if ( !mods )
 165                                         mods = buffer;
 166                                 else
 167                                         mods++;
 168
 169                                 if (( *(col-2) == '.' ) && ( *(col-1) == 'o' ))
 170                                         ext = 2;
 171
 172                                 mod = bb_xstrndup ( mods, col - mods - ext );
 173
 174                                 if ( !current ) {
 175                                         first = current = (struct dep_t *) xmalloc ( sizeof ( struct dep_t ));
 176                                 }
 177                                 else {
 178                                         current-> m_next = (struct dep_t *) xmalloc ( sizeof ( struct dep_t ));
 179                                         current = current-> m_next;
 180                                 }
 181                                 current-> m_module  = mod;
 182                                 current-> m_options = 0;
 183                                 current-> m_isalias = 0;
 184                                 current-> m_depcnt  = 0;
 185                                 current-> m_deparr  = 0;
 186                                 current-> m_next    = 0;
 187
 188                                 //printf ( "%s:\n", mod );
 189
 190                                 p = col + 1;
 191                         }
 192                         else
 193                                 p = 0;
 194                 }
 195                 else
 196                         p = buffer;
 197
 198                 if ( p && *p ) {
 199                         char *end = &buffer [l-1];
 200                         char *deps = strrchr ( end, '/' );
 201                         char *dep;
 202                         int ext = 0;
 203
 204                         while ( isblank ( *end ) || ( *end == '\\' ))
 205                                 end--;
 206
 207                         deps = strrchr ( p, '/' );
 208
 209                         if ( !deps || ( deps < p )) {
 210                                 deps = p;
 211
 212                                 while ( isblank ( *deps ))
 213                                         deps++;
 214                         }
 215                         else
 216                                 deps++;
 217
 218                         if (( *(end-1) == '.' ) && ( *end == 'o' ))
 219                                 ext = 2;
 220
 221                         /* Cope with blank lines */
 222                         if ((end-deps-ext+1) <= 0)
 223                                 continue;
 224
 225                         dep = bb_xstrndup ( deps, end - deps - ext + 1 );
 226
 227                         current-> m_depcnt++;
 228                         current-> m_deparr = (char **) xrealloc ( current-> m_deparr, sizeof ( char *) * current-> m_depcnt );
 229                         current-> m_deparr [current-> m_depcnt - 1] = dep;
 230
 231                         //printf ( "    %d) %s\n", current-> m_depcnt, current-> m_deparr [current-> m_depcnt -1] );
 232                 }
 233
 234                 if ( buffer [l-1] == '\\' )
 235                         continuation_line = 1;
 236                 else
 237                         continuation_line = 0;
 238         }
 239         close ( fd );
 240
 241         // alias parsing is not 100% correct (no correct handling of continuation lines within an alias) !
 242
 243         if (( fd = open ( "/etc/modules.conf", O_RDONLY )) < 0 )
 244                 if (( fd = open ( "/etc/conf.modules", O_RDONLY )) < 0 )
 245                         return first;
 246
 247         continuation_line = 0;
 248         while ( reads ( fd, buffer, sizeof( buffer ))) {
 249                 int l;
 250                 char *p;
 251
 252                 p = strchr ( buffer, '#' );
 253                 if ( p )
 254                         *p = 0;
 255
 256                 l = bb_strlen ( buffer );
 257
 258                 while ( l && isspace ( buffer [l-1] )) {
 259                         buffer [l-1] = 0;
 260                         l--;
 261                 }
 262
 263                 if ( l == 0 ) {
 264                         continuation_line = 0;
 265                         continue;
 266                 }
 267
 268                 if ( !continuation_line ) {
 269                         if (( strncmp ( buffer, "alias", 5 ) == 0 ) && isspace ( buffer [5] )) {
 270                                 char *alias, *mod;
 271
 272                                 if ( parse_tag_value ( buffer + 6, &alias, &mod )) {
 273                                         // fprintf ( stderr, "ALIAS: '%s' -> '%s'\n", alias, mod );
 274
 275                                         if ( !current ) {
 276                                                 first = current = (struct dep_t *) xmalloc ( sizeof ( struct dep_t ));
 277                                         }
 278                                         else {
 279                                                 current-> m_next = (struct dep_t *) xmalloc ( sizeof ( struct dep_t ));
 280                                                 current = current-> m_next;
 281                                         }
 282                                         current-> m_module  = bb_xstrdup ( alias );
 283                                         current-> m_isalias = 1;
 284
 285                                         if (( strcmp ( alias, "off" ) == 0 ) || ( strcmp ( alias, "null" ) == 0 )) {
 286                                                 current-> m_depcnt = 0;
 287                                                 current-> m_deparr = 0;
 288                                         }
 289                                         else {
 290                                                 current-> m_depcnt  = 1;
 291                                                 current-> m_deparr  = xmalloc ( 1 * sizeof( char * ));
 292                                                 current-> m_deparr[0] = bb_xstrdup ( mod );
 293                                         }
 294                                         current-> m_next    = 0;
 295                                 }
 296                         }
 297                         else if (( strncmp ( buffer, "options", 7 ) == 0 ) && isspace ( buffer [7] )) {
 298                                 char *mod, *opt;
 299
 300                                 if ( parse_tag_value ( buffer + 8, &mod, &opt )) {
 301                                         struct dep_t *dt;
 302
 303                                         for ( dt = first; dt; dt = dt-> m_next ) {
 304                                                 if ( strcmp ( dt-> m_module, mod ) == 0 )
 305                                                         break;
 306                                         }
 307                                         if ( dt ) {
 308                                                 dt-> m_options = xrealloc ( dt-> m_options, bb_strlen( opt ) + 1 );
 309                                                 strcpy ( dt-> m_options, opt );
 310
 311                                                 // fprintf ( stderr, "OPTION: '%s' -> '%s'\n", dt-> m_module, dt-> m_options );
 312                                         }
 313                                 }
 314                         }
 315                 }
 316         }
 317         close ( fd );
 318
 319         return first;
 320 }
 321
 322 /* return 1 = loaded, 0 = not loaded, -1 = can't tell */
 323 static int already_loaded (const char *name)
 324 {
 325         int fd;
 326         char buffer[256];
 327
 328         fd = open ("/proc/modules", O_RDONLY);
 329         if (fd < 0)
 330                 return -1;
 331
 332         while ( reads ( fd, buffer, sizeof( buffer ))) {
 333                 char *p;
 334
 335                 p = strchr (buffer, ' ');
 336                 if (p) {
 337                         *p = 0;
 338                         if (strcmp (name, buffer) == 0) {
 339                                 close (fd);
 340                                 return 1;
 341                         }
 342                 }
 343         }
 344
 345         close (fd);
 346         return 0;
 347 }
 348
 349 static int mod_process ( struct mod_list_t *list, int do_insert )
 350 {
 351         char lcmd [256];
 352         int rc = 1;
 353
 354         while ( list ) {
 355                 *lcmd = '\0';
 356                 if ( do_insert ) {
 357                         if (already_loaded (list->m_module) != 1)
 358                                 snprintf ( lcmd, sizeof( lcmd ) - 1, "insmod %s %s %s %s %s", do_syslog ? "-s" : "", autoclean ? "-k" : "", quiet ? "-q" : "", list-> m_module, list-> m_options ? list-> m_options : "" );
 359                 } else {
 360                         if (already_loaded (list->m_module) != 0)
 361                                 snprintf ( lcmd, sizeof( lcmd ) - 1, "rmmod %s %s", do_syslog ? "-s" : "", list-> m_module );
 362                 }
 363
 364                 if ( verbose )
 365                         printf ( "%s\n", lcmd );
 366                 if ( !show_only && *lcmd) {
 367                         int rc2 = system ( lcmd );
 368                         if (do_insert) rc = rc2; /* only last module matters */
 369                         else if (!rc2) rc = 0; /* success if remove any mod */
 370                 }
 371
 372                 list = do_insert ? list-> m_prev : list-> m_next;
 373         }
 374         return (show_only) ? 0 : rc;
 375 }
 376
 377 static void check_dep ( char *mod, struct mod_list_t **head, struct mod_list_t **tail )
 378 {
 379         struct mod_list_t *find;
 380         struct dep_t *dt;
 381         char *opt = 0;
 382         int lm;
 383
 384         // remove .o extension
 385         lm = bb_strlen ( mod );
 386         if (( mod [lm-2] == '.' ) && ( mod [lm-1] == 'o' ))
 387                 mod [lm-2] = 0;
 388
 389         // check dependencies
 390         for ( dt = depend; dt; dt = dt-> m_next ) {
 391                 if ( strcmp ( dt-> m_module, mod ) == 0 ) {
 392                         opt = dt-> m_options;
 393                         break;
 394                 }
 395         }
 396
 397         // resolve alias names
 398         while ( dt && dt-> m_isalias ) {
 399                 if ( dt-> m_depcnt == 1 ) {
 400                         struct dep_t *adt;
 401
 402                         for ( adt = depend; adt; adt = adt-> m_next ) {
 403                                 if ( strcmp ( adt-> m_module, dt-> m_deparr [0] ) == 0 )
 404                                         break;
 405                         }
 406                         if ( adt ) {
 407                                 dt = adt;
 408                                 mod = dt-> m_module;
 409                                 if ( !opt )
 410                                         opt = dt-> m_options;
 411                         }
 412                         else
 413                                 return;
 414                 }
 415                 else
 416                         return;
 417         }
 418
 419         // search for duplicates
 420         for ( find = *head; find; find = find-> m_next ) {
 421                 if ( !strcmp ( mod, find-> m_module )) {
 422                         // found -> dequeue it
 423
 424                         if ( find-> m_prev )
 425                                 find-> m_prev-> m_next = find-> m_next;
 426                         else
 427                                 *head = find-> m_next;
 428
 429                         if ( find-> m_next )
 430                                 find-> m_next-> m_prev = find-> m_prev;
 431                         else
 432                                 *tail = find-> m_prev;
 433
 434                         break; // there can be only one duplicate
 435                 }
 436         }
 437
 438         if ( !find ) { // did not find a duplicate
 439                 find = (struct mod_list_t *) xmalloc ( sizeof(struct mod_list_t));
 440                 find-> m_module = mod;
 441                 find-> m_options = opt;
 442         }
 443
 444         // enqueue at tail
 445         if ( *tail )
 446                 (*tail)-> m_next = find;
 447         find-> m_prev   = *tail;
 448         find-> m_next   = 0;
 449
 450         if ( !*head )
 451                 *head = find;
 452         *tail = find;
 453
 454         if ( dt ) {
 455                 int i;
 456
 457                 for ( i = 0; i < dt-> m_depcnt; i++ )
 458                         check_dep ( dt-> m_deparr [i], head, tail );
 459         }
 460 }
 461
 462
 463
 464 static int mod_insert ( char *mod, int argc, char **argv )
 465 {
 466         struct mod_list_t *tail = 0;
 467         struct mod_list_t *head = 0;
 468         int rc;
 469
 470         // get dep list for module mod
 471         check_dep ( mod, &head, &tail );
 472
 473         if ( head && tail ) {
 474                 if ( argc ) {
 475                         int i;
 476                         int l = 0;
 477
 478                         // append module args
 479                         for ( i = 0; i < argc; i++ )
 480                                 l += ( bb_strlen ( argv [i] ) + 1 );
 481
 482                         head-> m_options = xrealloc ( head-> m_options, l + 1 );
 483                         head-> m_options [0] = 0;
 484
 485                         for ( i = 0; i < argc; i++ ) {
 486                                 strcat ( head-> m_options, argv [i] );
 487                                 strcat ( head-> m_options, " " );
 488                         }
 489                 }
 490
 491                 // process tail ---> head
 492                 rc = mod_process ( tail, 1 );
 493         }
 494         else
 495                 rc = 1;
 496
 497         return rc;
 498 }
 499
 500 static int mod_remove ( char *mod )
 501 {
 502         int rc;
 503         static struct mod_list_t rm_a_dummy = { "-a", 0, 0 };
 504
 505         struct mod_list_t *head = 0;
 506         struct mod_list_t *tail = 0;
 507
 508         if ( mod )
 509                 check_dep ( mod, &head, &tail );
 510         else  // autoclean
 511                 head = tail = &rm_a_dummy;
 512
 513         if ( head && tail )
 514                 rc = mod_process ( head, 0 );  // process head ---> tail
 515         else
 516                 rc = 1;
 517         return rc;
 518
 519 }
 520
 521
 522
 523 extern int modprobe_main(int argc, char** argv)
 524 {
 525         int     opt;
 526         int remove_opt = 0;
 527
 528         autoclean = show_only = quiet = do_syslog = verbose = 0;
 529
 530         while ((opt = getopt(argc, argv, "acdklnqrst:vVC:")) != -1) {
 531                 switch(opt) {
 532                 case 'c': // no config used
 533                 case 'l': // no pattern matching
 534                         return EXIT_SUCCESS;
 535                         break;
 536                 case 'C': // no config used
 537                 case 't': // no pattern matching
 538                         bb_error_msg_and_die("-t and -C not supported");
 539
 540                 case 'a': // ignore
 541                 case 'd': // ignore
 542                         break;
 543                 case 'k':
 544                         autoclean++;
 545                         break;
 546                 case 'n':
 547                         show_only++;
 548                         break;
 549                 case 'q':
 550                         quiet++;
 551                         break;
 552                 case 'r':
 553                         remove_opt++;
 554                         break;
 555                 case 's':
 556                         do_syslog++;
 557                         break;
 558                 case 'v':
 559                         verbose++;
 560                         break;
 561                 case 'V':
 562                 default:
 563                         bb_show_usage();
 564                         break;
 565                 }
 566         }
 567
 568         depend = build_dep ( );
 569
 570         if ( !depend )
 571                 bb_error_msg_and_die ( "could not parse modules.dep\n" );
 572
 573         if (remove_opt) {
 574                 int rc = EXIT_SUCCESS;
 575                 do {
 576                         if (mod_remove ( optind < argc ?
 577                                          bb_xstrdup (argv [optind]) : NULL )) {
 578                                 bb_error_msg ("failed to remove module %s",
 579                                         argv [optind] );
 580                                 rc = EXIT_FAILURE;
 581                         }
 582                 } while ( ++optind < argc );
 583
 584                 return rc;
 585         }
 586
 587         if (optind >= argc)
 588                 bb_error_msg_and_die ( "No module or pattern provided\n" );
 589
 590         if ( mod_insert ( bb_xstrdup ( argv [optind] ), argc - optind - 1, argv + optind + 1 ))
 591                 bb_error_msg_and_die ( "failed to load module %s", argv [optind] );
 592
 593         return EXIT_SUCCESS;
 594 }
 595
 596