Add integration tests for injector

4 files changed, 90 insertions, 0 deletions

diff --git a/zluda_inject/build.rs b/zluda_inject/build.rs
new file mode 100644
index 0000000..c9146e0
--- /dev/null
+++ b/zluda_inject/build.rs
@@ -0,0 +1,68 @@
+use std::{

+    env::{self, VarError},

+    fs::{self, DirEntry},

+    io,

+    path::{self, PathBuf},

+    process::Command,

+};

+

+fn main() -> Result<(), VarError> {

+    println!("cargo:rerun-if-changed=build.rs");

+    if env::var("PROFILE")? != "debug" {

+        return Ok(());

+    }

+    let rustc_exe = env::var("RUSTC")?;

+    let out_dir = env::var("OUT_DIR")?;

+    let target = env::var("TARGET")?;

+    let is_msvc = env::var("CARGO_CFG_TARGET_ENV")? == "msvc";

+    let opt_level = env::var("OPT_LEVEL")?;

+    let debug = str::parse::<bool>(env::var("DEBUG")?.as_str()).unwrap();

+    let mut helpers_dir = PathBuf::from(env::var("CARGO_MANIFEST_DIR")?);

+    helpers_dir.push("tests");

+    helpers_dir.push("helpers");

+    let helpers_dir_as_string = helpers_dir.to_string_lossy();

+    for rust_file in fs::read_dir(&helpers_dir).unwrap().filter_map(rust_file) {

+        let full_file_path = format!(

+            "{}{}{}",

+            helpers_dir_as_string,

+            path::MAIN_SEPARATOR,

+            rust_file

+        );

+        println!("cargo:rerun-if-changed={}", full_file_path);

+        let mut rustc_cmd = Command::new(&*rustc_exe);

+        if debug {

+            rustc_cmd.arg("-g");

+        }

+        rustc_cmd.arg(format!("-Lnative={}", helpers_dir_as_string));

+        if !is_msvc {

+            // HACK ALERT

+            // I have no idea why the extra library below have to be linked

+            rustc_cmd.arg(r"-lucrt");

+        }

+        rustc_cmd

+            .arg("-ldylib=nvcuda")

+            .arg("-C")

+            .arg(format!("opt-level={}", opt_level))

+            .arg("--out-dir")

+            .arg(format!("{}", out_dir))

+            .arg("--target")

+            .arg(format!("{}", target))

+            .arg(full_file_path);

+        assert!(rustc_cmd.status().unwrap().success());

+    }

+    println!("cargo:rustc-env=HELPERS_OUT_DIR={}", &out_dir);

+    Ok(())

+}

+

+fn rust_file(entry: io::Result<DirEntry>) -> Option<String> {

+    entry.ok().and_then(|e| {

+        let os_file_name = e.file_name();

+        let file_name = os_file_name.to_string_lossy();

+        let is_file = e.file_type().ok().map(|t| t.is_file()).unwrap_or(false);

+        if is_file && file_name.ends_with(".rs") {

+            Some(file_name.to_string())

+        } else {

+            None

+        }

+    })

+}

diff --git a/zluda_inject/tests/helpers/direct_cuinit.rs b/zluda_inject/tests/helpers/direct_cuinit.rs
new file mode 100644
index 0000000..8341a60
--- /dev/null
+++ b/zluda_inject/tests/helpers/direct_cuinit.rs
@@ -0,0 +1,9 @@
+#![crate_type = "bin"]

+

+extern "system" {

+    fn cuInit(flags: u32) -> u32;

+}

+

+fn main() {

+    unsafe { cuInit(0) };

+}

diff --git a/zluda_inject/tests/helpers/nvcuda.lib b/zluda_inject/tests/helpers/nvcuda.lib
new file mode 100644
index 0000000..b793c56
--- /dev/null
+++ b/zluda_inject/tests/helpers/nvcuda.lib
diff --git a/zluda_inject/tests/inject.rs b/zluda_inject/tests/inject.rs
new file mode 100644
index 0000000..f63186a
--- /dev/null
+++ b/zluda_inject/tests/inject.rs
@@ -0,0 +1,13 @@
+use std::{env, io, process::Command};

+

+#[test]

+fn direct_cuinit() -> io::Result<()> {

+    let helpers_dir = env!("HELPERS_OUT_DIR");

+    let mut main_exe = Command::new(format!(

+        "{}{}direct_cuinit.exe",

+        helpers_dir,

+        std::path::MAIN_SEPARATOR

+    ));

+    assert!(main_exe.status()?.success());

+    Ok(())

+}