From affe1ecb4ff16a7f0962d5a77a52f4b46372b3f6 Mon Sep 17 00:00:00 2001
From: Evgeniy Stratonikov <evgeniy@nspcc.ru>
Date: Wed, 25 May 2022 17:14:05 +0300
Subject: [PATCH] encoding: add bigint fuzz test

Signed-off-by: Evgeniy Stratonikov <evgeniy@nspcc.ru>
---
 pkg/encoding/bigint/fuzz_test.go | 52 ++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)
 create mode 100644 pkg/encoding/bigint/fuzz_test.go

diff --git a/pkg/encoding/bigint/fuzz_test.go b/pkg/encoding/bigint/fuzz_test.go
new file mode 100644
index 000000000..63609353b
--- /dev/null
+++ b/pkg/encoding/bigint/fuzz_test.go
@@ -0,0 +1,52 @@
+//go:build go1.18
+// +build go1.18
+
+package bigint
+
+import (
+	"bytes"
+	"crypto/rand"
+	"math/big"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func FuzzFromBytes(f *testing.F) {
+	for _, tc := range testCases {
+		f.Add(tc.buf)
+	}
+	for i := 0; i < 50; i++ {
+		for j := 1; j < MaxBytesLen; j++ {
+			b := make([]byte, j)
+			_, err := rand.Read(b)
+			require.NoError(f, err)
+			f.Add(b)
+		}
+	}
+	f.Fuzz(func(t *testing.T, raw []byte) {
+		var bi *big.Int
+		require.NotPanics(t, func() { bi = FromBytes(raw) })
+
+		var actual []byte
+		require.NotPanics(t, func() { actual = ToBytes(bi) })
+		require.True(t, len(actual) <= len(raw), "actual: %x, raw: %x", actual, raw)
+
+		require.True(t, bytes.Equal(actual, raw[:len(actual)]), "actual: %x, raw: %x", actual, raw)
+		if len(actual) == len(raw) {
+			return
+		}
+
+		var b byte
+		if bi.Sign() == -1 {
+			b = 0xFF
+		}
+		for i := len(actual); i < len(raw); i++ {
+			require.Equal(t, b, raw[i], "invalid prefix")
+		}
+
+		newRaw := ToBytes(bi)
+		newBi := FromBytes(newRaw)
+		require.Equal(t, bi, newBi)
+	})
+}