Skip to content

Commit d7bfcb7

Browse files
committed
fix to apply snake case
Signed-off-by: Jun Kimura <jun.kimura@datachain.jp>
1 parent 8718f75 commit d7bfcb7

File tree

17 files changed

+108
-106
lines changed

17 files changed

+108
-106
lines changed

crates/collaterals/src/lib.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,6 @@ pub mod enclave_identity;
44
pub mod enclave_report;
55
pub mod quote;
66
pub mod sgx_extensions;
7-
pub mod tcbinfo;
7+
pub mod tcb_info;
88
pub mod utils;
99
pub use openssl;
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
use dcap_types::tcbinfo::{
1+
use dcap_types::tcb_info::{
22
TcbComponent, TcbInfoV3, TcbInfoV3Inner, TcbInfoV3TcbLevel, TcbInfoV3TcbLevelItem, TdxModule,
33
TdxModuleIdentities,
44
};

crates/pcs/src/client.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ impl PCSClient {
7575
let update_policy = self.update_policy();
7676

7777
// get the TCB info of the platform
78-
let (tcbinfo_json, sgx_tcb_signing_der) = {
78+
let (tcb_info_json, sgx_tcb_signing_der) = {
7979
let fmspc = hex::encode_upper(sgx_extensions.fmspc);
8080
let res = http_get(format!(
8181
"{base_url}/tcb?fmspc={fmspc}&update={update_policy}"
@@ -86,7 +86,7 @@ impl PCSClient {
8686
};
8787

8888
// get the QE identity
89-
let qeidentity_json = http_get(format!("{base_url}/qe/identity?update={update_policy}"))?
89+
let qe_identity_json = http_get(format!("{base_url}/qe/identity?update={update_policy}"))?
9090
.bytes()?
9191
.to_vec();
9292

@@ -111,8 +111,8 @@ impl PCSClient {
111111
.to_vec();
112112

113113
Ok(QvCollateral {
114-
tcbinfo_json,
115-
qeidentity_json,
114+
tcb_info_json,
115+
qe_identity_json,
116116
sgx_intel_root_ca_der: sgx_root_cert_der,
117117
sgx_tcb_signing_der,
118118
sgx_intel_root_ca_crl_der,

crates/quote-verifier/src/cert.rs

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ use crate::crypto::verify_p256_signature_der;
22
use anyhow::bail;
33
use core::str::FromStr;
44
use dcap_types::cert::{SgxExtensionTcbLevel, SgxExtensions};
5-
use dcap_types::tcbinfo::{TcbComponent, TcbInfoV3};
5+
use dcap_types::tcb_info::{TcbComponent, TcbInfoV3};
66
use dcap_types::TcbInfoV3TcbStatus;
77
use dcap_types::{SGX_TEE_TYPE, TDX_TEE_TYPE};
88
use x509_parser::prelude::*;
@@ -83,23 +83,23 @@ pub fn get_x509_issuer_cn(cert: &X509Certificate) -> String {
8383
/// * `tee_type` - The type of TEE (SGX or TDX)
8484
/// * `tee_tcb_svn` - The TCB SVN of the TEE (only for TDX)
8585
/// * `sgx_extensions` - The SGX Extensions from the PCK Certificate
86-
/// * `tcbinfov3` - The TCB Info V3
86+
/// * `tcb_info_v3` - The TCB Info V3
8787
/// # Returns
8888
/// * `(sgx_tcb_status, tdx_tcb_status, advisory_ids)` - The TCB status of the SGX and TDX, and the advisory IDs
8989
pub fn get_sgx_tdx_tcb_status_v3(
9090
tee_type: u32,
9191
tee_tcb_svn: Option<[u8; 16]>,
9292
sgx_extensions: &SgxExtensions,
93-
tcbinfov3: &TcbInfoV3,
93+
tcb_info_v3: &TcbInfoV3,
9494
) -> crate::Result<(TcbInfoV3TcbStatus, Option<TcbInfoV3TcbStatus>, Vec<String>)> {
9595
if tee_type == SGX_TEE_TYPE {
96-
if tcbinfov3.tcb_info.id != "SGX" {
96+
if tcb_info_v3.tcb_info.id != "SGX" {
9797
bail!("Invalid TCB Info ID for SGX TEE Type");
9898
} else if tee_tcb_svn.is_some() {
9999
bail!("SGX TCB SVN is not needed");
100100
}
101101
} else if tee_type == TDX_TEE_TYPE {
102-
if tcbinfov3.tcb_info.id != "TDX" {
102+
if tcb_info_v3.tcb_info.id != "TDX" {
103103
bail!("Invalid TCB Info ID for TDX TEE Type");
104104
} else if tee_tcb_svn.is_none() {
105105
bail!("TDX TCB SVN is missing");
@@ -108,27 +108,27 @@ pub fn get_sgx_tdx_tcb_status_v3(
108108
bail!("Unsupported TEE type: {}", tee_type);
109109
}
110110

111-
if sgx_extensions.fmspc != tcbinfov3.tcb_info.fmspc()? {
111+
if sgx_extensions.fmspc != tcb_info_v3.tcb_info.fmspc()? {
112112
// ref. https://github.com/intel/SGX-TDX-DCAP-QuoteVerificationLibrary/blob/812e0fa140a284b772b2d8b08583c761e23ec3b3/Src/AttestationLibrary/src/Verifiers/QuoteVerifier.cpp#L117
113113
bail!(
114114
"FMSPC does not match: {:x?} != {:x?}",
115115
sgx_extensions.fmspc,
116-
tcbinfov3.tcb_info.fmspc()?
116+
tcb_info_v3.tcb_info.fmspc()?
117117
);
118-
} else if sgx_extensions.pceid != tcbinfov3.tcb_info.pce_id()? {
118+
} else if sgx_extensions.pceid != tcb_info_v3.tcb_info.pce_id()? {
119119
// ref. https://github.com/intel/SGX-TDX-DCAP-QuoteVerificationLibrary/blob/812e0fa140a284b772b2d8b08583c761e23ec3b3/Src/AttestationLibrary/src/Verifiers/QuoteVerifier.cpp#L124
120120
bail!(
121121
"PCE ID does not match: {:x?} != {:x?}",
122122
sgx_extensions.pceid,
123-
tcbinfov3.tcb_info.pce_id()?
123+
tcb_info_v3.tcb_info.pce_id()?
124124
);
125125
}
126126

127127
let mut sgx_tcb_status: Option<TcbInfoV3TcbStatus> = None;
128128
let tcb = &sgx_extensions.tcb;
129129
let extension_pcesvn = tcb.pcesvn;
130130

131-
for tcb_level in tcbinfov3.tcb_info.tcb_levels.iter() {
131+
for tcb_level in tcb_info_v3.tcb_info.tcb_levels.iter() {
132132
if sgx_tcb_status.is_none()
133133
&& match_sgxtcbcomp(tcb, &tcb_level.tcb.sgxtcbcomponents)
134134
&& extension_pcesvn >= tcb_level.tcb.pcesvn

crates/quote-verifier/src/collateral.rs

Lines changed: 26 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
use anyhow::bail;
22
use dcap_types::enclave_identity::EnclaveIdentityV2;
3-
use dcap_types::tcbinfo::TcbInfoV3;
3+
use dcap_types::tcb_info::TcbInfoV3;
44
use dcap_types::utils::{parse_crl_der, parse_x509_der};
55
use x509_parser::{certificate::X509Certificate, revocation_list::CertificateRevocationList};
66

@@ -12,10 +12,10 @@ use crate::Result;
1212
pub struct QvCollateral {
1313
/// TCBInfo in JSON format
1414
/// ref. <https://api.portal.trustedservices.intel.com/content/documentation.html#pcs-tcb-info-model-v3>
15-
pub tcbinfo_json: Vec<u8>,
15+
pub tcb_info_json: Vec<u8>,
1616
/// QEIdentity in JSON format
1717
/// ref. <https://api.portal.trustedservices.intel.com/content/documentation.html#pcs-enclave-identity-model-v2>
18-
pub qeidentity_json: Vec<u8>,
18+
pub qe_identity_json: Vec<u8>,
1919
/// SGX Intel Root CA certificate in DER format
2020
/// ref. <https://certificates.trustedservices.intel.com/Intel_SGX_Provisioning_Certification_RootCA.pem>
2121
pub sgx_intel_root_ca_der: Vec<u8>,
@@ -42,24 +42,24 @@ impl QvCollateral {
4242

4343
// get the total length
4444
let total_length = 4 * 6
45-
+ self.tcbinfo_json.len()
46-
+ self.qeidentity_json.len()
45+
+ self.tcb_info_json.len()
46+
+ self.qe_identity_json.len()
4747
+ self.sgx_intel_root_ca_der.len()
4848
+ self.sgx_tcb_signing_der.len()
4949
+ self.sgx_intel_root_ca_crl_der.len()
5050
+ self.sgx_pck_crl_der.len();
5151

5252
// create the vec and copy the data
5353
let mut data = Vec::with_capacity(total_length);
54-
data.extend_from_slice(&(self.tcbinfo_json.len() as u32).to_le_bytes());
55-
data.extend_from_slice(&(self.qeidentity_json.len() as u32).to_le_bytes());
54+
data.extend_from_slice(&(self.tcb_info_json.len() as u32).to_le_bytes());
55+
data.extend_from_slice(&(self.qe_identity_json.len() as u32).to_le_bytes());
5656
data.extend_from_slice(&(self.sgx_intel_root_ca_der.len() as u32).to_le_bytes());
5757
data.extend_from_slice(&(self.sgx_tcb_signing_der.len() as u32).to_le_bytes());
5858
data.extend_from_slice(&(self.sgx_intel_root_ca_crl_der.len() as u32).to_le_bytes());
5959
data.extend_from_slice(&(self.sgx_pck_crl_der.len() as u32).to_le_bytes());
6060

61-
data.extend_from_slice(&self.tcbinfo_json);
62-
data.extend_from_slice(&self.qeidentity_json);
61+
data.extend_from_slice(&self.tcb_info_json);
62+
data.extend_from_slice(&self.qe_identity_json);
6363
data.extend_from_slice(&self.sgx_intel_root_ca_der);
6464
data.extend_from_slice(&self.sgx_tcb_signing_der);
6565
data.extend_from_slice(&self.sgx_intel_root_ca_crl_der);
@@ -77,8 +77,8 @@ impl QvCollateral {
7777

7878
// reverse the serialization process
7979
// each length is 4 bytes long, we have a total of 6 members
80-
let tcbinfo_json_len = u32::from_le_bytes(slice[0..4].try_into()?) as usize;
81-
let qeidentity_json_len = u32::from_le_bytes(slice[4..8].try_into()?) as usize;
80+
let tcb_info_json_len = u32::from_le_bytes(slice[0..4].try_into()?) as usize;
81+
let qe_identity_json_len = u32::from_le_bytes(slice[4..8].try_into()?) as usize;
8282
let sgx_intel_root_ca_der_len = u32::from_le_bytes(slice[8..12].try_into()?) as usize;
8383
let sgx_tcb_signing_der_len = u32::from_le_bytes(slice[12..16].try_into()?) as usize;
8484
let sgx_intel_root_ca_crl_der_len = u32::from_le_bytes(slice[16..20].try_into()?) as usize;
@@ -88,8 +88,8 @@ impl QvCollateral {
8888

8989
if slice.len()
9090
< offset
91-
+ tcbinfo_json_len
92-
+ qeidentity_json_len
91+
+ tcb_info_json_len
92+
+ qe_identity_json_len
9393
+ sgx_intel_root_ca_der_len
9494
+ sgx_tcb_signing_der_len
9595
+ sgx_intel_root_ca_crl_der_len
@@ -98,10 +98,10 @@ impl QvCollateral {
9898
bail!("Invalid QvCollateral length");
9999
}
100100

101-
let tcbinfo_json = slice[offset..offset + tcbinfo_json_len].to_vec();
102-
offset += tcbinfo_json_len;
103-
let qeidentity_json = slice[offset..offset + qeidentity_json_len].to_vec();
104-
offset += qeidentity_json_len;
101+
let tcb_info_json = slice[offset..offset + tcb_info_json_len].to_vec();
102+
offset += tcb_info_json_len;
103+
let qe_identity_json = slice[offset..offset + qe_identity_json_len].to_vec();
104+
offset += qe_identity_json_len;
105105
let sgx_intel_root_ca_der = slice[offset..offset + sgx_intel_root_ca_der_len].to_vec();
106106
offset += sgx_intel_root_ca_der_len;
107107
let sgx_tcb_signing_der = slice[offset..offset + sgx_tcb_signing_der_len].to_vec();
@@ -113,8 +113,8 @@ impl QvCollateral {
113113
offset += sgx_pck_crl_der_len;
114114

115115
Ok(QvCollateral {
116-
tcbinfo_json,
117-
qeidentity_json,
116+
tcb_info_json,
117+
qe_identity_json,
118118
sgx_intel_root_ca_der,
119119
sgx_tcb_signing_der,
120120
sgx_intel_root_ca_crl_der,
@@ -123,17 +123,17 @@ impl QvCollateral {
123123
}
124124

125125
/// Returns the TCBInfoV3 struct from the TCBInfo JSON bytes
126-
pub fn get_tcbinfov3(&self) -> Result<TcbInfoV3> {
127-
let tcbinfo: TcbInfoV3 = serde_json::from_slice(&self.tcbinfo_json)?;
128-
if tcbinfo.tcb_info.version != 3 {
129-
bail!("Invalid TCB Info version: {}", tcbinfo.tcb_info.version);
126+
pub fn get_tcb_info_v3(&self) -> Result<TcbInfoV3> {
127+
let tcb_info_v3: TcbInfoV3 = serde_json::from_slice(&self.tcb_info_json)?;
128+
if tcb_info_v3.tcb_info.version != 3 {
129+
bail!("Invalid TCB Info version: {}", tcb_info_v3.tcb_info.version);
130130
}
131-
Ok(tcbinfo)
131+
Ok(tcb_info_v3)
132132
}
133133

134134
/// Returns the EnclaveIdentityV2 struct from the QEIdentity JSON bytes
135-
pub fn get_qeidentityv2(&self) -> Result<EnclaveIdentityV2> {
136-
let qe: EnclaveIdentityV2 = serde_json::from_slice(&self.qeidentity_json)?;
135+
pub fn get_qe_identity_v2(&self) -> Result<EnclaveIdentityV2> {
136+
let qe: EnclaveIdentityV2 = serde_json::from_slice(&self.qe_identity_json)?;
137137
if qe.enclave_identity.version != 2 {
138138
bail!(
139139
"Invalid QE Identity version: {}",

crates/quote-verifier/src/enclave_identity.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -64,9 +64,9 @@ pub fn validate_qe_identityv2(
6464
/// ref. <https://github.com/intel/SGX-TDX-DCAP-QuoteVerificationLibrary/blob/812e0fa140a284b772b2d8b08583c761e23ec3b3/Src/AttestationLibrary/src/Verifiers/EnclaveIdentityV2.cpp#L326>
6565
pub fn get_qe_tcb_status(
6666
qe_report_isv_svn: u16,
67-
qeidentityv2_tcb_levels: &[EnclaveIdentityV2TcbLevelItem],
67+
qe_identity_v2_tcb_levels: &[EnclaveIdentityV2TcbLevelItem],
6868
) -> Result<(EnclaveIdentityV2TcbStatus, Vec<String>)> {
69-
for tcb_level in qeidentityv2_tcb_levels {
69+
for tcb_level in qe_identity_v2_tcb_levels {
7070
if tcb_level.tcb.isvsvn <= qe_report_isv_svn {
7171
return Ok((
7272
EnclaveIdentityV2TcbStatus::from_str(&tcb_level.tcb_status)?,

crates/quote-verifier/src/lib.rs

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ pub mod enclave_identity;
66
pub mod pck;
77
pub mod quotes;
88
pub mod sgx_extensions;
9-
pub mod tcbinfo;
9+
pub mod tcb_info;
1010
pub mod tdx_module;
1111
#[cfg(test)]
1212
pub mod tests;
@@ -23,10 +23,10 @@ mod quote_verifier_tests {
2323
use crate::collateral::QvCollateral;
2424
use crate::crypto::keccak256sum;
2525
use crate::quotes::{version_3::verify_quote_v3, version_4::verify_quote_v4};
26-
use crate::tcbinfo::validate_tcbinfov3;
26+
use crate::tcb_info::validate_tcb_info_v3;
2727
use crate::verifier::{QuoteVerificationOutput, Status};
2828
use dcap_types::quotes::{version_3::QuoteV3, version_4::QuoteV4};
29-
use dcap_types::tcbinfo::TcbInfoV3;
29+
use dcap_types::tcb_info::TcbInfoV3;
3030
use dcap_types::utils::{parse_crl_der, parse_pem, parse_x509_der, pem_to_der};
3131
use dcap_types::{SGX_TEE_TYPE, TDX_TEE_TYPE};
3232

@@ -43,25 +43,25 @@ mod quote_verifier_tests {
4343
}
4444

4545
#[test]
46-
fn test_tcbinfov3_sgx() {
47-
let tcbinfov3: TcbInfoV3 =
46+
fn test_tcb_info_v3_sgx() {
47+
let tcb_info_v3: TcbInfoV3 =
4848
serde_json::from_str(include_str!("../../../data/v3/tcbinfov3_00906ED50000.json"))
4949
.unwrap();
5050
let sgx_signing_cert_pem =
5151
&parse_pem(include_bytes!("../../../data/v3/signing_cert.pem")).unwrap()[0];
5252
let sgx_signing_cert = parse_x509_der(&sgx_signing_cert_pem.contents).unwrap();
53-
assert!(validate_tcbinfov3(SGX_TEE_TYPE, &tcbinfov3, &sgx_signing_cert).is_ok());
53+
assert!(validate_tcb_info_v3(SGX_TEE_TYPE, &tcb_info_v3, &sgx_signing_cert).is_ok());
5454
}
5555

5656
#[test]
57-
fn test_tcbinfov3_tdx() {
58-
let tcbinfov3: TcbInfoV3 =
57+
fn test_tcb_info_v3_tdx() {
58+
let tcb_info_v3: TcbInfoV3 =
5959
serde_json::from_str(include_str!("../../../data/tcbinfov3_00806f050000.json"))
6060
.unwrap();
6161
let sgx_signing_cert_pem =
6262
&parse_pem(include_bytes!("../../../data/signing_cert.pem")).unwrap()[0];
6363
let sgx_signing_cert = parse_x509_der(&sgx_signing_cert_pem.contents).unwrap();
64-
assert!(validate_tcbinfov3(TDX_TEE_TYPE, &tcbinfov3, &sgx_signing_cert).is_ok());
64+
assert!(validate_tcb_info_v3(TDX_TEE_TYPE, &tcb_info_v3, &sgx_signing_cert).is_ok());
6565
}
6666

6767
#[test]
@@ -76,8 +76,8 @@ mod quote_verifier_tests {
7676
#[test]
7777
fn test_verifyv3() {
7878
let collaterals = QvCollateral {
79-
tcbinfo_json: include_bytes!("../../../data/v3/tcbinfov3_00906ED50000.json").to_vec(),
80-
qeidentity_json: include_bytes!("../../../data/v3/qeidentityv2.json").to_vec(),
79+
tcb_info_json: include_bytes!("../../../data/v3/tcbinfov3_00906ED50000.json").to_vec(),
80+
qe_identity_json: include_bytes!("../../../data/v3/qeidentityv2.json").to_vec(),
8181
sgx_intel_root_ca_der: include_bytes!(
8282
"../../../data/Intel_SGX_Provisioning_Certification_RootCA.cer"
8383
)
@@ -134,8 +134,8 @@ mod quote_verifier_tests {
134134
#[test]
135135
fn test_verifyv4() {
136136
let collaterals = QvCollateral {
137-
tcbinfo_json: include_bytes!("../../../data/v4/tcbinfov3_00806f050000.json").to_vec(),
138-
qeidentity_json: include_bytes!("../../../data/v4/qeidentityv2_apiv4.json").to_vec(),
137+
tcb_info_json: include_bytes!("../../../data/v4/tcbinfov3_00806f050000.json").to_vec(),
138+
qe_identity_json: include_bytes!("../../../data/v4/qeidentityv2_apiv4.json").to_vec(),
139139
sgx_intel_root_ca_der: include_bytes!(
140140
"../../../data/Intel_SGX_Provisioning_Certification_RootCA.cer"
141141
)

0 commit comments

Comments
 (0)