vmm: api: Add a VMM shutdown command

This shuts the current VM down, if any, and then exits the VMM process.

Signed-off-by: Samuel Ortiz <sameo@linux.intel.com>

vmm/src/api/http.rs

@@ -5,7 +5,7 @@
 
 extern crate threadpool;
 
-use crate::api::http_endpoint::{VmActionHandler, VmCreate, VmInfo};
+use crate::api::http_endpoint::{VmActionHandler, VmCreate, VmInfo, VmmShutdown};
 use crate::api::{ApiRequest, VmAction};
 use crate::{Error, Result};
 use micro_http::{HttpConnection, Request, Response, StatusCode, Version};
@@ -60,6 +60,7 @@ lazy_static! {
         r.routes.insert(endpoint!("/vm.info"), Box::new(VmInfo {}));
         r.routes.insert(endpoint!("/vm.shutdown"), Box::new(VmActionHandler::new(VmAction::Shutdown)));
         r.routes.insert(endpoint!("/vm.reboot"), Box::new(VmActionHandler::new(VmAction::Reboot)));
+        r.routes.insert(endpoint!("/vmm.shutdown"), Box::new(VmmShutdown {}));
 
         r
     };

vmm/src/api/http_endpoint.rs

@@ -5,8 +5,8 @@
 
 use crate::api::http::EndpointHandler;
 use crate::api::{
-    vm_boot, vm_create, vm_delete, vm_info, vm_reboot, vm_shutdown, ApiError, ApiRequest,
-    ApiResult, VmAction, VmConfig,
+    vm_boot, vm_create, vm_delete, vm_info, vm_reboot, vm_shutdown, vmm_shutdown, ApiError,
+    ApiRequest, ApiResult, VmAction, VmConfig,
 };
 use micro_http::{Body, Method, Request, Response, StatusCode, Version};
 use serde_json::Error as SerdeError;
@@ -37,6 +37,9 @@ pub enum HttpError {
 
     /// Could not act on a VM
     VmAction(ApiError),
+
+    /// Could not shut the VMM down
+    VmmShutdown(ApiError),
 }
 
 fn error_response(error: HttpError, status: StatusCode) -> Response {
@@ -155,3 +158,25 @@ impl EndpointHandler for VmInfo {
         }
     }
 }
+
+// /api/v1/vmm.shutdown handler
+pub struct VmmShutdown {}
+
+impl EndpointHandler for VmmShutdown {
+    fn handle_request(
+        &self,
+        req: &Request,
+        api_notifier: EventFd,
+        api_sender: Sender<ApiRequest>,
+    ) -> Response {
+        match req.method() {
+            Method::Put => {
+                match vmm_shutdown(api_notifier, api_sender).map_err(HttpError::VmmShutdown) {
+                    Ok(_) => Response::new(Version::Http11, StatusCode::OK),
+                    Err(e) => error_response(e, StatusCode::InternalServerError),
+                }
+            }
+            _ => Response::new(Version::Http11, StatusCode::BadRequest),
+        }
+    }
+}

vmm/src/api/mod.rs

@@ -87,6 +87,9 @@ pub enum ApiError {
 
     /// The VM could not reboot.
     VmReboot(VmError),
+
+    /// The VMM could not shutdown.
+    VmmShutdown(VmError),
 }
 pub type ApiResult<T> = std::result::Result<T, ApiError>;
 
@@ -138,6 +141,11 @@ pub enum ApiRequest {
     /// If the VM was not previously booted or created, the VMM API server
     /// will send a VmReboot error back.
     VmReboot(Sender<ApiResponse>),
+
+    /// Shut the VMM down.
+    /// This will shutdown and delete the current VM, if any, and then exit the
+    /// VMM process.
+    VmmShutdown(Sender<ApiResponse>),
 }
 
 pub fn vm_create(
@@ -226,3 +234,17 @@ pub fn vm_info(api_evt: EventFd, api_sender: Sender<ApiRequest>) -> ApiResult<Vm
         _ => Err(ApiError::ResponsePayloadType),
     }
 }
+
+pub fn vmm_shutdown(api_evt: EventFd, api_sender: Sender<ApiRequest>) -> ApiResult<()> {
+    let (response_sender, response_receiver) = channel();
+
+    // Send the VMM shutdown request.
+    api_sender
+        .send(ApiRequest::VmmShutdown(response_sender))
+        .map_err(ApiError::RequestSend)?;
+    api_evt.write(1).map_err(ApiError::EventFdWrite)?;
+
+    response_receiver.recv().map_err(ApiError::ResponseRecv)??;
+
+    Ok(())
+}

vmm/src/api/openapi/cloud-hypervisor.yaml

@@ -23,6 +23,15 @@ paths:
               schema:
                 $ref: '#/components/schemas/VmmInfo'
 
+  /vmm.shutdown:
+    put:
+      summary: Shuts the cloud-hypervisor VMM.
+      operationId: shutdownVMM
+      responses:
+        201:
+          description: The VMM successfully shutdown.
+          content: {}
+
   /vm.info:
     get:
       summary: Returns general information about the cloud-hypervisor Virtual Machine (VM) instance.