devicename.c revision 332154 
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/11/stand/uboot/lib/devicename.c 332154 2018-04-06 21:37:25Z kevans $");
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->dd.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->dd.d_dev = dv;
150	if (dev == NULL) {
151		free(idev);
152	} else {
153		*dev = idev;
154	}
155	return (0);
156
157fail:
158	free(idev);
159	return (err);
160}
161
162
163char *
164uboot_fmtdev(void *vdev)
165{
166	struct uboot_devdesc *dev = (struct uboot_devdesc *)vdev;
167	static char buf[128];
168
169	switch(dev->dd.d_dev->dv_type) {
170	case DEVT_NONE:
171		strcpy(buf, "(no device)");
172		break;
173
174	case DEVT_DISK:
175#ifdef LOADER_DISK_SUPPORT
176		return (disk_fmtdev(vdev));
177#endif
178
179	case DEVT_NET:
180		sprintf(buf, "%s%d:", dev->dd.d_dev->dv_name, dev->dd.d_unit);
181		break;
182	}
183	return(buf);
184}
185
186/*
187 * Set currdev to suit the value being supplied in (value).
188 */
189int
190uboot_setcurrdev(struct env_var *ev, int flags, const void *value)
191{
192	struct uboot_devdesc *ncurr;
193	int rv;
194
195	if ((rv = uboot_parsedev(&ncurr, value, NULL)) != 0)
196		return (rv);
197	free(ncurr);
198	env_setenv(ev->ev_name, flags | EV_NOHOOK, value, NULL, NULL);
199	return (0);
200}
201