plugin: esc: fixed Auto mode

IrineSistiana · Dec 6, 2020 · 9061ba8 · 9061ba8
1 parent 5b37ecb
commit 9061ba8
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/dispatcher/plugin/functional/ecs/ecs.go b/dispatcher/plugin/functional/ecs/ecs.go
@@ -104,8 +104,12 @@ func (e ecsPlugin) Do(_ context.Context, qCtx *handler.Context) (err error) {
 
 	if checkMsgHasECS(qCtx.Q) == false || e.args.ForceOverwrite {
 		if e.args.Auto {
-			s := qCtx.From.String()
-			ip := net.ParseIP(s)
+			addr := qCtx.From.String()
+			ipStr, _, _ := net.SplitHostPort(addr)
+			if len(ipStr) == 0 {
+				return
+			}
+			ip := net.ParseIP(ipStr)
 			if ip == nil {
 				qCtx.Logf(logrus.WarnLevel, "client address [%s] can not be parsed as ip", addr)
 				return nil