use std::env;
use std::path::Path;
use std::process::Command;

fn main() {
    let project_dir = env::var("CARGO_MANIFEST_DIR").unwrap();
    if !Path::new("./linux").exists() {
        Command::new("git")
            .args(&["clone", "https://github.com/docfate111/linux-lkl.git", "linux"])
            .current_dir(&Path::new(&project_dir))
            .status()
            .unwrap();
        Command::new("make")
            .args(&["CC=/home/t/libafl_cc-for-lkl/target/release/libafl_cc", "-C", "linux/tools/lkl", "-j16", 
		"LLVM=1", "CROSS_COMPILE=x86_64-linux-gnu"])
            .current_dir(&Path::new(&project_dir))
            .status()
            .unwrap();
    }
    if !Path::new("./ext4-00.img").exists() {
         Command::new("dd")
            .args(&["if=/dev/zero", "of=ext4-00.img", "bs=4k", "count=2048"])
            .current_dir(&Path::new(&project_dir))
            .status()
            .unwrap();
        Command::new("mkfs.ext4")
            .args(&["ext4-00.img"])
            .current_dir(&Path::new(&project_dir))
            .status()
            .unwrap();
    }
    println!("cargo:rustc-link-search=native=linux/tools/lkl");
    println!("cargo:rustc-link-lib=lkl");
}