hongshaorou/rustdesk
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

bind_multicast

Browse Source
This commit is contained in:
open-trade 2022-01-10 18:58:51 +08:00
parent 9a92b6ac4a
commit 9c9855877d
1 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
libs/hbb_common/src/udp.rs
View File

@ -3,8 +3,8 @@ use anyhow::anyhow;
use bytes::{Bytes, BytesMut};
use futures::{SinkExt, StreamExt};
use protobuf::Message;
use socket2::{Domain, Socket, Type};
use std::net::SocketAddr;
use socket2::{Domain, Protocol, Socket, Type};
use std::net::{SocketAddr, SocketAddrV4};
use tokio::net::{ToSocketAddrs, UdpSocket};
use tokio_socks::{udp::Socks5UdpFramed, IntoTargetAddr, TargetAddr, ToProxyAddrs};
use tokio_util::{codec::BytesCodec, udp::UdpFramed};
@ -142,3 +142,18 @@ impl FramedSocket {
}
}
}
// const DEFAULT_MULTICAST: &str = "239.255.42.98";
pub fn bind_multicast(addr: &SocketAddrV4, multi_addr: &SocketAddrV4) -> ResultType<FramedSocket> {
assert!(multi_addr.ip().is_multicast(), "Must be multcast address");
let socket = Socket::new(Domain::ipv4(), Type::dgram(), Some(Protocol::udp()))?;
socket.set_reuse_address(true)?;
socket.bind(&socket2::SockAddr::from(*addr))?;
socket.set_multicast_loop_v4(true)?;
socket.join_multicast_v4(multi_addr.ip(), addr.ip())?;
Ok(FramedSocket::Direct(UdpFramed::new(
UdpSocket::from_std(socket.into_udp_socket())?,
BytesCodec::new(),
)))
}