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

Side by Side Diff: courgette/rel32_finder_win32_x86.h

Issue 1543643002: Switch to standard integer types in courgette/. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: fix Created 4 years, 12 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/region.h ('k') | courgette/rel32_finder_win32_x86.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 #ifndef COURGETTE_REL32_FINDER_WIN32_X86_H_ 5 #ifndef COURGETTE_REL32_FINDER_WIN32_X86_H_
6 #define COURGETTE_REL32_FINDER_WIN32_X86_H_ 6 #define COURGETTE_REL32_FINDER_WIN32_X86_H_
7 7
8 #include <stdint.h>
9
8 #if COURGETTE_HISTOGRAM_TARGETS 10 #if COURGETTE_HISTOGRAM_TARGETS
9 #include <map> 11 #include <map>
10 #endif 12 #endif
11 #include <vector> 13 #include <vector>
12 14
13 #include "base/basictypes.h"
14 #include "courgette/image_utils.h" 15 #include "courgette/image_utils.h"
15 16
16 namespace courgette { 17 namespace courgette {
17 18
18 // A helper class to scan through a section of code to extract RVAs. 19 // A helper class to scan through a section of code to extract RVAs.
19 class Rel32FinderWin32X86 { 20 class Rel32FinderWin32X86 {
20 public: 21 public:
21 Rel32FinderWin32X86(RVA relocs_start_rva, RVA relocs_end_rva, 22 Rel32FinderWin32X86(RVA relocs_start_rva, RVA relocs_end_rva,
22 RVA image_end_rva); 23 RVA image_end_rva);
23 virtual ~Rel32FinderWin32X86(); 24 virtual ~Rel32FinderWin32X86();
24 25
25 bool IsValidRVA(RVA rva) const { return rva < image_end_rva_; } 26 bool IsValidRVA(RVA rva) const { return rva < image_end_rva_; }
26 27
27 // Swaps data in |rel32_locations_| to |dest|. 28 // Swaps data in |rel32_locations_| to |dest|.
28 void SwapRel32Locations(std::vector<RVA>* dest); 29 void SwapRel32Locations(std::vector<RVA>* dest);
29 30
30 #if COURGETTE_HISTOGRAM_TARGETS 31 #if COURGETTE_HISTOGRAM_TARGETS
31 // Swaps data in |rel32_target_rvas_| to |dest|. 32 // Swaps data in |rel32_target_rvas_| to |dest|.
32 void SwapRel32TargetRVAs(std::map<RVA, int>* dest); 33 void SwapRel32TargetRVAs(std::map<RVA, int>* dest);
33 #endif 34 #endif
34 35
35 // Scans through [|start_pointer|, |end_pointer|) for rel32 addresses. Seeks 36 // Scans through [|start_pointer|, |end_pointer|) for rel32 addresses. Seeks
36 // RVAs that satisfy the following: 37 // RVAs that satisfy the following:
37 // - Do not collide with |abs32_pos| (assumed sorted). 38 // - Do not collide with |abs32_pos| (assumed sorted).
38 // - Do not collide with |base_relocation_table|'s RVA range, 39 // - Do not collide with |base_relocation_table|'s RVA range,
39 // - Whose targets are in [|start_rva|, |end_rva|). 40 // - Whose targets are in [|start_rva|, |end_rva|).
40 // The sorted results are written to |rel32_locations_|. 41 // The sorted results are written to |rel32_locations_|.
41 virtual void Find(const uint8* start_pointer, 42 virtual void Find(const uint8_t* start_pointer,
42 const uint8* end_pointer, 43 const uint8_t* end_pointer,
43 RVA start_rva, 44 RVA start_rva,
44 RVA end_rva, 45 RVA end_rva,
45 const std::vector<RVA>& abs32_locations) = 0; 46 const std::vector<RVA>& abs32_locations) = 0;
46 47
47 protected: 48 protected:
48 const RVA relocs_start_rva_; 49 const RVA relocs_start_rva_;
49 const RVA relocs_end_rva_; 50 const RVA relocs_end_rva_;
50 const RVA image_end_rva_; 51 const RVA image_end_rva_;
51 52
52 std::vector<RVA> rel32_locations_; 53 std::vector<RVA> rel32_locations_;
53 54
54 #if COURGETTE_HISTOGRAM_TARGETS 55 #if COURGETTE_HISTOGRAM_TARGETS
55 std::map<RVA, int> rel32_target_rvas_; 56 std::map<RVA, int> rel32_target_rvas_;
56 #endif 57 #endif
57 }; 58 };
58 59
59 // The basic implementation performs naive scan for rel32 JMP and Jcc opcodes 60 // The basic implementation performs naive scan for rel32 JMP and Jcc opcodes
60 // (excluding JPO/JPE) disregarding instruction alignment. 61 // (excluding JPO/JPE) disregarding instruction alignment.
61 class Rel32FinderWin32X86_Basic : public Rel32FinderWin32X86 { 62 class Rel32FinderWin32X86_Basic : public Rel32FinderWin32X86 {
62 public: 63 public:
63 Rel32FinderWin32X86_Basic(RVA relocs_start_rva, RVA relocs_end_rva, 64 Rel32FinderWin32X86_Basic(RVA relocs_start_rva, RVA relocs_end_rva,
64 RVA image_end_rva); 65 RVA image_end_rva);
65 virtual ~Rel32FinderWin32X86_Basic(); 66 virtual ~Rel32FinderWin32X86_Basic();
66 67
67 // Rel32FinderWin32X86 implementation. 68 // Rel32FinderWin32X86 implementation.
68 void Find(const uint8* start_pointer, 69 void Find(const uint8_t* start_pointer,
69 const uint8* end_pointer, 70 const uint8_t* end_pointer,
70 RVA start_rva, 71 RVA start_rva,
71 RVA end_rva, 72 RVA end_rva,
72 const std::vector<RVA>& abs32_locations) override; 73 const std::vector<RVA>& abs32_locations) override;
73 }; 74 };
74 75
75 } // namespace courgette 76 } // namespace courgette
76 77
77 #endif // COURGETTE_REL32_FINDER_WIN32_X86_H_ 78 #endif // COURGETTE_REL32_FINDER_WIN32_X86_H_
OLDNEW
« no previous file with comments | « courgette/region.h ('k') | courgette/rel32_finder_win32_x86.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698