mirror of
https://github.com/open-goal/jak-project.git
synced 2024-10-20 21:27:52 -04:00
135 lines
4.5 KiB
C++
135 lines
4.5 KiB
C++
/*!
|
|
* @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
|