LIEF/api/python/OAT/objects/pyHeader.cpp
2018-05-04 12:46:28 +02:00

150 lines
4.7 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 "LIEF/OAT/Header.hpp"
#include "LIEF/OAT/hash.hpp"
#include "pyOAT.hpp"
namespace LIEF {
namespace OAT {
template<class T>
using getter_t = T (Header::*)(void) const;
template<class T>
using setter_t = void (Header::*)(T);
template<class T>
using no_const_getter = T (Header::*)(void);
template<>
void create<Header>(py::module& m) {
py::class_<Header, LIEF::Object>(m, "Header", "OAT Header representation")
.def(py::init<>())
.def_property_readonly("key_values",
static_cast<no_const_getter<Header::it_key_values_t>>(&Header::key_values),
py::return_value_policy::reference_internal
)
.def_property_readonly("keys",
&Header::keys,
py::return_value_policy::reference_internal
)
.def_property_readonly("values",
&Header::values,
py::return_value_policy::move
)
.def_property_readonly("magic",
static_cast<getter_t<Header::magic_t>>(&Header::magic))
.def_property_readonly("version",
static_cast<getter_t<oat_version_t>>(&Header::version))
.def_property_readonly("checksum",
static_cast<getter_t<uint32_t>>(&Header::checksum))
.def_property_readonly("instruction_set",
static_cast<getter_t<INSTRUCTION_SETS>>(&Header::instruction_set))
.def_property_readonly("nb_dex_files",
static_cast<getter_t<uint32_t>>(&Header::nb_dex_files))
.def_property_readonly("oat_dex_files_offset",
static_cast<getter_t<uint32_t>>(&Header::oat_dex_files_offset))
.def_property_readonly("executable_offset",
static_cast<getter_t<uint32_t>>(&Header::executable_offset))
.def_property_readonly("i2i_bridge_offset",
static_cast<getter_t<uint32_t>>(&Header::i2i_bridge_offset))
.def_property_readonly("i2c_code_bridge_offset",
static_cast<getter_t<uint32_t>>(&Header::i2c_code_bridge_offset))
.def_property_readonly("jni_dlsym_lookup_offset",
static_cast<getter_t<uint32_t>>(&Header::jni_dlsym_lookup_offset))
.def_property_readonly("quick_generic_jni_trampoline_offset",
static_cast<getter_t<uint32_t>>(&Header::quick_generic_jni_trampoline_offset))
.def_property_readonly("quick_imt_conflict_trampoline_offset",
static_cast<getter_t<uint32_t>>(&Header::quick_imt_conflict_trampoline_offset))
.def_property_readonly("quick_resolution_trampoline_offset",
static_cast<getter_t<uint32_t>>(&Header::quick_resolution_trampoline_offset))
.def_property_readonly("quick_to_interpreter_bridge_offset",
static_cast<getter_t<uint32_t>>(&Header::quick_to_interpreter_bridge_offset))
.def_property_readonly("image_patch_delta",
static_cast<getter_t<int32_t>>(&Header::image_patch_delta))
.def_property_readonly("image_file_location_oat_checksum",
static_cast<getter_t<uint32_t>>(&Header::image_file_location_oat_checksum))
.def_property_readonly("image_file_location_oat_data_begin",
static_cast<getter_t<uint32_t>>(&Header::image_file_location_oat_data_begin))
.def_property_readonly("key_value_size",
static_cast<getter_t<uint32_t>>(&Header::key_value_size))
.def("get",
static_cast<const std::string& (Header::*)(HEADER_KEYS) const>(&Header::get),
"key"_a,
py::return_value_policy::move)
.def("set",
static_cast<Header& (Header::*)(HEADER_KEYS, const std::string&)>(&Header::set),
"key"_a, "value"_a,
py::return_value_policy::reference)
.def("__getitem__",
static_cast<const std::string& (Header::*)(HEADER_KEYS) const>(&Header::operator[]),
"",
py::return_value_policy::move)
.def("__setitem__",
static_cast<Header& (Header::*)(HEADER_KEYS, const std::string&)>(&Header::set),
"",
py::return_value_policy::reference)
.def("__eq__", &Header::operator==)
.def("__ne__", &Header::operator!=)
.def("__hash__",
[] (const Header& header) {
return Hash::hash(header);
})
.def("__str__",
[] (const Header& header) {
std::ostringstream stream;
stream << header;
return stream.str();
});
}
}
}