Pyarmor-Static-Unpack-1shot/pyc_object.cpp

#include "pyc_object.h"
#include "pyc_module.h"
#include "pyc_numeric.h"
#include "pyc_code.h"
#include "data.h"
#include <cstdio>

PycRef<PycObject> Pyc_NULL = (PycObject*)0;
PycRef<PycObject> Pyc_None = new PycObject(PycObject::TYPE_NONE);
PycRef<PycObject> Pyc_Ellipsis = new PycObject(PycObject::TYPE_ELLIPSIS);
PycRef<PycObject> Pyc_StopIteration = new PycObject(PycObject::TYPE_STOPITER);
PycRef<PycObject> Pyc_False = new PycObject(PycObject::TYPE_FALSE);
PycRef<PycObject> Pyc_True = new PycObject(PycObject::TYPE_TRUE);

PycRef<PycObject> CreateObject(int type)
{
    switch (type) {
    case PycObject::TYPE_NULL:
        return Pyc_NULL;
    case PycObject::TYPE_NONE:
        return Pyc_None;
    case PycObject::TYPE_FALSE:
        return Pyc_False;
    case PycObject::TYPE_TRUE:
        return Pyc_True;
    case PycObject::TYPE_STOPITER:
        return Pyc_StopIteration;
    case PycObject::TYPE_ELLIPSIS:
        return Pyc_Ellipsis;
    case PycObject::TYPE_INT:
        return new PycInt();
    case PycObject::TYPE_INT64:
        return new PycLong(PycObject::TYPE_INT64);
    case PycObject::TYPE_FLOAT:
        return new PycFloat();
    case PycObject::TYPE_BINARY_FLOAT:
        return new PycCFloat();
    case PycObject::TYPE_COMPLEX:
        return new PycComplex();
    case PycObject::TYPE_BINARY_COMPLEX:
        return new PycCComplex();
    case PycObject::TYPE_LONG:
        return new PycLong();
    case PycObject::TYPE_STRING:
        return new PycString();
    case PycObject::TYPE_INTERNED:
        return new PycString(PycObject::TYPE_INTERNED);
    case PycObject::TYPE_STRINGREF:
        return new PycString(PycObject::TYPE_STRINGREF);
    case PycObject::TYPE_TUPLE:
        return new PycTuple();
    case PycObject::TYPE_LIST:
        return new PycList();
    case PycObject::TYPE_DICT:
        return new PycDict();
    case PycObject::TYPE_CODE:
    case PycObject::TYPE_CODE2:
        return new PycCode();
    case PycObject::TYPE_UNICODE:
        return new PycString(PycObject::TYPE_UNICODE);
    case PycObject::TYPE_SET:
        return new PycSet();
    case PycObject::TYPE_FROZENSET:
        return new PycSet(PycObject::TYPE_FROZENSET);
    default:
        fprintf(stderr, "CreateObject: Got unsupported type 0x%X\n", type);
        return Pyc_NULL;
    }
}

PycRef<PycObject> LoadObject(PycData* stream, PycModule* mod)
{
    PycRef<PycObject> obj = CreateObject(stream->getByte());
    if (obj != Pyc_NULL)
        obj->load(stream, mod);
    return obj;
}