DynamicLoader.cpp 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585
  1. /*
  2. * Copyright (c) 2019-2020, Andrew Kaster <andrewdkaster@gmail.com>
  3. * Copyright (c) 2020, Itamar S. <itamar8910@gmail.com>
  4. * Copyright (c) 2021, Andreas Kling <kling@serenityos.org>
  5. * All rights reserved.
  6. *
  7. * Redistribution and use in source and binary forms, with or without
  8. * modification, are permitted provided that the following conditions are met:
  9. *
  10. * 1. Redistributions of source code must retain the above copyright notice, this
  11. * list of conditions and the following disclaimer.
  12. *
  13. * 2. Redistributions in binary form must reproduce the above copyright notice,
  14. * this list of conditions and the following disclaimer in the documentation
  15. * and/or other materials provided with the distribution.
  16. *
  17. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  18. * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  19. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  20. * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
  21. * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  22. * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  23. * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
  24. * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  25. * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  26. * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  27. */
  28. #include <AK/Debug.h>
  29. #include <AK/Optional.h>
  30. #include <AK/StringBuilder.h>
  31. #include <LibELF/DynamicLinker.h>
  32. #include <LibELF/DynamicLoader.h>
  33. #include <LibELF/Validation.h>
  34. #include <assert.h>
  35. #include <dlfcn.h>
  36. #include <stdio.h>
  37. #include <stdlib.h>
  38. #include <string.h>
  39. #include <sys/mman.h>
  40. #include <sys/stat.h>
  41. #ifndef __serenity__
  42. static void* mmap_with_name(void* addr, size_t length, int prot, int flags, int fd, off_t offset, const char*)
  43. {
  44. return mmap(addr, length, prot, flags, fd, offset);
  45. }
  46. # define MAP_RANDOMIZED 0
  47. #endif
  48. namespace ELF {
  49. RefPtr<DynamicLoader> DynamicLoader::try_create(int fd, String filename)
  50. {
  51. struct stat stat;
  52. if (fstat(fd, &stat) < 0) {
  53. perror("DynamicLoader::try_create fstat");
  54. return {};
  55. }
  56. ASSERT(stat.st_size >= 0);
  57. auto size = static_cast<size_t>(stat.st_size);
  58. if (size < sizeof(Elf32_Ehdr))
  59. return {};
  60. String file_mmap_name = String::formatted("ELF_DYN: {}", filename);
  61. auto* data = mmap_with_name(nullptr, size, PROT_READ, MAP_PRIVATE, fd, 0, file_mmap_name.characters());
  62. if (data == MAP_FAILED) {
  63. perror("DynamicLoader::try_create mmap");
  64. return {};
  65. }
  66. return adopt(*new DynamicLoader(fd, move(filename), data, size));
  67. }
  68. DynamicLoader::DynamicLoader(int fd, String filename, void* data, size_t size)
  69. : m_filename(move(filename))
  70. , m_file_size(size)
  71. , m_image_fd(fd)
  72. , m_file_data(data)
  73. , m_elf_image((u8*)m_file_data, m_file_size)
  74. {
  75. m_tls_size = calculate_tls_size();
  76. m_valid = validate();
  77. }
  78. DynamicLoader::~DynamicLoader()
  79. {
  80. if (munmap(m_file_data, m_file_size) < 0) {
  81. perror("munmap");
  82. ASSERT_NOT_REACHED();
  83. }
  84. if (close(m_image_fd) < 0) {
  85. perror("close");
  86. ASSERT_NOT_REACHED();
  87. }
  88. }
  89. const DynamicObject& DynamicLoader::dynamic_object() const
  90. {
  91. if (!m_cached_dynamic_object) {
  92. VirtualAddress dynamic_section_address;
  93. m_elf_image.for_each_program_header([&dynamic_section_address](auto program_header) {
  94. if (program_header.type() == PT_DYNAMIC) {
  95. dynamic_section_address = VirtualAddress(program_header.raw_data());
  96. }
  97. return IterationDecision::Continue;
  98. });
  99. ASSERT(!dynamic_section_address.is_null());
  100. m_cached_dynamic_object = ELF::DynamicObject::create(VirtualAddress(m_elf_image.base_address()), dynamic_section_address);
  101. }
  102. return *m_cached_dynamic_object;
  103. }
  104. size_t DynamicLoader::calculate_tls_size() const
  105. {
  106. size_t tls_size = 0;
  107. m_elf_image.for_each_program_header([&tls_size](auto program_header) {
  108. if (program_header.type() == PT_TLS) {
  109. tls_size = program_header.size_in_memory();
  110. }
  111. return IterationDecision::Continue;
  112. });
  113. return tls_size;
  114. }
  115. bool DynamicLoader::validate()
  116. {
  117. if (!m_elf_image.is_valid())
  118. return false;
  119. auto* elf_header = (Elf32_Ehdr*)m_file_data;
  120. if (!validate_elf_header(*elf_header, m_file_size))
  121. return false;
  122. if (!validate_program_headers(*elf_header, m_file_size, (u8*)m_file_data, m_file_size, &m_program_interpreter))
  123. return false;
  124. return true;
  125. }
  126. void* DynamicLoader::symbol_for_name(const StringView& name)
  127. {
  128. auto result = m_dynamic_object->hash_section().lookup_symbol(name);
  129. if (!result.has_value())
  130. return nullptr;
  131. auto symbol = result.value();
  132. if (symbol.is_undefined())
  133. return nullptr;
  134. return m_dynamic_object->base_address().offset(symbol.value()).as_ptr();
  135. }
  136. RefPtr<DynamicObject> DynamicLoader::map()
  137. {
  138. ASSERT(!m_dynamic_object);
  139. if (!m_valid) {
  140. dbgln("DynamicLoader::map failed: image is invalid");
  141. return nullptr;
  142. }
  143. load_program_headers();
  144. m_dynamic_object = DynamicObject::create(m_text_segment_load_address, m_dynamic_section_address);
  145. m_dynamic_object->set_tls_offset(m_tls_offset);
  146. m_dynamic_object->set_tls_size(m_tls_size);
  147. return m_dynamic_object;
  148. }
  149. bool DynamicLoader::link(unsigned flags, size_t total_tls_size)
  150. {
  151. return load_stage_2(flags, total_tls_size);
  152. }
  153. bool DynamicLoader::load_stage_2(unsigned flags, size_t total_tls_size)
  154. {
  155. ASSERT(flags & RTLD_GLOBAL);
  156. #if DYNAMIC_LOAD_DEBUG
  157. m_dynamic_object->dump();
  158. #endif
  159. if (m_dynamic_object->has_text_relocations()) {
  160. ASSERT(m_text_segment_load_address.get() != 0);
  161. #ifndef AK_OS_MACOS
  162. // Remap this text region as private.
  163. if (mremap(m_text_segment_load_address.as_ptr(), m_text_segment_size, m_text_segment_size, MAP_PRIVATE) == MAP_FAILED) {
  164. perror("mremap .text: MAP_PRIVATE");
  165. return false;
  166. }
  167. #endif
  168. if (0 > mprotect(m_text_segment_load_address.as_ptr(), m_text_segment_size, PROT_READ | PROT_WRITE)) {
  169. perror("mprotect .text: PROT_READ | PROT_WRITE"); // FIXME: dlerror?
  170. return false;
  171. }
  172. }
  173. do_main_relocations(total_tls_size);
  174. return true;
  175. }
  176. void DynamicLoader::do_main_relocations(size_t total_tls_size)
  177. {
  178. auto do_single_relocation = [&](const ELF::DynamicObject::Relocation& relocation) {
  179. switch (do_relocation(total_tls_size, relocation)) {
  180. case RelocationResult::Failed:
  181. dbgln("Loader.so: {} unresolved symbol '{}'", m_filename, relocation.symbol().name());
  182. ASSERT_NOT_REACHED();
  183. case RelocationResult::ResolveLater:
  184. m_unresolved_relocations.append(relocation);
  185. break;
  186. case RelocationResult::Success:
  187. break;
  188. }
  189. return IterationDecision::Continue;
  190. };
  191. m_dynamic_object->relocation_section().for_each_relocation(do_single_relocation);
  192. m_dynamic_object->plt_relocation_section().for_each_relocation(do_single_relocation);
  193. }
  194. RefPtr<DynamicObject> DynamicLoader::load_stage_3(unsigned flags, size_t total_tls_size)
  195. {
  196. do_lazy_relocations(total_tls_size);
  197. if (flags & RTLD_LAZY) {
  198. if (m_dynamic_object->has_plt())
  199. setup_plt_trampoline();
  200. }
  201. if (mprotect(m_text_segment_load_address.as_ptr(), m_text_segment_size, PROT_READ | PROT_EXEC) < 0) {
  202. perror("mprotect .text: PROT_READ | PROT_EXEC"); // FIXME: dlerror?
  203. return nullptr;
  204. }
  205. if (m_relro_segment_size) {
  206. if (mprotect(m_relro_segment_address.as_ptr(), m_relro_segment_size, PROT_READ) < 0) {
  207. perror("mprotect .relro: PROT_READ");
  208. return nullptr;
  209. }
  210. #if __serenity__
  211. if (set_mmap_name(m_relro_segment_address.as_ptr(), m_relro_segment_size, String::formatted("{}: .relro", m_filename).characters()) < 0) {
  212. perror("set_mmap_name .relro");
  213. return nullptr;
  214. }
  215. #endif
  216. }
  217. call_object_init_functions();
  218. dbgln_if(DYNAMIC_LOAD_DEBUG, "Loaded {}", m_filename);
  219. return m_dynamic_object;
  220. }
  221. void DynamicLoader::do_lazy_relocations(size_t total_tls_size)
  222. {
  223. for (const auto& relocation : m_unresolved_relocations) {
  224. if (auto res = do_relocation(total_tls_size, relocation); res != RelocationResult::Success) {
  225. dbgln("Loader.so: {} unresolved symbol '{}'", m_filename, relocation.symbol().name());
  226. ASSERT_NOT_REACHED();
  227. }
  228. }
  229. }
  230. void DynamicLoader::load_program_headers()
  231. {
  232. Optional<ProgramHeaderRegion> text_region;
  233. Optional<ProgramHeaderRegion> data_region;
  234. Optional<ProgramHeaderRegion> tls_region;
  235. Optional<ProgramHeaderRegion> relro_region;
  236. VirtualAddress dynamic_region_desired_vaddr;
  237. m_elf_image.for_each_program_header([&](const Image::ProgramHeader& program_header) {
  238. ProgramHeaderRegion region {};
  239. region.set_program_header(program_header.raw_header());
  240. if (region.is_tls_template()) {
  241. ASSERT(!tls_region.has_value());
  242. tls_region = region;
  243. } else if (region.is_load()) {
  244. if (region.is_executable()) {
  245. ASSERT(!text_region.has_value());
  246. text_region = region;
  247. } else {
  248. ASSERT(!data_region.has_value());
  249. data_region = region;
  250. }
  251. } else if (region.is_dynamic()) {
  252. dynamic_region_desired_vaddr = region.desired_load_address();
  253. } else if (region.is_relro()) {
  254. ASSERT(!relro_region.has_value());
  255. relro_region = region;
  256. }
  257. return IterationDecision::Continue;
  258. });
  259. ASSERT(text_region.has_value());
  260. ASSERT(data_region.has_value());
  261. // Process regions in order: .text, .data, .tls
  262. void* requested_load_address = m_elf_image.is_dynamic() ? nullptr : text_region.value().desired_load_address().as_ptr();
  263. int reservation_mmap_flags = MAP_ANON | MAP_PRIVATE | MAP_NORESERVE;
  264. if (m_elf_image.is_dynamic())
  265. reservation_mmap_flags |= MAP_RANDOMIZED;
  266. else
  267. reservation_mmap_flags |= MAP_FIXED;
  268. ASSERT(!text_region.value().is_writable());
  269. // First, we make a dummy reservation mapping, in order to allocate enough VM
  270. // to hold both text+data contiguously in the address space.
  271. FlatPtr ph_text_base = text_region.value().desired_load_address().page_base().get();
  272. FlatPtr ph_text_end = round_up_to_power_of_two(text_region.value().desired_load_address().offset(text_region.value().size_in_memory()).get(), PAGE_SIZE);
  273. FlatPtr ph_data_base = data_region.value().desired_load_address().page_base().get();
  274. FlatPtr ph_data_end = round_up_to_power_of_two(data_region.value().desired_load_address().offset(data_region.value().size_in_memory()).get(), PAGE_SIZE);
  275. size_t total_mapping_size = ph_data_end - ph_text_base;
  276. size_t text_segment_size = ph_text_end - ph_text_base;
  277. size_t data_segment_size = ph_data_end - ph_data_base;
  278. auto* reservation = mmap(requested_load_address, total_mapping_size, PROT_NONE, reservation_mmap_flags, 0, 0);
  279. if (reservation == MAP_FAILED) {
  280. perror("mmap reservation");
  281. ASSERT_NOT_REACHED();
  282. }
  283. // Then we unmap the reservation.
  284. if (munmap(reservation, total_mapping_size) < 0) {
  285. perror("munmap reservation");
  286. ASSERT_NOT_REACHED();
  287. }
  288. // Now we can map the text segment at the reserved address.
  289. auto* text_segment_begin = (u8*)mmap_with_name(
  290. reservation,
  291. text_segment_size,
  292. PROT_READ,
  293. MAP_FILE | MAP_SHARED | MAP_FIXED,
  294. m_image_fd,
  295. text_region.value().offset(),
  296. String::formatted("{}: .text", m_filename).characters());
  297. if (text_segment_begin == MAP_FAILED) {
  298. perror("mmap text");
  299. ASSERT_NOT_REACHED();
  300. }
  301. ASSERT(requested_load_address == nullptr || requested_load_address == text_segment_begin);
  302. m_text_segment_size = text_segment_size;
  303. m_text_segment_load_address = VirtualAddress { (FlatPtr)text_segment_begin };
  304. if (relro_region.has_value()) {
  305. m_relro_segment_size = relro_region->size_in_memory();
  306. m_relro_segment_address = m_text_segment_load_address.offset(relro_region->desired_load_address().get());
  307. }
  308. if (m_elf_image.is_dynamic())
  309. m_dynamic_section_address = dynamic_region_desired_vaddr.offset(m_text_segment_load_address.get());
  310. else
  311. m_dynamic_section_address = dynamic_region_desired_vaddr;
  312. FlatPtr data_segment_offset_from_text = ph_data_base - ph_text_base;
  313. // Finally, we make an anonymous mapping for the data segment. Contents are then copied from the file.
  314. auto* data_segment_address = (u8*)text_segment_begin + data_segment_offset_from_text;
  315. auto* data_segment = (u8*)mmap_with_name(
  316. data_segment_address,
  317. data_segment_size,
  318. PROT_READ | PROT_WRITE,
  319. MAP_ANONYMOUS | MAP_PRIVATE | MAP_FIXED,
  320. 0,
  321. 0,
  322. String::formatted("{}: .data", m_filename).characters());
  323. if (MAP_FAILED == data_segment) {
  324. perror("mmap data");
  325. ASSERT_NOT_REACHED();
  326. }
  327. VirtualAddress data_segment_start;
  328. if (m_elf_image.is_dynamic())
  329. data_segment_start = data_region.value().desired_load_address().offset((FlatPtr)text_segment_begin);
  330. else
  331. data_segment_start = data_region.value().desired_load_address();
  332. memcpy(data_segment_start.as_ptr(), (u8*)m_file_data + data_region.value().offset(), data_region.value().size_in_image());
  333. // FIXME: Initialize the values in the TLS section. Currently, it is zeroed.
  334. }
  335. DynamicLoader::RelocationResult DynamicLoader::do_relocation(size_t total_tls_size, const ELF::DynamicObject::Relocation& relocation)
  336. {
  337. dbgln_if(DYNAMIC_LOAD_DEBUG, "Relocation symbol: {}, type: {}", relocation.symbol().name(), relocation.type());
  338. FlatPtr* patch_ptr = nullptr;
  339. if (is_dynamic())
  340. patch_ptr = (FlatPtr*)(m_dynamic_object->base_address().as_ptr() + relocation.offset());
  341. else
  342. patch_ptr = (FlatPtr*)(FlatPtr)relocation.offset();
  343. dbgln_if(DYNAMIC_LOAD_DEBUG, "dynamic object base address: {:p}", m_dynamic_object->base_address().as_ptr());
  344. dbgln_if(DYNAMIC_LOAD_DEBUG, "relocation offset: {:#08x}", relocation.offset());
  345. dbgln_if(DYNAMIC_LOAD_DEBUG, "patch_ptr: {:p}", patch_ptr);
  346. switch (relocation.type()) {
  347. case R_386_NONE:
  348. // Apparently most loaders will just skip these?
  349. // Seems if the 'link editor' generates one something is funky with your code
  350. dbgln_if(DYNAMIC_LOAD_DEBUG, "None relocation. No symbol, no nothing.");
  351. break;
  352. case R_386_32: {
  353. auto symbol = relocation.symbol();
  354. dbgln_if(DYNAMIC_LOAD_DEBUG, "Absolute relocation: name: '{}', value: {}", symbol.name(), symbol.value());
  355. auto res = lookup_symbol(symbol);
  356. if (!res.has_value()) {
  357. if (symbol.bind() == STB_WEAK)
  358. return RelocationResult::ResolveLater;
  359. dbgln("ERROR: symbol not found: {}.", symbol.name());
  360. ASSERT_NOT_REACHED();
  361. }
  362. auto symbol_address = res.value().address;
  363. *patch_ptr += symbol_address.get();
  364. dbgln_if(DYNAMIC_LOAD_DEBUG, " Symbol address: {:p}", *patch_ptr);
  365. break;
  366. }
  367. case R_386_PC32: {
  368. auto symbol = relocation.symbol();
  369. dbgln_if(DYNAMIC_LOAD_DEBUG, "PC-relative relocation: '{}', value: {:p}", symbol.name(), symbol.value());
  370. auto result = lookup_symbol(symbol);
  371. ASSERT(result.has_value());
  372. auto relative_offset = result.value().address - m_dynamic_object->base_address().offset(relocation.offset());
  373. *patch_ptr += relative_offset.get();
  374. dbgln_if(DYNAMIC_LOAD_DEBUG, " Symbol address: {:p}", *patch_ptr);
  375. break;
  376. }
  377. case R_386_GLOB_DAT: {
  378. auto symbol = relocation.symbol();
  379. dbgln_if(DYNAMIC_LOAD_DEBUG, "Global data relocation: '{}', value: {:p}", symbol.name(), symbol.value());
  380. auto res = lookup_symbol(symbol);
  381. if (!res.has_value()) {
  382. // We do not support these
  383. // TODO: Can we tell gcc not to generate the piece of code that uses these?
  384. // (--disable-tm-clone-registry flag in gcc conifugraion?)
  385. if (symbol.name().is_one_of("__deregister_frame_info", "_ITM_registerTMCloneTable", "_ITM_deregisterTMCloneTable", "__register_frame_info"))
  386. break;
  387. if (symbol.bind() == STB_WEAK)
  388. return RelocationResult::ResolveLater;
  389. // Symbol not found
  390. return RelocationResult::Failed;
  391. }
  392. dbgln_if(DYNAMIC_LOAD_DEBUG, "symbol found, location: {:#08x}", res.value().address);
  393. dbgln_if(DYNAMIC_LOAD_DEBUG, "object: {}", m_filename);
  394. auto symbol_location = res.value().address;
  395. ASSERT(symbol_location != m_dynamic_object->base_address());
  396. *patch_ptr = symbol_location.get();
  397. dbgln_if(DYNAMIC_LOAD_DEBUG, " Symbol address: {:p}", *patch_ptr);
  398. break;
  399. }
  400. case R_386_RELATIVE: {
  401. // FIXME: According to the spec, R_386_relative ones must be done first.
  402. // We could explicitly do them first using m_number_of_relocatoins from DT_RELCOUNT
  403. // However, our compiler is nice enough to put them at the front of the relocations for us :)
  404. dbgln_if(DYNAMIC_LOAD_DEBUG, "Load address relocation at offset {:#08x}", relocation.offset());
  405. dbgln_if(DYNAMIC_LOAD_DEBUG, " patch ptr == {:p}, adding load base address ({:p}) to it and storing {:p}", *patch_ptr, m_dynamic_object->base_address().as_ptr(), *patch_ptr + m_dynamic_object->base_address().as_ptr());
  406. *patch_ptr += (FlatPtr)m_dynamic_object->base_address().as_ptr(); // + addend for RelA (addend for Rel is stored at addr)
  407. break;
  408. }
  409. case R_386_TLS_TPOFF32:
  410. case R_386_TLS_TPOFF: {
  411. dbgln_if(DYNAMIC_LOAD_DEBUG, "Relocation type: R_386_TLS_TPOFF at offset {:#08x}", relocation.offset());
  412. auto symbol = relocation.symbol();
  413. // For some reason, LibC has a R_386_TLS_TPOFF that refers to the undefined symbol.. huh
  414. if (relocation.symbol_index() == 0)
  415. break;
  416. dbgln_if(DYNAMIC_LOAD_DEBUG, "Symbol index: {}", symbol.index());
  417. dbgln_if(DYNAMIC_LOAD_DEBUG, "Symbol is_undefined?: {}", symbol.is_undefined());
  418. dbgln_if(DYNAMIC_LOAD_DEBUG, "TLS relocation: '{}', value: {:p}", symbol.name(), symbol.value());
  419. auto res = lookup_symbol(symbol);
  420. if (!res.has_value())
  421. break;
  422. u32 symbol_value = res.value().value;
  423. dbgln_if(DYNAMIC_LOAD_DEBUG, "symbol value: {}", symbol_value);
  424. auto* dynamic_object_of_symbol = res.value().dynamic_object;
  425. ASSERT(dynamic_object_of_symbol);
  426. size_t offset_of_tls_end = dynamic_object_of_symbol->tls_offset().value() + dynamic_object_of_symbol->tls_size().value();
  427. dbgln_if(DYNAMIC_LOAD_DEBUG, "patch ptr: {:p}", patch_ptr);
  428. dbgln_if(DYNAMIC_LOAD_DEBUG, "tls end offset: {}, total tls size: {}", offset_of_tls_end, total_tls_size);
  429. *patch_ptr = (offset_of_tls_end - total_tls_size - symbol_value - sizeof(Elf32_Addr));
  430. dbgln_if(DYNAMIC_LOAD_DEBUG, "*patch ptr: {}", (i32)*patch_ptr);
  431. break;
  432. }
  433. case R_386_JMP_SLOT: {
  434. // FIXME: Or BIND_NOW flag passed in?
  435. if (m_dynamic_object->must_bind_now()) {
  436. // Eagerly BIND_NOW the PLT entries, doing all the symbol looking goodness
  437. // The patch method returns the address for the LAZY fixup path, but we don't need it here
  438. dbgln_if(DYNAMIC_LOAD_DEBUG, "patching plt reloaction: {:p}", relocation.offset_in_section());
  439. m_dynamic_object->patch_plt_entry(relocation.offset_in_section());
  440. } else {
  441. u8* relocation_address = relocation.address().as_ptr();
  442. if (m_elf_image.is_dynamic())
  443. *(u32*)relocation_address += (FlatPtr)m_dynamic_object->base_address().as_ptr();
  444. }
  445. break;
  446. }
  447. default:
  448. // Raise the alarm! Someone needs to implement this relocation type
  449. dbgln("Found a new exciting relocation type {}", relocation.type());
  450. // printf("DynamicLoader: Found unknown relocation type %d\n", relocation.type());
  451. ASSERT_NOT_REACHED();
  452. }
  453. return RelocationResult::Success;
  454. }
  455. // Defined in <arch>/plt_trampoline.S
  456. extern "C" void _plt_trampoline(void) __attribute__((visibility("hidden")));
  457. void DynamicLoader::setup_plt_trampoline()
  458. {
  459. ASSERT(m_dynamic_object);
  460. ASSERT(m_dynamic_object->has_plt());
  461. VirtualAddress got_address = m_dynamic_object->plt_got_base_address();
  462. auto* got_ptr = (FlatPtr*)got_address.as_ptr();
  463. got_ptr[1] = (FlatPtr)m_dynamic_object.ptr();
  464. got_ptr[2] = (FlatPtr)&_plt_trampoline;
  465. dbgln_if(DYNAMIC_LOAD_DEBUG, "Set GOT PLT entries at {:p}: [0] = {:p} [1] = {:p}, [2] = {:p}", got_ptr, (void*)got_ptr[0], (void*)got_ptr[1], (void*)got_ptr[2]);
  466. }
  467. // Called from our ASM routine _plt_trampoline.
  468. // Tell the compiler that it might be called from other places:
  469. extern "C" FlatPtr _fixup_plt_entry(DynamicObject* object, u32 relocation_offset);
  470. extern "C" FlatPtr _fixup_plt_entry(DynamicObject* object, u32 relocation_offset)
  471. {
  472. return object->patch_plt_entry(relocation_offset).get();
  473. }
  474. void DynamicLoader::call_object_init_functions()
  475. {
  476. typedef void (*InitFunc)();
  477. if (m_dynamic_object->has_init_section()) {
  478. auto init_function = (InitFunc)(m_dynamic_object->init_section().address().as_ptr());
  479. dbgln_if(DYNAMIC_LOAD_DEBUG, "Calling DT_INIT at {:p}", init_function);
  480. (init_function)();
  481. }
  482. if (m_dynamic_object->has_init_array_section()) {
  483. auto init_array_section = m_dynamic_object->init_array_section();
  484. InitFunc* init_begin = (InitFunc*)(init_array_section.address().as_ptr());
  485. InitFunc* init_end = init_begin + init_array_section.entry_count();
  486. while (init_begin != init_end) {
  487. // Android sources claim that these can be -1, to be ignored.
  488. // 0 definitely shows up. Apparently 0/-1 are valid? Confusing.
  489. if (!*init_begin || ((FlatPtr)*init_begin == (FlatPtr)-1))
  490. continue;
  491. dbgln_if(DYNAMIC_LOAD_DEBUG, "Calling DT_INITARRAY entry at {:p}", *init_begin);
  492. (*init_begin)();
  493. ++init_begin;
  494. }
  495. }
  496. }
  497. Optional<DynamicObject::SymbolLookupResult> DynamicLoader::lookup_symbol(const ELF::DynamicObject::Symbol& symbol)
  498. {
  499. dbgln_if(DYNAMIC_LOAD_DEBUG, "looking up symbol: {}", symbol.name());
  500. if (symbol.is_undefined() || symbol.bind() == STB_WEAK)
  501. return DynamicLinker::lookup_global_symbol(symbol.name());
  502. dbgln_if(DYNAMIC_LOAD_DEBUG, "symbol is defined in its object");
  503. return DynamicObject::SymbolLookupResult { symbol.value(), symbol.address(), symbol.bind(), &symbol.object() };
  504. }
  505. } // end namespace ELF