use core::sync::atomic::fence

kernel/src/arch/aarch64/cpu.rs

@@ -6,7 +6,3 @@ pub fn id() -> usize {
     // TODO: cpu id
     0
 }
-
-pub fn fence() {
-    unsafe { asm!("dmb ish" ::: "memory"); }
-}

kernel/src/arch/riscv32/cpu.rs

@@ -35,7 +35,3 @@ pub unsafe fn start_others(hart_mask: usize) {
 pub fn halt() {
     unsafe { riscv::asm::wfi() }
 }
-
-pub fn fence() {
-    unsafe { asm!("fence" ::: "memory"); }
-}

kernel/src/arch/x86_64/cpu.rs

@@ -30,7 +30,3 @@ pub fn halt() {
     use x86_64::instructions::hlt;
     hlt();
 }
-
-pub fn fence() {
-    unsafe { asm!("mfence" ::: "memory"); }
-}

kernel/src/drivers/net/virtio_net.rs

@@ -4,6 +4,7 @@ use alloc::prelude::*;
 use alloc::sync::Arc;
 use core::mem::size_of;
 use core::slice;
+use core::sync::atomic::{fence, Ordering};
 
 use bitflags::*;
 use device_tree::Node;
@@ -215,7 +216,7 @@ impl phy::TxToken for VirtIONetTxToken {
         desc.len.write((len + size_of::<VirtIONetHeader>()) as u32);
 
         // memory barrier
-        crate::arch::cpu::fence();
+        fence(Ordering::SeqCst);
         
         // add desc to available ring
         ring.idx.write(ring.idx.read() + 1);
@@ -376,7 +377,7 @@ pub fn virtio_net_init(node: &Node) {
             desc.flags.write(0);
         }
         // memory barrier
-        crate::arch::cpu::fence();
+        fence(Ordering::SeqCst);
 
 
         if queue == VIRTIO_QUEUE_RECEIVE {