ozrs_test.go

package ozcoin

import (
	"math/big"
	"testing"
)

func TestOZRSSign(t *testing.T) {
	prevAmt := uint64(5000000000)
	amts := []uint64{1, 4999999998}
	rcpts := []WalletPublicKey{
		NewPrivateKey().PublicKey(),
		NewPrivateKey().PublicKey(),
	}

	pks, sec := pksAndSecret()
	ics, yi := commitmentsAndBF(prevAmt)
	outputs, bf := BuildOutputs(amts, rcpts)

	txn := Txn{
		Body: TxnBody{
			Outputs: outputs,
			Fee:     1,
		},
	}

	txn.OZRSSign(pks, ics, sec, yi, 0, bf)

	if !txn.VerifyOZRS(pks, ics) {
		t.Error("OZRS Failed to verify")
	}
}

func BenchmarkOZRSSign(b *testing.B) {
	prevAmt := uint64(5000000000)
	amts := []uint64{1, 4999999998}
	rcpts := []WalletPublicKey{
		NewPrivateKey().PublicKey(),
		NewPrivateKey().PublicKey(),
	}

	pks, sec := pksAndSecret()
	ics, yi := commitmentsAndBF(prevAmt)
	outputs, bf := BuildOutputs(amts, rcpts)

	txn := Txn{
		Body: TxnBody{
			Outputs: outputs,
			Fee:     1,
		},
	}

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		txn.OZRSSign(pks, ics, sec, yi, 0, bf)
	}
}

func BenchmarkOZRSVerify(b *testing.B) {
	prevAmt := uint64(5000000000)
	amts := []uint64{1, 4999999998}
	rcpts := []WalletPublicKey{
		NewPrivateKey().PublicKey(),
		NewPrivateKey().PublicKey(),
	}

	pks, sec := pksAndSecret()
	ics, yi := commitmentsAndBF(prevAmt)
	outputs, bf := BuildOutputs(amts, rcpts)

	txn := Txn{
		Body: TxnBody{
			Outputs: outputs,
			Fee:     1,
		},
	}

	txn.OZRSSign(pks, ics, sec, yi, 0, bf)
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		txn.VerifyOZRS(pks, ics)
	}
}

func pksAndSecret() ([]ECCPoint, *big.Int) {
	var sec *big.Int
	pks := []ECCPoint{}
	for i := 0; i < TXN_NUM_INPUTS; i++ {
		s := RandomInt()
		if i == 0 {
			sec = s
		}
		pkx, pky := CURVE.Params().ScalarBaseMult(s.Bytes())
		pks = append(pks, ECCPoint{pkx, pky})
	}

	return pks, sec
}

func commitmentsAndBF(amt uint64) ([]ECCPoint, *big.Int) {
	var yi *big.Int
	ics := []ECCPoint{}
	for i := 0; i < TXN_NUM_INPUTS; i++ {
		//b := RandomInt()
		b := &big.Int{}
		commit := RangeCommit(amt, b)
		if i == 0 {
			yi = b
		}
		ics = append(ics, commit.ECCPoint)
	}

	return ics, yi
}