/* Copyright 2017 - 2022 R. Thomas * Copyright 2017 - 2022 Quarkslab * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ #include #include #include "LIEF/MachO/Parser.hpp" #include "LIEF/MachO/FatBinary.hpp" #include "pyMachO.hpp" namespace LIEF { namespace MachO { template<> void create(py::module& m) { // Parser (Parser) m.def("parse", static_cast (*) (const std::string&, const ParserConfig&)>(&LIEF::MachO::Parser::parse), R"delim( Parse the given binary and return a :class:`~lief.MachO.FatBinary` object One can configure the parsing with the ``config`` parameter. See :class:`~lief.MachO.ParserConfig`, )delim", "filename"_a, "config"_a = ParserConfig::deep(), py::return_value_policy::take_ownership); m.def("parse", static_cast (*) (const std::vector&, const std::string&, const ParserConfig&)>(&LIEF::MachO::Parser::parse), R"delim( Parse the given binary (from raw bytes) and return a :class:`~lief.MachO.FatBinary` object One can configure the parsing with the ``config`` parameter. See :class:`~lief.MachO.ParserConfig` )delim", "raw"_a, "name"_a = "", "config"_a = ParserConfig::quick(), 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; } 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().c_str()); } std::string raw_str = static_cast(rawio.attr("readall")()); std::vector 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); }, R"delim( Parse the given binary from a Python IO interface and return a :class:`~lief.MachO.FatBinary` object One can configure the parsing with the ``config`` parameter. See :class:`~lief.MachO.ParserConfig` )delim", "io"_a, "name"_a = "", "config"_a = ParserConfig::quick(), py::return_value_policy::take_ownership); } } }