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
32 #include "bootstrap.h"
33 #include "disk.h"
34 #include "libuserboot.h"
35
36 #include "libzfs.h"
37
38 static int userboot_parsedev(struct devdesc **, const char *, const char **);
39
40 /*
41 * Point (dev) at an allocated device specifier for the device matching the
42 * path in (devspec). If it contains an explicit device specification,
43 * use that. If not, use the default device.
44 */
45 int
userboot_getdev(void ** vdev,const char * devspec,const char ** path)46 userboot_getdev(void **vdev, const char *devspec, const char **path)
47 {
48 struct devdesc **dev = (struct devdesc **)vdev;
49 int rv;
50
51 /*
52 * If it looks like this is just a path and no
53 * device, go with the current device.
54 */
55 if ((devspec == NULL) || (devspec[0] == '/') ||
56 (strchr(devspec, ':') == NULL)) {
57
58 rv = userboot_parsedev(dev, getenv("currdev"), NULL);
59 if (rv == 0 && 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 */
84 static int
userboot_parsedev(struct devdesc ** dev,const char * devspec,const char ** path)85 userboot_parsedev(struct devdesc **dev, const char *devspec, const char **path)
86 {
87 struct devdesc *idev;
88 struct devsw *dv;
89 int i, unit, err;
90 const 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 dv = devsw[i];
100 if (strncmp(devspec, dv->dv_name, strlen(dv->dv_name)) == 0)
101 break;
102 }
103 if (devsw[i] == NULL)
104 return (ENOENT);
105
106 np = devspec + strlen(dv->dv_name);
107 idev = NULL;
108 err = 0;
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 idev = malloc(sizeof (struct disk_devdesc));
116 if (idev == NULL)
117 return (ENOMEM);
118
119 err = disk_parsedev((struct disk_devdesc *)idev, np, path);
120 if (err != 0)
121 goto fail;
122 break;
123
124 case DEVT_ZFS:
125 idev = malloc(sizeof (struct zfs_devdesc));
126 if (idev == NULL)
127 return (ENOMEM);
128
129 err = zfs_parsedev((struct zfs_devdesc *)idev, np, path);
130 if (err != 0)
131 goto fail;
132 break;
133
134 case DEVT_CD:
135 case DEVT_NET:
136 idev = malloc(sizeof (struct devdesc));
137 if (idev == NULL)
138 return (ENOMEM);
139
140 unit = 0;
141
142 /* get unit number if present */
143 if (*np && (*np != ':')) {
144 unit = strtol(np, (char **)&cp, 0);
145 if (cp == np) {
146 err = EUNIT;
147 goto fail;
148 }
149 } else {
150 cp = np;
151 }
152 if (*cp && (*cp != ':')) {
153 err = EINVAL;
154 goto fail;
155 }
156
157 idev->d_unit = unit;
158 if (path != NULL)
159 *path = (*cp == 0) ? cp : cp + 1;
160 break;
161
162 default:
163 err = EINVAL;
164 goto fail;
165 }
166
167 idev->d_dev = dv;
168
169 if (dev == NULL) {
170 free(idev);
171 } else {
172 *dev = idev;
173 }
174 return (0);
175
176 fail:
177 free(idev);
178 return (err);
179 }
180
181
182 char *
userboot_fmtdev(void * vdev)183 userboot_fmtdev(void *vdev)
184 {
185 struct devdesc *dev = (struct devdesc *)vdev;
186 static char buf[128]; /* XXX device length constant? */
187
188 switch (dev->d_dev->dv_type) {
189 case DEVT_NONE:
190 (void) strlcpy(buf, "(no device)", sizeof (buf));
191 break;
192
193 case DEVT_DISK:
194 return (disk_fmtdev(vdev));
195
196 case DEVT_ZFS:
197 return (zfs_fmtdev(vdev));
198 break;
199
200 default:
201 (void) snprintf(buf, sizeof (buf), "%s%d:", dev->d_dev->dv_name,
202 dev->d_unit);
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 devdesc *ncurr;
216 int rv;
217
218 if ((rv = userboot_parsedev(&ncurr, value, NULL)) != 0)
219 return (rv);
220
221 free(ncurr);
222 (void) env_setenv(ev->ev_name, flags | EV_NOHOOK, value, NULL, NULL);
223 return (0);
224 }
225