ejschwartz commited on
Commit
39ad796
·
1 Parent(s): 5cd1f1d
Files changed (1) hide show
  1. main.py +3 -2
main.py CHANGED
@@ -11,14 +11,15 @@ def predict(decompiled_code):
11
  with tempfile.TemporaryDirectory() as temp_dir:
12
 
13
  code_file_name = os.path.join(temp_dir, "code.decompiled")
 
14
 
15
  with open(code_file_name, "w") as f:
16
  f.write('#include "/home/ReSym/clang-parser/defs.hh"\n' + decompiled_code)
17
 
18
  # subprocess.check_output(["python", "/home/ReSym/process_data/prep_decompiled.py", code_file_name, file_save_dir, parsed_save_dir])
19
- output = subprocess.check_output(["/home/ReSym/clang-parser/build/field_access", code_file_name])
20
 
21
- return output
22
 
23
 
24
  def run():
 
11
  with tempfile.TemporaryDirectory() as temp_dir:
12
 
13
  code_file_name = os.path.join(temp_dir, "code.decompiled")
14
+ field_file_name = os.path.join(temp_dir, "code.fields")
15
 
16
  with open(code_file_name, "w") as f:
17
  f.write('#include "/home/ReSym/clang-parser/defs.hh"\n' + decompiled_code)
18
 
19
  # subprocess.check_output(["python", "/home/ReSym/process_data/prep_decompiled.py", code_file_name, file_save_dir, parsed_save_dir])
20
+ output = subprocess.run(["/home/ReSym/clang-parser/build/field_access", code_file_name, field_file_name])
21
 
22
+ return output.stdout.decode("utf-8")
23
 
24
 
25
  def run():