mirror of
https://github.com/bitdefender/bddisasm.git
synced 2024-11-14 03:19:24 +00:00
Merge pull request #20 from akisari/master
Added support for byte-like objects as input value in pybddisasm.
This commit is contained in:
commit
65d6f52740
@ -967,18 +967,60 @@ static uint8_t _pybddisasm_py_stack_to_disasm(uint8_t stack)
|
||||
}
|
||||
|
||||
|
||||
static char *_pybddisasm_get_contiguous_buffer(Py_buffer *view)
|
||||
{
|
||||
char *buffer = NULL;
|
||||
|
||||
if (PyBuffer_IsContiguous(view, 'C'))
|
||||
{
|
||||
return view->buf;
|
||||
}
|
||||
|
||||
buffer = (char *)malloc(sizeof(char) * view->len);
|
||||
if (!buffer)
|
||||
{
|
||||
PyErr_SetString(PyExc_ValueError, "failed to allocate memory for contiguous buffer!");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
if (PyBuffer_ToContiguous(buffer, view, view->len, 'C') < 0)
|
||||
{
|
||||
PyErr_SetString(PyExc_ValueError, "PyBuffer_ToContiguous failed!");
|
||||
free(buffer);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
return buffer;
|
||||
}
|
||||
|
||||
|
||||
static void _pybddisasm_release_contiguous_buffer(Py_buffer *view, char *buffer)
|
||||
{
|
||||
if (!PyBuffer_IsContiguous(view, 'C'))
|
||||
{
|
||||
free(buffer);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
static PyObject *pybddisasm_decode_ex(PyObject *self, PyObject *args)
|
||||
{
|
||||
char *buf = NULL;
|
||||
Py_ssize_t bufsize = 0;
|
||||
char *buffer = NULL;
|
||||
uint64_t rip = 0;
|
||||
uint8_t code, data;
|
||||
Py_buffer view;
|
||||
|
||||
(void)self;
|
||||
|
||||
if (!PyArg_ParseTuple(args, "s#BB|K", &buf, &bufsize, &code, &data, &rip))
|
||||
if (!PyArg_ParseTuple(args, "y*BB|K", &view, &code, &data, &rip))
|
||||
{
|
||||
PyErr_SetString(PyExc_ValueError, "invalid arguments. expected: <buffer[byte-like object]> <code> <data> <rip>");
|
||||
Py_RETURN_NONE;
|
||||
}
|
||||
|
||||
buffer = _pybddisasm_get_contiguous_buffer(&view);
|
||||
if (!buffer)
|
||||
{
|
||||
PyErr_SetString(PyExc_ValueError, "invalid arguments. expected: <buffer[bytes]> <bufsize> <code> <data> <rip>");
|
||||
Py_RETURN_NONE;
|
||||
}
|
||||
|
||||
@ -992,29 +1034,38 @@ static PyObject *pybddisasm_decode_ex(PyObject *self, PyObject *args)
|
||||
|
||||
INSTRUX instr;
|
||||
|
||||
NDSTATUS status = nd_decode_ex(&instr, buf, bufsize, code, data);
|
||||
NDSTATUS status = nd_decode_ex(&instr, buffer, view.len, code, data);
|
||||
if (!ND_SUCCESS(status))
|
||||
{
|
||||
Py_RETURN_NONE;
|
||||
}
|
||||
|
||||
_pybddisasm_release_contiguous_buffer(&view, buffer);
|
||||
PyBuffer_Release(&view);
|
||||
|
||||
return _pybddisasm_build_instr_dict(&instr, rip);
|
||||
}
|
||||
|
||||
|
||||
static PyObject *pybddisasm_decode_ex2(PyObject *self, PyObject *args)
|
||||
{
|
||||
char *buf = NULL;
|
||||
char *str_vend = NULL;
|
||||
Py_ssize_t bufsize = 0;
|
||||
char *buffer = NULL;
|
||||
uint64_t rip = 0;
|
||||
uint8_t code, data, stack, vend;
|
||||
Py_buffer view;
|
||||
|
||||
(void)self;
|
||||
|
||||
if (!PyArg_ParseTuple(args, "s#BBBs|K", &buf, &bufsize, &code, &data, &stack, &str_vend, &rip))
|
||||
if (!PyArg_ParseTuple(args, "y*BBBs|K", &view, &code, &data, &stack, &str_vend, &rip))
|
||||
{
|
||||
PyErr_SetString(PyExc_ValueError, "invalid arguments. expected: <buffer[byte-like object]> <code> <data> <stack> <vendor> <rip>");
|
||||
Py_RETURN_NONE;
|
||||
}
|
||||
|
||||
buffer = _pybddisasm_get_contiguous_buffer(&view);
|
||||
if (!buffer)
|
||||
{
|
||||
PyErr_SetString(PyExc_ValueError, "invalid arguments. expected: <buffer[bytes]> <bufsize> <code> <data> <stack> <vendor> <rip>");
|
||||
Py_RETURN_NONE;
|
||||
}
|
||||
|
||||
@ -1055,28 +1106,37 @@ static PyObject *pybddisasm_decode_ex2(PyObject *self, PyObject *args)
|
||||
|
||||
INSTRUX instr;
|
||||
|
||||
NDSTATUS status = nd_decode_ex2(&instr, buf, bufsize, code, data, stack, vend);
|
||||
NDSTATUS status = nd_decode_ex2(&instr, buffer, view.len, code, data, stack, vend);
|
||||
if (!ND_SUCCESS(status))
|
||||
{
|
||||
Py_RETURN_NONE;
|
||||
}
|
||||
|
||||
_pybddisasm_release_contiguous_buffer(&view, buffer);
|
||||
PyBuffer_Release(&view);
|
||||
|
||||
return _pybddisasm_build_instr_dict(&instr, rip);
|
||||
}
|
||||
|
||||
|
||||
static PyObject *pybddisasm_to_text(PyObject *self, PyObject *args)
|
||||
{
|
||||
char *buf = NULL;
|
||||
Py_ssize_t bufsize = 0;
|
||||
char *buffer = NULL;
|
||||
uint64_t rip = 0;
|
||||
uint8_t code, data;
|
||||
Py_buffer view;
|
||||
|
||||
(void)self;
|
||||
|
||||
if (!PyArg_ParseTuple(args, "s#BB|K", &buf, &bufsize, &code, &data, &rip))
|
||||
if (!PyArg_ParseTuple(args, "y*BB|K", &view, &code, &data, &rip))
|
||||
{
|
||||
PyErr_SetString(PyExc_ValueError, "invalid arguments. expected: <buffer[byte-like object]> <code> <data> <rip>");
|
||||
Py_RETURN_NONE;
|
||||
}
|
||||
|
||||
buffer = _pybddisasm_get_contiguous_buffer(&view);
|
||||
if (!buffer)
|
||||
{
|
||||
PyErr_SetString(PyExc_ValueError, "invalid arguments. expected: <buffer[bytes]> <bufsize> <code> <data> <rip>");
|
||||
Py_RETURN_NONE;
|
||||
}
|
||||
|
||||
@ -1090,7 +1150,7 @@ static PyObject *pybddisasm_to_text(PyObject *self, PyObject *args)
|
||||
|
||||
INSTRUX instr;
|
||||
|
||||
NDSTATUS status = nd_decode_ex(&instr, buf, bufsize, code, data);
|
||||
NDSTATUS status = nd_decode_ex(&instr, buffer, view.len, code, data);
|
||||
if (!ND_SUCCESS(status))
|
||||
{
|
||||
Py_RETURN_NONE;
|
||||
@ -1104,6 +1164,9 @@ static PyObject *pybddisasm_to_text(PyObject *self, PyObject *args)
|
||||
Py_RETURN_NONE;
|
||||
}
|
||||
|
||||
_pybddisasm_release_contiguous_buffer(&view, buffer);
|
||||
PyBuffer_Release(&view);
|
||||
|
||||
return Py_BuildValue("{s,s,s,y#}", "text", instr_text, "bytes", instr.InstructionBytes, instr.Length);
|
||||
}
|
||||
|
||||
|
@ -11,7 +11,7 @@ import re
|
||||
from setuptools import find_packages, setup, Command, Extension, Distribution
|
||||
from codecs import open
|
||||
|
||||
VERSION = (0, 1, 2)
|
||||
VERSION = (0, 1, 3)
|
||||
LIBRARY_VERSION = (1, 28, 1)
|
||||
LIBRARY_INSTRUX_SIZE = 856
|
||||
|
||||
@ -75,7 +75,7 @@ setup(
|
||||
package_dir={'pybddisasm': 'pybddisasm'},
|
||||
platforms = ["Windows", "Linux"],
|
||||
include_package_data=True,
|
||||
python_requires=">=3.4",
|
||||
python_requires=">=3.5",
|
||||
setup_requires=['wheel'],
|
||||
install_requires=requires,
|
||||
zip_safe=False,
|
||||
@ -85,14 +85,14 @@ setup(
|
||||
'Programming Language :: Python :: 3.7',
|
||||
'Programming Language :: Python :: 3.8',
|
||||
'License :: OSI Approved :: Apache Software License',
|
||||
'Operating System :: OS Independent',
|
||||
'Operating System :: Microsoft :: Windows',
|
||||
'Operating System :: POSIX :: Linux'
|
||||
],
|
||||
ext_modules = [Extension("_pybddisasm",
|
||||
sources = ["_pybddisasm/_pybddisasm.c", "_pybddisasm/pybddisasm.c"],
|
||||
define_macros = [('AMD64', None), ('Py_LIMITED_API', None), ('LIBRARY_INSTRUX_SIZE', LIBRARY_INSTRUX_SIZE)],
|
||||
define_macros = [('AMD64', None), ('LIBRARY_INSTRUX_SIZE', LIBRARY_INSTRUX_SIZE)],
|
||||
include_dirs = ['../inc'],
|
||||
libraries = ['bddisasm'],
|
||||
library_dirs = ['/usr/local/lib', '../bin/x64/Release'],
|
||||
py_limited_api=True)],
|
||||
library_dirs = ['/usr/local/lib', '../bin/x64/Release'])],
|
||||
distclass=BinaryDistribution
|
||||
)
|
||||
|
Loading…
Reference in New Issue
Block a user