agent: Add API EmitEvent

Author: kvinwang

guest-agent/rpc/proto/agent_rpc.proto

@@ -39,6 +39,9 @@ service DstackGuest {
   // Generates a TDX quote with given report data.
   rpc GetQuote(RawQuoteArgs) returns (GetQuoteResponse) {}
 
+  // Emit an event. This extends the event to RTMR3 on TDX platform.
+  rpc EmitEvent(EmitEventArgs) returns (google.protobuf.Empty) {}
+
   // Get worker info
   rpc Info(google.protobuf.Empty) returns (WorkerInfo) {}
 }
@@ -161,6 +164,13 @@ message GetQuoteResponse {
   bytes report_data = 3;
 }
 
+message EmitEventArgs {
+  // The event name
+  string event = 1;
+  // The event data
+  bytes payload = 2;
+}
+
 // The request to derive a key
 message WorkerInfo {
   // App ID

guest-agent/src/rpc_service.rs

@@ -6,9 +6,9 @@ use dstack_guest_agent_rpc::{
     dstack_guest_server::{DstackGuestRpc, DstackGuestServer},
     tappd_server::{TappdRpc, TappdServer},
     worker_server::{WorkerRpc, WorkerServer},
-    DeriveK256KeyResponse, DeriveKeyArgs, GetKeyArgs, GetKeyResponse, GetQuoteResponse,
-    GetTlsKeyArgs, GetTlsKeyResponse, RawQuoteArgs, TdxQuoteArgs, TdxQuoteResponse, WorkerInfo,
-    WorkerVersion,
+    DeriveK256KeyResponse, DeriveKeyArgs, EmitEventArgs, GetKeyArgs, GetKeyResponse,
+    GetQuoteResponse, GetTlsKeyArgs, GetTlsKeyResponse, RawQuoteArgs, TdxQuoteArgs,
+    TdxQuoteResponse, WorkerInfo, WorkerVersion,
 };
 use dstack_types::AppKeys;
 use fs_err as fs;
@@ -161,6 +161,13 @@ impl DstackGuestRpc for InternalRpcHandler {
         })
     }
 
+    async fn emit_event(self, request: EmitEventArgs) -> Result<()> {
+        if self.state.config().simulator.enabled {
+            return Ok(());
+        }
+        tdx_attest::extend_rtmr3(&request.event, &request.payload)
+    }
+
     async fn info(self) -> Result<WorkerInfo> {
         ExternalRpcHandler { state: self.state }.info().await
     }