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
|
// SPDX-License-Identifier: GPL-2.0
//! Nova-core auxbus data. Contains all the methods used by the auxbus drivers
//! to interact with nova-core.
use core::pin::Pin;
use kernel::{
auxiliary,
device::Bound,
pci,
prelude::*,
types::ForLt, //
};
pub use crate::gpu::Spec;
use crate::{
driver,
gpu::Gpu,
gsp::commands::GetGspStaticInfoReply, //
};
/// API handle for the auxiliary bus child drivers to interact with nova-core.
pub struct NovaCoreApi<'bound> {
pub(crate) gpu: Pin<&'bound Gpu<'bound>>,
pub(crate) pdev: &'bound pci::Device<Bound>,
}
impl NovaCoreApi<'_> {
/// Obtain a [`NovaCoreApi`] handle from an auxiliary device registered
/// by nova-core.
pub fn of(adev: &auxiliary::Device<Bound>) -> Result<NovaCoreApiHandle<'_>> {
NovaCoreApiHandle::of(adev)
}
/// Returns the GPU [`GetGspStaticInfoReply`].
pub fn gsp_static_info(&self) -> &GetGspStaticInfoReply {
&self.gpu.gsp_static_info
}
/// Returns the GPU [`Spec`].
pub fn spec(&self) -> &Spec {
&self.gpu.spec
}
/// Returns the size of the PCIe BAR used for accessing VRAM, typically
/// BAR1.
pub fn bar1_size(&self) -> Result<u64> {
let bar1 = driver::bar1_resource_index(self.pdev)?;
self.pdev.resource_len(bar1)
}
}
/// Expose a handle to nova-core API
pub struct NovaCoreApiHandle<'a> {
adev: &'a auxiliary::Device<Bound>,
}
impl<'a> NovaCoreApiHandle<'a> {
fn of(adev: &'a auxiliary::Device<Bound>) -> Result<Self> {
adev.registration_data_with::<ForLt!(NovaCoreApi<'_>), ()>(|_| ())?;
Ok(Self { adev })
}
/// Access the [`NovaCoreApi`] through a closure.
pub fn with<R>(&self, f: impl for<'b> FnOnce(Pin<&'a NovaCoreApi<'b>>) -> R) -> R {
self.adev
.registration_data_with::<ForLt!(NovaCoreApi<'_>), R>(f)
.expect("TypeId was validated in NovaCoreApiHandle::of()")
}
}
|
