consensus/monero/address/crypto_test.go

package address

import (
	"git.gammaspectra.live/P2Pool/consensus/v3/monero/crypto"
	"git.gammaspectra.live/P2Pool/consensus/v3/types"
	"os"
	"path"
	"runtime"
	"strconv"
	"strings"
	"testing"
)

func init() {
	_, filename, _, _ := runtime.Caller(0)
	// The ".." may change depending on you folder structure
	dir := path.Join(path.Dir(filename), "../..")
	err := os.Chdir(dir)
	if err != nil {
		panic(err)
	}

}

func GetTestEntries(name string, n int) chan []string {
	buf, err := os.ReadFile("testdata/crypto_tests.txt")
	if err != nil {
		return nil
	}
	result := make(chan []string)
	go func() {
		defer close(result)
		for _, line := range strings.Split(string(buf), "\n") {
			entries := strings.Split(strings.TrimSpace(line), " ")
			if entries[0] == name && len(entries) >= (n+1) {
				result <- entries[1:]
			}
		}
	}()
	return result
}

func TestDerivePublicKey(t *testing.T) {
	results := GetTestEntries("derive_public_key", 4)
	if results == nil {
		t.Fatal()
	}

	hasher := crypto.GetKeccak256Hasher()
	defer crypto.PutKeccak256Hasher(hasher)

	for e := range results {
		var expectedDerivedKey types.Hash

		derivation := crypto.PublicKeyBytes(types.MustHashFromString(e[0]))
		outputIndex, _ := strconv.ParseUint(e[1], 10, 0)

		base := crypto.PublicKeyBytes(types.MustHashFromString(e[2]))

		result := e[3] == "true"
		if result {
			expectedDerivedKey = types.MustHashFromString(e[4])
		}

		point2 := base.AsPoint()

		if result == false && point2 == nil {
			//expected failure
			continue
		} else if point2 == nil {
			t.Errorf("invalid point %s / %s", derivation.String(), base.String())
			continue
		}

		sharedData := crypto.GetDerivationSharedDataForOutputIndex(&derivation, outputIndex)

		sharedData2, _ := crypto.GetDerivationSharedDataAndViewTagForOutputIndexNoAllocate(derivation, outputIndex, hasher)

		if sharedData.AsBytes() != crypto.PrivateKeyFromScalar(&sharedData2).AsBytes() {
			t.Errorf("derive_public_key differs from no_allocate: %s != %s", sharedData, crypto.PrivateKeyFromScalar(&sharedData2))
		}

		var addr PackedAddress
		addr[0] = base
		derivedKey := GetPublicKeyForSharedData(&addr, sharedData)

		derivedKey2, _ := GetEphemeralPublicKeyAndViewTagNoAllocate(base.AsPoint().Point(), derivation, outputIndex, hasher)

		if derivedKey.AsBytes() != derivedKey2 {
			t.Errorf("derive_public_key differs from no_allocate: %s != %s", derivedKey, &derivedKey2)
		}

		if result {
			if expectedDerivedKey.String() != derivedKey.String() {
				t.Errorf("expected %s, got %s", expectedDerivedKey.String(), derivedKey.String())
			}
		}
	}
}

var testThrowawayAddress = FromBase58("42ecNLuoGtn1qC9SPSD9FPMNfsv35RE66Eu8WJJtyEHKfEsEiALVVp7GBCeAYFb7PHcSZmz9sDUtRMnKk2as1KfuLuTQJ3i")
var signatureMessage = []byte("test_message")
var signatureMessage2 = []byte("test_message2")

const (
	SignatureSpendSuccess      = "SigV2DnQYF11xZ1ahLJxddohCroiEJRnUe1tgwD5ksmFMzQ9NcRdbxLPrEdQW3e8w4sLpqhSup5tU9igQqeAR8j7r7Sty"
	SignatureSpendWrongMessage = "SigV26RKRd31efizGHrWHwtYG6EN2MmwvF1rjU4ygZQuDmSxvCJnky1GJTzaM49naQeKvXbaGcnpZ1b3k8gVQLaFMFiBJ"
	SignatureViewSuccess       = "SigV2b7LaAuXrFvPAXwU11SJwHbcXJoKfQ5aBJ9FwMJNxvMTu78AebqNUCWPH1BVfNRvy1f3GCTLjHfWvuRJMZtSHu5uj"
	SignatureViewWrongMessage  = "SigV2AxWUATswZvnHSR5mMRsn9GcJe2gSCv3SbFwHv6J8THkj5KvmR8gUnTidHovZVyxgNHcUuiunM2dfVhbZvBTS6sZZ"
)

func TestSignature(t *testing.T) {
	result := VerifyMessage(testThrowawayAddress, signatureMessage, SignatureSpendSuccess)
	if result != ResultSuccessSpend {
		t.Fatalf("unexpected %d", result)
	}
	result = VerifyMessage(testThrowawayAddress, signatureMessage, SignatureSpendWrongMessage)
	if result != ResultFail {
		t.Fatalf("unexpected %d", result)
	}
	result = VerifyMessage(testThrowawayAddress, signatureMessage2, SignatureSpendSuccess)
	if result != ResultFail {
		t.Fatalf("unexpected %d", result)
	}
	result = VerifyMessage(testThrowawayAddress, signatureMessage, SignatureViewSuccess)
	if result != ResultFailZeroSpend {
		t.Fatalf("unexpected %d", result)
	}
	result = VerifyMessage(testThrowawayAddress, signatureMessage, SignatureViewWrongMessage)
	if result != ResultFail {
		t.Fatalf("unexpected %d", result)
	}
	result = VerifyMessage(testThrowawayAddress, signatureMessage2, SignatureViewSuccess)
	if result != ResultFail {
		t.Fatalf("unexpected %d", result)
	}
	result = VerifyMessage(&ZeroPrivateKeyAddress, signatureMessage, SignatureViewSuccess)
	if result != ResultFail {
		t.Fatalf("unexpected %d", result)
	}
}