#include "ApplicationState.h"
#include "utils/WiiUScreen.h"
#include "utils/ScreenUtils.h"
#include "common/common.h"
#include "utils/utils.h"
#include "utils/StringTools.h"
#include "utils/rijndael.h"
#include "fs/FSUtils.h"
#include <sysapp/launch.h>
#include <coreinit/ios.h>
#include <iosuhax.h>
#include <malloc.h>
#include <coreinit/memdefaultheap.h>
#include <cstdarg>
#include <ntfs.h>
#include <WUD/DiscReaderDiscDrive.h>
#include <whb/proc.h>
#include <coreinit/debug.h>
#include <WUD/content/partitions/WiiUGMPartition.h>
#include <WUD/header/WiiUDiscHeader.h>
#include <WUD/entities/TMD/Content.h>
#include <WUD/entities/TMD/TitleMetaData.h>
#include <WUD/NUSDataProviderWUD.h>
#include <WUD/NUSTitle.h>

extern ntfs_md *ntfs_mounts;
extern int ntfs_mount_count;


unsigned int swap_uint32(unsigned int val) {
    val = ((val << 8) & 0xFF00FF00) | ((val >> 8) & 0xFF00FF);
    return (val << 16) | (val >> 16);
}

unsigned long long swap_uint64(unsigned long long val) {
    val = ((val << 8) & 0xFF00FF00FF00FF00ULL) | ((val >> 8) & 0x00FF00FF00FF00FFULL);
    val = ((val << 16) & 0xFFFF0000FFFF0000ULL) | ((val >> 16) & 0x0000FFFF0000FFFFULL);
    return (val << 32) | (val >> 32);
}

/*
 * Hash function used to create a hash of each sector
 * The hashes are then compared to find duplicate sectors
 */
void calculateHash256(unsigned char *data, unsigned int length, unsigned char *hashOut) {
    // cheap and simple hash implementation
    // you can replace this part with your favorite hash method
    memset(hashOut, 0x00, 32);
    for (unsigned int i = 0; i < length; i++) {
        hashOut[i % 32] ^= data[i];
        hashOut[(i + 7) % 32] += data[i];
    }
}

ApplicationState::ApplicationState() : log("fs:/vol/external01/wudump.log", CFile::WriteOnly) {
    this->log.fwrite("Started wudump\n");
    this->state = STATE_WELCOME_SCREEN;
}

void ApplicationState::printHeader() {
    WiiUScreen::drawLine("Wudump");
    WiiUScreen::drawLine("==================");
    WiiUScreen::drawLine("");
}

void ApplicationState::render() {
    WiiUScreen::clearScreen();

    if (this->state == STATE_ERROR) {
        WiiUScreen::drawLine();
        WiiUScreen::drawLinef("Error:       %s", ErrorMessage().c_str());
        WiiUScreen::drawLinef("Description: %s", ErrorDescription().c_str());
        WiiUScreen::drawLine();
        WiiUScreen::drawLine("Press A to return to the Wii U Menu.");
    } else if (this->state == STATE_WELCOME_SCREEN) {
        WiiUScreen::drawLine("Welcome to Wudump");
        WiiUScreen::drawLine("Press A to dump the currently inserted Disc");
        WiiUScreen::drawLine("");
        if (this->selectedOption == 0) {
            WiiUScreen::drawLine("> Dump as WUX       Dump as WUD       Dump as .app       Exit");
        } else if (this->selectedOption == 1) {
            WiiUScreen::drawLine("  Dump as WUX     > Dump as WUD       Dump as .app       Exit");
        } else if (this->selectedOption == 2) {
            WiiUScreen::drawLine("  Dump as WUX       Dump as WUD     > Dump as .app       Exit");
        } else if (this->selectedOption == 3) {
            WiiUScreen::drawLine("  Dump as WUX       Dump as WUD       Dump as .app     > Exit");
        }
    } else if (this->state == STATE_CHOOSE_TARGET) {
        printHeader();
        WiiUScreen::drawLine("Please choose your target:");
        std::vector<std::string> options;
        int32_t targetCount = 0;
        if (this->dumpFormat == DUMP_AS_APP) {
            options.emplace_back("SD");
            targetCount++;
        }
        for (int i = 0; i < ntfs_mount_count; i++) {
            options.emplace_back(ntfs_mounts[i].name);
            targetCount++;
        }

        std::string curLine = "";
        if (this->selectedOption == 0) {
            curLine = "> Back\t";
        } else {
            curLine = "  Back\t";
        }

        if (targetCount == 0) {
            WiiUScreen::drawLine("Please insert a NTFS formatted USB drive and restart wudump\n");
        } else {
            for (int32_t i = 0; i < targetCount; i++) {
                if (this->selectedOption - 1 == i) {
                    curLine += "> " + options[i];
                } else {
                    curLine += "  " + options[i];
                }
                curLine += "\t";
            }
        }
        WiiUScreen::drawLine(curLine.c_str());
    } else if (this->state == STATE_OPEN_ODD1) {
        WiiUScreen::drawLine("Open /dev/odd01");
    } else if (this->state == STATE_READ_DISC_INFO) {
        WiiUScreen::drawLine("Read disc information");
    } else if (this->state == STATE_READ_DISC_INFO_DONE) {
        WiiUScreen::drawLinef("Dumping: %s", this->discId);
    } else if (this->state == STATE_DUMP_TICKET) {
        WiiUScreen::drawLinef("Dumping game.key");
    } else if (this->state == STATE_DUMP_DISC_START || this->state == STATE_DUMP_DISC || this->state == STATE_WAIT_USER_ERROR_CONFIRM) {
        WiiUScreen::drawLinef("Dumping: %s", this->discId);

        float percent = this->currentSector / (WUD_FILE_SIZE / READ_SECTOR_SIZE * 1.0f) * 100.0f;
        WiiUScreen::drawLinef("Progress: %0.2f MiB / %5.2f MiB (%2.1f %%)", this->currentSector * (READ_SECTOR_SIZE / 1024.0f / 1024.0f), WUD_FILE_SIZE / 1024.0f / 1024.0f, percent);
        if (doWUX) {
            WiiUScreen::drawLinef("Written %0.2f MiB. Compression ratio 1:%0.2f", this->hashMap.size() * (READ_SECTOR_SIZE / 1024.0f / 1024.0f),
                                  1.0f / (this->hashMap.size() / (float) this->currentSector));
        }

        if (this->readResult < 0 || this->oddFd < 0) {
            WiiUScreen::drawLine();

            if (this->oddFd < 0) {
                WiiUScreen::drawLine("Failed to open disc, try again.");
            } else {
                WiiUScreen::drawLinef("Error: Failed to read sector - Error %d", this->readResult);
            }
            WiiUScreen::drawLine();
            WiiUScreen::drawLine("Press A to skip this sector (will be replaced by 0's)");
            WiiUScreen::drawLine("Press B to try again");
        } else {
            OSTime curTime = OSGetTime();
            float remaining = (WUD_FILE_SIZE - (READ_SECTOR_SIZE * this->currentSector)) / 1024.0f / 1024.0f;
            float curSpeed = READ_SECTOR_SIZE * ((this->readSectors / 1000.0f) / OSTicksToMilliseconds(curTime - startTime));
            int32_t remainingSec = remaining / curSpeed;
            int32_t minutes = (remainingSec / 60) % 60;
            int32_t seconds = remainingSec % 60;
            int32_t hours = remainingSec / 3600;

            WiiUScreen::drawLinef("Speed: %.2f MiB/s ETA: %02dh %02dm %02ds", curSpeed, remaining, hours, minutes, seconds);
        }

        WiiUScreen::drawLine();
        if (!this->skippedSectors.empty()) {
            WiiUScreen::drawLinef("Skipped dumping %d sectors", this->skippedSectors.size());
        }
    } else if (this->state == STATE_DUMP_DISC_DONE) {
        if (!flushWriteCache()) {
            setError(ERROR_WRITE_FAILED);
        }
        WiiUScreen::drawLinef("Dumping done! Press A to continue");
    } else if (this->state == STATE_DUMP_APP_FILES_DONE) {
        WiiUScreen::drawLinef("Dumping done! Press A to continue");
    }
    printFooter();
    WiiUScreen::flipBuffers();
}

void ApplicationState::update(Input *input) {
    if (this->state == STATE_ERROR) {
        OSEnableHomeButtonMenu(true);
        if (entrySelected(input)) {
            SYSLaunchMenu();
        }
    } else if (this->state == STATE_WELCOME_SCREEN) {
        proccessMenuNavigation(input, 4);
        if (entrySelected(input)) {
            if (this->selectedOption == 0) {
                this->retryCount = 10;
                this->state = STATE_CHOOSE_TARGET;
                this->dumpFormat = DUMP_AS_WUX;
            } else if (this->selectedOption == 1) {
                this->retryCount = 10;
                this->state = STATE_CHOOSE_TARGET;
                this->dumpFormat = DUMP_AS_WUD;
            } else if (this->selectedOption == 2) {
                this->retryCount = 10;
                this->state = STATE_CHOOSE_TARGET;
                this->dumpFormat = DUMP_AS_APP;
            } else {
                SYSLaunchMenu();
            }
            this->selectedOption = 0;
            return;
        }
    } else if (this->state == STATE_CHOOSE_TARGET) {
        WiiUScreen::drawLine("Please choose your target");
        std::vector<std::string> options;
        uint32_t targetCount = 0;

        if (this->dumpFormat == DUMP_AS_APP) {
            options.emplace_back("fs:/vol/external01/");
            targetCount++;
        }
        if (ntfs_mount_count > 0) {

            for (int i = 0; i < ntfs_mount_count; i++) {
                options.emplace_back(std::string(ntfs_mounts[i].name) + ":/");
                targetCount++;
            }
        }

        proccessMenuNavigation(input, targetCount + 1);
        if (entrySelected(input)) {
            if (this->selectedOption == 0) {
                this->state = STATE_WELCOME_SCREEN;
            } else if (targetCount > 0) {
                target = options[selectedOption - 1];
                this->state = STATE_OPEN_ODD1;
            }
            this->selectedOption = 0;
        }
    } else if (this->state == STATE_OPEN_ODD1) {
        if (this->readSectors > 0) {
            auto ret = IOSUHAX_FSA_RawOpen(gFSAfd, "/dev/odd01", &(this->oddFd));
            if (ret >= 0) {
                // continue!
                this->state = STATE_DUMP_DISC;
            } else {
                this->oddFd = -1;
                this->state = STATE_WAIT_USER_ERROR_CONFIRM;
            }
            return;
        }
        DEBUG_FUNCTION_LINE("STATE_OPEN_ODD1");
        if (this->retryCount-- <= 0) {
            this->setError(ERROR_OPEN_ODD1);
            return;
        }
        auto ret = IOSUHAX_FSA_RawOpen(gFSAfd, "/dev/odd01", &(this->oddFd));
        if (ret >= 0) {
            if (this->sectorBuf == nullptr) {
                this->sectorBuf = (void *) memalign(0x100, this->sectorBufSize);
                if (this->sectorBuf == nullptr) {
                    this->setError(ERROR_MALLOC_FAILED);
                    return;
                }
            }
            DEBUG_FUNCTION_LINE("Opened /dev/odd01 %d", this->oddFd);
            this->state = STATE_READ_DISC_INFO;
        }
    } else if (this->state == STATE_READ_DISC_INFO) {
        DEBUG_FUNCTION_LINE("STATE_READ_DISC_INFO");
        if (IOSUHAX_FSA_RawRead(gFSAfd, this->sectorBuf, READ_SECTOR_SIZE, 1, 0, this->oddFd) >= 0) {
            this->discId[10] = '\0';
            memcpy(this->discId, sectorBuf, 10);
            if (this->discId[0] == 0) {
                setError(ERROR_NO_DISC_ID);
                return;
            }
            this->state = STATE_READ_DISC_INFO_DONE;
            return;
        }

        this->setError(ERROR_READ_FIRST_SECTOR);
        return;
    } else if (this->state == STATE_READ_DISC_INFO_DONE) {
        DEBUG_FUNCTION_LINE("STATE_READ_DISC_INFO_DONE");
        this->state = STATE_DUMP_TICKET;
    } else if (this->state == STATE_DUMP_TICKET) {
        DEBUG_FUNCTION_LINE("STATE_DUMP_TICKET");

        auto res = IOSUHAX_FSA_RawRead(gFSAfd, this->sectorBuf, READ_SECTOR_SIZE, 1, 3, this->oddFd);
        uint8_t discKey[16];
        bool hasDiscKey = false;
        if (res >= 0) {
            if (((uint32_t *) this->sectorBuf)[0] != 0xCCA6E67B) {
                uint8_t iv[16];
                memset(iv, 0, 16);

                auto odm_handle = IOS_Open("/dev/odm", IOS_OPEN_READ);
                if (odm_handle >= 0) {
                    uint32_t io_buffer[0x20 / 4];
                    // disc encryption key, only works with patched IOSU
                    io_buffer[0] = 3;
                    if (IOS_Ioctl(odm_handle, 0x06, io_buffer, 0x14, io_buffer, 0x20) == 0) {
                        memcpy(discKey, io_buffer, 16);
                        hasDiscKey = true;
                    }
                    IOS_Close(odm_handle);
                }
            }
        }

        if (hasDiscKey) {
            if (!FSUtils::CreateSubfolder(StringTools::fmt("%swudump/%s", target.c_str(), discId))) {
                setError(ERROR_WRITE_FAILED);
                return;
            }
            if (!FSUtils::saveBufferToFile(StringTools::fmt("%swudump/%s/game.key", target.c_str(), discId), discKey, 16)) {
                setError(ERROR_WRITE_FAILED);
                return;
            }
        }
        if (this->dumpFormat == DUMP_AS_WUX || this->dumpFormat == DUMP_AS_WUD) {
            if (this->dumpFormat == DUMP_AS_WUX) {
                this->doWUX = true;
            }
            this->state = STATE_DUMP_DISC_START;
        } else {
            this->state = STATE_DUMP_APP_FILES;
        }
    } else if (this->state == STATE_DUMP_APP_FILES) {
        ApplicationState::dumpAppFiles();
        if (this->state != STATE_ERROR) {
            this->state = STATE_DUMP_APP_FILES_DONE;
        }
    } else if (this->state == STATE_DUMP_APP_FILES_DONE) {
        if (entrySelected(input)) {
            this->state = STATE_WELCOME_SCREEN;
            if (this->oddFd >= 0) {
                IOSUHAX_FSA_RawClose(gFSAfd, this->oddFd);
                this->oddFd = -1;
            }
            this->currentSector = 0;
            this->readSectors = 0;
        }
    } else if (this->state == STATE_DUMP_DISC_START) {
        ApplicationState::clearWriteCache();
        DEBUG_FUNCTION_LINE("STATE_DUMP_DISC_START");
        if (!FSUtils::CreateSubfolder(StringTools::fmt("%swudump/%s", target.c_str(), discId))) {
            setError(ERROR_WRITE_FAILED);
            return;
        }
        this->fileHandle = new CFile(StringTools::fmt("%swudump/%s/game.%s", target.c_str(), discId, doWUX ? "wux" : "wud"), CFile::WriteOnly);

        this->totalSectorCount = WUD_FILE_SIZE / SECTOR_SIZE;

        if (!this->fileHandle->isOpen()) {
            DEBUG_FUNCTION_LINE("Failed to open file");
            this->setError(ERROR_FILE_OPEN_FAILED);
            return;
        }

        if (doWUX) {
            wuxHeader_t wuxHeader = {0};
            wuxHeader.magic0 = WUX_MAGIC_0;
            wuxHeader.magic1 = WUX_MAGIC_1;
            wuxHeader.sectorSize = swap_uint32(SECTOR_SIZE);
            wuxHeader.uncompressedSize = swap_uint64(WUD_FILE_SIZE);
            wuxHeader.flags = 0;

            DEBUG_FUNCTION_LINE("Write header");
            this->fileHandle->write((uint8_t *) &wuxHeader, sizeof(wuxHeader_t));
            this->sectorTableStart = this->fileHandle->tell();

            this->sectorIndexTable = (void *) malloc(totalSectorCount * 4);
            if (sectorIndexTable == nullptr) {
                this->setError(ERROR_MALLOC_FAILED);
                return;
            }
            memset(this->sectorIndexTable, 0, totalSectorCount * 4);

            DEBUG_FUNCTION_LINE("Write empty sectorIndexTable");
            this->fileHandle->write((uint8_t *) this->sectorIndexTable, totalSectorCount * 4);

            DEBUG_FUNCTION_LINE("Get sector table end");
            this->sectorTableEnd = this->fileHandle->tell();
            uint64_t tableEnd = this->sectorTableEnd;

            this->sectorTableEnd += SECTOR_SIZE - 1;
            this->sectorTableEnd -= (this->sectorTableEnd % SECTOR_SIZE);

            uint64_t padding = this->sectorTableEnd - tableEnd;
            auto *paddingData = (uint8_t *) malloc(padding);
            memset(paddingData, 0, padding);
            this->fileHandle->write(reinterpret_cast<const uint8_t *>(paddingData), padding);
            free(paddingData);
            this->hashMap.clear();
        }

        this->writeBufferSize = READ_SECTOR_SIZE * WRITE_BUFFER_NUM_SECTORS;
        this->writeBuffer = (void *) memalign(0x1000, this->writeBufferSize);
        if (this->writeBuffer == nullptr) {
            this->setError(ERROR_MALLOC_FAILED);
            return;
        }
        this->writeBufferPos = 0;

        this->startTime = OSGetTime();

        this->state = STATE_DUMP_DISC;
        this->currentSector = 0;
        this->retryCount = 10;
        this->selectedOption = 0;
        this->readSectors = 0;
    } else if (this->state == STATE_DUMP_DISC) {
        //DEBUG_FUNCTION_LINE("STATE_DUMP_DISC");
        int32_t numSectors = this->currentSector + READ_NUM_SECTORS > this->totalSectorCount ? this->totalSectorCount - this->currentSector : READ_NUM_SECTORS;
        if ((this->readResult = IOSUHAX_FSA_RawRead(gFSAfd, sectorBuf, READ_SECTOR_SIZE, numSectors, this->currentSector, this->oddFd)) >= 0) {
            if (!writeDataToFile(this->sectorBuf, numSectors)) {
                this->setError(ERROR_WRITE_FAILED);
                return;
            }
            //DEBUG_FUNCTION_LINE("Read done %lld %lld", this->currentSector, this->totalSectorCount);
            this->retryCount = 10;
            if (this->currentSector >= this->totalSectorCount) {
                this->state = STATE_DUMP_DISC_DONE;

                if (this->fileHandle->isOpen()) {
                    if (!this->flushWriteCache()) {
                        this->setError(ERROR_WRITE_FAILED);
                        return;
                    }
                    if (doWUX) {
                        this->writeSectorIndexTable();
                    }
                    this->fileHandle->close();
                }
            }
        } else {
            this->state = STATE_WAIT_USER_ERROR_CONFIRM;
            if (this->oddFd >= 0) {
                IOSUHAX_FSA_RawClose(gFSAfd, this->oddFd);
                this->oddFd = -1;
            }
            return;
        }
    } else if (this->state == STATE_WAIT_USER_ERROR_CONFIRM) {
        if (this->autoSkip) {
            if (this->oddFd >= 0) {
                IOSUHAX_FSA_RawClose(gFSAfd, this->oddFd);
                this->oddFd = -1;
            }
        }
        if (this->autoSkip || (input->data.buttons_d & Input::BUTTON_A)) {
            this->log.fwrite("Skipped sector %d : 0x%ll016X-0x%ll016X, filled with 0's\n", this->currentSector, this->currentSector * READ_SECTOR_SIZE, (this->currentSector + 1) * READ_SECTOR_SIZE);
            this->state = STATE_OPEN_ODD1;
            this->skippedSectors.push_back(this->currentSector);
            // We can't use seek because we may have cached values.

            if (this->emptySector == nullptr) {
                this->emptySector = memalign(0x100, READ_SECTOR_SIZE);
                if (this->emptySector == nullptr) {
                    this->setError(ERROR_MALLOC_FAILED);
                    return;
                }
            }

            if (!this->writeCached(reinterpret_cast<uint32_t>(emptySector), READ_SECTOR_SIZE)) {
                this->setError(ERROR_WRITE_FAILED);
                return;
            }

            this->currentSector += 1;
            this->readResult = 0;
        } else if (input->data.buttons_d & Input::BUTTON_B) {
            this->state = STATE_OPEN_ODD1;
            this->readResult = 0;
        } else if (input->data.buttons_d & Input::BUTTON_Y) {
            this->autoSkip = true;
        }
    } else if (this->state == STATE_DUMP_DISC_DONE) {
        if (entrySelected(input)) {
            if (this->oddFd >= 0) {
                IOSUHAX_FSA_RawClose(gFSAfd, this->oddFd);
                this->oddFd = -1;
            }
            this->state = STATE_WELCOME_SCREEN;
            this->selectedOption = 0;
            this->currentSector = 0;
            this->readSectors = 0;
            this->writtenSector = 0;
            return;
        }
    }
}

std::string ApplicationState::ErrorMessage() {
    if (this->error == ERROR_NONE) {
        return "NONE";
    } else if (this->error == ERROR_IOSUHAX_FAILED) {
        return "ERROR_IOSUHAX_FAILED";
    } else if (this->error == ERROR_MALLOC_FAILED) {
        return "ERROR_MALLOC_FAILED";
    } else if (this->error == ERROR_FILE_OPEN_FAILED) {
        return "ERROR_FILE_OPEN_FAILED";
    } else if (this->error == ERROR_NO_DISC_ID) {
        return "ERROR_NO_DISC_ID";
    }
    DEBUG_FUNCTION_LINE("Error: %d", this->error);
    return "UNKNOWN_ERROR";
}

std::string ApplicationState::ErrorDescription() {
    if (this->error == ERROR_NONE) {
        return "-";
    } else if (this->error == ERROR_IOSUHAX_FAILED) {
        return "Failed to init IOSUHAX.";
    } else if (this->error == ERROR_MALLOC_FAILED) {
        return "Failed to allocate data.";
    } else if (this->error == ERROR_FILE_OPEN_FAILED) {
        return "Failed to create file";
    } else if (this->error == ERROR_NO_DISC_ID) {
        return "Failed to get the disc id";
    }
    DEBUG_FUNCTION_LINE("Error: %d", this->error);
    return "UNKNOWN_ERROR";
}

void ApplicationState::setError(eErrorState err) {
    this->state = STATE_ERROR;
    this->error = err;
    //OSEnableHomeButtonMenu(true);
}

void ApplicationState::printFooter() {
    ScreenUtils::printTextOnScreen(CONSOLE_SCREEN_TV, 0, 27, "By Maschell");
    ScreenUtils::printTextOnScreen(CONSOLE_SCREEN_DRC, 0, 17, "By Maschell");
}

void ApplicationState::proccessMenuNavigation(Input *input, int maxOptionValue) {
    if (input->data.buttons_d & Input::BUTTON_LEFT) {
        this->selectedOption--;
    } else if (input->data.buttons_d & Input::BUTTON_RIGHT) {
        this->selectedOption++;
    }
    if (this->selectedOption < 0) {
        this->selectedOption = maxOptionValue;
    } else if (this->selectedOption >= maxOptionValue) {
        this->selectedOption = 0;
    }
}

bool ApplicationState::entrySelected(Input *input) {
    return input->data.buttons_d & Input::BUTTON_A;
}

ApplicationState::~ApplicationState() {
    this->log.close();
    if (this->fileHandle->isOpen()) {
        if (!this->flushWriteCache()) {

        }
        if (doWUX) {
            this->writeSectorIndexTable();
        }
        this->fileHandle->close();
    }
    if (this->emptySector != nullptr) {
        free(this->emptySector);
        this->emptySector = nullptr;
    }
    if (this->writeBuffer != nullptr) {
        free(this->writeBuffer);
        this->writeBuffer = nullptr;
    }
    if (this->sectorIndexTable != nullptr) {
        free(this->sectorIndexTable);
        this->sectorIndexTable = nullptr;
    }
    if (this->sectorBuf != nullptr) {
        free(this->sectorBuf);
        this->sectorBuf = nullptr;
    }
    if (this->oddFd >= 0) {
        IOSUHAX_FSA_RawClose(gFSAfd, this->oddFd);
        this->oddFd = -1;
    }
}

bool ApplicationState::writeDataToFile(void *buffer, int numberOfSectors) {
    if (!doWUX) {
        if (!writeCached(reinterpret_cast<uint32_t>(buffer), numberOfSectors * READ_SECTOR_SIZE)) {
            return false;
        }
        this->currentSector += numberOfSectors;
        this->readSectors += numberOfSectors;
    } else {
        char hashOut[32];
        for (int i = 0; i < numberOfSectors; i++) {
            uint32_t addr = ((uint32_t) buffer) + (i * READ_SECTOR_SIZE);
            calculateHash256(reinterpret_cast<unsigned char *>(addr), READ_SECTOR_SIZE, reinterpret_cast<unsigned char *>(hashOut));
            char tmp[34];
            auto *test = (uint32_t *) hashOut;
            snprintf(tmp, 33, "%08X%08X%08X%08X", test[0], test[1], test[2], test[3]);
            std::string hash(tmp);

            uint32_t *indexTable = (uint32_t *) this->sectorIndexTable;

            auto it = hashMap.find(hash);
            if (it != hashMap.end()) {
                indexTable[this->currentSector] = swap_uint32(this->hashMap[hash]);
            } else {
                indexTable[this->currentSector] = swap_uint32(this->writtenSector);
                hashMap[hash] = this->writtenSector;
                if (this->fileHandle->isOpen()) {
                    if (!writeCached(addr, READ_SECTOR_SIZE)) {
                        return false;
                    }
                }
                this->writtenSector++;
            }
            this->currentSector++;
            this->readSectors++;
        }
    }
    return true;
}

bool ApplicationState::writeCached(uint32_t addr, uint32_t writeSize) {
    // DEBUG_FUNCTION_LINE("Lest write %d bytes", writeSize);

    if (writeSize == this->writeBufferSize) {
        if (!this->flushWriteCache()) {
            return false;
        }
        int32_t res = this->fileHandle->write(reinterpret_cast<const uint8_t *>(addr), writeSize);
        return res >= 0;
    }

    uint32_t toWrite = writeSize;
    if (toWrite == 0) {
        return true;
    }

    uint32_t written = 0;

    do {
        uint32_t curWrite = toWrite;

        if (this->writeBufferPos + curWrite > this->writeBufferSize) {
            curWrite = this->writeBufferSize - this->writeBufferPos;
        }
        // DEBUG_FUNCTION_LINE("Copy from %08X into %08X, size %08X, %d",(addr + written),((uint32_t) this->writeBuffer) + this->writeBufferPos, curWrite, this->writeBufferPos/READ_SECTOR_SIZE);
        OSBlockMove((void *) (((uint32_t) this->writeBuffer) + this->writeBufferPos), (void *) (addr + written), curWrite, 1);
        this->writeBufferPos += curWrite;

        if (this->writeBufferPos == this->writeBufferSize) {
            if (!flushWriteCache()) {
                return false;
            }
        }

        toWrite -= curWrite;
        written += curWrite;
    } while (toWrite > 0);
    return true;
}

bool ApplicationState::flushWriteCache() {
    if (this->writeBufferPos > 0) {
        int32_t res = this->fileHandle->write(static_cast<const uint8_t *>(this->writeBuffer), this->writeBufferPos);
        if (res < 0) {
            return false;
        }
        this->writeBufferPos = 0;
    }
    return true;
}

void ApplicationState::clearWriteCache() {
    this->writeBufferPos = 0;
}

void ApplicationState::writeSectorIndexTable() {
    if (this->fileHandle->isOpen() && doWUX) {
        this->fileHandle->seek(this->sectorTableStart, SEEK_SET);
        this->fileHandle->write((uint8_t *) this->sectorIndexTable, totalSectorCount * 4);
    }
}

void ApplicationState::printDumpState(const char *fmt, ...) {
    WiiUScreen::clearScreen();
    ApplicationState::printHeader();
    char *buf = (char *) MEMAllocFromDefaultHeapEx(PRINTF_BUFFER_LENGTH, 4);
    va_list va;

    if (!buf) {
        return;
    }

    va_start(va, fmt);
    vsnprintf(buf, PRINTF_BUFFER_LENGTH, fmt, va);

    WiiUScreen::drawLine(buf);

    MEMFreeToDefaultHeap(buf);
    va_end(va);
    ApplicationState::printFooter();
    WiiUScreen::flipBuffers();
}

void ApplicationState::dumpAppFiles() {
    uint8_t opt[0x400];
    IOSUHAX_read_otp(opt, 0x400);
    uint8_t cKey[0x10];
    memcpy(cKey, opt + 0xE0, 0x10);

    DEBUG_FUNCTION_LINE("Reading Partitions");

    printDumpState("Reading Partitions...");

    auto discReader = new DiscReaderDiscDrive();
    if (!discReader->IsReady()) {
        DEBUG_FUNCTION_LINE("!IsReady");
        this->setError(ERROR_OPEN_ODD1);
        delete discReader;
        return;
    }
    DEBUG_FUNCTION_LINE("Read DiscHeader");
    auto *discHeader = new WiiUDiscHeader(discReader, 0);
    bool forceExit = false;
    for (auto &partition: discHeader->wiiUContentsInformation->partitions->partitions) {
        auto gmPartition = dynamic_cast<WiiUGMPartition *>(partition);
        if (gmPartition != nullptr) {
            auto *nusTitle = NUSTitle::loadTitleFromGMPartition(gmPartition, discReader, cKey);
            if (nusTitle == nullptr) {
                DEBUG_FUNCTION_LINE("nusTitle was null");
                continue;
            }
            auto *dataProvider = nusTitle->dataProcessor->getDataProvider();

            uint64_t partitionSize = 0;
            uint64_t partitionSizeWritten = 0;
            for (auto &content: nusTitle->tmd->contentList) {
                partitionSize += ROUNDUP(content->encryptedFileSize, 16);
            }

            auto partitionDumpInfo = StringTools::strfmt("Partition: %s\n\tProgress: %.2f MiB / %.2f MiB\n", partition->getVolumeId().c_str(), partitionSizeWritten / 1024.0f / 1024.0f,
                                                         partitionSize / 1024.0f / 1024.0f);
            printDumpState("%s", partitionDumpInfo.c_str());

            char buffer[512];
            snprintf(buffer, 500, "%swudump/%s/%s", target.c_str(), this->discId, gmPartition->getVolumeId().c_str());
            FSUtils::CreateSubfolder(buffer);

            uint8_t *wBuffer = nullptr;
            uint32_t wBufferLen = 0;
            if (dataProvider->getRawTMD(&wBuffer, &wBufferLen)) {
                std::string fileName = std::string(buffer).append("/").append(WUD_TMD_FILENAME);
                printDumpState("%s\nSaving %s", partitionDumpInfo.c_str(), WUD_TMD_FILENAME);
                FSUtils::saveBufferToFile(fileName.c_str(), wBuffer, wBufferLen);
                free(wBuffer);
            }

            if (dataProvider->getRawTicket(&wBuffer, &wBufferLen)) {
                std::string fileName = std::string(buffer).append("/").append(WUD_TICKET_FILENAME);
                printDumpState("%s\nSaving %s", partitionDumpInfo.c_str(), WUD_TICKET_FILENAME);
                FSUtils::saveBufferToFile(fileName.c_str(), wBuffer, wBufferLen);
                free(wBuffer);
            }
            if (dataProvider->getRawCert(&wBuffer, &wBufferLen)) {
                std::string fileName = std::string(buffer).append("/").append(WUD_TICKET_FILENAME);
                printDumpState("%s\nSaving %s", partitionDumpInfo.c_str(), WUD_CERT_FILENAME);
                FSUtils::saveBufferToFile(fileName.c_str(), wBuffer, wBufferLen);
                free(wBuffer);
            }

            auto contentCount = nusTitle->tmd->contentList.size();
            auto contentI = 1;

            for (auto &content: nusTitle->tmd->contentList) {
                char bufApp[32];
                snprintf(bufApp, 31, "%08X.app", content->ID);
                std::string appFileName = std::string(buffer) + "/" + bufApp;

                partitionDumpInfo = StringTools::strfmt("Partition: %s\n\tProgress: %.2f MiB / %.2f MiB\n", partition->getVolumeId().c_str(), partitionSizeWritten / 1024.0f / 1024.0f,
                                                        partitionSize / 1024.0f / 1024.0f);
                auto contentDumpInfo = StringTools::strfmt("Saving %s (Content %02d/%02d)\n", bufApp, contentI, contentCount);

                printDumpState("%s\n%s", partitionDumpInfo.c_str(), contentDumpInfo.c_str());

                uint32_t bufferSize = READ_NUM_SECTORS * READ_SECTOR_SIZE * 2;

                auto *readBuffer = (uint8_t *) malloc(bufferSize);
                if (readBuffer == nullptr) {
                    DEBUG_FUNCTION_LINE("Failed to alloc buffer");
                    continue;
                }

                CFile file(appFileName, CFile::WriteOnly);
                if (!file.isOpen()) {
                    free(readBuffer);
                    continue;
                }
                uint32_t readSoFar = 0;
                uint64_t curOffset = 0;
                uint32_t size = ROUNDUP(content->encryptedFileSize, 16);
                OSTime startTimeApp = OSGetTime();
                do {
                    if (!WHBProcIsRunning()) {
                        forceExit = true;
                        break;
                    }
                    startTimeApp = OSGetTime();
                    WiiUScreen::clearScreen();

                    uint32_t toRead = size - readSoFar;
                    if (toRead > bufferSize) {
                        toRead = bufferSize;
                    }
                    dataProvider->readRawContent(content, readBuffer, curOffset, toRead);
                    if (file.write((const uint8_t *) readBuffer, toRead) != (int32_t) toRead) {
                        DEBUG_FUNCTION_LINE("Failed to write");
                        break;
                    }

                    OSTime curTime = OSGetTime();
                    auto curSpeed = (float) toRead / (float) OSTicksToMilliseconds(curTime - startTimeApp);

                    readSoFar += toRead;
                    curOffset += toRead;

                    partitionSizeWritten += toRead;

                    partitionDumpInfo = StringTools::strfmt("Partition: %s\n\tProgress: %.2f MiB / %.2f MiB\n", partition->getVolumeId().c_str(), partitionSizeWritten / 1024.0f / 1024.0f,
                                                            partitionSize / 1024.0f / 1024.0f);
                    printDumpState("%s\n%s\tProgress: %.2f MiB / %.2f MiB (%0.2f%%)\n\tSpeed: %0.2f MiB/s", partitionDumpInfo.c_str(), contentDumpInfo.c_str(), readSoFar / 1024.0f / 1024.0f,
                                   size / 1024.0f / 1024.0f, ((readSoFar * 1.0f) / size) * 100.0f, curSpeed / 1024.0f);
                } while (readSoFar < size);

                file.close();

                if (forceExit) {
                    break;
                }

                uint8_t *h3Data = nullptr;
                uint32_t h3Length = 0;
                if (dataProvider->getContentH3Hash(content, &h3Data, &h3Length)) {
                    char bufh3[32];
                    snprintf(bufh3, 31, "%08X.h3", content->ID);
                    std::string h3FileName = std::string(buffer) + "/" + bufh3;
                    printDumpState("%s\n%s", partitionDumpInfo.c_str(), contentDumpInfo.c_str());
                    FSUtils::saveBufferToFile(h3FileName.c_str(), h3Data, h3Length);
                }
                contentI++;
            }

            delete nusTitle;

            if (forceExit) {
                exit(0);
                break;
            }
        }
    }
    delete discHeader;
    delete discReader;
}