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
37 static 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 */
45 int
uboot_getdev(void ** vdev,const char * devspec,const char ** path)46 uboot_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 */
84 static int
uboot_parsedev(struct uboot_devdesc ** dev,const char * devspec,const char ** path)85 uboot_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
158 fail:
159 free(idev);
160 return (err);
161 }
162
163
164 char *
uboot_fmtdev(void * vdev)165 uboot_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 */
190 int
uboot_setcurrdev(struct env_var * ev,int flags,const void * value)191 uboot_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