builtins/
ioport_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.IoportResource protocol.
12pub struct IoportResource {
13    resource: Resource,
14}
15
16impl IoportResource {
17    /// `resource` must be the IOPORT 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::IoportResourceRequestStream,
25    ) -> Result<(), Error> {
26        if self.resource.info()?.kind != zx::sys::ZX_RSRC_KIND_IOPORT {
27            return Err(format_err!("invalid handle kind, expected IOPORT"));
28        }
29        while let Some(fkernel::IoportResourceRequest::Get { responder }) =
30            stream.try_next().await?
31        {
32            responder.send(self.resource.duplicate_handle(zx::Rights::SAME_RIGHTS)?)?;
33        }
34        Ok(())
35    }
36}
37
38#[cfg(test)]
39mod tests {
40    use super::*;
41    use fuchsia_component::client::connect_to_protocol;
42    use {fidl_fuchsia_kernel as fkernel, fuchsia_async as fasync};
43
44    async fn get_ioport_resource() -> Result<Resource, Error> {
45        let ioport_resource_provider = connect_to_protocol::<fkernel::IoportResourceMarker>()?;
46        let ioport_resource_handle = ioport_resource_provider.get().await?;
47        Ok(Resource::from(ioport_resource_handle))
48    }
49
50    async fn serve_ioport_resource() -> Result<fkernel::IoportResourceProxy, Error> {
51        let ioport_resource = get_ioport_resource().await?;
52
53        let (proxy, stream) =
54            fidl::endpoints::create_proxy_and_stream::<fkernel::IoportResourceMarker>();
55        fasync::Task::local(
56            IoportResource::new(ioport_resource)
57                .serve(stream)
58                .unwrap_or_else(|e| panic!("Error while serving IOPORT resource service: {}", e)),
59        )
60        .detach();
61        Ok(proxy)
62    }
63
64    #[fuchsia::test]
65    async fn fail_with_no_ioport_resource() -> Result<(), Error> {
66        let (_, stream) =
67            fidl::endpoints::create_proxy_and_stream::<fkernel::IoportResourceMarker>();
68        IoportResource::new(Resource::from(zx::Handle::invalid()))
69            .serve(stream)
70            .await
71            .expect_err("should fail to serve stream with an invalid resource");
72        Ok(())
73    }
74
75    #[fuchsia::test]
76    async fn kind_type_is_ioport() -> Result<(), Error> {
77        let ioport_resource_provider = serve_ioport_resource().await?;
78        let ioport_resource: Resource = ioport_resource_provider.get().await?;
79        let resource_info = ioport_resource.info()?;
80        assert_eq!(resource_info.kind, zx::sys::ZX_RSRC_KIND_IOPORT);
81        assert_eq!(resource_info.base, 0);
82        assert_eq!(resource_info.size, 0);
83        Ok(())
84    }
85}