blob: 501972ff19c55231d24795ba3245a478b8fb1cd5 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
//
// ZapRelocs.h
//
//
// Zapping of relocations
//
// ======================================================================================
#ifndef __ZAPRELOCS_H__
#define __ZAPRELOCS_H__
typedef BYTE ZapRelocationType; // IMAGE_REL_XXX enum
// Special NGEN-specific relocation type for fixups (absolute RVA in the middle 30 bits)
#define IMAGE_REL_BASED_ABSOLUTE_TAGGED 0x7E
// Special NGEN-specific relocation type for relative pointer (used to make NGen relocation section smaller)
#define IMAGE_REL_BASED_RELPTR 0x7D
#define IMAGE_REL_BASED_RELPTR32 0x7C
// Invalid reloc marker (used to mark end of the reloc array)
#define IMAGE_REL_INVALID 0xFF
// IMAGE_REL_BASED_PTR is architecture specific reloc of virtual address
#ifdef _TARGET_64BIT_
#define IMAGE_REL_BASED_PTR IMAGE_REL_BASED_DIR64
#else
#define IMAGE_REL_BASED_PTR IMAGE_REL_BASED_HIGHLOW
#endif
// Size of base relocs relocation page
#define RELOCATION_PAGE_SIZE 0x1000
//
// The ZapNode for regular PE base relocs
//
class ZapBaseRelocs : public ZapNode
{
ZapImage * m_pImage;
// The page currently being written
DWORD m_page;
COUNT_T m_pageIndex;
// Reloc writer output
SArray<USHORT> m_SerializedRelocs;
void FlushWriter();
public:
ZapBaseRelocs(ZapImage * pImage)
: m_pImage(pImage)
{
// Everything is zero initialized by the allocator
}
void WriteReloc(PVOID pSrc, int offset, ZapNode * pTarget, int targetOffset, ZapRelocationType type);
virtual DWORD GetSize()
{
return m_SerializedRelocs.GetCount() * sizeof(USHORT);
}
virtual ZapNodeType GetType()
{
return ZapNodeType_Relocs;
}
virtual void Save(ZapWriter * pZapWriter);
};
//
//
// Blob with associated relocations. Used for compiled code.
//
struct ZapReloc
{
ZapRelocationType m_type;
DWORD m_offset;
ZapNode * m_pTargetNode;
};
class ZapBlobWithRelocs : public ZapBlob
{
ZapReloc * m_pRelocs;
protected:
ZapBlobWithRelocs(SIZE_T cbSize)
: ZapBlob(cbSize)
{
}
public:
void SetRelocs(ZapReloc * pRelocs)
{
_ASSERTE(m_pRelocs == NULL);
_ASSERTE(pRelocs != NULL);
m_pRelocs = pRelocs;
}
ZapReloc * GetRelocs()
{
return m_pRelocs;
}
virtual PBYTE GetData()
{
return (PBYTE)(this + 1);
}
virtual void Save(ZapWriter * pZapWriter);
// Returns number of straddler relocs, assuming RVA of the node is dwPos
COUNT_T GetCountOfStraddlerRelocations(DWORD dwPos);
// Create new zap blob node. The node *does* own copy of the memory.
static ZapBlobWithRelocs * NewBlob(ZapWriter * pWriter, PVOID pData, SIZE_T cbSize);
// Create new aligned zap blob node.
static ZapBlobWithRelocs * NewAlignedBlob(ZapWriter * pWriter, PVOID pData, SIZE_T cbSize, SIZE_T cbAlignment);
};
#endif // __ZAPRELOCS_H__
|
