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