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 <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 * 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 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 * 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 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 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 * 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 /* 173 * mount new rootfs and unmount old, set "currdev" environment variable. 174 */ 175 int mount_currdev(struct env_var *ev, int flags, const void *value) 176 { 177 int rv; 178 179 /* mount new rootfs */ 180 rv = mount(value, "/", 0, NULL); 181 if (rv == 0) { 182 /* 183 * Note we unmount any previously mounted fs only after 184 * successfully mounting the new because we do not want to 185 * end up with unmounted rootfs. 186 */ 187 if (ev->ev_value != NULL) 188 unmount(ev->ev_value, 0); 189 env_setenv(ev->ev_name, flags | EV_NOHOOK, value, NULL, NULL); 190 } 191 return (rv); 192 } 193 194 /* 195 * Set currdev to suit the value being supplied in (value) 196 */ 197 int 198 gen_setcurrdev(struct env_var *ev, int flags, const void *value) 199 { 200 struct devdesc *ncurr; 201 int rv; 202 203 if ((rv = devparse(&ncurr, value, NULL)) != 0) 204 return (rv); 205 free(ncurr); 206 207 return (mount_currdev(ev, flags, value)); 208 } 209 210 /* 211 * Wrapper to set currdev and loaddev at the same time. 212 */ 213 void 214 set_currdev(const char *devname) 215 { 216 217 env_setenv("currdev", EV_VOLATILE, devname, gen_setcurrdev, 218 env_nounset); 219 /* 220 * Don't execute hook here; the loaddev hook makes it immutable 221 * once we've determined what the proper currdev is. 222 */ 223 env_setenv("loaddev", EV_VOLATILE | EV_NOHOOK, devname, env_noset, 224 env_nounset); 225 } 226