Add wasm helper, wasm test utility

cmd/test-wasm-runtime/main.go

@@ -0,0 +1,107 @@
+package main
+
+import (
+	"context"
+	"encoding/json"
+	"flag"
+	"fmt"
+	"git.gammaspectra.live/git/go-away/lib/challenge"
+	"github.com/tetratelabs/wazero/api"
+	"os"
+	"reflect"
+	"slices"
+)
+
+func main() {
+
+	pathToTest := flag.String("wasm", "", "Path to test file")
+	makeChallenge := flag.String("make-challenge", "", "Path to contents for MakeChallenge input")
+	makeChallengeOutput := flag.String("make-challenge-out", "", "Path to contents for expected MakeChallenge output")
+	verifyChallenge := flag.String("verify-challenge", "", "Path to contents for VerifyChallenge input")
+	verifyChallengeOutput := flag.Uint64("verify-challenge-out", uint64(challenge.VerifyChallengeOutputOK), "Path to contents for expected VerifyChallenge output")
+
+	flag.Parse()
+
+	if *pathToTest == "" || *makeChallenge == "" || *makeChallengeOutput == "" || *verifyChallenge == "" {
+		flag.PrintDefaults()
+		os.Exit(1)
+	}
+
+	wasmData, err := os.ReadFile(*pathToTest)
+	if err != nil {
+		panic(err)
+	}
+
+	runner := challenge.NewRunner(true)
+	defer runner.Close()
+
+	err = runner.Compile("test", wasmData)
+	if err != nil {
+		panic(err)
+	}
+
+	makeData, err := os.ReadFile(*makeChallenge)
+	if err != nil {
+		panic(err)
+	}
+	var makeIn challenge.MakeChallengeInput
+	err = json.Unmarshal(makeData, &makeIn)
+	if err != nil {
+		panic(err)
+	}
+
+	makeOutData, err := os.ReadFile(*makeChallengeOutput)
+	if err != nil {
+		panic(err)
+	}
+	var makeOut challenge.MakeChallengeOutput
+	err = json.Unmarshal(makeOutData, &makeOut)
+	if err != nil {
+		panic(err)
+	}
+
+	verifyData, err := os.ReadFile(*verifyChallenge)
+	if err != nil {
+		panic(err)
+	}
+	var verifyIn challenge.VerifyChallengeInput
+	err = json.Unmarshal(verifyData, &verifyIn)
+	if err != nil {
+		panic(err)
+	}
+
+	if slices.Compare(makeIn.Key, verifyIn.Key) != 0 {
+		panic("challenge keys do not match")
+	}
+
+	err = runner.Instantiate("test", func(ctx context.Context, mod api.Module) error {
+		out, err := challenge.MakeChallengeCall(ctx, mod, makeIn)
+		if err != nil {
+			return err
+		}
+
+		if !reflect.DeepEqual(*out, makeOut) {
+			return fmt.Errorf("challenge output did not match expected output, got %v, expected %v", *out, makeOut)
+		}
+		return nil
+	})
+	if err != nil {
+		panic(err)
+	}
+
+	err = runner.Instantiate("test", func(ctx context.Context, mod api.Module) error {
+		out, err := challenge.VerifyChallengeCall(ctx, mod, verifyIn)
+		if err != nil {
+			return err
+		}
+
+		if out != challenge.VerifyChallengeOutput(*verifyChallengeOutput) {
+			return fmt.Errorf("verify output did not match expected output, got %d expected %d", out, challenge.VerifyChallengeOutput(*verifyChallengeOutput))
+		}
+		return nil
+	})
+	if err != nil {
+		panic(err)
+	}
+
+}