zx/channel/
io_slice.rs

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
// Copyright 2024 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

use crate::sys::zx_channel_iovec_t;
use std::ops::Deref;

/// A reference to a readable slice of memory for channel writes and calls, analogous to
/// `std::io::IoSlice` but for Zircon channel I/O. ABI-compatible with `zx_channel_iovec_t`,
/// guaranteeing the pointed-to bytes are readable using lifetimes.
#[derive(Copy, Clone)]
#[repr(transparent)]
pub struct ChannelIoSlice<'a>(zx_channel_iovec_t, std::marker::PhantomData<&'a [u8]>);

impl<'a> ChannelIoSlice<'a> {
    /// Convert a Rust byte slice to a `ChannelIoSlice`. If the input slice is longer than can be
    /// referenced by a `zx_channel_iovec_t` the length will be truncated, although this is
    /// significantly longer than `ZX_CHANNEL_MAX_MSG_BYTES` in practice.
    pub fn new(buf: &'a [u8]) -> Self {
        let mut inner = zx_channel_iovec_t::default();
        inner.buffer = buf.as_ptr();
        inner.capacity = buf.len() as u32;
        Self(inner, std::marker::PhantomData)
    }
}

impl<'a> Deref for ChannelIoSlice<'a> {
    type Target = [u8];
    fn deref(&self) -> &[u8] {
        // SAFETY: lifetime marker guarantees these bytes are still live.
        unsafe { std::slice::from_raw_parts(self.0.buffer, self.0.capacity as usize) }
    }
}

impl<'a> std::cmp::PartialEq for ChannelIoSlice<'a> {
    fn eq(&self, rhs: &Self) -> bool {
        self.deref().eq(rhs.deref())
    }
}
impl<'a> std::cmp::Eq for ChannelIoSlice<'a> {}

impl<'a> std::hash::Hash for ChannelIoSlice<'a> {
    fn hash<H: std::hash::Hasher>(&self, h: &mut H) {
        self.deref().hash(h)
    }
}

impl<'a> std::fmt::Debug for ChannelIoSlice<'a> {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        self.deref().fmt(f)
    }
}

// SAFETY: this type has no meaningful drop impl other than releasing its borrow.
unsafe impl<'a> Send for ChannelIoSlice<'a> {}

// SAFETY: this type has no mutability.
unsafe impl<'a> Sync for ChannelIoSlice<'a> {}