github.com/danielpfeifer02/quic-go-prio-packs@v0.41.0-28/internal/qtls/client_session_cache.go (about)

     1  package qtls
     2  
     3  import (
     4  	"crypto/tls"
     5  	"sync"
     6  )
     7  
     8  type clientSessionCache struct {
     9  	mx      sync.Mutex
    10  	getData func(earlyData bool) []byte
    11  	setData func(data []byte, earlyData bool) (allowEarlyData bool)
    12  	wrapped tls.ClientSessionCache
    13  }
    14  
    15  var _ tls.ClientSessionCache = &clientSessionCache{}
    16  
    17  func (c *clientSessionCache) Put(key string, cs *tls.ClientSessionState) {
    18  	c.mx.Lock()
    19  	defer c.mx.Unlock()
    20  
    21  	if cs == nil {
    22  		c.wrapped.Put(key, nil)
    23  		return
    24  	}
    25  	ticket, state, err := cs.ResumptionState()
    26  	if err != nil || state == nil {
    27  		c.wrapped.Put(key, cs)
    28  		return
    29  	}
    30  	state.Extra = append(state.Extra, addExtraPrefix(c.getData(state.EarlyData)))
    31  	newCS, err := tls.NewResumptionState(ticket, state)
    32  	if err != nil {
    33  		// It's not clear why this would error. Just save the original state.
    34  		c.wrapped.Put(key, cs)
    35  		return
    36  	}
    37  	c.wrapped.Put(key, newCS)
    38  }
    39  
    40  func (c *clientSessionCache) Get(key string) (*tls.ClientSessionState, bool) {
    41  	c.mx.Lock()
    42  	defer c.mx.Unlock()
    43  
    44  	cs, ok := c.wrapped.Get(key)
    45  	if !ok || cs == nil {
    46  		return cs, ok
    47  	}
    48  	ticket, state, err := cs.ResumptionState()
    49  	if err != nil {
    50  		// It's not clear why this would error.
    51  		// Remove the ticket from the session cache, so we don't run into this error over and over again
    52  		c.wrapped.Put(key, nil)
    53  		return nil, false
    54  	}
    55  	// restore QUIC transport parameters and RTT stored in state.Extra
    56  	if extra := findExtraData(state.Extra); extra != nil {
    57  		earlyData := c.setData(extra, state.EarlyData)
    58  		if state.EarlyData {
    59  			state.EarlyData = earlyData
    60  		}
    61  	}
    62  	session, err := tls.NewResumptionState(ticket, state)
    63  	if err != nil {
    64  		// It's not clear why this would error.
    65  		// Remove the ticket from the session cache, so we don't run into this error over and over again
    66  		c.wrapped.Put(key, nil)
    67  		return nil, false
    68  	}
    69  	return session, true
    70  }