deal all auth packet

capture/network.go

@@ -134,10 +134,9 @@ func (nc *networkCard) listenNormal() {
 
 			// send FIN tcp packet to avoid not complete session cannot be released
 			tcpPkt := packet.TransportLayer().(*layers.TCP)
-			payLoad := tcpPkt.Payload
 			// deal FIN packet
 			// deal auth packet
-			if tcpPkt.FIN || (len(payLoad) >= 5 && sd.IsAuthPacket(payLoad[4])) {
+			if tcpPkt.FIN || sd.IsAuthPacket(tcpPkt.Payload) {
 				nc.parseTCPPackage(packet)
 				continue
 			}

session-dealer/controller.go

@@ -25,11 +25,12 @@ func CheckParams()  {
 	}
 }
 
-func IsAuthPacket(val byte) bool {
+func IsAuthPacket(payload []byte) bool {
 	switch serviceType {
 	case ServiceTypeMysql:
-		return mysql.IsAuthPacket(val)
+		return len(payload) >= 5 && mysql.IsAuth(payload[4])
+
 	default:
-		return mysql.IsAuthPacket(val)
+		return false
 	}
 }

session-dealer/mysql/session.go

@@ -208,7 +208,7 @@ func (ms *MysqlSession) readFromClient(seqID int64, bytes []byte) {
 	// ms.expectReceiveSize = ms.expectReceiveSize - int(contentSize)
 }
 
-func IsAuthPacket(val byte) bool {
+func IsAuth(val byte) bool {
 	return val > 32
 }
 
@@ -231,7 +231,7 @@ func (ms *MysqlSession) GenerateQueryPiece() (qp model.QueryPiece) {
 
 	var mqp *model.PooledMysqlQueryPiece
 	var querySQLInBytes []byte
-	if IsAuthPacket(ms.cachedStmtBytes[0]) {
+	if IsAuth(ms.cachedStmtBytes[0]) {
 		userName, dbName, err := parseAuthInfo(ms.cachedStmtBytes)
 		if err != nil {
 			log.Errorf("parse auth info failed <-- %s", err.Error())