builtins/
irq_resource.rs

1// Copyright 2020 The Fuchsia Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5use anyhow::{format_err, Error};
6use fidl_fuchsia_kernel as fkernel;
7use futures::prelude::*;
8use std::sync::Arc;
9use zx::{self as zx, HandleBased, Resource};
10
11/// An implementation of fuchsia.kernel.IrqResource protocol.
12pub struct IrqResource {
13    resource: Resource,
14}
15
16impl IrqResource {
17    /// `resource` must be the IRQ resource.
18    pub fn new(resource: Resource) -> Arc<Self> {
19        Arc::new(Self { resource })
20    }
21
22    pub async fn serve(
23        self: Arc<Self>,
24        mut stream: fkernel::IrqResourceRequestStream,
25    ) -> Result<(), Error> {
26        if self.resource.info()?.kind != zx::sys::ZX_RSRC_KIND_IRQ {
27            return Err(format_err!("invalid handle kind, expected IRQ"));
28        }
29        while let Some(fkernel::IrqResourceRequest::Get { responder }) = stream.try_next().await? {
30            responder.send(self.resource.duplicate_handle(zx::Rights::SAME_RIGHTS)?)?;
31        }
32        Ok(())
33    }
34}
35
36#[cfg(test)]
37mod tests {
38    use super::*;
39    use fuchsia_component::client::connect_to_protocol;
40    use {fidl_fuchsia_kernel as fkernel, fuchsia_async as fasync};
41
42    async fn get_irq_resource() -> Result<Resource, Error> {
43        let irq_resource_provider = connect_to_protocol::<fkernel::IrqResourceMarker>()?;
44        let irq_resource_handle = irq_resource_provider.get().await?;
45        Ok(Resource::from(irq_resource_handle))
46    }
47
48    async fn serve_irq_resource() -> Result<fkernel::IrqResourceProxy, Error> {
49        let irq_resource = get_irq_resource().await?;
50
51        let (proxy, stream) =
52            fidl::endpoints::create_proxy_and_stream::<fkernel::IrqResourceMarker>();
53        fasync::Task::local(
54            IrqResource::new(irq_resource)
55                .serve(stream)
56                .unwrap_or_else(|e| panic!("Error while serving IRQ resource service: {}", e)),
57        )
58        .detach();
59        Ok(proxy)
60    }
61
62    #[fuchsia::test]
63    async fn fail_with_no_irq_resource() -> Result<(), Error> {
64        let (_, stream) = fidl::endpoints::create_proxy_and_stream::<fkernel::IrqResourceMarker>();
65        IrqResource::new(Resource::from(zx::Handle::invalid()))
66            .serve(stream)
67            .await
68            .expect_err("should fail to serve stream with an invalid resource");
69        Ok(())
70    }
71
72    #[fuchsia::test]
73    async fn kind_type_is_irq() -> Result<(), Error> {
74        let irq_resource_provider = serve_irq_resource().await?;
75        let irq_resource: Resource = irq_resource_provider.get().await?;
76        let resource_info = irq_resource.info()?;
77        assert_eq!(resource_info.kind, zx::sys::ZX_RSRC_KIND_IRQ);
78        assert_eq!(resource_info.base, 0);
79        assert_eq!(resource_info.size, 0);
80        Ok(())
81    }
82}