Inline size of byte arrays.

gotranspile · Oct 16, 2022 · f921f98 · f921f98
1 parent 2e3e90d
commit f921f98
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/c_expr.go b/c_expr.go
@@ -1044,6 +1044,12 @@ func (e *CSizeofExpr) Uses() []types.Usage {
 }
 
 func sizeOf(typ types.Type) GoExpr {
+	switch typ := types.Unwrap(typ).(type) {
+	case types.ArrayType:
+		if !typ.IsSlice() && types.Unwrap(typ.Elem()) == types.UintT(1) {
+			return intLit(typ.Len())
+		}
+	}
 	x := typ.GoType()
 	switch types.Unwrap(typ).(type) {
 	case *types.StructType, types.ArrayType:

diff --git a/parse_test.go b/parse_test.go
@@ -572,10 +572,12 @@ var a int32 = int32(unsafe.Sizeof(int32(0)))
 void foo() {
 	size_t a;
 	int b[10];
+	unsigned char b2[10];
 	void* c;
 	int* d;
 	int (*e)(void);
 	a = sizeof(b);
+	a = sizeof(b2);
 	a = sizeof(c);
 	a = sizeof(d);
 	a = sizeof(e);
@@ -587,13 +589,16 @@ func foo() {
 	_ = a
 	var b [10]int32
 	_ = b
+	var b2 [10]uint8
+	_ = b2
 	var c unsafe.Pointer
 	_ = c
 	var d *int32
 	_ = d
 	var e func() int32
 	_ = e
 	a = size_t(unsafe.Sizeof([10]int32{}))
+	a = size_t(10)
 	a = size_t(unsafe.Sizeof(unsafe.Pointer(nil)))
 	a = size_t(unsafe.Sizeof((*int32)(nil)))
 	a = size_t(unsafe.Sizeof(uintptr(0)))