shadPS4/src/core/linker.h

#pragma once

#include <vector>
#include <mutex>
#include "core/loader/elf.h"
#include "core/loader/symbols_resolver.h"

namespace Core {

struct DynamicModuleInfo;
class Linker;

struct EntryParams {
    int argc;
    u32 padding;
    const char* argv[3];
};

struct ModuleInfo {
	std::string name;
    union {
		u64 value;
        struct {
			u32 name_offset;
			u08 version_minor;
			u08 version_major;
			u16 id;
		};
	};
	std::string enc_id;
};

struct LibraryInfo {
	std::string name;
    union {
		u64 value;
        struct {
			u32 name_offset;
			u16 version;
			u16 id;
		};
	};
	std::string enc_id;
};

struct DynamicModuleInfo {
    void* hash_table = nullptr;
    u64 hash_table_size = 0;

    char* str_table = nullptr;
    u64 str_table_size = 0;

    elf_symbol* symbol_table = nullptr;
    u64 symbol_table_total_size = 0;
    u64 symbol_table_entries_size = 0;

    u64 init_virtual_addr = 0;
    u64 fini_virtual_addr = 0;
    u64 pltgot_virtual_addr = 0;
    u64 init_array_virtual_addr = 0;
    u64 fini_array_virtual_addr = 0;
    u64 preinit_array_virtual_addr = 0;
    u64 init_array_size = 0;
    u64 fini_array_size = 0;
    u64 preinit_array_size = 0;

    elf_relocation* jmp_relocation_table = nullptr;
    u64 jmp_relocation_table_size = 0;
    s64 jmp_relocation_type = 0;

    elf_relocation* relocation_table = nullptr;
    u64 relocation_table_size = 0;
    u64 relocation_table_entries_size = 0;

    u64 debug = 0;
    u64 textrel = 0;
    u64 flags = 0;

    std::vector<const char*> needed;
    std::vector<ModuleInfo> import_modules;
    std::vector<ModuleInfo> export_modules;
    std::vector<LibraryInfo> import_libs;
    std::vector<LibraryInfo> export_libs;

    std::string filename; // Filename with absolute path
};

// This struct keeps neccesary info about loaded modules. Main executeable is included too as well
struct Module {
    Loader::Elf elf;
    u64 aligned_base_size = 0;
    u64 base_virtual_addr = 0; // Base virtual address

    Linker* linker = nullptr;

    std::vector<u08> m_dynamic;
    std::vector<u08> m_dynamic_data;
    DynamicModuleInfo dynamic_info{};

    Loader::SymbolsResolver export_sym;
    Loader::SymbolsResolver import_sym;
};

class Linker {
public:
	Linker();
	virtual ~Linker();

	Module* LoadModule(const std::string& elf_name);
	Module* FindModule(/*u32 id*/);
	void LoadModuleToMemory(Module* m);
	void LoadDynamicInfo(Module* m);
	void LoadSymbols(Module* m);
    Loader::SymbolsResolver& getHLESymbols() { return m_hle_symbols; }
	void Relocate(Module* m);
    void Resolve(const std::string& name, int Symtype, Module* m, Loader::SymbolRecord* return_info);
    void Execute();

private:
	const ModuleInfo* FindModule(const Module& m, const std::string& id);
	const LibraryInfo* FindLibrary(const Module& program, const std::string& id);

    std::vector<Module> m_modules;
    Loader::SymbolsResolver m_hle_symbols{};
    std::mutex m_mutex;
};

} // namespace Core