tribufu/rust-gamedig
1
0
Fork 0
mirror of https://github.com/tribufu/rust-gamedig synced 2026-06-01 09:42:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: backport pcap impl

Browse source
This commit is contained in:
Cain 2024-01-13 03:22:50 +00:00 • committed by Douile
parent 49096e46bb
commit 3d47180e85
No known key found for this signature in database
GPG key ID: E048586A5FF6585C
7 changed files with 272 additions and 235 deletions
Download patch file Download diff file Expand all files Collapse all files

12
crates/lib/src/socket.rs
View file

@ -213,13 +213,13 @@ pub mod capture {
}
/// A trait representing a provider of a network protocol.
pub trait ProtocolProvider {
pub(crate) trait ProtocolProvider {
/// Returns the protocol used by the provider.
fn protocol() -> Protocol;
}
/// Represents the TCP protocol provider.
pub struct ProtocolTCP;
pub(crate) struct ProtocolTCP;
impl ProtocolProvider for ProtocolTCP {
fn protocol() -> Protocol {
Protocol::TCP
@ -375,14 +375,14 @@ pub mod capture {
///
/// This type captures and processes UDP packets, wrapping around standard UDP socket
/// functionalities with additional packet capture capabilities.
pub type CapturedUdpSocket = WrappedCaptureSocket<UdpSocketImpl, ProtocolUDP>;
pub(crate) type CapturedUdpSocket = WrappedCaptureSocket<UdpSocketImpl, ProtocolUDP>;
/// A specialized `WrappedCaptureSocket` for TCP, using `TcpSocketImpl` as the inner socket
/// and `ProtocolTCP` as the protocol provider.
///
/// This type captures and processes TCP packets, wrapping around standard TCP socket
/// functionalities with additional packet capture capabilities.
pub type CapturedTcpSocket = WrappedCaptureSocket<TcpSocketImpl, ProtocolTCP>;
pub(crate) type CapturedTcpSocket = WrappedCaptureSocket<TcpSocketImpl, ProtocolTCP>;
}
#[cfg(not(feature = "packet_capture"))]
@ -391,9 +391,9 @@ pub type UdpSocket = UdpSocketImpl;
pub type TcpSocket = TcpSocketImpl;
#[cfg(feature = "packet_capture")]
pub type UdpSocket = capture::CapturedUdpSocket;
pub(crate) type UdpSocket = capture::CapturedUdpSocket;
#[cfg(feature = "packet_capture")]
pub type TcpSocket = capture::CapturedTcpSocket;
pub(crate) type TcpSocket = capture::CapturedTcpSocket;
#[cfg(test)]
mod tests {