address comments

Signed-off-by: xhe <[email protected]>
pingcap · Dec 2, 2023 · 59d7ab3 · 59d7ab3
1 parent cdf33ef
commit 59d7ab3
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/pkg/proxy/backend/authenticator.go b/pkg/proxy/backend/authenticator.go
@@ -233,15 +233,13 @@ loop:
 		if packetErr != nil {
 			// tiproxy pp enabled, tidb pp disabled, tls disabled => invalid sequence
 			// tiproxy pp disabled, tidb pp enabled, tls disabled => invalid sequence
-			for _, p := range []string{
-				"packets out of order",
-				"(nvali): d sequence",
-				"invalid sequence",
-				"PROXY Protocol",
-			} {
-				if strings.Contains(packetErr.Message, p) {
+			if pktIdx == 0 {
+				if packetErr.Code == 1156 || packetErr.Code == 8052 {
 					return errors.Wrap(ErrBackendPPV2, packetErr)
 				}
+				if strings.Contains(packetErr.Message, "PROXY Protocol") {
+					return ErrBackendPPV2
+				}
 			}
 			return errors.Wrap(ErrClientAuthFail, packetErr)
 		}