From 4f561c4158089d8f9c01d021197708c06e69784d Mon Sep 17 00:00:00 2001
From: zenground0 <ZenGround0@users.noreply.github.com>
Date: Sat, 14 May 2022 14:42:57 -0400
Subject: [PATCH] almost everything is there

---
 gen.go                    | 28 ++++++++++++++++++----------
 testing/roundtrip_test.go |  5 +++--
 2 files changed, 21 insertions(+), 12 deletions(-)

diff --git a/gen.go b/gen.go
index 6d83e5f..3bf7eb1 100644
--- a/gen.go
+++ b/gen.go
@@ -5,6 +5,7 @@ import (
 	"io"
 	"math/big"
 	"reflect"
+	"strconv"
 	"strings"
 	"text/template"
 
@@ -16,6 +17,7 @@ const MaxLength = 8192
 const ByteArrayMaxLen = 2 << 20
 
 const MaxLenTag = "maxlen"
+const NoUsrMaxLen = -1
 
 var (
 	cidType      = reflect.TypeOf(cid.Cid{})
@@ -34,6 +36,7 @@ func doTemplate(w io.Writer, info interface{}, templ string) error {
 			"ReadHeader": func(rdr string) string {
 				return fmt.Sprintf(`%s.ReadHeader()`, rdr)
 			},
+			//todo do it here
 		}).Parse(templ))
 
 	return t.Execute(w, info)
@@ -85,7 +88,6 @@ type Field struct {
 	IterLabel string
 
 	MaxLen int
-	//	MaxByteLen int
 }
 
 func typeName(pkg string, t reflect.Type) string {
@@ -177,19 +179,25 @@ func ParseTypeInfo(i interface{}) (*GenTypeInfo, error) {
 		}
 
 		mapk := f.Name
+		usrMaxLen := NoUsrMaxLen
 		tagval := f.Tag.Get("cborgen")
-		if tagval != "" {
+		if len(tagval) > len(MaxLenTag) && tagval[0:len(MaxLenTag)] == MaxLenTag {
+			var err error
+			usrMaxLen, err = strconv.Atoi(tagval[len(MaxLenTag):])
+			if err != nil {
+				return nil, fmt.Errorf("failed to parse specified length in max len tag %w", err)
+			}
+		} else if tagval != "" {
 			mapk = tagval
 		}
 
-		// todo and here
-
 		out.Fields = append(out.Fields, Field{
 			Name:    f.Name,
 			MapKey:  mapk,
 			Pointer: pointer,
 			Type:    ft,
 			Pkg:     pkg,
+			MaxLen:  usrMaxLen,
 		})
 	}
 
@@ -230,7 +238,7 @@ func emitCborMarshalStringField(w io.Writer, f Field) error {
 	}
 
 	return doTemplate(w, f, `
-	if len({{ .Name }}) > cbg.MaxLength {
+	if len({{ .Name }}) > {{ .MaxLen }} {
 		return xerrors.Errorf("Value in field {{ .Name | js }} was too long")
 	}
 
@@ -404,7 +412,7 @@ func emitCborMarshalSliceField(w io.Writer, f Field) error {
 	// Note: this re-slices the slice to deal with arrays.
 	if e.Kind() == reflect.Uint8 {
 		return doTemplate(w, f, `
-	if len({{ .Name }}) > cbg.ByteArrayMaxLen {
+	if len({{ .Name }}) > {{ .MaxLen }} {
 		return xerrors.Errorf("Byte array in field {{ .Name }} was too long")
 	}
 
@@ -421,7 +429,7 @@ func emitCborMarshalSliceField(w io.Writer, f Field) error {
 	}
 
 	err := doTemplate(w, f, `
-	if len({{ .Name }}) > cbg.MaxLength {
+	if len({{ .Name }}) > {{ .MaxLen }} {
 		return xerrors.Errorf("Slice value in field {{ .Name }} was too long")
 	}
 
@@ -876,7 +884,7 @@ func emitCborUnmarshalSliceField(w io.Writer, f Field) error {
 
 	if e.Kind() == reflect.Uint8 {
 		return doTemplate(w, f, `
-	if extra > cbg.ByteArrayMaxLen {
+	if extra > {{ .MaxLen }} {
 		return fmt.Errorf("{{ .Name }}: byte array too large (%d)", extra)
 	}
 	if maj != cbg.MajByteString {
@@ -900,7 +908,7 @@ func emitCborUnmarshalSliceField(w io.Writer, f Field) error {
 	}
 
 	if err := doTemplate(w, f, `
-	if extra > cbg.MaxLength {
+	if extra > {{ .MaxLen }} {
 		return fmt.Errorf("{{ .Name }}: array too large (%d)", extra)
 	}
 `); err != nil {
@@ -1200,7 +1208,7 @@ func (t *{{ .Name}}) UnmarshalCBOR(r io.Reader) (err error) {
 		return fmt.Errorf("cbor input should be of type map")
 	}
 
-	if extra > cbg.MaxLength {
+	if extra > {{ .MaxLen }} {
 		return fmt.Errorf("{{ .Name }}: map struct too large (%d)", extra)
 	}
 
diff --git a/testing/roundtrip_test.go b/testing/roundtrip_test.go
index b2cbb5c..63b93cd 100644
--- a/testing/roundtrip_test.go
+++ b/testing/roundtrip_test.go
@@ -346,7 +346,7 @@ func TestErrUnexpectedEOF(t *testing.T) {
 	}
 }
 
-func TestLargeFeild(t *testing.T) {
+func TestLargeField(t *testing.T) {
 	// 10 MB of data is the specified max so  4 MiB should work
 	bs := make([]byte, 2<<21)
 	bs[2<<20] = 0xaa // flags to check that serialization works
@@ -375,5 +375,6 @@ func TestLargeFeild(t *testing.T) {
 	if err == nil {
 		t.Fatal("buffer bigger than specified in struct tag should fail")
 	}
-
 }
+
+//TODO same for strings