RingSize Sanity Check Fix by Slixe

deroproject · Jun 13, 2023 · b9c5fa7 · b9c5fa7
1 parent 0951e98
commit b9c5fa7
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 6 deletions.
diff --git a/config/version.go b/config/version.go
@@ -20,4 +20,4 @@ import "github.com/blang/semver/v4"
 
 // right now it has to be manually changed
 // do we need to include git commitsha??
-var Version = semver.MustParse("3.5.3-117.DEROHE.STARGATE+18032023")
+var Version = semver.MustParse("3.5.3-140.DEROHE.STARGATE+13062023")
diff --git a/cryptography/crypto/protocol_structures.go b/cryptography/crypto/protocol_structures.go
@@ -16,11 +16,15 @@
 
 package crypto
 
-import "bytes"
-import "encoding/binary"
-import "math/big"
-import "github.com/deroproject/derohe/cryptography/bn256"
-import "github.com/deroproject/graviton"
+import (
+	"bytes"
+	"encoding/binary"
+	"fmt"
+	"math/big"
+
+	"github.com/deroproject/derohe/cryptography/bn256"
+	"github.com/deroproject/graviton"
+)
 
 type Statement struct {
 	RingSize                 uint64
@@ -98,6 +102,9 @@ func (s *Statement) Deserialize(r *bytes.Reader) error {
 		return err
 	}
 	s.RingSize = 1 << length
+	if s.RingSize > 128 {
+		return fmt.Errorf("ring size is too large")
+	}
 
 	s.Bytes_per_publickey, err = r.ReadByte()
 	if err != nil {
@@ -223,3 +230,4 @@ type Proof struct {
 	//ip *InnerProduct
 }
 */
+