7x31/sniffer-agent
1
0
Fork 0
mirror of https://github.com/zr-hebo/sniffer-agent.git synced 2025-08-11 17:53:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

deal all auth packet

Browse Source
This commit is contained in:
hebo
2019-11-13 13:47:56 +08:00
parent 1b0d2c8e91
commit 16722c878c
3 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
session-dealer/controller.go
View File

@@ -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
}
}

4
session-dealer/mysql/session.go
View File

@@ -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())