1 files changed, 146 insertions, 0 deletions
diff --git a/tests/src/main/java/org/whispersystems/libaxolotl/groups/GroupCipherTest.java b/tests/src/main/java/org/whispersystems/libaxolotl/groups/GroupCipherTest.java
new file mode 100644
index 00000000..7c9892b0
--- /dev/null
+++ b/tests/src/main/java/org/whispersystems/libaxolotl/groups/GroupCipherTest.java
@@ -0,0 +1,146 @@
+package org.whispersystems.libaxolotl.groups;
+
+import junit.framework.TestCase;
+
+import org.whispersystems.libaxolotl.DuplicateMessageException;
+import org.whispersystems.libaxolotl.InvalidMessageException;
+import org.whispersystems.libaxolotl.LegacyMessageException;
+import org.whispersystems.libaxolotl.NoSessionException;
+import org.whispersystems.libaxolotl.ecc.ECKeyPair;
+import org.whispersystems.libaxolotl.protocol.SenderKeyDistributionMessage;
+import org.whispersystems.libaxolotl.util.KeyHelper;
+
+import java.security.NoSuchAlgorithmException;
+import java.security.SecureRandom;
+import java.util.ArrayList;
+
+public class GroupCipherTest extends TestCase {
+
+  public void testBasicEncryptDecrypt()
+      throws LegacyMessageException, DuplicateMessageException, InvalidMessageException, NoSessionException
+  {
+    InMemorySenderKeyStore aliceStore = new InMemorySenderKeyStore();
+    InMemorySenderKeyStore bobStore   = new InMemorySenderKeyStore();
+
+    GroupSessionBuilder aliceSessionBuilder = new GroupSessionBuilder(aliceStore);
+    GroupSessionBuilder bobSessionBuilder   = new GroupSessionBuilder(bobStore);
+
+    GroupCipher aliceGroupCipher = new GroupCipher(aliceStore, "groupWithBobInIt");
+    GroupCipher bobGroupCipher   = new GroupCipher(bobStore, "groupWithBobInIt::aliceUserName");
+
+    byte[]    aliceSenderKey        = KeyHelper.generateSenderKey();
+    ECKeyPair aliceSenderSigningKey = KeyHelper.generateSenderSigningKey();
+    int       aliceSenderKeyId      = KeyHelper.generateSenderKeyId();
+
+    SenderKeyDistributionMessage aliceDistributionMessage =
+        aliceSessionBuilder.process("groupWithBobInIt", aliceSenderKeyId, 0,
+                                    aliceSenderKey, aliceSenderSigningKey);
+
+    bobSessionBuilder.process("groupWithBobInIt::aliceUserName", aliceDistributionMessage);
+
+    byte[] ciphertextFromAlice = aliceGroupCipher.encrypt("smert ze smert".getBytes());
+    byte[] plaintextFromAlice  = bobGroupCipher.decrypt(ciphertextFromAlice);
+
+    assertTrue(new String(plaintextFromAlice).equals("smert ze smert"));
+  }
+
+  public void testBasicRatchet()
+      throws LegacyMessageException, DuplicateMessageException, InvalidMessageException, NoSessionException
+  {
+    InMemorySenderKeyStore aliceStore = new InMemorySenderKeyStore();
+    InMemorySenderKeyStore bobStore   = new InMemorySenderKeyStore();
+
+    GroupSessionBuilder aliceSessionBuilder = new GroupSessionBuilder(aliceStore);
+    GroupSessionBuilder bobSessionBuilder   = new GroupSessionBuilder(bobStore);
+
+    GroupCipher aliceGroupCipher = new GroupCipher(aliceStore, "groupWithBobInIt");
+    GroupCipher bobGroupCipher   = new GroupCipher(bobStore, "groupWithBobInIt::aliceUserName");
+
+    byte[]    aliceSenderKey        = KeyHelper.generateSenderKey();
+    ECKeyPair aliceSenderSigningKey = KeyHelper.generateSenderSigningKey();
+    int       aliceSenderKeyId      = KeyHelper.generateSenderKeyId();
+
+    SenderKeyDistributionMessage aliceDistributionMessage =
+        aliceSessionBuilder.process("groupWithBobInIt", aliceSenderKeyId, 0,
+                                    aliceSenderKey, aliceSenderSigningKey);
+
+    bobSessionBuilder.process("groupWithBobInIt::aliceUserName", aliceDistributionMessage);
+
+    byte[] ciphertextFromAlice  = aliceGroupCipher.encrypt("smert ze smert".getBytes());
+    byte[] ciphertextFromAlice2 = aliceGroupCipher.encrypt("smert ze smert2".getBytes());
+    byte[] ciphertextFromAlice3 = aliceGroupCipher.encrypt("smert ze smert3".getBytes());
+
+    byte[] plaintextFromAlice   = bobGroupCipher.decrypt(ciphertextFromAlice);
+
+    try {
+      bobGroupCipher.decrypt(ciphertextFromAlice);
+      throw new AssertionError("Should have ratcheted forward!");
+    } catch (DuplicateMessageException dme) {
+      // good
+    }
+
+    byte[] plaintextFromAlice2  = bobGroupCipher.decrypt(ciphertextFromAlice2);
+    byte[] plaintextFromAlice3  = bobGroupCipher.decrypt(ciphertextFromAlice3);
+
+    assertTrue(new String(plaintextFromAlice).equals("smert ze smert"));
+    assertTrue(new String(plaintextFromAlice2).equals("smert ze smert2"));
+    assertTrue(new String(plaintextFromAlice3).equals("smert ze smert3"));
+  }
+
+  public void testOutOfOrder()
+      throws LegacyMessageException, DuplicateMessageException, InvalidMessageException, NoSessionException
+  {
+    InMemorySenderKeyStore aliceStore = new InMemorySenderKeyStore();
+    InMemorySenderKeyStore bobStore   = new InMemorySenderKeyStore();
+
+    GroupSessionBuilder aliceSessionBuilder = new GroupSessionBuilder(aliceStore);
+    GroupSessionBuilder bobSessionBuilder   = new GroupSessionBuilder(bobStore);
+
+    GroupCipher aliceGroupCipher = new GroupCipher(aliceStore, "groupWithBobInIt");
+    GroupCipher bobGroupCipher   = new GroupCipher(bobStore, "groupWithBobInIt::aliceUserName");
+
+    byte[]    aliceSenderKey        = KeyHelper.generateSenderKey();
+    ECKeyPair aliceSenderSigningKey = KeyHelper.generateSenderSigningKey();
+    int       aliceSenderKeyId      = KeyHelper.generateSenderKeyId();
+
+    SenderKeyDistributionMessage aliceDistributionMessage =
+        aliceSessionBuilder.process("groupWithBobInIt", aliceSenderKeyId, 0,
+                                    aliceSenderKey, aliceSenderSigningKey);
+
+    bobSessionBuilder.process("groupWithBobInIt::aliceUserName", aliceDistributionMessage);
+
+
+    ArrayList<byte[]> ciphertexts = new ArrayList<>(100);
+
+    for (int i=0;i<100;i++) {
+      ciphertexts.add(aliceGroupCipher.encrypt("up the punks".getBytes()));
+    }
+
+    while (ciphertexts.size() > 0) {
+      int    index      = randomInt() % ciphertexts.size();
+      byte[] ciphertext = ciphertexts.remove(index);
+      byte[] plaintext  = bobGroupCipher.decrypt(ciphertext);
+
+      assertTrue(new String(plaintext).equals("up the punks"));
+    }
+  }
+
+  public void testEncryptNoSession() {
+    InMemorySenderKeyStore aliceStore = new InMemorySenderKeyStore();
+    GroupCipher aliceGroupCipher = new GroupCipher(aliceStore, "groupWithBobInIt");
+    try {
+      aliceGroupCipher.encrypt("up the punks".getBytes());
+      throw new AssertionError("Should have failed!");
+    } catch (NoSessionException nse) {
+      // good
+    }
+  }
+
+  private int randomInt() {
+    try {
+      return SecureRandom.getInstance("SHA1PRNG").nextInt(Integer.MAX_VALUE);
+    } catch (NoSuchAlgorithmException e) {
+      throw new AssertionError(e);
+    }
+  }
+}