Generate valid secp256k1 key during key creation

Signed-off-by: sdimitrov9 <stoyan.dimitrov@limechain.tech>
hashgraph · Jan 6, 2025 · d3483f1 · d3483f1
1 parent 1520f14
commit d3483f1
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/hedera-mirror-common/src/test/java/com/hedera/mirror/common/domain/DomainBuilder.java b/hedera-mirror-common/src/test/java/com/hedera/mirror/common/domain/DomainBuilder.java
@@ -1147,14 +1147,23 @@ public byte[] key(KeyCase keyCase) {
         var key =
                 switch (keyCase) {
                     case ECDSA_SECP256K1 -> Key.newBuilder()
-                            .setECDSASecp256K1(ByteString.copyFrom(bytes(KEY_LENGTH_ECDSA)));
+                            .setECDSASecp256K1(ByteString.copyFrom(generateSecp256k1Key()));
                     case ED25519 -> Key.newBuilder().setEd25519(ByteString.copyFrom(bytes(KEY_LENGTH_ED25519)));
                     default -> throw new UnsupportedOperationException("Key type not supported");
                 };
 
         return key.build().toByteArray();
     }
 
+    private byte[] generateSecp256k1Key() {
+        byte[] xCoordinate = bytes(32); // The x-coordinate of the elliptic curve point
+        byte prefix = (byte) (Math.random() < 0.5 ? 0x02 : 0x03); // Randomly chosen even or odd y-coordinate
+        ByteBuffer compressedKey = ByteBuffer.allocate(33);
+        compressedKey.put(prefix);
+        compressedKey.put(xCoordinate);
+        return compressedKey.array();
+    }
+
     public long number() {
         return id.incrementAndGet();
     }