Skip to content

Commit b7554b6

Browse files
committed
[interpreter] Fix custom branch hints for try_table
1 parent 3b3d7aa commit b7554b6

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

interpreter/custom/handler_branch_hint.ml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -42,8 +42,8 @@ let flatten_instr_locs is =
4242
| i :: rest ->
4343
let group =
4444
match i.it with
45-
| Block (_, inner) -> [ i ] @ flatten inner
46-
| Loop (_, inner) -> [ i ] @ flatten inner
45+
| Block (_, inner) | Loop (_, inner)
46+
| TryTable (_, _, inner) -> [ i ] @ flatten inner
4747
| If (_, inner1, inner2) -> [ i ] @ flatten inner1 @ flatten inner2
4848
| _ -> [ i ]
4949
in
@@ -445,12 +445,12 @@ let check_one locs prev_hidx h =
445445
check_error h.at
446446
"@metadata.code.branch_hint annotation: invalid target")
447447

448-
let check_fun m fidx hs =
448+
let check_func m fidx hs =
449449
let f = get_func m fidx in
450450
let locs = flatten_instr_locs f.it.body in
451451
let prev_hidx = ref 0 in
452452
List.iter (check_one locs prev_hidx) hs
453453

454454
let check (m : module_) (fmt : format) =
455-
IdxMap.iter (check_fun m) fmt.it.func_hints;
455+
IdxMap.iter (check_func m) fmt.it.func_hints;
456456
()

0 commit comments

Comments
 (0)