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 GetExtension() *Extension
SetTcpID(tcpID *TcpID) SetTcpID(tcpID *TcpID)
SetCaptureTime(captureTime time.Time) SetCaptureTime(captureTime time.Time)
SetEmitter(emitter Emitter)
} }
type TcpStream interface { type TcpStream interface {

View File

@@ -155,3 +155,7 @@ func (reader *tcpReader) SetTcpID(tcpID *api.TcpID) {
func (reader *tcpReader) SetCaptureTime(captureTime time.Time) { func (reader *tcpReader) SetCaptureTime(captureTime time.Time) {
reader.captureTime = captureTime 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) tcpid := p.buildTcpId(chunk, ip, port)
tcpReader.SetTcpID(&tcpid) tcpReader.SetTcpID(&tcpid)
tlsEmitter := &tlsEmitter{ tcpReader.SetEmitter(&tlsEmitter{
delegate: tcpReader.GetEmitter(), delegate: tcpReader.GetEmitter(),
namespace: p.getNamespace(chunk.Pid), namespace: p.getNamespace(chunk.Pid),
} })
go dissect(extension, reader, tcpReader, tlsEmitter, options) go dissect(extension, reader, tcpReader, options)
return reader return reader
} }
func dissect(extension *api.Extension, reader *tlsReader, tcpReader api.TcpReader, func dissect(extension *api.Extension, reader *tlsReader, tcpReader api.TcpReader,
tlsEmitter *tlsEmitter, options *shared.TrafficFilteringOptions) { options *shared.TrafficFilteringOptions) {
b := bufio.NewReader(reader) b := bufio.NewReader(reader)
err := extension.Dissector.Dissect(b, tcpReader, options) err := extension.Dissector.Dissect(b, tcpReader, options)