From 83b22a7a2c73773deb74c83dc2feda3f984160ad Mon Sep 17 00:00:00 2001 From: Mizari Date: Thu, 6 Dec 2018 23:47:24 +0700 Subject: perfomance feature --- miasm2/expression/expression.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) (limited to 'miasm2/expression/expression.py') diff --git a/miasm2/expression/expression.py b/miasm2/expression/expression.py index d06b7e21..0e4b7629 100644 --- a/miasm2/expression/expression.py +++ b/miasm2/expression/expression.py @@ -317,15 +317,10 @@ class Expr(object): """Find and replace sub expression using dct @dct: dictionary of Expr -> * """ - if dct is None: - dct = {} - - def my_replace(expr, dct): - if expr in dct: - return dct[expr] - return expr + if not dct: + return self - return self.visit(lambda expr: my_replace(expr, dct)) + return self.visit(lambda expr: dct.get(expr, expr)) def canonize(self): "Canonize the Expression" -- cgit 1.4.1