Work on portmaster restructuring

network/reference/protocols.go

@@ -0,0 +1,43 @@
+package reference
+
+import "strconv"
+
+var (
+	protocolNames = map[uint8]string{
+		1:   "ICMP",
+		2:   "IGMP",
+		6:   "TCP",
+		17:  "UDP",
+		27:  "RDP",
+		33:  "DCCP",
+		136: "UDPLite",
+	}
+
+	protocolNumbers = map[string]uint8{
+		"ICMP":    1,
+		"IGMP":    2,
+		"TCP":     6,
+		"UDP":     17,
+		"RDP":     27,
+		"DCCP":    33,
+		"UDPLite": 136,
+	}
+)
+
+// GetProtocolName returns the name of a IP protocol number.
+func GetProtocolName(protocol uint8) (name string) {
+	name, ok := protocolNames[protocol]
+	if ok {
+		return name
+	}
+	return strconv.Itoa(int(protocol))
+}
+
+// GetProtocolNumber returns the number of a IP protocol name.
+func GetProtocolNumber(protocol string) (number uint8, ok bool) {
+	number, ok = protocolNumbers[protocol]
+	if ok {
+		return number, true
+	}
+	return 0, false
+}