// Copyright (c) 2022-2024 The MobileCoin Foundation

//! Builds the FFI type bindings for the common SGX SDK types

use mc_sgx_core_build::SgxParseCallbacks;

// The types to generate bindings for.
//
// To keep the noise out of the bindings, we use the underlying type and tell
// bindgen to map directly to `sgx_<name>` version.
//
// For example `_foo_name` would be the underlying type:
// ```C
//      typedef struct _foo_name {
//          int a;
//          float b;
//      } sgx_foo_name;
// ```
const CORE_TYPES: &[&str] = &[
    "_attributes_t",
    "_key_request_t",
    "_report_t",
    "_sgx_cpu_svn_t",
    "_sgx_key_id_t",
    "_sgx_measurement_t",
    "_sgx_misc_attribute_t",
    "_status_t",
    "sgx_config_id_t",
    "sgx_config_svn_t",
    "sgx_isv_svn_t",
    "sgx_isvext_prod_id_t",
    "sgx_isvfamily_id_t",
    "sgx_key_128bit_t",
    "sgx_mac_t",
    "sgx_misc_select_t",
    "sgx_prod_id_t",
    "_sgx_att_key_id_ext_t",
    "_sgx_ql_att_key_id_t",
    "_quote_nonce",
    "sgx_epid_group_id_t",
    "_spid_t",
    "_basename_t",
    "sgx_quote_sign_type_t",
    "_quote_t",
    "_platform_info",
    "_update_info_bit",
    "_att_key_id_t",
    "_qe_report_info_t",
];

const CORE_CONSTS: &[&str] = &[
    "SGX_CPUSVN_SIZE",
    "SGX_KEYID_SIZE",
    "SGX_KEY_REQUEST_RESERVED2_BYTES",
    "SGX_KEYSELECT_EINITTOKEN",
    "SGX_KEYSELECT_PROVISION",
    "SGX_KEYSELECT_PROVISION_SEAL",
    "SGX_KEYSELECT_REPORT",
    "SGX_KEYSELECT_SEAL",
    "SGX_KEYPOLICY_MRENCLAVE",
    "SGX_KEYPOLICY_MRSIGNER",
    "SGX_KEYPOLICY_NOISVPRODID",
    "SGX_KEYPOLICY_CONFIGID",
    "SGX_KEYPOLICY_ISVFAMILYID",
    "SGX_KEYPOLICY_ISVEXTPRODID",
    "SGX_REPORT_BODY_RESERVED1_BYTES",
    "SGX_REPORT_BODY_RESERVED2_BYTES",
    "SGX_REPORT_BODY_RESERVED3_BYTES",
    "SGX_REPORT_BODY_RESERVED4_BYTES",
    "SGX_HASH_SIZE",
    "SGX_ISVEXT_PROD_ID_SIZE",
    "SGX_REPORT_DATA_SIZE",
    "SGX_CONFIGID_SIZE",
    "SGX_ISV_FAMILY_ID_SIZE",
    "SGX_TARGET_INFO_RESERVED1_BYTES",
    "SGX_TARGET_INFO_RESERVED2_BYTES",
    "SGX_TARGET_INFO_RESERVED3_BYTES",
    "SGX_MAC_SIZE",
    "SGX_PLATFORM_INFO_SIZE",
    "SGX_FLAGS_INITTED",
    "SGX_FLAGS_DEBUG",
    "SGX_FLAGS_MODE64BIT",
    "SGX_FLAGS_PROVISION_KEY",
    "SGX_FLAGS_EINITTOKEN_KEY",
    "SGX_FLAGS_KSS",
    "SGX_FLAGS_AEX_NOTIFY",
    "SGX_FLAGS_NON_CHECK_BITS",
    "SGX_XFRM_LEGACY",
    "SGX_XFRM_AVX",
    "SGX_XFRM_AVX512",
    "SGX_XFRM_MPX",
    "SGX_XFRM_PKRU",
    "SGX_XFRM_AMX",
    "SGX_XFRM_RESERVED",
];

fn main() {
    let callback = SgxParseCallbacks::default()
        .enum_types(["sgx_status_t", "sgx_quote_sign_type_t"])
        .derive_copy([
            "sgx_update_info_bit_t",
            "sgx_ql_att_key_id_t",
            "sgx_att_key_id_ext_t",
            "sgx_qe_report_info_t",
            "sgx_quote_nonce_t",
            "sgx_report_t",
            "sgx_key_id_t",
            "sgx_cpu_svn_t",
            "sgx_measurement_t",
            "sgx_attributes_t",
        ])
        .dynamically_sized_types(["sgx_quote_t"])
        .serialize_types([
            "sgx_measurement_t",
            "sgx_attributes_t",
            "sgx_report_t",
            "sgx_key_id_t",
            "sgx_cpu_svn_t",
            "sgx_quote_nonce_t",
        ])
        .derive_default([
            "sgx_report_t",
            "sgx_attributes_t",
            "sgx_basename_t",
            "sgx_quote_nonce_t",
            "sgx_update_info_bit_t",
            "sgx_qe_report_info_t",
        ]);
    let mut builder = mc_sgx_core_build::sgx_builder()
        .header("wrapper.h")
        .parse_callbacks(Box::new(callback));

    for t in CORE_TYPES.iter() {
        builder = builder.allowlist_type(t)
    }

    for c in CORE_CONSTS.iter() {
        builder = builder.allowlist_var(c)
    }

    let bindings = builder.generate().expect("Unable to generate bindings");

    let out_path = mc_sgx_core_build::build_output_dir();
    bindings
        .write_to_file(out_path.join("bindings.rs"))
        .expect("Couldn't write bindings!");
}