get-derived-key: fix URL + add to KBS confidential-containers#3

Author: eldios

attestation-agent/kbs_protocol/src/evidence_provider/aa_ttrpc.rs

@@ -40,10 +40,9 @@ impl AAEvidenceProvider {
 #[async_trait]
 impl EvidenceProvider for AAEvidenceProvider {
     /// Get derived key using the provided key ID
-    async fn get_derived_key(&self, root_key_hint: &[u8], context: Vec<u8>) -> Result<Vec<u8>> {
+    async fn get_derived_key(&self, key_id: &[u8]) -> Result<Vec<u8>> {
         let req = GetDerivedKeyRequest {
-            RootKeyHint: root_key_hint.to_vec(),
-            Context: context,
+            key_id: key_id.to_vec(),
             ..Default::default()
         };
         let res = self
@@ -54,8 +53,9 @@ impl EvidenceProvider for AAEvidenceProvider {
             )
             .await
             .map_err(|e| Error::AAEvidenceProvider(format!("call ttrpc failed: {e}")))?;
-        Ok(res.DerivedKey)
+        Ok(res.derived_key)
     }
+
     /// Get evidence with as runtime data (report data, challege)
     async fn get_evidence(&self, runtime_data: Vec<u8>) -> Result<String> {
         let req = GetEvidenceRequest {

attestation-agent/kbs_protocol/src/evidence_provider/mock.rs

@@ -19,8 +19,8 @@ impl EvidenceProvider for MockedEvidenceProvider {
         Ok("test evidence".into())
     }
 
-    async fn get_derived_key(&self, _key_id: Vec<u8>) -> Result<String> {
-        Ok("amazing key!".into())
+    async fn get_derived_key(&self, _key_id: Vec<u8>) -> Result<Vec<u8>> {
+        Ok(vec![0u8; 32]) // Return a mock 32-byte key filled with zeros
     }
 
     async fn get_tee_type(&self) -> Result<Tee> {