Question

我有std::net::TcpStream。我想确定是否有数据可供阅读而尚未实际阅读。

我在TcpStream上可以找到的唯一相关API是read

不保证是否阻止等待数据

对于这个问题听起来并不鼓舞。

related question似乎下拉到文件描述符和read(2)以强制进行非阻塞读取。但是我无法弄清楚如何使用read(2)来查看fd而不实际阅读它。

我认为这是select(2)的工作，但为C参数构造fd_set似乎相当毛茸茸。当然没有Rust类型，并且我不会立即清楚如何发明一个。

Answer 1

我认为这是select（2）的工作，但是为C参数构造fd_sets看起来相当毛茸茸。

我认为poll(2)应该更方便。例如：

#![feature(std_misc, net, libc, os, io)]

extern crate libc;
use libc::{c_int, c_uint, c_short};
use std::thread::spawn;
use std::net::{TcpListener, TcpStream};
use std::os;
use std::io::Read;
use std::os::unix::AsRawFd;

#[repr(C)]
struct pollfd {
    fd: c_int,
    events: c_short,
    revents: c_short,
}

extern { fn poll(fds: *mut pollfd, nfds: c_uint, timeout: c_int) -> c_int; }

const POLLIN: c_short = 1;

fn handle_client(mut stream: TcpStream) {
    let mut fdset = pollfd { fd: stream.as_raw_fd(), events: POLLIN, revents: 0, };
    loop {
        match unsafe { poll(&mut fdset as *mut _, 1, -1) } {
            ret if ret < 0 => panic!("poll error: {}", os::last_os_error()),
            ret if ret > 0 && fdset.events == fdset.revents => {
                let mut byte: &mut [u8] = &mut [0];
                match stream.read(&mut byte).unwrap() {
                    0 => break,
                    1 => println!("A byte read: {}", byte[0]),
                    _ => unreachable!(),
                }
            },
            _ => break,
        }
    }
}

fn main() {
    let listener = TcpListener::bind("127.0.0.1:9999").unwrap();
    for stream in listener.incoming() {
        match stream {
            Ok(stream) => { spawn(move || handle_client(stream)); },
            Err(e) => panic!("connection error: {}", e),
        }
    }
}

确定TcpStream上的数据是否可用

1 个答案: