libublk 0.4.2

Library for building linux block device in userspace
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89

extern crate bindgen;

#[derive(Debug)]
pub struct Fix753 {}
impl bindgen::callbacks::ParseCallbacks for Fix753 {
    fn item_name(&self, original_item_name: &str) -> Option<String> {
        Some(original_item_name.trim_start_matches("Fix753_").to_owned())
    }
}

fn add_serialize(outdir: &std::path::Path) -> anyhow::Result<i32> {
    use std::fs::File;
    use std::io::Write;

    let res = std::fs::read_to_string(outdir.join("ublk_cmd.rs"))?;
    let data = format!(
        "use serde::{{Serialize, Deserialize}};\n{}",
        regex::Regex::new(r"#\s*\[\s*derive\s*\((?P<d>[^)]+)\)\s*\]\s*pub\s*(?P<s>struct|enum)")?
            .replace_all(&res, "#[derive($d, Serialize, Deserialize)] pub $s")
    );
    let mut fd = File::create(outdir.join("ublk_cmd.rs"))?;
    fd.write_all(data.as_bytes())?;

    Ok(0)
}

fn main() {
    use std::env;
    use std::path::PathBuf;

    const INCLUDE: &str = r#"
#include <asm/ioctl.h>
#include <linux/errno.h>
#include <linux/blkzoned.h>
#include "ublk_cmd.h"

#ifdef UBLK_F_CMD_IOCTL_ENCODE
#define MARK_FIX_753(req_name) const unsigned int Fix753_##req_name = req_name;
#else
#define MARK_FIX_753(req_name)
#endif
MARK_FIX_753(UBLK_U_CMD_GET_QUEUE_AFFINITY);
MARK_FIX_753(UBLK_U_CMD_GET_DEV_INFO);
MARK_FIX_753(UBLK_U_CMD_ADD_DEV);
MARK_FIX_753(UBLK_U_CMD_DEL_DEV);
MARK_FIX_753(UBLK_U_CMD_START_DEV);
MARK_FIX_753(UBLK_U_CMD_STOP_DEV);
MARK_FIX_753(UBLK_U_CMD_SET_PARAMS);
MARK_FIX_753(UBLK_U_CMD_GET_PARAMS);
MARK_FIX_753(UBLK_U_CMD_START_USER_RECOVERY);
MARK_FIX_753(UBLK_U_CMD_END_USER_RECOVERY);
MARK_FIX_753(UBLK_U_CMD_GET_DEV_INFO2);
MARK_FIX_753(UBLK_U_CMD_GET_FEATURES);
MARK_FIX_753(UBLK_U_IO_FETCH_REQ);
MARK_FIX_753(UBLK_U_IO_COMMIT_AND_FETCH_REQ);
MARK_FIX_753(UBLK_U_IO_NEED_GET_DATA);
MARK_FIX_753(UBLK_U_CMD_DEL_DEV_ASYNC);
MARK_FIX_753(UBLK_U_IO_REGISTER_IO_BUF);
MARK_FIX_753(UBLK_U_IO_UNREGISTER_IO_BUF);
const int Fix753_UBLK_IO_RES_ABORT = UBLK_IO_RES_ABORT;
    "#;

    let outdir = PathBuf::from(env::var("OUT_DIR").unwrap());
    let mut builder = bindgen::Builder::default();
    builder = builder.header_contents("include-file.h", INCLUDE);

    builder
        .ctypes_prefix("libc")
        .prepend_enum_name(false)
        .derive_default(true)
        .generate_comments(true)
        .use_core()
        .generate_inline_functions(true)
        .allowlist_var("UBLKSRV_.*|UBLK_.*|UBLK_U_.*|Fix753_.*")
        .allowlist_function("ublk_sqe_addr_to_auto_buf_reg")
        .allowlist_function("ublk_auto_buf_reg_to_sqe_addr")
        .allowlist_type("ublksrv_.*|ublk_.*")
        .allowlist_var("BLK_ZONE_.*")
        .allowlist_type("blk_zone_.*")
        .parse_callbacks(Box::new(Fix753 {}))
        .generate()
        .unwrap()
        .write_to_file(outdir.join("ublk_cmd.rs"))
        .unwrap();

    if let Err(error) = add_serialize(&outdir) {
        eprintln!("Error: {}", error)
    }
}