Rollup merge of #96334 - devnexen:socket_mark, r=dtolnay
socket `set_mark` addition. to be able to set a marker/id on the socket for network filtering (iptables/ipfw here) purpose.
This commit is contained in:
@@ -438,6 +438,17 @@ impl Socket {
|
||||
self.0.set_nonblocking(nonblocking)
|
||||
}
|
||||
|
||||
#[cfg(any(target_os = "linux", target_os = "freebsd", target_os = "openbsd"))]
|
||||
pub fn set_mark(&self, mark: u32) -> io::Result<()> {
|
||||
#[cfg(target_os = "linux")]
|
||||
let option = libc::SO_MARK;
|
||||
#[cfg(target_os = "freebsd")]
|
||||
let option = libc::SO_USER_COOKIE;
|
||||
#[cfg(target_os = "openbsd")]
|
||||
let option = libc::SO_RTABLE;
|
||||
setsockopt(self, libc::SOL_SOCKET, option, mark as libc::c_int)
|
||||
}
|
||||
|
||||
pub fn take_error(&self) -> io::Result<Option<io::Error>> {
|
||||
let raw: c_int = getsockopt(self, libc::SOL_SOCKET, libc::SO_ERROR)?;
|
||||
if raw == 0 { Ok(None) } else { Ok(Some(io::Error::from_raw_os_error(raw as i32))) }
|
||||
|
||||
Reference in New Issue
Block a user