usr.sbin/sysinstall/devices.c

   1 /*
   2  * The new sysinstall program.
   3  *
   4  * This is probably the last program in the `sysinstall' line - the next
   5  * generation being essentially a complete rewrite.
   6  *
   7  * $FreeBSD$
   8  *
   9  * Copyright (c) 1995
  10  *      Jordan Hubbard.  All rights reserved.
  11  *
  12  * Redistribution and use in source and binary forms, with or without
  13  * modification, are permitted provided that the following conditions
  14  * are met:
  15  * 1. Redistributions of source code must retain the above copyright
  16  *    notice, this list of conditions and the following disclaimer,
  17  *    verbatim and that no modifications are made prior to this
  18  *    point in the file.
  19  * 2. Redistributions in binary form must reproduce the above copyright
  20  *    notice, this list of conditions and the following disclaimer in the
  21  *    documentation and/or other materials provided with the distribution.
  22  *
  23  * THIS SOFTWARE IS PROVIDED BY JORDAN HUBBARD ``AS IS'' AND
  24  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  25  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  26  * ARE DISCLAIMED.  IN NO EVENT SHALL JORDAN HUBBARD OR HIS PETS BE LIABLE
  27  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  29  * OR SERVICES; LOSS OF USE, DATA, LIFE OR PROFITS; OR BUSINESS INTERRUPTION)
  30  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  31  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  32  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  33  * SUCH DAMAGE.
  34  *
  35  */
  36
  37 #include "sysinstall.h"
  38 #include <sys/fcntl.h>
  39 #include <sys/param.h>
  40 #include <sys/socket.h>
  41 #include <sys/ioctl.h>
  42 #include <sys/errno.h>
  43 #include <sys/time.h>
  44 #include <net/if.h>
  45 #include <net/if_var.h>
  46 #include <net/if_dl.h>
  47 #include <netinet/in.h>
  48 #include <netinet/in_var.h>
  49 #include <arpa/inet.h>
  50 #include <ctype.h>
  51
  52 /* how much to bias minor number for a given /dev/<ct#><un#>s<s#> slice */
  53 #define SLICE_DELTA     (0x10000)
  54
  55 static Device *Devices[DEV_MAX];
  56 static int numDevs;
  57
  58 static struct _devname {
  59     DeviceType type;
  60     char *name;
  61     char *description;
  62     int major, minor, delta, max;
  63 } device_names[] = {
  64     { DEVICE_TYPE_CDROM,        "cd%dc",        "SCSI CDROM drive",     15, 2, 8, 4                             },
  65     { DEVICE_TYPE_CDROM,        "mcd%da",       "Mitsumi (old model) CDROM drive",      29, 0, 8, 4             },
  66     { DEVICE_TYPE_CDROM,        "scd%da",       "Sony CDROM drive - CDU31/33A type",    45, 0, 8, 4             },
  67 #ifdef notdef
  68     { DEVICE_TYPE_CDROM,        "matcd%da",     "Matsushita CDROM ('sound blaster' type)", 46, 0, 8, 4          },
  69 #endif
  70     { DEVICE_TYPE_CDROM,        "acd%dc",       "ATAPI/IDE CDROM",      117, 2, 8, 4                            },
  71     { DEVICE_TYPE_TAPE,         "rsa%d",        "SCSI tape drive",      14, 0, 16, 4                            },
  72     { DEVICE_TYPE_TAPE,         "rwt%d",        "Wangtek tape drive",   10, 0, 1, 4                             },
  73     { DEVICE_TYPE_DISK,         "da%d",         "SCSI disk device",     13, 65538, 8, 16                        },
  74     { DEVICE_TYPE_DISK,         "ad%d",         "ATA/IDE disk device",  116, 65538, 8, 16                       },
  75     { DEVICE_TYPE_DISK,         "ar%d",         "ATA/IDE RAID device",  157, 65538, 8, 16                       },
  76     { DEVICE_TYPE_DISK,         "fla%d",        "M-Systems DiskOnChip Flash devicee",   102, 65538, 8, 16       },
  77     { DEVICE_TYPE_DISK,         "afd%d",        "ATAPI/IDE floppy device",      118, 65538, 8, 4                },
  78     { DEVICE_TYPE_DISK,         "mlxd%d",       "Mylex RAID disk",      131, 65538, 8, 4                        },
  79     { DEVICE_TYPE_DISK,         "amrd%d",       "AMI MegaRAID drive",   133, 65538, 8, 4                        },
  80     { DEVICE_TYPE_DISK,         "idad%d",       "Compaq RAID array",    109, 65538, 8, 4                        },
  81     { DEVICE_TYPE_DISK,         "twed%d",       "3ware ATA RAID array", 147, 65538, 8, 4                        },
  82     { DEVICE_TYPE_DISK,         "aacd%d",       "Adaptec FSA RAID array", 151, 65538, 8, 4                      },
  83     { DEVICE_TYPE_FLOPPY,       "fd%d",         "floppy drive unit A",  9, 0, 64, 4                             },
  84     { DEVICE_TYPE_NETWORK,      "aue",          "ADMtek USB ethernet adapter"                                   },
  85     { DEVICE_TYPE_NETWORK,      "bge",          "Broadcom BCM570x PCI gigabit ethernet card"                            },
  86     { DEVICE_TYPE_NETWORK,      "cue",          "CATC USB ethernet adapter"                                     },
  87     { DEVICE_TYPE_NETWORK,      "fpa",          "DEC DEFPA PCI FDDI card"                                       },
  88     { DEVICE_TYPE_NETWORK,      "sr",           "SDL T1/E1 sync serial PCI card"                                },
  89     { DEVICE_TYPE_NETWORK,      "cc3i",         "SDL HSSI sync serial PCI card"                                 },
  90     { DEVICE_TYPE_NETWORK,      "en",           "Efficient Networks ATM PCI card"                               },
  91     { DEVICE_TYPE_NETWORK,      "dc",           "DEC/Intel 21143 (and clones) PCI fast ethernet card"           },
  92     { DEVICE_TYPE_NETWORK,      "de",           "DEC DE435 PCI NIC or other DC21040-AA based card"              },
  93     { DEVICE_TYPE_NETWORK,      "fxp",          "Intel EtherExpress Pro/100B PCI Fast Ethernet card"            },
  94     { DEVICE_TYPE_NETWORK,      "ed",           "Novell NE1000/2000; 3C503; NE2000-compatible PCMCIA"           },
  95     { DEVICE_TYPE_NETWORK,      "ep",           "3Com 3C509 ethernet card/3C589 PCMCIA"                         },
  96     { DEVICE_TYPE_NETWORK,      "el",           "3Com 3C501 ethernet card"                                      },
  97     { DEVICE_TYPE_NETWORK,      "ex",           "Intel EtherExpress Pro/10 ethernet card"                       },
  98     { DEVICE_TYPE_NETWORK,      "fe",           "Fujitsu MB86960A/MB86965A ethernet card"                       },
  99     { DEVICE_TYPE_NETWORK,      "ie",           "AT&T StarLAN 10 and EN100; 3Com 3C507; NI5210"                 },
 100     { DEVICE_TYPE_NETWORK,      "ix",           "Intel Etherexpress ethernet card"                              },
 101     { DEVICE_TYPE_NETWORK,      "kue",          "Kawasaki LSI USB ethernet adapter"                             },
 102     { DEVICE_TYPE_NETWORK,      "le",           "DEC EtherWorks 2 or 3 ethernet card"                           },
 103     { DEVICE_TYPE_NETWORK,      "lnc",          "Lance/PCnet (Isolan/Novell NE2100/NE32-VL) ethernet"           },
 104     { DEVICE_TYPE_NETWORK,      "nge",          "NatSemi PCI gigabit ethernet card"                             },
 105     { DEVICE_TYPE_NETWORK,      "pcn",          "AMD Am79c79x PCI ethernet card"                                },
 106     { DEVICE_TYPE_NETWORK,      "rl",           "RealTek 8129/8139 PCI ethernet card"                           },
 107     { DEVICE_TYPE_NETWORK,      "sf",           "Adaptec AIC-6915 PCI ethernet card"                            },
 108     { DEVICE_TYPE_NETWORK,      "sis",          "SiS 900/SiS 7016 PCI ethernet card"                            },
 109 #ifdef PC98
 110     { DEVICE_TYPE_NETWORK,      "snc",          "SONIC ethernet card"                                           },
 111 #endif
 112     { DEVICE_TYPE_NETWORK,      "sn",           "SMC/Megahertz ethernet card"                                   },
 113     { DEVICE_TYPE_NETWORK,      "ste",          "Sundance ST201 PCI ethernet card"                              },
 114     { DEVICE_TYPE_NETWORK,      "sk",           "SysKonnect PCI gigabit ethernet card"                          },
 115     { DEVICE_TYPE_NETWORK,      "tx",           "SMC 9432TX ethernet card"                                      },
 116     { DEVICE_TYPE_NETWORK,      "txp",          "3Com 3cR990 ethernet card"                                     },
 117     { DEVICE_TYPE_NETWORK,      "ti",           "Alteon Networks PCI gigabit ethernet card"                     },
 118     { DEVICE_TYPE_NETWORK,      "tl",           "Texas Instruments ThunderLAN PCI ethernet card"                },
 119     { DEVICE_TYPE_NETWORK,      "vr",           "VIA VT3043/VT86C100A Rhine PCI ethernet card"                  },
 120     { DEVICE_TYPE_NETWORK,      "vx",           "3COM 3c590 / 3c595 ethernet card"                              },
 121     { DEVICE_TYPE_NETWORK,      "wb",           "Winbond W89C840F PCI ethernet card"                            },
 122     { DEVICE_TYPE_NETWORK,      "wx",           "Intel Gigabit Ethernet (82452) card"                   },
 123     { DEVICE_TYPE_NETWORK,      "xe",           "Xircom/Intel EtherExpress Pro100/16 ethernet card"             },
 124     { DEVICE_TYPE_NETWORK,      "xl",           "3COM 3c90x / 3c90xB PCI ethernet card"                         },
 125     { DEVICE_TYPE_NETWORK,      "cuaa%d",       "%s on device %s (COM%d)",      28, 128, 1, 16                  },
 126     { DEVICE_TYPE_NETWORK,      "lp",           "Parallel Port IP (PLIP) peer connection"                       },
 127     { DEVICE_TYPE_NETWORK,      "lo",           "Loop-back (local) network interface"                           },
 128 #ifdef PC98
 129     { DEVICE_TYPE_DISK,         "wd%d",         "IDE disk device",              3, 65538, 8, 16                 },
 130     { DEVICE_TYPE_CDROM,        "wcd%dc",       "ATAPI IDE CDROM",              69, 2, 8, 4                     },
 131     { DEVICE_TYPE_FLOPPY,       "wfd%d",        "ATAPI floppy drive unit A",    87, 0, 8, 4                     },
 132     { DEVICE_TYPE_DISK,         "wfd%d",        "ATAPI floppy device",          87, 65538, 8, 4                 },
 133 #endif
 134     { 0 },
 135 };
 136
 137 Device *
 138 new_device(char *name)
 139 {
 140     Device *dev;
 141
 142     dev = safe_malloc(sizeof(Device));
 143     bzero(dev, sizeof(Device));
 144     if (name)
 145         SAFE_STRCPY(dev->name, name);
 146     return dev;
 147 }
 148
 149 /* Stubs for unimplemented strategy routines */
 150 Boolean
 151 dummyInit(Device *dev)
 152 {
 153     return TRUE;
 154 }
 155
 156 FILE *
 157 dummyGet(Device *dev, char *dist, Boolean probe)
 158 {
 159     return NULL;
 160 }
 161
 162 void
 163 dummyShutdown(Device *dev)
 164 {
 165     return;
 166 }
 167
 168 static int
 169 deviceTry(struct _devname dev, char *try, int i)
 170 {
 171     int fd;
 172     char unit[80];
 173     mode_t m;
 174     dev_t d;
 175     int fail;
 176
 177     snprintf(unit, sizeof unit, dev.name, i);
 178     snprintf(try, FILENAME_MAX, "/dev/%s", unit);
 179     if (isDebug())
 180         msgDebug("deviceTry: attempting to open %s\n", try);
 181     fd = open(try, O_RDONLY);
 182     if (fd >= 0) {
 183         if (isDebug())
 184             msgDebug("deviceTry: open of %s succeeded on first try.\n", try);
 185         return fd;
 186     }
 187     m = 0640 | S_IFCHR;
 188     d = makedev(dev.major, dev.minor + (i * dev.delta));
 189     if (isDebug())
 190         msgDebug("deviceTry: Making %s device for %s [%d, %d]\n", m & S_IFCHR ? "raw" : "block", try, dev.major, dev.minor + (i * dev.delta));
 191     fail = mknod(try, m, d);
 192     fd = open(try, O_RDONLY);
 193     if (fd >= 0) {
 194         if (isDebug())
 195             msgDebug("deviceTry: open of %s succeeded on second try.\n", try);
 196         return fd;
 197     }
 198     else if (!fail)
 199         (void)unlink(try);
 200     /* Don't try a "make-under" here since we're using a fixit floppy in this case */
 201     snprintf(try, FILENAME_MAX, "/mnt/dev/%s", unit);
 202     fd = open(try, O_RDONLY);
 203     if (isDebug())
 204         msgDebug("deviceTry: final attempt for %s returns %d\n", try, fd);
 205     return fd;
 206 }
 207
 208 /* Register a new device in the devices array */
 209 Device *
 210 deviceRegister(char *name, char *desc, char *devname, DeviceType type, Boolean enabled,
 211                Boolean (*init)(Device *), FILE * (*get)(Device *, char *, Boolean),
 212                void (*shutdown)(Device *), void *private)
 213 {
 214     Device *newdev = NULL;
 215
 216     if (numDevs == DEV_MAX)
 217         msgFatal("Too many devices found!");
 218     else {
 219         newdev = new_device(name);
 220         newdev->description = desc;
 221         newdev->devname = devname;
 222         newdev->type = type;
 223         newdev->enabled = enabled;
 224         newdev->init = init ? init : dummyInit;
 225         newdev->get = get ? get : dummyGet;
 226         newdev->shutdown = shutdown ? shutdown : dummyShutdown;
 227         newdev->private = private;
 228         Devices[numDevs] = newdev;
 229         Devices[++numDevs] = NULL;
 230     }
 231     return newdev;
 232 }
 233
 234 /* Reset the registered device chain */
 235 void
 236 deviceReset(void)
 237 {
 238     int i;
 239
 240     for (i = 0; i < numDevs; i++) {
 241         DEVICE_SHUTDOWN(Devices[i]);
 242
 243         /* XXX this potentially leaks Devices[i]->private if it's being
 244          * used to point to something dynamic, but you're not supposed
 245          * to call this routine at such times that some open instance
 246          * has its private ptr pointing somewhere anyway. XXX
 247          */
 248         free(Devices[i]);
 249     }
 250     Devices[numDevs = 0] = NULL;
 251 }
 252
 253 /* Get all device information for devices we have attached */
 254 void
 255 deviceGetAll(void)
 256 {
 257     int i, j, fd, s;
 258     struct ifconf ifc;
 259     struct ifreq *ifptr, *end;
 260     int ifflags;
 261     char buffer[INTERFACE_MAX * sizeof(struct ifreq)];
 262     char **names;
 263
 264     msgNotify("Probing devices, please wait (this can take a while)...");
 265     /* First go for the network interfaces.  Stolen shamelessly from ifconfig! */
 266     ifc.ifc_len = sizeof(buffer);
 267     ifc.ifc_buf = buffer;
 268
 269     s = socket(AF_INET, SOCK_DGRAM, 0);
 270     if (s < 0)
 271         goto skipif;    /* Jump over network iface probing */
 272
 273     if (ioctl(s, SIOCGIFCONF, (char *) &ifc) < 0)
 274         goto skipif;    /* Jump over network iface probing */
 275
 276     close(s);
 277     ifflags = ifc.ifc_req->ifr_flags;
 278     end = (struct ifreq *) (ifc.ifc_buf + ifc.ifc_len);
 279     for (ifptr = ifc.ifc_req; ifptr < end; ifptr++) {
 280         char *descr;
 281
 282         /* If it's not a link entry, forget it */
 283         if (ifptr->ifr_ifru.ifru_addr.sa_family != AF_LINK)
 284             goto loopend;
 285
 286         /* Eliminate network devices that don't make sense */
 287         if (!strncmp(ifptr->ifr_name, "lo", 2))
 288             goto loopend;
 289
 290         /* If we have a slip device, don't register it */
 291         if (!strncmp(ifptr->ifr_name, "sl", 2)) {
 292             goto loopend;
 293         }
 294         /* And the same for ppp */
 295         if (!strncmp(ifptr->ifr_name, "tun", 3) || !strncmp(ifptr->ifr_name, "ppp", 3)) {
 296             goto loopend;
 297         }
 298         /* Try and find its description */
 299         for (i = 0, descr = NULL; device_names[i].name; i++) {
 300             int len = strlen(device_names[i].name);
 301
 302             if (!ifptr->ifr_name || !ifptr->ifr_name[0])
 303                 continue;
 304             else if (!strncmp(ifptr->ifr_name, device_names[i].name, len)) {
 305                 descr = device_names[i].description;
 306                 break;
 307             }
 308         }
 309         if (!descr)
 310             descr = "<unknown network interface type>";
 311
 312         deviceRegister(ifptr->ifr_name, descr, strdup(ifptr->ifr_name), DEVICE_TYPE_NETWORK, TRUE,
 313                        mediaInitNetwork, NULL, mediaShutdownNetwork, NULL);
 314         if (isDebug())
 315             msgDebug("Found a network device named %s\n", ifptr->ifr_name);
 316         close(s);
 317         if ((s = socket(AF_INET, SOCK_DGRAM, 0)) < 0)
 318             continue;
 319
 320 loopend:
 321         if (ifptr->ifr_addr.sa_len)     /* I'm not sure why this is here - it's inherited */
 322             ifptr = (struct ifreq *)((caddr_t)ifptr + ifptr->ifr_addr.sa_len - sizeof(struct sockaddr));
 323         close(s);
 324     }
 325
 326 skipif:
 327     /* Next, try to find all the types of devices one might need
 328      * during the second stage of the installation.
 329      */
 330     for (i = 0; device_names[i].name; i++) {
 331         for (j = 0; j < device_names[i].max; j++) {
 332             char try[FILENAME_MAX];
 333
 334             switch(device_names[i].type) {
 335             case DEVICE_TYPE_CDROM:
 336                 fd = deviceTry(device_names[i], try, j);
 337                 if (fd >= 0 || errno == EBUSY) {        /* EBUSY if already mounted */
 338                     char n[BUFSIZ];
 339
 340                     if (fd >= 0) close(fd);
 341                     snprintf(n, sizeof n, device_names[i].name, j);
 342                     deviceRegister(strdup(n), device_names[i].description, strdup(try),
 343                                          DEVICE_TYPE_CDROM, TRUE, mediaInitCDROM, mediaGetCDROM,
 344                                          mediaShutdownCDROM, NULL);
 345                     if (isDebug())
 346                         msgDebug("Found a CDROM device for %s\n", try);
 347                 }
 348                 break;
 349
 350             case DEVICE_TYPE_TAPE:
 351                 fd = deviceTry(device_names[i], try, j);
 352                 if (fd >= 0) {
 353                     char n[BUFSIZ];
 354
 355                     close(fd);
 356                     snprintf(n, sizeof n, device_names[i].name, j);
 357                     deviceRegister(strdup(n), device_names[i].description, strdup(try),
 358                                    DEVICE_TYPE_TAPE, TRUE, mediaInitTape, mediaGetTape, mediaShutdownTape, NULL);
 359                     if (isDebug())
 360                         msgDebug("Found a TAPE device for %s\n", try);
 361                 }
 362                 break;
 363
 364             case DEVICE_TYPE_DISK:
 365                 fd = deviceTry(device_names[i], try, j);
 366                 if (fd >= 0 && RunningAsInit) {
 367                     dev_t d;
 368                     mode_t m;
 369                     int s, fail;
 370                     char unit[80], slice[80];
 371
 372                     close(fd);
 373                     /* Make associated slice entries */
 374                     for (s = 1; s < 8; s++) {
 375                         snprintf(unit, sizeof unit, device_names[i].name, j);
 376                         snprintf(slice, sizeof slice, "/dev/%ss%d", unit, s);
 377                         d = makedev(device_names[i].major, device_names[i].minor +
 378                                     (j * device_names[i].delta) + (s * SLICE_DELTA));
 379                         m = 0640 | S_IFCHR;
 380                         fail = mknod(slice, m, d);
 381                         fd = open(slice, O_RDONLY);
 382                         if (fd >= 0)
 383                             close(fd);
 384                         else if (!fail)
 385                             (void)unlink(slice);
 386                     }
 387                 }
 388                 break;
 389
 390             case DEVICE_TYPE_FLOPPY:
 391                 fd = deviceTry(device_names[i], try, j);
 392                 if (fd >= 0) {
 393                     char n[BUFSIZ];
 394
 395                     close(fd);
 396                     snprintf(n, sizeof n, device_names[i].name, j);
 397                     deviceRegister(strdup(n), device_names[i].description, strdup(try),
 398                                    DEVICE_TYPE_FLOPPY, TRUE, mediaInitFloppy, mediaGetFloppy,
 399                                    mediaShutdownFloppy, NULL);
 400                     if (isDebug())
 401                         msgDebug("Found a floppy device for %s\n", try);
 402                 }
 403                 break;
 404
 405             case DEVICE_TYPE_NETWORK:
 406                 fd = deviceTry(device_names[i], try, j);
 407                 /* The only network devices that you can open this way are serial ones */
 408                 if (fd >= 0) {
 409                     char *newdesc, *cp;
 410
 411                     close(fd);
 412                     cp = device_names[i].description;
 413                     /* Serial devices get a slip and ppp device each, if supported */
 414                     newdesc = safe_malloc(strlen(cp) + 40);
 415                     sprintf(newdesc, cp, "SLIP interface", try, j + 1);
 416                     deviceRegister("sl0", newdesc, strdup(try), DEVICE_TYPE_NETWORK, TRUE, mediaInitNetwork,
 417                                    NULL, mediaShutdownNetwork, NULL);
 418                     msgDebug("Add mapping for %s to sl0\n", try);
 419                     newdesc = safe_malloc(strlen(cp) + 50);
 420                     sprintf(newdesc, cp, "PPP interface", try, j + 1);
 421                     deviceRegister("ppp0", newdesc, strdup(try), DEVICE_TYPE_NETWORK, TRUE, mediaInitNetwork,
 422                                    NULL, mediaShutdownNetwork, NULL);
 423                     if (isDebug())
 424                         msgDebug("Add mapping for %s to ppp0\n", try);
 425                 }
 426                 break;
 427
 428             default:
 429                 break;
 430             }
 431         }
 432     }
 433
 434     /* Finally, go get the disks and look for DOS partitions to register */
 435     if ((names = Disk_Names()) != NULL) {
 436         int i;
 437
 438         for (i = 0; names[i]; i++) {
 439             Chunk *c1;
 440             Disk *d;
 441
 442             /* Ignore memory disks */
 443             if (!strncmp(names[i], "md", 2))
 444                 continue;
 445
 446             d = Open_Disk(names[i]);
 447             if (!d) {
 448                 msgDebug("Unable to open disk %s", names[i]);
 449                 continue;
 450             }
 451
 452             deviceRegister(names[i], names[i], d->name, DEVICE_TYPE_DISK, FALSE,
 453                            dummyInit, dummyGet, dummyShutdown, d);
 454             if (isDebug())
 455                 msgDebug("Found a disk device named %s\n", names[i]);
 456
 457             /* Look for existing DOS partitions to register as "DOS media devices" */
 458             for (c1 = d->chunks->part; c1; c1 = c1->next) {
 459                 if (c1->type == fat || c1->type == extended) {
 460                     Device *dev;
 461                     char devname[80];
 462
 463                     /* Got one! */
 464                     snprintf(devname, sizeof devname, "/dev/%s", c1->name);
 465                     dev = deviceRegister(c1->name, c1->name, strdup(devname), DEVICE_TYPE_DOS, TRUE,
 466                                          mediaInitDOS, mediaGetDOS, mediaShutdownDOS, NULL);
 467                     dev->private = c1;
 468                     if (isDebug())
 469                         msgDebug("Found a DOS partition %s on drive %s\n", c1->name, d->name);
 470                 }
 471             }
 472         }
 473         free(names);
 474     }
 475     dialog_clear_norefresh();
 476 }
 477
 478 /* Rescan all devices, after closing previous set - convenience function */
 479 void
 480 deviceRescan(void)
 481 {
 482     deviceReset();
 483     deviceGetAll();
 484 }
 485
 486 /*
 487  * Find all devices that match the criteria, allowing "wildcarding" as well
 488  * by allowing NULL or ANY values to match all.  The array returned is static
 489  * and may be used until the next invocation of deviceFind().
 490  */
 491 Device **
 492 deviceFind(char *name, DeviceType class)
 493 {
 494     static Device *found[DEV_MAX];
 495     int i, j;
 496
 497     j = 0;
 498     for (i = 0; i < numDevs; i++) {
 499         if ((!name || !strcmp(Devices[i]->name, name))
 500             && (class == DEVICE_TYPE_ANY || class == Devices[i]->type))
 501             found[j++] = Devices[i];
 502     }
 503     found[j] = NULL;
 504     return j ? found : NULL;
 505 }
 506
 507 Device **
 508 deviceFindDescr(char *name, char *desc, DeviceType class)
 509 {
 510     static Device *found[DEV_MAX];
 511     int i, j;
 512
 513     j = 0;
 514     for (i = 0; i < numDevs; i++) {
 515         if ((!name || !strcmp(Devices[i]->name, name)) &&
 516             (!desc || !strcmp(Devices[i]->description, desc)) &&
 517             (class == DEVICE_TYPE_ANY || class == Devices[i]->type))
 518             found[j++] = Devices[i];
 519     }
 520     found[j] = NULL;
 521     return j ? found : NULL;
 522 }
 523
 524 int
 525 deviceCount(Device **devs)
 526 {
 527     int i;
 528
 529     if (!devs)
 530         return 0;
 531     for (i = 0; devs[i]; i++);
 532     return i;
 533 }
 534
 535 /*
 536  * Create a menu listing all the devices of a certain type in the system.
 537  * The passed-in menu is expected to be a "prototype" from which the new
 538  * menu is cloned.
 539  */
 540 DMenu *
 541 deviceCreateMenu(DMenu *menu, DeviceType type, int (*hook)(dialogMenuItem *d), int (*check)(dialogMenuItem *d))
 542 {
 543     Device **devs;
 544     int numdevs;
 545     DMenu *tmp = NULL;
 546     int i, j;
 547
 548     devs = deviceFind(NULL, type);
 549     numdevs = deviceCount(devs);
 550     if (!numdevs)
 551         return NULL;
 552     tmp = (DMenu *)safe_malloc(sizeof(DMenu) + (sizeof(dialogMenuItem) * (numdevs + 1)));
 553     bcopy(menu, tmp, sizeof(DMenu));
 554     for (i = 0; devs[i]; i++) {
 555         tmp->items[i].prompt = devs[i]->name;
 556         for (j = 0; j < numDevs; j++) {
 557             if (devs[i] == Devices[j]) {
 558                 tmp->items[i].title = Devices[j]->description;
 559                 break;
 560             }
 561         }
 562         if (j == numDevs)
 563             tmp->items[i].title = "<unknown device type>";
 564         tmp->items[i].fire = hook;
 565         tmp->items[i].checked = check;
 566     }
 567     tmp->items[i].title = NULL;
 568     return tmp;
 569 }