test(shiftLeft): add tests for shift left

xhanulik · Mar 24, 2024 · f3a6385 · f3a6385
1 parent 85cfc67
commit f3a6385
Showing 1 changed file with 128 additions and 0 deletions.
diff --git a/applet/src/test/java/tests/BigNatInternal/ShiftLeftTest.java b/applet/src/test/java/tests/BigNatInternal/ShiftLeftTest.java
@@ -0,0 +1,128 @@
+package tests.BigNatInternal;
+
+import javacard.framework.ISOException;
+import javacard.framework.JCSystem;
+import opencrypto.jcmathlib.BigNat;
+import opencrypto.jcmathlib.ResourceManager;
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.Test;
+
+public class ShiftLeftTest {
+    @Test
+    public void shiftLeft_0() {
+        ResourceManager rm = new ResourceManager((short) 256);
+        byte memoryType = JCSystem.MEMORY_TYPE_TRANSIENT_RESET;
+        BigNat bn1 = new BigNat((short) 10, memoryType, rm);
+
+        byte[] data1 = {0x05, 0x08};
+        bn1.fromByteArray(data1, (short) 0, (short) data1.length);
+        bn1.ctShiftLeft((short) 0);
+
+        byte[] expectedResult = {0x05, 0x08};
+        byte[] actualResult = new byte[2];
+        bn1.copyToByteArray(actualResult, (short) 0);
+        Assertions.assertArrayEquals(expectedResult, actualResult);
+    }
+
+    @Test
+    public void shiftLeft_1() {
+        ResourceManager rm = new ResourceManager((short) 256);
+        byte memoryType = JCSystem.MEMORY_TYPE_TRANSIENT_RESET;
+        BigNat bn1 = new BigNat((short) 10, memoryType, rm);
+
+        byte[] data1 = {0x05, 0x08};
+        bn1.fromByteArray(data1, (short) 0, (short) data1.length);
+        bn1.ctShiftLeft((short) 1);
+
+        byte[] expectedResult = {0x0A, 0x10};
+        byte[] actualResult = new byte[2];
+        bn1.copyToByteArray(actualResult, (short) 0);
+        Assertions.assertArrayEquals(expectedResult, actualResult);
+    }
+
+    @Test
+    public void shiftLeft_5() {
+        ResourceManager rm = new ResourceManager((short) 256);
+        byte memoryType = JCSystem.MEMORY_TYPE_TRANSIENT_RESET;
+        BigNat bn1 = new BigNat((short) 10, memoryType, rm);
+
+        byte[] data1 = {0x05, 0x08};
+        bn1.fromByteArray(data1, (short) 0, (short) data1.length);
+        bn1.shiftLeft((short) 5);
+
+        byte[] expectedResult = {(byte) 0xA1, (byte) 0x00};
+        byte[] actualResult = new byte[2];
+        bn1.copyToByteArray(actualResult, (short) 0);
+        Assertions.assertArrayEquals(expectedResult, actualResult);
+    }
+
+    @Test
+    public void shiftLeft_7() {
+        ResourceManager rm = new ResourceManager((short) 256);
+        byte memoryType = JCSystem.MEMORY_TYPE_TRANSIENT_RESET;
+        BigNat bn1 = new BigNat((short) 10, memoryType, rm);
+
+        byte[] data1 = {0x05, 0x08};
+        bn1.fromByteArray(data1, (short) 0, (short) data1.length);
+        bn1.shiftLeft((short) 7);
+
+        byte[] expectedResult = {(byte) 0x02, (byte) 0x84, (byte) 0x00};
+        byte[] actualResult = new byte[3];
+        bn1.copyToByteArray(actualResult, (short) 0);
+        Assertions.assertArrayEquals(expectedResult, actualResult);
+    }
+
+    @Test
+    public void shiftLeft_carry1() {
+        ResourceManager rm = new ResourceManager((short) 256);
+        byte memoryType = JCSystem.MEMORY_TYPE_TRANSIENT_RESET;
+        BigNat bn1 = new BigNat((short) 10, memoryType, rm);
+
+        byte[] data1 = {0x00, 0x01};
+        bn1.fromByteArray(data1, (short) 0, (short) data1.length);
+        bn1.ctShiftLeft((short) 1, (short) 1);
+
+        byte[] expectedResult = {0x00, 0x03};
+        byte[] actualResult = new byte[2];
+        bn1.copyToByteArray(actualResult, (short) 0);
+        Assertions.assertArrayEquals(expectedResult, actualResult);
+    }
+
+    @Test
+    public void shiftLeft_carry128() {
+        ResourceManager rm = new ResourceManager((short) 256);
+        byte memoryType = JCSystem.MEMORY_TYPE_TRANSIENT_RESET;
+        BigNat bn1 = new BigNat((short) 10, memoryType, rm);
+
+        byte[] data1 = {0x00, 0x01};
+        bn1.fromByteArray(data1, (short) 0, (short) data1.length);
+        bn1.ctShiftLeft((short) 1, (short) 4);
+
+        byte[] expectedResult = {0x00, 0x06};
+        byte[] actualResult = new byte[2];
+        bn1.copyToByteArray(actualResult, (short) 0);
+        Assertions.assertArrayEquals(expectedResult, actualResult);
+    }
+
+    @Test
+    public void shiftLeft_invalidShift8() {
+        ResourceManager rm = new ResourceManager((short) 256);
+        byte memoryType = JCSystem.MEMORY_TYPE_TRANSIENT_RESET;
+        BigNat bn1 = new BigNat((short) 10, memoryType, rm);
+
+        byte[] data1 = {0x00, 0x02};
+        bn1.fromByteArray(data1, (short) 0, (short) data1.length);
+        Assertions.assertThrows(ISOException.class, () -> bn1.ctShiftRight((short) 8));
+    }
+
+    @Test
+    public void shiftLeft_invalidShiftNegative() {
+        ResourceManager rm = new ResourceManager((short) 256);
+        byte memoryType = JCSystem.MEMORY_TYPE_TRANSIENT_RESET;
+        BigNat bn1 = new BigNat((short) 10, memoryType, rm);
+
+        byte[] data1 = {0x00, 0x02};
+        bn1.fromByteArray(data1, (short) 0, (short) data1.length);
+        Assertions.assertThrows(ISOException.class, () -> bn1.ctShiftRight((short) -1));
+    }
+}