Browse Source

master: increate max packet size for IPv6

master
Denis Drakhnia 2 months ago
parent
commit
3e91b49e5d
  1. 20
      master/src/master_server.rs
  2. 1
      master/src/stats/stub.rs

20
master/src/master_server.rs

@ -41,6 +41,8 @@ pub trait AddrExt: Sized + Eq + Hash + Display + Copy + ToSocketAddrs + ServerAd
fn extract(addr: SocketAddr) -> Result<Self, SocketAddr>; fn extract(addr: SocketAddr) -> Result<Self, SocketAddr>;
fn ip(&self) -> &Self::Ip; fn ip(&self) -> &Self::Ip;
fn wrap(self) -> SocketAddr; fn wrap(self) -> SocketAddr;
fn mtu() -> usize;
} }
impl AddrExt for SocketAddrV4 { impl AddrExt for SocketAddrV4 {
@ -61,6 +63,11 @@ impl AddrExt for SocketAddrV4 {
fn wrap(self) -> SocketAddr { fn wrap(self) -> SocketAddr {
SocketAddr::V4(self) SocketAddr::V4(self)
} }
#[inline(always)]
fn mtu() -> usize {
512
}
} }
impl AddrExt for SocketAddrV6 { impl AddrExt for SocketAddrV6 {
@ -81,10 +88,15 @@ impl AddrExt for SocketAddrV6 {
fn wrap(self) -> SocketAddr { fn wrap(self) -> SocketAddr {
SocketAddr::V6(self) SocketAddr::V6(self)
} }
#[inline(always)]
fn mtu() -> usize {
MAX_PACKET_SIZE
}
} }
/// The maximum size of UDP packets. /// The maximum size of UDP packets.
const MAX_PACKET_SIZE: usize = 512; const MAX_PACKET_SIZE: usize = 1280;
/// How many cleanup calls should be skipped before removing outdated servers. /// How many cleanup calls should be skipped before removing outdated servers.
const SERVER_CLEANUP_MAX: usize = 100; const SERVER_CLEANUP_MAX: usize = 100;
@ -333,7 +345,7 @@ impl<Addr: AddrExt> MasterServer<Addr> {
pub fn run(&mut self, sig_flag: &AtomicBool) -> Result<(), Error> { pub fn run(&mut self, sig_flag: &AtomicBool) -> Result<(), Error> {
let mut buf = [0; MAX_PACKET_SIZE]; let mut buf = [0; MAX_PACKET_SIZE];
while !sig_flag.load(Ordering::Relaxed) { while !sig_flag.load(Ordering::Relaxed) {
let (n, from) = match self.sock.recv_from(&mut buf) { let (n, from) = match self.sock.recv_from(&mut buf[..Addr::mtu()]) {
Ok(x) => x, Ok(x) => x,
Err(e) => match e.kind() { Err(e) => match e.kind() {
io::ErrorKind::Interrupted => break, io::ErrorKind::Interrupted => break,
@ -468,7 +480,7 @@ impl<Addr: AddrExt> MasterServer<Addr> {
}; };
trace!("{}: send {:?}", from, p); trace!("{}: send {:?}", from, p);
let mut buf = [0; MAX_PACKET_SIZE]; let mut buf = [0; MAX_PACKET_SIZE];
let n = p.encode(&mut buf)?; let n = p.encode(&mut buf[..Addr::mtu()])?;
self.sock.send_to(&buf[..n], from)?; self.sock.send_to(&buf[..n], from)?;
} }
} }
@ -663,7 +675,7 @@ impl<Addr: AddrExt> MasterServer<Addr> {
let mut offset = 0; let mut offset = 0;
let mut list = master::QueryServersResponse::new(key); let mut list = master::QueryServersResponse::new(key);
while offset < servers.len() { while offset < servers.len() {
let (n, c) = list.encode(&mut buf, &servers[offset..])?; let (n, c) = list.encode(&mut buf[..Addr::mtu()], &servers[offset..])?;
offset += c; offset += c;
self.sock.send_to(&buf[..n], &to)?; self.sock.send_to(&buf[..n], &to)?;
} }

1
master/src/stats/stub.rs

@ -1,5 +1,6 @@
use crate::config::StatConfig; use crate::config::StatConfig;
#[allow(dead_code)]
#[derive(Default)] #[derive(Default)]
struct Counters; struct Counters;

Loading…
Cancel
Save