devicename.c revision 181282 
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: head/sys/boot/i386/libi386/devicename.c 181282 2008-08-04 07:01:42Z cperciva $");
29
30#include <stand.h>
31#include <string.h>
32#include <sys/disklabel.h>
33#include "bootstrap.h"
34#include "libi386.h"
35
36static 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 */
43int
44i386_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 */
83static int
84i386_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 == 'p') {		/* got a GPT partition */
124		np = cp + 1;
125		slice = strtol(np, &cp, 10);
126		if (cp == np) {
127		    err = ESLICE;
128		    goto fail;
129		}
130		if (*cp && (*cp != ':')) {
131		    err = EINVAL;
132		    goto fail;
133		}
134		partition = 0xff;
135	    } else {
136		if (*cp == 's') {		/* got a slice number */
137		    np = cp + 1;
138		    slice = strtol(np, &cp, 10);
139		    if (cp == np) {
140			err = ESLICE;
141			goto fail;
142		    }
143		}
144		if (*cp && (*cp != ':')) {
145		    partition = *cp - 'a';	/* got a partition number */
146		    if ((partition < 0) || (partition >= MAXPARTITIONS)) {
147			err = EPART;
148			goto fail;
149		    }
150		    cp++;
151		}
152	    }
153	} else {
154		cp = np;
155	}
156	if (*cp && (*cp != ':')) {
157	    err = EINVAL;
158	    goto fail;
159	}
160
161	idev->d_unit = unit;
162	idev->d_kind.biosdisk.slice = slice;
163	idev->d_kind.biosdisk.partition = partition;
164	if (path != NULL)
165	    *path = (*cp == 0) ? cp : cp + 1;
166	break;
167
168    case DEVT_CD:
169    case DEVT_NET:
170	unit = 0;
171
172	if (*np && (*np != ':')) {
173	    unit = strtol(np, &cp, 0);	/* get unit number if present */
174	    if (cp == np) {
175		err = EUNIT;
176		goto fail;
177	    }
178	} else {
179		cp = np;
180	}
181	if (*cp && (*cp != ':')) {
182	    err = EINVAL;
183	    goto fail;
184	}
185
186	idev->d_unit = unit;
187	if (path != NULL)
188	    *path = (*cp == 0) ? cp : cp + 1;
189	break;
190
191    default:
192	err = EINVAL;
193	goto fail;
194    }
195    idev->d_dev = dv;
196    idev->d_type = dv->dv_type;
197    if (dev == NULL) {
198	free(idev);
199    } else {
200	*dev = idev;
201    }
202    return(0);
203
204 fail:
205    free(idev);
206    return(err);
207}
208
209
210char *
211i386_fmtdev(void *vdev)
212{
213    struct i386_devdesc	*dev = (struct i386_devdesc *)vdev;
214    static char		buf[128];	/* XXX device length constant? */
215    char		*cp;
216
217    switch(dev->d_type) {
218    case DEVT_NONE:
219	strcpy(buf, "(no device)");
220	break;
221
222    case DEVT_CD:
223	sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_unit);
224	break;
225
226    case DEVT_DISK:
227	cp = buf;
228	cp += sprintf(cp, "%s%d", dev->d_dev->dv_name, dev->d_unit);
229	if (dev->d_kind.biosdisk.partition == 0xff) {
230	    cp += sprintf(cp, "p%d", dev->d_kind.biosdisk.slice);
231	} else {
232	    if (dev->d_kind.biosdisk.slice > 0)
233		cp += sprintf(cp, "s%d", dev->d_kind.biosdisk.slice);
234	    if (dev->d_kind.biosdisk.partition >= 0)
235		cp += sprintf(cp, "%c", dev->d_kind.biosdisk.partition + 'a');
236	}
237	strcat(cp, ":");
238	break;
239
240    case DEVT_NET:
241	sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_unit);
242	break;
243    }
244    return(buf);
245}
246
247
248/*
249 * Set currdev to suit the value being supplied in (value)
250 */
251int
252i386_setcurrdev(struct env_var *ev, int flags, const void *value)
253{
254    struct i386_devdesc	*ncurr;
255    int			rv;
256
257    if ((rv = i386_parsedev(&ncurr, value, NULL)) != 0)
258	return(rv);
259    free(ncurr);
260    env_setenv(ev->ev_name, flags | EV_NOHOOK, value, NULL, NULL);
261    return(0);
262}
263