object/read/coff/
relocation.rs

Help
1use alloc::fmt;
2use core::slice;
3
4use crate::endian::LittleEndian as LE;
5use crate::pe;
6use crate::read::{
7    ReadRef, Relocation, RelocationEncoding, RelocationFlags, RelocationKind, RelocationTarget,
8    SymbolIndex,
9};
10
11use super::{CoffFile, CoffHeader};
12
13/// An iterator for the relocations in a [`CoffBigSection`](super::CoffBigSection).
14pub type CoffBigRelocationIterator<'data, 'file, R = &'data [u8]> =
15    CoffRelocationIterator<'data, 'file, R, pe::AnonObjectHeaderBigobj>;
16
17/// An iterator for the relocations in a [`CoffSection`](super::CoffSection).
18pub struct CoffRelocationIterator<
19    'data,
20    'file,
21    R: ReadRef<'data> = &'data [u8],
22    Coff: CoffHeader = pe::ImageFileHeader,
23> {
24    pub(super) file: &'file CoffFile<'data, R, Coff>,
25    pub(super) iter: slice::Iter<'data, pe::ImageRelocation>,
26}
27
28impl<'data, 'file, R: ReadRef<'data>, Coff: CoffHeader> Iterator
29    for CoffRelocationIterator<'data, 'file, R, Coff>
30{
31    type Item = (u64, Relocation);
32
33    fn next(&mut self) -> Option<Self::Item> {
34        use RelocationEncoding as E;
35        use RelocationKind as K;
36
37        self.iter.next().map(|relocation| {
38            let typ = relocation.typ.get(LE);
39            let flags = RelocationFlags::Coff { typ };
40            let g = E::Generic;
41            let unknown = (K::Unknown, E::Generic, 0, 0);
42            let (kind, encoding, size, addend) = match self.file.header.machine() {
43                pe::IMAGE_FILE_MACHINE_ARMNT => match typ {
44                    pe::IMAGE_REL_ARM_ADDR32 => (K::Absolute, g, 32, 0),
45                    pe::IMAGE_REL_ARM_ADDR32NB => (K::ImageOffset, g, 32, 0),
46                    pe::IMAGE_REL_ARM_REL32 => (K::Relative, g, 32, -4),
47                    pe::IMAGE_REL_ARM_SECTION => (K::SectionIndex, g, 16, 0),
48                    pe::IMAGE_REL_ARM_SECREL => (K::SectionOffset, g, 32, 0),
49                    _ => unknown,
50                },
51                pe::IMAGE_FILE_MACHINE_ARM64 | pe::IMAGE_FILE_MACHINE_ARM64EC => match typ {
52                    pe::IMAGE_REL_ARM64_ADDR32 => (K::Absolute, g, 32, 0),
53                    pe::IMAGE_REL_ARM64_ADDR32NB => (K::ImageOffset, g, 32, 0),
54                    pe::IMAGE_REL_ARM64_SECREL => (K::SectionOffset, g, 32, 0),
55                    pe::IMAGE_REL_ARM64_SECTION => (K::SectionIndex, g, 16, 0),
56                    pe::IMAGE_REL_ARM64_ADDR64 => (K::Absolute, g, 64, 0),
57                    pe::IMAGE_REL_ARM64_REL32 => (K::Relative, g, 32, -4),
58                    pe::IMAGE_REL_ARM64_BRANCH26 => (K::Relative, E::AArch64Call, 26, 0),
59                    _ => unknown,
60                },
61                pe::IMAGE_FILE_MACHINE_I386 => match typ {
62                    pe::IMAGE_REL_I386_DIR16 => (K::Absolute, g, 16, 0),
63                    pe::IMAGE_REL_I386_REL16 => (K::Relative, g, 16, 0),
64                    pe::IMAGE_REL_I386_DIR32 => (K::Absolute, g, 32, 0),
65                    pe::IMAGE_REL_I386_DIR32NB => (K::ImageOffset, g, 32, 0),
66                    pe::IMAGE_REL_I386_SECTION => (K::SectionIndex, g, 16, 0),
67                    pe::IMAGE_REL_I386_SECREL => (K::SectionOffset, g, 32, 0),
68                    pe::IMAGE_REL_I386_SECREL7 => (K::SectionOffset, g, 7, 0),
69                    pe::IMAGE_REL_I386_REL32 => (K::Relative, g, 32, -4),
70                    _ => unknown,
71                },
72                pe::IMAGE_FILE_MACHINE_AMD64 => match typ {
73                    pe::IMAGE_REL_AMD64_ADDR64 => (K::Absolute, g, 64, 0),
74                    pe::IMAGE_REL_AMD64_ADDR32 => (K::Absolute, g, 32, 0),
75                    pe::IMAGE_REL_AMD64_ADDR32NB => (K::ImageOffset, g, 32, 0),
76                    pe::IMAGE_REL_AMD64_REL32 => (K::Relative, g, 32, -4),
77                    pe::IMAGE_REL_AMD64_REL32_1 => (K::Relative, g, 32, -5),
78                    pe::IMAGE_REL_AMD64_REL32_2 => (K::Relative, g, 32, -6),
79                    pe::IMAGE_REL_AMD64_REL32_3 => (K::Relative, g, 32, -7),
80                    pe::IMAGE_REL_AMD64_REL32_4 => (K::Relative, g, 32, -8),
81                    pe::IMAGE_REL_AMD64_REL32_5 => (K::Relative, g, 32, -9),
82                    pe::IMAGE_REL_AMD64_SECTION => (K::SectionIndex, g, 16, 0),
83                    pe::IMAGE_REL_AMD64_SECREL => (K::SectionOffset, g, 32, 0),
84                    pe::IMAGE_REL_AMD64_SECREL7 => (K::SectionOffset, g, 7, 0),
85                    _ => unknown,
86                },
87                _ => unknown,
88            };
89            let target = RelocationTarget::Symbol(relocation.symbol());
90            (
91                u64::from(relocation.virtual_address.get(LE)),
92                Relocation {
93                    kind,
94                    encoding,
95                    size,
96                    target,
97                    addend,
98                    implicit_addend: true,
99                    flags,
100                },
101            )
102        })
103    }
104}
105
106impl<'data, 'file, R: ReadRef<'data>, Coff: CoffHeader> fmt::Debug
107    for CoffRelocationIterator<'data, 'file, R, Coff>
108{
109    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
110        f.debug_struct("CoffRelocationIterator").finish()
111    }
112}
113
114impl pe::ImageRelocation {
115    /// Get the index of the symbol referenced by this relocation.
116    pub fn symbol(&self) -> SymbolIndex {
117        SymbolIndex(self.symbol_table_index.get(LE) as usize)
118    }
119}
object/read/coff/relocation.rs

object/read/coff/
relocation.rs
