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