Add device_mut method to DeviceWrapper

To prepare the mutability refactoring, we add a device_mut method to
DeviceWrapper that can be used to obtain a mutable reference to the
wrapped device.

1 files changed, 7 insertions, 0 deletions

diff --git a/src/device.rs b/src/device.rs
index c985802..462e9dc 100644
--- a/src/device.rs
+++ b/src/device.rs
@@ -728,6 +728,13 @@ impl DeviceWrapper {
             DeviceWrapper::Pro(ref pro) => pro,
         }
     }
+
+    fn device_mut(&mut self) -> &mut dyn Device {
+        match *self {
+            DeviceWrapper::Storage(ref mut storage) => storage,
+            DeviceWrapper::Pro(ref mut pro) => pro,
+        }
+    }
 }
 
 impl From<Pro> for DeviceWrapper {