blob: 15bd9d8b36586c7ce147427303bc070c87fc578f (
plain) (
blame)
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
|
from builtins import range
from idc import get_wide_byte, get_segm_end
from idautils import Segments
from idaapi import is_mapped
from miasm.core.utils import int_to_byte
from miasm.core.bin_stream import bin_stream_str
class bin_stream_ida(bin_stream_str):
"""
bin_stream implementation for IDA
Don't generate xrange using address computation:
It can raise error on overflow 7FFFFFFF with 32 bit python
"""
def _getbytes(self, start, l=1):
out = []
for ad in range(l):
offset = ad + start + self.base_address
if not is_mapped(offset):
raise IOError(f"not enough bytes @ offset {offset:x}")
out.append(int_to_byte(get_wide_byte(offset)))
return b''.join(out)
def readbs(self, l=1):
if self.offset + l > self.l:
raise IOError("not enough bytes")
content = self.getbytes(self.offset)
self.offset += l
return content
def __str__(self):
raise NotImplementedError('Not fully functional')
def setoffset(self, val):
self.offset = val
def getlen(self):
# Lazy version
if hasattr(self, "_getlen"):
return self._getlen
max_addr = get_segm_end(list(Segments())[-1] - (self.offset - self.base_address))
self._getlen = max_addr
return max_addr
|