#include #include "file.h" #include "fs.h" #include "print.h" #include "interrupt.h" #include "kbd.h" #include "tss.h" #include "paging.h" #include "v86defs.h" #include "tests.h" #include "progs.h" #include "helper.h" #include "disk.h" typedef unsigned short word; char check_apic() { uint32_t eax, ebx, ecx, edx; asm("cpuid" : "=a"(eax), "=b"(ebx), "=c"(ecx), "=d"(edx) : "a"(1)); return (edx & (1 << 9)) != 0; } char check_sse() { uint32_t eax, ebx, ecx, edx; asm("cpuid" : "=a"(eax), "=b"(ebx), "=c"(ecx), "=d"(edx) : "a"(1)); return (edx & (1 << 25)) != 0; } char check_cmov() { uint32_t eax, ebx, ecx, edx; asm("cpuid" : "=a"(eax), "=b"(ebx), "=c"(ecx), "=d"(edx) : "a"(1)); return (edx & (1 << 15)) != 0; } void enable_sse() { asm volatile( "mov %%cr0, %%eax\n" "and $0xFFFB, %%ax\n" "or $0x2, %%ax\n" "mov %%eax, %%cr0\n" "mov %%cr4, %%eax\n" "or $0x600, %%ax\n" "mov %%eax, %%cr4\n" : : : "%eax" ); } uint32_t get_flags() { uint32_t flags; asm volatile("pushfd\npop %%eax":"=a"(flags)); return flags; } uint32_t get_cr0() { uint32_t reg; asm volatile("mov %%cr0, %%eax":"=a"(reg)); return reg; } uint32_t get_cr3() { uint32_t reg; asm volatile("mov %%cr3, %%eax":"=a"(reg)); return reg; } uint32_t get_cr4() { uint32_t reg; asm volatile("mov %%cr4, %%eax":"=a"(reg)); return reg; } extern char _edata, _v86code, _ev86code, _bstart, _bend; void setup_binary() { // Put V86 code in proper place based on linker char *s = &_edata; char *d = &_v86code; while (d < &_ev86code) *d++ = *s++; // Clear BSS area for (d = &_bstart; d < &_bend; d++) *d = 0; } uint16_t error_screen[80*50]; // 50-line VGA screen of error content extern uint16_t *ivt; uint16_t ivt_bkup[0x200]; uint8_t bios_bkup[0x40000]; void backup_ivtbios() { for (int i = 0; i < 0x200; i++) ivt_bkup[i] = ivt[i]; for (int i = 0; i < 0x40000; i++) bios_bkup[i] = ((uint8_t*)0xC0000)[i]; } // This should only be used during fault handling, as it is expensive void ensure_v86env() { for (int i = 0; i < 0x200; i++) ivt[i] = ivt_bkup[i]; for (int i = 0; i < 0x40000; i++) ((uint8_t*)0xC0000)[i] = bios_bkup[i]; char *s = &_edata; char *d = &_v86code; while (d < &_ev86code) *d++ = *s++; } uint32_t _ERRORCODE = 0; __attribute__((__no_caller_saved_registers__)) uint32_t check_error_code() { uint32_t v = _ERRORCODE; _ERRORCODE = 0; return v; } __attribute__((__no_caller_saved_registers__)) __attribute__((__noreturn__)) extern void return_prev_task(); __attribute__((__no_caller_saved_registers__)) __attribute__((__noreturn__)) void error_environment(uint32_t stack0, uint32_t stack1, uint32_t stack2, uint32_t stack3, uint32_t stack4, uint32_t stack5) { ensure_v86env(); setup_interrupts(); // just in case for (int i = 0; i < 80*50; i++) if (!(error_screen[i] & 0xFF00)) error_screen[i] = 0x0f00 | (error_screen[i] & 0x00FF); union V86Regs_t regs; FARPTR v86_entry = i386LinearToFp(v86TextMode); enter_v86(0x8000, 0xFF00, FP_SEG(v86_entry), FP_OFF(v86_entry), ®s); char str0[] = "Oh noes!!! System error! ;c STKDMP:"; printStr(str0, &error_screen[80]); uint16_t *tmp = &error_screen[80+sizeof(str0)-1]; tmp += printDword(stack0, tmp); tmp++; tmp += printDword(stack1, tmp); tmp++; tmp += printDword(stack2, tmp); tmp = &error_screen[80*2+sizeof(str0)-1]; tmp += printDword(stack3, tmp); tmp++; tmp += printDword(stack4, tmp); tmp++; tmp += printDword(stack5, tmp); printStr("Press E for a fun recovery :3", &error_screen[80*2]); printStr("Press R to return to previous task", &error_screen[80*3]); uint16_t *vga_text = ((uint16_t*)0xB8000); for (int i = 0; i < 80*50; i++) vga_text[i] = error_screen[i]; for(;;) { uint8_t key = get_scancode() & 0xff; if (key == KEY_E) { v86_entry = i386LinearToFp(v86TransFlag); enter_v86(0x8000, 0xFF00, FP_SEG(v86_entry), FP_OFF(v86_entry), ®s); } if (key == KEY_R) break; } // reset error screen for (int i = 0; i < (80*50)/2; i++) ((uint32_t*)error_screen)[i] = 0x0f000f00; _ERRORCODE = -1; return_prev_task(); } uint32_t GetFreeStack() { uint32_t stack; asm volatile("mov %%esp,%%eax":"=a"(stack)); stack = ((stack - 0x2000) / 0x1000) * 0x1000; return stack; } /* Real Mode Accessible (First MB) 00000 - 00400 IVT (1kB) 00400 - 01000 Unused (3kB) 01000 - 04000 Free (12kB) 04000 - 07C00 V86 Code (15kB) 07C00 - 08000 Boot & V86 Code (512B) 08000 - 20000 Free (96kB) 20000 - 30000 Disk Buffer (64kB) 30000 - 40000 V86 Data (64kB) 40000 - 80000 Free (256kB) 80000 - 90000 V86 Stack (64kB) 90000 - A0000 Free (64kB) A0000 - C0000 VGA (128kB) C0000 - FFFFF BIOS Area (256kB) Protected Only (1MB+) 100000 - 200000 Kernel Code (1mB) 200000 - 200080 TSS (128B) 200080 - 202080 TSS IOMAP (8kB) 202080 - 208000 Free (~24kB) 208000 - 240000 Kernel File Stack (224kB) 240000 - 280000 Active Filesystems (128kB) 280000 - 300000 Disk Cache (512kB) 300000 - 310000 Task Stack (64kB) 310000 - 320000 Interrupt Stack (64kB) 320000 - 400000 Free (896kB) 400000 - 700000 Free (3mB) 700000 - 800000 Kernel Stack (1mB) 800000 - F00000 Usermode Code (7mB) F00000 - 1000000 Usermode Stack (1mB) */ void DrawScreen(uint16_t *vga) { uint16_t *vga_text = vga; // clear screen for (int i = 0; i < 80*25; i++) vga_text[i] = 0x1f00; // draw border for (int c = 1; c < 79; c++) { vga_text[c] = 0x1fc4; // top line vga_text[(2*80)+c] = 0x1fc4; // 3rd line vga_text[(24*80)+c] = 0x1fc4; // bottom line } for (int l = 1; l < 24; l++) { vga_text[80*l] = 0x1fb3; vga_text[(80*l)+79] = 0x1fb3; } vga_text[0] = 0x1fda; vga_text[79] = 0x1fbf; vga_text[2*80] = 0x1fc3; vga_text[2*80+79] = 0x1fb4; vga_text[24*80] = 0x1fc0; vga_text[24*80+79] = 0x1fd9; // name vga_text[80+34] = 0x1f00 | '-'; vga_text[80+35] = 0x1f00 | ' '; vga_text[80+36] = 0x1f00 | 'L'; vga_text[80+37] = 0x1f00 | 'u'; vga_text[80+38] = 0x1f00 | 'c'; vga_text[80+39] = 0x1f00 | 'i'; vga_text[80+40] = 0x1f00 | 'a'; vga_text[80+41] = 0x1f00 | 'O'; vga_text[80+42] = 0x1f00 | 'S'; vga_text[80+43] = 0x1f00 | ' '; vga_text[80+44] = 0x1f00 | '-'; } void SetPalette() { union V86Regs_t regs; regs.w.bx = 1; // index regs.h.dh = 0x3D; regs.h.ch = 0x2A; regs.h.cl = 0x2E; regs.w.ax = 0x1010; // set DAC color register V8086Int(0x10, ®s); } // NOTE Does not set text mode void RestoreVGA() { SetVideo25Lines(); SetCursorDisabled(); SetPalette(); } int32_t fileCount, fileOffset; extern char _USERMODE; dirent *const DirEntries = (dirent*)0x400000; #define MAXDISPFILES 16 void PrintFileList(uint16_t *vga) { uint16_t *vga_text = &((uint16_t *)vga)[80*6+3]; for (int i = 0; (i + fileOffset) < fileCount && i < MAXDISPFILES; i++) { dirent *de = &DirEntries[i + fileOffset]; de->name[de->namelen < 20 ? de->namelen : 20] = 0; vga_text += printStr(de->name, vga_text); vga_text += printStr(" ", vga_text); vga_text += printDec(de->size, vga_text); *(uint8_t*)vga_text++ = 'B'; vga_text = nextLine(vga_text, vga) + 3; } } char IsDir(dirent *de) { return de->type == FT_DIR; } void ScancodeTest() { uint16_t *vga = (uint16_t*)0xb8000; uint16_t scancode; while(((scancode = get_scancode()) & 0xff) != KEY_F1) { vga += printWord(scancode, vga); vga++; if ((uintptr_t)vga >= 0xb8000+80*25*2) vga = (uint16_t*)0xb8000; } } extern void create_child(uint32_t esp, uint32_t eip, uint32_t argc, ...); uint16_t FileSelectScreen[80*25]; char ValidFilesystems[256]; void FileSelect() { ActiveFilesystemBitmap(ValidFilesystems); uint8_t currentFsId = 0; char current_path[80]; uintptr_t current_path_end; for (int i = 0; i < sizeof(current_path); i++) current_path[i] = 0; current_path[0] = '0'; current_path[1] = '/'; current_path_end = 2; fileCount = 5; uint16_t *vga_text = (uint16_t *)FileSelectScreen; int32_t fileHovered = 0; fileOffset = 0; for (char reload = 1;;) { DrawScreen(vga_text); // Info line (4) { uint16_t *vga = &vga_text[80*4 + 79 - 24]; printStr("X to view in hex", vga); vga += 80; printStr("T to view as text", vga); vga += 80; printStr("P to load as program", vga); vga += 80; printStr("O to open directory", vga); vga += 80; printStr("S to switch volume", vga); vga += 80; printStr("F4 to run tests", vga); } printStr(current_path, &vga_text[80*4 + 2]); for (int i = 2; i < 15; i++) *(uint8_t*)&vga_text[80*5 + i] = '-'; DIR dir; if (reload) { current_path[current_path_end] = 0; dir_open(&dir, current_path); GetFileList(&dir, DirEntries, &fileCount, INT32_MAX); reload = 0; } if (fileHovered >= fileCount) { fileOffset = fileCount - MAXDISPFILES; fileHovered = fileCount - 1; } if ((fileHovered - fileOffset) >= MAXDISPFILES) fileOffset = fileHovered - MAXDISPFILES + 1; else if ((fileHovered - fileOffset) < 0) fileOffset = fileHovered; PrintFileList(vga_text); for (int i = 6; i < 24; i++) { *(uint8_t*)&vga_text[80*i+2] = ' '; } *(uint8_t*)&vga_text[80*(6+(fileHovered-fileOffset))+2] = '>'; // Copy to real VGA for (int i = 0; i < 80*25; i++) ((uint16_t*)0xb8000)[i] = vga_text[i]; uint16_t key = get_scancode(); uint8_t path[13]; switch (key & 0xff) { // scancode component case KEY_DOWN: // down fileHovered++; if (fileHovered >= fileCount) { fileHovered = 0; fileOffset = 0; } break; case KEY_UP: // up fileHovered--; if (fileHovered < 0) fileHovered = fileCount - 1; break; case KEY_F4: create_child(GetFreeStack(), (uintptr_t)RunTests, 0); // Set Text mode, tests might return in gfx { union V86Regs_t regs; regs.w.ax = 3; V8086Int(0x10, ®s); } RestoreVGA(); reload = 1; break; case KEY_P: if (IsDir(&DirEntries[fileHovered])) break; for (int i = 0; i < DirEntries[fileHovered].namelen; i++) current_path[current_path_end + i] = DirEntries[fileHovered].name[i]; current_path[current_path_end + DirEntries[fileHovered].namelen] = 0; create_child(GetFreeStack(), (uintptr_t)ProgramLoadTest, 1, current_path); current_path[current_path_end] = 0; RestoreVGA(); reload = 1; break; case KEY_X: if (IsDir(&DirEntries[fileHovered])) break; for (int i = 0; i < DirEntries[fileHovered].namelen; i++) current_path[current_path_end + i] = DirEntries[fileHovered].name[i]; current_path[current_path_end + DirEntries[fileHovered].namelen] = 0; { // Give userspace mem (8MB) uintptr_t process_page0 = add_pages(4|2|1); uintptr_t process_page1 = add_pages(4|2|1); asm("xchg %bx,%bx"); // Page blocks are 4MB move_pages(process_page0 >> 22, (uintptr_t)&_USERMODE >> 22); move_pages(process_page1 >> 22, ((uintptr_t)&_USERMODE >> 22) + 1); asm("xchg %bx,%bx"); create_child(GetFreeStack(), (uintptr_t)HexEditor, 1, current_path); // Free userspace mem free_pages(process_page0 >> 22); free_pages(process_page1 >> 22); } current_path[current_path_end] = 0; RestoreVGA(); reload = 1; break; case KEY_T: if (IsDir(&DirEntries[fileHovered])) break; //TextViewTest(path, &vi); for (int i = 0; i < DirEntries[fileHovered].namelen; i++) current_path[current_path_end + i] = DirEntries[fileHovered].name[i]; current_path[current_path_end + DirEntries[fileHovered].namelen] = 0; { // Give userspace mem (8MB) uintptr_t process_page0 = add_pages(4|2|1); uintptr_t process_page1 = add_pages(4|2|1); // Page blocks are 4MB move_pages(process_page0 >> 22, (uintptr_t)&_USERMODE >> 22); move_pages(process_page1 >> 22, ((uintptr_t)&_USERMODE >> 22) + 1); create_child(GetFreeStack(), (uintptr_t)TextViewTest, 1, current_path); // Free userspace mem free_pages(process_page0 >> 22); free_pages(process_page1 >> 22); } current_path[current_path_end] = 0; RestoreVGA(); reload = 1; break; case KEY_O: case 0x9C: // enter release if (IsDir(&DirEntries[fileHovered])) { uint8_t tmp_path[80]; { int i; for (i = 0; i < DirEntries[fileHovered].namelen && i < sizeof(tmp_path)-1; i++) tmp_path[i] = DirEntries[fileHovered].name[i]; tmp_path[i] = 0; } if ((*(uint32_t*)tmp_path & 0xffff) == ('.' | 0x0000)) { // Current dir, do nothing break; } else if ((*(uint32_t*)tmp_path & 0xffffff) == ('.' | '.' << 8 | 0x000000)) { // Up one directory for (current_path_end--;current_path_end >= 1 && current_path[current_path_end - 1] != '/'; current_path_end--); current_path[current_path_end] = 0; reload = 1; fileHovered = 0; fileOffset = 0; break; } for (int i = 0; (i + current_path_end) < sizeof(current_path); i++) current_path[current_path_end + i] = tmp_path[i]; for (;current_path_end < sizeof(current_path) - 2 && current_path[current_path_end]; current_path_end++); current_path[current_path_end] = '/'; current_path_end++; current_path[current_path_end] = 0; reload = 1; fileHovered = 0; fileOffset = 0; } break; case KEY_S: // Next filesystem TODO Support over 077 I'm so lazy right now for (currentFsId = (currentFsId + 1) % 64; !ValidFilesystems[currentFsId]; currentFsId = (currentFsId + 1) % 64); if (currentFsId < 8) { current_path[0] = '0' + currentFsId; current_path[1] = '/'; current_path[2] = 0; current_path_end = 2; } else { current_path[0] = '0' + (currentFsId >> 3); current_path[1] = '0' + (currentFsId & 7); current_path[2] = '/'; current_path[3] = 0; current_path_end = 3; } reload = 1; fileHovered = 0; fileOffset = 0; break; case KEY_F6: ScancodeTest(); reload = 1; break; default: break; } } } // TODO Move this somewhere else, maybe configurable & with timer handler void PITPgrm() { const uint32_t base_clk = 3579545; const uint16_t reload_v = 1194; // (base_clk / 3) / reload_v = ~999.31463Hz // 32.32 Fixed point const uint64_t irq_ms = (uint64_t)reload_v * 3000LL * ((1LL << 42) / (uint64_t)base_clk / (1LL << 10)); // irq_ms = 4294818000 (0xfffdb8d0) // We don't actually use this here, but in handler.nasm // for now, eventually this should be in C asm volatile( "cli\n" "movb $0b00110100, %%al\n" "outb %%al, $0x43\n" :::"%eax" ); uint32_t clobber_eax; asm volatile( "outb %%al, $0x40\n" "movb %%ah, %%al\n" "outb %%al, $0x40\n" "sti\n" :"=a"(clobber_eax):"a"(reload_v) ); } int MakeSystemVolume(uint8_t sysPartition); void MakeMBRPartitions(); void SystemRun(uint8_t sysPartition) { uint16_t *vga_text = (word *)0xb8000; RestoreVGA(); DrawScreen((uint16_t*)0xb8000); InitDisk(); PITPgrm(); // Check for FAT partition { while (1) { create_child(GetFreeStack(), (uintptr_t)MakeSystemVolume, 1, sysPartition); if (!check_error_code()) break; vga_text = &((word*)0xb8000)[80*4 + 2]; vga_text += printStr("Error loading file select. Ensure the disk has a valid MBR and FAT partition.", vga_text); vga_text = &((word*)0xb8000)[80*5 + 2]; vga_text += printStr("Press R to retry.", vga_text); for (;(get_scancode() & 0xff) != KEY_R;); } create_child(GetFreeStack(), (uintptr_t)MakeMBRPartitions, 0); } for (;;) { create_child(GetFreeStack(), (uintptr_t)FileSelect, 0); // should never return, so if it does, // we have an error { union V86Regs_t regs; FARPTR v86_entry = i386LinearToFp(v86TextMode); enter_v86(0x8000, 0xFF00, FP_SEG(v86_entry), FP_OFF(v86_entry), ®s); } RestoreVGA(); DrawScreen((uint16_t*)0xb8000); vga_text = &((word*)0xb8000)[80*4 + 2]; vga_text += printStr("Error loading file select. Ensure the disk has a valid MBR and FAT partition.", vga_text); vga_text = &((word*)0xb8000)[80*5 + 2]; vga_text += printStr("Press R to retry.", vga_text); for (;(get_scancode() & 0xff) != KEY_R;); } } __attribute__((__noreturn__)) extern void triple_fault(); uint32_t kernel_check = 0x12345678; void start() { word *vga_text = (word *)0xb8000; char h[] = "LuciaOS"; for (int i = 0; i < sizeof(h); i++) *(char *)&vga_text[i] = h[i]; vga_text = &vga_text[80]; // DL *should* be preserved uint16_t boot_dx; asm volatile("nop":"=d"(boot_dx)); if (!check_cmov()) { char cmov_err[] = "NO CMOV"; for (int i = 0; i < sizeof(cmov_err); i++) *(char *)&vga_text[i] = cmov_err[i]; for (;;) asm volatile("hlt"); } vga_text += printWord(boot_dx, vga_text); vga_text++; uint32_t o; asm("mov %%esp, %%eax" : "=a"(o) : :); vga_text += printDword(o, vga_text); vga_text++; asm("mov %%ebp, %%eax" : "=a"(o) : :); vga_text += printDword(o, vga_text); vga_text++; //char c[] = "APIC support: "; //char apic; //printByte(apic = check_apic(), (short*)&vga_text[80*3 + sizeof(c) - 1]); //for (int i = 0; i < sizeof(c); i++) // *(char *)&vga_text[i+(80*3)] = c[i]; //if (!apic) return; //char sse_str[] = "SSE support: "; //vga_text += printStr("SSE: ", vga_text); //char sse = check_sse(); //if (!sse) { // *vga_text = 'N'; //} //vga_text += printStr("Y ", vga_text); //enable_sse(); //print_flags(); vga_text += printStr("CR0:", vga_text); vga_text += printDword(get_cr0(), vga_text); vga_text++; //print_cr3(); //print_cr4(); // Setup system setup_binary(); setup_interrupts(); setup_tss(); init_paging(); backup_ivtbios(); // DL contained disk number, DH contained active partition uint8_t SystemPartition = boot_dx >> 8; create_child(GetFreeStack(), (uintptr_t)SystemRun, 1, SystemPartition); // If this returns, something is *very* wrong, reboot the system // TODO Maybe try to recover? // Triple fault triple_fault(); }