fernlader2/src/main.c

455 lines
9.5 KiB
C

#include "std.h"
#include <bootboot.h>
#define COM1_BASE_PORT 0x3F8
static inline uint8_t
inb(uint16_t port)
{
uint8_t data;
__asm__ ("inb %%dx, %%al" : "=a"(data) : "d"(port));
return data;
}
static inline void
outb(uint16_t port, uint8_t data)
{
__asm__ ("outb %%al, %%dx" :: "a"(data), "d"(port));
}
extern void bios_write(const char *msg);
extern void *bios_getmap(void *buffer);
void
debug_write(const char *msg)
{
for (const char *c = msg; *c; c++) {
outb(0xE9, *c);
}
bios_write(msg);
}
extern BOOTBOOT bootboot;
#define E820_TYPE_FREE 0x1
#define E820_TYPE_RESERVED 0x2
#define E820_TYPE_RECLAIMABLE 0x3
#define E820_TYPE_NONVOLATILE 0x4
#define E820_TYPE_BADMEM 0x5
struct e820_entry {
uint64_t base;
uint64_t length;
uint32_t type;
uint32_t xattr;
};
struct e820_entry memmap[128];
struct vesa_info {
char signature[4];
uint16_t version;
uint16_t oem_name[2];
uint8_t capab[4];
uint16_t video_mode_offset;
uint16_t video_mode_segment;
uint16_t num_64k_blocks;
uint8_t reserved[492];
};
extern uint16_t bios_func(uint8_t inum, uint32_t reg[]);
void
blank_screen(uint16_t attr)
{
uint32_t reg[7];
reg[0] = (0x06 << 8);
reg[1] = 0;
reg[2] = 0xFFFF;
reg[3] = attr << 8;
reg[4] = 0;
reg[5] = 0;
reg[6] = 0;
bios_func(0x10, reg);
}
void
move_cursor(uint8_t row, uint8_t col)
{
uint32_t reg[7];
reg[0] = (0x02 << 8);
reg[1] = 0;
reg[2] = (row << 8) | col;
reg[3] = 0;
reg[4] = 0;
reg[5] = 0;
reg[6] = 0;
bios_func(0x10, reg);
}
void
display_string(uint16_t attr, const char *str)
{
uint32_t reg[7];
reg[0] = (0x13 << 8) | 0x01;
reg[1] = strlen(str);
reg[2] = 0xFFFF;
reg[3] = attr;
reg[4] = (uint32_t)str;
reg[5] = 0;
reg[6] = 0;
bios_func(0x10, reg);
}
void
debug_write_uint(unsigned v)
{
char buf[20], *p = buf + sizeof buf;
p--;
*p = 0;
do {
p--;
*p = (v % 10) + '0';
v /= 10;
} while (v);
debug_write(p);
}
#define PG_PRESENT 0x001
#define PG_WRITE 0x002
#define PG_USER 0x004
#define PG_SIZE 0x080
#define PG_GLOBAL 0x100
__attribute__((aligned(4096))) uint64_t pml4[512];
__attribute__((aligned(4096))) uint64_t pdp[512];
__attribute__((aligned(4096))) uint64_t pd[512];
uint64_t
pg_setup()
{
pml4[0] = (uint64_t)(uint32_t)pdp | PG_WRITE | PG_PRESENT;
pdp[0] = (uint64_t)(uint32_t)pd | PG_WRITE | PG_PRESENT;
for (unsigned i = 0; i < 512; i++) {
pd[i] = (i * (uint64_t)0x200000) | PG_WRITE | PG_PRESENT | PG_SIZE;
}
uint64_t cr3 = (uint64_t)(uint32_t)pml4;
return cr3;
}
struct PXENV {
uint8_t signature[6];
uint16_t version;
uint8_t length;
uint8_t checksum;
uint32_t rmentry;
uint32_t pmoffset;
};
struct exPXE {
uint8_t signature[4];
uint8_t length;
uint8_t checksum;
uint8_t revision;
uint8_t reserved;
uint32_t undiromid;
uint32_t bcromid;
uint32_t rmentry;
uint32_t pmentry;
};
struct bootph {
uint8_t opcode;
uint8_t hardware;
uint8_t hardlen;
uint8_t gatehops;
uint32_t ident;
uint16_t seconds;
uint16_t flags;
uint32_t cip;
uint32_t yip;
uint32_t sip;
uint32_t gip;
uint8_t caddr[16];
uint8_t sname[64];
uint8_t bootfile[128];
uint8_t vendor[64];
};
extern uint32_t pxe_entry;
extern uint32_t pxe_call(uint16_t func, uint16_t segment, uint16_t offset);
extern unsigned char pxe_cmd_buf[512];
extern struct PXENV *PXENV;
extern struct exPXE *exPXE;
#define PXENV_GET_CACHED 0x0071
#define PXENV_TFTP_OPEN 0x0020
#define PXENV_TFTP_READ 0x0022
typedef struct __attribute__((packed)) s_PXENV_GET_CACHED {
uint16_t status;
uint16_t packet_type;
uint16_t buffer_size;
uint16_t buffer_offset;
uint16_t buffer_seg;
uint16_t buffer_limit;
} t_PXENV_GET_CACHED;
typedef struct __attribute__((packed)) s_PXENV_TFTP_OPEN {
uint16_t status;
uint32_t server_ip;
uint32_t gateway_ip;
unsigned char filename[128];
uint16_t tftp_port;
uint16_t packet_size;
} t_PXENV_TFTP_OPEN;
typedef struct __attribute__((packed)) s_PXENV_TFTP_READ {
uint16_t status;
uint16_t packet_number;
uint16_t buffer_size;
uint16_t buffer_offset;
uint16_t buffer_seg;
} t_PXENV_TFTP_READ;
/**
* return type for fs drivers
*/
typedef struct {
uint8_t *ptr;
uint32_t size;
} file_t;
/**
* convert ascii octal number to binary number
*/
int octbin(unsigned char *str,int size)
{
int s=0;
unsigned char *c=str;
while(size-->0){
s*=8;
s+=*c-'0';
c++;
}
return s;
}
/**
* convert ascii hex number to binary number
*/
int hexbin(unsigned char *str, int size)
{
int v=0;
while(size-->0){
v <<= 4;
if(*str>='0' && *str<='9')
v += (int)((unsigned char)(*str)-'0');
else if(*str >= 'A' && *str <= 'F')
v += (int)((unsigned char)(*str)-'A'+10);
str++;
}
return v;
}
#define PRINTFS(fsname,kernel) do {\
debug_write(fsname);\
debug_write(" ");\
debug_write(kernel);\
debug_write("\r\n");\
} while (0)
#include "fs.h"
uint32_t pxe_sip;
uint32_t pxe_gip;
void
get_cached()
{
t_PXENV_GET_CACHED *t_cached = (void *)pxe_cmd_buf;
t_cached->status = 0;
t_cached->packet_type = 2;
t_cached->buffer_size = 0;
t_cached->buffer_offset = 0;
t_cached->buffer_seg = 0;
t_cached->buffer_limit = 0xFFFFu;
debug_write("Performing PXE call\r\n");
unsigned ret = pxe_call(PXENV_GET_CACHED, (uint16_t)(uintptr_t)pxe_cmd_buf, 0x0000);
debug_write("Still alive!\r\n");
debug_write("ret=");
debug_write_uint(ret);
debug_write("\r\nstatus=");
debug_write_uint(t_cached->status);
debug_write("\r\n");
debug_write("buffer_off=");
debug_write_uint(t_cached->buffer_offset);
debug_write("\r\nbuffer_seg=");
debug_write_uint(t_cached->buffer_seg);
struct bootph *bootph = (void *)(t_cached->buffer_offset + 16 * t_cached->buffer_seg);
pxe_sip = bootph->sip;
pxe_gip = bootph->gip;
debug_write("\r\nsip=");
debug_write_uint(pxe_sip);
debug_write("\r\ngip=");
debug_write_uint(pxe_gip);
debug_write("\r\n");
}
void
receive_file(void)
{
unsigned ret;
t_PXENV_TFTP_OPEN *t_open = (void *)pxe_cmd_buf;
t_open->status = 0;
t_open->server_ip = pxe_sip;
t_open->gateway_ip = pxe_gip;
memcpy(t_open->filename, "/initrd", 8);
t_open->tftp_port = 69 << 8;
t_open->packet_size = 512;
debug_write("Performing PXE OPEN call\r\n");
ret = pxe_call(PXENV_TFTP_OPEN, (uint16_t)(uintptr_t)pxe_cmd_buf, 0x0000);
debug_write("ret=");
debug_write_uint(ret);
debug_write(" status=");
debug_write_uint(t_open->status);
debug_write("\r\n");
uint16_t packet_size = t_open->packet_size;
#if 0
for (;;) {
t_PXENV_TFTP_READ *t_read = (void *)pxe_cmd_buf;
t_read->status = 0;
t_read->buffer_offset = ;
t_read->buffer_seg = ;
ret = pxe_call(PXENV_TFTP_READ, (uint16_t)(uintptr_t)pxe_cmd_buf, 0x0000);
debug_write("ret=");
debug_write_uint(ret);
debug_write(" status=");
debug_write_uint(t_read->status);
debug_write("\r\n");
if (t_read->buffer_size < packet_size) {
break;
}
}
#endif
}
void
main()
{
blank_screen(0x10);
move_cursor(0, 0);
//display_string(0x14, "Netboot via fernlader v2 ...\r\n");
debug_write("Going well ...\r\n");
if (memcmp(PXENV->signature, "PXENV+", 6) != 0) {
debug_write("missing PXENV+ signature\r\n");
}
if (PXENV->version >= 0x0201) {
debug_write("!PXE version\r\n");
if (memcmp(exPXE->signature, "!PXE", 4) != 0) {
debug_write("missing !PXE signature\r\n");
}
pxe_entry = exPXE->rmentry;
}
memcpy(bootboot.magic, BOOTBOOT_MAGIC, sizeof bootboot.magic);
bootboot.size = 128;
bootboot.protocol = PROTOCOL_MINIMAL | LOADER_BIOS;
bootboot.numcores = 1;
//display_string(0x0E, "init\r\n");
struct e820_entry *end = bios_getmap(memmap);
//display_string(0x0E, "karlos\r\n");
for (int i = 0; ; i++) {
if (&memmap[i] >= end) break;
char buf[2];
buf[0] = memmap[i].type < 10 ? memmap[i].type + '0' : '?';
buf[1] = 0;
debug_write(buf);
}
pg_setup();
get_cached();
receive_file();
#if 0
#if 1
t_PXENV_GET_CACHED *t_cached = (void *)pxe_cmd_buf;
t_cached->status = 0;
t_cached->packet_type = 2;
t_cached->buffer_size = 0;
t_cached->buffer_offset = 0;
t_cached->buffer_seg = 0;
t_cached->buffer_limit = 0xFFFFu;
debug_write("Performing PXE call\r\n");
unsigned ret = pxe_call(PXENV_GET_CACHED, (uint16_t)(uintptr_t)pxe_cmd_buf, 0x0000);
debug_write("Still alive!\r\n");
debug_write("ret=");
debug_write_uint(ret);
debug_write("\r\nstatus=");
debug_write_uint(t_cached->status);
debug_write("\r\n");
#endif
debug_write("buffer_off=");
debug_write_uint(t_cached->buffer_offset);
debug_write("\r\nbuffer_seg=");
debug_write_uint(t_cached->buffer_seg);
struct bootph *bootph = (void *)(t_cached->buffer_offset + 16 * t_cached->buffer_seg);
uint32_t sip = bootph->sip;
uint32_t gip = bootph->gip;
debug_write("\r\nsip=");
debug_write_uint(sip);
debug_write("\r\ngip=");
debug_write_uint(gip);
debug_write("\r\n");
#if 1
t_PXENV_TFTP_OPEN *t_open = (void *)pxe_cmd_buf;
t_open->status = 0;
t_open->server_ip = sip;
t_open->gateway_ip = gip;
memcpy(t_open->filename, "/initrd", 8);
t_open->tftp_port = 69 << 8;
t_open->packet_size = 512;
debug_write("Performing PXE OPEN call\r\n");
ret = pxe_call(PXENV_TFTP_OPEN, (uint16_t)(uintptr_t)pxe_cmd_buf, 0x0000);
debug_write("Still alive!\r\n");
debug_write("ret=");
debug_write_uint(ret);
debug_write("\r\nstatus=");
debug_write_uint(t_open->status);
debug_write("\r\n");
#endif
#if 0
uint32_t reg[7];
reg[0] = (0x0E << 8) + 'J';
reg[1] = 0;
reg[2] = 0;
reg[3] = 0;
reg[4] = 0;
reg[5] = 0;
reg[6] = 0;
bios_func(0x10, reg);
debug_write("what?\r\n");
#endif
#endif
__asm__ ("cli");
for (;;) {
//__asm__ ("sti");
//__asm__ ("hlt");
}
}