Merge pull request rust-osdev#113 from rust-osdev/make-bootsector-a-staticlib

Make bootsector a staticlib

Author: katietheqt-old

.cargo/config

@@ -1,3 +1,2 @@
-[build]
-target = "x86_64-real_mode.json"
-rustflags = ["-C", "link-arg=-Tlinker.ld"]
+[alias]
+xbuild = "build -Zbuild-std=core"

.github/workflows/build.yml

@@ -41,20 +41,15 @@ jobs:
 
     - name: "Install Rustup Components"
       run: rustup component add rust-src llvm-tools-preview
-    - name: "Install cargo-xbuild"
-      run: cargo install cargo-xbuild --debug
     - name: "Install cargo-binutils"
       run: cargo install cargo-binutils --version 0.1.7 --debug
 
-    - run: cargo xbuild
-      working-directory: test-kernel
-      name: 'Build Test Kernel'
-
     - name: 'Build Bootloader'
       run: cargo xbuild --release
+      working-directory: src/stage_2
 
     - name: 'Convert Bootloader ELF to Binary'
-      run: cargo objcopy -- -I elf64-i386 -O binary --binary-architecture=i386:x86-64 target/x86_64-real_mode/release/bootsector target/x86_64-real_mode/release/bootsector.bin
+      run: cargo objcopy -- -I elf64-i386 -O binary --binary-architecture=i386:x86-64 target/x86_64-real_mode/release/stage_2 target/x86_64-real_mode/release/image.bin
 
     #  # install QEMU
     # - name: Install QEMU (Linux)
@@ -108,8 +103,6 @@ jobs:
       if: runner.os == 'macOS'
     - name: "Install Rustup Components"
       run: rustup component add rust-src llvm-tools-preview
-    - name: "Install cargo-xbuild"
-      run: cargo install cargo-xbuild --debug
     - name: 'Build Example Kernel'
       run: cargo xbuild
       working-directory: example-kernel

Cargo.lock

@@ -4,10 +4,15 @@ members = [
   "src/v86",
   "src/bootsector",
   "src/stage_2",
-#  "example-kernel",
-#  "test-kernel",
+  "example-kernel",
+  "test-kernel",
 ]
 
 [profile.release]
 opt-level = "z"
-panic = "abort"
+panic = "abort"
+lto = true
+
+[profile.release.package.bootsector]
+opt-level = "s"
+codegen-units = 1

Cargo.toml

@@ -6,6 +6,9 @@ edition = "2018"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
+[lib]
+name = "bootsector"
+crate-type = ["staticlib"]
+
 [dependencies]
 shared = { path = "../shared" }
-stage_2 = { path = "../stage_2" }

src/bootsector/Cargo.toml

@@ -1,16 +1,16 @@
-#[inline(never)]
-pub fn println(s: &[u8]) {
+#[no_mangle]
+pub fn real_mode_println(s: &[u8]) {
     print(s);
     print_char(b'\n');
 }
 
 pub fn print(s: &[u8]) {
-	let mut i = 0;
+    let mut i = 0;
 
-	while i < s.len() {
-		print_char(s[i]);
-		i += 1;
-	}
+    while i < s.len() {
+        print_char(s[i]);
+        i += 1;
+    }
 }
 
 #[inline(always)]
@@ -19,4 +19,4 @@ pub fn print_char(c: u8) {
     unsafe {
         llvm_asm!("int 0x10" :: "{ax}"(ax) :: "intel" );
     }
-}
+}

src/bootsector/src/console.rs

@@ -1,18 +1,18 @@
+use super::console::real_mode_println;
 use shared::utils;
-use super::console::println;
 
 #[no_mangle]
-pub extern "C" fn dap_load_failed() -> ! {
-    println(b"[!] DAP Load Failed");
+extern "C" fn dap_load_failed() -> ! {
+    real_mode_println(b"[!] DAP Load Failed");
     loop {
         utils::hlt()
     }
 }
 
 #[no_mangle]
-pub extern "C" fn no_int13h_extensions() -> ! {
-    println(b"[!] No int13h Extensions");
+extern "C" fn no_int13h_extensions() -> ! {
+    real_mode_println(b"[!] No int13h Extensions");
     loop {
         utils::hlt()
     }
-}
+}

src/bootsector/src/errors.rs

@@ -0,0 +1,54 @@
+#![feature(llvm_asm, global_asm)]
+#![no_std]
+#![allow(dead_code)]
+
+mod console;
+mod errors;
+
+use self::console::real_mode_println;
+use core::panic::PanicInfo;
+use shared::{dap, linker_symbol, utils};
+
+extern "C" {
+    fn second_stage();
+}
+global_asm!(include_str!("bootstrap.s"));
+
+#[no_mangle]
+extern "C" fn rust_start(disk_number: u16) -> ! {
+    real_mode_println(b"Stage 1");
+
+    check_int13h_extensions(disk_number);
+
+    let dap = dap::DiskAddressPacket::new(
+        linker_symbol!(_rest_of_bootloader_start) as u16,
+        (linker_symbol!(_rest_of_bootloader_start) - linker_symbol!(_bootloader_start)) as u64,
+        linker_symbol!(_rest_of_bootloader_end) - linker_symbol!(_rest_of_bootloader_start),
+    );
+
+    unsafe { dap.perform_load(disk_number) };
+
+    unsafe { second_stage() };
+
+    loop {
+        utils::hlt();
+    }
+}
+
+fn check_int13h_extensions(disk_number: u16) {
+    unsafe {
+        llvm_asm!("
+			int 0x13
+    		jc no_int13h_extensions
+        " :: "{ah}"(0x41), "{bx}"(0x55aa), "{dl}"(disk_number) :: "intel", "volatile");
+    }
+}
+
+#[panic_handler]
+fn panic(_info: &PanicInfo) -> ! {
+    real_mode_println(b"[Panic]");
+
+    loop {
+        utils::hlt()
+    }
+}

src/bootsector/src/lib.rs

@@ -36,4 +36,4 @@ impl Writer {
             llvm_asm!("int 0x10" :: "{ax}"(ax), "{bx}"(0) :: "intel", "volatile");
         }
     }
-}
+}