add decompiler

2026-08-14 04:33:04 -04:00 · 2020-08-22 23:30:17 -04:00
parent ba3c3af43e
commit c3aff47886
52 changed files with 10200 additions and 5 deletions
@@ -0,0 +1,131 @@
+/*!
+ * @file LinkedObjectFile.h
+ * An object file's data with linking information included.
+ */
+
+#ifndef NEXT_LINKEDOBJECTFILE_H
+#define NEXT_LINKEDOBJECTFILE_H
+
+#include <cstdint>
+#include <vector>
+#include <string>
+#include <unordered_map>
+#include <unordered_set>
+#include "LinkedWord.h"
+#include "decompiler/Function/Function.h"
+#include "decompiler/util/LispPrint.h"
+
+
+/*!
+ * A label to a location in this object file.
+ * Doesn't have to be word aligned.
+ */
+struct Label {
+  std::string name;
+  int target_segment;
+  int offset; // in bytes
+};
+
+/*!
+ * An object file's data with linking information included.
+ */
+class LinkedObjectFile {
+public:
+  LinkedObjectFile() = default;
+  void set_segment_count(int n_segs);
+  void push_back_word_to_segment(uint32_t word, int segment);
+  int get_label_id_for(int seg, int offset);
+  int get_label_at(int seg, int offset) const;
+  bool label_points_to_code(int label_id) const;
+  bool pointer_link_word(int source_segment, int source_offset, int dest_segment, int dest_offset);
+  void pointer_link_split_word(int source_segment, int source_hi_offset, int source_lo_offset, int dest_segment, int dest_offset);
+  void symbol_link_word(int source_segment, int source_offset, const char* name, LinkedWord::Kind kind);
+  void symbol_link_offset(int source_segment, int source_offset, const char* name);
+  Function& get_function_at_label(int label_id);
+  std::string get_label_name(int label_id) const;
+  uint32_t set_ordered_label_names();
+  void find_code();
+  std::string print_words();
+  void find_functions();
+  void disassemble_functions();
+  void process_fp_relative_links();
+  std::string print_scripts();
+  std::string print_disassembly();
+  bool has_any_functions();
+  void append_word_to_string(std::string& dest, const LinkedWord& word) const;
+
+  struct Stats {
+    uint32_t total_code_bytes = 0;
+    uint32_t total_v2_code_bytes = 0;
+    uint32_t total_v2_pointers = 0;
+    uint32_t total_v2_pointer_seeks = 0;
+    uint32_t total_v2_link_bytes = 0;
+    uint32_t total_v2_symbol_links = 0;
+    uint32_t total_v2_symbol_count = 0;
+
+    uint32_t v3_code_bytes = 0;
+    uint32_t v3_pointers = 0;
+    uint32_t v3_split_pointers = 0;
+    uint32_t v3_word_pointers = 0;
+    uint32_t v3_pointer_seeks = 0;
+    uint32_t v3_link_bytes = 0;
+
+    uint32_t v3_symbol_count = 0;
+    uint32_t v3_symbol_link_offset = 0;
+    uint32_t v3_symbol_link_word = 0;
+
+    uint32_t data_bytes = 0;
+    uint32_t code_bytes = 0;
+
+    uint32_t function_count = 0;
+    uint32_t decoded_ops = 0;
+
+    uint32_t n_fp_reg_use = 0;
+    uint32_t n_fp_reg_use_resolved = 0;
+
+
+    void add(const Stats& other) {
+      total_code_bytes += other.total_code_bytes;
+      total_v2_code_bytes += other.total_v2_code_bytes;
+      total_v2_pointers += other.total_v2_pointers;
+      total_v2_pointer_seeks += other.total_v2_pointer_seeks;
+      total_v2_link_bytes += other.total_v2_link_bytes;
+      total_v2_symbol_links += other.total_v2_symbol_links;
+      total_v2_symbol_count += other.total_v2_symbol_count;
+      v3_code_bytes += other.v3_code_bytes;
+      v3_pointers += other.v3_pointers;
+      v3_pointer_seeks += other.v3_pointer_seeks;
+      v3_link_bytes += other.v3_link_bytes;
+      v3_word_pointers += other.v3_word_pointers;
+      v3_split_pointers += other.v3_split_pointers;
+      v3_symbol_count += other.v3_symbol_count;
+      v3_symbol_link_offset += other.v3_symbol_link_offset;
+      v3_symbol_link_word += other.v3_symbol_link_word;
+      data_bytes += other.data_bytes;
+      code_bytes += other.code_bytes;
+      function_count += other.function_count;
+      decoded_ops += other.decoded_ops;
+      n_fp_reg_use += other.n_fp_reg_use;
+      n_fp_reg_use_resolved += other.n_fp_reg_use_resolved;
+    }
+  } stats;
+
+  int segments = 0;
+  std::vector<std::vector<LinkedWord>> words_by_seg;
+  std::vector<uint32_t> offset_of_data_zone_by_seg;
+  std::vector<std::vector<Function>> functions_by_seg;
+  std::vector<Label> labels;
+
+private:
+  std::shared_ptr<Form> to_form_script(int seg, int word_idx, std::vector<bool>& seen);
+  std::shared_ptr<Form> to_form_script_object(int seg, int byte_idx, std::vector<bool> &seen);
+  bool is_empty_list(int seg, int byte_idx);
+  bool is_string(int seg, int byte_idx);
+  std::string get_goal_string(int seg, int word_idx);
+
+  std::vector<std::unordered_map<int, int>> label_per_seg_by_offset;
+};
+
+
+
+#endif //NEXT_LINKEDOBJECTFILE_H