From b5c665b60236b7868833b2c1d31a7510e1b54ddc Mon Sep 17 00:00:00 2001
From: David Levanon <dvdlevanon@gmail.com>
Date: Tue, 26 Apr 2022 14:37:44 +0300
Subject: [PATCH] set capture time for every packet, so long living readers
 would be accurate (#1043)

---
 tap/tlstapper/tls_poller.go | 10 +++++-----
 tap/tlstapper/tls_reader.go |  1 +
 2 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/tap/tlstapper/tls_poller.go b/tap/tlstapper/tls_poller.go
index 7b521a9a9..dc8d103d3 100644
--- a/tap/tlstapper/tls_poller.go
+++ b/tap/tlstapper/tls_poller.go
@@ -131,6 +131,7 @@ func (p *tlsPoller) handleTlsChunk(chunk *tlsChunk, extension *api.Extension,
 		p.readers[key] = reader
 	}
 
+	reader.timer.CaptureTime = time.Now()
 	reader.chunks <- chunk
 
 	if os.Getenv("MIZU_VERBOSE_TLS_TAPPER") == "true" {
@@ -150,6 +151,9 @@ func (p *tlsPoller) startNewTlsReader(chunk *tlsChunk, ip net.IP, port uint16, k
 			p.closeReader(key, r)
 		},
 		progress: &api.ReadProgress{},
+		timer: api.SuperTimer{
+			CaptureTime: time.Now(),
+		},
 	}
 
 	tcpid := p.buildTcpId(chunk, ip, port)
@@ -166,13 +170,9 @@ func (p *tlsPoller) startNewTlsReader(chunk *tlsChunk, ip net.IP, port uint16, k
 func dissect(extension *api.Extension, reader *tlsReader, isRequest bool, tcpid *api.TcpID,
 	tlsEmitter *tlsEmitter, options *api.TrafficFilteringOptions, reqResMatcher api.RequestResponseMatcher) {
 	b := bufio.NewReader(reader)
-	
-	timer := api.SuperTimer{
-		CaptureTime: time.Now(),
-	}
 
 	err := extension.Dissector.Dissect(b, reader.progress, api.Ebpf, isRequest, tcpid, &api.CounterPair{},
-		&timer, &api.SuperIdentifier{}, tlsEmitter, options, reqResMatcher)
+		&reader.timer, &api.SuperIdentifier{}, tlsEmitter, options, reqResMatcher)
 
 	if err != nil {
 		logger.Log.Warningf("Error dissecting TLS %v - %v", tcpid, err)
diff --git a/tap/tlstapper/tls_reader.go b/tap/tlstapper/tls_reader.go
index 908cbf1ad..67a1fae50 100644
--- a/tap/tlstapper/tls_reader.go
+++ b/tap/tlstapper/tls_reader.go
@@ -13,6 +13,7 @@ type tlsReader struct {
 	data        []byte
 	doneHandler func(r *tlsReader)
 	progress    *api.ReadProgress
+	timer        api.SuperTimer
 }
 
 func (r *tlsReader) Read(p []byte) (int, error) {