Fix examples.

Author: qwandor

examples/aarch64/src/main.rs

@@ -46,6 +46,7 @@ use virtio_drivers::{
         },
         DeviceType, Transport,
     },
+    UniqueMmioPointer,
 };
 
 /// Base memory-mapped address of the primary PL011 UART device.
@@ -142,7 +143,9 @@ fn main(x0: u64, x1: u64, x2: u64, x3: u64) -> ! {
                 debug!("Found VirtIO MMIO device at {:?}", region);
 
                 let header = NonNull::new(region.starting_address as *mut VirtIOHeader).unwrap();
-                match unsafe { MmioTransport::new(header, region.size.unwrap()) } {
+                match unsafe {
+                    MmioTransport::new(UniqueMmioPointer::new(header), region.size.unwrap())
+                } {
                     Err(e) => warn!("Error creating VirtIO MMIO transport: {}", e),
                     Ok(transport) => {
                         info!(

examples/riscv/src/main.rs

@@ -23,6 +23,7 @@ use virtio_drivers::{
         mmio::{MmioTransport, VirtIOHeader},
         DeviceType, Transport,
     },
+    UniqueMmioPointer,
 };
 use virtio_impl::HalImpl;
 
@@ -69,7 +70,7 @@ fn virtio_probe(node: FdtNode) {
             node.compatible().map(Compatible::first),
         );
         let header = NonNull::new(vaddr as *mut VirtIOHeader).unwrap();
-        match unsafe { MmioTransport::new(header, size) } {
+        match unsafe { MmioTransport::new(UniqueMmioPointer::new(header), size) } {
             Err(e) => warn!("Error creating VirtIO MMIO transport: {}", e),
             Ok(transport) => {
                 info!(