From 755f8078c184af38325bba261d07892000341468 Mon Sep 17 00:00:00 2001 From: SajjadPourali Date: Wed, 3 Jul 2024 16:41:42 -0400 Subject: [PATCH] Improve readability --- src/stream/tcp.rs | 21 ++++++--------------- 1 file changed, 6 insertions(+), 15 deletions(-) diff --git a/src/stream/tcp.rs b/src/stream/tcp.rs index 40f5968..5e5d149 100644 --- a/src/stream/tcp.rs +++ b/src/stream/tcp.rs @@ -108,21 +108,12 @@ impl IpStackTcpStream { ); tcp_header.acknowledgment_number = self.tcb.get_ack(); - if flags & SYN != 0 { - tcp_header.syn = true; - } - if flags & ACK != 0 { - tcp_header.ack = true; - } - if flags & RST != 0 { - tcp_header.rst = true; - } - if flags & FIN != 0 { - tcp_header.fin = true; - } - if flags & PSH != 0 { - tcp_header.psh = true; - } + tcp_header.syn = flags & SYN != 0; + tcp_header.ack = flags & ACK != 0; + tcp_header.rst = flags & RST != 0; + tcp_header.fin = flags & FIN != 0; + tcp_header.psh = flags & PSH != 0; + let ip_header = match (self.dst_addr.ip(), self.src_addr.ip()) { (std::net::IpAddr::V4(dst), std::net::IpAddr::V4(src)) => {