miasm2.core.sembuilder.MiasmTransformer Class Reference

Inheritance diagram for miasm2.core.sembuilder.MiasmTransformer:

Collaboration diagram for miasm2.core.sembuilder.MiasmTransformer:

Public Member Functions
def	visit_Call
def	visit_Subscript
def	visit_IfExp

Static Public Attributes
tuple	parse_integer = re.compile("^i([0-9]+)$")
tuple	parse_mem = re.compile("^mem([0-9]+)$")

Detailed Description

AST visitor translating DSL to Miasm expression

memX[Y]       -> ExprMem(Y, X)
iX(Y)         -> ExprIntX(Y)
X if Y else Z -> ExprCond(Y, X, Z)
'X'(Y)        -> ExprOp('X', Y)
('X' % Y)(Z)  -> ExprOp('X' % Y, Z)

Definition at line 11 of file sembuilder.py.

Member Function Documentation

def miasm2.core.sembuilder.MiasmTransformer.visit_Call	(		self,
			node
	)

iX(Y) -> ExprIntX(Y),
'X'(Y) -> ExprOp('X', Y), ('X' % Y)(Z) -> ExprOp('X' % Y, Z)

Definition at line 26 of file sembuilder.py.

    def visit_Call(self, node):
        """iX(Y) -> ExprIntX(Y),
        'X'(Y) -> ExprOp('X', Y), ('X' % Y)(Z) -> ExprOp('X' % Y, Z)"""

        # Recursive visit
        node = self.generic_visit(node)

        if isinstance(node.func, ast.Name):
            # iX(Y) -> ExprIntX(Y)
            fc_name = node.func.id

            # Match the function name
            new_name = fc_name
            integer = self.parse_integer.search(fc_name)

            # Do replacement
            if integer is not None:
                new_name = "ExprInt%s" % integer.groups()[0]

            # Replace in the node
            node.func.id = new_name

        elif (isinstance(node.func, ast.Str) or
              (isinstance(node.func, ast.BinOp) and
               isinstance(node.func.op, ast.Mod) and
               isinstance(node.func.left, ast.Str))):
            # 'op'(args...) -> ExprOp('op', args...)
            # ('op' % (fmt))(args...) -> ExprOp('op' % (fmt), args...)
            op_name = node.func

            # Do replacement
            node.func = ast.Name(id="ExprOp", ctx=ast.Load())
            node.args[0:0] = [op_name]

        return node