fix: file closing

Signed-off-by: Sarapulov Vasilii <[email protected]>
VyacheslavIurevich · Jul 11, 2024 · 92ba761 · 92ba761
1 parent 81033d8
commit 92ba761
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 49 deletions.
diff --git a/src/code_receiving.py b/src/code_receiving.py
diff --git a/src/write_program_data_types b/src/write_program_data_types
diff --git a/src/write_program_data_types.py b/src/write_program_data_types.py
@@ -0,0 +1,14 @@
+'''Dumping program data types'''
+#pylint: disable=wrong-import-position
+import pyhidra # pylint: disable=import-error
+pyhidra.start()
+import ghidra # pylint: disable=import-error
+from java.io import PrintWriter # pylint: disable=import-error
+
+def write_program_data_types(program, file, monitor):
+    """Dumping program data types"""
+    dtm = program.getDataTypeManager()
+    c_file_writer = PrintWriter(file)
+    data_type_writer = ghidra.program.model.data.DataTypeWriter(dtm, c_file_writer, False)
+    data_type_writer.write(dtm, monitor)
+    c_file_writer.close()