Fix session key length generation

ProtonMail · Feb 29, 2024 · 31b9581 · 31b9581
1 parent 9d2beb2
commit 31b9581
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/openpgp/write.go b/openpgp/write.go
@@ -444,7 +444,13 @@ func encrypt(keyWriter io.Writer, dataWriter io.Writer, to []*Entity, signed *En
 		}
 	}
 
-	symKey := make([]byte, cipher.KeySize())
+	var symKey []byte
+	if aeadSupported {
+		symKey = make([]byte, aeadCipherSuite.Cipher.KeySize())
+	} else {
+		symKey = make([]byte, cipher.KeySize())
+	}
+
 	if _, err := io.ReadFull(config.Random(), symKey); err != nil {
 		return nil, err
 	}