mirror of
https://github.com/wiiu-env/wudd.git
synced 2024-11-11 04:45:07 +01:00
112 lines
4.1 KiB
C++
112 lines
4.1 KiB
C++
/****************************************************************************
|
|
* Copyright (C) 2021 Maschell
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that 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/>.
|
|
****************************************************************************/
|
|
#include "WUXFileWriter.h"
|
|
#include "WUDDumperState.h"
|
|
#include "utils/StringTools.h"
|
|
#include <utils/logger.h>
|
|
|
|
WUXFileWriter::WUXFileWriter(const char *path, int32_t cacheSize, int32_t sectorSize, bool split) : WUDFileWriter(path, cacheSize, sectorSize, split) {
|
|
wuxHeader_t wuxHeader = {0};
|
|
wuxHeader.magic0 = WUX_MAGIC_0;
|
|
wuxHeader.magic1 = WUX_MAGIC_1;
|
|
wuxHeader.sectorSize = swap_uint32(this->sectorSize);
|
|
wuxHeader.uncompressedSize = swap_uint64(WUD_FILE_SIZE);
|
|
wuxHeader.flags = 0;
|
|
|
|
this->write((uint8_t *) &wuxHeader, sizeof(wuxHeader_t));
|
|
this->sectorTableStart = this->tell();
|
|
this->totalSectorCount = WUD_FILE_SIZE / this->sectorSize;
|
|
|
|
this->sectorIndexTable = (void *) memalign(0x40, ROUNDUP(totalSectorCount * 4, 0x40));
|
|
if (sectorIndexTable == nullptr) {
|
|
DEBUG_FUNCTION_LINE_ERR("Failed to alloc");
|
|
WUDFileWriter::close();
|
|
return;
|
|
}
|
|
memset(this->sectorIndexTable, 0, totalSectorCount * 4);
|
|
|
|
if (this->write((uint8_t *) this->sectorIndexTable, totalSectorCount * 4) < 0) {
|
|
WUDFileWriter::close();
|
|
return;
|
|
}
|
|
|
|
this->sectorTableEnd = this->tell();
|
|
uint64_t tableEnd = this->sectorTableEnd;
|
|
|
|
this->sectorTableEnd += this->sectorSize - 1;
|
|
this->sectorTableEnd -= (this->sectorTableEnd % this->sectorSize);
|
|
|
|
uint64_t padding = this->sectorTableEnd - tableEnd;
|
|
auto *paddingData = (uint8_t *) memalign(0x40, ROUNDUP(padding, 0x40));
|
|
memset(paddingData, 0, padding);
|
|
this->write(reinterpret_cast<const uint8_t *>(paddingData), padding);
|
|
free(paddingData);
|
|
this->hashMap.clear();
|
|
flush();
|
|
}
|
|
|
|
|
|
int32_t WUXFileWriter::writeSector(const uint8_t *buffer, uint32_t numberOfSectors) {
|
|
int32_t curWrittenSectors = 0;
|
|
for (uint32_t i = 0; i < numberOfSectors; i++) {
|
|
uint32_t addr = ((uint32_t) buffer) + (i * this->sectorSize);
|
|
std::array<uint8_t, 32> hashOut{};
|
|
calculateHash256(reinterpret_cast<unsigned char *>(addr), this->sectorSize, reinterpret_cast<unsigned char *>(hashOut.data()));
|
|
|
|
auto *indexTable = (uint32_t *) this->sectorIndexTable;
|
|
|
|
auto it = hashMap.find(hashOut);
|
|
if (it != hashMap.end()) {
|
|
indexTable[this->currentSector] = swap_uint32(this->hashMap[hashOut]);
|
|
} else {
|
|
indexTable[this->currentSector] = swap_uint32(this->writtenSector);
|
|
hashMap[hashOut] = writtenSector;
|
|
if (isOpen()) {
|
|
if (!write((uint8_t *) addr, this->sectorSize)) {
|
|
DEBUG_FUNCTION_LINE_ERR("Write failed");
|
|
return -1;
|
|
}
|
|
}
|
|
this->writtenSector++;
|
|
curWrittenSectors++;
|
|
}
|
|
this->currentSector++;
|
|
}
|
|
return curWrittenSectors;
|
|
}
|
|
|
|
void WUXFileWriter::writeSectorIndexTable() {
|
|
if (this->isOpen()) {
|
|
flush();
|
|
// We need to make sure to call CFile::seek!
|
|
seek((int64_t) sectorTableStart, SEEK_SET_BASE_CLASS);
|
|
write((uint8_t *) sectorIndexTable, totalSectorCount * 4);
|
|
flush();
|
|
}
|
|
}
|
|
|
|
WUXFileWriter::~WUXFileWriter() {
|
|
WUXFileWriter::close();
|
|
free(sectorIndexTable);
|
|
}
|
|
|
|
void WUXFileWriter::finalize() {
|
|
WUDFileWriter::finalize();
|
|
writeSectorIndexTable();
|
|
WUXFileWriter::close();
|
|
}
|