60e4eb6316
Showing CACHE instructions and the extra PyCode fields in pycdas is now controlled by command-line arguments.
345 lines
12 KiB
C++
345 lines
12 KiB
C++
#include <cstdio>
|
|
#include <cstring>
|
|
#include <cstdarg>
|
|
#include <string>
|
|
#include "pyc_module.h"
|
|
#include "pyc_numeric.h"
|
|
#include "bytecode.h"
|
|
|
|
#ifdef WIN32
|
|
# define PATHSEP '\\'
|
|
#else
|
|
# define PATHSEP '/'
|
|
#endif
|
|
|
|
static const char* flag_names[] = {
|
|
"CO_OPTIMIZED", "CO_NEWLOCALS", "CO_VARARGS", "CO_VARKEYWORDS",
|
|
"CO_NESTED", "CO_GENERATOR", "CO_NOFREE", "CO_COROUTINE",
|
|
"CO_ITERABLE_COROUTINE", "<0x200>", "<0x400>", "<0x800>",
|
|
"CO_GENERATOR_ALLOWED", "CO_FUTURE_DIVISION",
|
|
"CO_FUTURE_ABSOLUTE_IMPORT", "CO_FUTURE_WITH_STATEMENT",
|
|
"CO_FUTURE_PRINT_FUNCTION", "CO_FUTURE_UNICODE_LITERALS",
|
|
"CO_FUTURE_BARRY_AS_BDFL", "CO_FUTURE_GENERATOR_STOP",
|
|
"<0x100000>", "<0x200000>", "<0x400000>", "<0x800000>",
|
|
"<0x1000000>", "<0x2000000>", "<0x4000000>", "<0x8000000>",
|
|
"<0x10000000>", "<0x20000000>", "<0x40000000>", "<0x80000000>"
|
|
};
|
|
|
|
static void print_coflags(unsigned long flags)
|
|
{
|
|
if (flags == 0) {
|
|
fputs("\n", pyc_output);
|
|
return;
|
|
}
|
|
|
|
fputs(" (", pyc_output);
|
|
unsigned long f = 1;
|
|
int k = 0;
|
|
while (k < 32) {
|
|
if ((flags & f) != 0) {
|
|
flags &= ~f;
|
|
if (flags == 0)
|
|
fputs(flag_names[k], pyc_output);
|
|
else
|
|
fprintf(pyc_output, "%s | ", flag_names[k]);
|
|
}
|
|
++k;
|
|
f <<= 1;
|
|
}
|
|
fputs(")\n", pyc_output);
|
|
}
|
|
|
|
static void iputs(int indent, const char* text)
|
|
{
|
|
for (int i=0; i<indent; i++)
|
|
fputs(" ", pyc_output);
|
|
fputs(text, pyc_output);
|
|
}
|
|
|
|
static void ivprintf(int indent, const char* fmt, va_list varargs)
|
|
{
|
|
for (int i=0; i<indent; i++)
|
|
fputs(" ", pyc_output);
|
|
vfprintf(pyc_output, fmt, varargs);
|
|
}
|
|
|
|
static void iprintf(int indent, const char* fmt, ...)
|
|
{
|
|
va_list varargs;
|
|
va_start(varargs, fmt);
|
|
ivprintf(indent, fmt, varargs);
|
|
va_end(varargs);
|
|
}
|
|
|
|
void output_object(PycRef<PycObject> obj, PycModule* mod, int indent,
|
|
unsigned flags)
|
|
{
|
|
if (obj == NULL) {
|
|
iputs(indent, "<NULL>");
|
|
return;
|
|
}
|
|
|
|
switch (obj->type()) {
|
|
case PycObject::TYPE_CODE:
|
|
case PycObject::TYPE_CODE2:
|
|
{
|
|
PycRef<PycCode> codeObj = obj.cast<PycCode>();
|
|
iputs(indent, "[Code]\n");
|
|
iprintf(indent + 1, "File Name: %s\n", codeObj->fileName()->value());
|
|
iprintf(indent + 1, "Object Name: %s\n", codeObj->name()->value());
|
|
if (mod->verCompare(3, 11) >= 0)
|
|
iprintf(indent + 1, "Qualified Name: %s\n", codeObj->qualName()->value());
|
|
iprintf(indent + 1, "Arg Count: %d\n", codeObj->argCount());
|
|
if (mod->verCompare(3, 8) >= 0)
|
|
iprintf(indent + 1, "Pos Only Arg Count: %d\n", codeObj->posOnlyArgCount());
|
|
if (mod->majorVer() >= 3)
|
|
iprintf(indent + 1, "KW Only Arg Count: %d\n", codeObj->kwOnlyArgCount());
|
|
if (mod->verCompare(3, 11) < 0)
|
|
iprintf(indent + 1, "Locals: %d\n", codeObj->numLocals());
|
|
if (mod->verCompare(1, 5) >= 0)
|
|
iprintf(indent + 1, "Stack Size: %d\n", codeObj->stackSize());
|
|
if (mod->verCompare(1, 3) >= 0) {
|
|
iprintf(indent + 1, "Flags: 0x%08X", codeObj->flags());
|
|
print_coflags(codeObj->flags());
|
|
}
|
|
|
|
iputs(indent + 1, "[Names]\n");
|
|
for (int i=0; i<codeObj->names()->size(); i++)
|
|
output_object(codeObj->names()->get(i), mod, indent + 2, flags);
|
|
|
|
if (mod->verCompare(1, 3) >= 0 && mod->verCompare(3, 11) < 0) {
|
|
if (mod->verCompare(3, 11) >= 0)
|
|
iputs(indent + 1, "[Locals+Names]\n");
|
|
else
|
|
iputs(indent + 1, "[Var Names]\n");
|
|
for (int i=0; i<codeObj->localNames()->size(); i++)
|
|
output_object(codeObj->localNames()->get(i), mod, indent + 2, flags);
|
|
}
|
|
|
|
if (mod->verCompare(3, 11) >= 0 && (flags & Pyc::DISASM_PYCODE_VERBOSE) != 0) {
|
|
iputs(indent + 1, "[Locals+Kinds]\n");
|
|
output_object(codeObj->localKinds().cast<PycObject>(), mod, indent + 2, flags);
|
|
}
|
|
|
|
if (mod->verCompare(2, 1) >= 0 && mod->verCompare(3, 11) < 0) {
|
|
iputs(indent + 1, "[Free Vars]\n");
|
|
for (int i=0; i<codeObj->freeVars()->size(); i++)
|
|
output_object(codeObj->freeVars()->get(i), mod, indent + 2, flags);
|
|
|
|
iputs(indent + 1, "[Cell Vars]\n");
|
|
for (int i=0; i<codeObj->cellVars()->size(); i++)
|
|
output_object(codeObj->cellVars()->get(i), mod, indent + 2, flags);
|
|
}
|
|
|
|
iputs(indent + 1, "[Constants]\n");
|
|
for (int i=0; i<codeObj->consts()->size(); i++)
|
|
output_object(codeObj->consts()->get(i), mod, indent + 2, flags);
|
|
|
|
iputs(indent + 1, "[Disassembly]\n");
|
|
bc_disasm(codeObj, mod, indent + 2, flags);
|
|
|
|
if (mod->verCompare(1, 5) >= 0 && (flags & Pyc::DISASM_PYCODE_VERBOSE) != 0) {
|
|
iputs(indent + 1, "[Line Number Table]\n");
|
|
output_object(codeObj->lnTable().cast<PycObject>(), mod, indent + 2, flags);
|
|
}
|
|
|
|
if (mod->verCompare(3, 11) >= 0 && (flags & Pyc::DISASM_PYCODE_VERBOSE) != 0) {
|
|
iputs(indent + 1, "[Exception Table]\n");
|
|
output_object(codeObj->exceptTable().cast<PycObject>(), mod, indent + 2, flags);
|
|
}
|
|
}
|
|
break;
|
|
case PycObject::TYPE_STRING:
|
|
iputs(indent, "");
|
|
OutputString(obj.cast<PycString>(), mod->strIsUnicode() ? 'b' : 0);
|
|
fputs("\n", pyc_output);
|
|
break;
|
|
case PycObject::TYPE_UNICODE:
|
|
iputs(indent, "");
|
|
OutputString(obj.cast<PycString>(), mod->strIsUnicode() ? 0 : 'u');
|
|
fputs("\n", pyc_output);
|
|
break;
|
|
case PycObject::TYPE_INTERNED:
|
|
case PycObject::TYPE_ASCII:
|
|
case PycObject::TYPE_ASCII_INTERNED:
|
|
case PycObject::TYPE_SHORT_ASCII:
|
|
case PycObject::TYPE_SHORT_ASCII_INTERNED:
|
|
iputs(indent, "");
|
|
if (mod->majorVer() >= 3)
|
|
OutputString(obj.cast<PycString>(), 0);
|
|
else
|
|
OutputString(obj.cast<PycString>(), mod->strIsUnicode() ? 'b' : 0);
|
|
fputs("\n", pyc_output);
|
|
break;
|
|
case PycObject::TYPE_TUPLE:
|
|
case PycObject::TYPE_SMALL_TUPLE:
|
|
{
|
|
iputs(indent, "(\n");
|
|
for (const auto& val : obj.cast<PycTuple>()->values())
|
|
output_object(val, mod, indent + 1, flags);
|
|
iputs(indent, ")\n");
|
|
}
|
|
break;
|
|
case PycObject::TYPE_LIST:
|
|
{
|
|
iputs(indent, "[\n");
|
|
for (const auto& val : obj.cast<PycList>()->values())
|
|
output_object(val, mod, indent + 1, flags);
|
|
iputs(indent, "]\n");
|
|
}
|
|
break;
|
|
case PycObject::TYPE_DICT:
|
|
{
|
|
iputs(indent, "{\n");
|
|
PycDict::key_t keys = obj.cast<PycDict>()->keys();
|
|
PycDict::value_t values = obj.cast<PycDict>()->values();
|
|
PycDict::key_t::const_iterator ki = keys.begin();
|
|
PycDict::value_t::const_iterator vi = values.begin();
|
|
while (ki != keys.end()) {
|
|
output_object(*ki, mod, indent + 1, flags);
|
|
output_object(*vi, mod, indent + 2, flags);
|
|
++ki, ++vi;
|
|
}
|
|
iputs(indent, "}\n");
|
|
}
|
|
break;
|
|
case PycObject::TYPE_SET:
|
|
{
|
|
iputs(indent, "{\n");
|
|
for (const auto& val : obj.cast<PycSet>()->values())
|
|
output_object(val, mod, indent + 1, flags);
|
|
iputs(indent, "}\n");
|
|
}
|
|
break;
|
|
case PycObject::TYPE_NONE:
|
|
iputs(indent, "None\n");
|
|
break;
|
|
case PycObject::TYPE_FALSE:
|
|
iputs(indent, "False\n");
|
|
break;
|
|
case PycObject::TYPE_TRUE:
|
|
iputs(indent, "True\n");
|
|
break;
|
|
case PycObject::TYPE_ELLIPSIS:
|
|
iputs(indent, "...\n");
|
|
break;
|
|
case PycObject::TYPE_INT:
|
|
iprintf(indent, "%d\n", obj.cast<PycInt>()->value());
|
|
break;
|
|
case PycObject::TYPE_LONG:
|
|
iprintf(indent, "%s\n", obj.cast<PycLong>()->repr().c_str());
|
|
break;
|
|
case PycObject::TYPE_FLOAT:
|
|
iprintf(indent, "%s\n", obj.cast<PycFloat>()->value());
|
|
break;
|
|
case PycObject::TYPE_COMPLEX:
|
|
iprintf(indent, "(%s+%sj)\n", obj.cast<PycComplex>()->value(),
|
|
obj.cast<PycComplex>()->imag());
|
|
break;
|
|
case PycObject::TYPE_BINARY_FLOAT:
|
|
iprintf(indent, "%g\n", obj.cast<PycCFloat>()->value());
|
|
break;
|
|
case PycObject::TYPE_BINARY_COMPLEX:
|
|
iprintf(indent, "(%g+%gj)\n", obj.cast<PycCComplex>()->value(),
|
|
obj.cast<PycCComplex>()->imag());
|
|
break;
|
|
default:
|
|
iprintf(indent, "<TYPE: %d>\n", obj->type());
|
|
}
|
|
}
|
|
|
|
int main(int argc, char* argv[])
|
|
{
|
|
const char* infile = nullptr;
|
|
bool marshalled = false;
|
|
const char* version = nullptr;
|
|
unsigned disasm_flags = 0;
|
|
|
|
for (int arg = 1; arg < argc; ++arg) {
|
|
if (strcmp(argv[arg], "-o") == 0) {
|
|
if (arg + 1 < argc) {
|
|
const char* filename = argv[++arg];
|
|
FILE* outfile = fopen(filename, "w");
|
|
if (!outfile) {
|
|
fprintf(stderr, "Error opening file '%s' for writing\n",
|
|
argv[arg]);
|
|
return 1;
|
|
}
|
|
pyc_output = outfile;
|
|
} else {
|
|
fputs("Option '-o' requires a filename\n", stderr);
|
|
return 1;
|
|
}
|
|
} else if (strcmp(argv[arg], "-c") == 0) {
|
|
marshalled = true;
|
|
} else if (strcmp(argv[arg], "-v") == 0) {
|
|
if (arg + 1 < argc) {
|
|
version = argv[++arg];
|
|
} else {
|
|
fputs("Option '-v' requires a version\n", stderr);
|
|
return 1;
|
|
}
|
|
} else if (strcmp(argv[arg], "--pycode-extra") == 0) {
|
|
disasm_flags |= Pyc::DISASM_PYCODE_VERBOSE;
|
|
} else if (strcmp(argv[arg], "--show-caches") == 0) {
|
|
disasm_flags |= Pyc::DISASM_SHOW_CACHES;
|
|
} else if (strcmp(argv[arg], "--help") == 0 || strcmp(argv[arg], "-h") == 0) {
|
|
fprintf(stderr, "Usage: %s [options] input.pyc\n\n", argv[0]);
|
|
fputs("Options:\n", stderr);
|
|
fputs(" -o <filename> Write output to <filename> (default: stdout)\n", stderr);
|
|
fputs(" -c Specify loading a compiled code object. Requires the version to be set\n", stderr);
|
|
fputs(" -v <x.y> Specify a Python version for loading a compiled code object\n", stderr);
|
|
fputs(" --pycode-extra Show extra fields in PyCode object dumps\n", stderr);
|
|
fputs(" --show-caches Don't suprress CACHE instructions in Python 3.11+ disassembly\n", stderr);
|
|
fputs(" --help Show this help text and then exit\n", stderr);
|
|
return 0;
|
|
} else if (argv[arg][0] == '-') {
|
|
fprintf(stderr, "Error: Unrecognized argument %s\n", argv[arg]);
|
|
return 1;
|
|
} else {
|
|
infile = argv[arg];
|
|
}
|
|
}
|
|
|
|
if (!infile) {
|
|
fputs("No input file specified\n", stderr);
|
|
return 1;
|
|
}
|
|
|
|
PycModule mod;
|
|
if (!marshalled) {
|
|
try {
|
|
mod.loadFromFile(infile);
|
|
} catch (std::exception &ex) {
|
|
fprintf(stderr, "Error disassembling %s: %s\n", infile, ex.what());
|
|
return 1;
|
|
}
|
|
} else {
|
|
if (!version) {
|
|
fputs("Opening raw code objects requires a version to be specified\n", stderr);
|
|
return 1;
|
|
}
|
|
std::string s(version);
|
|
auto dot = s.find('.');
|
|
if (dot == std::string::npos || dot == s.size()-1) {
|
|
fputs("Unable to parse version string (use the format x.y)\n", stderr);
|
|
return 1;
|
|
}
|
|
int major = std::stoi(s.substr(0, dot));
|
|
int minor = std::stoi(s.substr(dot+1, s.size()));
|
|
mod.loadFromMarshalledFile(infile, major, minor);
|
|
}
|
|
const char* dispname = strrchr(infile, PATHSEP);
|
|
dispname = (dispname == NULL) ? infile : dispname + 1;
|
|
fprintf(pyc_output, "%s (Python %d.%d%s)\n", dispname, mod.majorVer(), mod.minorVer(),
|
|
(mod.majorVer() < 3 && mod.isUnicode()) ? " -U" : "");
|
|
try {
|
|
output_object(mod.code().try_cast<PycObject>(), &mod, 0, disasm_flags);
|
|
} catch (std::exception& ex) {
|
|
fprintf(stderr, "Error disassembling %s: %s\n", infile, ex.what());
|
|
return 1;
|
|
}
|
|
|
|
return 0;
|
|
}
|