From 0235ed3388ca31809cc49536dfdd5fafd7071688 Mon Sep 17 00:00:00 2001
From: Bo Chen <chen.bo@intel.com>
Date: Thu, 13 Oct 2022 11:43:18 -0700
Subject: [PATCH] virtio-devices: mem: Report error instead of panic

Signed-off-by: Bo Chen <chen.bo@intel.com>
---
 virtio-devices/src/mem.rs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/virtio-devices/src/mem.rs b/virtio-devices/src/mem.rs
index 2aeab2165..ce2536027 100644
--- a/virtio-devices/src/mem.rs
+++ b/virtio-devices/src/mem.rs
@@ -139,6 +139,8 @@ pub enum Error {
     NotActivatedByGuest,
     #[error("Unknown request type: {0}")]
     UnkownRequestType(u16),
+    #[error("Failed adding used index: {0}")]
+    QueueAddUsed(virtio_queue::Error),
 }
 
 #[repr(C)]
@@ -614,7 +616,7 @@ impl MemEpollHandler {
             let len = r.send_response(desc_chain.memory(), resp_type, resp_state)?;
             self.queue
                 .add_used(desc_chain.memory(), desc_chain.head_index(), len)
-                .unwrap();
+                .map_err(Error::QueueAddUsed)?;
             used_descs = true;
         }