Enhance the Python IO interface

Related to #832
This commit is contained in:
Romain Thomas
2022-12-09 10:22:53 +01:00
parent 15d8b2ffaf
commit 2a10494b8a
18 changed files with 295 additions and 166 deletions
+9 -30
View File
@@ -16,9 +16,12 @@
#include <string>
#include <memory>
#include "PyIOStream.hpp"
#include "LIEF/MachO/Parser.hpp"
#include "LIEF/MachO/FatBinary.hpp"
#include "logging.hpp"
#include "pyMachO.hpp"
namespace LIEF {
@@ -62,37 +65,13 @@ void create<Parser>(py::module& m) {
py::return_value_policy::take_ownership);
m.def("parse",
[] (py::object byteio, std::string name, const ParserConfig& config) {
const auto& io = py::module::import("io");
const auto& RawIOBase = io.attr("RawIOBase");
const auto& BufferedIOBase = io.attr("BufferedIOBase");
const auto& TextIOBase = io.attr("TextIOBase");
py::object rawio;
if (py::isinstance(byteio, RawIOBase)) {
rawio = byteio;
[] (py::object byteio, std::string name, const ParserConfig& config) -> py::object {
if (auto stream = PyIOStream::from_python(byteio)) {
auto ptr = std::make_unique<PyIOStream>(std::move(*stream));
return py::cast(MachO::Parser::parse(std::move(ptr), config));
}
else if (py::isinstance(byteio, BufferedIOBase)) {
rawio = byteio.attr("raw");
}
else if (py::isinstance(byteio, TextIOBase)) {
rawio = byteio.attr("buffer").attr("raw");
}
else {
throw py::type_error(py::repr(byteio).cast<std::string>().c_str());
}
std::string raw_str = static_cast<py::bytes>(rawio.attr("readall")());
std::vector<uint8_t> raw = {
std::make_move_iterator(std::begin(raw_str)),
std::make_move_iterator(std::end(raw_str))};
return LIEF::MachO::Parser::parse(std::move(raw), name, config);
LIEF_ERR("Can't create a LIEF stream interface over the provided io");
return py::none();
},
R"delim(
Parse the given binary from a Python IO interface and return a :class:`~lief.MachO.FatBinary` object