diff --git a/crates/cli/src/main.rs b/crates/cli/src/main.rs index 5594593..2bafb77 100644 --- a/crates/cli/src/main.rs +++ b/crates/cli/src/main.rs @@ -304,7 +304,7 @@ fn output_result_bson_base64(result: T) { if let bson::Bson::Document(document) = bson { let bytes = bson::to_vec(&document).unwrap(); - println!("{}", base64::prelude::BASE64_STANDARD.encode(&bytes)); + println!("{}", base64::prelude::BASE64_STANDARD.encode(bytes)); } else { panic!("Failed to convert result to BSON Base64"); } diff --git a/crates/lib/src/capture/packet.rs b/crates/lib/src/capture/packet.rs index f7c3c3c..ef3ae62 100644 --- a/crates/lib/src/capture/packet.rs +++ b/crates/lib/src/capture/packet.rs @@ -24,9 +24,9 @@ pub(crate) enum Direction { #[derive(Clone, Copy, Debug, PartialEq)] pub(crate) enum Protocol { /// Transmission Control Protocol. - TCP, + Tcp, /// User Datagram Protocol. - UDP, + Udp, } /// Trait for handling different types of IP addresses (IPv4, IPv6). @@ -41,7 +41,7 @@ pub(crate) trait IpAddress: Sized { pub(crate) struct CapturePacket<'a> { /// Direction of the packet (Send/Receive). pub(crate) direction: Direction, - /// Protocol of the packet (TCP/UDP). + /// Protocol of the packet (Tcp/UDP). pub(crate) protocol: Protocol, /// Remote socket address. pub(crate) remote_address: &'a SocketAddr, @@ -135,14 +135,14 @@ mod tests { fn test_ports_by_direction() { let packet_send = CapturePacket { direction: Direction::Send, - protocol: Protocol::TCP, + protocol: Protocol::Tcp, local_address: &socket_addr("127.0.0.1:8080"), remote_address: &socket_addr("192.168.1.1:80"), }; let packet_receive = CapturePacket { direction: Direction::Receive, - protocol: Protocol::TCP, + protocol: Protocol::Tcp, local_address: &socket_addr("127.0.0.1:8080"), remote_address: &socket_addr("192.168.1.1:80"), }; @@ -155,14 +155,14 @@ mod tests { fn test_ip_addr() { let packet_send = CapturePacket { direction: Direction::Send, - protocol: Protocol::TCP, + protocol: Protocol::Tcp, local_address: &socket_addr("127.0.0.1:8080"), remote_address: &socket_addr("192.168.1.1:80"), }; let packet_receive = CapturePacket { direction: Direction::Receive, - protocol: Protocol::TCP, + protocol: Protocol::Tcp, local_address: &socket_addr("127.0.0.1:8080"), remote_address: &socket_addr("192.168.1.1:80"), }; @@ -187,7 +187,7 @@ mod tests { fn test_ip_by_direction_type_specific() { let packet = CapturePacket { direction: Direction::Send, - protocol: Protocol::TCP, + protocol: Protocol::Tcp, local_address: &socket_addr("127.0.0.1:8080"), remote_address: &socket_addr("192.168.1.1:80"), }; diff --git a/crates/lib/src/capture/pcap.rs b/crates/lib/src/capture/pcap.rs index 0613619..f041b3a 100644 --- a/crates/lib/src/capture/pcap.rs +++ b/crates/lib/src/capture/pcap.rs @@ -56,7 +56,7 @@ impl Pcap { let (source_port, dest_port) = info.ports_by_direction(); match info.protocol { - Protocol::TCP => { + Protocol::Tcp => { let buf_size = { let mut tcp = MutableTcpPacket::new(&mut buf).unwrap(); tcp.set_source(source_port); @@ -123,7 +123,7 @@ impl Pcap { vec![EnhancedPacketOption::Comment("Generated TCP ACK".into())], ); } - Protocol::UDP => { + Protocol::Udp => { let buf_size = { let mut udp = MutableUdpPacket::new(&mut buf).unwrap(); udp.set_source(source_port); diff --git a/crates/lib/src/socket.rs b/crates/lib/src/socket.rs index 4548720..eeba5eb 100644 --- a/crates/lib/src/socket.rs +++ b/crates/lib/src/socket.rs @@ -210,13 +210,13 @@ pub mod capture { /// Represents the TCP protocol provider. pub(crate) struct ProtocolTCP; impl ProtocolProvider for ProtocolTCP { - fn protocol() -> Protocol { Protocol::TCP } + fn protocol() -> Protocol { Protocol::Tcp } } /// Represents the UDP protocol provider. pub(crate) struct ProtocolUDP; impl ProtocolProvider for ProtocolUDP { - fn protocol() -> Protocol { Protocol::UDP } + fn protocol() -> Protocol { Protocol::Udp } } /// A socket wrapper that allows capturing packets.