Remove invalid loops

Fix #104 Signed-off-by: Wei Liu <[email protected]>
intel · Jan 4, 2024 · 3acbf6c · 3acbf6c
1 parent 4fe0628
commit 3acbf6c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/devices/vsock/src/transport/vmcall.rs b/src/devices/vsock/src/transport/vmcall.rs
@@ -137,7 +137,7 @@ impl VmcallVsock {
         tdx::tdvmcall_service(command, response, VMCALL_VECTOR as u64, timeout)
             .map_err(|e| VsockTransportError::Vmcall(e))?;
 
-        while !wait_for_event(&VMCALL_FLAG, self.timer.as_ref()) {
+        if !wait_for_event(&VMCALL_FLAG, self.timer.as_ref()) {
             return Err(VsockTransportError::Timeout);
         }
         self.timer.reset_timeout();
@@ -176,7 +176,7 @@ impl VmcallVsock {
         // TO DO:
         // Refactor the waiting logic
         loop {
-            while !wait_for_event(&VMCALL_FLAG, self.timer.as_ref()) {
+            if !wait_for_event(&VMCALL_FLAG, self.timer.as_ref()) {
                 return Err(VsockTransportError::Timeout);
             }
             self.timer.reset_timeout();