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 * $FreeBSD: src/sys/boot/i386/libi386/devicename.c,v 1.6 2003/08/25 23:28:31 obrien Exp $
27 * $DragonFly: src/sys/boot/pc32/libi386/devicename.c,v 1.4 2007/06/18 05:13:42 dillon Exp $
28 */
29
30 #include <stand.h>
31 #include <string.h>
32 #include <sys/disklabel32.h>
33 #include "bootstrap.h"
34 #include "libi386.h"
35
36 static int i386_parsedev(struct i386_devdesc **dev, const char *devspec, const char **path);
37
38 /*
39 * Point (dev) at an allocated device specifier for the device matching the
40 * path in (devspec). If it contains an explicit device specification,
41 * use that. If not, use the default device.
42 */
43 int
i386_getdev(void ** vdev,const char * devspec,const char ** path)44 i386_getdev(void **vdev, const char *devspec, const char **path)
45 {
46 struct i386_devdesc **dev = (struct i386_devdesc **)vdev;
47 int rv;
48
49 /*
50 * If it looks like this is just a path and no
51 * device, go with the current device.
52 */
53 if ((devspec == NULL) ||
54 (devspec[0] == '/') ||
55 (strchr(devspec, ':') == NULL)) {
56
57 if (((rv = i386_parsedev(dev, getenv("currdev"), NULL)) == 0) &&
58 (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
i386_parsedev(struct i386_devdesc ** dev,const char * devspec,const char ** path)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, slice, partition, 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 if (!strncmp(devspec, devsw[i]->dv_name, strlen(devsw[i]->dv_name))) {
99 dv = devsw[i];
100 break;
101 }
102 }
103 if (dv == NULL)
104 return(ENOENT);
105 idev = malloc(sizeof(struct i386_devdesc));
106 err = 0;
107 np = (devspec + strlen(dv->dv_name));
108
109 switch(dv->dv_type) {
110 case DEVT_NONE: /* XXX what to do here? Do we care? */
111 break;
112
113 case DEVT_DISK:
114 unit = -1;
115 slice = -1;
116 partition = -1;
117 if (*np && (*np != ':')) {
118 unit = strtol(np, &cp, 10); /* next comes the unit number */
119 if (cp == np) {
120 err = EUNIT;
121 goto fail;
122 }
123 if (*cp == 's') { /* got a slice number */
124 np = cp + 1;
125 slice = strtol(np, &cp, 10);
126 if (cp == np) {
127 err = ESLICE;
128 goto fail;
129 }
130 }
131 if (*cp && (*cp != ':')) {
132 partition = *cp - 'a'; /* get a partition number */
133 if ((partition < 0) || (partition >= MAXPARTITIONS32)) {
134 err = EPART;
135 goto fail;
136 }
137 cp++;
138 }
139 }
140 if (*cp && (*cp != ':')) {
141 err = EINVAL;
142 goto fail;
143 }
144
145 idev->d_kind.biosdisk.unit = unit;
146 idev->d_kind.biosdisk.slice = slice;
147 idev->d_kind.biosdisk.partition = partition;
148 if (path != NULL)
149 *path = (*cp == 0) ? cp : cp + 1;
150 break;
151
152 case DEVT_CD:
153 case DEVT_NET:
154 unit = 0;
155
156 if (*np && (*np != ':')) {
157 unit = strtol(np, &cp, 0); /* get unit number if present */
158 if (cp == np) {
159 err = EUNIT;
160 goto fail;
161 }
162 }
163 if (*cp && (*cp != ':')) {
164 err = EINVAL;
165 goto fail;
166 }
167
168 if (dv->dv_type == DEVT_NET)
169 idev->d_kind.netif.unit = unit;
170 else
171 idev->d_kind.bioscd.unit = unit;
172 if (path != NULL)
173 *path = (*cp == 0) ? cp : cp + 1;
174 break;
175
176 default:
177 err = EINVAL;
178 goto fail;
179 }
180 idev->d_dev = dv;
181 idev->d_type = dv->dv_type;
182 if (dev == NULL) {
183 free(idev);
184 } else {
185 *dev = idev;
186 }
187 return(0);
188
189 fail:
190 free(idev);
191 return(err);
192 }
193
194
195 char *
i386_fmtdev(void * vdev)196 i386_fmtdev(void *vdev)
197 {
198 struct i386_devdesc *dev = (struct i386_devdesc *)vdev;
199 static char buf[128]; /* XXX device length constant? */
200 char *cp;
201
202 switch(dev->d_type) {
203 case DEVT_NONE:
204 strcpy(buf, "(no device)");
205 break;
206
207 case DEVT_CD:
208 sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_kind.bioscd.unit);
209 break;
210
211 case DEVT_DISK:
212 cp = buf;
213 cp += sprintf(cp, "%s%d", dev->d_dev->dv_name, dev->d_kind.biosdisk.unit);
214 if (dev->d_kind.biosdisk.slice > 0)
215 cp += sprintf(cp, "s%d", dev->d_kind.biosdisk.slice);
216 if (dev->d_kind.biosdisk.partition >= 0)
217 cp += sprintf(cp, "%c", dev->d_kind.biosdisk.partition + 'a');
218 strcat(cp, ":");
219 break;
220
221 case DEVT_NET:
222 sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_kind.netif.unit);
223 break;
224 }
225 return(buf);
226 }
227
228
229 /*
230 * Set currdev to suit the value being supplied in (value)
231 */
232 int
i386_setcurrdev(struct env_var * ev,int flags,const void * value)233 i386_setcurrdev(struct env_var *ev, int flags, const void *value)
234 {
235 struct i386_devdesc *ncurr;
236 int rv;
237
238 if ((rv = i386_parsedev(&ncurr, value, NULL)) != 0)
239 return(rv);
240 free(ncurr);
241 env_setenv(ev->ev_name, flags | EV_NOHOOK, value, NULL, NULL);
242 return(0);
243 }
244
245