support list of non-ptr containers (ferranbt#151)

itsdevbear · Mar 25, 2024 · 3fad963 · 3fad963
1 parent 274d5da
commit 3fad963
Show file tree

Hide file tree

Showing 3 changed files with 314 additions and 1 deletion.
diff --git a/sszgen/generator/unmarshal.go b/sszgen/generator/unmarshal.go
@@ -342,7 +342,11 @@ func (v *Value) createSlice(useNumVariable bool) string {
 
 	case TypeContainer:
 		// []*(ref.)Struct{}
-		return fmt.Sprintf("::.%s = make([]*%s, %s)", v.name, v.e.objRef(), size)
+		ptr := "*"
+		if v.e.noPtr {
+			ptr = ""
+		}
+		return fmt.Sprintf("::.%s = make([]%s%s, %s)", v.name, ptr, v.e.objRef(), size)
 
 	case TypeBytes:
 		// [][]byte

diff --git a/sszgen/testcases/list.go b/sszgen/testcases/list.go
@@ -0,0 +1,15 @@
+package testcases
+
+//go:generate go run ../main.go --path list.go
+
+type BytesWrapper struct {
+	Bytes []byte `ssz-size:"48"`
+}
+
+type ListC struct {
+	Elems []BytesWrapper `ssz-max:"32"`
+}
+
+type ListP struct {
+	Elems []*BytesWrapper `ssz-max:"32"`
+}
diff --git a/sszgen/testcases/list_encoding.go b/sszgen/testcases/list_encoding.go