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
37static int	userboot_parsedev(struct disk_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
45userboot_getdev(void **vdev, const char *devspec, const char **path)
46{
47    struct disk_devdesc **dev = (struct disk_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 = userboot_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(userboot_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
85userboot_parsedev(struct disk_devdesc **dev, const char *devspec, const char **path)
86{
87    struct disk_devdesc *idev;
88    struct devsw	*dv;
89    int			i, unit, 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 disk_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	err = disk_parsedev(idev, np, path);
116	if (err != 0)
117	    goto fail;
118	break;
119
120    case DEVT_CD:
121    case DEVT_NET:
122    case DEVT_ZFS:
123	unit = 0;
124
125	if (*np && (*np != ':')) {
126	    unit = strtol(np, &cp, 0);	/* get unit number if present */
127	    if (cp == np) {
128		err = EUNIT;
129		goto fail;
130	    }
131	} else {
132		cp = np;
133	}
134	if (*cp && (*cp != ':')) {
135	    err = EINVAL;
136	    goto fail;
137	}
138
139	idev->d_unit = unit;
140	if (path != NULL)
141	    *path = (*cp == 0) ? cp : cp + 1;
142	break;
143
144    default:
145	err = EINVAL;
146	goto fail;
147    }
148    idev->d_dev = dv;
149    idev->d_type = dv->dv_type;
150    if (dev == NULL) {
151	free(idev);
152    } else {
153	*dev = idev;
154    }
155    return(0);
156
157 fail:
158    free(idev);
159    return(err);
160}
161
162
163char *
164userboot_fmtdev(void *vdev)
165{
166    struct disk_devdesc	*dev = (struct disk_devdesc *)vdev;
167    static char		buf[128];	/* XXX device length constant? */
168
169    switch(dev->d_type) {
170    case DEVT_NONE:
171	strcpy(buf, "(no device)");
172	break;
173
174    case DEVT_CD:
175	sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_unit);
176	break;
177
178    case DEVT_DISK:
179	return (disk_fmtdev(vdev));
180
181    case DEVT_NET:
182    case DEVT_ZFS:
183	sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_unit);
184	break;
185    }
186    return(buf);
187}
188
189
190/*
191 * Set currdev to suit the value being supplied in (value)
192 */
193int
194userboot_setcurrdev(struct env_var *ev, int flags, const void *value)
195{
196    struct disk_devdesc	*ncurr;
197    int			rv;
198
199    if ((rv = userboot_parsedev(&ncurr, value, NULL)) != 0)
200	return(rv);
201    free(ncurr);
202    env_setenv(ev->ev_name, flags | EV_NOHOOK, value, NULL, NULL);
203    return(0);
204}
205