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