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 "libuboot.h"
36
37static int uboot_parsedev(struct uboot_devdesc **dev, const char *devspec,
38    const char **path);
39
40/*
41 * Point (dev) at an allocated device specifier for the device matching the
42 * path in (devspec). If it contains an explicit device specification,
43 * use that.  If not, use the default device.
44 */
45int
46uboot_getdev(void **vdev, const char *devspec, const char **path)
47{
48	struct uboot_devdesc **dev = (struct uboot_devdesc **)vdev;
49	int rv;
50
51	/*
52	 * If it looks like this is just a path and no
53	 * device, go with the current device.
54	 */
55	if ((devspec == NULL) || (devspec[0] == '/') ||
56	    (strchr(devspec, ':') == NULL)) {
57
58		if (((rv = uboot_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 (uboot_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>[<partition>]:
82 *
83 */
84static int
85uboot_parsedev(struct uboot_devdesc **dev, const char *devspec,
86    const char **path)
87{
88	struct uboot_devdesc *idev;
89	struct devsw *dv;
90	char *cp;
91	const char *np;
92	int i, unit, err;
93
94	/* minimum length check */
95	if (strlen(devspec) < 2)
96		return(EINVAL);
97
98	/* look for a device that matches */
99	for (i = 0, dv = NULL; devsw[i] != NULL; i++) {
100		if (!strncmp(devspec, devsw[i]->dv_name,
101		    strlen(devsw[i]->dv_name))) {
102			dv = devsw[i];
103			break;
104		}
105	}
106	if (dv == NULL)
107		return(ENOENT);
108	idev = malloc(sizeof(struct uboot_devdesc));
109	err = 0;
110	np = (devspec + strlen(dv->dv_name));
111
112	switch(dv->dv_type) {
113	case DEVT_NONE:
114		break;
115
116#ifdef LOADER_DISK_SUPPORT
117	case DEVT_DISK:
118		err = disk_parsedev((struct disk_devdesc *)idev, np, path);
119		if (err != 0)
120			goto fail;
121		break;
122#endif
123
124	case DEVT_NET:
125		unit = 0;
126
127		if (*np && (*np != ':')) {
128			/* get unit number if present */
129			unit = strtol(np, &cp, 0);
130			if (cp == np) {
131				err = EUNIT;
132				goto fail;
133			}
134		}
135		if (*cp && (*cp != ':')) {
136			err = EINVAL;
137			goto fail;
138		}
139		idev->d_unit = unit;
140
141		if (path != NULL)
142			*path = (*cp == 0) ? cp : cp + 1;
143		break;
144
145	default:
146		err = EINVAL;
147		goto fail;
148	}
149	idev->d_dev = dv;
150	idev->d_type = dv->dv_type;
151	if (dev == NULL) {
152		free(idev);
153	} else {
154		*dev = idev;
155	}
156	return (0);
157
158fail:
159	free(idev);
160	return (err);
161}
162
163
164char *
165uboot_fmtdev(void *vdev)
166{
167	struct uboot_devdesc *dev = (struct uboot_devdesc *)vdev;
168	static char buf[128];
169
170	switch(dev->d_type) {
171	case DEVT_NONE:
172		strcpy(buf, "(no device)");
173		break;
174
175	case DEVT_DISK:
176#ifdef LOADER_DISK_SUPPORT
177		return (disk_fmtdev(vdev));
178#endif
179
180	case DEVT_NET:
181		sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_unit);
182		break;
183	}
184	return(buf);
185}
186
187/*
188 * Set currdev to suit the value being supplied in (value).
189 */
190int
191uboot_setcurrdev(struct env_var *ev, int flags, const void *value)
192{
193	struct uboot_devdesc *ncurr;
194	int rv;
195
196	if ((rv = uboot_parsedev(&ncurr, value, NULL)) != 0)
197		return (rv);
198	free(ncurr);
199	env_setenv(ev->ev_name, flags | EV_NOHOOK, value, NULL, NULL);
200	return (0);
201}
202