Bug fix.

8 years ago · 942b0a0c01
parent 8fe02202de
commit 942b0a0c01
1 changed files with 7 additions and 6 deletions
--- a/sgfw/policy.go
+++ b/sgfw/policy.go
@ -259,12 +259,13 @@ func printPacket(pkt *nfqueue.NFQPacket, hostname string, pinfo *procsnitch.Info
 	SrcPort, DstPort := uint16(0), uint16(0)
 	SrcIp, DstIp := getPacketIP4Addrs(pkt)

-	switch pkt.Packet.TransportLayer().TransportFlow().EndpointType() {
-		case 4:
+//	switch pkt.Packet.TransportLayer().TransportFlow().EndpointType() {
+	if pkt.Packet.Layer(layers.LayerTypeTCP) != nil {
+//		case 4:
 		proto = "TCP"
-		case 5:
+	} else if pkt.Packet.Layer(layers.LayerTypeUDP) != nil {
+//		case 5:
 		proto = "UDP"
-		default:
 	}

 	if proto == "TCP" {