forked from mozilla/neqo
-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
8a3aaa2
commit f8a3d7e
Showing
13 changed files
with
199 additions
and
41 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -10,3 +10,6 @@ members = [ | |
| "neqo-interop", | ||
| "test-fixture", | ||
| ] | ||
|
|
||
| [patch.crates-io] | ||
| nix = { path = "./../nix" } | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,12 @@ | ||
| [package] | ||
| name = "neqo-helper" | ||
| version = "0.6.7" | ||
| authors = ["Lars Eggert <lars@eggert.org>"] | ||
| edition = "2018" | ||
| rust-version = "1.65.0" | ||
| license = "MIT OR Apache-2.0" | ||
|
|
||
| [dependencies] | ||
| neqo-common = { path="./../neqo-common" } | ||
| nix = { version = "0.27.1", features = ["socket", "net", "uio"] } | ||
| log = {version = "0.4.0", default-features = false} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,136 @@ | ||
| // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or | ||
| // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license | ||
| // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your | ||
| // option. This file may not be copied, modified, or distributed | ||
| // except according to those terms. | ||
|
|
||
| use std::{ | ||
| io::{self, Error, ErrorKind, IoSlice, IoSliceMut}, | ||
| net::{SocketAddr, UdpSocket}, | ||
| process::exit, | ||
| }; | ||
|
|
||
| use nix::{ | ||
| cmsg_space, | ||
| errno::Errno::{EAGAIN, EINTR}, | ||
| sys::socket::{ | ||
| recvmsg, sendmsg, setsockopt, | ||
| sockopt::{IpDontFrag, IpRecvTos, IpTos, Ipv6DontFrag, Ipv6RecvTClass, Ipv6TClass}, | ||
| AddressFamily, ControlMessage, ControlMessageOwned, MsgFlags, SockaddrLike, | ||
| SockaddrStorage, | ||
| }, | ||
| }; | ||
|
|
||
| use neqo_common::{Datagram, qdebug}; | ||
|
|
||
| // Bind a UDPO socket and set some default socket options. | ||
| pub fn bind(local_addr: SocketAddr) -> io::Result<UdpSocket> { | ||
| let socket = match UdpSocket::bind(local_addr) { | ||
| Err(e) => { | ||
| eprintln!("Unable to bind UDP socket: {}", e); | ||
| exit(1) | ||
| } | ||
| Ok(s) => { | ||
| // Don't let the host stack or network path fragment our IP packets | ||
| // (RFC9000, Section 14). | ||
| let res = match local_addr { | ||
| SocketAddr::V4(..) => setsockopt(&s, IpDontFrag, &true), | ||
| SocketAddr::V6(..) => setsockopt(&s, Ipv6DontFrag, &true), | ||
| }; | ||
| assert!(res.is_ok()); | ||
|
|
||
| // Mark all outgoing IP packets with ECT(0). This is a kludge until proper ECN | ||
| // validation is supported (RFC9000, Section 13.4.2). | ||
| let res = match local_addr { | ||
| SocketAddr::V4(..) => setsockopt(&s, IpTos, &0b00000010), | ||
| SocketAddr::V6(..) => setsockopt(&s, Ipv6TClass, &0b00000010), | ||
| }; | ||
| assert!(res.is_ok()); | ||
|
|
||
| let res = match local_addr { | ||
| SocketAddr::V4(..) => setsockopt(&s, IpRecvTos, &true), | ||
| SocketAddr::V6(..) => setsockopt(&s, Ipv6RecvTClass, &true), | ||
| }; | ||
| assert!(res.is_ok()); | ||
|
|
||
| // let res = match local_addr { | ||
| // SocketAddr::V4(..) => setsockopt(&s, IpRecvTtl, &true), | ||
| // SocketAddr::V6(..) => setsockopt(&s, Ipv6HopLimit, &true), | ||
| // }; | ||
| // assert!(res.is_ok()); | ||
| s | ||
| } | ||
| }; | ||
| Ok(socket) | ||
| } | ||
|
|
||
| fn to_sockaddr(addr: SocketAddr) -> SockaddrStorage { | ||
| SockaddrStorage::from(addr) | ||
| } | ||
|
|
||
| pub fn emit_datagram(fd: i32, d: Datagram) -> io::Result<()> { | ||
| let iov = [IoSlice::new(&d[..])]; | ||
| let tos = d.tos(); | ||
| let cmsg = ControlMessage::IpTos(&tos); | ||
| let sent = sendmsg( | ||
| fd, | ||
| &iov, | ||
| &[cmsg], | ||
| MsgFlags::empty(), | ||
| Some(&to_sockaddr(d.destination())), | ||
| ) | ||
| .unwrap(); | ||
| if sent != d.len() { | ||
| eprintln!("Unable to send all {} bytes of datagram", d.len()); | ||
| } | ||
| qdebug!("TX TOS {:#04x}", tos); | ||
| Ok(()) | ||
| } | ||
|
|
||
| fn to_socket_addr(addr: &SockaddrStorage) -> SocketAddr { | ||
| let fam = addr.family().unwrap(); | ||
|
|
||
| let port = match fam { | ||
| AddressFamily::Inet => addr.as_sockaddr_in().unwrap().port(), | ||
| AddressFamily::Inet6 => addr.as_sockaddr_in6().unwrap().port(), | ||
| _ => todo!(), | ||
| }; | ||
|
|
||
| match fam { | ||
| AddressFamily::Inet => SocketAddr::new( | ||
| std::net::IpAddr::V4(addr.as_sockaddr_in().unwrap().ip()), | ||
| port, | ||
| ), | ||
| AddressFamily::Inet6 => SocketAddr::new( | ||
| std::net::IpAddr::V6(addr.as_sockaddr_in6().unwrap().ip()), | ||
| port, | ||
| ), | ||
| _ => todo!(), | ||
| } | ||
| } | ||
|
|
||
| pub fn recv_datagram(fd: i32, buf: &mut [u8], tos: &mut u8) -> io::Result<(usize, SocketAddr)> { | ||
| let mut iov = [IoSliceMut::new(buf)]; | ||
| let mut cmsg = cmsg_space!(IpTos, Ipv6TClass); | ||
| let flags = MsgFlags::empty(); | ||
|
|
||
| match recvmsg::<SockaddrStorage>(fd, &mut iov, Some(&mut cmsg), flags) { | ||
| Err(e) if e == EAGAIN => Err(Error::new(ErrorKind::WouldBlock, e)), | ||
| Err(e) if e == EINTR => Err(Error::new(ErrorKind::Interrupted, e)), | ||
| Err(e) => { | ||
| eprintln!("UDP error: {}", e); | ||
| exit(1) | ||
| } | ||
| Ok(res) => { | ||
| for cmsg in res.cmsgs() { | ||
| *tos = match cmsg { | ||
| ControlMessageOwned::IpTos(t) => t, | ||
| ControlMessageOwned::Ipv6TClass(t) => t, | ||
| _ => todo!(), | ||
| }; | ||
| } | ||
| qdebug!("RX TOS {:#04x}", tos); | ||
| Ok((res.bytes, to_socket_addr(&res.address.unwrap()))) | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.