From 0a58e9e806853392c2c6f9bc5cca4c8cab3f280a Mon Sep 17 00:00:00 2001
From: bjorn3 <17426603+bjorn3@users.noreply.github.com>
Date: Sun, 7 Sep 2025 12:03:37 +0200
Subject: [PATCH] Allow cfg(dtb) for check-cfg

---
 build.rs | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/build.rs b/build.rs
index 8720635b4b..8d4c6b400c 100644
--- a/build.rs
+++ b/build.rs
@@ -48,7 +48,8 @@ fn parse_kconfig(arch: &str) -> Option<()> {
 }
 
 fn main() {
-    println!("cargo:rustc-env=TARGET={}", env::var("TARGET").unwrap());
+    println!("cargo::rustc-env=TARGET={}", env::var("TARGET").unwrap());
+    println!("cargo::rustc-check-cfg=cfg(dtb)");
 
     let out_dir = env::var("OUT_DIR").unwrap();
     let cfg = Cfg::of(env::var("TARGET").unwrap().as_str()).unwrap();
@@ -56,10 +57,10 @@ fn main() {
 
     match arch_str {
         "aarch64" => {
-            println!("cargo:rustc-cfg=dtb");
+            println!("cargo::rustc-cfg=dtb");
         }
         "x86" => {
-            println!("cargo:rerun-if-changed=src/asm/x86/trampoline.asm");
+            println!("cargo::rerun-if-changed=src/asm/x86/trampoline.asm");
 
             let status = Command::new("nasm")
                 .arg("-f")
@@ -74,7 +75,7 @@ fn main() {
             }
         }
         "x86_64" => {
-            println!("cargo:rerun-if-changed=src/asm/x86_64/trampoline.asm");
+            println!("cargo::rerun-if-changed=src/asm/x86_64/trampoline.asm");
 
             let status = Command::new("nasm")
                 .arg("-f")
@@ -89,7 +90,7 @@ fn main() {
             }
         }
         "riscv64" => {
-            println!("cargo:rustc-cfg=dtb");
+            println!("cargo::rustc-cfg=dtb");
         }
         _ => (),
     }