feat(galoisd): sha256 test

galoisd/pkg/sha256/api.go

@@ -283,7 +283,6 @@ func padding(api frontend.API, input []frontend.Variable, size frontend.Variable
 
 	// helpers
 	inputLen := len(input)
-	paddingLen := inputLen % 64
 
 	// t is start index of inputBitLen encoding
 	var t int
@@ -304,16 +303,6 @@ func padding(api frontend.API, input []frontend.Variable, size frontend.Variable
 		out[i] = 0
 	}
 
-	// return if no padding required
-	if paddingLen == 0 {
-
-		// overwrite into fixed size slice
-		for i := 0; i < inputLen; i++ {
-			out[i] = input[i]
-		}
-		return out
-	}
-
 	// existing bytes into out
 	for i := 0; i < inputLen; i++ {
 		out[i] = input[i]

galoisd/pkg/sha256/api_test.go

@@ -0,0 +1,79 @@
+package sha256
+
+import (
+	"crypto/rand"
+	"crypto/sha256"
+	"testing"
+
+	"github.com/consensys/gnark-crypto/ecc"
+	"github.com/consensys/gnark/backend"
+	"github.com/consensys/gnark/frontend"
+	"github.com/consensys/gnark/test"
+)
+
+const compressThreshold = 1000
+
+const MaxPreimageLength = 64
+
+const ImageLength = 32
+
+type sha256Circuit struct {
+	PreimageLength frontend.Variable
+	Preimage       [MaxPreimageLength]frontend.Variable
+	Image          [ImageLength]frontend.Variable
+}
+
+func (c *sha256Circuit) Define(api frontend.API) error {
+	api.AssertIsLessOrEqual(c.PreimageLength, MaxPreimageLength)
+	hash := NewSHA256(api)
+	actualPreimage := make([]frontend.Variable, MaxPreimageLength)
+	for i := 0; i < MaxPreimageLength; i++ {
+		actualPreimage[i] = c.Preimage[i]
+	}
+	image := hash.Hash(actualPreimage, c.PreimageLength)
+	for i := 0; i < ImageLength; i++ {
+		api.AssertIsEqual(image[i], c.Image[i])
+	}
+	return nil
+}
+
+func TestSha256(t *testing.T) {
+	message := make([]byte, MaxPreimageLength)
+	_, err := rand.Read(message)
+	if err != nil {
+		panic(err)
+	}
+
+	nativeHasher := sha256.New()
+	nativeHasher.Write(message)
+	final := nativeHasher.Sum(nil)
+
+	var preimage [MaxPreimageLength]frontend.Variable
+	for i := 0; i < MaxPreimageLength; i++ {
+		if i < len(message) {
+			preimage[i] = message[i]
+		} else {
+			preimage[i] = 0
+		}
+	}
+
+	var image [ImageLength]frontend.Variable
+	for i := 0; i < ImageLength; i++ {
+		image[i] = final[i]
+	}
+
+	circuit := sha256Circuit{}
+	assignment := sha256Circuit{
+		Preimage:       preimage,
+		PreimageLength: len(message),
+		Image:          image,
+	}
+	test.NewAssert(t).ProverSucceeded(
+		&circuit,
+		&assignment,
+		test.WithCurves(ecc.BN254),
+		test.NoFuzzing(),
+		test.WithCurves(ecc.BN254),
+		test.WithBackends(backend.GROTH16),
+	)
+}