Export pickles to python2

ash09 · Jul 30, 2020 · b3715e9 · b3715e9
1 parent 8001621
commit b3715e9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/BB-weight-angr.py b/BB-weight-angr.py
@@ -200,7 +200,7 @@ def dump_analysis(filename):
 		a,b = fweight[addr]
 		fweight[addr] = (1.0/a,b)
 	with open("%s.pkl"%filename,"wb") as f:
-		pickle.dump(fweight,f)
+		pickle.dump(fweight,f,protocol=2)
 	# dump cmp operands analysis
 	cmp_imm_operands_hex = []
 	cmp_imm_operands2 = list(cmp_imm_operands)
@@ -210,7 +210,7 @@ def dump_analysis(filename):
 			cmp_imm_operands.remove(op)
 	cmp_imm_operands_hex = set().union(cmp_imm_operands_hex)
 	with open("%s.names" % filename,'wb') as f:
-		pickle.dump((cmp_imm_operands,cmp_imm_operands_hex),f)
+		pickle.dump((cmp_imm_operands,cmp_imm_operands_hex),f,protocol=2)
 
 def find_CMP_operands(proj,cfg,binName):
 	global cmp_imm_operands