xref: /dragonfly/stand/boot/pc32/libi386/devicename.c (revision 655933d6)
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
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
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 *
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
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