fidl_test_componentmanager_stresstests/

fidl_test_componentmanager_stresstests.rs

// WARNING: This file is machine generated by fidlgen.

#![warn(clippy::all)]
#![allow(unused_parens, unused_mut, unused_imports, nonstandard_style)]

use bitflags::bitflags;
use fidl::client::QueryResponseFut;
use fidl::encoding::{MessageBufFor, ProxyChannelBox, ResourceDialect};
use fidl::endpoints::{ControlHandle as _, Responder as _};
pub use fidl_test_componentmanager_stresstests_common::*;
use futures::future::{self, MaybeDone, TryFutureExt};
use zx_status;

#[derive(Debug, Copy, Clone, Eq, PartialEq, Ord, PartialOrd, Hash)]
pub struct ChildRealmMarker;

impl fidl::endpoints::ProtocolMarker for ChildRealmMarker {
    type Proxy = ChildRealmProxy;
    type RequestStream = ChildRealmRequestStream;
    #[cfg(target_os = "fuchsia")]
    type SynchronousProxy = ChildRealmSynchronousProxy;

    const DEBUG_NAME: &'static str = "test.componentmanager.stresstests.ChildRealm";
}
impl fidl::endpoints::DiscoverableProtocolMarker for ChildRealmMarker {}

pub trait ChildRealmProxyInterface: Send + Sync {
    fn r#stop(&self) -> Result<(), fidl::Error>;
    type StopChildrenResponseFut: std::future::Future<Output = Result<(), fidl::Error>> + Send;
    fn r#stop_children(&self) -> Self::StopChildrenResponseFut;
    type CreateChildrenResponseFut: std::future::Future<Output = Result<(), fidl::Error>> + Send;
    fn r#create_children(
        &self,
        direct_children: u16,
        tree_height: u16,
    ) -> Self::CreateChildrenResponseFut;
}
#[derive(Debug)]
#[cfg(target_os = "fuchsia")]
pub struct ChildRealmSynchronousProxy {
    client: fidl::client::sync::Client,
}

#[cfg(target_os = "fuchsia")]
impl fidl::endpoints::SynchronousProxy for ChildRealmSynchronousProxy {
    type Proxy = ChildRealmProxy;
    type Protocol = ChildRealmMarker;

    fn from_channel(inner: fidl::Channel) -> Self {
        Self::new(inner)
    }

    fn into_channel(self) -> fidl::Channel {
        self.client.into_channel()
    }

    fn as_channel(&self) -> &fidl::Channel {
        self.client.as_channel()
    }
}

#[cfg(target_os = "fuchsia")]
impl ChildRealmSynchronousProxy {
    pub fn new(channel: fidl::Channel) -> Self {
        let protocol_name = <ChildRealmMarker as fidl::endpoints::ProtocolMarker>::DEBUG_NAME;
        Self { client: fidl::client::sync::Client::new(channel, protocol_name) }
    }

    pub fn into_channel(self) -> fidl::Channel {
        self.client.into_channel()
    }

    /// Waits until an event arrives and returns it. It is safe for other
    /// threads to make concurrent requests while waiting for an event.
    pub fn wait_for_event(
        &self,
        deadline: zx::MonotonicInstant,
    ) -> Result<ChildRealmEvent, fidl::Error> {
        ChildRealmEvent::decode(self.client.wait_for_event(deadline)?)
    }

    /// Component will gracefully exit.
    pub fn r#stop(&self) -> Result<(), fidl::Error> {
        self.client.send::<fidl::encoding::EmptyPayload>(
            (),
            0x36c34aa66f2a56ab,
            fidl::encoding::DynamicFlags::empty(),
        )
    }

    pub fn r#stop_children(&self, ___deadline: zx::MonotonicInstant) -> Result<(), fidl::Error> {
        let _response =
            self.client.send_query::<fidl::encoding::EmptyPayload, fidl::encoding::EmptyPayload>(
                (),
                0x61c8762c3aa05df7,
                fidl::encoding::DynamicFlags::empty(),
                ___deadline,
            )?;
        Ok(_response)
    }

    /// Create nested children in this realm.
    /// `direct_children` - Num of direct children of this realm.
    /// `tree_height` - Height of the realm tree to create.
    /// Returns once all children are running.
    pub fn r#create_children(
        &self,
        mut direct_children: u16,
        mut tree_height: u16,
        ___deadline: zx::MonotonicInstant,
    ) -> Result<(), fidl::Error> {
        let _response = self
            .client
            .send_query::<ChildRealmCreateChildrenRequest, fidl::encoding::EmptyPayload>(
                (direct_children, tree_height),
                0x37a93e012377095c,
                fidl::encoding::DynamicFlags::empty(),
                ___deadline,
            )?;
        Ok(_response)
    }
}

#[derive(Debug, Clone)]
pub struct ChildRealmProxy {
    client: fidl::client::Client<fidl::encoding::DefaultFuchsiaResourceDialect>,
}

impl fidl::endpoints::Proxy for ChildRealmProxy {
    type Protocol = ChildRealmMarker;

    fn from_channel(inner: ::fidl::AsyncChannel) -> Self {
        Self::new(inner)
    }

    fn into_channel(self) -> Result<::fidl::AsyncChannel, Self> {
        self.client.into_channel().map_err(|client| Self { client })
    }

    fn as_channel(&self) -> &::fidl::AsyncChannel {
        self.client.as_channel()
    }
}

impl ChildRealmProxy {
    /// Create a new Proxy for test.componentmanager.stresstests/ChildRealm.
    pub fn new(channel: ::fidl::AsyncChannel) -> Self {
        let protocol_name = <ChildRealmMarker as fidl::endpoints::ProtocolMarker>::DEBUG_NAME;
        Self { client: fidl::client::Client::new(channel, protocol_name) }
    }

    /// Get a Stream of events from the remote end of the protocol.
    ///
    /// # Panics
    ///
    /// Panics if the event stream was already taken.
    pub fn take_event_stream(&self) -> ChildRealmEventStream {
        ChildRealmEventStream { event_receiver: self.client.take_event_receiver() }
    }

    /// Component will gracefully exit.
    pub fn r#stop(&self) -> Result<(), fidl::Error> {
        ChildRealmProxyInterface::r#stop(self)
    }

    pub fn r#stop_children(
        &self,
    ) -> fidl::client::QueryResponseFut<(), fidl::encoding::DefaultFuchsiaResourceDialect> {
        ChildRealmProxyInterface::r#stop_children(self)
    }

    /// Create nested children in this realm.
    /// `direct_children` - Num of direct children of this realm.
    /// `tree_height` - Height of the realm tree to create.
    /// Returns once all children are running.
    pub fn r#create_children(
        &self,
        mut direct_children: u16,
        mut tree_height: u16,
    ) -> fidl::client::QueryResponseFut<(), fidl::encoding::DefaultFuchsiaResourceDialect> {
        ChildRealmProxyInterface::r#create_children(self, direct_children, tree_height)
    }
}

impl ChildRealmProxyInterface for ChildRealmProxy {
    fn r#stop(&self) -> Result<(), fidl::Error> {
        self.client.send::<fidl::encoding::EmptyPayload>(
            (),
            0x36c34aa66f2a56ab,
            fidl::encoding::DynamicFlags::empty(),
        )
    }

    type StopChildrenResponseFut =
        fidl::client::QueryResponseFut<(), fidl::encoding::DefaultFuchsiaResourceDialect>;
    fn r#stop_children(&self) -> Self::StopChildrenResponseFut {
        fn _decode(
            mut _buf: Result<<fidl::encoding::DefaultFuchsiaResourceDialect as fidl::encoding::ResourceDialect>::MessageBufEtc, fidl::Error>,
        ) -> Result<(), fidl::Error> {
            let _response = fidl::client::decode_transaction_body::<
                fidl::encoding::EmptyPayload,
                fidl::encoding::DefaultFuchsiaResourceDialect,
                0x61c8762c3aa05df7,
            >(_buf?)?;
            Ok(_response)
        }
        self.client.send_query_and_decode::<fidl::encoding::EmptyPayload, ()>(
            (),
            0x61c8762c3aa05df7,
            fidl::encoding::DynamicFlags::empty(),
            _decode,
        )
    }

    type CreateChildrenResponseFut =
        fidl::client::QueryResponseFut<(), fidl::encoding::DefaultFuchsiaResourceDialect>;
    fn r#create_children(
        &self,
        mut direct_children: u16,
        mut tree_height: u16,
    ) -> Self::CreateChildrenResponseFut {
        fn _decode(
            mut _buf: Result<<fidl::encoding::DefaultFuchsiaResourceDialect as fidl::encoding::ResourceDialect>::MessageBufEtc, fidl::Error>,
        ) -> Result<(), fidl::Error> {
            let _response = fidl::client::decode_transaction_body::<
                fidl::encoding::EmptyPayload,
                fidl::encoding::DefaultFuchsiaResourceDialect,
                0x37a93e012377095c,
            >(_buf?)?;
            Ok(_response)
        }
        self.client.send_query_and_decode::<ChildRealmCreateChildrenRequest, ()>(
            (direct_children, tree_height),
            0x37a93e012377095c,
            fidl::encoding::DynamicFlags::empty(),
            _decode,
        )
    }
}

pub struct ChildRealmEventStream {
    event_receiver: fidl::client::EventReceiver<fidl::encoding::DefaultFuchsiaResourceDialect>,
}

impl std::marker::Unpin for ChildRealmEventStream {}

impl futures::stream::FusedStream for ChildRealmEventStream {
    fn is_terminated(&self) -> bool {
        self.event_receiver.is_terminated()
    }
}

impl futures::Stream for ChildRealmEventStream {
    type Item = Result<ChildRealmEvent, fidl::Error>;

    fn poll_next(
        mut self: std::pin::Pin<&mut Self>,
        cx: &mut std::task::Context<'_>,
    ) -> std::task::Poll<Option<Self::Item>> {
        match futures::ready!(futures::stream::StreamExt::poll_next_unpin(
            &mut self.event_receiver,
            cx
        )?) {
            Some(buf) => std::task::Poll::Ready(Some(ChildRealmEvent::decode(buf))),
            None => std::task::Poll::Ready(None),
        }
    }
}

#[derive(Debug)]
pub enum ChildRealmEvent {
    OnConnected {},
}

impl ChildRealmEvent {
    #[allow(irrefutable_let_patterns)]
    pub fn into_on_connected(self) -> Option<()> {
        if let ChildRealmEvent::OnConnected {} = self {
            Some(())
        } else {
            None
        }
    }

    /// Decodes a message buffer as a [`ChildRealmEvent`].
    fn decode(
        mut buf: <fidl::encoding::DefaultFuchsiaResourceDialect as fidl::encoding::ResourceDialect>::MessageBufEtc,
    ) -> Result<ChildRealmEvent, fidl::Error> {
        let (bytes, _handles) = buf.split_mut();
        let (tx_header, _body_bytes) = fidl::encoding::decode_transaction_header(bytes)?;
        debug_assert_eq!(tx_header.tx_id, 0);
        match tx_header.ordinal {
            0x68a2f3384c114a61 => {
                let mut out = fidl::new_empty!(
                    fidl::encoding::EmptyPayload,
                    fidl::encoding::DefaultFuchsiaResourceDialect
                );
                fidl::encoding::Decoder::<fidl::encoding::DefaultFuchsiaResourceDialect>::decode_into::<fidl::encoding::EmptyPayload>(&tx_header, _body_bytes, _handles, &mut out)?;
                Ok((ChildRealmEvent::OnConnected {}))
            }
            _ => Err(fidl::Error::UnknownOrdinal {
                ordinal: tx_header.ordinal,
                protocol_name: <ChildRealmMarker as fidl::endpoints::ProtocolMarker>::DEBUG_NAME,
            }),
        }
    }
}

/// A Stream of incoming requests for test.componentmanager.stresstests/ChildRealm.
pub struct ChildRealmRequestStream {
    inner: std::sync::Arc<fidl::ServeInner<fidl::encoding::DefaultFuchsiaResourceDialect>>,
    is_terminated: bool,
}

impl std::marker::Unpin for ChildRealmRequestStream {}

impl futures::stream::FusedStream for ChildRealmRequestStream {
    fn is_terminated(&self) -> bool {
        self.is_terminated
    }
}

impl fidl::endpoints::RequestStream for ChildRealmRequestStream {
    type Protocol = ChildRealmMarker;
    type ControlHandle = ChildRealmControlHandle;

    fn from_channel(channel: ::fidl::AsyncChannel) -> Self {
        Self { inner: std::sync::Arc::new(fidl::ServeInner::new(channel)), is_terminated: false }
    }

    fn control_handle(&self) -> Self::ControlHandle {
        ChildRealmControlHandle { inner: self.inner.clone() }
    }

    fn into_inner(
        self,
    ) -> (::std::sync::Arc<fidl::ServeInner<fidl::encoding::DefaultFuchsiaResourceDialect>>, bool)
    {
        (self.inner, self.is_terminated)
    }

    fn from_inner(
        inner: std::sync::Arc<fidl::ServeInner<fidl::encoding::DefaultFuchsiaResourceDialect>>,
        is_terminated: bool,
    ) -> Self {
        Self { inner, is_terminated }
    }
}

impl futures::Stream for ChildRealmRequestStream {
    type Item = Result<ChildRealmRequest, fidl::Error>;

    fn poll_next(
        mut self: std::pin::Pin<&mut Self>,
        cx: &mut std::task::Context<'_>,
    ) -> std::task::Poll<Option<Self::Item>> {
        let this = &mut *self;
        if this.inner.check_shutdown(cx) {
            this.is_terminated = true;
            return std::task::Poll::Ready(None);
        }
        if this.is_terminated {
            panic!("polled ChildRealmRequestStream after completion");
        }
        fidl::encoding::with_tls_decode_buf::<_, fidl::encoding::DefaultFuchsiaResourceDialect>(
            |bytes, handles| {
                match this.inner.channel().read_etc(cx, bytes, handles) {
                    std::task::Poll::Ready(Ok(())) => {}
                    std::task::Poll::Pending => return std::task::Poll::Pending,
                    std::task::Poll::Ready(Err(zx_status::Status::PEER_CLOSED)) => {
                        this.is_terminated = true;
                        return std::task::Poll::Ready(None);
                    }
                    std::task::Poll::Ready(Err(e)) => {
                        return std::task::Poll::Ready(Some(Err(fidl::Error::ServerRequestRead(
                            e.into(),
                        ))))
                    }
                }

                // A message has been received from the channel
                let (header, _body_bytes) = fidl::encoding::decode_transaction_header(bytes)?;

                std::task::Poll::Ready(Some(match header.ordinal {
                    0x36c34aa66f2a56ab => {
                        header.validate_request_tx_id(fidl::MethodType::OneWay)?;
                        let mut req = fidl::new_empty!(
                            fidl::encoding::EmptyPayload,
                            fidl::encoding::DefaultFuchsiaResourceDialect
                        );
                        fidl::encoding::Decoder::<fidl::encoding::DefaultFuchsiaResourceDialect>::decode_into::<fidl::encoding::EmptyPayload>(&header, _body_bytes, handles, &mut req)?;
                        let control_handle = ChildRealmControlHandle { inner: this.inner.clone() };
                        Ok(ChildRealmRequest::Stop { control_handle })
                    }
                    0x61c8762c3aa05df7 => {
                        header.validate_request_tx_id(fidl::MethodType::TwoWay)?;
                        let mut req = fidl::new_empty!(
                            fidl::encoding::EmptyPayload,
                            fidl::encoding::DefaultFuchsiaResourceDialect
                        );
                        fidl::encoding::Decoder::<fidl::encoding::DefaultFuchsiaResourceDialect>::decode_into::<fidl::encoding::EmptyPayload>(&header, _body_bytes, handles, &mut req)?;
                        let control_handle = ChildRealmControlHandle { inner: this.inner.clone() };
                        Ok(ChildRealmRequest::StopChildren {
                            responder: ChildRealmStopChildrenResponder {
                                control_handle: std::mem::ManuallyDrop::new(control_handle),
                                tx_id: header.tx_id,
                            },
                        })
                    }
                    0x37a93e012377095c => {
                        header.validate_request_tx_id(fidl::MethodType::TwoWay)?;
                        let mut req = fidl::new_empty!(
                            ChildRealmCreateChildrenRequest,
                            fidl::encoding::DefaultFuchsiaResourceDialect
                        );
                        fidl::encoding::Decoder::<fidl::encoding::DefaultFuchsiaResourceDialect>::decode_into::<ChildRealmCreateChildrenRequest>(&header, _body_bytes, handles, &mut req)?;
                        let control_handle = ChildRealmControlHandle { inner: this.inner.clone() };
                        Ok(ChildRealmRequest::CreateChildren {
                            direct_children: req.direct_children,
                            tree_height: req.tree_height,

                            responder: ChildRealmCreateChildrenResponder {
                                control_handle: std::mem::ManuallyDrop::new(control_handle),
                                tx_id: header.tx_id,
                            },
                        })
                    }
                    _ => Err(fidl::Error::UnknownOrdinal {
                        ordinal: header.ordinal,
                        protocol_name:
                            <ChildRealmMarker as fidl::endpoints::ProtocolMarker>::DEBUG_NAME,
                    }),
                }))
            },
        )
    }
}

/// The way for parent to conrtol lifecycle of its child in a stress test.
#[derive(Debug)]
pub enum ChildRealmRequest {
    /// Component will gracefully exit.
    Stop {
        control_handle: ChildRealmControlHandle,
    },
    StopChildren {
        responder: ChildRealmStopChildrenResponder,
    },
    /// Create nested children in this realm.
    /// `direct_children` - Num of direct children of this realm.
    /// `tree_height` - Height of the realm tree to create.
    /// Returns once all children are running.
    CreateChildren {
        direct_children: u16,
        tree_height: u16,
        responder: ChildRealmCreateChildrenResponder,
    },
}

impl ChildRealmRequest {
    #[allow(irrefutable_let_patterns)]
    pub fn into_stop(self) -> Option<(ChildRealmControlHandle)> {
        if let ChildRealmRequest::Stop { control_handle } = self {
            Some((control_handle))
        } else {
            None
        }
    }

    #[allow(irrefutable_let_patterns)]
    pub fn into_stop_children(self) -> Option<(ChildRealmStopChildrenResponder)> {
        if let ChildRealmRequest::StopChildren { responder } = self {
            Some((responder))
        } else {
            None
        }
    }

    #[allow(irrefutable_let_patterns)]
    pub fn into_create_children(self) -> Option<(u16, u16, ChildRealmCreateChildrenResponder)> {
        if let ChildRealmRequest::CreateChildren { direct_children, tree_height, responder } = self
        {
            Some((direct_children, tree_height, responder))
        } else {
            None
        }
    }

    /// Name of the method defined in FIDL
    pub fn method_name(&self) -> &'static str {
        match *self {
            ChildRealmRequest::Stop { .. } => "stop",
            ChildRealmRequest::StopChildren { .. } => "stop_children",
            ChildRealmRequest::CreateChildren { .. } => "create_children",
        }
    }
}

#[derive(Debug, Clone)]
pub struct ChildRealmControlHandle {
    inner: std::sync::Arc<fidl::ServeInner<fidl::encoding::DefaultFuchsiaResourceDialect>>,
}

impl fidl::endpoints::ControlHandle for ChildRealmControlHandle {
    fn shutdown(&self) {
        self.inner.shutdown()
    }
    fn shutdown_with_epitaph(&self, status: zx_status::Status) {
        self.inner.shutdown_with_epitaph(status)
    }

    fn is_closed(&self) -> bool {
        self.inner.channel().is_closed()
    }
    fn on_closed(&self) -> fidl::OnSignalsRef<'_> {
        self.inner.channel().on_closed()
    }

    #[cfg(target_os = "fuchsia")]
    fn signal_peer(
        &self,
        clear_mask: zx::Signals,
        set_mask: zx::Signals,
    ) -> Result<(), zx_status::Status> {
        use fidl::Peered;
        self.inner.channel().signal_peer(clear_mask, set_mask)
    }
}

impl ChildRealmControlHandle {
    pub fn send_on_connected(&self) -> Result<(), fidl::Error> {
        self.inner.send::<fidl::encoding::EmptyPayload>(
            (),
            0,
            0x68a2f3384c114a61,
            fidl::encoding::DynamicFlags::empty(),
        )
    }
}

#[must_use = "FIDL methods require a response to be sent"]
#[derive(Debug)]
pub struct ChildRealmStopChildrenResponder {
    control_handle: std::mem::ManuallyDrop<ChildRealmControlHandle>,
    tx_id: u32,
}

/// Set the the channel to be shutdown (see [`ChildRealmControlHandle::shutdown`])
/// if the responder is dropped without sending a response, so that the client
/// doesn't hang. To prevent this behavior, call `drop_without_shutdown`.
impl std::ops::Drop for ChildRealmStopChildrenResponder {
    fn drop(&mut self) {
        self.control_handle.shutdown();
        // Safety: drops once, never accessed again
        unsafe { std::mem::ManuallyDrop::drop(&mut self.control_handle) };
    }
}

impl fidl::endpoints::Responder for ChildRealmStopChildrenResponder {
    type ControlHandle = ChildRealmControlHandle;

    fn control_handle(&self) -> &ChildRealmControlHandle {
        &self.control_handle
    }

    fn drop_without_shutdown(mut self) {
        // Safety: drops once, never accessed again due to mem::forget
        unsafe { std::mem::ManuallyDrop::drop(&mut self.control_handle) };
        // Prevent Drop from running (which would shut down the channel)
        std::mem::forget(self);
    }
}

impl ChildRealmStopChildrenResponder {
    /// Sends a response to the FIDL transaction.
    ///
    /// Sets the channel to shutdown if an error occurs.
    pub fn send(self) -> Result<(), fidl::Error> {
        let _result = self.send_raw();
        if _result.is_err() {
            self.control_handle.shutdown();
        }
        self.drop_without_shutdown();
        _result
    }

    /// Similar to "send" but does not shutdown the channel if an error occurs.
    pub fn send_no_shutdown_on_err(self) -> Result<(), fidl::Error> {
        let _result = self.send_raw();
        self.drop_without_shutdown();
        _result
    }

    fn send_raw(&self) -> Result<(), fidl::Error> {
        self.control_handle.inner.send::<fidl::encoding::EmptyPayload>(
            (),
            self.tx_id,
            0x61c8762c3aa05df7,
            fidl::encoding::DynamicFlags::empty(),
        )
    }
}

#[must_use = "FIDL methods require a response to be sent"]
#[derive(Debug)]
pub struct ChildRealmCreateChildrenResponder {
    control_handle: std::mem::ManuallyDrop<ChildRealmControlHandle>,
    tx_id: u32,
}

/// Set the the channel to be shutdown (see [`ChildRealmControlHandle::shutdown`])
/// if the responder is dropped without sending a response, so that the client
/// doesn't hang. To prevent this behavior, call `drop_without_shutdown`.
impl std::ops::Drop for ChildRealmCreateChildrenResponder {
    fn drop(&mut self) {
        self.control_handle.shutdown();
        // Safety: drops once, never accessed again
        unsafe { std::mem::ManuallyDrop::drop(&mut self.control_handle) };
    }
}

impl fidl::endpoints::Responder for ChildRealmCreateChildrenResponder {
    type ControlHandle = ChildRealmControlHandle;

    fn control_handle(&self) -> &ChildRealmControlHandle {
        &self.control_handle
    }

    fn drop_without_shutdown(mut self) {
        // Safety: drops once, never accessed again due to mem::forget
        unsafe { std::mem::ManuallyDrop::drop(&mut self.control_handle) };
        // Prevent Drop from running (which would shut down the channel)
        std::mem::forget(self);
    }
}

impl ChildRealmCreateChildrenResponder {
    /// Sends a response to the FIDL transaction.
    ///
    /// Sets the channel to shutdown if an error occurs.
    pub fn send(self) -> Result<(), fidl::Error> {
        let _result = self.send_raw();
        if _result.is_err() {
            self.control_handle.shutdown();
        }
        self.drop_without_shutdown();
        _result
    }

    /// Similar to "send" but does not shutdown the channel if an error occurs.
    pub fn send_no_shutdown_on_err(self) -> Result<(), fidl::Error> {
        let _result = self.send_raw();
        self.drop_without_shutdown();
        _result
    }

    fn send_raw(&self) -> Result<(), fidl::Error> {
        self.control_handle.inner.send::<fidl::encoding::EmptyPayload>(
            (),
            self.tx_id,
            0x37a93e012377095c,
            fidl::encoding::DynamicFlags::empty(),
        )
    }
}

mod internal {
    use super::*;
}