1 files changed, 142 insertions, 0 deletions
diff --git a/pg/src/main/java/org/spongycastle/openpgp/operator/jcajce/JcePBEKeyEncryptionMethodGenerator.java b/pg/src/main/java/org/spongycastle/openpgp/operator/jcajce/JcePBEKeyEncryptionMethodGenerator.java
new file mode 100644
index 00000000..510ca578
--- /dev/null
+++ b/pg/src/main/java/org/spongycastle/openpgp/operator/jcajce/JcePBEKeyEncryptionMethodGenerator.java
@@ -0,0 +1,142 @@
+package org.spongycastle.openpgp.operator.jcajce;
+
+import java.security.InvalidAlgorithmParameterException;
+import java.security.InvalidKeyException;
+import java.security.Provider;
+import java.security.SecureRandom;
+
+import javax.crypto.BadPaddingException;
+import javax.crypto.Cipher;
+import javax.crypto.IllegalBlockSizeException;
+import javax.crypto.SecretKey;
+import javax.crypto.spec.IvParameterSpec;
+import javax.crypto.spec.SecretKeySpec;
+
+import org.spongycastle.bcpg.S2K;
+import org.spongycastle.jcajce.util.DefaultJcaJceHelper;
+import org.spongycastle.jcajce.util.NamedJcaJceHelper;
+import org.spongycastle.jcajce.util.ProviderJcaJceHelper;
+import org.spongycastle.openpgp.PGPException;
+import org.spongycastle.openpgp.operator.PBEKeyEncryptionMethodGenerator;
+import org.spongycastle.openpgp.operator.PGPDigestCalculator;
+
+/**
+ * JCE based generator for password based encryption (PBE) data protection methods.
+ */
+public class JcePBEKeyEncryptionMethodGenerator
+    extends PBEKeyEncryptionMethodGenerator
+{
+    private OperatorHelper helper = new OperatorHelper(new DefaultJcaJceHelper());
+
+    /**
+     * Create a PBE encryption method generator using the provided digest and the default S2K count
+     * for key generation.
+     *
+     * @param passPhrase the passphrase to use as the primary source of key material.
+     * @param s2kDigestCalculator the digest calculator to use for key calculation.
+     */
+    public JcePBEKeyEncryptionMethodGenerator(char[] passPhrase, PGPDigestCalculator s2kDigestCalculator)
+    {
+        super(passPhrase, s2kDigestCalculator);
+    }
+
+    /**
+     * Create a PBE encryption method generator using the default SHA-1 digest and the default S2K
+     * count for key generation.
+     *
+     * @param passPhrase the passphrase to use as the primary source of key material.
+     */
+    public JcePBEKeyEncryptionMethodGenerator(char[] passPhrase)
+    {
+        this(passPhrase, new SHA1PGPDigestCalculator());
+    }
+
+    /**
+     * Create a PBE encryption method generator using the provided calculator and S2K count for key
+     * generation.
+     *
+     * @param passPhrase the passphrase to use as the primary source of key material.
+     * @param s2kDigestCalculator the digest calculator to use for key calculation.
+     * @param s2kCount the single byte {@link S2K} count to use.
+     */
+    public JcePBEKeyEncryptionMethodGenerator(char[] passPhrase, PGPDigestCalculator s2kDigestCalculator, int s2kCount)
+    {
+        super(passPhrase, s2kDigestCalculator, s2kCount);
+    }
+
+    /**
+     * Create a PBE encryption method generator using the default SHA-1 digest calculator and a S2K
+     * count other than the default for key generation.
+     *
+     * @param passPhrase the passphrase to use as the primary source of key material.
+     * @param s2kCount the single byte {@link S2K} count to use.
+     */
+    public JcePBEKeyEncryptionMethodGenerator(char[] passPhrase, int s2kCount)
+    {
+        super(passPhrase, new SHA1PGPDigestCalculator(), s2kCount);
+    }
+
+    /**
+     * Sets the JCE provider to source cryptographic primitives from.
+     *
+     * @param provider the JCE provider to use.
+     * @return the current generator.
+     */
+    public JcePBEKeyEncryptionMethodGenerator setProvider(Provider provider)
+    {
+        this.helper = new OperatorHelper(new ProviderJcaJceHelper(provider));
+
+        return this;
+    }
+
+    /**
+     * Sets the JCE provider to source cryptographic primitives from.
+     *
+     * @param providerName the name of the JCE provider to use.
+     * @return the current generator.
+     */
+    public JcePBEKeyEncryptionMethodGenerator setProvider(String providerName)
+    {
+        this.helper = new OperatorHelper(new NamedJcaJceHelper(providerName));
+
+        return this;
+    }
+
+    public PBEKeyEncryptionMethodGenerator setSecureRandom(SecureRandom random)
+    {
+        super.setSecureRandom(random);
+
+        return this;
+    }
+
+    protected byte[] encryptSessionInfo(int encAlgorithm, byte[] key, byte[] sessionInfo)
+        throws PGPException
+    {
+        try
+        {
+            String cName = PGPUtil.getSymmetricCipherName(encAlgorithm);
+            Cipher c = helper.createCipher(cName + "/CFB/NoPadding");
+            SecretKey sKey = new SecretKeySpec(key, PGPUtil.getSymmetricCipherName(encAlgorithm));
+
+            c.init(Cipher.ENCRYPT_MODE, sKey, new IvParameterSpec(new byte[c.getBlockSize()]));
+
+            return c.doFinal(sessionInfo, 0, sessionInfo.length);
+        }
+        catch (IllegalBlockSizeException e)
+        {
+            throw new PGPException("illegal block size: " + e.getMessage(), e);
+        }
+        catch (BadPaddingException e)
+        {
+            throw new PGPException("bad padding: " + e.getMessage(), e);
+        }
+        catch (InvalidAlgorithmParameterException e)
+        {
+            throw new PGPException("IV invalid: " + e.getMessage(), e);
+        }
+        catch (InvalidKeyException e)
+        {
+            throw new PGPException("key invalid: " + e.getMessage(), e);
+        }
+    }
+}