Selaa lähdekoodia

- added error handling for when a packet can't be parsed

Tobias Begalke 7 vuotta sitten
vanhempi
commit
593ccea416
1 muutettua tiedostoa jossa 10 lisäystä ja 2 poistoa
  1. 10 2
      main.go

+ 10 - 2
main.go

@@ -231,9 +231,17 @@ func processPacket(packet gopacket.Packet) {
 
 	switch config.Protocol {
 	case "http":
-		processHTTP(&request, applicationLayer.Payload())
+		err := processHTTP(&request, applicationLayer.Payload())
+		if err != nil {
+			log.Println(err)
+			return
+		}
 	case "ajp13":
-		processAJP13(&request, applicationLayer.Payload())
+		err := processAJP13(&request, applicationLayer.Payload())
+		if err != nil {
+			log.Println(err)
+			return
+		}
 	}
 
 	if config.UseXForwardedAsSource && request.XForwardedFor != "" {