python2->python3 fix

panda-re · Aug 12, 2024 · 5f2cf80 · 5f2cf80
1 parent 65cb91b
commit 5f2cf80
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/scripts/fninstr.py b/scripts/fninstr.py
@@ -1,7 +1,7 @@
 import sys
 import yaml
 import re
-import cPickle as pickle
+import pickle
 import argparse
 
 debug = True
@@ -126,14 +126,14 @@ def merge(v, vors):
                 fpa = FnPtrAssign(x['fnPtrAssign'])
                 addtohl(fpas, fpa.name, fpa)
 
-    f = open("getfns.pickle", "w")
+    f = open("getfns.pickle", "wb")
     pickle.dump(fundefs, f)
     pickle.dump(prots, f)
     pickle.dump(calls, f)
     pickle.dump(fpas, f)
     f.close()
 else:
-    f = open("getfns.pickle", "r")
+    f = open("getfns.pickle", "rb")
     fundefs = pickle.load(f)
     prots = pickle.load(f)
     calls = pickle.load(f)