Line data    Source code

       1              : use nix::fcntl::{fcntl, OFlag, F_GETFL, F_SETFL};
       2              : use std::os::unix::io::RawFd;
       3              : 
       4              : /// Put a file descriptor into non-blocking mode
       5           16 : pub fn set_nonblock(fd: RawFd) -> Result<(), std::io::Error> {
       6           16 :     let bits = fcntl(fd, F_GETFL)?;
       7              : 
       8              :     // If F_GETFL returns some unknown bits, they should be valid
       9              :     // for passing back to F_SETFL, too. If we left them out, the F_SETFL
      10              :     // would effectively clear them, which is not what we want.
      11           16 :     let mut flags = OFlag::from_bits_retain(bits);
      12           16 :     flags |= OFlag::O_NONBLOCK;
      13           16 : 
      14           16 :     fcntl(fd, F_SETFL(flags))?;
      15              : 
      16           16 :     Ok(())
      17           16 : }