Fix reference leak when encountering error during compilation of function body.

markshannon · markshannon · commit 55d9d2b90f22 · 2020-11-13T11:51:10.000Z
diff --git a/Python/compile.c b/Python/compile.c
@@ -2276,7 +2276,7 @@ compiler_function(struct compiler *c, stmt_ty s, int is_async)
     c->u->u_posonlyargcount = asdl_seq_LEN(args->posonlyargs);
     c->u->u_kwonlyargcount = asdl_seq_LEN(args->kwonlyargs);
     for (i = docstring ? 1 : 0; i < asdl_seq_LEN(body); i++) {
-        VISIT(c, stmt, (stmt_ty)asdl_seq_GET(body, i));
+        VISIT_IN_SCOPE(c, stmt, (stmt_ty)asdl_seq_GET(body, i));
     }
     co = assemble(c, 1);
     qualname = c->u->u_qualname;

Original file line number	Diff line number	Diff line change
`@@ -2276,7 +2276,7 @@ compiler_function(struct compiler *c, stmt_ty s, int is_async)`
`2276`	`2276`	`c->u->u_posonlyargcount = asdl_seq_LEN(args->posonlyargs);`
`2277`	`2277`	`c->u->u_kwonlyargcount = asdl_seq_LEN(args->kwonlyargs);`
`2278`	`2278`	`for (i = docstring ? 1 : 0; i < asdl_seq_LEN(body); i++) {`
`2279`		`- VISIT(c, stmt, (stmt_ty)asdl_seq_GET(body, i));`
	`2279`	`+ VISIT_IN_SCOPE(c, stmt, (stmt_ty)asdl_seq_GET(body, i));`
`2280`	`2280`	`}`
`2281`	`2281`	`co = assemble(c, 1);`
`2282`	`2282`	`qualname = c->u->u_qualname;`