7x31/sniffer-agent
1
0
Fork 0
mirror of https://github.com/zr-hebo/sniffer-agent.git synced 2025-10-09 22:29:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

add response parse module

Browse Source
This commit is contained in:
hebo
2019-12-24 22:39:30 +08:00
parent 0ed1b77305
commit 6c2fcd13b2
6 changed files with 141 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -91,9 +91,15 @@ func (ms *MysqlSession) ReceiveTCPPacket(newPkt *model.TCPPacket) {
ms.readFromClient(newPkt.Seq, newPkt.Payload)
} else {
ms.readFromServer(newPkt.Seq, newPkt.Payload)
ok, respVal, err := ms.readFromServer(newPkt.Seq, newPkt.Payload)
if err != nil {
log.Debug(err.Error())
return
}
qp := ms.GenerateQueryPiece()
if qp != nil {
qp.SetResponse(ok, respVal)
ms.queryPieceReceiver <- qp
}
}
@@ -103,20 +109,6 @@ func (ms *MysqlSession) resetBeginTime() {
ms.stmtBeginTime = time.Now().UnixNano() / millSecondUnit
}
func (ms *MysqlSession) readFromServer(respSeq int64, bytes []byte) {
if ms.expectSendSize < 1 && len(bytes) > 4 {
ms.expectSendSize = extractMysqlPayloadSize(bytes[:4])
contents := bytes[4:]
if ms.prepareInfo != nil && contents[0] == 0 {
ms.prepareInfo.prepareStmtID = bytesToInt(contents[1:5])
}
}
if ms.coverRanges.head.next == nil || ms.coverRanges.head.next.end != respSeq {
ms.clear()
}
}
func (ms *MysqlSession) checkFinish() bool {
if ms.coverRanges.head == nil || ms.coverRanges.head.next == nil {
return false
@@ -147,6 +139,32 @@ func (ms *MysqlSession) clear() {
ms.coverRanges.clear()
}
func (ms *MysqlSession) readFromServer(respSeq int64, bytes []byte) (ok, val int64, err error) {
defer func() {
// 检查返回包的seqid和请求需要的id是否连续
if ms.coverRanges.head.next == nil || ms.coverRanges.head.next.end != respSeq {
ms.clear()
}
if err != nil {
ms.clear()
}
}()
if ms.expectSendSize < 1 && len(bytes) > 4 {
ms.expectSendSize = extractMysqlPayloadSize(bytes[:4])
contents := bytes[4:]
if ms.prepareInfo != nil && contents[0] == 0 {
ms.prepareInfo.prepareStmtID = bytesToInt(contents[1:5])
}
return parseResponseHeader(contents)
}
err = fmt.Errorf("not need packet")
return
}
func (ms *MysqlSession) readFromClient(seqID int64, bytes []byte) {
contentSize := int64(len(bytes))
@@ -318,7 +336,6 @@ func (ms *MysqlSession) GenerateQueryPiece() (qp model.QueryPiece) {
if mqp == nil {
return nil
}
mqp.GenerateJsonBytes()
return mqp
}