remove build.rs

Download the needed resources before running the tests.
Do not use include_bytes! macro because it makes cargo check fail when
the bzimage path does not work.

Signed-off-by: Andreea Florescu <fandree@amazon.com>

Author: andreeaflorescu

benches/x86_64/mod.rs

@@ -10,6 +10,9 @@
 extern crate linux_loader;
 extern crate vm_memory;
 
+use std::fs::File;
+use std::io::{Cursor, Read};
+
 use linux_loader::configurator::pvh::PvhBootConfigurator;
 use linux_loader::configurator::{BootConfigurator, BootParams};
 #[cfg(feature = "bzimage")]
@@ -19,8 +22,6 @@ use linux_loader::loader::elf::Elf;
 use linux_loader::loader::KernelLoader;
 use vm_memory::{Address, GuestAddress, GuestMemoryMmap};
 
-use std::io::Cursor;
-
 use criterion::{black_box, Criterion};
 
 const MEM_SIZE: usize = 0x100_0000;
@@ -67,13 +68,29 @@ fn build_pvh_boot_params() -> BootParams {
     boot_params
 }
 
+#[cfg(feature = "bzimage")]
+fn download_resources() {
+    use std::process::Command;
+
+    let command = "./.buildkite/download_resources.sh";
+    let status = Command::new(command).status().unwrap();
+    if !status.success() {
+        panic!("Cannot run build script");
+    }
+}
+
 #[cfg(feature = "bzimage")]
 fn create_bzimage() -> Vec<u8> {
-    include_bytes!(concat!(
+    download_resources();
+    let mut v = Vec::new();
+    let path = concat!(
         env!("CARGO_MANIFEST_DIR"),
         "/src/loader/x86_64/bzimage/bzimage"
-    ))
-    .to_vec()
+    );
+    let mut f = File::open(path).unwrap();
+    f.read_to_end(&mut v).unwrap();
+
+    v
 }
 
 pub fn criterion_benchmark(c: &mut Criterion) {

build.rs

@@ -173,7 +173,9 @@ impl KernelLoader for BzImage {
 mod tests {
     use super::*;
 
+    use std::fs::File;
     use std::io::Cursor;
+    use std::process::Command;
     use vm_memory::{Address, GuestAddress, GuestMemoryMmap};
 
     const MEM_SIZE: u64 = 0x100_0000;
@@ -182,9 +184,24 @@ mod tests {
         GuestMemoryMmap::from_ranges(&[(GuestAddress(0x0), (MEM_SIZE as usize))]).unwrap()
     }
 
+    fn download_resources() {
+        let command = "./.buildkite/download_resources.sh";
+        let status = Command::new(command).status().unwrap();
+        if !status.success() {
+            panic!("Cannot run build script");
+        }
+    }
+
     fn make_bzimage() -> Vec<u8> {
+        download_resources();
         let mut v = Vec::new();
-        v.extend_from_slice(include_bytes!("bzimage"));
+        let path = concat!(
+            env!("CARGO_MANIFEST_DIR"),
+            "/src/loader/x86_64/bzimage/bzimage"
+        );
+        let mut f = File::open(path).unwrap();
+        f.read_to_end(&mut v).unwrap();
+
         v
     }
 

src/loader/x86_64/bzimage/mod.rs

@@ -10,7 +10,6 @@
 //! Bindgen autogenerated structs for `x86_64` boot parameters.
 
 #![cfg(any(target_arch = "x86", target_arch = "x86_64"))]
-
 #![cfg_attr(feature = "cargo-clippy", allow(clippy::all))]
 #![allow(dead_code)]
 #![allow(missing_docs)]