refactor: move linker script args to build.rs

Author: mkroening

build.rs

@@ -1,3 +1,24 @@
+use std::env;
+
 fn main() {
 	built::write_built_file().expect("Failed to acquire build-time information");
+
+	set_linker_script();
+}
+
+fn set_linker_script() {
+	let cfg_target_arch = env::var("CARGO_CFG_TARGET_ARCH").unwrap();
+	let cfg_feature = env::var("CARGO_CFG_FEATURE").unwrap();
+
+	let linker_script = match cfg_target_arch.as_str() {
+		"aarch64" if cfg_feature.contains("elf") => "link.ld",
+		"riscv64" if cfg_feature.contains("sbi") => "link.ld",
+		"x86_64" if cfg_feature.contains("linux") => "platform/linux/link.ld",
+		"x86_64" if cfg_feature.contains("multiboot") => "platform/multiboot/link.ld",
+		_ => return,
+	};
+
+	let linker_script = format_args!("src/arch/{cfg_target_arch}/{linker_script}");
+	println!("cargo:rerun-if-changed={linker_script}");
+	println!("cargo:rustc-link-arg=-T{linker_script}");
 }

xtask/src/target.rs

@@ -74,17 +74,8 @@ impl Target {
 
 	pub fn rustflags(&self) -> &'static [&'static str] {
 		match self {
-			Self::X86_64Linux => &[
-				"-Clink-arg=-Tsrc/arch/x86_64/platform/linux/link.ld",
-				"-Crelocation-model=static",
-			],
-			Self::X86_64Multiboot => &[
-				"-Clink-arg=-Tsrc/arch/x86_64/platform/multiboot/link.ld",
-				"-Crelocation-model=static",
-			],
-			Self::X86_64Uefi => &[],
-			Self::Aarch64Elf | Self::Aarch64BeElf => &["-Clink-arg=-Tsrc/arch/aarch64/link.ld"],
-			Self::Riscv64Sbi => &["-Clink-arg=-Tsrc/arch/riscv64/link.ld"],
+			Self::X86_64Linux | Self::X86_64Multiboot => &["-Crelocation-model=static"],
+			_ => &[],
 		}
 	}