1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
// -*- coding: utf-8 -*-
//
// disktest - Hard drive tester
//
// Copyright 2020-2023 Michael Buesch <m@bues.ch>
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License along
// with this program; if not, write to the Free Software Foundation, Inc.,
// 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
//
use crate::disktest::DisktestQuiet;
use std::collections::HashMap;
use std::sync::mpsc::{channel, Receiver, Sender};
pub struct BufCache {
snd: HashMap<u32, Sender<Vec<u8>>>,
quiet_level: DisktestQuiet,
}
impl BufCache {
pub fn new(quiet_level: DisktestQuiet) -> BufCache {
BufCache {
snd: HashMap::new(),
quiet_level,
}
}
pub fn new_consumer(&mut self, cons_id: u32) -> BufCacheCons {
let (snd, rcv) = channel();
self.snd.insert(cons_id, snd);
BufCacheCons { rcv }
}
pub fn push(&mut self, cons_id: u32, buf: Vec<u8>) {
let Some(snd) = self.snd.get(&cons_id) else {
panic!("BufCache: Consumer {} does not exist.", cons_id);
};
if let Err(e) = snd.send(buf) {
if self.quiet_level < DisktestQuiet::NoWarn {
eprintln!("BufCache: Failed to send: {}", e);
}
}
}
}
pub struct BufCacheCons {
rcv: Receiver<Vec<u8>>,
}
impl BufCacheCons {
pub fn pull(&mut self, buf_len: usize) -> Vec<u8> {
let mut buf = match self.rcv.try_recv() {
Ok(buf) => buf,
Err(_) => Vec::with_capacity(buf_len),
};
if buf.len() != buf_len {
buf.resize(buf_len, 0);
}
buf
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_bufcache() {
let mut cache = BufCache::new(DisktestQuiet::Normal);
let mut cons0 = cache.new_consumer(42);
let mut cons1 = cache.new_consumer(43);
let buf = cons0.pull(4);
assert_eq!(buf.len(), 4);
assert_eq!(buf, vec![0, 0, 0, 0]);
cache.push(42, vec![0xDE, 0xAD, 0xBE, 0xEF]);
let buf = cons0.pull(4);
assert_eq!(buf.len(), 4);
assert_eq!(buf, vec![0xDE, 0xAD, 0xBE, 0xEF]);
let buf = cons0.pull(4);
assert_eq!(buf.len(), 4);
assert_eq!(buf, vec![0, 0, 0, 0]);
cache.push(43, vec![0xCA, 0xFE, 0xAF, 0xFE]);
let buf = cons0.pull(4);
assert_eq!(buf.len(), 4);
assert_eq!(buf, vec![0, 0, 0, 0]);
let buf = cons1.pull(4);
assert_eq!(buf.len(), 4);
assert_eq!(buf, vec![0xCA, 0xFE, 0xAF, 0xFE]);
}
#[test]
#[should_panic(expected = "Consumer 42 does not exist")]
fn test_bufcache_cons_invalid() {
let mut cache = BufCache::new(DisktestQuiet::Normal);
cache.push(42, vec![]);
}
}
// vim: ts=4 sw=4 expandtab
|
