sys/boot/userboot/userboot/devicename.c

   1 /*-
   2  * Copyright (c) 1998 Michael Smith <msmith@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
  27 #include <sys/cdefs.h>
  28 __FBSDID("$FreeBSD$");
  29
  30 #include <stand.h>
  31 #include <string.h>
  32
  33 #include "bootstrap.h"
  34 #include "disk.h"
  35 #include "libuserboot.h"
  36
  37 #if defined(USERBOOT_ZFS_SUPPORT)
  38 #include "../zfs/libzfs.h"
  39 #endif
  40
  41 static int      userboot_parsedev(struct disk_devdesc **dev, const char *devspec, const char **path);
  42
  43 /*
  44  * Point (dev) at an allocated device specifier for the device matching the
  45  * path in (devspec). If it contains an explicit device specification,
  46  * use that.  If not, use the default device.
  47  */
  48 int
  49 userboot_getdev(void **vdev, const char *devspec, const char **path)
  50 {
  51     struct disk_devdesc **dev = (struct disk_devdesc **)vdev;
  52     int                         rv;
  53
  54     /*
  55      * If it looks like this is just a path and no
  56      * device, go with the current device.
  57      */
  58     if ((devspec == NULL) ||
  59         (devspec[0] == '/') ||
  60         (strchr(devspec, ':') == NULL)) {
  61
  62         if (((rv = userboot_parsedev(dev, getenv("currdev"), NULL)) == 0) &&
  63             (path != NULL))
  64                 *path = devspec;
  65         return(rv);
  66     }
  67
  68     /*
  69      * Try to parse the device name off the beginning of the devspec
  70      */
  71     return(userboot_parsedev(dev, devspec, path));
  72 }
  73
  74 /*
  75  * Point (dev) at an allocated device specifier matching the string version
  76  * at the beginning of (devspec).  Return a pointer to the remaining
  77  * text in (path).
  78  *
  79  * In all cases, the beginning of (devspec) is compared to the names
  80  * of known devices in the device switch, and then any following text
  81  * is parsed according to the rules applied to the device type.
  82  *
  83  * For disk-type devices, the syntax is:
  84  *
  85  * disk<unit>[s<slice>][<partition>]:
  86  *
  87  */
  88 static int
  89 userboot_parsedev(struct disk_devdesc **dev, const char *devspec, const char **path)
  90 {
  91     struct disk_devdesc *idev;
  92     struct devsw        *dv;
  93     int                 i, unit, err;
  94     char                *cp;
  95     const char          *np;
  96
  97     /* minimum length check */
  98     if (strlen(devspec) < 2)
  99         return(EINVAL);
 100
 101     /* look for a device that matches */
 102     for (i = 0, dv = NULL; devsw[i] != NULL; i++) {
 103         if (!strncmp(devspec, devsw[i]->dv_name, strlen(devsw[i]->dv_name))) {
 104             dv = devsw[i];
 105             break;
 106         }
 107     }
 108     if (dv == NULL)
 109         return(ENOENT);
 110     idev = malloc(sizeof(struct disk_devdesc));
 111     err = 0;
 112     np = (devspec + strlen(dv->dv_name));
 113
 114     switch(dv->dv_type) {
 115     case DEVT_NONE:                     /* XXX what to do here?  Do we care? */
 116         break;
 117
 118     case DEVT_DISK:
 119         err = disk_parsedev(idev, np, path);
 120         if (err != 0)
 121             goto fail;
 122         break;
 123
 124     case DEVT_CD:
 125     case DEVT_NET:
 126         unit = 0;
 127
 128         if (*np && (*np != ':')) {
 129             unit = strtol(np, &cp, 0);  /* get unit number if present */
 130             if (cp == np) {
 131                 err = EUNIT;
 132                 goto fail;
 133             }
 134         } else {
 135                 cp = np;
 136         }
 137         if (*cp && (*cp != ':')) {
 138             err = EINVAL;
 139             goto fail;
 140         }
 141
 142         idev->d_unit = unit;
 143         if (path != NULL)
 144             *path = (*cp == 0) ? cp : cp + 1;
 145         break;
 146
 147     case DEVT_ZFS:
 148 #if defined(USERBOOT_ZFS_SUPPORT)
 149             err = zfs_parsedev((struct zfs_devdesc *)idev, np, path);
 150             if (err != 0)
 151                     goto fail;
 152             break;
 153 #else
 154             /* FALLTHROUGH */
 155 #endif
 156
 157     default:
 158         err = EINVAL;
 159         goto fail;
 160     }
 161     idev->d_dev = dv;
 162     idev->d_type = dv->dv_type;
 163     if (dev == NULL) {
 164         free(idev);
 165     } else {
 166         *dev = idev;
 167     }
 168     return(0);
 169
 170  fail:
 171     free(idev);
 172     return(err);
 173 }
 174
 175
 176 char *
 177 userboot_fmtdev(void *vdev)
 178 {
 179     struct disk_devdesc *dev = (struct disk_devdesc *)vdev;
 180     static char         buf[128];       /* XXX device length constant? */
 181
 182     switch(dev->d_type) {
 183     case DEVT_NONE:
 184         strcpy(buf, "(no device)");
 185         break;
 186
 187     case DEVT_CD:
 188         sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_unit);
 189         break;
 190
 191     case DEVT_DISK:
 192         return (disk_fmtdev(vdev));
 193
 194     case DEVT_NET:
 195         sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_unit);
 196         break;
 197
 198     case DEVT_ZFS:
 199 #if defined(USERBOOT_ZFS_SUPPORT)
 200         return (zfs_fmtdev(vdev));
 201 #else
 202         sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_unit);
 203 #endif
 204         break;
 205     }
 206     return(buf);
 207 }
 208
 209
 210 /*
 211  * Set currdev to suit the value being supplied in (value)
 212  */
 213 int
 214 userboot_setcurrdev(struct env_var *ev, int flags, const void *value)
 215 {
 216     struct disk_devdesc *ncurr;
 217     int                 rv;
 218
 219     if ((rv = userboot_parsedev(&ncurr, value, NULL)) != 0)
 220         return(rv);
 221     free(ncurr);
 222     env_setenv(ev->ev_name, flags | EV_NOHOOK, value, NULL, NULL);
 223     return(0);
 224 }