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 29 #include <stand.h> 30 #include <string.h> 31 #include <sys/param.h> 32 #include "bootstrap.h" 33 #include "disk.h" 34 #include "libi386.h" 35 #include "libzfs.h" 36 37 static int i386_parsedev(struct i386_devdesc **, const char *, const char **); 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 */ 44 int 45 i386_getdev(void **vdev, const char *devspec, const char **path) 46 { 47 struct i386_devdesc **dev = (struct i386_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] == '/') || (strchr(devspec, ':') == NULL)) { 56 57 rv = i386_parsedev(dev, getenv("currdev"), NULL); 58 if (rv == 0 && path != NULL) 59 *path = devspec; 60 return (rv); 61 } 62 63 /* 64 * Try to parse the device name off the beginning of the devspec 65 */ 66 return (i386_parsedev(dev, devspec, path)); 67 } 68 69 /* 70 * Point (dev) at an allocated device specifier matching the string version 71 * at the beginning of (devspec). Return a pointer to the remaining 72 * text in (path). 73 * 74 * In all cases, the beginning of (devspec) is compared to the names 75 * of known devices in the device switch, and then any following text 76 * is parsed according to the rules applied to the device type. 77 * 78 * For disk-type devices, the syntax is: 79 * 80 * disk<unit>[s<slice>][<partition>]: 81 * 82 */ 83 static int 84 i386_parsedev(struct i386_devdesc **dev, const char *devspec, const char **path) 85 { 86 struct i386_devdesc *idev; 87 struct devsw *dv; 88 int i, unit, err; 89 char *cp; 90 const char *np; 91 92 /* minimum length check */ 93 if (strlen(devspec) < 2) 94 return (EINVAL); 95 96 /* look for a device that matches */ 97 for (i = 0, dv = NULL; devsw[i] != NULL; i++) { 98 dv = devsw[i]; 99 if (strncmp(devspec, dv->dv_name, strlen(dv->dv_name)) == 0) 100 break; 101 } 102 if (devsw[i] == NULL) 103 return (ENOENT); 104 105 np = devspec + strlen(dv->dv_name); 106 idev = NULL; 107 err = 0; 108 109 switch (dv->dv_type) { 110 case DEVT_NONE: 111 break; 112 113 case DEVT_DISK: 114 idev = malloc(sizeof (struct i386_devdesc)); 115 if (idev == NULL) 116 return (ENOMEM); 117 118 err = disk_parsedev((struct disk_devdesc *)idev, np, path); 119 if (err != 0) 120 goto fail; 121 break; 122 123 case DEVT_ZFS: 124 idev = malloc(sizeof (struct zfs_devdesc)); 125 if (idev == NULL) 126 return (ENOMEM); 127 128 err = zfs_parsedev((struct zfs_devdesc *)idev, np, path); 129 if (err != 0) 130 goto fail; 131 break; 132 133 default: 134 idev = malloc(sizeof (struct devdesc)); 135 if (idev == NULL) 136 return (ENOMEM); 137 138 unit = 0; 139 cp = (char *)np; 140 141 if (*np && (*np != ':')) { 142 /* get unit number if present */ 143 unit = strtol(np, &cp, 0); 144 if (cp == np) { 145 err = EUNIT; 146 goto fail; 147 } 148 } 149 if (*cp && (*cp != ':')) { 150 err = EINVAL; 151 goto fail; 152 } 153 154 idev->dd.d_unit = unit; 155 if (path != NULL) 156 *path = (*cp == '\0') ? cp : cp + 1; 157 break; 158 } 159 160 idev->dd.d_dev = dv; 161 162 if (dev != NULL) 163 *dev = idev; 164 else 165 free(idev); 166 return (0); 167 168 fail: 169 free(idev); 170 return (err); 171 } 172 173 174 char * 175 i386_fmtdev(void *vdev) 176 { 177 struct i386_devdesc *dev = (struct i386_devdesc *)vdev; 178 static char buf[SPECNAMELEN + 1]; 179 180 switch (dev->dd.d_dev->dv_type) { 181 case DEVT_NONE: 182 strlcpy(buf, "(no device)", sizeof (buf)); 183 break; 184 185 case DEVT_DISK: 186 return (disk_fmtdev(vdev)); 187 188 case DEVT_ZFS: 189 return (zfs_fmtdev(vdev)); 190 191 default: 192 snprintf(buf, sizeof (buf), "%s%d:", dev->dd.d_dev->dv_name, 193 dev->dd.d_unit); 194 break; 195 } 196 return (buf); 197 } 198 199 200 /* 201 * Set currdev to suit the value being supplied in (value) 202 */ 203 int 204 i386_setcurrdev(struct env_var *ev, int flags, const void *value) 205 { 206 struct i386_devdesc *ncurr; 207 int rv; 208 209 if ((rv = i386_parsedev(&ncurr, value, NULL)) != 0) 210 return (rv); 211 212 free(ncurr); 213 env_setenv(ev->ev_name, flags | EV_NOHOOK, value, NULL, NULL); 214 return (0); 215 } 216