Make PGPEncryptedDataList.addSessionKeyDecryptionMethod() return PGPSessionKeyEncryptedData object

Author: vanitasvitae

pg/src/main/java/org/bouncycastle/openpgp/PGPEncryptedDataList.java

@@ -134,11 +134,13 @@ public PGPEncryptedDataList(
      * session key.
      *
      * @param sessionKey session key for message decryption
+     * @return session key encrypted data
      */
-    public void addSessionKeyDecryptionMethod(PGPSessionKey sessionKey)
+    public PGPSessionKeyEncryptedData addSessionKeyDecryptionMethod(PGPSessionKey sessionKey)
     {
         PGPSessionKeyEncryptedData sessionKeyEncryptedData = new PGPSessionKeyEncryptedData(sessionKey, data);
         methods.add(sessionKeyEncryptedData);
+        return sessionKeyEncryptedData;
     }
 
     /**

pg/src/test/java/org/bouncycastle/openpgp/test/PGPSessionKeyTest.java

@@ -291,11 +291,11 @@ private void decryptMessageWithoutEskUsingSessionKey()
         PGPEncryptedDataList encryptedData = (PGPEncryptedDataList) objectFactory.nextObject();
         isEquals(0, encryptedData.size()); // there is no encrypted session key packet
 
-        encryptedData.addSessionKeyDecryptionMethod(sessionKey); // Add decryption method using a session key
+        // Add decryption method using a session key
+        PGPSessionKeyEncryptedData sessionKeyEncData = encryptedData.addSessionKeyDecryptionMethod(sessionKey);
         isEquals(1, encryptedData.size());
 
         SessionKeyDataDecryptorFactory decryptorFactory = new BcSessionKeyDataDecryptorFactory(sessionKey);
-        PGPSessionKeyEncryptedData sessionKeyEncData = (PGPSessionKeyEncryptedData) encryptedData.get(0);
         InputStream decrypted = sessionKeyEncData.getDataStream(decryptorFactory);
 
         objectFactory = new BcPGPObjectFactory(decrypted);