Use the modified tlsEmitter

This commit is contained in:
M. Mert Yildiran
2022-04-26 17:02:35 +03:00
parent 9aed5b2941
commit 6e9064dc56
3 changed files with 9 additions and 4 deletions

View File

@@ -429,6 +429,7 @@ type TcpReader interface {
GetExtension() *Extension
SetTcpID(tcpID *TcpID)
SetCaptureTime(captureTime time.Time)
SetEmitter(emitter Emitter)
}
type TcpStream interface {

View File

@@ -155,3 +155,7 @@ func (reader *tcpReader) SetTcpID(tcpID *api.TcpID) {
func (reader *tcpReader) SetCaptureTime(captureTime time.Time) {
reader.captureTime = captureTime
}
func (reader *tcpReader) SetEmitter(emitter api.Emitter) {
reader.emitter = emitter
}

View File

@@ -173,17 +173,17 @@ func (p *tlsPoller) startNewTlsReader(chunk *tlsChunk, ip net.IP, port uint16, k
tcpid := p.buildTcpId(chunk, ip, port)
tcpReader.SetTcpID(&tcpid)
tlsEmitter := &tlsEmitter{
tcpReader.SetEmitter(&tlsEmitter{
delegate: tcpReader.GetEmitter(),
namespace: p.getNamespace(chunk.Pid),
}
})
go dissect(extension, reader, tcpReader, tlsEmitter, options)
go dissect(extension, reader, tcpReader, options)
return reader
}
func dissect(extension *api.Extension, reader *tlsReader, tcpReader api.TcpReader,
tlsEmitter *tlsEmitter, options *shared.TrafficFilteringOptions) {
options *shared.TrafficFilteringOptions) {
b := bufio.NewReader(reader)
err := extension.Dissector.Dissect(b, tcpReader, options)