From c2761234c88bf1a86495713bc4308b717e55d43e Mon Sep 17 00:00:00 2001
From: Matthew Gyurgyik <matthew@gyurgyik.io>
Date: Sun, 7 Jun 2020 18:33:08 +0000
Subject: [PATCH] use layer type for string conversion

---
 main.go | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/main.go b/main.go
index 3c32332..271e14f 100644
--- a/main.go
+++ b/main.go
@@ -157,7 +157,7 @@ func printPacketInfo(src string, dst string, packet gopacket.Packet) {
 
 	if eapLayer != nil {
 		eap, _ := eapLayer.(*layers.EAP)
-		codeString := EAPTypeString(eap.Code)
+		codeString := EAPCodeToString(eap.Code)
 		line += fmt.Sprintf(", %s (%d) id %d", codeString, eap.Code, eap.Id)
 	}
 
@@ -165,15 +165,15 @@ func printPacketInfo(src string, dst string, packet gopacket.Packet) {
 	fmt.Println(line)
 }
 
-func EAPTypeString(code layers.EAPCode) string {
+func EAPCodeToString(code layers.EAPCode) string {
 	switch code {
-	case 1:
+	case layers.EAPCodeRequest:
 		return "Request"
-	case 2:
+	case layers.EAPCodeResponse:
 		return "Response"
-	case 3:
+	case layers.EAPCodeSuccess:
 		return "Success"
-	case 4:
+	case layers.EAPCodeFailure:
 		return "Failure"
 	}
 	return "Unknown"