#include <sys/param.h>
#include <sys/reboot.h>
#include <lib/libsa/stand.h>
#include <lib/libkern/libkern.h>
#include "loadfile.h"
#include "libi386.h"
#include "bootinfo.h"
#include "bootmod.h"
#include "vbe.h"
#ifdef SUPPORT_PS2
#include "biosmca.h"
#endif
#ifdef EFIBOOT
#include "efiboot.h"
#include "biosdisk.h"
#include "efidisk.h"
#undef DEBUG
#endif
#define BOOT_NARGS 6
#ifndef PAGE_SIZE
#define PAGE_SIZE 4096
#endif
#define MODULE_WARNING_SEC 5
#define MAXMODNAME 32
extern struct btinfo_console btinfo_console;
extern struct btinfo_rootdevice bi_root;
boot_module_t *boot_modules;
bool boot_modules_enabled = true;
bool kernel_loaded;
typedef struct userconf_command {
char *uc_text;
size_t uc_len;
struct userconf_command *uc_next;
} userconf_command_t;
userconf_command_t *userconf_commands = NULL;
struct btinfo_framebuffer btinfo_framebuffer;
struct btinfo_modulelist *btinfo_modulelist;
static size_t btinfo_modulelist_size;
static uint32_t image_end;
static char module_base[64] = "/";
static int howto;
static struct btinfo_userconfcommands *btinfo_userconfcommands = NULL;
static size_t btinfo_userconfcommands_size = 0;
static void module_init(const char *);
static void module_add_common(const char *, uint8_t);
static void userconf_init(void);
static void extract_device(const char *, char *, size_t);
static void module_base_path(char *, size_t, const char *);
static int module_open(boot_module_t *, int, const char *, const char *,
bool);
void
framebuffer_configure(struct btinfo_framebuffer *fb)
{
if (fb)
btinfo_framebuffer = *fb;
else {
btinfo_framebuffer.physaddr = 0;
btinfo_framebuffer.flags = 0;
}
}
void
module_add(char *name)
{
return module_add_common(name, BM_TYPE_KMOD);
}
void
splash_add(char *name)
{
return module_add_common(name, BM_TYPE_IMAGE);
}
void
rnd_add(char *name)
{
return module_add_common(name, BM_TYPE_RND);
}
void
fs_add(char *name)
{
return module_add_common(name, BM_TYPE_FS);
}
void
module_add_split(const char *name, uint8_t type)
{
char mod_name[MAXMODNAME];
int i;
const char *mp = name;
char *ep;
while (*mp) {
i = MAXMODNAME;
ep = mod_name;
while (--i) {
*ep = *mp;
if (*ep == '/')
*ep = '\0';
if (*ep == 0 ) {
if (ep != mod_name)
module_add_common(mod_name, type);
break;
}
ep++; mp++;
}
if (*ep != 0) {
printf("module name too long\n");
return;
}
if (*mp == '/') {
mp++;
}
}
}
static void
module_add_common(const char *name, uint8_t type)
{
boot_module_t *bm, *bmp;
size_t len;
char *str;
while (*name == ' ' || *name == '\t')
++name;
for (bm = boot_modules; bm != NULL; bm = bm->bm_next)
if (bm->bm_type == type && strcmp(bm->bm_path, name) == 0)
return;
bm = alloc(sizeof(boot_module_t));
len = strlen(name) + 1;
str = alloc(len);
if (bm == NULL || str == NULL) {
printf("couldn't allocate module\n");
return;
}
memcpy(str, name, len);
bm->bm_path = str;
bm->bm_next = NULL;
bm->bm_type = type;
if (boot_modules == NULL)
boot_modules = bm;
else {
for (bmp = boot_modules; bmp->bm_next;
bmp = bmp->bm_next)
;
bmp->bm_next = bm;
}
}
void
userconf_add(char *cmd)
{
userconf_command_t *uc;
size_t len;
char *text;
while (*cmd == ' ' || *cmd == '\t')
++cmd;
uc = alloc(sizeof(*uc));
if (uc == NULL) {
printf("couldn't allocate command\n");
return;
}
len = strlen(cmd) + 1;
text = alloc(len);
if (text == NULL) {
dealloc(uc, sizeof(*uc));
printf("couldn't allocate command\n");
return;
}
memcpy(text, cmd, len);
uc->uc_text = text;
uc->uc_len = len;
uc->uc_next = NULL;
if (userconf_commands == NULL)
userconf_commands = uc;
else {
userconf_command_t *ucp;
for (ucp = userconf_commands; ucp->uc_next != NULL;
ucp = ucp->uc_next)
;
ucp->uc_next = uc;
}
}
struct btinfo_prekern bi_prekern;
int has_prekern = 0;
static int
common_load_prekern(const char *file, u_long *basemem, u_long *extmem,
physaddr_t loadaddr, int floppy, u_long marks[MARK_MAX])
{
paddr_t kernpa_start, kernpa_end;
char prekernpath[] = "/prekern";
u_long prekern_start;
int fd, flags;
*extmem = getextmem();
*basemem = getbasemem();
marks[MARK_START] = loadaddr;
flags = LOAD_KERNEL & ~(LOAD_HDR|LOAD_SYM);
if ((fd = loadfile(prekernpath, marks, flags)) == -1)
return errno;
close(fd);
prekern_start = marks[MARK_START];
marks[MARK_START] = loadaddr;
marks[MARK_END] = loadaddr + (1UL << 21);
kernpa_start = (1UL << 21);
flags = (LOAD_KERNEL | LOAD_DYN) & ~(floppy ? LOAD_BACKWARDS : 0);
if ((fd = loadfile(file, marks, flags)) == -1)
return errno;
close(fd);
kernpa_end = marks[MARK_END] - loadaddr;
if (fsmod != NULL)
module_add_split(fsmod, BM_TYPE_KMOD);
bi_prekern.kernpa_start = kernpa_start;
bi_prekern.kernpa_end = kernpa_end;
BI_ADD(&bi_prekern, BTINFO_PREKERN, sizeof(struct btinfo_prekern));
#ifdef PASS_BIOSGEOM
bi_getbiosgeom();
#endif
#ifdef PASS_MEMMAP
bi_getmemmap();
#endif
marks[MARK_START] = prekern_start;
marks[MARK_END] = (((u_long)marks[MARK_END] + sizeof(int) - 1)) &
(-sizeof(int));
image_end = marks[MARK_END];
kernel_loaded = true;
return 0;
}
static int
common_load_kernel(const char *file, u_long *basemem, u_long *extmem,
physaddr_t loadaddr, int floppy, u_long marks[MARK_MAX])
{
int fd;
*extmem = getextmem();
*basemem = getbasemem();
marks[MARK_START] = loadaddr;
if ((fd = loadfile(file, marks,
LOAD_KERNEL & ~(floppy ? LOAD_BACKWARDS : 0))) == -1)
return errno;
close(fd);
if (fsmod != NULL)
module_add_split(fsmod, BM_TYPE_KMOD);
#ifdef PASS_BIOSGEOM
bi_getbiosgeom();
#endif
#ifdef PASS_MEMMAP
bi_getmemmap();
#endif
marks[MARK_END] = (((u_long) marks[MARK_END] + sizeof(int) - 1)) &
(-sizeof(int));
image_end = marks[MARK_END];
kernel_loaded = true;
return 0;
}
int
exec_netbsd(const char *file, physaddr_t loadaddr, int boothowto, int floppy,
void (*callback)(void))
{
uint32_t boot_argv[BOOT_NARGS];
u_long marks[MARK_MAX];
struct btinfo_symtab btinfo_symtab;
u_long extmem;
u_long basemem;
u_long entry;
int error;
#ifdef EFIBOOT
int i;
#endif
#ifdef DEBUG
printf("exec: file=%s loadaddr=0x%lx\n", file ? file : "NULL",
loadaddr);
#endif
BI_ALLOC(BTINFO_MAX);
BI_ADD(&btinfo_console, BTINFO_CONSOLE, sizeof(struct btinfo_console));
if (bi_root.devname[0])
BI_ADD(&bi_root, BTINFO_ROOTDEVICE, sizeof(struct btinfo_rootdevice));
howto = boothowto;
memset(marks, 0, sizeof(marks));
if (has_prekern) {
error = common_load_prekern(file, &basemem, &extmem, loadaddr,
floppy, marks);
} else {
error = common_load_kernel(file, &basemem, &extmem, loadaddr,
floppy, marks);
}
if (error) {
errno = error;
goto out;
}
#ifdef EFIBOOT
BI_ADD(&bi_disk, BTINFO_BOOTDISK, sizeof(bi_disk));
BI_ADD(&bi_wedge, BTINFO_BOOTWEDGE, sizeof(bi_wedge));
efidisk_getbiosgeom();
efi_load_start = marks[MARK_START];
marks[MARK_START] -= efi_loadaddr;
marks[MARK_ENTRY] -= efi_loadaddr;
marks[MARK_DATA] -= efi_loadaddr;
marks[MARK_SYM] -= efi_loadaddr;
marks[MARK_END] -= efi_loadaddr;
#endif
boot_argv[0] = boothowto;
boot_argv[1] = 0;
boot_argv[2] = vtophys(bootinfo);
boot_argv[3] = marks[MARK_END];
boot_argv[4] = extmem;
boot_argv[5] = basemem;
if (boot_modules_enabled) {
module_init(file);
if (btinfo_modulelist) {
#ifdef EFIBOOT
struct bi_modulelist_entry *bim;
bim = (void *)(btinfo_modulelist + 1);
for (i = 0; i < btinfo_modulelist->num; i++, bim++)
bim->base -= efi_loadaddr;
btinfo_modulelist->endpa -= efi_loadaddr;
#endif
BI_ADD(btinfo_modulelist, BTINFO_MODULELIST,
btinfo_modulelist_size);
}
}
userconf_init();
if (btinfo_userconfcommands != NULL)
BI_ADD(btinfo_userconfcommands, BTINFO_USERCONFCOMMANDS,
btinfo_userconfcommands_size);
#ifdef DEBUG
printf("Start @ 0x%lx [%ld=0x%lx-0x%lx]...\n", marks[MARK_ENTRY],
marks[MARK_NSYM], marks[MARK_SYM], marks[MARK_END]);
#endif
btinfo_symtab.nsym = marks[MARK_NSYM];
btinfo_symtab.ssym = marks[MARK_SYM];
btinfo_symtab.esym = marks[MARK_END];
BI_ADD(&btinfo_symtab, BTINFO_SYMTAB, sizeof(struct btinfo_symtab));
vbe_commit();
BI_ADD(&btinfo_framebuffer, BTINFO_FRAMEBUFFER,
sizeof(struct btinfo_framebuffer));
if (callback != NULL)
(*callback)();
entry = marks[MARK_ENTRY];
#ifdef EFIBOOT
for (i = 0; i < bootinfo->nentries; i++) {
struct btinfo_common *bi = (void *)(u_long)bootinfo->entry[i];
char *p = alloc(bi->len);
memcpy(p, bi, bi->len);
bootinfo->entry[i] = vtophys(p);
}
efi_kernel_start = marks[MARK_START];
efi_kernel_size = image_end - (efi_loadaddr + efi_kernel_start);
switch (efi_reloc_type) {
case RELOC_NONE:
entry += (efi_load_start - efi_kernel_start);
efi_kernel_start = efi_load_start;
break;
case RELOC_ADDR:
entry += (efi_kernel_reloc - efi_kernel_start);
efi_kernel_start = efi_kernel_reloc;
break;
case RELOC_DEFAULT:
default:
break;
}
#endif
startprog(entry, BOOT_NARGS, boot_argv,
x86_trunc_page(basemem * 1024));
panic("exec returned");
out:
BI_FREE();
bootinfo = NULL;
return -1;
}
static void
extract_device(const char *path, char *buf, size_t buflen)
{
size_t i;
if (strchr(path, ':') != NULL) {
for (i = 0; i < buflen - 2 && path[i] != ':'; i++)
buf[i] = path[i];
buf[i++] = ':';
buf[i] = '\0';
} else
buf[0] = '\0';
}
static const char *
module_path(boot_module_t *bm, const char *kdev, const char *base_path)
{
static char buf[256];
char name_buf[256], dev_buf[64];
const char *name, *name2, *p;
name = bm->bm_path;
for (name2 = name; *name2; ++name2) {
if (*name2 == ' ' || *name2 == '\t') {
strlcpy(name_buf, name, sizeof(name_buf));
if ((uintptr_t)name2 - (uintptr_t)name < sizeof(name_buf))
name_buf[name2 - name] = '\0';
name = name_buf;
break;
}
}
if ((p = strchr(name, ':')) != NULL) {
if (p[1] == '/')
snprintf(buf, sizeof(buf), "%s", name);
else {
p++;
extract_device(name, dev_buf, sizeof(dev_buf));
snprintf(buf, sizeof(buf), "%s%s/%s/%s.kmod",
dev_buf, base_path, p, p);
}
} else {
if (name[0] == '/')
snprintf(buf, sizeof(buf), "%s%s", kdev, name);
else
snprintf(buf, sizeof(buf), "%s%s/%s/%s.kmod",
kdev, base_path, name, name);
}
return buf;
}
static int
module_open(boot_module_t *bm, int mode, const char *kdev,
const char *base_path, bool doload)
{
int fd;
const char *path;
path = module_path(bm, kdev, base_path);
fd = open(path, mode);
if (fd != -1) {
if ((howto & AB_SILENT) == 0 && doload)
printf("Loading %s ", path);
} else {
fd = open(bm->bm_path, mode);
if (fd != -1 && (howto & AB_SILENT) == 0 && doload)
printf("Loading %s ", bm->bm_path);
}
if (!doload && fd == -1) {
printf("WARNING: couldn't open %s", bm->bm_path);
if (strcmp(bm->bm_path, path) != 0)
printf(" (%s)", path);
printf("\n");
}
return fd;
}
static void
module_base_path(char *buf, size_t bufsize, const char *kernel_path)
{
#ifdef KERNEL_DIR
char *ptr = strrchr(kernel_path, '/');
if (ptr) *ptr = '\0';
snprintf(buf, bufsize, "%s/modules", kernel_path);
if (ptr) *ptr = '/';
#else
const char *machine;
switch (netbsd_elf_class) {
case ELFCLASS32:
machine = "i386";
break;
case ELFCLASS64:
machine = "amd64";
break;
default:
machine = "generic";
break;
}
if (netbsd_version / 1000000 % 100 == 99) {
snprintf(buf, bufsize,
"/stand/%s/%d.%d.%d/modules", machine,
netbsd_version / 100000000,
netbsd_version / 1000000 % 100,
netbsd_version / 100 % 10000);
} else if (netbsd_version != 0) {
snprintf(buf, bufsize,
"/stand/%s/%d.%d/modules", machine,
netbsd_version / 100000000,
netbsd_version / 1000000 % 100);
}
#endif
}
static void
module_init(const char *kernel_path)
{
struct bi_modulelist_entry *bi;
struct stat st;
char kdev[64];
char *buf;
boot_module_t *bm;
ssize_t len;
off_t off;
int err, fd, nfail = 0;
extract_device(kernel_path, kdev, sizeof(kdev));
module_base_path(module_base, sizeof(module_base), kernel_path);
len = sizeof(struct btinfo_modulelist);
for (bm = boot_modules; bm; bm = bm->bm_next) {
fd = module_open(bm, 0, kdev, module_base, false);
if (fd == -1) {
bm->bm_len = -1;
++nfail;
continue;
}
err = fstat(fd, &st);
if (err == -1 || st.st_size == -1) {
printf("WARNING: couldn't stat %s\n", bm->bm_path);
close(fd);
bm->bm_len = -1;
++nfail;
continue;
}
bm->bm_len = st.st_size;
close(fd);
len += sizeof(struct bi_modulelist_entry);
}
btinfo_modulelist = alloc(len);
if (btinfo_modulelist == NULL) {
printf("WARNING: couldn't allocate module list\n");
wait_sec(MODULE_WARNING_SEC);
return;
}
memset(btinfo_modulelist, 0, len);
btinfo_modulelist_size = len;
buf = (char *)btinfo_modulelist;
btinfo_modulelist->num = 0;
off = sizeof(struct btinfo_modulelist);
for (bm = boot_modules; bm; bm = bm->bm_next) {
if (bm->bm_len == -1)
continue;
fd = module_open(bm, 0, kdev, module_base, true);
if (fd == -1)
continue;
image_end = (image_end + PAGE_SIZE - 1) & ~(PAGE_SIZE - 1);
len = pread(fd, (void *)(uintptr_t)image_end, SSIZE_MAX);
if (len < bm->bm_len) {
if ((howto & AB_SILENT) != 0)
printf("Loading %s ", bm->bm_path);
printf(" FAILED\n");
} else {
btinfo_modulelist->num++;
bi = (struct bi_modulelist_entry *)(buf + off);
off += sizeof(struct bi_modulelist_entry);
strncpy(bi->path, bm->bm_path, sizeof(bi->path) - 1);
bi->base = image_end;
bi->len = len;
switch (bm->bm_type) {
case BM_TYPE_KMOD:
bi->type = BI_MODULE_ELF;
break;
case BM_TYPE_IMAGE:
bi->type = BI_MODULE_IMAGE;
break;
case BM_TYPE_FS:
bi->type = BI_MODULE_FS;
break;
case BM_TYPE_RND:
default:
bi->type = BI_MODULE_RND;
break;
}
if ((howto & AB_SILENT) == 0)
printf(" \n");
}
if (len > 0)
image_end += len;
close(fd);
}
btinfo_modulelist->endpa = image_end;
if (nfail > 0) {
printf("WARNING: %d module%s failed to load\n",
nfail, nfail == 1 ? "" : "s");
#if notyet
wait_sec(MODULE_WARNING_SEC);
#endif
}
}
static void
userconf_init(void)
{
size_t count, len;
userconf_command_t *uc;
char *buf;
off_t off;
count = 0;
for (uc = userconf_commands; uc != NULL; uc = uc->uc_next)
count++;
len = sizeof(*btinfo_userconfcommands) +
count * sizeof(struct bi_userconfcommand);
btinfo_userconfcommands = alloc(len);
if (btinfo_userconfcommands == NULL) {
printf("WARNING: couldn't allocate userconf commands list\n");
return;
}
memset(btinfo_userconfcommands, 0, len);
btinfo_userconfcommands_size = len;
buf = (char *)btinfo_userconfcommands;
off = sizeof(*btinfo_userconfcommands);
btinfo_userconfcommands->num = 0;
for (uc = userconf_commands; uc != NULL; uc = uc->uc_next) {
struct bi_userconfcommand *bi;
bi = (struct bi_userconfcommand *)(buf + off);
strncpy(bi->text, uc->uc_text, sizeof(bi->text) - 1);
off += sizeof(*bi);
btinfo_userconfcommands->num++;
}
}
int
exec_multiboot(const char *file, char *args)
{
physaddr_t loadaddr = 0;
u_long marks[MARK_MAX];
u_long extmem;
u_long basemem;
struct multiboot_package *mbp = NULL;
#ifndef NO_MULTIBOOT2
if ((mbp = probe_multiboot2(file)) != NULL)
goto is_multiboot;
#endif
if ((mbp = probe_multiboot1(file)) != NULL) {
#ifdef EFIBOOT
printf("EFI boot requires multiboot 2 kernel\n");
goto out;
#else
goto is_multiboot;
#endif
}
#ifndef NO_MULTIBOOT2
printf("%s is not a multiboot kernel\n", file);
#else
printf("%s is not a multiboot 1 kernel "
"(multiboot 2 support is not built in)\n", file);
#endif
goto out;
is_multiboot:
#ifdef EFIBOOT
loadaddr = efi_loadaddr;
#endif
if (common_load_kernel(file, &basemem, &extmem, loadaddr, 0, marks))
goto out;
if (boot_modules_enabled)
module_init(file);
mbp->mbp_args = args;
mbp->mbp_basemem = basemem;
mbp->mbp_extmem = extmem;
mbp->mbp_loadaddr = loadaddr;
mbp->mbp_marks = marks;
(void)mbp->mbp_exec(mbp);
out:
if (mbp != NULL)
mbp->mbp_cleanup(mbp);
return -1;
}
void
x86_progress(const char *fmt, ...)
{
va_list ap;
if ((howto & AB_SILENT) != 0)
return;
va_start(ap, fmt);
vprintf(fmt, ap);
va_end(ap);
}
