From e6b55f3c963947d9419a4ca3eaae0bc1c4202506 Mon Sep 17 00:00:00 2001 From: Peter LaFosse Date: Fri, 3 Sep 2021 18:00:11 -0400 Subject: type hints for mediumlevelil.py --- python/mediumlevelil.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) (limited to 'python') diff --git a/python/mediumlevelil.py b/python/mediumlevelil.py index d10080f2..8941b3d4 100644 --- a/python/mediumlevelil.py +++ b/python/mediumlevelil.py @@ -54,6 +54,7 @@ MediumLevelILOperandType = Union[ 'variable.Variable', 'SSAVariable', List[int], + List['variable.Variable'], List['SSAVariable'], List['MediumLevelILInstruction'], Mapping[int, int] @@ -1039,8 +1040,8 @@ class MediumLevelILRet(MediumLevelILInstruction, Return): return self.get_expr_list(0, 1) @property - def operands(self) -> List[MediumLevelILInstruction]: - return [self.src] # TODO: Expand output and params before regeneration + def operands(self) -> List[List[MediumLevelILInstruction]]: + return [self.src] @dataclass(frozen=True, repr=False) @@ -1491,7 +1492,7 @@ class MediumLevelILSyscall(MediumLevelILInstruction, Syscall): @property def operands(self) -> List[MediumLevelILOperandType]: - return [self.output, self.params] # TODO: Expand output and params before regeneration + return [self.output, self.params] @dataclass(frozen=True, repr=False) @@ -1811,7 +1812,7 @@ class MediumLevelILIntrinsic(MediumLevelILInstruction): @property def operands(self) -> List[MediumLevelILOperandType]: - return [self.output, self.intrinsic, self.params] # TODO: Expand output and params before regeneration + return [self.output, self.intrinsic, self.params] @dataclass(frozen=True, repr=False) @@ -1842,7 +1843,7 @@ class MediumLevelILIntrinsic_ssa(MediumLevelILInstruction, SSA): @property def operands(self) -> List[MediumLevelILOperandType]: - return [self.output, self.intrinsic, self.params] # TODO: Expand output and params before regeneration + return [self.output, self.intrinsic, self.params] @dataclass(frozen=True, repr=False) @@ -2118,7 +2119,7 @@ class MediumLevelILCall(MediumLevelILCallBase): @property def operands(self) -> List[MediumLevelILOperandType]: - return [self.output, self.dest, self.params] # TODO: Expand output and params before regeneration + return [self.output, self.dest, self.params] @dataclass(frozen=True, repr=False) @@ -2258,7 +2259,7 @@ class MediumLevelILTailcall(MediumLevelILCallBase, Tailcall): @property def operands(self) -> List[MediumLevelILOperandType]: - return [self.output, self.dest, self.params] # TODO: Expand output and params before regeneration + return [self.output, self.dest, self.params] @dataclass(frozen=True, repr=False) -- cgit v1.3.1