pkg/compiler, prog: fix template parent lens

It's possible that a struct can have 2+ parents,
which is the same template (differs only by arguments).
See the new test case.
Support such case.

1 files changed, 7 insertions, 1 deletions

diff --git a/prog/size.go b/prog/size.go
index 36928a8bc..e2451fe15 100644
--- a/prog/size.go
+++ b/prog/size.go
@@ -5,6 +5,7 @@ package prog
 
 import (
 	"fmt"
+	"strings"
 )
 
 func (target *Target) generateSize(arg Arg, lenType *LenType) uint64 {
@@ -67,7 +68,12 @@ func (target *Target) assignSizes(args []Arg, parentsMap map[Arg]Arg) {
 
 			sizeAssigned := false
 			for parent := parentsMap[arg]; parent != nil; parent = parentsMap[parent] {
-				if typ.Buf == parent.Type().Name() {
+				parentName := parent.Type().Name()
+				if pos := strings.IndexByte(parentName, '['); pos != -1 {
+					// For template parents, strip arguments.
+					parentName = parentName[:pos]
+				}
+				if typ.Buf == parentName {
 					a.Val = parent.Size()
 					if typ.BitSize != 0 {
 						a.Val = a.Val * 8 / typ.BitSize