/*
 * Copyright (c) 2018-2019 Atmosphère-NX
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms and conditions of the GNU General Public License,
 * version 2, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#pragma once
#include <switch.h>
#include <stratosphere.hpp>
#include "creport_threads.hpp"

namespace sts::creport {

    class ModuleList {
        private:
            static constexpr size_t ModuleCountMax      = 0x60;
            static constexpr size_t ModuleNameLengthMax = 0x20;
            static constexpr size_t ModuleBuildIdLength = 0x20;

            struct ModuleInfo {
                char name[ModuleNameLengthMax];
                u8   build_id[ModuleBuildIdLength];
                u64  start_address;
                u64  end_address;
            };
        private:
            Handle debug_handle;
            size_t num_modules;
            ModuleInfo modules[ModuleCountMax];

            /* For pretty-printing. */
            char address_str_buf[0x280];
        public:
            ModuleList() : debug_handle(INVALID_HANDLE), num_modules(0) {
                std::memset(this->modules, 0, sizeof(this->modules));
            }

            size_t GetModuleCount() const {
                return this->num_modules;
            }

            u64 GetModuleStartAddress(size_t i) const {
                return this->modules[i].start_address;
            }

            void FindModulesFromThreadInfo(Handle debug_handle, const ThreadInfo &thread);
            const char *GetFormattedAddressString(uintptr_t address);
            void SaveToFile(FILE *f_report);
        private:
            bool TryFindModule(uintptr_t *out_address, uintptr_t guess);
            void TryAddModule(uintptr_t guess);
            void GetModuleName(char *out_name, uintptr_t text_start, uintptr_t ro_start);
            void GetModuleBuildId(u8 *out_build_id, uintptr_t ro_start);
    };

}