def test_getmemsz(self): seg = DataElfSegment() self.assertRaises(InvalidArgument, seg.get_memsz) data = ByteArray() data.memsz = 10 seg.set_data(data) self.assertEquals(seg.get_memsz(), 10) seg = HeaderElfSegment() self.assertRaises(Unprepared, seg.get_memsz)
def test_getmemsz(self): seg = DataElfSegment() self.assertRaises(InvalidArgument, seg.get_memsz) data = ByteArray() data.memsz = 10 seg.set_data(data) self.assertEquals(seg.get_memsz(), 10) seg = HeaderElfSegment() self.assertRaises(Unprepared, seg.get_memsz)
def test_get_program_header(self): data = ByteArray() data.memsz = 0 seg = DataElfSegment(data=data) self.assertRaises(Unprepared, seg.get_program_header, "<", 32) seg.prepare(0) self.assertRaises(InvalidArgument, seg.get_program_header, "<", 33) self.assertRaises(InvalidArgument, seg.get_program_header, "ads", 32) ph = seg.get_program_header("<", 32) self.assertEquals(ph.todata(), Elf32ProgramHeader("<").todata())
def test_get_program_header(self): data = ByteArray() data.memsz = 0 seg = DataElfSegment(None, data = data) self.assertRaises(Unprepared, seg.get_program_header, '<', 32) seg.prepare(0) self.assertRaises(InvalidArgument, seg.get_program_header, '<', 33) self.assertRaises(InvalidArgument, seg.get_program_header, 'ads', 32) ph = seg.get_program_header('<', 32) self.assertEquals(ph.todata(), Elf32ProgramHeader('<').todata())
def test_getfilesz(self): seg = DataElfSegment() self.assertRaises(InvalidArgument, seg.get_filesz) data = ByteArray("1234567890") data.memsz = 10 seg.set_data(data) self.assertEquals(seg.get_filesz(), 10) seg = HeaderElfSegment() self.assertRaises(Unprepared, seg.get_filesz) seg = SectionedElfSegment() self.assertEqual(seg.get_filesz(), 0)
def test_getfilesz(self): seg = DataElfSegment() self.assertRaises(InvalidArgument, seg.get_filesz) data = ByteArray("1234567890") data.memsz = 10 seg.set_data(data) self.assertEquals(seg.get_filesz(), 10) seg = HeaderElfSegment() self.assertRaises(Unprepared, seg.get_filesz) seg = SectionedElfSegment() self.assertEqual(seg.get_filesz(), 0)