Applied fix for dop251#148 to for-of loops.

lebronyang1 · Apr 14, 2020 · 3347463 · 3347463
1 parent 00c4500
commit 3347463
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/compiler_stmt.go b/compiler_stmt.go
@@ -345,7 +345,7 @@ func (c *compiler) compileForOfStatement(v *ast.ForOfStatement, needResult bool)
 
 func (c *compiler) compileLabeledForOfStatement(v *ast.ForOfStatement, needResult bool, label unistring.String) {
 	c.block = &block{
-		typ:        blockLoop,
+		typ:        blockLoopEnum,
 		outer:      c.block,
 		label:      label,
 		needResult: needResult,

diff --git a/compiler_test.go b/compiler_test.go
@@ -2032,6 +2032,17 @@ func TestReturnFromForInLoop(t *testing.T) {
 	testScript1(SCRIPT, valueTrue, t)
 }
 
+func TestReturnFromForOfLoop(t *testing.T) {
+	const SCRIPT = `
+	(function f() {
+		for (var i of [1]) {
+			return true;
+		}
+	})();
+	`
+	testScript1(SCRIPT, valueTrue, t)
+}
+
 func TestIfStackLeaks(t *testing.T) {
 	const SCRIPT = `
 	var t = 0;