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/i386/libi386/devicename.c 346476 2019-04-21 03:36:05Z kevans $");
29
30#include <stand.h>
31#include <string.h>
32#include "bootstrap.h"
33#include "disk.h"
34#include "libi386.h"
35#include "libzfs.h"
36
37static int	i386_parsedev(struct i386_devdesc **dev, const char *devspec, const char **path);
38
39/*
40 * Point (dev) at an allocated device specifier for the device matching the
41 * path in (devspec). If it contains an explicit device specification,
42 * use that.  If not, use the default device.
43 */
44int
45i386_getdev(void **vdev, const char *devspec, const char **path)
46{
47    struct i386_devdesc **dev = (struct i386_devdesc **)vdev;
48    int				rv;
49
50    /*
51     * If it looks like this is just a path and no
52     * device, go with the current device.
53     */
54    if ((devspec == NULL) ||
55	(devspec[0] == '/') ||
56	(strchr(devspec, ':') == NULL)) {
57
58	if (((rv = i386_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(i386_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>[s<slice>][<partition>]:
82 *
83 */
84static int
85i386_parsedev(struct i386_devdesc **dev, const char *devspec, const char **path)
86{
87    struct i386_devdesc *idev;
88    struct devsw	*dv;
89    int			i, unit, err;
90    char		*cp;
91    const char		*np;
92
93    /* minimum length check */
94    if (strlen(devspec) < 2)
95	return(EINVAL);
96
97    /* look for a device that matches */
98    for (i = 0, dv = NULL; devsw[i] != NULL; i++) {
99	if (!strncmp(devspec, devsw[i]->dv_name, strlen(devsw[i]->dv_name))) {
100	    dv = devsw[i];
101	    break;
102	}
103    }
104    if (dv == NULL)
105	return(ENOENT);
106
107    np = (devspec + strlen(dv->dv_name));
108    idev = NULL;
109    err = 0;
110
111    switch(dv->dv_type) {
112    case DEVT_NONE:
113	break;
114
115    case DEVT_DISK:
116	idev = malloc(sizeof(struct i386_devdesc));
117	if (idev == NULL)
118	    return (ENOMEM);
119
120	err = disk_parsedev((struct disk_devdesc *)idev, np, path);
121	if (err != 0)
122	    goto fail;
123	break;
124
125    case DEVT_ZFS:
126	idev = malloc(sizeof (struct zfs_devdesc));
127	if (idev == NULL)
128	    return (ENOMEM);
129
130	err = zfs_parsedev((struct zfs_devdesc *)idev, np, path);
131	if (err != 0)
132	    goto fail;
133	break;
134
135    default:
136	idev = malloc(sizeof (struct devdesc));
137	if (idev == NULL)
138	    return (ENOMEM);
139
140	unit = 0;
141	cp = (char *)np;
142
143	if (*np && (*np != ':')) {
144	    unit = strtol(np, &cp, 0);	/* get unit number if present */
145	    if (cp == np) {
146		err = EUNIT;
147		goto fail;
148	    }
149	}
150
151	if (*cp && (*cp != ':')) {
152	    err = EINVAL;
153	    goto fail;
154	}
155
156	idev->dd.d_unit = unit;
157	if (path != NULL)
158	    *path = (*cp == 0) ? cp : cp + 1;
159	break;
160    }
161    idev->dd.d_dev = dv;
162    if (dev != NULL)
163	*dev = idev;
164    else
165	free(idev);
166
167    return(0);
168
169 fail:
170    free(idev);
171    return(err);
172}
173
174
175char *
176i386_fmtdev(void *vdev)
177{
178    struct i386_devdesc	*dev = (struct i386_devdesc *)vdev;
179    static char		buf[128];	/* XXX device length constant? */
180
181    switch(dev->dd.d_dev->dv_type) {
182    case DEVT_NONE:
183	strcpy(buf, "(no device)");
184	break;
185
186    case DEVT_CD:
187    case DEVT_NET:
188	sprintf(buf, "%s%d:", dev->dd.d_dev->dv_name, dev->dd.d_unit);
189	break;
190
191    case DEVT_DISK:
192	return (disk_fmtdev(vdev));
193
194    case DEVT_ZFS:
195	return(zfs_fmtdev(vdev));
196    }
197    return(buf);
198}
199
200
201/*
202 * Set currdev to suit the value being supplied in (value)
203 */
204int
205i386_setcurrdev(struct env_var *ev, int flags, const void *value)
206{
207    struct i386_devdesc	*ncurr;
208    int			rv;
209
210    if ((rv = i386_parsedev(&ncurr, value, NULL)) != 0)
211	return(rv);
212    free(ncurr);
213    env_setenv(ev->ev_name, flags | EV_NOHOOK, value, NULL, NULL);
214    return(0);
215}
216