Merge pull request #398 from Dstack-TEE/vmm-update-urls

vmm-rpc: Add support for update kms/gw urls

Author: kvinwang

vmm/rpc/proto/vmm_rpc.proto

@@ -131,6 +131,12 @@ message UpdateVmRequest {
   bool update_ports = 5;
   // Port mapping
   repeated PortMapping ports = 7;
+  // Optional update KMS URLs.
+  bool update_kms_urls = 8;
+  repeated string kms_urls = 9;
+  // Optional update gateway URLs.
+  bool update_gateway_urls = 10;
+  repeated string gateway_urls = 11;
   // gpus
   GpuConfig gpus = 13;
   // Optional compute resource updates. Leave unset to keep current configuration.

vmm/src/main_service.rs

@@ -398,6 +398,12 @@ impl VmmRpc for RpcHandler {
                 })
                 .collect::<Result<Vec<_>>>()?;
         }
+        if request.update_kms_urls {
+            manifest.kms_urls = request.kms_urls.clone();
+        }
+        if request.update_gateway_urls {
+            manifest.gateway_urls = request.gateway_urls.clone();
+        }
         vm_work_dir
             .put_manifest(&manifest)
             .context("Failed to put manifest")?;