mirror of
https://github.com/lief-project/LIEF
synced 2026-06-08 15:30:44 +00:00
Romain Thomas
101 lines
3.6 KiB
C++
101 lines
3.6 KiB
C++
/* 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 "pyELF.hpp"
|
|
|
|
#include "LIEF/ELF/Parser.hpp"
|
|
#include "LIEF/ELF/Binary.hpp"
|
|
|
|
#include <string>
|
|
|
|
namespace LIEF {
|
|
namespace ELF {
|
|
|
|
template<>
|
|
void create<Parser>(py::module& m) {
|
|
|
|
// Parser (Parser)
|
|
m.def("parse",
|
|
static_cast<std::unique_ptr<Binary> (*) (const std::string&, DYNSYM_COUNT_METHODS)>(&Parser::parse),
|
|
R"delim(
|
|
Parse the ELF binary from the given **file path** and return a :class:`lief.ELF.Binary` object
|
|
|
|
For *weird* binaries (e.g sectionless) you can choose the method to use to count dynamic symbols
|
|
(:class:`lief.ELF.DYNSYM_COUNT_METHODS`). By default, the value is set to
|
|
:attr:`lief.ELF.DYNSYM_COUNT_METHODS.COUNT_AUTO`
|
|
)delim",
|
|
"filename"_a, "dynsym_count_method"_a = DYNSYM_COUNT_METHODS::COUNT_AUTO,
|
|
py::return_value_policy::take_ownership);
|
|
|
|
m.def("parse",
|
|
static_cast<std::unique_ptr<Binary>(*)(const std::vector<uint8_t>&, const std::string&, DYNSYM_COUNT_METHODS)>(&Parser::parse),
|
|
R"delim(
|
|
Parse the ELF binary from the given **list of bytes** and return a :class:`lief.ELF.Binary` object
|
|
|
|
For *weird* binaries (e.g sectionless) you can choose the method to use to count dynamic symbols
|
|
(:class:`lief.ELF.DYNSYM_COUNT_METHODS`). By default, the value is set to
|
|
:attr:`lief.ELF.DYNSYM_COUNT_METHODS.COUNT_AUTO`
|
|
)delim",
|
|
|
|
"raw"_a, "name"_a = "", "dynsym_count_method"_a = DYNSYM_COUNT_METHODS::COUNT_AUTO,
|
|
py::return_value_policy::take_ownership);
|
|
|
|
|
|
m.def("parse",
|
|
[] (py::object byteio, const std::string& name, DYNSYM_COUNT_METHODS count) {
|
|
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<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::ELF::Parser::parse(std::move(raw), name, count);
|
|
},
|
|
R"delim(
|
|
Parse the ELF binary from a Python IO stream and return a :class:`lief.ELF.Binary` object
|
|
|
|
For *weird* binaries (e.g sectionless) you can choose the method to use to count dynamic symbols
|
|
(:class:`lief.ELF.lief.ELF.DYNSYM_COUNT_METHODS`). By default, the value is set to
|
|
:attr:`lief.ELF.lief.ELF.DYNSYM_COUNT_METHODS.COUNT_AUTO`
|
|
)delim",
|
|
"io"_a, "name"_a = "", "dynsym_count_method"_a = DYNSYM_COUNT_METHODS::COUNT_AUTO,
|
|
py::return_value_policy::take_ownership);
|
|
}
|
|
}
|
|
}
|