bin_stream: add getitem

1 files changed, 11 insertions, 0 deletions

diff --git a/miasm/core/bin_stream.py b/miasm/core/bin_stream.py
index 5db15bcc..9236e037 100644
--- a/miasm/core/bin_stream.py
+++ b/miasm/core/bin_stream.py
@@ -39,6 +39,17 @@ class bin_stream(object):
     def hexdump(self, offset, l):
         return
 
+    def __getitem__(self, item):
+        if not type(item) is slice: # integer
+            self.offset = item
+            return self.readbs(1)
+        start = item.start
+        stop  = item.stop
+        step  = item.step
+        self.offset = start
+        s = self.readbs(stop-start)
+        return s[::step]
+
 class bin_stream_str(bin_stream):
     def __init__(self, bin ="", offset = 0L):
         if offset>len(bin):