Use switch

netbirdio · Dec 21, 2024 · 0970b75 · 0970b75
1 parent fa38d8e
commit 0970b75
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/client/firewall/uspfilter/conntrack/tcp.go b/client/firewall/uspfilter/conntrack/tcp.go
@@ -209,12 +209,13 @@ func (t *TCPTracker) updateState(conn *TCPConnTrack, flags uint8, isOutbound boo
 		}
 
 	case TCPStateFinWait1:
-		if flags&TCPFin != 0 && flags&TCPAck != 0 {
+		switch {
+		case flags&TCPFin != 0 && flags&TCPAck != 0:
 			// Simultaneous close
 			conn.State = TCPStateClosing
-		} else if flags&TCPFin != 0 {
+		case flags&TCPFin != 0:
 			conn.State = TCPStateFinWait2
-		} else if flags&TCPAck != 0 {
+		case flags&TCPAck != 0:
 			conn.State = TCPStateFinWait2
 		}