prog: add bytesizeN types

1 files changed, 19 insertions, 2 deletions

diff --git a/sysgen/sysgen.go b/sysgen/sysgen.go
index 5bda2ed2b..92258150d 100644
--- a/sysgen/sysgen.go
+++ b/sysgen/sysgen.go
@@ -465,7 +465,7 @@ func generateArg(
 			failf("wrong number of arguments for %v arg %v, want %v, got %v", typ, name, want, len(a))
 		}
 		fmt.Fprintf(out, "&VmaType{%v}", common())
-	case "len", "bytesize":
+	case "len", "bytesize", "bytesize2", "bytesize4", "bytesize8":
 		canBeArg = true
 		size := uint64(ptrSize)
 		bigEndian := false
@@ -479,7 +479,11 @@ func generateArg(
 				failf("wrong number of arguments for %v arg %v, want %v, got %v", typ, name, want, len(a))
 			}
 		}
-		fmt.Fprintf(out, "&LenType{%v, Buf: \"%v\", TypeSize: %v, BigEndian: %v, ByteSize: %v}", common(), a[0], size, bigEndian, typ == "bytesize")
+		byteSize := uint8(0)
+		if typ != "len" {
+			byteSize = decodeByteSizeType(typ)
+		}
+		fmt.Fprintf(out, "&LenType{%v, Buf: \"%v\", TypeSize: %v, BigEndian: %v, ByteSize: %v}", common(), a[0], size, bigEndian, byteSize)
 	case "flags":
 		canBeArg = true
 		size := uint64(ptrSize)
@@ -683,6 +687,19 @@ func decodeIntType(typ string) (uint64, bool) {
 	return uint64(sz / 8), bigEndian
 }
 
+func decodeByteSizeType(typ string) uint8 {
+	switch typ {
+	case "bytesize", "bytesize2", "bytesize4", "bytesize8":
+	default:
+		failf("unknown type %v", typ)
+	}
+	sz := int64(1)
+	if typ != "bytesize" {
+		sz, _ = strconv.ParseInt(typ[8:], 10, 8)
+	}
+	return uint8(sz)
+}
+
 func isIdentifier(s string) bool {
 	for i, c := range s {
 		if c == '_' || (c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z') || i > 0 && (c >= '0' && c <= '9') {