From 29d083234d837dbcb449bf2b9797257b9d7e324f Mon Sep 17 00:00:00 2001
From: Philipp Oppermann <dev@phil-opp.com>
Date: Wed, 8 Jan 2020 10:23:22 +0100
Subject: [PATCH] Move architecture checks from build script into lib.rs

The build script always gets compiled for the host platform, so it only checked that the _host_ was `x86_64`. With the check in lib.rs, we correctly verify that the target is `x86_64`.
---
 build.rs   | 11 +----------
 src/lib.rs |  9 +++++++++
 2 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/build.rs b/build.rs
index a83dc9a2..779ec2e2 100644
--- a/build.rs
+++ b/build.rs
@@ -1,14 +1,5 @@
 #[cfg(not(feature = "binary"))]
-fn main() {
-    #[cfg(target_arch = "x86")]
-    compile_error!(
-        "This crate currently does not support 32-bit protected mode. \
-         See https://github.com/rust-osdev/bootloader/issues/70 for more information."
-    );
-
-    #[cfg(not(any(target_arch = "x86_64", target_arch = "x86")))]
-    compile_error!("This crate only supports the x86_64 architecture.");
-}
+fn main() {}
 
 #[cfg(feature = "binary")]
 #[derive(Default)]
diff --git a/src/lib.rs b/src/lib.rs
index 00d3d069..90d8eebf 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -12,6 +12,15 @@ pub use crate::bootinfo::BootInfo;
 
 pub mod bootinfo;
 
+#[cfg(target_arch = "x86")]
+compile_error!(
+    "This crate currently does not support 32-bit protected mode. \
+         See https://github.com/rust-osdev/bootloader/issues/70 for more information."
+);
+
+#[cfg(not(any(target_arch = "x86_64", target_arch = "x86")))]
+compile_error!("This crate only supports the x86_64 architecture.");
+
 /// Defines the entry point function.
 ///
 /// The function must have the signature `fn(&'static BootInfo) -> !`.