mirrors/uquic
1
0
Fork 0
mirror of https://github.com/refraction-networking/uquic.git synced 2025-04-04 12:47:36 +03:00
Code Issues Projects Releases Packages Wiki Activity

speed up updatableAEAD.Overhead()

Browse source
This commit is contained in:
Marten Seemann 2019-08-20 13:47:36 +07:00
parent 2133d01956
commit fd62eb67ee
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
internal/handshake/updatable_aead.go
View file

@ -59,6 +59,8 @@ type updatableAEAD struct {
numSentWithCurrentKey uint64 numSentWithCurrentKey uint64
rcvAEAD cipher.AEAD rcvAEAD cipher.AEAD
sendAEAD cipher.AEAD sendAEAD cipher.AEAD
// caches cipher.AEAD.Overhead(). This speeds up calls to Overhead().
aeadOverhead int
nextRcvAEAD cipher.AEAD nextRcvAEAD cipher.AEAD
nextSendAEAD cipher.AEAD nextSendAEAD cipher.AEAD
@ -120,6 +122,7 @@ func (a *updatableAEAD) SetReadKey(suite cipherSuite, trafficSecret []byte) {
if a.suite == nil { if a.suite == nil {
a.nonceBuf = make([]byte, a.rcvAEAD.NonceSize()) a.nonceBuf = make([]byte, a.rcvAEAD.NonceSize())
a.hpMask = make([]byte, a.hpDecrypter.BlockSize()) a.hpMask = make([]byte, a.hpDecrypter.BlockSize())
a.aeadOverhead = a.rcvAEAD.Overhead()
a.suite = suite a.suite = suite
} }
@ -135,6 +138,7 @@ func (a *updatableAEAD) SetWriteKey(suite cipherSuite, trafficSecret []byte) {
if a.suite == nil { if a.suite == nil {
a.nonceBuf = make([]byte, a.sendAEAD.NonceSize()) a.nonceBuf = make([]byte, a.sendAEAD.NonceSize())
a.hpMask = make([]byte, a.hpEncrypter.BlockSize()) a.hpMask = make([]byte, a.hpEncrypter.BlockSize())
a.aeadOverhead = a.sendAEAD.Overhead()
a.suite = suite a.suite = suite
} }
@ -238,7 +242,7 @@ func (a *updatableAEAD) KeyPhase() protocol.KeyPhaseBit {
} }
func (a *updatableAEAD) Overhead() int { func (a *updatableAEAD) Overhead() int {
return a.sendAEAD.Overhead() return a.aeadOverhead
} }
func (a *updatableAEAD) EncryptHeader(sample []byte, firstByte *byte, pnBytes []byte) { func (a *updatableAEAD) EncryptHeader(sample []byte, firstByte *byte, pnBytes []byte) {