initial commit
[freebsd-arm:freebsd-arm.git] / boot / i386 / libi386 / devicename.c
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 #include <sys/disklabel.h>
33 #include "bootstrap.h"
34 #include "libi386.h"
35
36 static int      i386_parsedev(struct i386_devdesc **dev, const char *devspec, const char **path);
37
38 /* 
39  * Point (dev) at an allocated device specifier for the device matching the
40  * path in (devspec). If it contains an explicit device specification,
41  * use that.  If not, use the default device.
42  */
43 int
44 i386_getdev(void **vdev, const char *devspec, const char **path)
45 {
46     struct i386_devdesc **dev = (struct i386_devdesc **)vdev;
47     int                         rv;
48     
49     /*
50      * If it looks like this is just a path and no
51      * device, go with the current device.
52      */
53     if ((devspec == NULL) || 
54         (devspec[0] == '/') || 
55         (strchr(devspec, ':') == NULL)) {
56
57         if (((rv = i386_parsedev(dev, getenv("currdev"), NULL)) == 0) &&
58             (path != NULL))
59                 *path = devspec;
60         return(rv);
61     }
62     
63     /*
64      * Try to parse the device name off the beginning of the devspec
65      */
66     return(i386_parsedev(dev, devspec, path));
67 }
68
69 /*
70  * Point (dev) at an allocated device specifier matching the string version
71  * at the beginning of (devspec).  Return a pointer to the remaining
72  * text in (path).
73  *
74  * In all cases, the beginning of (devspec) is compared to the names
75  * of known devices in the device switch, and then any following text
76  * is parsed according to the rules applied to the device type.
77  *
78  * For disk-type devices, the syntax is:
79  *
80  * disk<unit>[s<slice>][<partition>]:
81  * 
82  */
83 static int
84 i386_parsedev(struct i386_devdesc **dev, const char *devspec, const char **path)
85 {
86     struct i386_devdesc *idev;
87     struct devsw        *dv;
88     int                 i, unit, slice, partition, err;
89     char                *cp;
90     const char          *np;
91
92     /* minimum length check */
93     if (strlen(devspec) < 2)
94         return(EINVAL);
95
96     /* look for a device that matches */
97     for (i = 0, dv = NULL; devsw[i] != NULL; i++) {
98         if (!strncmp(devspec, devsw[i]->dv_name, strlen(devsw[i]->dv_name))) {
99             dv = devsw[i];
100             break;
101         }
102     }
103     if (dv == NULL)
104         return(ENOENT);
105     idev = malloc(sizeof(struct i386_devdesc));
106     err = 0;
107     np = (devspec + strlen(dv->dv_name));
108         
109     switch(dv->dv_type) {
110     case DEVT_NONE:                     /* XXX what to do here?  Do we care? */
111         break;
112
113     case DEVT_DISK:
114         unit = -1;
115         slice = -1;
116         partition = -1;
117         if (*np && (*np != ':')) {
118             unit = strtol(np, &cp, 10); /* next comes the unit number */
119             if (cp == np) {
120                 err = EUNIT;
121                 goto fail;
122             }
123 #ifdef LOADER_GPT_SUPPORT
124             if (*cp == 'p') {           /* got a GPT partition */
125                 np = cp + 1;
126                 slice = strtol(np, &cp, 10);
127                 if (cp == np) {
128                     err = ESLICE;
129                     goto fail;
130                 }
131                 if (*cp && (*cp != ':')) {
132                     err = EINVAL;
133                     goto fail;
134                 }
135                 partition = 0xff;
136             } else {
137 #endif
138                 if (*cp == 's') {               /* got a slice number */
139                     np = cp + 1;
140                     slice = strtol(np, &cp, 10);
141                     if (cp == np) {
142                         err = ESLICE;
143                         goto fail;
144                     }
145                 }
146                 if (*cp && (*cp != ':')) {
147                     partition = *cp - 'a';      /* got a partition number */
148                     if ((partition < 0) || (partition >= MAXPARTITIONS)) {
149                         err = EPART;
150                         goto fail;
151                     }
152                     cp++;
153                 }
154 #ifdef LOADER_GPT_SUPPORT
155             }
156 #endif
157         } else {
158                 cp = np;
159         }
160         if (*cp && (*cp != ':')) {
161             err = EINVAL;
162             goto fail;
163         }
164
165         idev->d_unit = unit;
166         idev->d_kind.biosdisk.slice = slice;
167         idev->d_kind.biosdisk.partition = partition;
168         if (path != NULL)
169             *path = (*cp == 0) ? cp : cp + 1;
170         break;
171
172     case DEVT_CD:
173     case DEVT_NET:
174     case DEVT_ZFS:
175         unit = 0;
176
177         if (*np && (*np != ':')) {
178             unit = strtol(np, &cp, 0);  /* get unit number if present */
179             if (cp == np) {
180                 err = EUNIT;
181                 goto fail;
182             }
183         } else {
184                 cp = np;
185         }
186         if (*cp && (*cp != ':')) {
187             err = EINVAL;
188             goto fail;
189         }
190
191         idev->d_unit = unit;
192         if (path != NULL)
193             *path = (*cp == 0) ? cp : cp + 1;
194         break;
195
196     default:
197         err = EINVAL;
198         goto fail;
199     }
200     idev->d_dev = dv;
201     idev->d_type = dv->dv_type;
202     if (dev == NULL) {
203         free(idev);
204     } else {
205         *dev = idev;
206     }
207     return(0);
208
209  fail:
210     free(idev);
211     return(err);
212 }
213
214
215 char *
216 i386_fmtdev(void *vdev)
217 {
218     struct i386_devdesc *dev = (struct i386_devdesc *)vdev;
219     static char         buf[128];       /* XXX device length constant? */
220     char                *cp;
221     
222     switch(dev->d_type) {
223     case DEVT_NONE:
224         strcpy(buf, "(no device)");
225         break;
226
227     case DEVT_CD:
228         sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_unit);
229         break;
230
231     case DEVT_DISK:
232         cp = buf;
233         cp += sprintf(cp, "%s%d", dev->d_dev->dv_name, dev->d_unit);
234 #ifdef LOADER_GPT_SUPPORT
235         if (dev->d_kind.biosdisk.partition == 0xff) {
236             cp += sprintf(cp, "p%d", dev->d_kind.biosdisk.slice);
237         } else {
238 #endif
239             if (dev->d_kind.biosdisk.slice > 0)
240                 cp += sprintf(cp, "s%d", dev->d_kind.biosdisk.slice);
241             if (dev->d_kind.biosdisk.partition >= 0)
242                 cp += sprintf(cp, "%c", dev->d_kind.biosdisk.partition + 'a');
243 #ifdef LOADER_GPT_SUPPORT
244         }
245 #endif
246         strcat(cp, ":");
247         break;
248
249     case DEVT_NET:
250     case DEVT_ZFS:
251         sprintf(buf, "%s%d:", dev->d_dev->dv_name, dev->d_unit);
252         break;
253     }
254     return(buf);
255 }
256
257
258 /*
259  * Set currdev to suit the value being supplied in (value)
260  */
261 int
262 i386_setcurrdev(struct env_var *ev, int flags, const void *value)
263 {
264     struct i386_devdesc *ncurr;
265     int                 rv;
266
267     if ((rv = i386_parsedev(&ncurr, value, NULL)) != 0)
268         return(rv);
269     free(ncurr);
270     env_setenv(ev->ev_name, flags | EV_NOHOOK, value, NULL, NULL);
271     return(0);
272 }