DragonOS/kernel/build.rs

extern crate bindgen;
extern crate cc;
// use ::std::env;

use std::path::PathBuf;

use cc::Build;

fn main() {
    // Tell cargo to look for shared libraries in the specified directory
    println!("cargo:rustc-link-search=src");
    println!("cargo:rerun-if-changed=src/include/bindings/wrapper.h");

    generate_bindings();
    CFilesBuilder::build();
}

fn generate_bindings() {
    // let crate_dir = env::var("CARGO_MANIFEST_DIR").unwrap();
    let out_path = PathBuf::from(String::from("src/include/bindings/"));

    // The bindgen::Builder is the main entry point
    // to bindgen, and lets you build up options for
    // the resulting bindings.
    {
        let bindings = bindgen::Builder::default()
            .clang_arg("-I./src")
            .clang_arg("-I./src/include")
            .clang_arg("-I./src/arch/x86_64/include") // todo: 当引入多种架构之后，需要修改这里，对于不同的架构编译时，include不同的路径
            // The input header we would like to generate
            // bindings for.
            .header("src/include/bindings/wrapper.h")
            .blocklist_file("src/include/bindings/bindings.h")
            .clang_arg("--target=x86_64-none-none")
            .clang_arg("-v")
            // 使用core，并将c语言的类型改为core::ffi，而不是使用std库。
            .use_core()
            .ctypes_prefix("::core::ffi")
            .generate_inline_functions(true)
            .raw_line("#![allow(dead_code)]")
            .raw_line("#![allow(non_upper_case_globals)]")
            .raw_line("#![allow(non_camel_case_types)]")
            // Tell cargo to invalidate the built crate whenever any of the
            // included header files changed.
            .parse_callbacks(Box::new(bindgen::CargoCallbacks))
            // Finish the builder and generate the bindings.
            .generate()
            // Unwrap the Result and panic on failure.
            .expect("Unable to generate bindings");

        bindings
            .write_to_file(out_path.join("bindings.rs"))
            .expect("Couldn't write bindings!");
    }
}

/// 构建项目的c文件
struct CFilesBuilder;

impl CFilesBuilder {
    fn build() {
        let mut c = cc::Build::new();
        Self::setup_global_flags(&mut c);
        Self::setup_defines(&mut c);
        Self::setup_global_include_dir(&mut c);
        Self::setup_files(&mut c);
        c.compile("dragonos_kernel_cfiles");
    }

    fn setup_global_flags(c: &mut Build) {
        c.flag("-mcmodel=large")
            .flag("-fno-builtin")
            .flag("-nostdlib")
            .flag("-fno-stack-protector")
            .flag("-fno-pie")
            .flag("-Wno-expansion-to-defined")
            .flag("-Wno-unused-parameter")
            .flag("-m64")
            .flag("-O1");
    }

    fn setup_defines(c: &mut Build) {
        if let Ok(k) = std::env::var("EMULATOR") {
            c.define("EMULATOR", Some(k.as_str()));
        } else {
            c.define("EMULATOR", "__NO_EMULATION__");
        }

        {
            #[cfg(target_arch = "x86_64")]
            c.define("__x86_64__", None);
        }
    }

    fn setup_global_include_dir(c: &mut Build) {
        c.include("src/include");
        c.include("src");
        c.include(".");

        #[cfg(target_arch = "x86_64")]
        c.include("src/arch/x86_64/include");
    }

    /// 设置需要编译的文件
    fn setup_files(c: &mut Build) {
        let mut files = Vec::new();

        #[cfg(target_arch = "x86_64")]
        Self::setup_files_x86_64(&mut files);

        Self::set_rerun_if_files_changed(&files);
        c.files(files.as_slice());
    }

    /// 设置x86_64架构下需要编译的C文件
    fn setup_files_x86_64(files: &mut Vec<PathBuf>) {
        files.push(PathBuf::from("src/arch/x86_64/driver/hpet.c"));
        // 获取`kernel/src/arch/x86_64/driver/apic`下的所有C文件
        files.append(&mut FileUtils::list_all_files(
            &PathBuf::from("src/arch/x86_64/driver/apic"),
            Some("c"),
            true,
        ));
    }

    /// 设置Cargo对文件更改的监听
    fn set_rerun_if_files_changed(files: &Vec<PathBuf>) {
        for f in files {
            println!("cargo:rerun-if-changed={}", f.to_str().unwrap());
        }
    }
}

struct FileUtils;

impl FileUtils {
    /// 列出指定目录下的所有文件
    ///
    /// ## 参数
    ///
    /// - `path` - 指定的目录
    /// - `ext_name` - 文件的扩展名，如果为None，则列出所有文件
    /// - `recursive` - 是否递归列出所有文件
    pub fn list_all_files(path: &PathBuf, ext_name: Option<&str>, recursive: bool) -> Vec<PathBuf> {
        let mut queue: Vec<PathBuf> = Vec::new();
        let mut result = Vec::new();
        queue.push(path.clone());

        while !queue.is_empty() {
            let path = queue.pop().unwrap();
            let d = std::fs::read_dir(path);
            if d.is_err() {
                continue;
            }
            let d = d.unwrap();

            d.for_each(|ent| {
                if let Ok(ent) = ent {
                    if let Ok(file_type) = ent.file_type() {
                        if file_type.is_file() {
                            if let Some(e) = ext_name {
                                if let Some(ext) = ent.path().extension() {
                                    if ext == e {
                                        result.push(ent.path());
                                    }
                                }
                            }
                        } else if file_type.is_dir() && recursive {
                            queue.push(ent.path());
                        }
                    }
                }
            });
        }

        return result;
    }
}