diff --git a/tap/tlstapper/golang_connection.go b/tap/tlstapper/golang_connection.go index 419c2f161..9c4eb1282 100644 --- a/tap/tlstapper/golang_connection.go +++ b/tap/tlstapper/golang_connection.go @@ -17,12 +17,15 @@ func NewGolangConnection(pid uint32, connAddr uint32, extension *api.Extension, stream := &tlsStream{} counterPair := &api.CounterPair{} reqResMatcher := extension.Dissector.NewResponseRequestMatcher() + clientReader := NewGolangReader(extension, true, emitter, counterPair, stream, reqResMatcher) + serverReader := NewGolangReader(extension, false, emitter, counterPair, stream, reqResMatcher) + stream.reader = clientReader return &golangConnection{ Pid: pid, ConnAddr: connAddr, Stream: stream, - ClientReader: NewGolangReader(extension, true, emitter, counterPair, stream, reqResMatcher), - ServerReader: NewGolangReader(extension, false, emitter, counterPair, stream, reqResMatcher), + ClientReader: clientReader, + ServerReader: serverReader, } } diff --git a/tap/tlstapper/tls_stream.go b/tap/tlstapper/tls_stream.go index d0077ba59..ff966f763 100644 --- a/tap/tlstapper/tls_stream.go +++ b/tap/tlstapper/tls_stream.go @@ -3,7 +3,7 @@ package tlstapper import "github.com/up9inc/mizu/tap/api" type tlsStream struct { - reader *tlsReader + reader api.TcpReader protocol *api.Protocol } @@ -16,7 +16,7 @@ func (t *tlsStream) SetProtocol(protocol *api.Protocol) { } func (t *tlsStream) GetReqResMatchers() []api.RequestResponseMatcher { - return []api.RequestResponseMatcher{t.reader.reqResMatcher} + return []api.RequestResponseMatcher{t.reader.GetReqResMatcher()} } func (t *tlsStream) GetIsTapTarget() bool {