Handle recursive structs.

markkurossi · Apr 1, 2020 · 9b7b78e · 9b7b78e
1 parent 455ce13
commit 9b7b78e
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/compiler/ast/package.go b/compiler/ast/package.go
@@ -183,12 +183,16 @@ func flattenStruct(t types.Info) circuit.IO {
 	}
 
 	for _, f := range t.Struct {
-		// XXX recursive structs.
-		result = append(result, circuit.IOArg{
-			Name: f.Name,
-			Type: f.Type.String(),
-			Size: f.Type.Bits,
-		})
+		if f.Type.Type == types.Struct {
+			ios := flattenStruct(f.Type)
+			result = append(result, ios...)
+		} else {
+			result = append(result, circuit.IOArg{
+				Name: f.Name,
+				Type: f.Type.String(),
+				Size: f.Type.Bits,
+			})
+		}
 	}
 
 	return result