vmm: Add support for init stopped VMs

Author: kvinwang

vmm/rpc/proto/vmm_rpc.proto

@@ -77,6 +77,8 @@ message VmConfiguration {
   repeated string kms_urls = 14;
   // Gateway URLs
   repeated string gateway_urls = 15;
+  // The VM is stopped
+  bool stopped = 16;
 }
 
 message GpuConfig {

vmm/src/app/qemu.rs

@@ -113,6 +113,7 @@ impl VmInfo {
                     .as_ref()
                     .map(|c| c.gateway_urls.clone())
                     .unwrap_or_default();
+                let stopped = !workdir.started().unwrap_or(false);
 
                 Some(pb::VmConfiguration {
                     name: self.manifest.name.clone(),
@@ -153,6 +154,7 @@ impl VmInfo {
                     }),
                     kms_urls,
                     gateway_urls,
+                    stopped,
                 })
             },
             app_url: self

vmm/src/main_service.rs

@@ -185,7 +185,7 @@ impl VmmRpc for RpcHandler {
             .put_manifest(&manifest)
             .context("Failed to write manifest")?;
         let work_dir = self.prepare_work_dir(&id, &request, &app_id)?;
-        if let Err(err) = vm_work_dir.set_started(true) {
+        if let Err(err) = vm_work_dir.set_started(!request.stopped) {
             warn!("Failed to set started: {}", err);
         }
 
@@ -195,7 +195,13 @@ impl VmmRpc for RpcHandler {
             .await
             .context("Failed to load VM");
         let result = match result {
-            Ok(()) => self.app.start_vm(&id).await,
+            Ok(()) => {
+                if !request.stopped {
+                    self.app.start_vm(&id).await
+                } else {
+                    Ok(())
+                }
+            }
             Err(err) => Err(err),
         };
         if let Err(err) = result {

vmm/src/vmm-cli.py

@@ -504,6 +504,7 @@ def create_vm(self, args) -> None:
             "ports": [parse_port_mapping(port) for port in args.port or []],
             "hugepages": args.hugepages,
             "pin_numa": args.pin_numa,
+            "stopped": args.stopped,
         }
 
         if args.ppcie:
@@ -899,6 +900,8 @@ def main():
                                help='KMS URL')
     deploy_parser.add_argument('--gateway-url', action='append', type=str,
                                help='Gateway URL')
+    deploy_parser.add_argument('--stopped', action='store_true',
+                               help='Create VM in stopped state (requires dstack-vmm >= 0.5.4)')
 
     # Images command
     _images_parser = subparsers.add_parser(