Don't yield None in BinaryView hlil|mlil_functions

Author: zznop

python/binaryview.py

@@ -2986,15 +2986,13 @@ def llil_basic_blocks(self) -> Generator['lowlevelil.LowLevelILBasicBlock', None
 	def mlil_basic_blocks(self) -> Generator['mediumlevelil.MediumLevelILBasicBlock', None, None]:
 		"""A generator of all MediumLevelILBasicBlock objects in the BinaryView"""
 		for func in self.mlil_functions():
-			if func is not None:
-				yield from func.basic_blocks
+			yield from func.basic_blocks
 
 	@property
 	def hlil_basic_blocks(self) -> Generator['highlevelil.HighLevelILBasicBlock', None, None]:
 		"""A generator of all HighLevelILBasicBlock objects in the BinaryView"""
 		for func in self.hlil_functions():
-			if func is not None:
-				yield from func.basic_blocks
+			yield from func.basic_blocks
 
 	@property
 	def instructions(self) -> InstructionsType:
@@ -3183,7 +3181,9 @@ def mlil_functions(
 		for func in AdvancedILFunctionList(
 		    self, self.preload_limit if preload_limit is None else preload_limit, function_generator
 		):
-			yield func.mlil
+			mlil = func.mlil
+			if mlil is not None:
+				yield mlil
 
 	def hlil_functions(
 	    self, preload_limit: Optional[int] = None,
@@ -3196,7 +3196,9 @@ def hlil_functions(
 		for func in AdvancedILFunctionList(
 		    self, self.preload_limit if preload_limit is None else preload_limit, function_generator
 		):
-			yield func.hlil
+			hlil = func.hlil
+			if hlil is not None:
+				yield hlil
 
 	@property
 	def has_functions(self) -> bool:

python/function.py

@@ -1546,13 +1546,21 @@ def comment(self, comment: str) -> None:
 	@property
 	def llil_basic_blocks(self) -> Generator['lowlevelil.LowLevelILBasicBlock', None, None]:
 		"""A generator of all LowLevelILBasicBlock objects in the current function"""
-		for block in self.llil:
+		llil = self.llil
+		if llil is None:
+			return
+
+		for block in llil:
 			yield block
 
 	@property
 	def mlil_basic_blocks(self) -> Generator['mediumlevelil.MediumLevelILBasicBlock', None, None]:
 		"""A generator of all MediumLevelILBasicBlock objects in the current function"""
-		for block in self.mlil:
+		mlil = self.mlil
+		if mlil is None:
+			return
+
+		for block in mlil:
 			yield block
 
 	@property
@@ -1880,6 +1888,10 @@ def get_llils_at(self, addr: int,
 			>>> func.get_llils_at(func.start)
 			[<il: push(rbp)>]
 		"""
+		llil = self.llil
+		if llil is None:
+			return []
+
 		if arch is None:
 			arch = self.arch
 		count = ctypes.c_ulonglong()
@@ -1888,7 +1900,7 @@ def get_llils_at(self, addr: int,
 		try:
 			result = []
 			for i in range(0, count.value):
-				result.append(self.llil[instrs[i]])
+				result.append(llil[instrs[i]])
 			return result
 		finally:
 			core.BNFreeILInstructionList(instrs)