import source files, forgetting about 938 prior commits

scripts/testbed/_dump_fnames.py

@@ -0,0 +1,46 @@
+'''
+IDAPython script to dump JSON file of functions names { fva: fname }.
+Meant to be run on benign files with PDB information. IDA should apply function names from the PDB files automatically.
+Can also be run on annotated IDA database files.
+
+Example usage (via IDA autonomous mode):
+  ida.exe -A -S_dump_fnames.py "<output path>" <sample_path>
+'''
+
+import json
+
+import idc
+import idautils
+
+
+def main():
+    if len(idc.ARGV) != 2:
+        # requires output file path argument
+        idc.qexit(-1)
+
+    # wait for auto-analysis to finish
+    idc.auto_wait()
+
+    INF_SHORT_DN_ATTR = idc.get_inf_attr(idc.INF_SHORT_DN)  # short form of demangled names
+
+    fnames = {}
+    for f in idautils.Functions():
+        fname = idc.get_name(f)
+        if fname.startswith("sub_"):
+            continue
+
+        name_demangled = idc.demangle_name(fname, INF_SHORT_DN_ATTR)
+        if name_demangled:
+            fname = name_demangled
+
+        fnames[f] = fname
+
+    with open(idc.ARGV[1], "w") as f:
+        json.dump(fnames, f)
+
+    # exit IDA
+    idc.qexit(0)
+
+
+if __name__ == "__main__":
+    main()