pe-parse/pepy/pepy.cpp

/*
 * Copyright (c) 2013, Wesley Shields <wxs@atarininja.org>. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include <Python.h>
#include <pe-parse/parse.h>
#include <structmember.h>

using namespace peparse;

/* NOTE(ww): These don't necessarily have to be the same, but currently are.
 */
#define PEPY_VERSION PEPARSE_VERSION

/*
 * Add some definition for compatibility between python2 and python3
 */
#if PY_MAJOR_VERSION >= 3
#define PyString_FromString PyUnicode_FromString
#endif

/*
 * Some macro only available after python 2.6
 * Needed for compatibility with python3
 */
#ifndef PyVarObject_HEAD_INIT
#define PyVarObject_HEAD_INIT(type, size) PyObject_HEAD_INIT(type) size,
#endif

#ifndef Py_TYPE
#define Py_TYPE(_ob_) (((PyObject *) (_ob_))->ob_type)
#endif

/* These are used to across multiple objects. */
#define PEPY_OBJECT_GET(OBJ, ATTR)                                          \
  static PyObject *pepy_##OBJ##_get_##ATTR(PyObject *self, void *closure) { \
    Py_INCREF(((pepy_##OBJ *) self)->ATTR);                                 \
    return ((pepy_##OBJ *) self)->ATTR;                                     \
  }

#define OBJECTGETTER(OBJ, ATTR, DOC)                         \
  {                                                          \
    (char *) #ATTR, (getter) pepy_##OBJ##_get_##ATTR,        \
        (setter) pepy_attr_not_writable, (char *) #DOC, NULL \
  }

/* 'OPTIONAL' references the fact that these are from the Optional Header */
#define OBJECTGETTER_OPTIONAL(ATTR, DOC)                      \
  {                                                           \
    (char *) #ATTR, (getter) pepy_parsed_get_optional_##ATTR, \
        (setter) pepy_attr_not_writable, (char *) #DOC, NULL  \
  }

static PyObject *pepy_error;

struct pepy {
  PyObject_HEAD
};

struct pepy_parsed {
  PyObject_HEAD parsed_pe *pe;
};

struct pepy_section {
  PyObject_HEAD PyObject *name;
  PyObject *base;
  PyObject *length;
  PyObject *virtaddr;
  PyObject *virtsize;
  PyObject *numrelocs;
  PyObject *numlinenums;
  PyObject *characteristics;
  PyObject *data;
};

struct pepy_resource {
  PyObject_HEAD PyObject *type_str;
  PyObject *name_str;
  PyObject *lang_str;
  PyObject *type;
  PyObject *name;
  PyObject *lang;
  PyObject *codepage;
  PyObject *RVA;
  PyObject *size;
  PyObject *data;
};

struct pepy_import {
  PyObject_HEAD PyObject *name;
  PyObject *sym;
  PyObject *addr;
};

struct pepy_export {
  PyObject_HEAD PyObject *mod;
  PyObject *func;
  PyObject *addr;
};

struct pepy_relocation {
  PyObject_HEAD PyObject *type;
  PyObject *addr;
};

/* None of the attributes in these objects are writable. */
static int
pepy_attr_not_writable(PyObject *self, PyObject *value, void *closure) {
  PyErr_SetString(PyExc_TypeError, "Attribute not writable.");
  return -1;
}

static PyObject *
pepy_import_new(PyTypeObject *type, PyObject *args, PyObject *kwds) {
  pepy_import *self;

  self = (pepy_import *) type->tp_alloc(type, 0);

  return (PyObject *) self;
}

static int pepy_import_init(pepy_import *self, PyObject *args, PyObject *kwds) {
  if (!PyArg_ParseTuple(
          args, "OOO:pepy_import_init", &self->name, &self->sym, &self->addr))
    return -1;
  return 0;
}

static void pepy_import_dealloc(pepy_import *self) {
  Py_XDECREF(self->name);
  Py_XDECREF(self->sym);
  Py_XDECREF(self->addr);
  Py_TYPE(self)->tp_free((PyObject *) self);
}

PEPY_OBJECT_GET(import, name)
PEPY_OBJECT_GET(import, sym)
PEPY_OBJECT_GET(import, addr)

static PyGetSetDef pepy_import_getseters[] = {
    OBJECTGETTER(import, name, "Name"),
    OBJECTGETTER(import, sym, "Symbol"),
    OBJECTGETTER(import, addr, "Address"),
    {NULL}};

static PyTypeObject pepy_import_type = {
    PyVarObject_HEAD_INIT(NULL, 0)    /* ob_size */
    "pepy.import",                    /* tp_name */
    sizeof(pepy_import),              /* tp_basicsize */
    0,                                /* tp_itemsize */
    (destructor) pepy_import_dealloc, /* tp_dealloc */
    0,                                /* tp_print */
    0,                                /* tp_getattr */
    0,                                /* tp_setattr */
    0,                                /* tp_compare */
    0,                                /* tp_repr */
    0,                                /* tp_as_number */
    0,                                /* tp_as_sequence */
    0,                                /* tp_as_mapping */
    0,                                /* tp_hash */
    0,                                /* tp_call */
    0,                                /* tp_str */
    0,                                /* tp_getattro */
    0,                                /* tp_setattro */
    0,                                /* tp_as_buffer */
    Py_TPFLAGS_DEFAULT,               /* tp_flags */
    "pepy import object",             /* tp_doc */
    0,                                /* tp_traverse */
    0,                                /* tp_clear */
    0,                                /* tp_richcompare */
    0,                                /* tp_weaklistoffset */
    0,                                /* tp_iter */
    0,                                /* tp_iternext */
    0,                                /* tp_methods */
    0,                                /* tp_members */
    pepy_import_getseters,            /* tp_getset */
    0,                                /* tp_base */
    0,                                /* tp_dict */
    0,                                /* tp_descr_get */
    0,                                /* tp_descr_set */
    0,                                /* tp_dictoffset */
    (initproc) pepy_import_init,      /* tp_init */
    0,                                /* tp_alloc */
    pepy_import_new                   /* tp_new */
};

static PyObject *
pepy_export_new(PyTypeObject *type, PyObject *args, PyObject *kwds) {
  pepy_export *self;

  self = (pepy_export *) type->tp_alloc(type, 0);

  return (PyObject *) self;
}

static int pepy_export_init(pepy_export *self, PyObject *args, PyObject *kwds) {
  if (!PyArg_ParseTuple(
          args, "OOO:pepy_export_init", &self->mod, &self->func, &self->addr))
    return -1;
  return 0;
}

static void pepy_export_dealloc(pepy_export *self) {
  Py_XDECREF(self->mod);
  Py_XDECREF(self->func);
  Py_XDECREF(self->addr);
  Py_TYPE(self)->tp_free((PyObject *) self);
}

PEPY_OBJECT_GET(export, mod)
PEPY_OBJECT_GET(export, func)
PEPY_OBJECT_GET(export, addr)

static PyGetSetDef pepy_export_getseters[] = {
    OBJECTGETTER(export, mod, "Module"),
    OBJECTGETTER(export, func, "Function"),
    OBJECTGETTER(export, addr, "Address"),
    {NULL}};

static PyTypeObject pepy_export_type = {
    PyVarObject_HEAD_INIT(NULL, 0)    /* ob_size */
    "pepy.export",                    /* tp_name */
    sizeof(pepy_export),              /* tp_basicsize */
    0,                                /* tp_itemsize */
    (destructor) pepy_export_dealloc, /* tp_dealloc */
    0,                                /* tp_print */
    0,                                /* tp_getattr */
    0,                                /* tp_setattr */
    0,                                /* tp_compare */
    0,                                /* tp_repr */
    0,                                /* tp_as_number */
    0,                                /* tp_as_sequence */
    0,                                /* tp_as_mapping */
    0,                                /* tp_hash */
    0,                                /* tp_call */
    0,                                /* tp_str */
    0,                                /* tp_getattro */
    0,                                /* tp_setattro */
    0,                                /* tp_as_buffer */
    Py_TPFLAGS_DEFAULT,               /* tp_flags */
    "pepy export object",             /* tp_doc */
    0,                                /* tp_traverse */
    0,                                /* tp_clear */
    0,                                /* tp_richcompare */
    0,                                /* tp_weaklistoffset */
    0,                                /* tp_iter */
    0,                                /* tp_iternext */
    0,                                /* tp_methods */
    0,                                /* tp_members */
    pepy_export_getseters,            /* tp_getset */
    0,                                /* tp_base */
    0,                                /* tp_dict */
    0,                                /* tp_descr_get */
    0,                                /* tp_descr_set */
    0,                                /* tp_dictoffset */
    (initproc) pepy_export_init,      /* tp_init */
    0,                                /* tp_alloc */
    pepy_export_new                   /* tp_new */
};

static PyObject *
pepy_relocation_new(PyTypeObject *type, PyObject *args, PyObject *kwds) {
  pepy_relocation *self;

  self = (pepy_relocation *) type->tp_alloc(type, 0);

  return (PyObject *) self;
}

static int
pepy_relocation_init(pepy_relocation *self, PyObject *args, PyObject *kwds) {
  if (!PyArg_ParseTuple(
          args, "OO:pepy_relocation_init", &self->type, &self->addr))
    return -1;
  return 0;
}

static void pepy_relocation_dealloc(pepy_relocation *self) {
  Py_XDECREF(self->type);
  Py_XDECREF(self->addr);
  Py_TYPE(self)->tp_free((PyObject *) self);
}

PEPY_OBJECT_GET(relocation, type)
PEPY_OBJECT_GET(relocation, addr)

static PyGetSetDef pepy_relocation_getseters[] = {
    OBJECTGETTER(relocation, type, "Type"),
    OBJECTGETTER(relocation, addr, "Address"),
    {NULL}};

static PyTypeObject pepy_relocation_type = {
    PyVarObject_HEAD_INIT(NULL, 0)        /* ob_size */
    "pepy.relocation",                    /* tp_name */
    sizeof(pepy_relocation),              /* tp_basicsize */
    0,                                    /* tp_itemsize */
    (destructor) pepy_relocation_dealloc, /* tp_dealloc */
    0,                                    /* tp_print */
    0,                                    /* tp_getattr */
    0,                                    /* tp_setattr */
    0,                                    /* tp_compare */
    0,                                    /* tp_repr */
    0,                                    /* tp_as_number */
    0,                                    /* tp_as_sequence */
    0,                                    /* tp_as_mapping */
    0,                                    /* tp_hash */
    0,                                    /* tp_call */
    0,                                    /* tp_str */
    0,                                    /* tp_getattro */
    0,                                    /* tp_setattro */
    0,                                    /* tp_as_buffer */
    Py_TPFLAGS_DEFAULT,                   /* tp_flags */
    "pepy relocation object",             /* tp_doc */
    0,                                    /* tp_traverse */
    0,                                    /* tp_clear */
    0,                                    /* tp_richcompare */
    0,                                    /* tp_weaklistoffset */
    0,                                    /* tp_iter */
    0,                                    /* tp_iternext */
    0,                                    /* tp_methods */
    0,                                    /* tp_members */
    pepy_relocation_getseters,            /* tp_getset */
    0,                                    /* tp_base */
    0,                                    /* tp_dict */
    0,                                    /* tp_descr_get */
    0,                                    /* tp_descr_set */
    0,                                    /* tp_dictoffset */
    (initproc) pepy_relocation_init,      /* tp_init */
    0,                                    /* tp_alloc */
    pepy_relocation_new                   /* tp_new */
};

static PyObject *
pepy_section_new(PyTypeObject *type, PyObject *args, PyObject *kwds) {
  pepy_section *self;

  self = (pepy_section *) type->tp_alloc(type, 0);

  return (PyObject *) self;
}

static int
pepy_section_init(pepy_section *self, PyObject *args, PyObject *kwds) {
  if (!PyArg_ParseTuple(args,
                        "OOOOOOOOO:pepy_section_init",
                        &self->name,
                        &self->base,
                        &self->length,
                        &self->virtaddr,
                        &self->virtsize,
                        &self->numrelocs,
                        &self->numlinenums,
                        &self->characteristics,
                        &self->data))
    return -1;
  return 0;
}

static void pepy_section_dealloc(pepy_section *self) {
  Py_XDECREF(self->name);
  Py_XDECREF(self->base);
  Py_XDECREF(self->length);
  Py_XDECREF(self->virtaddr);
  Py_XDECREF(self->virtsize);
  Py_XDECREF(self->numrelocs);
  Py_XDECREF(self->numlinenums);
  Py_XDECREF(self->characteristics);
  Py_XDECREF(self->data);
  Py_TYPE(self)->tp_free((PyObject *) self);
}

PEPY_OBJECT_GET(section, name)
PEPY_OBJECT_GET(section, base)
PEPY_OBJECT_GET(section, length)
PEPY_OBJECT_GET(section, virtaddr)
PEPY_OBJECT_GET(section, virtsize)
PEPY_OBJECT_GET(section, numrelocs)
PEPY_OBJECT_GET(section, numlinenums)
PEPY_OBJECT_GET(section, characteristics)
PEPY_OBJECT_GET(section, data)

static PyGetSetDef pepy_section_getseters[] = {
    OBJECTGETTER(section, name, "Name"),
    OBJECTGETTER(section, base, "Base address"),
    OBJECTGETTER(section, length, "Length"),
    OBJECTGETTER(section, virtaddr, "Virtual address"),
    OBJECTGETTER(section, virtsize, "Virtual size"),
    OBJECTGETTER(section, numrelocs, "Number of relocations"),
    OBJECTGETTER(section, numlinenums, "Number of line numbers"),
    OBJECTGETTER(section, characteristics, "Characteristics"),
    OBJECTGETTER(section, data, "Section data"),
    {NULL}};

static PyTypeObject pepy_section_type = {
    PyVarObject_HEAD_INIT(NULL, 0)     /* ob_size */
    "pepy.section",                    /* tp_name */
    sizeof(pepy_section),              /* tp_basicsize */
    0,                                 /* tp_itemsize */
    (destructor) pepy_section_dealloc, /* tp_dealloc */
    0,                                 /* tp_print */
    0,                                 /* tp_getattr */
    0,                                 /* tp_setattr */
    0,                                 /* tp_compare */
    0,                                 /* tp_repr */
    0,                                 /* tp_as_number */
    0,                                 /* tp_as_sequence */
    0,                                 /* tp_as_mapping */
    0,                                 /* tp_hash */
    0,                                 /* tp_call */
    0,                                 /* tp_str */
    0,                                 /* tp_getattro */
    0,                                 /* tp_setattro */
    0,                                 /* tp_as_buffer */
    Py_TPFLAGS_DEFAULT,                /* tp_flags */
    "pepy section object",             /* tp_doc */
    0,                                 /* tp_traverse */
    0,                                 /* tp_clear */
    0,                                 /* tp_richcompare */
    0,                                 /* tp_weaklistoffset */
    0,                                 /* tp_iter */
    0,                                 /* tp_iternext */
    0,                                 /* tp_methods */
    0,                                 /* tp_members */
    pepy_section_getseters,            /* tp_getset */
    0,                                 /* tp_base */
    0,                                 /* tp_dict */
    0,                                 /* tp_descr_get */
    0,                                 /* tp_descr_set */
    0,                                 /* tp_dictoffset */
    (initproc) pepy_section_init,      /* tp_init */
    0,                                 /* tp_alloc */
    pepy_section_new                   /* tp_new */
};

static PyObject *
pepy_resource_new(PyTypeObject *type, PyObject *args, PyObject *kwds) {
  pepy_resource *self;

  self = (pepy_resource *) type->tp_alloc(type, 0);

  return (PyObject *) self;
}

static int
pepy_resource_init(pepy_resource *self, PyObject *args, PyObject *kwds) {
  if (!PyArg_ParseTuple(args,
                        "OOOOOOOOOO:pepy_resource_init",
                        &self->type_str,
                        &self->name_str,
                        &self->lang_str,
                        &self->type,
                        &self->name,
                        &self->lang,
                        &self->codepage,
                        &self->RVA,
                        &self->size,
                        &self->data))
    return -1;

  return 0;
}

static void pepy_resource_dealloc(pepy_resource *self) {
  Py_XDECREF(self->type_str);
  Py_XDECREF(self->name_str);
  Py_XDECREF(self->lang_str);
  Py_XDECREF(self->type);
  Py_XDECREF(self->name);
  Py_XDECREF(self->lang);
  Py_XDECREF(self->codepage);
  Py_XDECREF(self->RVA);
  Py_XDECREF(self->size);
  Py_XDECREF(self->data);
  Py_TYPE(self)->tp_free((PyObject *) self);
}

PEPY_OBJECT_GET(resource, type_str)
PEPY_OBJECT_GET(resource, name_str)
PEPY_OBJECT_GET(resource, lang_str)
PEPY_OBJECT_GET(resource, type)
PEPY_OBJECT_GET(resource, name)
PEPY_OBJECT_GET(resource, lang)
PEPY_OBJECT_GET(resource, codepage)
PEPY_OBJECT_GET(resource, RVA)
PEPY_OBJECT_GET(resource, size)
PEPY_OBJECT_GET(resource, data)

static PyObject *pepy_resource_type_as_str(PyObject *self, PyObject *args) {
  PyObject *ret;
  char *str;
  long type;

  type = PyLong_AsUnsignedLong(((pepy_resource *) self)->type);
  if (type == -1) {
    if (PyErr_Occurred()) {
      PyErr_PrintEx(0);
      return NULL;
    }
  }
  switch ((resource_type) type) {
    case (RT_CURSOR):
      str = (char *) "CURSOR";
      break;
    case (RT_BITMAP):
      str = (char *) "BITMAP";
      break;
    case (RT_ICON):
      str = (char *) "ICON";
      break;
    case (RT_MENU):
      str = (char *) "MENU";
      break;
    case (RT_DIALOG):
      str = (char *) "DIALOG";
      break;
    case (RT_STRING):
      str = (char *) "STRING";
      break;
    case (RT_FONTDIR):
      str = (char *) "FONTDIR";
      break;
    case (RT_FONT):
      str = (char *) "FONT";
      break;
    case (RT_ACCELERATOR):
      str = (char *) "ACCELERATOR";
      break;
    case (RT_RCDATA):
      str = (char *) "RCDATA";
      break;
    case (RT_MESSAGETABLE):
      str = (char *) "MESSAGETABLE";
      break;
    case (RT_GROUP_CURSOR):
      str = (char *) "GROUP_CURSOR";
      break;
    case (RT_GROUP_ICON):
      str = (char *) "GROUP_ICON";
      break;
    case (RT_VERSION):
      str = (char *) "VERSION";
      break;
    case (RT_DLGINCLUDE):
      str = (char *) "DLGINCLUDE";
      break;
    case (RT_PLUGPLAY):
      str = (char *) "PLUGPLAY";
      break;
    case (RT_VXD):
      str = (char *) "VXD";
      break;
    case (RT_ANICURSOR):
      str = (char *) "ANICURSOR";
      break;
    case (RT_ANIICON):
      str = (char *) "ANIICON";
      break;
    case (RT_HTML):
      str = (char *) "HTML";
      break;
    case (RT_MANIFEST):
      str = (char *) "MANIFEST";
      break;
    default:
      str = (char *) "UNKNOWN";
      break;
  }

  ret = PyString_FromString(str);
  if (!ret) {
    PyErr_SetString(pepy_error, "Unable to create return string.");
    return NULL;
  }

  return ret;
}

static PyMethodDef pepy_resource_methods[] = {
    {"type_as_str",
     pepy_resource_type_as_str,
     METH_NOARGS,
     "Return the resource type as a string."},
    {NULL}};

static PyGetSetDef pepy_resource_getseters[] = {
    OBJECTGETTER(resource, type_str, "Type string"),
    OBJECTGETTER(resource, name_str, "Name string"),
    OBJECTGETTER(resource, lang_str, "Lang string"),
    OBJECTGETTER(resource, type, "Type"),
    OBJECTGETTER(resource, name, "Name"),
    OBJECTGETTER(resource, lang, "Language"),
    OBJECTGETTER(resource, codepage, "Codepage"),
    OBJECTGETTER(resource, RVA, "RVA"),
    OBJECTGETTER(resource, size, "Size (specified in RDAT)"),
    OBJECTGETTER(resource, data, "Resource data"),
    {NULL}};

static PyTypeObject pepy_resource_type = {
    PyVarObject_HEAD_INIT(NULL, 0)      /* ob_size */
    "pepy.resource",                    /* tp_name */
    sizeof(pepy_resource),              /* tp_basicsize */
    0,                                  /* tp_itemsize */
    (destructor) pepy_resource_dealloc, /* tp_dealloc */
    0,                                  /* tp_print */
    0,                                  /* tp_getattr */
    0,                                  /* tp_setattr */
    0,                                  /* tp_compare */
    0,                                  /* tp_repr */
    0,                                  /* tp_as_number */
    0,                                  /* tp_as_sequence */
    0,                                  /* tp_as_mapping */
    0,                                  /* tp_hash */
    0,                                  /* tp_call */
    0,                                  /* tp_str */
    0,                                  /* tp_getattro */
    0,                                  /* tp_setattro */
    0,                                  /* tp_as_buffer */
    Py_TPFLAGS_DEFAULT,                 /* tp_flags */
    "pepy resource object",             /* tp_doc */
    0,                                  /* tp_traverse */
    0,                                  /* tp_clear */
    0,                                  /* tp_richcompare */
    0,                                  /* tp_weaklistoffset */
    0,                                  /* tp_iter */
    0,                                  /* tp_iternext */
    pepy_resource_methods,              /* tp_methods */
    0,                                  /* tp_members */
    pepy_resource_getseters,            /* tp_getset */
    0,                                  /* tp_base */
    0,                                  /* tp_dict */
    0,                                  /* tp_descr_get */
    0,                                  /* tp_descr_set */
    0,                                  /* tp_dictoffset */
    (initproc) pepy_resource_init,      /* tp_init */
    0,                                  /* tp_alloc */
    pepy_resource_new                   /* tp_new */
};

static PyObject *
pepy_parsed_new(PyTypeObject *type, PyObject *args, PyObject *kwds) {
  pepy_parsed *self;

  self = (pepy_parsed *) type->tp_alloc(type, 0);

  return (PyObject *) self;
}

static int pepy_parsed_init(pepy_parsed *self, PyObject *args, PyObject *kwds) {
  char *pe_path;

  if (!PyArg_ParseTuple(args, "s:pepy_parse", &pe_path))
    return -1;

  if (!pe_path)
    return -1;

  self->pe = ParsePEFromFile(pe_path);
  if (!self->pe) {
    return -2;
  }

  return 0;
}

static void pepy_parsed_dealloc(pepy_parsed *self) {
  DestructParsedPE(self->pe);
  Py_TYPE(self)->tp_free((PyObject *) self);
}

static PyObject *pepy_parsed_get_entry_point(PyObject *self, PyObject *args) {
  VA entrypoint;
  PyObject *ret;

  if (!GetEntryPoint(((pepy_parsed *) self)->pe, entrypoint))
    Py_RETURN_NONE;

  ret = PyLong_FromUnsignedLongLong(entrypoint);
  if (!ret) {
    PyErr_SetString(pepy_error, "Unable to create return object.");
    return NULL;
  }

  return ret;
}

static PyObject *pepy_parsed_get_machine_as_str(PyObject *self,
                                                PyObject *args) {
  PyObject *ret;
  const char *str;

  str = GetMachineAsString(((pepy_parsed *) self)->pe);
  if (!str)
    Py_RETURN_NONE;

  ret = PyString_FromString(str);
  if (!ret) {
    PyErr_SetString(pepy_error, "Unable to create return string.");
    return NULL;
  }

  return ret;
}

static PyObject *pepy_parsed_get_subsystem_as_str(PyObject *self,
                                                  PyObject *args) {
  PyObject *ret;
  const char *str;

  str = GetSubsystemAsString(((pepy_parsed *) self)->pe);
  if (!str)
    Py_RETURN_NONE;

  ret = PyString_FromString(str);
  if (!ret) {
    PyErr_SetString(pepy_error, "Unable to create return string.");
    return NULL;
  }

  return ret;
}

static PyObject *pepy_parsed_get_bytes(PyObject *self, PyObject *args) {
  uint64_t start;
  Py_ssize_t len, idx;
  PyObject *ret;

  if (!PyArg_ParseTuple(args, "KK:pepy_parsed_get_bytes", &start, &len))
    return NULL;

  /*
   * XXX: a new implementation read all bytes in char* and use
   * PybyteArray_FromStringAndSize
   */

  uint8_t *buf = new (std::nothrow) uint8_t[len];
  if (!buf) {
    /* in case allocation failed */
    PyErr_SetString(pepy_error,
                    "Unable to create initial buffer (allocation failure).");
    return NULL;
  }

  for (idx = 0; idx < len; idx++) {
    if (!ReadByteAtVA(((pepy_parsed *) self)->pe, start + idx, buf[idx]))
      break;
  }

  /* use idx as content length, if we get less than asked for */
  ret = PyByteArray_FromStringAndSize(reinterpret_cast<char *>(buf), idx);
  if (!ret) {
    PyErr_SetString(pepy_error, "Unable to create new byte array.");
    return NULL;
  }

  delete[] buf;
  return ret;
}

/*
 * This is used to convert bounded buffers into python byte array objects.
 * In case the buffer is NULL, return an empty bytearray.
 */
static PyObject *pepy_data_converter(bounded_buffer *data) {
  PyObject *ret;
  const char *str;
  Py_ssize_t len;

  if (!data || !data->buf) {
    str = "";
    len = 0;
  } else {
    str = (const char *) data->buf;
    len = data->bufLen;
  }

  ret = PyByteArray_FromStringAndSize(str, len);
  if (!ret) {
    PyErr_SetString(pepy_error, "Unable to convert data to byte array.");
    return NULL;
  }

  return ret;
}

int section_callback(void *cbd,
                     const VA &base,
                     const std::string &name,
                     const image_section_header &s,
                     const bounded_buffer *data) {
  uint32_t buflen;
  PyObject *sect;
  PyObject *tuple;
  PyObject *list = (PyObject *) cbd;

  /*
   * I've seen some interesting binaries with a section where the
   * PointerToRawData and SizeOfRawData are invalid. The parser library
   * handles this by setting sectionData to NULL as returned by splitBuffer().
   * The sectionData (passed in to us as *data) is converted using
   * pepy_data_converter() which will return an empty string object.
   * However, we need to address the fact that we pass an invalid length
   * via data->bufLen.
   */
  if (!data) {
    buflen = 0;
  } else {
    buflen = data->bufLen;
  }

  /*
   * The tuple item order is important here. It is passed into the
   * section type initialization and parsed there.
   */
  tuple = Py_BuildValue("sKKIIHHIO&",
                        name.c_str(),
                        base,
                        buflen,
                        s.VirtualAddress,
                        s.Misc.VirtualSize,
                        s.NumberOfRelocations,
                        s.NumberOfLinenumbers,
                        s.Characteristics,
                        pepy_data_converter,
                        data);
  if (!tuple)
    return 1;

  sect = pepy_section_new(&pepy_section_type, NULL, NULL);
  if (!sect) {
    Py_DECREF(tuple);
    return 1;
  }

  if (pepy_section_init((pepy_section *) sect, tuple, NULL) == -1) {
    PyErr_SetString(pepy_error, "Unable to init new section.");
    return 1;
  }

  if (PyList_Append(list, sect) == -1) {
    Py_DECREF(tuple);
    Py_DECREF(sect);
    return 1;
  }

  return 0;
}

static PyObject *pepy_parsed_get_sections(PyObject *self, PyObject *args) {
  PyObject *ret = PyList_New(0);
  if (!ret) {
    PyErr_SetString(pepy_error, "Unable to create new list.");
    return NULL;
  }

  IterSec(((pepy_parsed *) self)->pe, section_callback, ret);

  return ret;
}

int resource_callback(void *cbd, const resource &r) {
  PyObject *rsrc;
  PyObject *tuple;
  PyObject *list = (PyObject *) cbd;

  /*
   * The tuple item order is important here. It is passed into the
   * section type initialization and parsed there.
   */
  tuple = Py_BuildValue("s#s#s#IIIIIIO&",
                        r.type_str.c_str(),
                        r.type_str.length(),
                        r.name_str.c_str(),
                        r.name_str.length(),
                        r.lang_str.c_str(),
                        r.lang_str.length(),
                        r.type,
                        r.name,
                        r.lang,
                        r.codepage,
                        r.RVA,
                        r.size,
                        pepy_data_converter,
                        r.buf);
  if (!tuple)
    return 1;

  rsrc = pepy_resource_new(&pepy_resource_type, NULL, NULL);
  if (!rsrc) {
    Py_DECREF(tuple);
    return 1;
  }

  if (pepy_resource_init((pepy_resource *) rsrc, tuple, NULL) == -1) {
    PyErr_SetString(pepy_error, "Unable to init new resource.");
    return 1;
  }

  if (PyList_Append(list, rsrc) == -1) {
    Py_DECREF(tuple);
    Py_DECREF(rsrc);
    return 1;
  }

  return 0;
}

static PyObject *pepy_parsed_get_resources(PyObject *self, PyObject *args) {
  PyObject *ret = PyList_New(0);
  if (!ret) {
    PyErr_SetString(pepy_error, "Unable to create new list.");
    return NULL;
  }

  IterRsrc(((pepy_parsed *) self)->pe, resource_callback, ret);

  return ret;
}

int import_callback(void *cbd,
                    const VA &addr,
                    const std::string &name,
                    const std::string &sym) {
  PyObject *imp;
  PyObject *tuple;
  PyObject *list = (PyObject *) cbd;

  /*
   * The tuple item order is important here. It is passed into the
   * import type initialization and parsed there.
   */
  tuple = Py_BuildValue("ssI", name.c_str(), sym.c_str(), addr);
  if (!tuple)
    return 1;

  imp = pepy_import_new(&pepy_import_type, NULL, NULL);
  if (!imp) {
    Py_DECREF(tuple);
    return 1;
  }

  if (pepy_import_init((pepy_import *) imp, tuple, NULL) == -1) {
    PyErr_SetString(pepy_error, "Unable to init new section.");
    return 1;
  }

  if (PyList_Append(list, imp) == -1) {
    Py_DECREF(tuple);
    Py_DECREF(imp);
    return 1;
  }

  return 0;
}

static PyObject *pepy_parsed_get_imports(PyObject *self, PyObject *args) {
  PyObject *ret = PyList_New(0);
  if (!ret) {
    PyErr_SetString(pepy_error, "Unable to create new list.");
    return NULL;
  }

  IterImpVAString(((pepy_parsed *) self)->pe, import_callback, ret);

  return ret;
}

int export_callback(void *cbd,
                    const VA &addr,
                    const std::string &mod,
                    const std::string &func) {
  PyObject *exp;
  PyObject *tuple;
  PyObject *list = (PyObject *) cbd;

  /*
   * The tuple item order is important here. It is passed into the
   * export type initialization and parsed there.
   */
  tuple = Py_BuildValue("ssI", mod.c_str(), func.c_str(), addr);
  if (!tuple)
    return 1;

  exp = pepy_export_new(&pepy_export_type, NULL, NULL);
  if (!exp) {
    Py_DECREF(tuple);
    return 1;
  }

  if (pepy_export_init((pepy_export *) exp, tuple, NULL) == -1) {
    PyErr_SetString(pepy_error, "Unable to init new section.");
    return 1;
  }

  if (PyList_Append(list, exp) == -1) {
    Py_DECREF(tuple);
    Py_DECREF(exp);
    return 1;
  }

  return 0;
}

static PyObject *pepy_parsed_get_exports(PyObject *self, PyObject *args) {
  PyObject *ret = PyList_New(0);
  if (!ret) {
    PyErr_SetString(pepy_error, "Unable to create new list.");
    return NULL;
  }

  /*
   * This could use the same callback and object as imports but the names
   * of the attributes would be slightly off.
   */
  IterExpVA(((pepy_parsed *) self)->pe, export_callback, ret);

  return ret;
}

int reloc_callback(void *cbd, const VA &addr, const reloc_type &type) {
  PyObject *reloc;
  PyObject *tuple;
  PyObject *list = (PyObject *) cbd;

  /*
   * The tuple item order is important here. It is passed into the
   * relocation type initialization and parsed there.
   */
  tuple = Py_BuildValue("II", type, addr);
  if (!tuple)
    return 1;

  reloc = pepy_relocation_new(&pepy_relocation_type, NULL, NULL);
  if (!reloc) {
    Py_DECREF(tuple);
    return 1;
  }

  if (pepy_relocation_init((pepy_relocation *) reloc, tuple, NULL) == -1) {
    PyErr_SetString(pepy_error, "Unable to init new section.");
    return 1;
  }

  if (PyList_Append(list, reloc) == -1) {
    Py_DECREF(tuple);
    Py_DECREF(reloc);
    return 1;
  }

  return 0;
}

static PyObject *pepy_parsed_get_relocations(PyObject *self, PyObject *args) {
  PyObject *ret = PyList_New(0);
  if (!ret) {
    PyErr_SetString(pepy_error, "Unable to create new list.");
    return NULL;
  }

  IterRelocs(((pepy_parsed *) self)->pe, reloc_callback, ret);

  return ret;
}

#define PEPY_PARSED_GET(ATTR, VAL)                                         \
  static PyObject *pepy_parsed_get_##ATTR(PyObject *self, void *closure) { \
    PyObject *ret = PyLong_FromUnsignedLongLong(                           \
        ((pepy_parsed *) self)->pe->peHeader.nt.VAL);                      \
    if (!ret)                                                              \
      PyErr_SetString(PyExc_AttributeError, "Error getting attribute.");   \
    return ret;                                                            \
  }

PEPY_PARSED_GET(signature, Signature)
PEPY_PARSED_GET(machine, FileHeader.Machine)
PEPY_PARSED_GET(numberofsections, FileHeader.NumberOfSections)
PEPY_PARSED_GET(timedatestamp, FileHeader.TimeDateStamp)
PEPY_PARSED_GET(numberofsymbols, FileHeader.NumberOfSymbols)
PEPY_PARSED_GET(characteristics, FileHeader.Characteristics)
PEPY_PARSED_GET(magic, OptionalMagic)

/*
 * This is used to get things from the optional header, which can be either
 * the PE32 or PE32+ version, depending upon the magic value. Technically
 * the magic is stored in the OptionalHeader, but to make life easier pe-parse
 * stores the value in nt_header_32 along with the appropriate optional header.
 * This is why "magic" is handled above, and not here.
 */
#define PEPY_PARSED_GET_OPTIONAL(ATTR, VAL)                                \
  static PyObject *pepy_parsed_get_optional_##ATTR(PyObject *self,         \
                                                   void *closure) {        \
    PyObject *ret = NULL;                                                  \
    if (((pepy_parsed *) self)->pe->peHeader.nt.OptionalMagic ==           \
        NT_OPTIONAL_32_MAGIC) {                                            \
      ret = PyLong_FromUnsignedLongLong(                                   \
          ((pepy_parsed *) self)->pe->peHeader.nt.OptionalHeader.VAL);     \
      if (!ret)                                                            \
        PyErr_SetString(PyExc_AttributeError, "Error getting attribute."); \
    } else if (((pepy_parsed *) self)->pe->peHeader.nt.OptionalMagic ==    \
               NT_OPTIONAL_64_MAGIC) {                                     \
      ret = PyLong_FromUnsignedLongLong(                                   \
          ((pepy_parsed *) self)->pe->peHeader.nt.OptionalHeader64.VAL);   \
      if (!ret)                                                            \
        PyErr_SetString(PyExc_AttributeError, "Error getting attribute."); \
    } else {                                                               \
      PyErr_SetString(pepy_error, "Bad magic value.");                     \
    }                                                                      \
    return ret;                                                            \
  }

PEPY_PARSED_GET_OPTIONAL(majorlinkerver, MajorLinkerVersion)
PEPY_PARSED_GET_OPTIONAL(minorlinkerver, MinorLinkerVersion)
PEPY_PARSED_GET_OPTIONAL(codesize, SizeOfCode);
PEPY_PARSED_GET_OPTIONAL(initdatasize, SizeOfInitializedData);
PEPY_PARSED_GET_OPTIONAL(uninitdatasize, SizeOfUninitializedData);
PEPY_PARSED_GET_OPTIONAL(entrypointaddr, AddressOfEntryPoint);
PEPY_PARSED_GET_OPTIONAL(baseofcode, BaseOfCode);
PEPY_PARSED_GET_OPTIONAL(imagebase, ImageBase);
PEPY_PARSED_GET_OPTIONAL(sectionalignement, SectionAlignment);
PEPY_PARSED_GET_OPTIONAL(filealignment, FileAlignment);
PEPY_PARSED_GET_OPTIONAL(majorosver, MajorOperatingSystemVersion);
PEPY_PARSED_GET_OPTIONAL(minorosver, MinorOperatingSystemVersion);
PEPY_PARSED_GET_OPTIONAL(win32ver, Win32VersionValue);
PEPY_PARSED_GET_OPTIONAL(imagesize, SizeOfImage);
PEPY_PARSED_GET_OPTIONAL(headersize, SizeOfHeaders);
PEPY_PARSED_GET_OPTIONAL(checksum, CheckSum);
PEPY_PARSED_GET_OPTIONAL(subsystem, Subsystem);
PEPY_PARSED_GET_OPTIONAL(dllcharacteristics, DllCharacteristics);
PEPY_PARSED_GET_OPTIONAL(stackreservesize, SizeOfStackReserve);
PEPY_PARSED_GET_OPTIONAL(stackcommitsize, SizeOfStackCommit);
PEPY_PARSED_GET_OPTIONAL(heapreservesize, SizeOfHeapReserve);
PEPY_PARSED_GET_OPTIONAL(heapcommitsize, SizeOfHeapCommit);
PEPY_PARSED_GET_OPTIONAL(loaderflags, LoaderFlags);
PEPY_PARSED_GET_OPTIONAL(rvasandsize, NumberOfRvaAndSizes);

/*
 * BaseOfData is only in PE32, not PE32+. Thus, it uses a non-standard
 * getter function compared to the other shared fields.
 */
static PyObject *pepy_parsed_get_optional_baseofdata(PyObject *self,
                                                     void *closure) {
  PyObject *ret = NULL;
  if (((pepy_parsed *) self)->pe->peHeader.nt.OptionalMagic ==
      NT_OPTIONAL_32_MAGIC) {
    ret = PyLong_FromUnsignedLong(
        ((pepy_parsed *) self)->pe->peHeader.nt.OptionalHeader.BaseOfData);
    if (!ret)
      PyErr_SetString(PyExc_AttributeError, "Error getting attribute.");
  } else if (((pepy_parsed *) self)->pe->peHeader.nt.OptionalMagic ==
             NT_OPTIONAL_64_MAGIC) {
    PyErr_SetString(PyExc_AttributeError, "Not available on PE32+.");
  } else {
    PyErr_SetString(pepy_error, "Bad magic value.");
  }
  return ret;
}

static PyGetSetDef pepy_parsed_getseters[] = {
    OBJECTGETTER(parsed, signature, "PE Signature"),
    OBJECTGETTER(parsed, machine, "Machine"),
    OBJECTGETTER(parsed, numberofsections, "Number of sections"),
    OBJECTGETTER(parsed, timedatestamp, "Timedate stamp"),
    OBJECTGETTER(parsed, numberofsymbols, "Number of symbols"),
    OBJECTGETTER(parsed, characteristics, "Characteristics"),
    OBJECTGETTER(parsed, magic, "Magic"),
    OBJECTGETTER_OPTIONAL(majorlinkerver, "Major linker version"),
    OBJECTGETTER_OPTIONAL(minorlinkerver, "Minor linker version"),
    OBJECTGETTER_OPTIONAL(codesize, "Size of code"),
    OBJECTGETTER_OPTIONAL(initdatasize, "Size of initialized data"),
    OBJECTGETTER_OPTIONAL(uninitdatasize, "Size of uninitialized data"),
    OBJECTGETTER_OPTIONAL(entrypointaddr, "Address of entry point"),
    OBJECTGETTER_OPTIONAL(baseofcode, "Base address of code"),
    OBJECTGETTER_OPTIONAL(imagebase, "Image base address"),
    OBJECTGETTER_OPTIONAL(sectionalignement, "Section alignment"),
    OBJECTGETTER_OPTIONAL(filealignment, "File alignment"),
    OBJECTGETTER_OPTIONAL(majorosver, "Major OS version"),
    OBJECTGETTER_OPTIONAL(minorosver, "Minor OS version"),
    OBJECTGETTER_OPTIONAL(win32ver, "Win32 version"),
    OBJECTGETTER_OPTIONAL(imagesize, "Size of image"),
    OBJECTGETTER_OPTIONAL(headersize, "Size of headers"),
    OBJECTGETTER_OPTIONAL(checksum, "Checksum"),
    OBJECTGETTER_OPTIONAL(subsystem, "Subsystem"),
    OBJECTGETTER_OPTIONAL(dllcharacteristics, "DLL characteristics"),
    OBJECTGETTER_OPTIONAL(stackreservesize, "Size of stack reserve"),
    OBJECTGETTER_OPTIONAL(stackcommitsize, "Size of stack commit"),
    OBJECTGETTER_OPTIONAL(heapreservesize, "Size of heap reserve"),
    OBJECTGETTER_OPTIONAL(heapcommitsize, "Size of heap commit"),
    OBJECTGETTER_OPTIONAL(loaderflags, "Loader flags"),
    OBJECTGETTER_OPTIONAL(rvasandsize, "Number of RVA and sizes"),
    /* Base of data is only available in PE32, not PE32+. */
    {(char *) "baseofdata",
     (getter) pepy_parsed_get_optional_baseofdata,
     (setter) pepy_attr_not_writable,
     (char *) "Base address of data",
     NULL},
    {NULL}};

static PyMethodDef pepy_parsed_methods[] = {
    {"get_entry_point",
     pepy_parsed_get_entry_point,
     METH_NOARGS,
     "Return the entry point address."},
    {"get_machine_as_str",
     pepy_parsed_get_machine_as_str,
     METH_NOARGS,
     "Return the machine as a human readable string."},
    {"get_subsystem_as_str",
     pepy_parsed_get_subsystem_as_str,
     METH_NOARGS,
     "Return the subsystem as a human readable string."},
    {"get_bytes",
     pepy_parsed_get_bytes,
     METH_VARARGS,
     "Return the first N bytes at a given address."},
    {"get_sections",
     pepy_parsed_get_sections,
     METH_NOARGS,
     "Return a list of section objects."},
    {"get_imports",
     pepy_parsed_get_imports,
     METH_NOARGS,
     "Return a list of import objects."},
    {"get_exports",
     pepy_parsed_get_exports,
     METH_NOARGS,
     "Return a list of export objects."},
    {"get_relocations",
     pepy_parsed_get_relocations,
     METH_NOARGS,
     "Return a list of relocation objects."},
    {"get_resources",
     pepy_parsed_get_resources,
     METH_NOARGS,
     "Return a list of resource objects."},
    {NULL}};

static PyTypeObject pepy_parsed_type = {
    PyVarObject_HEAD_INIT(NULL, 0)            /* ob_size */
    "pepy.parsed",                            /* tp_name */
    sizeof(pepy_parsed),                      /* tp_basicsize */
    0,                                        /* tp_itemsize */
    (destructor) pepy_parsed_dealloc,         /* tp_dealloc */
    0,                                        /* tp_print */
    0,                                        /* tp_getattr */
    0,                                        /* tp_setattr */
    0,                                        /* tp_compare */
    0,                                        /* tp_repr */
    0,                                        /* tp_as_number */
    0,                                        /* tp_as_sequence */
    0,                                        /* tp_as_mapping */
    0,                                        /* tp_hash */
    0,                                        /* tp_call */
    0,                                        /* tp_str */
    0,                                        /* tp_getattro */
    0,                                        /* tp_setattro */
    0,                                        /* tp_as_buffer */
    Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
    "pepy parsed object",                     /* tp_doc */
    0,                                        /* tp_traverse */
    0,                                        /* tp_clear */
    0,                                        /* tp_richcompare */
    0,                                        /* tp_weaklistoffset */
    0,                                        /* tp_iter */
    0,                                        /* tp_iternext */
    pepy_parsed_methods,                      /* tp_methods */
    0,                                        /* tp_members */
    pepy_parsed_getseters,                    /* tp_getset */
    0,                                        /* tp_base */
    0,                                        /* tp_dict */
    0,                                        /* tp_descr_get */
    0,                                        /* tp_descr_set */
    0,                                        /* tp_dictoffset */
    (initproc) pepy_parsed_init,              /* tp_init */
    0,                                        /* tp_alloc */
    pepy_parsed_new                           /* tp_new */
};

static PyObject *pepy_parse(PyObject *self, PyObject *args) {
  PyObject *parsed;
  int ret;
  char *err_str = NULL;

  parsed = pepy_parsed_new(&pepy_parsed_type, NULL, NULL);
  if (!parsed) {
    PyErr_SetString(pepy_error, "Unable to make new parsed object.");
    return NULL;
  }

  ret = pepy_parsed_init((pepy_parsed *) parsed, args, NULL);
  if (ret < 0) {
    if (ret == -2) {
      // error (loc)
      size_t len = GetPEErrString().length() + GetPEErrLoc().length() + 4;
      err_str = (char *) malloc(len);
      if (!err_str)
        return PyErr_NoMemory();
      snprintf(err_str,
               len,
               "%s (%s)",
               GetPEErrString().c_str(),
               GetPEErrLoc().c_str());
      PyErr_SetString(pepy_error, err_str);
    } else
      PyErr_SetString(pepy_error, "Unable to init new parsed object.");
    return NULL;
  }

  return parsed;
}

static PyMethodDef pepy_methods[] = {
    {"parse", pepy_parse, METH_VARARGS, "Parse PE from file."}, {NULL}};

static PyObject *pepi_module_init(void) {
  PyObject *m;

  if (PyType_Ready(&pepy_parsed_type) < 0 ||
      PyType_Ready(&pepy_section_type) < 0 ||
      PyType_Ready(&pepy_import_type) < 0 ||
      PyType_Ready(&pepy_export_type) < 0 ||
      PyType_Ready(&pepy_relocation_type) < 0 ||
      PyType_Ready(&pepy_resource_type) < 0)
    return NULL;

#if PY_MAJOR_VERSION >= 3
  static struct PyModuleDef moduledef = {
      PyModuleDef_HEAD_INIT,
      "pepy",
      "Python interface to pe-parse.",
      -1,
      pepy_methods,
      NULL,
      NULL,
      NULL,
      NULL,
  };
#endif

#if PY_MAJOR_VERSION >= 3
  m = PyModule_Create(&moduledef);
#else
  m = Py_InitModule3("pepy", pepy_methods, "Python interface to pe-parse.");
#endif
  if (!m)
    return NULL;

  pepy_error = PyErr_NewException((char *) "pepy.error", NULL, NULL);
  Py_INCREF(pepy_error);
  PyModule_AddObject(m, "error", pepy_error);

  Py_INCREF(&pepy_parsed_type);
  PyModule_AddObject(m, "pepy_parsed", (PyObject *) &pepy_parsed_type);

  Py_INCREF(&pepy_section_type);
  PyModule_AddObject(m, "pepy_section", (PyObject *) &pepy_section_type);

  Py_INCREF(&pepy_import_type);
  PyModule_AddObject(m, "pepy_import", (PyObject *) &pepy_import_type);

  Py_INCREF(&pepy_export_type);
  PyModule_AddObject(m, "pepy_export", (PyObject *) &pepy_export_type);

  Py_INCREF(&pepy_relocation_type);
  PyModule_AddObject(m, "pepy_relocation", (PyObject *) &pepy_relocation_type);

  Py_INCREF(&pepy_resource_type);
  PyModule_AddObject(m, "pepy_resource", (PyObject *) &pepy_resource_type);

  PyModule_AddStringMacro(m, PEPY_VERSION);
  PyModule_AddStringMacro(m, PEPARSE_VERSION);
  PyModule_AddStringConstant(m, "__version__", PEPY_VERSION);

  PyModule_AddIntMacro(m, MZ_MAGIC);
  PyModule_AddIntMacro(m, NT_MAGIC);
  PyModule_AddIntMacro(m, NUM_DIR_ENTRIES);
  PyModule_AddIntMacro(m, NT_OPTIONAL_32_MAGIC);
  PyModule_AddIntMacro(m, NT_SHORT_NAME_LEN);
  PyModule_AddIntMacro(m, DIR_EXPORT);
  PyModule_AddIntMacro(m, DIR_IMPORT);
  PyModule_AddIntMacro(m, DIR_RESOURCE);
  PyModule_AddIntMacro(m, DIR_EXCEPTION);
  PyModule_AddIntMacro(m, DIR_SECURITY);
  PyModule_AddIntMacro(m, DIR_BASERELOC);
  PyModule_AddIntMacro(m, DIR_DEBUG);
  PyModule_AddIntMacro(m, DIR_ARCHITECTURE);
  PyModule_AddIntMacro(m, DIR_GLOBALPTR);
  PyModule_AddIntMacro(m, DIR_TLS);
  PyModule_AddIntMacro(m, DIR_LOAD_CONFIG);
  PyModule_AddIntMacro(m, DIR_BOUND_IMPORT);
  PyModule_AddIntMacro(m, DIR_IAT);
  PyModule_AddIntMacro(m, DIR_DELAY_IMPORT);
  PyModule_AddIntMacro(m, DIR_COM_DESCRIPTOR);

  PyModule_AddIntMacro(m, IMAGE_SCN_TYPE_NO_PAD);
  PyModule_AddIntMacro(m, IMAGE_SCN_CNT_CODE);
  PyModule_AddIntMacro(m, IMAGE_SCN_CNT_INITIALIZED_DATA);
  PyModule_AddIntMacro(m, IMAGE_SCN_CNT_UNINITIALIZED_DATA);
  PyModule_AddIntMacro(m, IMAGE_SCN_LNK_OTHER);
  PyModule_AddIntMacro(m, IMAGE_SCN_LNK_INFO);
  PyModule_AddIntMacro(m, IMAGE_SCN_LNK_REMOVE);
  PyModule_AddIntMacro(m, IMAGE_SCN_LNK_COMDAT);
  PyModule_AddIntMacro(m, IMAGE_SCN_NO_DEFER_SPEC_EXC);
  PyModule_AddIntMacro(m, IMAGE_SCN_GPREL);
  PyModule_AddIntMacro(m, IMAGE_SCN_MEM_FARDATA);
  PyModule_AddIntMacro(m, IMAGE_SCN_MEM_PURGEABLE);
  PyModule_AddIntMacro(m, IMAGE_SCN_MEM_16BIT);
  PyModule_AddIntMacro(m, IMAGE_SCN_MEM_LOCKED);
  PyModule_AddIntMacro(m, IMAGE_SCN_MEM_PRELOAD);
  PyModule_AddIntMacro(m, IMAGE_SCN_ALIGN_1BYTES);
  PyModule_AddIntMacro(m, IMAGE_SCN_ALIGN_2BYTES);
  PyModule_AddIntMacro(m, IMAGE_SCN_ALIGN_4BYTES);
  PyModule_AddIntMacro(m, IMAGE_SCN_ALIGN_8BYTES);
  PyModule_AddIntMacro(m, IMAGE_SCN_ALIGN_16BYTES);
  PyModule_AddIntMacro(m, IMAGE_SCN_ALIGN_32BYTES);
  PyModule_AddIntMacro(m, IMAGE_SCN_ALIGN_64BYTES);
  PyModule_AddIntMacro(m, IMAGE_SCN_ALIGN_128BYTES);
  PyModule_AddIntMacro(m, IMAGE_SCN_ALIGN_256BYTES);
  PyModule_AddIntMacro(m, IMAGE_SCN_ALIGN_512BYTES);
  PyModule_AddIntMacro(m, IMAGE_SCN_ALIGN_1024BYTES);
  PyModule_AddIntMacro(m, IMAGE_SCN_ALIGN_2048BYTES);
  PyModule_AddIntMacro(m, IMAGE_SCN_ALIGN_4096BYTES);
  PyModule_AddIntMacro(m, IMAGE_SCN_ALIGN_8192BYTES);
  PyModule_AddIntMacro(m, IMAGE_SCN_ALIGN_MASK);
  PyModule_AddIntMacro(m, IMAGE_SCN_LNK_NRELOC_OVFL);
  PyModule_AddIntMacro(m, IMAGE_SCN_MEM_DISCARDABLE);
  PyModule_AddIntMacro(m, IMAGE_SCN_MEM_NOT_CACHED);
  PyModule_AddIntMacro(m, IMAGE_SCN_MEM_NOT_PAGED);
  PyModule_AddIntMacro(m, IMAGE_SCN_MEM_SHARED);
  PyModule_AddIntMacro(m, IMAGE_SCN_MEM_EXECUTE);
  PyModule_AddIntMacro(m, IMAGE_SCN_MEM_READ);
  PyModule_AddIntMacro(m, IMAGE_SCN_MEM_WRITE);

  return m;
}

#if PY_MAJOR_VERSION >= 3
PyMODINIT_FUNC PyInit_pepy(void) {
  return pepi_module_init();
}
#else
PyMODINIT_FUNC initpepy(void) {
  pepi_module_init();
}
#endif