1 files changed, 82 insertions, 0 deletions

diff --git a/src/test/java/org/gnunet/util/CryptoECCTest.java b/src/test/java/org/gnunet/util/CryptoECCTest.java
new file mode 100644
index 0000000..cc88409
--- /dev/null
+++ b/src/test/java/org/gnunet/util/CryptoECCTest.java
@@ -0,0 +1,82 @@
+package org.gnunet.util;
+
+
+import org.junit.Assert;
+import org.junit.Test;
+
+import java.util.Random;
+
+public class CryptoECCTest {
+    /**
+     * Check that signed messages can be verified correctly.
+     */
+    @Test
+    public void test_sign_success() {
+        Random r = new Random();
+        // the test uses random data, repeat it multiple times!
+        for (int i = 0; i < 10; i++) {
+            byte[] msg = new byte[16];
+            r.nextBytes(msg);
+
+            CryptoECC.PrivateKey privateKey = new CryptoECC.PrivateKey();
+            privateKey.d = new byte[32];
+            r.nextBytes(privateKey.d);
+            CryptoECC.PublicKey publicKey = CryptoECC.computePublicKey(privateKey);
+
+            CryptoECC.Signature sig = CryptoECC.sign(msg, privateKey, publicKey);
+
+            boolean valid = CryptoECC.verify(sig, msg, publicKey);
+
+            Assert.assertTrue(valid);
+        }
+    }
+
+    /**
+     * Check that signature verification fails for manipulated data.
+     */
+    @Test
+    public void test_sign_failure() {
+        Random r = new Random();
+        // the test uses random data, repeat it multiple times!
+        for (int i = 0; i < 10; i++) {
+            byte[] msg = new byte[16];
+            r.nextBytes(msg);
+
+            CryptoECC.PrivateKey privateKey = new CryptoECC.PrivateKey();
+            privateKey.d = new byte[32];
+            r.nextBytes(privateKey.d);
+            CryptoECC.PublicKey publicKey = CryptoECC.computePublicKey(privateKey);
+
+            CryptoECC.Signature sig = CryptoECC.sign(msg, privateKey, publicKey);
+
+            msg[0] = (byte) (msg[0] + 1);
+
+            boolean valid = CryptoECC.verify(sig, msg, publicKey);
+
+            Assert.assertFalse(valid);
+        }
+    }
+
+    /**
+     * Check whether ecdh key coincide
+     */
+    @Test
+    public void test_ecdh() {
+        Random r = new Random();
+
+        CryptoECC.PrivateKey privateAlice = new CryptoECC.PrivateKey();
+        privateAlice.d = new byte[32];
+        r.nextBytes(privateAlice.d);
+        CryptoECC.PublicKey publicAlice = CryptoECC.computePublicKey(privateAlice);
+
+        CryptoECC.PrivateKey privateBob = new CryptoECC.PrivateKey();
+        privateBob.d = new byte[32];
+        r.nextBytes(privateBob.d);
+        CryptoECC.PublicKey publicBob = CryptoECC.computePublicKey(privateBob);
+
+        HashCode ssAlice = CryptoECC.ecdh(privateAlice, publicBob);
+        HashCode ssBob = CryptoECC.ecdh(privateBob, publicAlice);
+
+        Assert.assertArrayEquals(ssAlice.data, ssBob.data);
+    }
+}