jak-disassembler/LinkedObjectFile.h at master · himham-jak/jak-disassembler · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
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 "Function/Function.h"
#include "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