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$");
29
30#include <stand.h>
31#include <string.h>
32
33#include "bootstrap.h"
34#include "disk.h"
35#include "libuserboot.h"
36
37#if defined(USERBOOT_ZFS_SUPPORT)
38#include "../zfs/libzfs.h"
39#endif
40
41static int	userboot_parsedev(struct disk_devdesc **dev, const char *devspec, const char **path);
42
43/*
44 * Point (dev) at an allocated device specifier for the device matching the
45 * path in (devspec). If it contains an explicit device specification,
46 * use that.  If not, use the default device.
47 */
48int
49userboot_getdev(void **vdev, const char *devspec, const char **path)
50{
51    struct disk_devdesc **dev = (struct disk_devdesc **)vdev;
52    int				rv;
53
54    /*
55     * If it looks like this is just a path and no
56     * device, go with the current device.
57     */
58    if ((devspec == NULL) ||
59	(devspec[0] == '/') ||
60	(strchr(devspec, ':') == NULL)) {
61
62	if (((rv = userboot_parsedev(dev, getenv("currdev"), NULL)) == 0) &&
63	    (path != NULL))
64		*path = devspec;
65	return(rv);
66    }
67
68    /*
69     * Try to parse the device name off the beginning of the devspec
70     */
71    return(userboot_parsedev(dev, devspec, path));
72}
73
74/*
75 * Point (dev) at an allocated device specifier matching the string version
76 * at the beginning of (devspec).  Return a pointer to the remaining
77 * text in (path).
78 *
79 * In all cases, the beginning of (devspec) is compared to the names
80 * of known devices in the device switch, and then any following text
81 * is parsed according to the rules applied to the device type.
82 *
83 * For disk-type devices, the syntax is:
84 *
85 * disk<unit>[s<slice>][<partition>]:
86 *
87 */
88static int
89userboot_parsedev(struct disk_devdesc **dev, const char *devspec, const char **path)
90{
91    struct disk_devdesc *idev;
92    struct devsw	*dv;
93    int			i, unit, err;
94    char		*cp;
95    const char		*np;
96
97    /* minimum length check */
98    if (strlen(devspec) < 2)
99	return(EINVAL);
100
101    /* look for a device that matches */
102    for (i = 0, dv = NULL; devsw[i] != NULL; i++) {
103	if (!strncmp(devspec, devsw[i]->dv_name, strlen(devsw[i]->dv_name))) {
104	    dv = devsw[i];
105	    break;
106	}
107    }
108    if (dv == NULL)
109	return(ENOENT);
110    idev = malloc(sizeof(struct disk_devdesc));
111    err = 0;
112    np = (devspec + strlen(dv->dv_name));
113
114    switch(dv->dv_type) {
115    case DEVT_NONE:			/* XXX what to do here?  Do we care? */
116	break;
117
118    case DEVT_DISK:
119	err = disk_parsedev(idev, np, path);
120	if (err != 0)
121	    goto fail;
122	break;
123
124    case DEVT_CD:
125    case DEVT_NET:
126	unit = 0;
127
128	if (*np && (*np != ':')) {
129	    unit = strtol(np, &cp, 0);	/* get unit number if present */
130	    if (cp == np) {
131		err = EUNIT;
132		goto fail;
133	    }
134	} else {
135		cp = np;
136	}
137	if (*cp && (*cp != ':')) {
138	    err = EINVAL;
139	    goto fail;
140	}
141
142	idev->d_unit = unit;
143	if (path != NULL)
144	    *path = (*cp == 0) ? cp : cp + 1;
145	break;
146
147    case DEVT_ZFS:
148#if defined(USERBOOT_ZFS_SUPPORT)
149	    err = zfs_parsedev((struct zfs_devdesc *)idev, np, path);
150	    if (err != 0)
151		    goto fail;
152	    break;
153#else
154	    /* FALLTHROUGH */
155#endif
156
157    default:
158	err = EINVAL;
159	goto fail;
160    }
161    idev->d_dev = dv;
162    idev->d_type = dv->dv_type;
163    if (dev == NULL) {
164	free(idev);
165    } else {
166	*dev = idev;
167    }
168    return(0);
169
170 fail:
171    free(idev);
172    return(err);
173}
174
175
176char *
177userboot_fmtdev(void *vdev)
178{
179    struct disk_devdesc	*dev = (struct disk_devdesc *)vdev;
180    static char		buf[128];	/* XXX device length constant? */
181
182    switch(dev->d_type) {
183    case DEVT_NONE:
184	strcpy(buf, "(no device)");
185	break;
186
187    case DEVT_CD:
188	sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_unit);
189	break;
190
191    case DEVT_DISK:
192	return (disk_fmtdev(vdev));
193
194    case DEVT_NET:
195	sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_unit);
196	break;
197
198    case DEVT_ZFS:
199#if defined(USERBOOT_ZFS_SUPPORT)
200	return (zfs_fmtdev(vdev));
201#else
202	sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_unit);
203#endif
204	break;
205    }
206    return(buf);
207}
208
209
210/*
211 * Set currdev to suit the value being supplied in (value)
212 */
213int
214userboot_setcurrdev(struct env_var *ev, int flags, const void *value)
215{
216    struct disk_devdesc	*ncurr;
217    int			rv;
218
219    if ((rv = userboot_parsedev(&ncurr, value, NULL)) != 0)
220	return(rv);
221    free(ncurr);
222    env_setenv(ev->ev_name, flags | EV_NOHOOK, value, NULL, NULL);
223    return(0);
224}
225