Rename get_eth_key to get_key

Author: kvinwang

guest-agent/rpc/proto/agent_rpc.proto

@@ -11,7 +11,7 @@ service Tappd {
   rpc DeriveKey(GetTlsKeyArgs) returns (GetTlsKeyResponse) {}
 
   // Derives a new ECDSA key with k256 EC curve.
-  rpc DeriveK256Key(GetEthKeyArgs) returns (DeriveK256KeyResponse) {}
+  rpc DeriveK256Key(GetKeyArgs) returns (DeriveK256KeyResponse) {}
 
   // Generates a TDX quote
   rpc TdxQuote(TdxQuoteArgs) returns (TdxQuoteResponse) {}
@@ -34,7 +34,7 @@ service DstackGuest {
   rpc GetTlsKey(GetTlsKeyArgs) returns (GetTlsKeyResponse) {}
 
   // Derives a new ECDSA key with k256 EC curve.
-  rpc GetEthKey(GetEthKeyArgs) returns (GetEthKeyResponse) {}
+  rpc GetKey(GetKeyArgs) returns (GetKeyResponse) {}
 
   // Generates a TDX quote with given report data.
   rpc GetQuote(RawQuoteArgs) returns (GetQuoteResponse) {}
@@ -72,7 +72,7 @@ message GetTlsKeyResponse {
 }
 
 // The request to derive a new ECDSA key with k256 EC curve
-message GetEthKeyArgs {
+message GetKeyArgs {
   // Path to the key to derive
   string path = 1;
   // Purpose of the key
@@ -88,7 +88,7 @@ message DeriveK256KeyResponse {
 }
 
 // The response to a GetEthKey request
-message GetEthKeyResponse {
+message GetKeyResponse {
   // Derived k256 key
   bytes key = 1;
   // Derived k256 signature chain

guest-agent/src/rpc_service.rs

@@ -6,7 +6,7 @@ use dstack_guest_agent_rpc::{
     dstack_guest_server::{DstackGuestRpc, DstackGuestServer},
     tappd_server::{TappdRpc, TappdServer},
     worker_server::{WorkerRpc, WorkerServer},
-    DeriveK256KeyResponse, GetEthKeyArgs, GetEthKeyResponse, GetQuoteResponse, GetTlsKeyArgs,
+    DeriveK256KeyResponse, GetKeyArgs, GetKeyResponse, GetQuoteResponse, GetTlsKeyArgs,
     GetTlsKeyResponse, RawQuoteArgs, TdxQuoteArgs, TdxQuoteResponse, WorkerInfo, WorkerVersion,
 };
 use dstack_types::AppKeys;
@@ -114,7 +114,7 @@ impl DstackGuestRpc for InternalRpcHandler {
         })
     }
 
-    async fn get_eth_key(self, request: GetEthKeyArgs) -> Result<GetEthKeyResponse> {
+    async fn get_key(self, request: GetKeyArgs) -> Result<GetKeyResponse> {
         let k256_app_key = &self.state.inner.keys.k256_key;
         let derived_k256_key = derive_ecdsa_key(k256_app_key, &[request.path.as_bytes()], 32)
             .context("Failed to derive k256 key")?;
@@ -131,7 +131,7 @@ impl DstackGuestRpc for InternalRpcHandler {
         let mut signature = signature.to_vec();
         signature.push(recid.to_byte());
 
-        Ok(GetEthKeyResponse {
+        Ok(GetKeyResponse {
             key: derived_k256_key.to_bytes().to_vec(),
             signature_chain: vec![signature, self.state.inner.keys.k256_signature.clone()],
         })
@@ -185,9 +185,9 @@ impl TappdRpc for InternalRpcHandlerV0 {
             .await
     }
 
-    async fn derive_k256_key(self, request: GetEthKeyArgs) -> Result<DeriveK256KeyResponse> {
+    async fn derive_k256_key(self, request: GetKeyArgs) -> Result<DeriveK256KeyResponse> {
         let res = InternalRpcHandler { state: self.state }
-            .get_eth_key(request)
+            .get_key(request)
             .await?;
         Ok(DeriveK256KeyResponse {
             k256_key: res.key,