Pain

Author: Scags

.gitignore

@@ -0,0 +1,3 @@
+
+.vscode/*
+iconbuster.py

findmyfunc.py

@@ -2,19 +2,16 @@
 import idaapi
 
 def getsigloc(sig):
-	segend = ida_segment.get_segm_by_name(".text").end_ea
-	addr = ida_search.find_binary(0, segend, sig, 0, idc.SEARCH_DOWN|idc.SEARCH_NEXT)
+	segend = idaapi.get_segm_by_name(".text").end_ea
+	addr = idaapi.find_binary(0, segend, sig, 0, idc.SEARCH_DOWN|idc.SEARCH_NEXT)
 	return addr
 
-def main():
-	sig = ida_kernwin.ask_str("", 0, "Insert signature: ")
+sig = idaapi.ask_str("", 0, "Insert signature: ")
 
-	oldsig = sig
-	sig = sig.replace(r"\x", " ").replace("2A", "?").replace("2a", "?").strip()
+oldsig = sig
+sig = sig.replace(r"\x", " ").replace("2A", "?").replace("2a", "?").strip()
 
+def main():
 	loc = getsigloc(sig)
 	if loc != idc.BADADDR:
-		ida_kernwin.jumpto(loc)
-
-if __name__ == "__main__":
-	main()
+		idaapi.jumpto(loc)

gamedata_checker.py

@@ -5,11 +5,11 @@
 
 from sys import version_info
 
-FUNCS_SEGEND = ida_segment.get_segm_by_name(".text").end_ea
+FUNCS_SEGEND = idaapi.get_segm_by_name(".text").end_ea
 
 def get_os():
 	# Lazy af lol
-	return "linux" if ida_nalt.get_root_filename().endswith(".so") else "windows"
+	return "linux" if idaapi.get_root_filename().endswith(".so") else "windows"
 
 def checksig(sig):
 	if sig[0] == '@':
@@ -19,10 +19,10 @@ def checksig(sig):
 	sig = sig.replace(r"\x", " ").replace("2A", "?").replace("2a", "?").replace("\\", "").strip()
 	count = 0
 	addr = 0
-	addr = ida_search.find_binary(addr, FUNCS_SEGEND, sig, 0, idc.SEARCH_DOWN|idc.SEARCH_NEXT)
+	addr = idaapi.find_binary(addr, FUNCS_SEGEND, sig, 0, idc.SEARCH_DOWN|idc.SEARCH_NEXT)
 	while addr != idc.BADADDR:
 		count = count + 1
-		addr = ida_search.find_binary(addr, FUNCS_SEGEND, sig, 0, idc.SEARCH_DOWN|idc.SEARCH_NEXT)
+		addr = idaapi.find_binary(addr, FUNCS_SEGEND, sig, 0, idc.SEARCH_DOWN|idc.SEARCH_NEXT)
 
 	return count == 1
 
@@ -36,7 +36,7 @@ def get_gamedir(kv):
 	gamedir = ""
 	# If we've got multiple games supported, so let's just ask
 	if len(kv.items()) > 1:
-		gamedir = ida_kernwin.ask_str("", 0, "There are multiple supported games with this file. Which game directory is this for?")
+		gamedir = idaapi.ask_str("", 0, "There are multiple supported games with this file. Which game directory is this for?")
 		# Not in the basic game shit, check for support in default
 		if gamedir not in kv.keys():
 			default = kv.get("#default")
@@ -58,7 +58,7 @@ def get_gamedir(kv):
 			supported = kv.get("#supported")
 			if supported:
 				if len(supported.items()) > 1:
-					gamedir = ida_kernwin.ask_str("", 0, "There are multiple supported games with this file. Which game directory is this for?")
+					gamedir = idaapi.ask_str("", 0, "There are multiple supported games with this file. Which game directory is this for?")
 					if gamedir in default["#supported"].values():
 						return gamedir
 					return ""
@@ -86,10 +86,10 @@ def read_vtable(funcname, ea):
 	offset = 0
 	while ea != idc.BADADDR:
 		offs = idc.get_wide_dword(ea)
-		if not ida_bytes.is_code(ida_bytes.get_full_flags(offs)):
+		if not idaapi.is_code(idaapi.get_full_flags(offs)):
 			break
 
-		name = idc.get_name(offs, ida_name.GN_VISIBLE)
+		name = idc.get_name(offs, idaapi.GN_VISIBLE)
 		demangled = idc.demangle_name(name, idc.get_inf_attr(idc.INF_SHORT_DN))
 		if demangled == None:
 			demangled = name
@@ -99,7 +99,7 @@ def read_vtable(funcname, ea):
 		funcs[demangled.lower()] = offset
 
 		offset += 1
-		ea = ida_bytes.next_not_tail(ea)
+		ea = idaapi.next_not_tail(ea)
 
 	# We've got a list of function names, let's do this really shittily because idk any other way
 
@@ -152,7 +152,7 @@ def try_get_voffset(funcname):
 
 	# Let's chug along all of these functions, woohoo for option 2!
 	for func in idautils.Functions():
-		name = idc.get_name(func, ida_name.GN_VISIBLE)
+		name = idc.get_name(func, idaapi.GN_VISIBLE)
 		if not name or funcname not in name:	# funcname should only be a plain function decl, so it would be unfettered in a mangled name
 			continue
 
@@ -181,18 +181,22 @@ def try_get_voffset(funcname):
 
 def main():
 	kv = None
-	with open(ida_kernwin.ask_file(0, "*.txt", "Select a gamedata file")) as f:
+	filereq = idaapi.ask_file(0, "*.txt", "Select a gamedata file")
+	if filereq == None:
+		return
+
+	with open(filereq) as f:
 		kv = vdf.load(f)
 
 	if kv == None:
-		ida_kernwin.warning("Could not load file!")
+		idaapi.warning("Could not load file!")
 		return
 
 	kv = list(kv.values())[0]
 	os = get_os()
 	gamedir = get_gamedir(kv)
 	if not gamedir:
-		ida_kernwin.warning("Could not find game directory in file")
+		idaapi.warning("Could not find game directory in file")
 		return
 
 	kv = kv[gamedir]
@@ -238,7 +242,6 @@ def main():
 #	if os == "windows" and kv.get("Offsets"):
 #		print("Offset checking is not supported on Windows binaries")
 
-	ida_kernwin.warning("Check console for output")
+	idaapi.ask_form("Validated\nCheck console for output")
 
-if __name__ == "__main__":
-	main()
+main()

getfuncoffset.py

@@ -10,5 +10,4 @@ def main():
 
 	print("Offset from %X to %X:\n%d (0x%X)" % (funcstart, addr, addr - funcstart, addr - funcstart))
 
-if __name__ == "__main__":
-	main()
+main()

isgoodsig.py

@@ -1,10 +1,10 @@
 import idc
 import idaapi
 
-FUNCS_SEGEND = ida_segment.get_segm_by_name(".text").end_ea
+FUNCS_SEGEND = idaapi.get_segm_by_name(".text").end_ea
 
 def main():
-	sig = ida_kernwin.ask_str("", 0, "Insert signature: ")
+	sig = idaapi.ask_str("", 0, "Insert signature: ")
 
 	# wtfwtfwtfwtf
 	oldsig = sig
@@ -24,12 +24,11 @@ def main():
 def checksig(sig):
 	count = 0
 	addr = 0
-	addr = ida_search.find_binary(addr, FUNCS_SEGEND, sig, 0, idc.SEARCH_DOWN|idc.SEARCH_NEXT)
+	addr = idaapi.find_binary(addr, FUNCS_SEGEND, sig, 0, idc.SEARCH_DOWN|idc.SEARCH_NEXT)
 	while addr != idc.BADADDR:
 		count = count + 1
-		addr = ida_search.find_binary(addr, FUNCS_SEGEND, sig, 0, idc.SEARCH_DOWN|idc.SEARCH_NEXT)
+		addr = idaapi.find_binary(addr, FUNCS_SEGEND, sig, 0, idc.SEARCH_DOWN|idc.SEARCH_NEXT)
 
 	return count
 
-if __name__ == "__main__":
-	main()
+main()

makesig.py

@@ -2,33 +2,27 @@
 import idautils
 import idaapi
 
-FUNCS_SEGEND  = ida_segment.get_segm_by_name(".text").end_ea
+FUNCS_SEGEND  = idaapi.get_segm_by_name(".text").end_ea
 
 def get_dt_size(dtype):
-	if dtype == ida_ua.dt_byte:
-		return 1
-	elif dtype == ida_ua.dt_word:
-		return 2
-	elif dtype == ida_ua.dt_dword:
-		return 4
-	elif dtype == ida_ua.dt_float:
-		return 4
-	elif dtype == ida_ua.dt_double:
-		return 8
-	else:
-		print("Unknown type size (%d)" % dtype)
-		return -1
+	return {
+		idaapi.dt_byte: 1,
+		idaapi.dt_word: 2,
+		idaapi.dt_dword: 4,
+		idaapi.dt_float: 4,
+		idaapi.dt_double: 8,		
+	}.get(dtype, -1)
 
 def print_wildcards(count):
 	return "? " * count
 
 def is_good_sig(sig):
 	count = 0
 	addr = 0
-	addr = ida_search.find_binary(addr, FUNCS_SEGEND, sig, 0, idc.SEARCH_DOWN|idc.SEARCH_NEXT)
+	addr = idaapi.find_binary(addr, FUNCS_SEGEND, sig, 0, idc.SEARCH_DOWN|idc.SEARCH_NEXT)
 	while count < 2 and addr != idc.BADADDR:
 		count = count + 1
-		addr = ida_search.find_binary(addr, FUNCS_SEGEND, sig, 0, idc.SEARCH_DOWN|idc.SEARCH_NEXT)
+		addr = idaapi.find_binary(addr, FUNCS_SEGEND, sig, 0, idc.SEARCH_DOWN|idc.SEARCH_NEXT)
 
 	return count == 1
 
@@ -40,7 +34,7 @@ def makesig():
 		print("Make sure you are in a function!")
 		return
 
-	name = idc.get_name(addr, ida_name.GN_VISIBLE);
+	name = idc.get_name(addr, idaapi.GN_VISIBLE);
 	funcend = idc.get_func_attr(addr, idc.FUNCATTR_END);
 
 	sig = ""
@@ -49,12 +43,12 @@ def makesig():
 
 	addr = funcstart
 	while addr != idc.BADADDR:
-		info = ida_ua.insn_t()
-		if not ida_ua.decode_insn(info, addr):
+		info = idaapi.insn_t()
+		if not idaapi.decode_insn(info, addr):
 			return None
 
 		done = 0
-		if info.Op1.type == ida_ua.o_near or info.Op1.type == ida_ua.o_far:
+		if info.Op1.type == idaapi.o_near or info.Op1.type == idaapi.o_far:
 			if (idc.get_wide_byte(addr)) == 0x0F: 	# Two-byte instruction
 				sig = sig + ("0F %02X " % idc.get_wide_byte(addr + 1)) + print_wildcards(get_dt_size(info.Op1.dtype))
 			else:
@@ -66,7 +60,7 @@ def makesig():
 			size = idc.get_item_size(addr)
 			while 1:	# Screw u python
 				loc = addr + i
-				if ((idc.get_fixup_target_type(loc) & 0xF) == ida_fixup.FIXUP_OFF32):
+				if ((idc.get_fixup_target_type(loc) & 0xF) == idaapi.FIXUP_OFF32):
 					sig = sig + print_wildcards(4)
 					i = i + 3
 				else:
@@ -105,5 +99,4 @@ def makesig():
 def main():
 	makesig()
 
-if __name__ == "__main__":
-	main()
+main()

makesigfromhere.py

@@ -2,33 +2,27 @@
 import idautils
 import idaapi
 
-FUNCS_SEGEND  = ida_segment.get_segm_by_name(".text").end_ea
+FUNCS_SEGEND  = idaapi.get_segm_by_name(".text").end_ea
 
 def get_dt_size(dtype):
-	if dtype == ida_ua.dt_byte:
-		return 1
-	elif dtype == ida_ua.dt_word:
-		return 2
-	elif dtype == ida_ua.dt_dword:
-		return 4
-	elif dtype == ida_ua.dt_float:
-		return 4
-	elif dtype == ida_ua.dt_double:
-		return 8
-	else:
-		print("Unknown type size (%d)" % dtype)
-		return -1
+	return {
+		idaapi.dt_byte: 1,
+		idaapi.dt_word: 2,
+		idaapi.dt_dword: 4,
+		idaapi.dt_float: 4,
+		idaapi.dt_double: 8,
+	}.get(dtype, -1)
 
 def print_wildcards(count):
 	return "? " * count
 
 def is_good_sig(sig):
 	count = 0
 	addr = 0
-	addr = ida_search.find_binary(addr, FUNCS_SEGEND, sig, 0, idc.SEARCH_DOWN|idc.SEARCH_NEXT)
+	addr = idaapi.find_binary(addr, FUNCS_SEGEND, sig, 0, idc.SEARCH_DOWN|idc.SEARCH_NEXT)
 	while count < 2 and addr != idc.BADADDR:
 		count = count + 1
-		addr = ida_search.find_binary(addr, FUNCS_SEGEND, sig, 0, idc.SEARCH_DOWN|idc.SEARCH_NEXT)
+		addr = idaapi.find_binary(addr, FUNCS_SEGEND, sig, 0, idc.SEARCH_DOWN|idc.SEARCH_NEXT)
 
 	return count == 1
 
@@ -40,20 +34,20 @@ def makesig():
 		return
 
 	funcend = idc.get_func_attr(addr, idc.FUNCATTR_END);
-	name = idc.get_name(funcstart, ida_name.GN_VISIBLE);
+	name = idc.get_name(funcstart, idaapi.GN_VISIBLE);
 
 	sig = ""
 	found = 0
 	done = 0
 	startaddr = addr
 
 	while addr != idc.BADADDR:
-		info = ida_ua.insn_t()
-		if not ida_ua.decode_insn(info, addr):
+		info = idaapi.insn_t()
+		if not idaapi.decode_insn(info, addr):
 			return None
 
 		done = 0
-		if info.Op1.type == ida_ua.o_near or info.Op1.type == ida_ua.o_far:
+		if info.Op1.type == idaapi.o_near or info.Op1.type == idaapi.o_far:
 			if (idc.get_wide_byte(addr)) == 0x0F: 	# Two-byte instruction
 				sig = sig + ("0F %02X " % idc.get_wide_byte(addr + 1)) + print_wildcards(get_dt_size(info.Op1.dtype))
 			else:
@@ -65,7 +59,7 @@ def makesig():
 			size = idc.get_item_size(addr)
 			while 1:	# Screw u python
 				loc = addr + i
-				if ((idc.get_fixup_target_type(loc) & 0xF) == ida_fixup.FIXUP_OFF32):
+				if ((idc.get_fixup_target_type(loc) & 0xF) == idaapi.FIXUP_OFF32):
 					sig = sig + print_wildcards(4)
 					i = i + 3
 				else:
@@ -104,5 +98,4 @@ def makesig():
 def main():
 	makesig()
 
-if __name__ == "__main__":
-	main()
+main()

nameresetter.py

@@ -6,17 +6,16 @@ def main():
 	segstart = 0
 	segend = None
 
-	segm = ida_segment.get_segm_by_name(".text")
+	segm = idaapi.get_segm_by_name(".text")
 	if segm:
 		segstart = segm.start_ea
 		segend = segm.end_ea
 
 	for fea in idautils.Functions(segstart, segend):
-		flags = idc.get_func_attr(fea, FUNCATTR_FLAGS)
-		if flags & ida_funcs.FUNC_LIB:
+		flags = idc.get_func_attr(fea, idc.FUNCATTR_FLAGS)
+		if flags & idaapi.FUNC_LIB:
 			continue
 
 		idc.set_name(fea, "")
 
-if __name__ == "__main__":
-	main()
+main()