diff --git a/bytecode.cpp b/bytecode.cpp index c6b7cba..0b067b2 100644 --- a/bytecode.cpp +++ b/bytecode.cpp @@ -600,3 +600,18 @@ void bc_disasm(std::ostream& pyc_output, PycRef code, PycModule* mod, pyc_output << "\n"; } } + +void bc_exceptiontable(std::ostream& pyc_output, PycRef code, + int indent) +{ + for (auto tuple: code->exceptTableEntries()) { + + for (int i=0; i(tuple) << " to " << std::get<1>(tuple); + pyc_output << " -> " << std::get<2>(tuple) << " "; + pyc_output << "[" << std::get<3>(tuple) << "] " << (std::get<4>(tuple) ? "lasti": ""); + pyc_output << "\n"; + } +} \ No newline at end of file diff --git a/bytecode.h b/bytecode.h index 7e4179e..3c0d9d3 100644 --- a/bytecode.h +++ b/bytecode.h @@ -32,3 +32,5 @@ void print_const(std::ostream& pyc_output, PycRef obj, PycModule* mod void bc_next(PycBuffer& source, PycModule* mod, int& opcode, int& operand, int& pos); void bc_disasm(std::ostream& pyc_output, PycRef code, PycModule* mod, int indent, unsigned flags); +void bc_exceptiontable(std::ostream& pyc_output, PycRef code, + int indent); diff --git a/pyc_code.cpp b/pyc_code.cpp index ba63eed..ec8be8e 100644 --- a/pyc_code.cpp +++ b/pyc_code.cpp @@ -128,3 +128,44 @@ PycRef PycCode::getCellVar(PycModule* mod, int idx) const ? m_freeVars->get(idx - m_cellVars->size()).cast() : m_cellVars->get(idx).cast(); } + +int _parse_varint(PycBuffer& data, int& pos) { + int b = data.getByte(); + pos += 1; + + int val = b & 63; + while (b & 64) { + val <<= 6; + + b = data.getByte(); + pos += 1; + + val |= (b & 63); + } + return val; +} + +std::vector PycCode::exceptTableEntries() const +{ + PycBuffer data(m_exceptTable->value(), m_exceptTable->length()); + + std::vector entries; + + int pos = 0; + while (!data.atEof()) { + + int start = _parse_varint(data, pos) * 2; + int length = _parse_varint(data, pos) * 2; + int end = start + length; + + int target = _parse_varint(data, pos) * 2; + int dl = _parse_varint(data, pos); + + int depth = dl >> 1; + bool lasti = bool(dl & 1); + + entries.emplace_back(start, end, target, depth, lasti); + } + + return entries; +} \ No newline at end of file diff --git a/pyc_code.h b/pyc_code.h index e6b2ce9..0a64ee5 100644 --- a/pyc_code.h +++ b/pyc_code.h @@ -87,6 +87,10 @@ public: m_globalsUsed.emplace_back(std::move(varname)); } + typedef std::tuple exception_table_entry_t; + + std::vector exceptTableEntries() const; + private: int m_argCount, m_posOnlyArgCount, m_kwOnlyArgCount, m_numLocals; int m_stackSize, m_flags; diff --git a/pycdas.cpp b/pycdas.cpp index b73410f..7b326b1 100644 --- a/pycdas.cpp +++ b/pycdas.cpp @@ -145,16 +145,16 @@ void output_object(PycRef obj, PycModule* mod, int indent, iputs(pyc_output, indent + 1, "[Disassembly]\n"); bc_disasm(pyc_output, codeObj, mod, indent + 2, flags); + if (mod->verCompare(3, 11) >= 0) { + iputs(pyc_output, indent + 1, "[Exception Table]\n"); + bc_exceptiontable(pyc_output, codeObj, indent+2); + } + if (mod->verCompare(1, 5) >= 0 && (flags & Pyc::DISASM_PYCODE_VERBOSE) != 0) { iprintf(pyc_output, indent + 1, "First Line: %d\n", codeObj->firstLine()); iputs(pyc_output, indent + 1, "[Line Number Table]\n"); output_object(codeObj->lnTable().cast(), mod, indent + 2, flags, pyc_output); } - - if (mod->verCompare(3, 11) >= 0 && (flags & Pyc::DISASM_PYCODE_VERBOSE) != 0) { - iputs(pyc_output, indent + 1, "[Exception Table]\n"); - output_object(codeObj->exceptTable().cast(), mod, indent + 2, flags, pyc_output); - } } break; case PycObject::TYPE_STRING: