Shares the RTT estimates through the connection context

agents/handshake_agent.go

@@ -8,6 +8,7 @@ import (
 	. "github.com/QUIC-Tracker/quic-tracker"
 	"github.com/davecgh/go-spew/spew"
 	"strings"
+	"time"
 )
 
 type HandshakeStatus struct {
@@ -45,6 +46,7 @@ func (a *HandshakeAgent) Run(conn *Connection) {
 
 	firstInitialReceived := false
 	tlsCompleted := false
+	pingTimer := time.NewTimer(0)
 	var tlsPacket Packet
 
 	go func() {
@@ -89,6 +91,7 @@ func (a *HandshakeAgent) Run(conn *Connection) {
 				default:
 					a.HandshakeStatus.Submit(HandshakeStatus{false, p.(Packet), errors.New("received incorrect packet type during handshake")})
 				}
+				pingTimer.Reset(time.Duration(conn.SmoothedRTT + conn.RTTVar) * time.Microsecond)
 			case p := <-outPackets:
 				if !tlsCompleted {
 					break
@@ -115,6 +118,11 @@ func (a *HandshakeAgent) Run(conn *Connection) {
 			case i := <-socketStatus:
 				if strings.Contains(i.(error).Error(), "connection refused") {
 					a.HandshakeStatus.Submit(HandshakeStatus{false, nil , i.(error)})
+					return
+				}
+			case <-pingTimer.C:
+				if firstInitialReceived {
+					conn.PreparePacket.Submit(EncryptionLevelBest)
 				}
 			case <-conn.ConnectionRestarted:
 				incPackets = conn.IncomingPackets.RegisterNewChan(1000)

agents/rtt_agent.go

@@ -8,6 +8,7 @@ import (
 
 type RTTAgent struct {
 	BaseAgent
+	conn *Connection
 	MinRTT             uint64
 	LatestRTT          uint64
 	SmoothedRTT        uint64
@@ -25,6 +26,7 @@ type SentPacket struct {
 
 func (a *RTTAgent) Run(conn *Connection) {
 	a.Init("RTTAgent", conn.OriginalDestinationCID)
+	a.conn = conn
 	a.MinRTT = math.MaxUint64
 
 	a.SentPackets = map[PNSpace]map[PacketNumber]SentPacket{
@@ -106,5 +108,10 @@ func (a *RTTAgent) UpdateRTT(ackDelay uint64, ackOnly bool) { // TODO: https://t
 		a.RTTVar = uint64(0.75 * float64(a.RTTVar) + 0.25 * float64(RTTVarSample))
 		a.SmoothedRTT = uint64(0.875 * float64(a.SmoothedRTT) + 0.125 * float64(a.LatestRTT))
 	}
+
+	a.conn.MinRTT = a.MinRTT
+	a.conn.SmoothedRTT = a.SmoothedRTT
+	a.conn.RTTVar = a.RTTVar
+
 	a.Logger.Printf("LatestRTT = %d, MinRTT = %d, SmoothedRTT = %d, RTTVar = %d", a.LatestRTT, a.MinRTT, a.SmoothedRTT, a.RTTVar)
 }

connection.go

@@ -67,6 +67,10 @@ type Connection struct {
 	LargestPNsReceived     map[PNSpace]PacketNumber // Stores the largest PN received
 	LargestPNsAcknowledged map[PNSpace]PacketNumber // Stores the largest PN we have sent that were acknowledged by the peer
 
+	MinRTT             uint64
+	SmoothedRTT        uint64
+	RTTVar             uint64
+
 	AckQueue             map[PNSpace][]PacketNumber // Stores the packet numbers to be acked TODO: This should be a channel actually
 	Logger               *log.Logger
 }