sdk: Add API EmitEvent

Author: kvinwang

sdk/curl/api.md

@@ -165,6 +165,33 @@ curl --unix-socket /var/run/dstack.sock http://dstack/Info
 }
 ```
 
+### 5. Emit Event
+
+Emit an event to be extended to RTMR3 on TDX platform. This API requires Dstack OS 0.5.0 or later.
+
+**Endpoint:** `/EmitEvent`
+
+**Request Parameters:**
+
+| Field | Type | Description | Example |
+|-------|------|-------------|----------|
+| `event` | string | The event name | `"custom-event"` |
+| `payload` | string | Hex-encoded payload data | `"deadbeef"` |
+
+**Example:**
+```bash
+curl --unix-socket /var/run/dstack.sock -X POST \
+  http://dstack/EmitEvent \
+  -H 'Content-Type: application/json' \
+  -d '{
+    "event": "custom-event",
+    "payload": "deadbeef"
+  }'
+```
+
+**Response:**
+Empty response with HTTP 200 status code on success.
+
 ## Error Responses
 
 All endpoints may return the following HTTP status codes:

sdk/go/dstack/client.go

@@ -417,3 +417,15 @@ func (c *DstackClient) Info(ctx context.Context) (*InfoResponse, error) {
 
 	return &response, nil
 }
+
+// EmitEvent sends an event to be extended to RTMR3 on TDX platform.
+// The event will be extended to RTMR3 with the provided name and payload.
+//
+// Requires Dstack OS 0.5.0 or later.
+func (c *DstackClient) EmitEvent(ctx context.Context, event string, payload []byte) error {
+	_, err := c.sendRPCRequest(ctx, "/EmitEvent", map[string]interface{}{
+		"event":   event,
+		"payload": hex.EncodeToString(payload),
+	})
+	return err
+}

sdk/js/src/index.ts

@@ -314,4 +314,28 @@ export class DstackClient {
       tcb_info: JSON.parse(result.tcb_info) as TcbInfo,
     })
   }
+
+  /**
+   * Emit an event. This extends the event to RTMR3 on TDX platform.
+   *
+   * Requires Dstack OS 0.5.0 or later.
+   *
+   * @param event The event name
+   * @param payload The event data as string or Buffer or Uint8Array
+   */
+  async emitEvent(event: string, payload: string | Buffer | Uint8Array): Promise<void> {
+    if (!event) {
+      throw new Error('Event name cannot be empty')
+    }
+
+    const hexPayload = to_hex(payload)
+    await send_rpc_request(
+      this.endpoint,
+      '/EmitEvent',
+      JSON.stringify({
+        event: event,
+        payload: hexPayload
+      })
+    )
+  }
 }

sdk/python/src/dstack_sdk/dstack_client.py

@@ -203,6 +203,33 @@ def info(self) -> InfoResponse:
         result = self._send_rpc_request("/Info", {})
         return InfoResponse.model_validate(result)
 
+    def emit_event(
+        self,
+        event: str,
+        payload: str | bytes,
+    ) -> None:
+        """
+        Emit an event. This extends the event to RTMR3 on TDX platform.
+
+        Requires Dstack OS 0.5.0 or later.
+
+        Args:
+            event: The event name
+            payload: The event data as string or bytes
+
+        Returns:
+            None
+        """
+        if not event:
+            raise ValueError("event name cannot be empty")
+
+        if isinstance(payload, str):
+            payload = payload.encode()
+
+        hex_payload = binascii.hexlify(payload).decode()
+        self._send_rpc_request("/EmitEvent", {"event": event, "payload": hex_payload})
+        return None
+
     def get_tls_key(
         self,
         subject: str | None = None,
@@ -285,6 +312,33 @@ async def info(self) -> InfoResponse:
         result = await self._send_rpc_request("/Info", {})
         return InfoResponse.model_validate(result)
 
+    async def emit_event(
+        self,
+        event: str,
+        payload: str | bytes,
+    ) -> None:
+        """
+        Emit an event. This extends the event to RTMR3 on TDX platform.
+
+        Requires Dstack OS 0.5.0 or later.
+
+        Args:
+            event: The event name
+            payload: The event data as string or bytes
+
+        Returns:
+            None
+        """
+        if not event:
+            raise ValueError("event name cannot be empty")
+
+        if isinstance(payload, str):
+            payload = payload.encode()
+
+        hex_payload = binascii.hexlify(payload).decode()
+        await self._send_rpc_request("/EmitEvent", {"event": event, "payload": hex_payload})
+        return None
+
     async def get_tls_key(
         self,
         subject: str | None = None,