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/10/sys/boot/powerpc/ps3/devicename.c 224857 2011-08-14 00:20:37Z nwhitehorn $");
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 
39 static 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  */
47 int
ps3_getdev(void ** vdev,const char * devspec,const char ** path)48 ps3_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  */
86 static int
ps3_parsedev(struct ps3_devdesc ** dev,const char * devspec,const char ** path)87 ps3_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 
183 fail:
184 	free(idev);
185 	return (err);
186 }
187 
188 
189 char *
ps3_fmtdev(void * vdev)190 ps3_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  */
227 int
ps3_setcurrdev(struct env_var * ev,int flags,const void * value)228 ps3_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