virtio-device: move driver feat hook-up into set dev status

Signed-off-by: João Peixoto <[email protected]>

Author: joaopeixoto13

virtio-device/src/virtio_config.rs

@@ -10,6 +10,7 @@ use std::sync::Arc;
 
 use log::error;
 
+use crate::status::{ACKNOWLEDGE, DRIVER, FEATURES_OK};
 use crate::{VirtioDevice, WithDriverSelect};
 use virtio_queue::{Queue, QueueT};
 
@@ -183,10 +184,6 @@ where
             // Accessing an unknown page has no effect.
             _ => features,
         };
-
-        if page == 0 {
-            <Self as VirtioDeviceActions>::negotiate_driver_features(self);
-        }
     }
 
     fn device_status(&self) -> u8 {
@@ -195,6 +192,9 @@ where
 
     fn set_device_status(&mut self, status: u8) {
         self.borrow_mut().device_status = status;
+        if self.borrow_mut().device_status == (ACKNOWLEDGE | DRIVER | FEATURES_OK) {
+            <Self as VirtioDeviceActions>::negotiate_driver_features(self);
+        }
     }
 
     fn activate(&mut self) -> Result<(), Self::E> {