From a57b9c70aad09d3c61d9cca8cbf8f210a4d0023c Mon Sep 17 00:00:00 2001
From: Marten Seemann <martenseemann@gmail.com>
Date: Tue, 20 Aug 2019 13:36:40 +0700
Subject: [PATCH] speed up header encryption / decryption

---
 internal/handshake/updatable_aead.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/internal/handshake/updatable_aead.go b/internal/handshake/updatable_aead.go
index ddc463b9..c573037a 100644
--- a/internal/handshake/updatable_aead.go
+++ b/internal/handshake/updatable_aead.go
@@ -242,7 +242,7 @@ func (a *updatableAEAD) Overhead() int {
 }
 
 func (a *updatableAEAD) EncryptHeader(sample []byte, firstByte *byte, pnBytes []byte) {
-	if len(sample) != a.hpEncrypter.BlockSize() {
+	if len(sample) != len(a.hpMask) {
 		panic("invalid sample size")
 	}
 	a.hpEncrypter.Encrypt(a.hpMask, sample)
@@ -253,7 +253,7 @@ func (a *updatableAEAD) EncryptHeader(sample []byte, firstByte *byte, pnBytes []
 }
 
 func (a *updatableAEAD) DecryptHeader(sample []byte, firstByte *byte, pnBytes []byte) {
-	if len(sample) != a.hpDecrypter.BlockSize() {
+	if len(sample) != len(a.hpMask) {
 		panic("invalid sample size")
 	}
 	a.hpDecrypter.Encrypt(a.hpMask, sample)