From 522e32219cd4c818dbc04628b002b810a9a22e26 Mon Sep 17 00:00:00 2001
From: Bo Chen <chen.bo@intel.com>
Date: Mon, 12 Sep 2022 14:48:48 -0700
Subject: [PATCH] virtio-devices: pmem: Avoid panic and propagate errors
 properly

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

diff --git a/virtio-devices/src/pmem.rs b/virtio-devices/src/pmem.rs
index 8589ea327..fa8c15d01 100644
--- a/virtio-devices/src/pmem.rs
+++ b/virtio-devices/src/pmem.rs
@@ -90,6 +90,8 @@ enum Error {
     BufferLengthTooSmall,
     #[error("Invalid request")]
     InvalidRequest,
+    #[error("Failed adding used index: {0}")]
+    QueueAddUsed(virtio_queue::Error),
 }
 
 #[derive(Debug, PartialEq, Eq)]
@@ -197,7 +199,7 @@ impl PmemEpollHandler {
 
             self.queue
                 .add_used(desc_chain.memory(), desc_chain.head_index(), len)
-                .unwrap();
+                .map_err(Error::QueueAddUsed)?;
             used_descs = true;
         }