Add session key size check (#62)

* Add session key size check

Co-authored-by: Daniel Huigens <d.huigens@protonmail.com>
This commit is contained in:
wussler 2020-07-20 11:43:36 +02:00 committed by GitHub
parent 3b2e53c586
commit 8c04ff64a5
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 50 additions and 4 deletions

View file

@ -81,6 +81,25 @@ func TestSymmetricKeyPacket(t *testing.T) {
assert.Exactly(t, testSessionKey, outputSymmetricKey)
}
func TestSymmetricKeyPacketWrongSize(t *testing.T) {
r, err := RandomToken(symKeyAlgos[constants.AES256].KeySize())
if err != nil {
t.Fatal("Expected no error while generating session key, got:", err)
}
sk := &SessionKey{
Key: r,
Algo: constants.AES128,
}
password := []byte("I like encryption")
_, err = EncryptSessionKeyWithPassword(sk, password)
if err == nil {
t.Fatal("Expected error while generating key packet with wrong sized key")
}
}
func TestDataPacketEncryption(t *testing.T) {
var message = NewPlainMessageFromString("The secret code is... 1, 2, 3, 4, 5")