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