elfkit/
segment.rs

1use error::Error;
2use types;
3use header::Header;
4
5use std::io::BufWriter;
6use std::io::{Read, Write};
7use num_traits::{FromPrimitive, ToPrimitive};
8
9#[derive(Default, Debug, Clone)]
10pub struct SegmentHeader {
11    pub phtype: types::SegmentType,
12    pub flags: types::SegmentFlags,
13    pub offset: u64,
14    pub vaddr: u64,
15    pub paddr: u64,
16    pub filesz: u64,
17    pub memsz: u64,
18    pub align: u64,
19}
20
21impl SegmentHeader {
22    pub fn entsize(eh: &Header) -> usize {
23        match eh.ident_class {
24            types::Class::Class64 => 4 + 4 + 6 * 8,
25            types::Class::Class32 => 4 + 4 + 6 * 4,
26        }
27    }
28
29    pub fn from_reader<R>(io: &mut R, eh: &Header) -> Result<SegmentHeader, Error>
30    where
31        R: Read,
32    {
33        let mut r = SegmentHeader::default();
34        let reb = elf_read_u32!(eh, io)?;
35        r.phtype = match types::SegmentType::from_u32(reb) {
36            Some(v) => v,
37            None => return Err(Error::InvalidSegmentType(reb)),
38        };
39
40        match eh.ident_class {
41            types::Class::Class64 => {
42                r.flags = types::SegmentFlags::from_bits_truncate(elf_read_u32!(eh, io)? as u64);
43                r.offset = elf_read_u64!(eh, io)?;
44                r.vaddr = elf_read_u64!(eh, io)?;
45                r.paddr = elf_read_u64!(eh, io)?;
46                r.filesz = elf_read_u64!(eh, io)?;
47                r.memsz = elf_read_u64!(eh, io)?;
48                r.align = elf_read_u64!(eh, io)?;
49            }
50            types::Class::Class32 => {
51                r.offset = elf_read_u32!(eh, io)? as u64;
52                r.vaddr = elf_read_u32!(eh, io)? as u64;
53                r.paddr = elf_read_u32!(eh, io)? as u64;
54                r.filesz = elf_read_u32!(eh, io)? as u64;
55                r.memsz = elf_read_u32!(eh, io)? as u64;
56                r.flags = types::SegmentFlags::from_bits_truncate(elf_read_u32!(eh, io)? as u64);
57                r.align = elf_read_u32!(eh, io)? as u64;
58            }
59        };
60        Ok(r)
61    }
62    pub fn to_writer<R>(&self, eh: &Header, io: &mut R) -> Result<(), Error>
63    where
64        R: Write,
65    {
66        let mut w = BufWriter::new(io);
67        elf_write_u32!(eh, w, self.phtype.to_u32().unwrap())?;
68        match eh.ident_class {
69            types::Class::Class64 => {
70                elf_write_u32!(eh, w, self.flags.bits() as u32)?;
71                elf_write_u64!(eh, w, self.offset)?;
72                elf_write_u64!(eh, w, self.vaddr)?;
73                elf_write_u64!(eh, w, self.paddr)?;
74                elf_write_u64!(eh, w, self.filesz)?;
75                elf_write_u64!(eh, w, self.memsz)?;
76                elf_write_u64!(eh, w, self.align)?;
77            }
78            types::Class::Class32 => {
79                elf_write_u32!(eh, w, self.offset as u32)?;
80                elf_write_u32!(eh, w, self.vaddr as u32)?;
81                elf_write_u32!(eh, w, self.paddr as u32)?;
82                elf_write_u32!(eh, w, self.filesz as u32)?;
83                elf_write_u32!(eh, w, self.memsz as u32)?;
84                elf_write_u32!(eh, w, self.flags.bits() as u32)?;
85                elf_write_u32!(eh, w, self.align as u32)?;
86            }
87        };
88        Ok(())
89    }
90}
elfkit/segment.rs

elfkit/
segment.rs
