#include "pyc_code.h" #include "pyc_module.h" #include "data.h" /* == Marshal structure for Code object == 1.0 1.3 1.5 2.1 2.3 3.0 3.8 3.11 argcount short short short long long long long posonlyargc long long kwonlyargc long long long nlocals short short short long long long stacksize short short long long long long flags short short short long long long long code Obj Obj Obj Obj Obj Obj Obj Obj consts Obj Obj Obj Obj Obj Obj Obj Obj names Obj Obj Obj Obj Obj Obj Obj Obj varnames Obj Obj Obj Obj Obj Obj freevars Obj Obj Obj Obj cellvars Obj Obj Obj Obj locals+names Obj locals+kinds Obj filename Obj Obj Obj Obj Obj Obj Obj Obj name Obj Obj Obj Obj Obj Obj Obj Obj qualname Obj firstline short short long long long long lntable Obj Obj Obj Obj Obj Obj exceptiontable Obj */ void PycCode::load(PycData* stream, PycModule* mod) { if (mod->verCompare(1, 3) >= 0 && mod->verCompare(2, 3) < 0) m_argCount = stream->get16(); else if (mod->verCompare(2, 3) >= 0) m_argCount = stream->get32(); if (mod->verCompare(3, 8) >= 0) m_posOnlyArgCount = stream->get32(); else m_posOnlyArgCount = 0; if (mod->majorVer() >= 3) m_kwOnlyArgCount = stream->get32(); else m_kwOnlyArgCount = 0; if (mod->verCompare(1, 3) >= 0 && mod->verCompare(2, 3) < 0) m_numLocals = stream->get16(); else if (mod->verCompare(2, 3) >= 0 && mod->verCompare(3, 11) < 0) m_numLocals = stream->get32(); else m_numLocals = 0; if (mod->verCompare(1, 5) >= 0 && mod->verCompare(2, 3) < 0) m_stackSize = stream->get16(); else if (mod->verCompare(2, 3) >= 0) m_stackSize = stream->get32(); else m_stackSize = 0; if (mod->verCompare(1, 3) >= 0 && mod->verCompare(2, 3) < 0) m_flags = stream->get16(); else if (mod->verCompare(2, 3) >= 0) m_flags = stream->get32(); else m_flags = 0; if (mod->verCompare(3, 8) < 0) { // Remap flags to new values introduced in 3.8 if (m_flags & 0xF0000000) throw std::runtime_error("Cannot remap unexpected flags"); m_flags = (m_flags & 0xFFFF) | ((m_flags & 0xFFF0000) << 4); } m_code = LoadObject(stream, mod).cast(); m_consts = LoadObject(stream, mod).cast(); m_names = LoadObject(stream, mod).cast(); if (mod->verCompare(1, 3) >= 0) m_localNames = LoadObject(stream, mod).cast(); else m_localNames = new PycTuple; if (mod->verCompare(3, 11) >= 0) m_localKinds = LoadObject(stream, mod).cast(); else m_localKinds = new PycString; if (mod->verCompare(2, 1) >= 0 && mod->verCompare(3, 11) < 0) m_freeVars = LoadObject(stream, mod).cast(); else m_freeVars = new PycTuple; if (mod->verCompare(2, 1) >= 0 && mod->verCompare(3, 11) < 0) m_cellVars = LoadObject(stream, mod).cast(); else m_cellVars = new PycTuple; m_fileName = LoadObject(stream, mod).cast(); m_name = LoadObject(stream, mod).cast(); if (mod->verCompare(3, 11) >= 0) m_qualName = LoadObject(stream, mod).cast(); else m_qualName = new PycString; if (mod->verCompare(1, 5) >= 0 && mod->verCompare(2, 3) < 0) m_firstLine = stream->get16(); else if (mod->verCompare(2, 3) >= 0) m_firstLine = stream->get32(); if (mod->verCompare(1, 5) >= 0) m_lnTable = LoadObject(stream, mod).cast(); else m_lnTable = new PycString; if (mod->verCompare(3, 11) >= 0) m_exceptTable = LoadObject(stream, mod).cast(); else m_exceptTable = new PycString; } PycRef PycCode::getCellVar(PycModule* mod, int idx) const { if (mod->verCompare(3, 11) >= 0) return getLocal(idx); return (idx >= m_cellVars->size()) ? m_freeVars->get(idx - m_cellVars->size()).cast() : m_cellVars->get(idx).cast(); }