diff options
Diffstat (limited to 'msg_socket/src/msg_on_socket.rs')
-rw-r--r-- | msg_socket/src/msg_on_socket.rs | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/msg_socket/src/msg_on_socket.rs b/msg_socket/src/msg_on_socket.rs index 4149098..3dc59c0 100644 --- a/msg_socket/src/msg_on_socket.rs +++ b/msg_socket/src/msg_on_socket.rs @@ -196,6 +196,57 @@ impl<T: MsgOnSocket> MsgOnSocket for Option<T> { } } +impl<O: MsgOnSocket, E: MsgOnSocket> MsgOnSocket for Result<O, E> { + fn uses_fd() -> bool { + O::uses_fd() || E::uses_fd() + } + + fn msg_size(&self) -> usize { + 1 + match self { + Ok(v) => v.msg_size(), + Err(e) => e.msg_size(), + } + } + + fn fd_count(&self) -> usize { + match self { + Ok(v) => v.fd_count(), + Err(e) => e.fd_count(), + } + } + + unsafe fn read_from_buffer(buffer: &[u8], fds: &[RawFd]) -> MsgResult<(Self, usize)> { + match buffer.get(0).ok_or(MsgError::WrongMsgBufferSize)? { + 0 => { + let (value, size) = O::read_from_buffer(&buffer[1..], &fds)?; + Ok((Ok(value), size)) + } + 1 => { + let (value, size) = E::read_from_buffer(&buffer[1..], &fds)?; + Ok((Err(value), size)) + } + _ => Err(MsgError::InvalidType), + } + } + + fn write_to_buffer(&self, buffer: &mut [u8], mut fds: &mut [RawFd]) -> MsgResult<usize> { + if buffer.is_empty() { + return Err(MsgError::WrongMsgBufferSize); + } + + match self { + Ok(value) => { + buffer[0] = 0; + Ok(value.write_to_buffer(&mut buffer[1..], &mut fds)?) + } + Err(error) => { + buffer[0] = 1; + Ok(error.write_to_buffer(&mut buffer[1..], &mut fds)?) + } + } + } +} + impl<T: MsgOnSocket> MsgOnSocket for Mutex<T> { fn uses_fd() -> bool { T::uses_fd() |