github/kubeshark
1
0
Fork 0
mirror of https://github.com/kubeshark/kubeshark.git synced 2025-07-29 15:37:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Set ConnectionInfo in HTTP handlers

Browse Source
This commit is contained in:
M. Mert Yildiran 2021-08-18 11:14:29 +03:00
parent 902b6bff87
commit d7c3b04bcc
No known key found for this signature in database
GPG Key ID: D42ADB236521BF7A
1 changed files with 32 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
tap/extensions/http/handlers.go
View File

@ -33,6 +33,15 @@ func handleHTTP2Stream(grpcAssembler *GrpcAssembler, tcpID *api.TcpID, emitter a
streamID, streamID,
) )
item = reqResMatcher.registerRequest(ident, &messageHTTP1, time.Now()) item = reqResMatcher.registerRequest(ident, &messageHTTP1, time.Now())
if item != nil {
item.ConnectionInfo = &api.ConnectionInfo{
ClientIP: tcpID.SrcIP,
ClientPort: tcpID.SrcPort,
ServerIP: tcpID.DstIP,
ServerPort: tcpID.DstPort,
IsOutgoing: true,
}
}
case http.Response: case http.Response:
responseCounter++ responseCounter++
ident := fmt.Sprintf( ident := fmt.Sprintf(
@ -44,6 +53,15 @@ func handleHTTP2Stream(grpcAssembler *GrpcAssembler, tcpID *api.TcpID, emitter a
streamID, streamID,
) )
item = reqResMatcher.registerResponse(ident, &messageHTTP1, time.Now()) item = reqResMatcher.registerResponse(ident, &messageHTTP1, time.Now())
if item != nil {
item.ConnectionInfo = &api.ConnectionInfo{
ClientIP: tcpID.DstIP,
ClientPort: tcpID.DstPort,
ServerIP: tcpID.SrcIP,
ServerPort: tcpID.SrcPort,
IsOutgoing: false,
}
}
} }
if item != nil { if item != nil {
@ -83,6 +101,13 @@ func handleHTTP1ClientStream(b *bufio.Reader, tcpID *api.TcpID, emitter api.Emit
) )
item := reqResMatcher.registerRequest(ident, req, time.Now()) item := reqResMatcher.registerRequest(ident, req, time.Now())
if item != nil { if item != nil {
item.ConnectionInfo = &api.ConnectionInfo{
ClientIP: tcpID.SrcIP,
ClientPort: tcpID.SrcPort,
ServerIP: tcpID.DstIP,
ServerPort: tcpID.DstPort,
IsOutgoing: true,
}
emitter.Emit(item) emitter.Emit(item)
} }
return nil return nil
@ -128,6 +153,13 @@ func handleHTTP1ServerStream(b *bufio.Reader, tcpID *api.TcpID, emitter api.Emit
) )
item := reqResMatcher.registerResponse(ident, res, time.Now()) item := reqResMatcher.registerResponse(ident, res, time.Now())
if item != nil { if item != nil {
item.ConnectionInfo = &api.ConnectionInfo{
ClientIP: tcpID.DstIP,
ClientPort: tcpID.DstPort,
ServerIP: tcpID.SrcIP,
ServerPort: tcpID.SrcPort,
IsOutgoing: false,
}
emitter.Emit(item) emitter.Emit(item)
} }
return nil return nil