devicename.c revision 185029 
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 185029 2008-11-17 20:49:29Z pjd $");
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    case DEVT_ZFS:
171	unit = 0;
172
173	if (*np && (*np != ':')) {
174	    unit = strtol(np, &cp, 0);	/* get unit number if present */
175	    if (cp == np) {
176		err = EUNIT;
177		goto fail;
178	    }
179	} else {
180		cp = np;
181	}
182	if (*cp && (*cp != ':')) {
183	    err = EINVAL;
184	    goto fail;
185	}
186
187	idev->d_unit = unit;
188	if (path != NULL)
189	    *path = (*cp == 0) ? cp : cp + 1;
190	break;
191
192    default:
193	err = EINVAL;
194	goto fail;
195    }
196    idev->d_dev = dv;
197    idev->d_type = dv->dv_type;
198    if (dev == NULL) {
199	free(idev);
200    } else {
201	*dev = idev;
202    }
203    return(0);
204
205 fail:
206    free(idev);
207    return(err);
208}
209
210
211char *
212i386_fmtdev(void *vdev)
213{
214    struct i386_devdesc	*dev = (struct i386_devdesc *)vdev;
215    static char		buf[128];	/* XXX device length constant? */
216    char		*cp;
217
218    switch(dev->d_type) {
219    case DEVT_NONE:
220	strcpy(buf, "(no device)");
221	break;
222
223    case DEVT_CD:
224	sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_unit);
225	break;
226
227    case DEVT_DISK:
228	cp = buf;
229	cp += sprintf(cp, "%s%d", dev->d_dev->dv_name, dev->d_unit);
230	if (dev->d_kind.biosdisk.partition == 0xff) {
231	    cp += sprintf(cp, "p%d", dev->d_kind.biosdisk.slice);
232	} else {
233	    if (dev->d_kind.biosdisk.slice > 0)
234		cp += sprintf(cp, "s%d", dev->d_kind.biosdisk.slice);
235	    if (dev->d_kind.biosdisk.partition >= 0)
236		cp += sprintf(cp, "%c", dev->d_kind.biosdisk.partition + 'a');
237	}
238	strcat(cp, ":");
239	break;
240
241    case DEVT_NET:
242    case DEVT_ZFS:
243	sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_unit);
244	break;
245    }
246    return(buf);
247}
248
249
250/*
251 * Set currdev to suit the value being supplied in (value)
252 */
253int
254i386_setcurrdev(struct env_var *ev, int flags, const void *value)
255{
256    struct i386_devdesc	*ncurr;
257    int			rv;
258
259    if ((rv = i386_parsedev(&ncurr, value, NULL)) != 0)
260	return(rv);
261    free(ncurr);
262    env_setenv(ev->ev_name, flags | EV_NOHOOK, value, NULL, NULL);
263    return(0);
264}
265