Core/cpu: '_' label

Use '_' in assembly to reference address next to current instruction

1 files changed, 6 insertions, 0 deletions

diff --git a/miasm2/core/cpu.py b/miasm2/core/cpu.py
index 4dd7fba3..ffca318b 100644
--- a/miasm2/core/cpu.py
+++ b/miasm2/core/cpu.py
@@ -961,6 +961,9 @@ class instruction(object):
     def get_asm_offset(self, expr):
         return m2_expr.ExprInt(self.offset, expr.size)
 
+    def get_asm_next_offset(self, expr):
+        return m2_expr.ExprInt(self.offset+self.l, expr.size)
+
     def resolve_args_with_symbols(self, symbols=None):
         if symbols is None:
             symbols = {}
@@ -977,6 +980,9 @@ class instruction(object):
                     if name == '$':
                         fixed_ids[x] = self.get_asm_offset(x)
                         continue
+                    if name == '_':
+                        fixed_ids[x] = self.get_asm_next_offset(x)
+                        continue
                     if not name in symbols:
                         raise ValueError('unresolved symbol! %r' % x)
                 else: