net_backend/
tests.rs

1// Copyright (c) Microsoft Corporation.
2// Licensed under the MIT License.
3
4//! Types to help test backends.
5
6use crate::BufferAccess;
7use crate::RxBufferSegment;
8use crate::RxId;
9use crate::RxMetadata;
10use guestmem::GuestMemory;
11use memory_range::MemoryRange;
12use parking_lot::Mutex;
13use std::sync::Arc;
14use vm_topology::memory::MemoryLayout;
15
16pub fn test_layout() -> MemoryLayout {
17    MemoryLayout::new(
18        64 * 4096,
19        &[
20            MemoryRange::new(64 * 4096..65 * 4096),
21            MemoryRange::new(65 * 4096..66 * 4096),
22        ],
23        &[],
24        &[],
25        None,
26    )
27    .unwrap()
28}
29
30#[derive(Clone)]
31pub struct Bufs {
32    inner: Arc<BufsInner>,
33}
34
35struct BufsInner {
36    rx_metadata: Vec<Mutex<Option<RxMetadata>>>,
37    guest_memory: GuestMemory,
38}
39
40impl Bufs {
41    pub fn new(guest_memory: GuestMemory) -> Self {
42        let mut rx_metadata = Vec::new();
43        rx_metadata.resize_with(128, Default::default);
44        Self {
45            inner: Arc::new(BufsInner {
46                rx_metadata,
47                guest_memory,
48            }),
49        }
50    }
51}
52
53impl BufferAccess for Bufs {
54    fn guest_memory(&self) -> &GuestMemory {
55        &self.inner.guest_memory
56    }
57
58    fn push_guest_addresses(&self, id: RxId, buf: &mut Vec<RxBufferSegment>) {
59        let gpa = id.0 as u64 * 2048;
60        buf.push(RxBufferSegment { gpa, len: 2048 });
61    }
62
63    fn capacity(&self, _id: RxId) -> u32 {
64        2048
65    }
66
67    fn write_data(&mut self, id: RxId, buf: &[u8]) {
68        self.inner
69            .guest_memory
70            .write_at(id.0 as u64 * 2048, buf)
71            .unwrap();
72    }
73
74    fn write_header(&mut self, id: RxId, metadata: &RxMetadata) {
75        *self.inner.rx_metadata[id.0 as usize].lock() = Some(*metadata);
76    }
77}
net_backend/tests.rs

net_backend/
tests.rs
