LIEF/api/python/MachO/objects/pyFatBinary.cpp
Romain Thomas 3602643f5d Fix memory leaks in the MachO Python API and create FatBinary
API Changes:

LIEF::MachO::Parser won't return a 'std::vector' of MachO::Binary*
but a pointer to MachO::FatBinary object
It's a kind of wrapper on std::vector<MachO::Binary*>
2017-09-29 13:06:08 +02:00

61 lines
1.6 KiB
C++

/* Copyright 2017 R. Thomas
* Copyright 2017 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 <algorithm>
#include "LIEF/MachO/FatBinary.hpp"
#include "pyMachO.hpp"
void init_MachO_FatBinary_class(py::module& m) {
py::class_<FatBinary>(m, "FatBinary")
.def_property_readonly("size",
&FatBinary::size,
"Number of " RST_CLASS_REF(lief.MachO.Binary) " registred")
.def("at",
static_cast<Binary& (FatBinary::*)(size_t)>(&FatBinary::at),
"Return the " RST_CLASS_REF(lief.MachO.Binary) " at the given index",
"index"_a)
.def("__len__",
&FatBinary::size)
.def("__getitem__",
static_cast<Binary& (FatBinary::*)(size_t)>(&FatBinary::operator[]),
"",
py::return_value_policy::reference)
.def("__iter__",
static_cast<it_binaries (FatBinary::*)(void)>(&FatBinary::begin),
py::return_value_policy::reference_internal)
.def("__str__",
[] (const FatBinary& fat_binary)
{
std::ostringstream stream;
stream << fat_binary;
std::string str = stream.str();
return str;
});
}