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 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 */ 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