diff options
Diffstat (limited to 'core/src/xmake/binutils/macho/rpath.c')
| -rw-r--r-- | core/src/xmake/binutils/macho/rpath.c | 135 |
1 files changed, 135 insertions, 0 deletions
diff --git a/core/src/xmake/binutils/macho/rpath.c b/core/src/xmake/binutils/macho/rpath.c new file mode 100644 index 000000000..1c3b8faf1 --- /dev/null +++ b/core/src/xmake/binutils/macho/rpath.c @@ -0,0 +1,135 @@ +/*!A cross-platform build utility based on Lua + * + * 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. + * + * Copyright (C) 2015-present, Xmake Open Source Community. + * + * @author ruki + * @file rpath.c + * + */ + +/* ////////////////////////////////////////////////////////////////////////////////////// + * trace + */ +#define TB_TRACE_MODULE_NAME "rpath_macho" +#define TB_TRACE_MODULE_DEBUG (0) + +/* ////////////////////////////////////////////////////////////////////////////////////// + * includes + */ +#include "prefix.h" + +/* ////////////////////////////////////////////////////////////////////////////////////// + * implementation + */ +tb_bool_t xm_binutils_macho_rpath_list(tb_stream_ref_t istream, tb_hize_t base_offset, lua_State *lua) { + tb_assert_and_check_return_val(istream && lua, tb_false); + + // read Mach-O header + xm_macho_header_t header; + if (!tb_stream_seek(istream, base_offset)) { + return tb_false; + } + if (!tb_stream_bread(istream, (tb_byte_t*)&header, sizeof(header))) { + return tb_false; + } + + tb_bool_t swap = tb_false; + tb_bool_t is64 = tb_false; + + // check magic + if (header.magic == XM_MACHO_MAGIC_32) { + is64 = tb_false; + } else if (header.magic == XM_MACHO_MAGIC_32_BE) { + is64 = tb_false; + swap = tb_true; + } else if (header.magic == XM_MACHO_MAGIC_64) { + is64 = tb_true; + } else if (header.magic == XM_MACHO_MAGIC_64_BE) { + is64 = tb_true; + swap = tb_true; + } else { + return tb_false; // Not a Mach-O file + } + + // re-read header for 64-bit if needed, or just use 32-bit part and skip reserved + tb_uint32_t ncmds = 0; + + if (is64) { + xm_macho_header_64_t header64; + if (!tb_stream_seek(istream, base_offset)) { + return tb_false; + } + if (!tb_stream_bread(istream, (tb_byte_t*)&header64, sizeof(header64))) { + return tb_false; + } + xm_binutils_macho_swap_header_64(&header64, swap); + ncmds = header64.ncmds; + } else { + xm_binutils_macho_swap_header_32(&header, swap); + ncmds = header.ncmds; + } + + // skip header to reach load commands + tb_size_t header_size = is64 ? sizeof(xm_macho_header_64_t) : sizeof(xm_macho_header_t); + if (!tb_stream_seek(istream, base_offset + header_size)) { + return tb_false; + } + + tb_size_t result_count = 0; + + // iterate load commands + for (tb_uint32_t i = 0; i < ncmds; i++) { + xm_macho_load_command_t lc; + tb_hize_t current_cmd_offset = tb_stream_offset(istream); + + if (!tb_stream_bread(istream, (tb_byte_t*)&lc, sizeof(lc))) { + return tb_false; + } + xm_binutils_macho_swap_load_command(&lc, swap); + + // check for LC_RPATH + if (lc.cmd == XM_MACHO_LC_RPATH) { + + xm_macho_rpath_command_t rc; + if (tb_stream_seek(istream, current_cmd_offset)) { + if (tb_stream_bread(istream, (tb_byte_t*)&rc, sizeof(rc))) { + xm_binutils_macho_swap_rpath_command(&rc, swap); + + tb_uint32_t name_offset = rc.path_offset; + if (name_offset < lc.cmdsize) { + // name is at current_cmd_offset + name_offset + tb_char_t rpath[TB_PATH_MAXN]; + tb_size_t max_len = lc.cmdsize - name_offset; + if (max_len > sizeof(rpath) - 1) max_len = sizeof(rpath) - 1; + + if (xm_binutils_read_string(istream, current_cmd_offset + name_offset, rpath, max_len + 1) && rpath[0]) { + lua_pushinteger(lua, result_count + 1); + lua_pushstring(lua, rpath); + lua_settable(lua, -3); + result_count++; + } + } + } + } + } + + // move to next command + if (!tb_stream_seek(istream, current_cmd_offset + lc.cmdsize)) { + break; + } + } + + return tb_true; +} |
