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 } 202