Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(78)

Side by Side Diff: courgette/rel32_finder_win32_x86.cc

Issue 1676683002: [Courgette] Clean up Disassembler; fix ELF Memory leaks. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Wrap #include <iostream> under #if COURGETTE_HISTOGRAM_TARGETS. Created 4 years, 9 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « courgette/rel32_finder_win32_x86.h ('k') | courgette/rel32_finder_win32_x86_unittest.cc » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 // Copyright 2015 The Chromium Authors. All rights reserved. 1 // Copyright 2015 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be 2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file. 3 // found in the LICENSE file.
4 4
5 #include <stdint.h> 5 #include <stdint.h>
6 6
7 #include "courgette/rel32_finder_win32_x86.h" 7 #include "courgette/rel32_finder_win32_x86.h"
8 8
9 namespace courgette { 9 namespace courgette {
10 10
11 Rel32FinderWin32X86::Rel32FinderWin32X86( 11 Rel32FinderWin32X86::Rel32FinderWin32X86(RVA relocs_start_rva,
12 RVA relocs_start_rva, RVA relocs_end_rva, RVA image_end_rva) 12 RVA relocs_end_rva)
13 : relocs_start_rva_(relocs_start_rva), 13 : relocs_start_rva_(relocs_start_rva), relocs_end_rva_(relocs_end_rva) {
14 relocs_end_rva_(relocs_end_rva),
15 image_end_rva_(image_end_rva) {
16 } 14 }
17 15
18 Rel32FinderWin32X86::~Rel32FinderWin32X86() { 16 Rel32FinderWin32X86::~Rel32FinderWin32X86() {
19 } 17 }
20 18
21 void Rel32FinderWin32X86::SwapRel32Locations(std::vector<RVA>* dest) { 19 void Rel32FinderWin32X86::SwapRel32Locations(std::vector<RVA>* dest) {
22 dest->swap(rel32_locations_); 20 dest->swap(rel32_locations_);
23 } 21 }
24 22
25 #if COURGETTE_HISTOGRAM_TARGETS 23 #if COURGETTE_HISTOGRAM_TARGETS
26 void Rel32FinderWin32X86::SwapRel32TargetRVAs(std::map<RVA, int>* dest) { 24 void Rel32FinderWin32X86::SwapRel32TargetRVAs(std::map<RVA, int>* dest) {
27 dest->swap(rel32_target_rvas_); 25 dest->swap(rel32_target_rvas_);
28 } 26 }
29 #endif 27 #endif
30 28
31 Rel32FinderWin32X86_Basic::Rel32FinderWin32X86_Basic( 29 Rel32FinderWin32X86_Basic::Rel32FinderWin32X86_Basic(RVA relocs_start_rva,
32 RVA relocs_start_rva, RVA relocs_end_rva, RVA image_end_rva) 30 RVA relocs_end_rva)
33 : Rel32FinderWin32X86(relocs_start_rva, relocs_end_rva, image_end_rva) { 31 : Rel32FinderWin32X86(relocs_start_rva, relocs_end_rva) {
34 } 32 }
35 33
36 Rel32FinderWin32X86_Basic::~Rel32FinderWin32X86_Basic() { 34 Rel32FinderWin32X86_Basic::~Rel32FinderWin32X86_Basic() {
37 } 35 }
38 36
39 void Rel32FinderWin32X86_Basic::Find(const uint8_t* start_pointer, 37 void Rel32FinderWin32X86_Basic::Find(const uint8_t* start_pointer,
40 const uint8_t* end_pointer, 38 const uint8_t* end_pointer,
41 RVA start_rva, 39 RVA start_rva,
42 RVA end_rva, 40 RVA end_rva,
43 const std::vector<RVA>& abs32_locations) { 41 const std::vector<RVA>& abs32_locations) {
44 // Quick way to convert from Pointer to RVA within a single Section is to 42 // Quick way to convert from Pointer to RVA within a single Section is to
45 // subtract 'pointer_to_rva'. 43 // subtract 'pointer_to_rva'.
46 const uint8_t* const adjust_pointer_to_rva = start_pointer - start_rva; 44 const uint8_t* const adjust_pointer_to_rva = start_pointer - start_rva;
47 45
48 std::vector<RVA>::const_iterator abs32_pos = abs32_locations.begin(); 46 std::vector<RVA>::const_iterator abs32_pos = abs32_locations.begin();
49 47
50 // Find the rel32 relocations. 48 // Find the rel32 relocations.
51 const uint8_t* p = start_pointer; 49 const uint8_t* p = start_pointer;
52 while (p < end_pointer) { 50 while (p < end_pointer) {
53 RVA current_rva = static_cast<RVA>(p - adjust_pointer_to_rva); 51 RVA current_rva = static_cast<RVA>(p - adjust_pointer_to_rva);
52
53 // Skip the base reloation table if we encounter it.
54 // Note: We're not bothering to handle the edge case where a Rel32 pointer
55 // collides with |relocs_start_rva_| by being {1, 2, 3}-bytes before it.
54 if (current_rva == relocs_start_rva_) { 56 if (current_rva == relocs_start_rva_) {
55 if (relocs_start_rva_ < relocs_end_rva_) { 57 if (relocs_start_rva_ < relocs_end_rva_) {
56 p += relocs_end_rva_ - relocs_start_rva_; 58 p += relocs_end_rva_ - relocs_start_rva_;
57 continue; 59 continue;
58 } 60 }
59 } 61 }
60 62
61 //while (abs32_pos != abs32_locations.end() && *abs32_pos < current_rva)
62 // ++abs32_pos;
63
64 // Heuristic discovery of rel32 locations in instruction stream: are the 63 // Heuristic discovery of rel32 locations in instruction stream: are the
65 // next few bytes the start of an instruction containing a rel32 64 // next few bytes the start of an instruction containing a rel32
66 // addressing mode? 65 // addressing mode?
67 const uint8_t* rel32 = NULL; 66 const uint8_t* rel32 = nullptr;
68 67
69 if (p + 5 <= end_pointer) { 68 if (p + 5 <= end_pointer) {
70 if (*p == 0xE8 || *p == 0xE9) { // jmp rel32 and call rel32 69 if (*p == 0xE8 || *p == 0xE9) { // jmp rel32 and call rel32
71 rel32 = p + 1; 70 rel32 = p + 1;
72 } 71 }
73 } 72 }
74 if (p + 6 <= end_pointer) { 73 if (p + 6 <= end_pointer) {
75 if (*p == 0x0F && (*(p+1) & 0xF0) == 0x80) { // Jcc long form 74 if (*p == 0x0F && (*(p+1) & 0xF0) == 0x80) { // Jcc long form
76 if (p[1] != 0x8A && p[1] != 0x8B) // JPE/JPO unlikely 75 if (p[1] != 0x8A && p[1] != 0x8B) // JPE/JPO unlikely
77 rel32 = p + 2; 76 rel32 = p + 2;
(...skipping 10 matching lines...) Expand all
88 if (abs32_pos != abs32_locations.end()) { 87 if (abs32_pos != abs32_locations.end()) {
89 if (*abs32_pos < rel32_rva + 4) { 88 if (*abs32_pos < rel32_rva + 4) {
90 // Beginning of abs32 reloc is before end of rel32 reloc so they 89 // Beginning of abs32 reloc is before end of rel32 reloc so they
91 // overlap. Skip four bytes past the abs32 reloc. 90 // overlap. Skip four bytes past the abs32 reloc.
92 p += (*abs32_pos + 4) - current_rva; 91 p += (*abs32_pos + 4) - current_rva;
93 continue; 92 continue;
94 } 93 }
95 } 94 }
96 95
97 RVA target_rva = rel32_rva + 4 + Read32LittleEndian(rel32); 96 RVA target_rva = rel32_rva + 4 + Read32LittleEndian(rel32);
98 // To be valid, rel32 target must be within image, and within this 97 // Valid, rel32 target must be within image, and within this section.
99 // section. 98 // Subsumes |target_rva| != |kUnassignedRVA|.
100 if (IsValidRVA(target_rva) && 99 if (start_rva <= target_rva && target_rva < end_rva) {
101 start_rva <= target_rva && target_rva < end_rva) {
102 rel32_locations_.push_back(rel32_rva); 100 rel32_locations_.push_back(rel32_rva);
103 #if COURGETTE_HISTOGRAM_TARGETS 101 #if COURGETTE_HISTOGRAM_TARGETS
104 ++rel32_target_rvas_[target_rva]; 102 ++rel32_target_rvas_[target_rva];
105 #endif 103 #endif
106 p = rel32 + 4; 104 p = rel32 + 4;
107 continue; 105 continue;
108 } 106 }
109 } 107 }
110 p += 1; 108 p += 1;
111 } 109 }
112 } 110 }
113 111
114 } // namespace courgette 112 } // namespace courgette
OLDNEW
« no previous file with comments | « courgette/rel32_finder_win32_x86.h ('k') | courgette/rel32_finder_win32_x86_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698