[*]优化协议名称逻辑，避免存在首字符为数字的协议名

lcvvvv · Nov 4, 2022 · ed5e688 · ed5e688
1 parent eb41d44
commit ed5e688
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/gonmap.go b/gonmap.go
@@ -3,6 +3,7 @@ package gonmap
 import (
 	"log"
 	"os"
+	"regexp"
 	"strings"
 	"time"
 )
@@ -186,6 +187,8 @@ func GuessProtocol(port int) string {
 	return protocol
 }
 
+var regexpFirstNum = regexp.MustCompile(`^\d`)
+
 func FixProtocol(oldProtocol string) string {
 	//进行最后输出修饰
 	if oldProtocol == "ssl/http" {
@@ -229,6 +232,9 @@ func FixProtocol(oldProtocol string) string {
 			return oldProtocol[4:] + "-ssl"
 		}
 	}
+	if regexpFirstNum.MatchString(oldProtocol) {
+		oldProtocol = "S" + oldProtocol
+	}
 	oldProtocol = strings.ReplaceAll(oldProtocol, "_", "-")
 	return oldProtocol
 }