summaryrefslogtreecommitdiff
path: root/python
diff options
context:
space:
mode:
authorPeter LaFosse <peter@vector35.com>2021-09-03 18:00:11 -0400
committerPeter LaFosse <peter@vector35.com>2021-09-06 11:46:44 -0400
commite6b55f3c963947d9419a4ca3eaae0bc1c4202506 (patch)
tree40249691e9f9b787f55475fb5226865a27968be0 /python
parent6263d15f0c5818cf2df62e2e6d4103a978251f63 (diff)
type hints for mediumlevelil.py
Diffstat (limited to 'python')
-rw-r--r--python/mediumlevelil.py15
1 files changed, 8 insertions, 7 deletions
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)