sys/boot/uboot/lib/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 "libuboot.h"
  36
  37 static int uboot_parsedev(struct uboot_devdesc **dev, const char *devspec,
  38     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 uboot_getdev(void **vdev, const char *devspec, const char **path)
  47 {
  48         struct uboot_devdesc **dev = (struct uboot_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) || (devspec[0] == '/') ||
  56             (strchr(devspec, ':') == NULL)) {
  57
  58                 if (((rv = uboot_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 (uboot_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>[<partition>]:
  82  *
  83  */
  84 static int
  85 uboot_parsedev(struct uboot_devdesc **dev, const char *devspec,
  86     const char **path)
  87 {
  88         struct uboot_devdesc *idev;
  89         struct devsw *dv;
  90         char *cp;
  91         const char *np;
  92         int i, unit, err;
  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,
 101                     strlen(devsw[i]->dv_name))) {
 102                         dv = devsw[i];
 103                         break;
 104                 }
 105         }
 106         if (dv == NULL)
 107                 return(ENOENT);
 108         idev = malloc(sizeof(struct uboot_devdesc));
 109         err = 0;
 110         np = (devspec + strlen(dv->dv_name));
 111
 112         switch(dv->dv_type) {
 113         case DEVT_NONE:
 114                 break;
 115
 116 #ifdef LOADER_DISK_SUPPORT
 117         case DEVT_DISK:
 118                 err = disk_parsedev((struct disk_devdesc *)idev, np, path);
 119                 if (err != 0)
 120                         goto fail;
 121                 break;
 122 #endif
 123
 124         case DEVT_NET:
 125                 unit = 0;
 126
 127                 if (*np && (*np != ':')) {
 128                         /* get unit number if present */
 129                         unit = strtol(np, &cp, 0);
 130                         if (cp == np) {
 131                                 err = EUNIT;
 132                                 goto fail;
 133                         }
 134                 }
 135                 if (*cp && (*cp != ':')) {
 136                         err = EINVAL;
 137                         goto fail;
 138                 }
 139                 idev->d_unit = unit;
 140
 141                 if (path != NULL)
 142                         *path = (*cp == 0) ? cp : cp + 1;
 143                 break;
 144
 145         default:
 146                 err = EINVAL;
 147                 goto fail;
 148         }
 149         idev->d_dev = dv;
 150         idev->d_type = dv->dv_type;
 151         if (dev == NULL) {
 152                 free(idev);
 153         } else {
 154                 *dev = idev;
 155         }
 156         return (0);
 157
 158 fail:
 159         free(idev);
 160         return (err);
 161 }
 162
 163
 164 char *
 165 uboot_fmtdev(void *vdev)
 166 {
 167         struct uboot_devdesc *dev = (struct uboot_devdesc *)vdev;
 168         static char buf[128];
 169
 170         switch(dev->d_type) {
 171         case DEVT_NONE:
 172                 strcpy(buf, "(no device)");
 173                 break;
 174
 175         case DEVT_DISK:
 176 #ifdef LOADER_DISK_SUPPORT
 177                 return (disk_fmtdev(vdev));
 178 #endif
 179
 180         case DEVT_NET:
 181                 sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_unit);
 182                 break;
 183         }
 184         return(buf);
 185 }
 186
 187 /*
 188  * Set currdev to suit the value being supplied in (value).
 189  */
 190 int
 191 uboot_setcurrdev(struct env_var *ev, int flags, const void *value)
 192 {
 193         struct uboot_devdesc *ncurr;
 194         int rv;
 195
 196         if ((rv = uboot_parsedev(&ncurr, value, NULL)) != 0)
 197                 return (rv);
 198         free(ncurr);
 199         env_setenv(ev->ev_name, flags | EV_NOHOOK, value, NULL, NULL);
 200         return (0);
 201 }