1 /*-
2 * Copyright (c) 1998 Michael Smith <[email protected]>
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 "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
userboot_getdev(void ** vdev,const char * devspec,const char ** path)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
userboot_parsedev(struct disk_devdesc ** dev,const char * devspec,const char ** path)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 *
userboot_fmtdev(void * vdev)176 userboot_fmtdev(void *vdev)
177 {
178 struct devdesc *dev = (struct devdesc *)vdev;
179 static char buf[128]; /* XXX device length constant? */
180
181 switch(dev->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->d_dev->dv_name, dev->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->d_dev->dv_name, dev->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->d_dev->dv_name, dev->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
userboot_setcurrdev(struct env_var * ev,int flags,const void * value)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