Implement struct_ref for return value

shinh · Dec 11, 2016 · 1121cdb · 1121cdb
1 parent 3dd51a8
commit 1121cdb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/gen.c b/gen.c
@@ -405,6 +405,11 @@ static void emit_load_struct_ref(Node *struc, Type *field, int off) {
         emit_expr(struc->operand);
         emit_lload(field, "rax", field->offset + off);
         break;
+    case AST_FUNCALL: {
+        emit_expr(struc);
+        emit_lload(field, "rax", field->offset + off);
+        break;
+    }
     default:
         error("internal error: %s", node2s(struc));
     }

diff --git a/test/struct.c b/test/struct.c
@@ -389,6 +389,7 @@ static void struct_return(void) {
     expect(21, sum_struct(6, return_struct_func(1, 2, 3, 4, 5)));
     expect(55, sum_struct(sum_struct(0, return_struct_func(1, 2, 3, 4, 5)),
                           return_struct_func(6, 7, 8, 9, 10)));
+    expect(99, return_struct_func(1, 2, 3, 99, 4).z.b);
 }
 
 static int struct_arg_func_small(int v, struct abi_check_small s) {