github/kubeshark
1
0
Fork 0
mirror of https://github.com/kubeshark/kubeshark.git synced 2025-07-23 04:31:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Implement the AMQP ConnectionClose

Browse Source
This commit is contained in:
M. Mert Yildiran 2021-08-21 16:28:44 +03:00
parent 8fa1506969
commit 21cac50ccb
No known key found for this signature in database
GPG Key ID: D42ADB236521BF7A
2 changed files with 65 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
tap/extensions/amqp/helpers.go
View File

@ -634,16 +634,59 @@ func representConnectionStart(event map[string]interface{}) []interface{} {
return rep
}
func printEventConnectionClose(eventConnectionClose ConnectionClose) {
return
fmt.Printf(
"[%s] ReplyCode: %d, ReplyText: %s, ClassId: %d, MethodId: %d\n",
connectionMethodMap[50],
eventConnectionClose.ReplyCode,
eventConnectionClose.ReplyText,
eventConnectionClose.ClassId,
eventConnectionClose.MethodId,
)
func emitConnectionClose(event ConnectionClose, connectionInfo *api.ConnectionInfo, emitter api.Emitter) {
request := &api.GenericMessage{
IsRequest: true,
CaptureTime: time.Now(),
Payload: AMQPPayload{
Type: "connection_close",
Data: &AMQPWrapper{
Method: connectionMethodMap[50],
Url: event.ReplyText,
Details: event,
},
},
}
item := &api.OutputChannelItem{
Protocol: protocol,
Timestamp: time.Now().UnixNano() / int64(time.Millisecond),
ConnectionInfo: connectionInfo,
Pair: &api.RequestResponsePair{
Request: *request,
Response: api.GenericMessage{},
},
}
emitter.Emit(item)
}
func representConnectionClose(event map[string]interface{}) []interface{} {
rep := make([]interface{}, 0)
details, _ := json.Marshal([]map[string]string{
{
"name": "Reply Code",
"value": fmt.Sprintf("%g", event["ReplyCode"].(float64)),
},
{
"name": "Reply Text",
"value": event["ReplyText"].(string),
},
{
"name": "Class ID",
"value": fmt.Sprintf("%g", event["ClassId"].(float64)),
},
{
"name": "Method ID",
"value": fmt.Sprintf("%g", event["MethodId"].(float64)),
},
})
rep = append(rep, map[string]string{
"type": "table",
"title": "Details",
"data": string(details),
})
return rep
}
func printEventQueueBind(eventQueueBind QueueBind) {

14
tap/extensions/amqp/main.go
View File

@ -190,7 +190,7 @@ func (d dissecting) Dissect(b *bufio.Reader, isClient bool, tcpID *api.TcpID, em
ClassId: m.ClassId,
MethodId: m.MethodId,
}
printEventConnectionClose(*eventConnectionClose)
emitConnectionClose(*eventConnectionClose, connectionInfo, emitter)
default:
@ -208,10 +208,14 @@ func (d dissecting) Analyze(item *api.OutputChannelItem, entryId string, resolve
entryBytes, _ := json.Marshal(item.Pair)
service := fmt.Sprintf("amqp")
var summary string
summary := ""
switch request["method"] {
case basicMethodMap[40]:
summary = reqDetails["Exchange"].(string)
break
case basicMethodMap[60]:
summary = reqDetails["Exchange"].(string)
break
case exchangeMethodMap[10]:
summary = reqDetails["Exchange"].(string)
break
@ -221,6 +225,9 @@ func (d dissecting) Analyze(item *api.OutputChannelItem, entryId string, resolve
case connectionMethodMap[10]:
summary = fmt.Sprintf("%g.%g", reqDetails["VersionMajor"].(float64), reqDetails["VersionMinor"].(float64))
break
case connectionMethodMap[50]:
summary = reqDetails["ReplyText"].(string)
break
}
return &api.MizuEntry{
@ -293,6 +300,9 @@ func (d dissecting) Represent(entry string) ([]byte, error) {
case connectionMethodMap[10]:
repRequest = representConnectionStart(details)
break
case connectionMethodMap[50]:
repRequest = representConnectionClose(details)
break
}
// response := root["response"].(map[string]interface{})["payload"].(map[string]interface{})
// repRequest := representRequest(request)