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
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
|
// SPDX-License-Identifier: GPL-2.0
//! Rust miscellaneous device sample.
use kernel::prelude::*;
use kernel::{
file::{self, File},
io_buffer::{IoBufferReader, IoBufferWriter},
miscdev,
sync::{Arc, ArcBorrow, CondVar, Mutex, UniqueArc},
};
module! {
type: RustMiscdev,
name: "rust_miscdev",
author: "Rust for Linux Contributors",
description: "Rust miscellaneous device sample",
license: "GPL",
}
const MAX_TOKENS: usize = 3;
struct SharedStateInner {
token_count: usize,
}
struct SharedState {
state_changed: CondVar,
inner: Mutex<SharedStateInner>,
}
impl SharedState {
fn try_new() -> Result<Arc<Self>> {
let mut state = Pin::from(UniqueArc::try_new(Self {
// SAFETY: `condvar_init!` is called below.
state_changed: unsafe { CondVar::new() },
// SAFETY: `mutex_init!` is called below.
inner: unsafe { Mutex::new(SharedStateInner { token_count: 0 }) },
})?);
// SAFETY: `state_changed` is pinned when `state` is.
let pinned = unsafe { state.as_mut().map_unchecked_mut(|s| &mut s.state_changed) };
kernel::condvar_init!(pinned, "SharedState::state_changed");
// SAFETY: `inner` is pinned when `state` is.
let pinned = unsafe { state.as_mut().map_unchecked_mut(|s| &mut s.inner) };
kernel::mutex_init!(pinned, "SharedState::inner");
Ok(state.into())
}
}
struct Token;
#[vtable]
impl file::Operations for Token {
type Data = Arc<SharedState>;
type OpenData = Arc<SharedState>;
fn open(shared: &Arc<SharedState>, _file: &File) -> Result<Self::Data> {
Ok(shared.clone())
}
fn read(
shared: ArcBorrow<'_, SharedState>,
_: &File,
data: &mut impl IoBufferWriter,
offset: u64,
) -> Result<usize> {
// Succeed if the caller doesn't provide a buffer or if not at the start.
if data.is_empty() || offset != 0 {
return Ok(0);
}
{
let mut inner = shared.inner.lock();
// Wait until we are allowed to decrement the token count or a signal arrives.
while inner.token_count == 0 {
if shared.state_changed.wait(&mut inner) {
return Err(EINTR);
}
}
// Consume a token.
inner.token_count -= 1;
}
// Notify a possible writer waiting.
shared.state_changed.notify_all();
// Write a one-byte 1 to the reader.
data.write_slice(&[1u8; 1])?;
Ok(1)
}
fn write(
shared: ArcBorrow<'_, SharedState>,
_: &File,
data: &mut impl IoBufferReader,
_offset: u64,
) -> Result<usize> {
{
let mut inner = shared.inner.lock();
// Wait until we are allowed to increment the token count or a signal arrives.
while inner.token_count == MAX_TOKENS {
if shared.state_changed.wait(&mut inner) {
return Err(EINTR);
}
}
// Increment the number of token so that a reader can be released.
inner.token_count += 1;
}
// Notify a possible reader waiting.
shared.state_changed.notify_all();
Ok(data.len())
}
}
struct RustMiscdev {
_dev: Pin<Box<miscdev::Registration<Token>>>,
}
impl kernel::Module for RustMiscdev {
fn init(name: &'static CStr, _module: &'static ThisModule) -> Result<Self> {
pr_info!("Rust miscellaneous device sample (init)\n");
let state = SharedState::try_new()?;
Ok(RustMiscdev {
_dev: miscdev::Registration::new_pinned(fmt!("{name}"), state)?,
})
}
}
impl Drop for RustMiscdev {
fn drop(&mut self) {
pr_info!("Rust miscellaneous device sample (exit)\n");
}
}
|
