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