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