Use Java MessageDigest for Sha256 hashing

LMAX-Exchange · Oct 4, 2024 · a6513dd · a6513dd
1 parent 3c47b7e
commit a6513dd
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 2 deletions.
diff --git a/src/main/java/com/lmax/solana4j/encoding/SolanaProgramDerivedAddress.java b/src/main/java/com/lmax/solana4j/encoding/SolanaProgramDerivedAddress.java
@@ -3,7 +3,7 @@
 import com.lmax.solana4j.api.ProgramDerivedAddress;
 import com.lmax.solana4j.api.PublicKey;
 import com.lmax.solana4j.util.Ed25519;
-import org.bitcoinj.core.Sha256Hash;
+import com.lmax.solana4j.util.Sha256;
 
 import java.nio.ByteBuffer;
 import java.nio.charset.StandardCharsets;
@@ -37,7 +37,7 @@ static ProgramDerivedAddress deriveProgramAddress(final List<byte[]> seeds, fina
             programId.write(seedsBuffer);
             seedsBuffer.put(PROGRAM_DERIVED_ADDRESS_BYTES);
 
-            final byte[] programAddress = Sha256Hash.hash(seedsBuffer.array());
+            final byte[] programAddress = Sha256.hash(seedsBuffer.array());
 
             if (isOffCurve(programAddress))
             {

diff --git a/src/main/java/com/lmax/solana4j/util/Sha256.java b/src/main/java/com/lmax/solana4j/util/Sha256.java
@@ -0,0 +1,21 @@
+package com.lmax.solana4j.util;
+
+import java.security.MessageDigest;
+import java.security.NoSuchAlgorithmException;
+
+public class Sha256
+{
+    public static byte[] hash(final byte[] input)
+    {
+        final MessageDigest digest;
+        try
+        {
+            digest = MessageDigest.getInstance("SHA-256");
+        }
+        catch (final NoSuchAlgorithmException e)
+        {
+            throw new RuntimeException(e);
+        }
+        return digest.digest(input);
+    }
+}