From ec94ae31eec87d82dc8c550c86e3ac3d65530d73 Mon Sep 17 00:00:00 2001
From: Bo Chen <chen.bo@intel.com>
Date: Wed, 23 Nov 2022 11:39:17 -0800
Subject: [PATCH] vmm: EpollContext: Allow to add custom epoll events for
 fuzzing

Signed-off-by: Bo Chen <chen.bo@intel.com>
---
 vmm/src/lib.rs | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/vmm/src/lib.rs b/vmm/src/lib.rs
index aab5a01ff..509fe28b1 100644
--- a/vmm/src/lib.rs
+++ b/vmm/src/lib.rs
@@ -232,6 +232,26 @@ impl EpollContext {
 
         Ok(())
     }
+
+    #[cfg(fuzzing)]
+    pub fn add_event_custom<T>(
+        &mut self,
+        fd: &T,
+        id: u64,
+        evts: epoll::Events,
+    ) -> result::Result<(), io::Error>
+    where
+        T: AsRawFd,
+    {
+        epoll::ctl(
+            self.epoll_file.as_raw_fd(),
+            epoll::ControlOptions::EPOLL_CTL_ADD,
+            fd.as_raw_fd(),
+            epoll::Event::new(evts, id),
+        )?;
+
+        Ok(())
+    }
 }
 
 impl AsRawFd for EpollContext {