来自原始fd的BufReader

问题描述 投票:1回答:1

我正在使用fdpass crate通过unix套接字将文件描述符从一个进程发送到另一个进程(我不关心兼容性,unix只是很好)。

使用mio我设法监听这些文件描述符上的事件:

let fd = fdpass::recv_fd(&mut client, vec!(0u8)).unwrap();
let efd = EventedFd(&fd.into_raw_fd());

poll.register(&efd, Token(0), Ready::readable(), PollOpt::level()).unwrap();

这完全正常,但我想使用BufReader逐行读取该文件描述符。我一直在试图找出一种方法来使用from_raw_fd()来实现BufReader不成功。它似乎只存在于文件或网络流等内容中。唯一的另一件事是Stdio没有实现Read,BufRead需要。

有关如何从原始fd获得BufReader而不使mio不安全使用的任何建议?

顺便说一句,文件描述符不是文件(尽管它们可能在某些时候),所以我不能使用File::,现在我只是通过fdpass将客户端的stdin作为原始fd发送。

rust
1个回答
2
投票

不幸的是FromRawFd只适用于hand full of structs。你需要事先知道你想要阅读什么样的“文件”,否则你会冒未定义的行为(因为Rust假定FD是一种它不是的类型)。

但是,您可以实现自己的结构,除了读取适用于所有文件描述符的内容之外别无其他。这可以通过对man (2) read的函数调用来完成。

use libc;
use std::ffi::OsStr;
use std::io::{Error, Read, Result};
use std::os::unix::ffi::OsStrExt;
use std::os::unix::io::{FromRawFd, RawFd};

pub struct RawFdReader {
    fd: RawFd,
}

impl FromRawFd for RawFdReader {
    unsafe fn from_raw_fd(fd: RawFd) -> Self {
        Self { fd }
    }   
}

impl Read for RawFdReader {
    fn read(&mut self, buf: &mut [u8]) -> Result<usize> {
        assert!(buf.len() <= isize::max_value() as usize);
        match unsafe { libc::read(self.fd, buf.as_mut_ptr() as _, buf.len()) } { 
            x if x < 0 => Err(Error::last_os_error()),
            x => Ok(x as usize),
        }
    }   
}

fn main() -> Result<()> {
    let mut reader = unsafe { RawFdReader::from_raw_fd(0) };
    let mut buffer = vec![0; 10];
    let len = reader.read(&mut buffer)?;
    println!("{:?}", OsStr::from_bytes(&buffer[..len]));
    Ok(())
}
© www.soinside.com 2019 - 2024. All rights reserved.