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 "libuserboot.h" 36 37static int userboot_parsedev(struct disk_devdesc **dev, const char *devspec, const char **path); 38 39/* 40 * Point (dev) at an allocated device specifier for the device matching the 41 * path in (devspec). If it contains an explicit device specification, 42 * use that. If not, use the default device. 43 */ 44int 45userboot_getdev(void **vdev, const char *devspec, const char **path) 46{ 47 struct disk_devdesc **dev = (struct disk_devdesc **)vdev; 48 int rv; 49 50 /* 51 * If it looks like this is just a path and no 52 * device, go with the current device. 53 */ 54 if ((devspec == NULL) || 55 (devspec[0] == '/') || 56 (strchr(devspec, ':') == NULL)) { 57 58 if (((rv = userboot_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(userboot_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>[s<slice>][<partition>]: 82 * 83 */ 84static int 85userboot_parsedev(struct disk_devdesc **dev, const char *devspec, const char **path) 86{ 87 struct disk_devdesc *idev; 88 struct devsw *dv; 89 int i, unit, err; 90 char *cp; 91 const char *np; 92 93 /* minimum length check */ 94 if (strlen(devspec) < 2) 95 return(EINVAL); 96 97 /* look for a device that matches */ 98 for (i = 0, dv = NULL; devsw[i] != NULL; i++) { 99 if (!strncmp(devspec, devsw[i]->dv_name, strlen(devsw[i]->dv_name))) { 100 dv = devsw[i]; 101 break; 102 } 103 } 104 if (dv == NULL) 105 return(ENOENT); 106 idev = malloc(sizeof(struct disk_devdesc)); 107 err = 0; 108 np = (devspec + strlen(dv->dv_name)); 109 110 switch(dv->dv_type) { 111 case DEVT_NONE: /* XXX what to do here? Do we care? */ 112 break; 113 114 case DEVT_DISK: 115 err = disk_parsedev(idev, np, path); 116 if (err != 0) 117 goto fail; 118 break; 119 120 case DEVT_CD: 121 case DEVT_NET: 122 case DEVT_ZFS: 123 unit = 0; 124 125 if (*np && (*np != ':')) { 126 unit = strtol(np, &cp, 0); /* get unit number if present */ 127 if (cp == np) { 128 err = EUNIT; 129 goto fail; 130 } 131 } else { 132 cp = np; 133 } 134 if (*cp && (*cp != ':')) { 135 err = EINVAL; 136 goto fail; 137 } 138 139 idev->d_unit = unit; 140 if (path != NULL) 141 *path = (*cp == 0) ? cp : cp + 1; 142 break; 143 144 default: 145 err = EINVAL; 146 goto fail; 147 } 148 idev->d_dev = dv; 149 idev->d_type = dv->dv_type; 150 if (dev == NULL) { 151 free(idev); 152 } else { 153 *dev = idev; 154 } 155 return(0); 156 157 fail: 158 free(idev); 159 return(err); 160} 161 162 163char * 164userboot_fmtdev(void *vdev) 165{ 166 struct disk_devdesc *dev = (struct disk_devdesc *)vdev; 167 static char buf[128]; /* XXX device length constant? */ 168 169 switch(dev->d_type) { 170 case DEVT_NONE: 171 strcpy(buf, "(no device)"); 172 break; 173 174 case DEVT_CD: 175 sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_unit); 176 break; 177 178 case DEVT_DISK: 179 return (disk_fmtdev(vdev)); 180 181 case DEVT_NET: 182 case DEVT_ZFS: 183 sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_unit); 184 break; 185 } 186 return(buf); 187} 188 189 190/* 191 * Set currdev to suit the value being supplied in (value) 192 */ 193int 194userboot_setcurrdev(struct env_var *ev, int flags, const void *value) 195{ 196 struct disk_devdesc *ncurr; 197 int rv; 198 199 if ((rv = userboot_parsedev(&ncurr, value, NULL)) != 0) 200 return(rv); 201 free(ncurr); 202 env_setenv(ev->ev_name, flags | EV_NOHOOK, value, NULL, NULL); 203 return(0); 204} 205