release/sysinstall/modules.c

   1 /*-
   2  * Copyright (c) 2000  "HOSOKAWA, Tatsumi" <hosokawa@FreeBSD.org>
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  24  * SUCH DAMAGE.
  25  *
  26  * $FreeBSD: src/release/sysinstall/modules.c,v 1.2.2.4 2002/06/05 20:13:50 jhb Exp $
  27  * $DragonFly: src/release/sysinstall/Attic/modules.c,v 1.2 2003/06/17 04:27:21 dillon Exp $
  28  */
  29
  30 #include "sysinstall.h"
  31 #include <stdio.h>
  32 #include <string.h>
  33 #include <sys/types.h>
  34 #include <sys/param.h>
  35 #include <sys/linker.h>
  36 #include <fcntl.h>
  37 #include <dirent.h>
  38 #include <fcntl.h>
  39 #include <fnmatch.h>
  40
  41 /* Prototypes */
  42 static int              kldModuleFire(dialogMenuItem *self);
  43
  44 #define MODULESDIR "/stand/modules"
  45 #define DISTMOUNT "/dist"
  46
  47 void
  48 moduleInitialize(void)
  49 {
  50     int fd, len;
  51     DIR *dirp;
  52     struct dirent *dp;
  53     char module[MAXPATHLEN], desc[MAXPATHLEN];
  54     char desc_str[BUFSIZ];
  55
  56     if (!RunningAsInit && !Fake) {
  57         /* It's not my job... */
  58         return;
  59     }
  60
  61     dirp = opendir(MODULESDIR);
  62     if (dirp) {
  63         while ((dp = readdir(dirp))) {
  64             if (dp->d_namlen < (sizeof(".ko") - 1)) continue;
  65             if (strcmp(dp->d_name + dp->d_namlen - (sizeof(".ko") - 1), ".ko") == 0) {
  66                 strcpy(module, MODULESDIR);
  67                 strcat(module, "/");
  68                 strcat(module, dp->d_name);
  69                 strcpy(desc, module);
  70                 len = strlen(desc);
  71                 strcpy(desc + (len - (sizeof(".ko") - 1)), ".dsc");
  72                 fd = open(module, O_RDONLY);
  73                 if (fd < 0) continue;
  74                 close(fd);
  75                 fd = open(desc, O_RDONLY);
  76                 if (fd < 0) {
  77                     desc_str[0] = 0;
  78                 }
  79                 else {
  80                     len = read(fd, desc_str, BUFSIZ);
  81                     close(fd);
  82                     if (len < BUFSIZ) desc_str[len] = 0;
  83                 }
  84                 if (desc_str[0])
  85                     msgDebug("Loading module %s (%s)\n", dp->d_name, desc_str);
  86                 else
  87                     msgDebug("Loading module %s\n", dp->d_name);
  88                 if (kldload(module) < 0 && errno != EEXIST) {
  89                     if (desc_str[0])
  90                         msgConfirm("Loading module %s failed\n%s", dp->d_name, desc_str);
  91                     else
  92                         msgConfirm("Loading module %s failed", dp->d_name);
  93                 }
  94             }
  95         }
  96         closedir(dirp);
  97     }
  98 }
  99
 100 int
 101 kldBrowser(dialogMenuItem *self)
 102 {
 103     DMenu       *menu;
 104     int         i, what = DITEM_SUCCESS, msize, count;
 105     DIR         *dir;
 106     struct dirent *de;
 107     char        *err;
 108
 109     err = NULL;
 110
 111     if (DITEM_STATUS(mediaSetFloppy(NULL)) == DITEM_FAILURE) {
 112         msgConfirm("Unable to set media device to floppy.");
 113         what |= DITEM_FAILURE;
 114         mediaClose();
 115         return what;
 116     }
 117
 118     if (!DEVICE_INIT(mediaDevice)) {
 119         msgConfirm("Unable to mount floppy filesystem.");
 120         what |= DITEM_FAILURE;
 121         mediaClose();
 122         return what;
 123     }
 124
 125     msize = sizeof(DMenu) + (sizeof(dialogMenuItem) * 2);
 126     count = 0;
 127     if ((menu = malloc(msize)) == NULL) {
 128         err = "Failed to allocate memory for menu";
 129         goto errout;
 130     }
 131
 132     bcopy(&MenuKLD, menu, sizeof(DMenu));
 133
 134     bzero(&menu->items[count], sizeof(menu->items[0]));
 135     menu->items[count].prompt = strdup("X Exit");
 136     menu->items[count].title = strdup("Exit this menu (returning to previous)");
 137     menu->items[count].fire = dmenuExit;
 138     count++;
 139
 140     if ((dir = opendir(DISTMOUNT)) == NULL) {
 141         err = "Couldn't open directory";
 142         goto errout;
 143     }
 144
 145     while ((de = readdir(dir)) != NULL) {
 146         if (fnmatch("*.ko", de->d_name, FNM_CASEFOLD))
 147             continue;
 148
 149         msize += sizeof(dialogMenuItem);
 150         if ((menu = realloc(menu, msize)) == NULL) {
 151             err = "Failed to allocate memory for menu item";
 152             goto errout;
 153         }
 154
 155         bzero(&menu->items[count], sizeof(menu->items[0]));
 156         menu->items[count].fire = kldModuleFire;
 157
 158         menu->items[count].prompt = strdup(de->d_name);
 159         menu->items[count].title = menu->items[count].prompt;
 160
 161         count++;
 162     }
 163
 164     closedir(dir);
 165
 166     menu->items[count].prompt = NULL;
 167     menu->items[count].title = NULL;
 168
 169     dmenuOpenSimple(menu, FALSE);
 170
 171     mediaClose();
 172
 173     deviceRescan();
 174
 175   errout:
 176     for (i = 0; i < count; i++)
 177         free(menu->items[i].prompt);
 178
 179     free(menu);
 180
 181     if (err != NULL) {
 182         what |= DITEM_FAILURE;
 183         if (!variable_get(VAR_NO_ERROR))
 184             msgConfirm(err);
 185     }
 186
 187     return (what);
 188 }
 189
 190 static int
 191 kldModuleFire(dialogMenuItem *self) {
 192     char        fname[256];
 193
 194     bzero(fname, sizeof(fname));
 195     snprintf(fname, sizeof(fname), "%s/%s", DISTMOUNT, self->prompt);
 196
 197     if (kldload(fname) < 0 && errno != EEXIST) {
 198         if (!variable_get(VAR_NO_ERROR))
 199             msgConfirm("Loading module %s failed\n", fname);
 200     } else {
 201         if (!variable_get(VAR_NO_ERROR))
 202             msgConfirm("Loaded module %s OK", fname);
 203     }
 204
 205     return(0);
 206  }