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 #include "bootstrap.h" 33 #include "disk.h" 34 #include "libi386.h" 35 #include "libzfs.h" 36 37 static int i386_parsedev(struct i386_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 */ 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] == '/') || 56 (strchr(devspec, ':') == NULL)) { 57 58 if (((rv = i386_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(i386_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 */ 84 static int 85 i386_parsedev(struct i386_devdesc **dev, const char *devspec, const char **path) 86 { 87 struct i386_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 107 np = (devspec + strlen(dv->dv_name)); 108 idev = NULL; 109 err = 0; 110 111 switch(dv->dv_type) { 112 case DEVT_NONE: 113 break; 114 115 case DEVT_DISK: 116 idev = malloc(sizeof(struct i386_devdesc)); 117 if (idev == NULL) 118 return (ENOMEM); 119 120 err = disk_parsedev((struct disk_devdesc *)idev, np, path); 121 if (err != 0) 122 goto fail; 123 break; 124 125 case DEVT_ZFS: 126 idev = malloc(sizeof (struct zfs_devdesc)); 127 if (idev == NULL) 128 return (ENOMEM); 129 130 err = zfs_parsedev((struct zfs_devdesc *)idev, np, path); 131 if (err != 0) 132 goto fail; 133 break; 134 135 default: 136 idev = malloc(sizeof (struct devdesc)); 137 if (idev == NULL) 138 return (ENOMEM); 139 140 unit = 0; 141 cp = (char *)np; 142 143 if (*np && (*np != ':')) { 144 unit = strtol(np, &cp, 0); /* get unit number if present */ 145 if (cp == np) { 146 err = EUNIT; 147 goto fail; 148 } 149 } 150 151 if (*cp && (*cp != ':')) { 152 err = EINVAL; 153 goto fail; 154 } 155 156 idev->dd.d_unit = unit; 157 if (path != NULL) 158 *path = (*cp == 0) ? cp : cp + 1; 159 break; 160 } 161 idev->dd.d_dev = dv; 162 if (dev != NULL) 163 *dev = idev; 164 else 165 free(idev); 166 167 return(0); 168 169 fail: 170 free(idev); 171 return(err); 172 } 173 174 175 char * 176 i386_fmtdev(void *vdev) 177 { 178 struct i386_devdesc *dev = (struct i386_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 case DEVT_NET: 188 sprintf(buf, "%s%d:", dev->dd.d_dev->dv_name, dev->dd.d_unit); 189 break; 190 191 case DEVT_DISK: 192 return (disk_fmtdev(vdev)); 193 194 case DEVT_ZFS: 195 return(zfs_fmtdev(vdev)); 196 } 197 return(buf); 198 } 199 200 201 /* 202 * Set currdev to suit the value being supplied in (value) 203 */ 204 int 205 i386_setcurrdev(struct env_var *ev, int flags, const void *value) 206 { 207 struct i386_devdesc *ncurr; 208 int rv; 209 210 if ((rv = i386_parsedev(&ncurr, value, NULL)) != 0) 211 return(rv); 212 free(ncurr); 213 env_setenv(ev->ev_name, flags | EV_NOHOOK, value, NULL, NULL); 214 return(0); 215 } 216