AA: add GetTeeType API

This new API is used for a caller to get current platform name. Signed-off-by: Xynnn007 <[email protected]>
confidential-containers · Jul 18, 2024 · dc48014 · dc48014
1 parent 4619b4b
commit dc48014
Show file tree

Hide file tree

Showing 10 changed files with 610 additions and 30 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/attestation-agent/attestation-agent/Cargo.toml b/attestation-agent/attestation-agent/Cargo.toml
@@ -23,6 +23,7 @@ config.workspace = true
 const_format.workspace = true
 env_logger = { workspace = true, optional = true }
 kbs_protocol = { path = "../kbs_protocol", default-features = false, optional = true }
+kbs-types.workspace = true
 log.workspace = true
 prost = { workspace = true, optional = true }
 protobuf = { workspace = true, optional = true }

diff --git a/attestation-agent/attestation-agent/src/bin/grpc-aa/server.rs b/attestation-agent/attestation-agent/src/bin/grpc-aa/server.rs
@@ -9,8 +9,9 @@ use attestation::attestation_agent_service_server::{
 };
 use attestation::{
     CheckInitDataRequest, CheckInitDataResponse, ExtendRuntimeMeasurementRequest,
-    ExtendRuntimeMeasurementResponse, GetEvidenceRequest, GetEvidenceResponse, GetTokenRequest,
-    GetTokenResponse, UpdateConfigurationRequest, UpdateConfigurationResponse,
+    ExtendRuntimeMeasurementResponse, GetEvidenceRequest, GetEvidenceResponse, GetTeeTypeRequest,
+    GetTeeTypeResponse, GetTokenRequest, GetTokenResponse, UpdateConfigurationRequest,
+    UpdateConfigurationResponse,
 };
 use attestation_agent::{AttestationAPIs, AttestationAgent};
 use log::{debug, error};
@@ -162,6 +163,31 @@ impl AttestationAgentService for AA {
 
         Result::Ok(Response::new(reply))
     }
+
+    async fn get_tee_type(
+        &self,
+        _request: Request<GetTeeTypeRequest>,
+    ) -> Result<Response<GetTeeTypeResponse>, Status> {
+        let mut attestation_agent = self.inner.lock().await;
+
+        debug!("AA (grpc): get tee type ...");
+
+        let tee = attestation_agent.get_tee_type();
+
+        let tee = serde_json::to_string(&tee)
+            .map_err(|e| {
+                error!("AA (ttrpc): get tee type failed:\n {e:?}");
+                Status::internal(format!("[ERROR:{AGENT_NAME}] AA get tee type failed"))
+            })?
+            .trim_end_matches('"')
+            .trim_start_matches('"')
+            .to_string();
+        debug!("AA (ttrpc): get tee type succeeded.");
+
+        let reply = GetTeeTypeResponse { tee };
+
+        Result::Ok(Response::new(reply))
+    }
 }
 
 pub async fn start_grpc_service(socket: SocketAddr, aa: AttestationAgent) -> Result<()> {

diff --git a/attestation-agent/attestation-agent/src/bin/ttrpc-aa/server.rs b/attestation-agent/attestation-agent/src/bin/ttrpc-aa/server.rs
@@ -16,8 +16,8 @@ use std::sync::Arc;
 
 use crate::ttrpc_protocol::attestation_agent::{
     ExtendRuntimeMeasurementRequest, ExtendRuntimeMeasurementResponse, GetEvidenceRequest,
-    GetEvidenceResponse, GetTokenRequest, GetTokenResponse, UpdateConfigurationRequest,
-    UpdateConfigurationResponse,
+    GetEvidenceResponse, GetTeeTypeRequest, GetTeeTypeResponse, GetTokenRequest, GetTokenResponse,
+    UpdateConfigurationRequest, UpdateConfigurationResponse,
 };
 use crate::ttrpc_protocol::attestation_agent_ttrpc::{
     create_attestation_agent_service, AttestationAgentService,
@@ -145,6 +145,35 @@ impl AttestationAgentService for AA {
         let reply = UpdateConfigurationResponse::new();
         ::ttrpc::Result::Ok(reply)
     }
+
+    async fn get_tee_type(
+        &self,
+        _ctx: &::ttrpc::r#async::TtrpcContext,
+        _req: GetTeeTypeRequest,
+    ) -> ::ttrpc::Result<GetTeeTypeResponse> {
+        debug!("AA (ttrpc): get tee type ...");
+
+        let mut attestation_agent = self.inner.lock().await;
+
+        let tee = attestation_agent.get_tee_type();
+
+        let res = serde_json::to_string(&tee)
+            .map_err(|e| {
+                error!("AA (ttrpc): get tee type failed:\n {e:?}");
+                let mut error_status = ::ttrpc::proto::Status::new();
+                error_status.set_code(Code::INTERNAL);
+                error_status
+                    .set_message(format!("[ERROR:{AGENT_NAME}] AA-KBC get tee type failed"));
+                ::ttrpc::Error::RpcStatus(error_status)
+            })?
+            .trim_end_matches('"')
+            .trim_start_matches('"')
+            .to_string();
+        debug!("AA (ttrpc): get tee type succeeded.");
+        let mut reply = GetTeeTypeResponse::new();
+        reply.tee = res;
+        ::ttrpc::Result::Ok(reply)
+    }
 }
 
 pub fn start_ttrpc_service(aa: AttestationAgent) -> Result<HashMap<String, Service>> {