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 <string.h>
28#include <stand.h>
29#include <bootstrap.h>
30
31/*
32 * Concatenate the (argc) elements of (argv) into a single string, and return
33 * a copy of same.
34 */
35char *
36unargv(int argc, char *argv[])
37{
38    size_t	hlong;
39    int		i;
40    char	*cp;
41
42    for (i = 0, hlong = 0; i < argc; i++)
43	hlong += strlen(argv[i]) + 2;
44
45    if(hlong == 0)
46	return(NULL);
47
48    cp = malloc(hlong);
49    cp[0] = 0;
50    for (i = 0; i < argc; i++) {
51	strcat(cp, argv[i]);
52	if (i < (argc - 1))
53	  strcat(cp, " ");
54    }
55
56    return(cp);
57}
58
59/*
60 * Get the length of a string in kernel space
61 */
62size_t
63strlenout(vm_offset_t src)
64{
65    char	c;
66    size_t	len;
67
68    for (len = 0; ; len++) {
69	archsw.arch_copyout(src++, &c, 1);
70	if (c == 0)
71	    break;
72    }
73    return(len);
74}
75
76/*
77 * Make a duplicate copy of a string in kernel space
78 */
79char *
80strdupout(vm_offset_t str)
81{
82    char	*result, *cp;
83
84    result = malloc(strlenout(str) + 1);
85    for (cp = result; ;cp++) {
86	archsw.arch_copyout(str++, cp, 1);
87	if (*cp == 0)
88	    break;
89    }
90    return(result);
91}
92
93/* Zero a region in kernel space. */
94void
95kern_bzero(vm_offset_t dest, size_t len)
96{
97	char buf[256];
98	size_t chunk, resid;
99
100	bzero(buf, sizeof(buf));
101	resid = len;
102	while (resid > 0) {
103		chunk = min(sizeof(buf), resid);
104		archsw.arch_copyin(buf, dest, chunk);
105		resid -= chunk;
106		dest += chunk;
107	}
108}
109
110/*
111 * Read the specified part of a file to kernel space.  Unlike regular
112 * pread, the file pointer is advanced to the end of the read data,
113 * and it just returns 0 if successful.
114 */
115int
116kern_pread(readin_handle_t fd, vm_offset_t dest, size_t len, off_t off)
117{
118
119	if (VECTX_LSEEK(fd, off, SEEK_SET) == -1) {
120#ifdef DEBUG
121		printf("\nlseek failed\n");
122#endif
123		return (-1);
124	}
125	if ((size_t)archsw.arch_readin(fd, dest, len) != len) {
126#ifdef DEBUG
127		printf("\nreadin failed\n");
128#endif
129		return (-1);
130	}
131	return (0);
132}
133
134/*
135 * Read the specified part of a file to a malloced buffer.  The file
136 * pointer is advanced to the end of the read data.
137 */
138/* coverity[ -tainted_data_return ] */
139void *
140alloc_pread(readin_handle_t fd, off_t off, size_t len)
141{
142	void *buf;
143
144	buf = malloc(len);
145	if (buf == NULL) {
146#ifdef DEBUG
147		printf("\nmalloc(%d) failed\n", (int)len);
148#endif
149		errno = ENOMEM;
150		return (NULL);
151	}
152	if (VECTX_LSEEK(fd, off, SEEK_SET) == -1) {
153#ifdef DEBUG
154		printf("\nlseek failed\n");
155#endif
156		free(buf);
157		return (NULL);
158	}
159	if ((size_t)VECTX_READ(fd, buf, len) != len) {
160#ifdef DEBUG
161		printf("\nread failed\n");
162#endif
163		free(buf);
164		return (NULL);
165	}
166	return (buf);
167}
168
169/*
170 * mount new rootfs and unmount old, set "currdev" environment variable.
171 */
172int mount_currdev(struct env_var *ev, int flags, const void *value)
173{
174	int rv;
175
176	/* mount new rootfs */
177	rv = mount(value, "/", 0, NULL);
178	if (rv == 0) {
179		/*
180		 * Note we unmount any previously mounted fs only after
181		 * successfully mounting the new because we do not want to
182		 * end up with unmounted rootfs.
183		 */
184		if (ev->ev_value != NULL)
185			unmount(ev->ev_value, 0);
186		env_setenv(ev->ev_name, flags | EV_NOHOOK, value, NULL, NULL);
187	}
188	return (rv);
189}
190
191/*
192 * Set currdev to suit the value being supplied in (value)
193 */
194int
195gen_setcurrdev(struct env_var *ev, int flags, const void *value)
196{
197	struct devdesc *ncurr;
198	int rv;
199
200	if ((rv = devparse(&ncurr, value, NULL)) != 0)
201		return (rv);
202	free(ncurr);
203
204	return (mount_currdev(ev, flags, value));
205}
206
207/*
208 * Wrapper to set currdev and loaddev at the same time.
209 */
210void
211set_currdev(const char *devname)
212{
213
214	env_setenv("currdev", EV_VOLATILE, devname, gen_setcurrdev,
215	    env_nounset);
216	/*
217	 * Don't execute hook here; the loaddev hook makes it immutable
218	 * once we've determined what the proper currdev is.
219	 */
220	env_setenv("loaddev", EV_VOLATILE | EV_NOHOOK, devname, env_noset,
221	    env_nounset);
222}
223