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