1 files changed, 143 insertions, 0 deletions
diff --git a/libraries/spongycastle/prov/src/test/java/org/spongycastle/jce/provider/test/BaseBlockCipherTest.java b/libraries/spongycastle/prov/src/test/java/org/spongycastle/jce/provider/test/BaseBlockCipherTest.java
new file mode 100644
index 000000000..f80ec5f32
--- /dev/null
+++ b/libraries/spongycastle/prov/src/test/java/org/spongycastle/jce/provider/test/BaseBlockCipherTest.java
@@ -0,0 +1,143 @@
+package org.spongycastle.jce.provider.test;
+
+import org.spongycastle.util.encoders.Hex;
+import org.spongycastle.util.test.SimpleTest;
+import org.spongycastle.util.test.TestFailedException;
+
+import javax.crypto.Cipher;
+import javax.crypto.KeyGenerator;
+import javax.crypto.SecretKey;
+import javax.crypto.spec.IvParameterSpec;
+import javax.crypto.spec.SecretKeySpec;
+import java.security.Key;
+
+public abstract class BaseBlockCipherTest
+    extends SimpleTest
+{
+    String algorithm;
+
+    BaseBlockCipherTest(
+        String algorithm)
+    {
+        this.algorithm = algorithm;
+    }
+
+    public String getName()
+    {
+        return algorithm;
+    }
+
+    protected void oidTest(String[] oids, String[] names, int groupSize)
+        throws Exception
+    {
+        byte[] data = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16};
+        IvParameterSpec ivSpec = new IvParameterSpec(new byte[16]);
+
+        for (int i = 0; i != oids.length; i++)
+        {
+            Cipher c1 = Cipher.getInstance(oids[i], "SC");
+            Cipher c2 = Cipher.getInstance(names[i], "SC");
+            KeyGenerator kg = KeyGenerator.getInstance(oids[i], "SC");
+
+            SecretKey k = kg.generateKey();
+
+            if (names[i].indexOf("/ECB/") > 0)
+            {
+                c1.init(Cipher.ENCRYPT_MODE, k);
+                c2.init(Cipher.DECRYPT_MODE, k);
+            }
+            else
+            {
+                c1.init(Cipher.ENCRYPT_MODE, k, ivSpec);
+                c2.init(Cipher.DECRYPT_MODE, k, ivSpec);
+            }
+
+            byte[] result = c2.doFinal(c1.doFinal(data));
+
+            if (!areEqual(data, result))
+            {
+                fail("failed OID test");
+            }
+
+            if (k.getEncoded().length != (16 + ((i / groupSize) * 8)))
+            {
+                fail("failed key length test");
+            }
+        }
+    }
+
+    protected void wrapOidTest(String[] oids, String name)
+        throws Exception
+    {
+        byte[] data = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16};
+
+        for (int i = 0; i != oids.length; i++)
+        {
+            Cipher c1 = Cipher.getInstance(oids[i], "SC");
+            Cipher c2 = Cipher.getInstance(name, "SC");
+            KeyGenerator kg = KeyGenerator.getInstance(oids[i], "SC");
+
+            SecretKey k = kg.generateKey();
+
+            c1.init(Cipher.WRAP_MODE, k);
+            c2.init(Cipher.UNWRAP_MODE, k);
+
+            Key wKey = c2.unwrap(c1.wrap(new SecretKeySpec(data, algorithm)), algorithm, Cipher.SECRET_KEY);
+
+            if (!areEqual(data, wKey.getEncoded()))
+            {
+                fail("failed wrap OID test");
+            }
+
+            if (k.getEncoded().length != (16 + (i * 8)))
+            {
+                fail("failed key length test");
+            }
+        }
+    }
+
+    protected void wrapTest(
+        int     id,
+        String  wrappingAlgorithm,
+        byte[]  kek,
+        byte[]  in,
+        byte[]  out)
+        throws Exception
+    {
+        Cipher wrapper = Cipher.getInstance(wrappingAlgorithm, "SC");
+
+        wrapper.init(Cipher.WRAP_MODE, new SecretKeySpec(kek, algorithm));
+
+        try
+        {
+            byte[]  cText = wrapper.wrap(new SecretKeySpec(in, algorithm));
+            if (!areEqual(cText, out))
+            {
+                fail("failed wrap test " + id  + " expected " + new String(Hex.encode(out)) + " got " + new String(Hex.encode(cText)));
+            }
+        }
+        catch (TestFailedException e)
+        {
+            throw e;
+        }
+        catch (Exception e)
+        {
+            fail("failed wrap test exception " + e.toString(), e);
+        }
+
+        wrapper.init(Cipher.UNWRAP_MODE, new SecretKeySpec(kek, algorithm));
+
+        try
+        {
+            Key  pText = wrapper.unwrap(out, algorithm, Cipher.SECRET_KEY);
+            if (!areEqual(pText.getEncoded(), in))
+            {
+                fail("failed unwrap test " + id  + " expected " + new String(Hex.encode(in)) + " got " + new String(Hex.encode(pText.getEncoded())));
+            }
+        }
+        catch (Exception e)
+        {
+            fail("failed unwrap test exception " + e.toString(), e);
+        }
+    }
+}