fix: emit source_loc around iterator_close so for/of return() stack traces report correct line

Fixes #1266

Author: mvanhorn

quickjs.c

@@ -23180,14 +23180,19 @@ static void emit_u32(JSParseState *s, uint32_t val)
     dbuf_put_u32(&s->cur_func->byte_code, val);
 }
 
-static void emit_source_loc(JSParseState *s)
+static void emit_source_loc_at(JSParseState *s, int line_num, int col_num)
 {
     JSFunctionDef *fd = s->cur_func;
     DynBuf *bc = &fd->byte_code;
 
     dbuf_putc(bc, OP_source_loc);
-    dbuf_put_u32(bc, s->token.line_num);
-    dbuf_put_u32(bc, s->token.col_num);
+    dbuf_put_u32(bc, line_num);
+    dbuf_put_u32(bc, col_num);
+}
+
+static void emit_source_loc(JSParseState *s)
+{
+    emit_source_loc_at(s, s->token.line_num, s->token.col_num);
 }
 
 static void emit_op(JSParseState *s, uint8_t val)
@@ -26032,8 +26037,11 @@ static int js_parse_destructuring_element(JSParseState *s, int tok,
     } else if (s->token.val == '[') {
         bool has_spread;
         int enum_depth;
+        int source_line_num, source_col_num;
         BlockEnv block_env;
 
+        source_line_num = s->token.line_num;
+        source_col_num = s->token.col_num;
         if (next_token(s))
             return -1;
         /* the block environment is only needed in generators in case
@@ -26129,6 +26137,7 @@ static int js_parse_destructuring_element(JSParseState *s, int tok,
         }
         /* close iterator object:
            if completed, enum_obj has been replaced by undefined */
+        emit_source_loc_at(s, source_line_num, source_col_num);
         emit_op(s, OP_iterator_close);
         pop_break_entry(s->cur_func);
         if (next_token(s))
@@ -28012,7 +28021,9 @@ static int is_let(JSParseState *s, int decl_mask)
 /* XXX: handle IteratorClose when exiting the loop before the
    enumeration is done */
 static __exception int js_parse_for_in_of(JSParseState *s, int label_name,
-                                          bool is_async)
+                                          bool is_async,
+                                          int source_line_num,
+                                          int source_col_num)
 {
     JSContext *ctx = s->ctx;
     JSFunctionDef *fd = s->cur_func;
@@ -28230,6 +28241,7 @@ static __exception int js_parse_for_in_of(JSParseState *s, int label_name,
     emit_label(s, label_break);
     if (is_for_of) {
         /* close and drop enum_rec */
+        emit_source_loc_at(s, source_line_num, source_col_num);
         emit_op(s, OP_iterator_close);
     } else {
         emit_op(s, OP_drop);
@@ -28464,8 +28476,11 @@ static __exception int js_parse_statement_or_decl(JSParseState *s,
             int pos_cont, pos_body, block_scope_level;
             BlockEnv break_entry;
             int tok, bits;
+            int source_line_num, source_col_num;
             bool is_async;
 
+            source_line_num = s->token.line_num;
+            source_col_num = s->token.col_num;
             if (next_token(s))
                 goto fail;
 
@@ -28489,7 +28504,8 @@ static __exception int js_parse_statement_or_decl(JSParseState *s,
 
             if (!(bits & SKIP_HAS_SEMI)) {
                 /* parse for/in or for/of */
-                if (js_parse_for_in_of(s, label_name, is_async))
+                if (js_parse_for_in_of(s, label_name, is_async,
+                                       source_line_num, source_col_num))
                     goto fail;
                 break;
             }

tests/test_for_of_line_numbers.js

@@ -0,0 +1,50 @@
+import { assert } from "./assert.js";
+
+function closingIterable(stacks)
+{
+    return {
+        [Symbol.iterator]() {
+            return {
+                next() {
+                    return { done: false, value: 1 };
+                },
+                return() {
+                    stacks.push(new Error().stack);
+                    return { done: true };
+                },
+            };
+        },
+    };
+}
+
+function assertCallerLine(frames, expectedLine, message)
+{
+    assert(frames.length >= 2, true, message);
+    assert(frames[1].getFileName().endsWith("test_for_of_line_numbers.js"), true, message);
+    assert(frames[1].getLineNumber(), expectedLine, message);
+}
+
+function test_for_of_empty_body_line_number()
+{
+    const stacks = [];
+    const expectedLine = 31;
+    for (const _ of closingIterable(stacks)) break;
+    assertCallerLine(stacks[0], expectedLine, "for-of iterator close");
+}
+
+function test_for_of_destructuring_line_number()
+{
+    const stacks = [];
+    const outer = [closingIterable(stacks)];
+    const expectedLine = 40;
+    for (const [_] of outer) break;
+    assertCallerLine(stacks[0], expectedLine, "destructuring iterator close");
+}
+
+Error.prepareStackTrace = (_, frames) => frames;
+try {
+    test_for_of_empty_body_line_number();
+    test_for_of_destructuring_line_number();
+} finally {
+    Error.prepareStackTrace = undefined;
+}