From 6922e25e786e32dbba79b7d8e8ec4930578cd01d Mon Sep 17 00:00:00 2001
From: Bo Chen <chen.bo@intel.com>
Date: Thu, 21 Mar 2024 10:14:45 -0700
Subject: [PATCH] vmm: Move VM shutdown event to Vmm::vm_shutdown

Signed-off-by: Bo Chen <chen.bo@intel.com>
---
 vmm/src/lib.rs | 8 +++++++-
 vmm/src/vm.rs  | 2 --
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/vmm/src/lib.rs b/vmm/src/lib.rs
index 702cc4da6..60b4feeda 100644
--- a/vmm/src/lib.rs
+++ b/vmm/src/lib.rs
@@ -1381,11 +1381,17 @@ impl RequestHandler for Vmm {
     }
 
     fn vm_shutdown(&mut self) -> result::Result<(), VmError> {
-        if let Some(ref mut vm) = self.vm.take() {
+        let r = if let Some(ref mut vm) = self.vm.take() {
             vm.shutdown()
         } else {
             Err(VmError::VmNotRunning)
+        };
+
+        if r.is_ok() {
+            event!("vm", "shutdown");
         }
+
+        r
     }
 
     fn vm_reboot(&mut self) -> result::Result<(), VmError> {
diff --git a/vmm/src/vm.rs b/vmm/src/vm.rs
index c61cfa643..80a96ea68 100644
--- a/vmm/src/vm.rs
+++ b/vmm/src/vm.rs
@@ -1397,8 +1397,6 @@ impl Vm {
         }
         *state = new_state;
 
-        event!("vm", "shutdown");
-
         Ok(())
     }