devicename.c revision 237766
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: stable/9/sys/boot/i386/libi386/devicename.c 237766 2012-06-29 10:19:15Z avg $");
29
30#include <stand.h>
31#include <string.h>
32#include <sys/disklabel.h>
33#include "bootstrap.h"
34#include "libi386.h"
35#include "../zfs/libzfs.h"
36
37static int	i386_parsedev(struct i386_devdesc **dev, const char *devspec, const char **path);
38
39/*
40 * Point (dev) at an allocated device specifier for the device matching the
41 * path in (devspec). If it contains an explicit device specification,
42 * use that.  If not, use the default device.
43 */
44int
45i386_getdev(void **vdev, const char *devspec, const char **path)
46{
47    struct i386_devdesc **dev = (struct i386_devdesc **)vdev;
48    int				rv;
49
50    /*
51     * If it looks like this is just a path and no
52     * device, go with the current device.
53     */
54    if ((devspec == NULL) ||
55	(devspec[0] == '/') ||
56	(strchr(devspec, ':') == NULL)) {
57
58	if (((rv = i386_parsedev(dev, getenv("currdev"), NULL)) == 0) &&
59	    (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(i386_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 */
84static int
85i386_parsedev(struct i386_devdesc **dev, const char *devspec, const char **path)
86{
87    struct i386_devdesc *idev;
88    struct devsw	*dv;
89    int			i, unit, slice, partition, err;
90    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	if (!strncmp(devspec, devsw[i]->dv_name, strlen(devsw[i]->dv_name))) {
100	    dv = devsw[i];
101	    break;
102	}
103    }
104    if (dv == NULL)
105	return(ENOENT);
106    idev = malloc(sizeof(struct i386_devdesc));
107    err = 0;
108    np = (devspec + strlen(dv->dv_name));
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	unit = -1;
116	slice = -1;
117	partition = -1;
118	if (*np && (*np != ':')) {
119	    unit = strtol(np, &cp, 10);	/* next comes the unit number */
120	    if (cp == np) {
121		err = EUNIT;
122		goto fail;
123	    }
124#ifdef LOADER_GPT_SUPPORT
125	    if (*cp == 'p') {		/* got a GPT partition */
126		np = cp + 1;
127		slice = strtol(np, &cp, 10);
128		if (cp == np) {
129		    err = ESLICE;
130		    goto fail;
131		}
132		if (*cp && (*cp != ':')) {
133		    err = EINVAL;
134		    goto fail;
135		}
136		partition = 0xff;
137	    } else {
138#endif
139		if (*cp == 's') {		/* got a slice number */
140		    np = cp + 1;
141		    slice = strtol(np, &cp, 10);
142		    if (cp == np) {
143			err = ESLICE;
144			goto fail;
145		    }
146		}
147		if (*cp && (*cp != ':')) {
148		    partition = *cp - 'a';	/* got a partition number */
149		    if ((partition < 0) || (partition >= MAXPARTITIONS)) {
150			err = EPART;
151			goto fail;
152		    }
153		    cp++;
154		}
155#ifdef LOADER_GPT_SUPPORT
156	    }
157#endif
158	} else {
159		cp = np;
160	}
161	if (*cp && (*cp != ':')) {
162	    err = EINVAL;
163	    goto fail;
164	}
165
166	idev->d_unit = unit;
167	idev->d_kind.biosdisk.slice = slice;
168	idev->d_kind.biosdisk.partition = partition;
169	if (path != NULL)
170	    *path = (*cp == 0) ? cp : cp + 1;
171	break;
172
173    case DEVT_CD:
174    case DEVT_NET:
175	unit = 0;
176
177	if (*np && (*np != ':')) {
178	    unit = strtol(np, &cp, 0);	/* get unit number if present */
179	    if (cp == np) {
180		err = EUNIT;
181		goto fail;
182	    }
183	} else {
184		cp = np;
185	}
186	if (*cp && (*cp != ':')) {
187	    err = EINVAL;
188	    goto fail;
189	}
190
191	idev->d_unit = unit;
192	if (path != NULL)
193	    *path = (*cp == 0) ? cp : cp + 1;
194	break;
195    case DEVT_ZFS:
196	err = zfs_parsedev((struct zfs_devdesc *)idev, np, path);
197	if (err != 0)
198	    goto fail;
199	break;
200    default:
201	err = EINVAL;
202	goto fail;
203    }
204    idev->d_dev = dv;
205    idev->d_type = dv->dv_type;
206    if (dev == NULL) {
207	free(idev);
208    } else {
209	*dev = idev;
210    }
211    return(0);
212
213 fail:
214    free(idev);
215    return(err);
216}
217
218
219char *
220i386_fmtdev(void *vdev)
221{
222    struct i386_devdesc	*dev = (struct i386_devdesc *)vdev;
223    static char		buf[128];	/* XXX device length constant? */
224    char		*cp;
225
226    switch(dev->d_type) {
227    case DEVT_NONE:
228	strcpy(buf, "(no device)");
229	break;
230
231    case DEVT_CD:
232	sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_unit);
233	break;
234
235    case DEVT_DISK:
236	cp = buf;
237	cp += sprintf(cp, "%s%d", dev->d_dev->dv_name, dev->d_unit);
238#ifdef LOADER_GPT_SUPPORT
239	if (dev->d_kind.biosdisk.partition == 0xff) {
240	    cp += sprintf(cp, "p%d", dev->d_kind.biosdisk.slice);
241	} else {
242#endif
243	    if (dev->d_kind.biosdisk.slice > 0)
244		cp += sprintf(cp, "s%d", dev->d_kind.biosdisk.slice);
245	    if (dev->d_kind.biosdisk.partition >= 0)
246		cp += sprintf(cp, "%c", dev->d_kind.biosdisk.partition + 'a');
247#ifdef LOADER_GPT_SUPPORT
248	}
249#endif
250	strcat(cp, ":");
251	break;
252
253    case DEVT_NET:
254	sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_unit);
255	break;
256    case DEVT_ZFS:
257	return(zfs_fmtdev(vdev));
258    }
259    return(buf);
260}
261
262
263/*
264 * Set currdev to suit the value being supplied in (value)
265 */
266int
267i386_setcurrdev(struct env_var *ev, int flags, const void *value)
268{
269    struct i386_devdesc	*ncurr;
270    int			rv;
271
272    if ((rv = i386_parsedev(&ncurr, value, NULL)) != 0)
273	return(rv);
274    free(ncurr);
275    env_setenv(ev->ev_name, flags | EV_NOHOOK, value, NULL, NULL);
276    return(0);
277}
278