summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--py/bc.c6
-rw-r--r--tests/basics/try_except_break.py73
-rw-r--r--tests/basics/try_except_break.py.exp3
-rwxr-xr-xtools/mpy-tool.py4
4 files changed, 83 insertions, 3 deletions
diff --git a/py/bc.c b/py/bc.c
index 7d0b13bd7..5625da8cf 100644
--- a/py/bc.c
+++ b/py/bc.c
@@ -292,7 +292,8 @@ continue2:;
#if MICROPY_PERSISTENT_CODE_LOAD || MICROPY_PERSISTENT_CODE_SAVE
// The following table encodes the number of bytes that a specific opcode
-// takes up. There are 3 special opcodes that always have an extra byte:
+// takes up. There are 4 special opcodes that always have an extra byte:
+// MP_BC_UNWIND_JUMP
// MP_BC_MAKE_CLOSURE
// MP_BC_MAKE_CLOSURE_DEFARGS
// MP_BC_RAISE_VARARGS
@@ -402,7 +403,8 @@ uint mp_opcode_format(const byte *ip, size_t *opcode_size, bool count_var_uint)
ip += 3;
} else {
int extra_byte = (
- *ip == MP_BC_RAISE_VARARGS
+ *ip == MP_BC_UNWIND_JUMP
+ || *ip == MP_BC_RAISE_VARARGS
|| *ip == MP_BC_MAKE_CLOSURE
|| *ip == MP_BC_MAKE_CLOSURE_DEFARGS
);
diff --git a/tests/basics/try_except_break.py b/tests/basics/try_except_break.py
new file mode 100644
index 000000000..a7683f218
--- /dev/null
+++ b/tests/basics/try_except_break.py
@@ -0,0 +1,73 @@
+# test deep unwind via break from nested try-except (22 of them)
+while True:
+ print(1)
+ try:
+ try:
+ try:
+ try:
+ try:
+ try:
+ try:
+ try:
+ try:
+ try:
+ try:
+ try:
+ try:
+ try:
+ try:
+ try:
+ try:
+ try:
+ try:
+ try:
+ try:
+ try:
+ print(2)
+ break
+ print(3)
+ except:
+ pass
+ except:
+ pass
+ except:
+ pass
+ except:
+ pass
+ except:
+ pass
+ except:
+ pass
+ except:
+ pass
+ except:
+ pass
+ except:
+ pass
+ except:
+ pass
+ except:
+ pass
+ except:
+ pass
+ except:
+ pass
+ except:
+ pass
+ except:
+ pass
+ except:
+ pass
+ except:
+ pass
+ except:
+ pass
+ except:
+ pass
+ except:
+ pass
+ except:
+ pass
+ except:
+ pass
+print(4)
diff --git a/tests/basics/try_except_break.py.exp b/tests/basics/try_except_break.py.exp
new file mode 100644
index 000000000..e8a01cd98
--- /dev/null
+++ b/tests/basics/try_except_break.py.exp
@@ -0,0 +1,3 @@
+1
+2
+4
diff --git a/tools/mpy-tool.py b/tools/mpy-tool.py
index 3d2a8fb7c..e5c8f0959 100755
--- a/tools/mpy-tool.py
+++ b/tools/mpy-tool.py
@@ -109,6 +109,7 @@ MP_OPCODE_VAR_UINT = 2
MP_OPCODE_OFFSET = 3
# extra bytes:
+MP_BC_UNWIND_JUMP = 0x46
MP_BC_MAKE_CLOSURE = 0x62
MP_BC_MAKE_CLOSURE_DEFARGS = 0x63
MP_BC_RAISE_VARARGS = 0x5c
@@ -215,7 +216,8 @@ def mp_opcode_format(bytecode, ip, count_var_uint, opcode_format=make_opcode_for
ip += 3
else:
extra_byte = (
- opcode == MP_BC_RAISE_VARARGS
+ opcode == MP_BC_UNWIND_JUMP
+ or opcode == MP_BC_RAISE_VARARGS
or opcode == MP_BC_MAKE_CLOSURE
or opcode == MP_BC_MAKE_CLOSURE_DEFARGS
)