blob: de7bc9719bc464856a41c465b4994a160dcf0c4a (
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
|
from idc import Byte, SegEnd
from idautils import Segments
from miasm2.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):
o = ""
for ad in xrange(l):
o += chr(Byte(ad + start - self.shift))
return o
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 = SegEnd(list(Segments())[-1] - (self.offset + self.shift))
self._getlen = max_addr
return max_addr
|