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 /* 29 * MD bootstrap main() and assorted miscellaneous 30 * commands. 31 */ 32 33 #include <stand.h> 34 #include <stddef.h> 35 #include <string.h> 36 #include <machine/bootinfo.h> 37 #include <machine/cpufunc.h> 38 #include <machine/psl.h> 39 #include <sys/disk.h> 40 #include <sys/reboot.h> 41 #include <common/drv.h> 42 43 #include "bootstrap.h" 44 #include "common/bootargs.h" 45 #include "libi386/libi386.h" 46 #include <smbios.h> 47 #include "btxv86.h" 48 49 #ifdef LOADER_ZFS_SUPPORT 50 #include <sys/zfs_bootenv.h> 51 #include "libzfs.h" 52 #endif 53 54 _Static_assert(sizeof(struct bootargs) == BOOTARGS_SIZE, "Bootarg size bad"); 55 _Static_assert(offsetof(struct bootargs, bootinfo) == BA_BOOTINFO, "BA_BOOTINFO"); 56 _Static_assert(offsetof(struct bootargs, bootflags) == BA_BOOTFLAGS, "BA_BOOTFLAGS"); 57 _Static_assert(offsetof(struct bootinfo, bi_size) == BI_SIZE, "BI_SIZE"); 58 59 /* Arguments passed in from the boot1/boot2 loader */ 60 static struct bootargs *kargs; 61 static uint32_t initial_howto; 62 static uint32_t initial_bootdev; 63 static struct bootinfo *initial_bootinfo; 64 65 struct arch_switch archsw; /* MI/MD interface boundary */ 66 67 static void extract_currdev(void); 68 static int isa_inb(int port); 69 static void isa_outb(int port, int value); 70 void exit(int code); 71 #ifdef LOADER_GELI_SUPPORT 72 #include "geliboot.h" 73 struct geli_boot_args *gargs; 74 struct geli_boot_data *gbdata; 75 #endif 76 #ifdef LOADER_ZFS_SUPPORT 77 struct zfs_boot_args *zargs; 78 static void i386_zfs_probe(void); 79 #endif 80 81 /* XXX debugging */ 82 extern char end[]; 83 84 static void *heap_top; 85 static void *heap_bottom; 86 87 caddr_t 88 ptov(uintptr_t x) 89 { 90 return (PTOV(x)); 91 } 92 93 int 94 main(void) 95 { 96 /* Pick up arguments */ 97 kargs = (void *)__args; 98 initial_howto = kargs->howto; 99 initial_bootdev = kargs->bootdev; 100 initial_bootinfo = kargs->bootinfo ? 101 (struct bootinfo *)PTOV(kargs->bootinfo) : NULL; 102 103 /* Initialize the v86 register set to a known-good state. */ 104 bzero(&v86, sizeof(v86)); 105 v86.efl = PSL_RESERVED_DEFAULT | PSL_I; 106 107 /* 108 * Initialise the heap as early as possible. 109 * Once this is done, malloc() is usable. 110 */ 111 bios_getmem(); 112 113 #if defined(LOADER_BZIP2_SUPPORT) || \ 114 defined(LOADER_GPT_SUPPORT) || defined(LOADER_ZFS_SUPPORT) 115 if (high_heap_size > 0) { 116 heap_top = PTOV(high_heap_base + high_heap_size); 117 heap_bottom = PTOV(high_heap_base); 118 if (high_heap_base < memtop_copyin) 119 memtop_copyin = high_heap_base; 120 } else 121 #endif 122 { 123 heap_top = (void *)PTOV(bios_basemem); 124 heap_bottom = (void *)end; 125 } 126 setheap(heap_bottom, heap_top); 127 128 /* 129 * Now that malloc is usable, allocate a buffer for tslog and start 130 * logging timestamps during the boot process. 131 */ 132 tslog_init(); 133 134 /* 135 * detect ACPI for future reference. This may set console to comconsole 136 * if we do have ACPI SPCR table. 137 */ 138 biosacpi_detect(); 139 140 /* 141 * XXX Chicken-and-egg problem; we want to have console output early, 142 * but some console attributes may depend on reading from eg. the boot 143 * device, which we can't do yet. 144 * 145 * We can use printf() etc. once this is done. 146 * If the previous boot stage has requested a serial console, 147 * prefer that. 148 */ 149 bi_setboothowto(initial_howto); 150 if (initial_howto & RB_MULTIPLE) { 151 if (initial_howto & RB_SERIAL) 152 setenv("console", "comconsole vidconsole", 1); 153 else 154 setenv("console", "vidconsole comconsole", 1); 155 } else if (initial_howto & RB_SERIAL) { 156 setenv("console", "comconsole", 1); 157 } else if (initial_howto & RB_MUTE) { 158 setenv("console", "nullconsole", 1); 159 } 160 cons_probe(); 161 162 /* Set up currdev variable to have hooks in place. */ 163 env_setenv("currdev", EV_VOLATILE | EV_NOHOOK, "", 164 gen_setcurrdev, env_nounset); 165 166 /* 167 * Initialise the block cache. Set the upper limit. 168 */ 169 bcache_init(32768, 512); 170 171 /* 172 * Special handling for PXE and CD booting. 173 */ 174 if (kargs->bootinfo == 0) { 175 /* 176 * We only want the PXE disk to try to init itself in the below 177 * walk through devsw if we actually booted off of PXE. 178 */ 179 if (kargs->bootflags & KARGS_FLAGS_PXE) 180 pxe_enable(kargs->pxeinfo ? 181 PTOV(kargs->pxeinfo) : NULL); 182 else if (kargs->bootflags & KARGS_FLAGS_CD) 183 bc_add(initial_bootdev); 184 } 185 186 archsw.arch_autoload = i386_autoload; 187 archsw.arch_getdev = i386_getdev; 188 archsw.arch_copyin = i386_copyin; 189 archsw.arch_copyout = i386_copyout; 190 archsw.arch_readin = i386_readin; 191 archsw.arch_isainb = isa_inb; 192 archsw.arch_isaoutb = isa_outb; 193 archsw.arch_hypervisor = x86_hypervisor; 194 #ifdef LOADER_ZFS_SUPPORT 195 archsw.arch_zfs_probe = i386_zfs_probe; 196 197 /* 198 * zfsboot and gptzfsboot have always passed KARGS_FLAGS_ZFS, 199 * so if that is set along with KARGS_FLAGS_EXTARG we know we 200 * can interpret the extarg data as a struct zfs_boot_args. 201 */ 202 #define KARGS_EXTARGS_ZFS (KARGS_FLAGS_EXTARG | KARGS_FLAGS_ZFS) 203 204 if ((kargs->bootflags & KARGS_EXTARGS_ZFS) == KARGS_EXTARGS_ZFS) { 205 zargs = (struct zfs_boot_args *)(kargs + 1); 206 } 207 #endif /* LOADER_ZFS_SUPPORT */ 208 209 #ifdef LOADER_GELI_SUPPORT 210 /* 211 * If we decided earlier that we have zfs_boot_args extarg data, 212 * and it is big enough to contain the embedded geli data 213 * (the early zfs_boot_args structs weren't), then init the gbdata 214 * pointer accordingly. If there is extarg data which isn't 215 * zfs_boot_args data, determine whether it is geli_boot_args data. 216 * Recent versions of gptboot set KARGS_FLAGS_GELI to indicate that. 217 * Earlier versions didn't, but we presume that's what we 218 * have if the extarg size exactly matches the size of the 219 * geli_boot_args struct during that pre-flag era. 220 */ 221 #define LEGACY_GELI_ARGS_SIZE 260 /* This can never change */ 222 223 #ifdef LOADER_ZFS_SUPPORT 224 if (zargs != NULL) { 225 if (zargs->size > offsetof(struct zfs_boot_args, gelidata)) { 226 gbdata = &zargs->gelidata; 227 } 228 } else 229 #endif /* LOADER_ZFS_SUPPORT */ 230 if ((kargs->bootflags & KARGS_FLAGS_EXTARG) != 0) { 231 gargs = (struct geli_boot_args *)(kargs + 1); 232 if ((kargs->bootflags & KARGS_FLAGS_GELI) || 233 gargs->size == LEGACY_GELI_ARGS_SIZE) { 234 gbdata = &gargs->gelidata; 235 } 236 } 237 238 if (gbdata != NULL) 239 import_geli_boot_data(gbdata); 240 #endif /* LOADER_GELI_SUPPORT */ 241 242 devinit(); 243 244 printf("BIOS %dkB/%dkB available memory\n", bios_basemem / 1024, 245 bios_extmem / 1024); 246 if (initial_bootinfo != NULL) { 247 initial_bootinfo->bi_basemem = bios_basemem / 1024; 248 initial_bootinfo->bi_extmem = bios_extmem / 1024; 249 } 250 251 /* detect SMBIOS for future reference */ 252 smbios_detect(NULL); 253 254 /* detect PCI BIOS for future reference */ 255 biospci_detect(); 256 257 printf("\n%s", bootprog_info); 258 259 extract_currdev(); /* set $currdev and $loaddev */ 260 autoload_font(true); 261 262 bios_getsmap(); 263 264 interact(); 265 266 /* if we ever get here, it is an error */ 267 return (1); 268 } 269 270 /* 271 * Set the 'current device' by (if possible) recovering the boot device as 272 * supplied by the initial bootstrap. 273 * 274 * XXX should be extended for netbooting. 275 */ 276 static void 277 extract_currdev(void) 278 { 279 struct i386_devdesc new_currdev; 280 #ifdef LOADER_ZFS_SUPPORT 281 char buf[20]; 282 char *bootonce; 283 #endif 284 int biosdev = -1; 285 286 /* Assume we are booting from a BIOS disk by default */ 287 new_currdev.dd.d_dev = &bioshd; 288 289 /* new-style boot loaders such as pxeldr and cdldr */ 290 if (kargs->bootinfo == 0) { 291 if ((kargs->bootflags & KARGS_FLAGS_CD) != 0) { 292 /* we are booting from a CD with cdboot */ 293 new_currdev.dd.d_dev = &bioscd; 294 new_currdev.dd.d_unit = bd_bios2unit(initial_bootdev); 295 } else if ((kargs->bootflags & KARGS_FLAGS_PXE) != 0) { 296 /* we are booting from pxeldr */ 297 new_currdev.dd.d_dev = &pxedisk; 298 new_currdev.dd.d_unit = 0; 299 } else { 300 /* we don't know what our boot device is */ 301 new_currdev.disk.d_slice = -1; 302 new_currdev.disk.d_partition = 0; 303 biosdev = -1; 304 } 305 #ifdef LOADER_ZFS_SUPPORT 306 } else if ((kargs->bootflags & KARGS_FLAGS_ZFS) != 0) { 307 /* 308 * zargs was set in main() if we have new style extended 309 * argument 310 */ 311 if (zargs != NULL && 312 zargs->size >= 313 offsetof(struct zfs_boot_args, primary_pool)) { 314 /* sufficient data is provided */ 315 new_currdev.zfs.pool_guid = zargs->pool; 316 new_currdev.zfs.root_guid = zargs->root; 317 if (zargs->size >= sizeof(*zargs) && 318 zargs->primary_vdev != 0) { 319 sprintf(buf, "%llu", zargs->primary_pool); 320 setenv("vfs.zfs.boot.primary_pool", buf, 1); 321 sprintf(buf, "%llu", zargs->primary_vdev); 322 setenv("vfs.zfs.boot.primary_vdev", buf, 1); 323 } 324 } else { 325 /* old style zfsboot block */ 326 new_currdev.zfs.pool_guid = kargs->zfspool; 327 new_currdev.zfs.root_guid = 0; 328 } 329 new_currdev.dd.d_dev = &zfs_dev; 330 331 if ((bootonce = malloc(VDEV_PAD_SIZE)) != NULL) { 332 if (zfs_get_bootonce(&new_currdev, OS_BOOTONCE_USED, 333 bootonce, VDEV_PAD_SIZE) == 0) { 334 setenv("zfs-bootonce", bootonce, 1); 335 } 336 free(bootonce); 337 (void) zfs_attach_nvstore(&new_currdev); 338 } 339 340 #endif 341 } else if ((initial_bootdev & B_MAGICMASK) != B_DEVMAGIC) { 342 /* The passed-in boot device is bad */ 343 new_currdev.disk.d_slice = -1; 344 new_currdev.disk.d_partition = 0; 345 biosdev = -1; 346 } else { 347 new_currdev.disk.d_slice = B_SLICE(initial_bootdev) - 1; 348 new_currdev.disk.d_partition = B_PARTITION(initial_bootdev); 349 biosdev = initial_bootinfo->bi_bios_dev; 350 351 /* 352 * If we are booted by an old bootstrap, we have to guess at 353 * the BIOS unit number. We will lose if there is more than 354 * one disk type and we are not booting from the 355 * lowest-numbered disk type (ie. SCSI when IDE also exists). 356 */ 357 if ((biosdev == 0) && (B_TYPE(initial_bootdev) != 2)) { 358 /* 359 * biosdev doesn't match major, assume harddisk 360 */ 361 biosdev = 0x80 + B_UNIT(initial_bootdev); 362 } 363 } 364 365 /* 366 * If we are booting off of a BIOS disk and we didn't succeed 367 * in determining which one we booted off of, just use disk0: 368 * as a reasonable default. 369 */ 370 if ((new_currdev.dd.d_dev->dv_type == bioshd.dv_type) && 371 ((new_currdev.dd.d_unit = bd_bios2unit(biosdev)) == -1)) { 372 printf("Can't work out which disk we are booting " 373 "from.\nGuessed BIOS device 0x%x not found by " 374 "probes, defaulting to disk0:\n", biosdev); 375 new_currdev.dd.d_unit = 0; 376 } 377 378 #ifdef LOADER_ZFS_SUPPORT 379 if (new_currdev.dd.d_dev->dv_type == DEVT_ZFS) 380 init_zfs_boot_options(devformat(&new_currdev.dd)); 381 #endif 382 383 set_currdev(devformat(&new_currdev.dd)); 384 } 385 386 COMMAND_SET(reboot, "reboot", "reboot the system", command_reboot); 387 388 static int 389 command_reboot(int argc, char *argv[]) 390 { 391 int i; 392 393 for (i = 0; devsw[i] != NULL; ++i) 394 if (devsw[i]->dv_cleanup != NULL) 395 (devsw[i]->dv_cleanup)(); 396 397 printf("Rebooting...\n"); 398 delay(1000000); 399 __exit(0); 400 } 401 402 /* provide this for panic, as it's not in the startup code */ 403 void 404 exit(int code) 405 { 406 __exit(code); 407 } 408 409 COMMAND_SET(heap, "heap", "show heap usage", command_heap); 410 411 static int 412 command_heap(int argc, char *argv[]) 413 { 414 mallocstats(); 415 printf("heap base at %p, top at %p, upper limit at %p\n", heap_bottom, 416 sbrk(0), heap_top); 417 return (CMD_OK); 418 } 419 420 /* ISA bus access functions for PnP. */ 421 static int 422 isa_inb(int port) 423 { 424 425 return (inb(port)); 426 } 427 428 static void 429 isa_outb(int port, int value) 430 { 431 432 outb(port, value); 433 } 434 435 #ifdef LOADER_ZFS_SUPPORT 436 static void 437 i386_zfs_probe(void) 438 { 439 char devname[32]; 440 struct i386_devdesc dev; 441 442 /* 443 * Open all the disks we can find and see if we can reconstruct 444 * ZFS pools from them. 445 */ 446 dev.dd.d_dev = &bioshd; 447 for (dev.dd.d_unit = 0; bd_unit2bios(&dev) >= 0; dev.dd.d_unit++) { 448 snprintf(devname, sizeof(devname), "%s%d:", bioshd.dv_name, 449 dev.dd.d_unit); 450 zfs_probe_dev(devname, NULL, true); 451 } 452 } 453 #endif 454