better instr removing

clayne · Dec 2, 2023 · 9bf64b6 · 9bf64b6
1 parent 972ba87
commit 9bf64b6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/herast/tree/ast_patch.py b/herast/tree/ast_patch.py
@@ -24,6 +24,12 @@ def remove_instr(item:idaapi.cinsn_t, ctx:ASTContext) -> bool:
 			unused_labels.append(lnum)
 
 	removed_labels = collect_labels(item)
+	for u in unused_labels:
+		try:
+			removed_labels.remove(u)
+		except ValueError:
+			pass
+
 	if len(removed_labels) > 0:
 		print("[!] failed removing item with labels in it")
 		return False