RagePhoto: G5E2P and G5E3P parser added
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
Syping 2020-11-08 18:05:44 +01:00
parent a798be0b53
commit b70396f77e
2 changed files with 304 additions and 154 deletions

View file

@ -23,18 +23,24 @@
#include <QDebug>
#include <QFile>
RagePhoto::RagePhoto(const QString &filePath) : p_filePath(filePath)
RagePhoto::RagePhoto(const QByteArray &data) : p_fileData(data)
{
p_inputMode = 0;
p_isLoaded = false;
}
RagePhoto::RagePhoto(QIODevice *ioDevice) : p_ioDevice(ioDevice)
RagePhoto::RagePhoto(const QString &filePath) : p_filePath(filePath)
{
p_inputMode = 1;
p_isLoaded = false;
}
RagePhoto::RagePhoto(QIODevice *ioDevice) : p_ioDevice(ioDevice)
{
p_inputMode = 2;
p_isLoaded = false;
}
bool RagePhoto::isLoaded()
{
return p_isLoaded;
@ -45,36 +51,31 @@ bool RagePhoto::load()
if (p_isLoaded)
clear();
QByteArray readData;
if (p_inputMode == 0) {
if (p_inputMode == 1) {
QFile pictureFile(p_filePath);
if (pictureFile.open(QIODevice::ReadOnly)) {
readData = pictureFile.readAll();
p_fileData = pictureFile.readAll();
}
pictureFile.close();
}
else if (p_inputMode == 1) {
else if (p_inputMode == 2) {
if (!p_ioDevice->isOpen()) {
if (!p_ioDevice->open(QIODevice::ReadOnly))
return false;
readData = p_ioDevice->readAll();
}
}
else {
return false;
p_fileData = p_ioDevice->readAll();
}
QBuffer dataBuffer(&readData);
QBuffer dataBuffer(&p_fileData);
dataBuffer.open(QIODevice::ReadOnly);
char formatHeader[4];
qint64 size = dataBuffer.read(formatHeader, 4);
if (size != 4)
return false;
quint32 format = charToUInt32(formatHeader);
if (format != PhotoFormat::GTA5)
return false;
quint32 format = charToUInt32LE(formatHeader);
if (format == PhotoFormat::GTA5) {
char photoHeader[256];
size = dataBuffer.read(photoHeader, 256);
if (size != 256)
@ -89,31 +90,31 @@ bool RagePhoto::load()
size = dataBuffer.read(checksum, 4);
if (size != 4)
return false;
p_headerCRC = charToUInt32(checksum);
p_headerSum = charToUInt32LE(checksum);
char endOfFile[4];
size = dataBuffer.read(endOfFile, 4);
if (size != 4)
return false;
p_endOfFile = charToUInt32(endOfFile);
p_endOfFile = charToUInt32LE(endOfFile);
char jsonOffset[4];
size = dataBuffer.read(jsonOffset, 4);
if (size != 4)
return false;
p_jsonOffset = charToUInt32(jsonOffset);
p_jsonOffset = charToUInt32LE(jsonOffset);
char titleOffset[4];
size = dataBuffer.read(titleOffset, 4);
if (size != 4)
return false;
p_titlOffset = charToUInt32(titleOffset);
p_titlOffset = charToUInt32LE(titleOffset);
char descOffset[4];
size = dataBuffer.read(descOffset, 4);
if (size != 4)
return false;
p_descOffset = charToUInt32(descOffset);
p_descOffset = charToUInt32LE(descOffset);
char jpegMarker[4];
size = dataBuffer.read(jpegMarker, 4);
@ -126,13 +127,13 @@ bool RagePhoto::load()
size = dataBuffer.read(jpegBuffer, 4);
if (size != 4)
return false;
p_jpegBuffer = charToUInt32(jpegBuffer);
p_jpegBuffer = charToUInt32LE(jpegBuffer);
char photoSize[4];
size = dataBuffer.read(photoSize, 4);
if (size != 4)
return false;
p_photoSize = charToUInt32(photoSize);
p_photoSize = charToUInt32LE(photoSize);
char photoData[p_photoSize];
size = dataBuffer.read(photoData, p_photoSize);
@ -152,7 +153,7 @@ bool RagePhoto::load()
size = dataBuffer.read(jsonSize, 4);
if (size != 4)
return false;
quint32 i_jsonSize = charToUInt32(jsonSize);
quint32 i_jsonSize = charToUInt32LE(jsonSize);
char jsonBytes[i_jsonSize];
size = dataBuffer.read(jsonBytes, i_jsonSize);
@ -181,7 +182,7 @@ bool RagePhoto::load()
size = dataBuffer.read(titlSize, 4);
if (size != 4)
return false;
quint32 i_titlSize = charToUInt32(titlSize);
quint32 i_titlSize = charToUInt32LE(titlSize);
char titlBytes[i_titlSize];
size = dataBuffer.read(titlBytes, i_titlSize);
@ -205,7 +206,7 @@ bool RagePhoto::load()
size = dataBuffer.read(descSize, 4);
if (size != 4)
return false;
quint32 i_descSize = charToUInt32(descSize);
quint32 i_descSize = charToUInt32LE(descSize);
char descBytes[i_descSize];
size = dataBuffer.read(descBytes, i_descSize);
@ -225,9 +226,143 @@ bool RagePhoto::load()
if (strncmp(jendMarker, "JEND", 4) != 0)
return false;
p_fileData.clear();
p_isLoaded = true;
return true;
}
else if (format == PhotoFormat::G5EX) {
char formatHeader[4];
qint64 size = dataBuffer.read(formatHeader, 4);
if (size != 4)
return false;
quint32 format = charToUInt32LE(formatHeader);
if (format == ExportFormat::G5E3P) {
char photoHeaderSize[4];
size = dataBuffer.peek(photoHeaderSize, 4);
if (size != 4)
return false;
quint32 i_photoHeaderSize = charToUInt32BE(photoHeaderSize) + 4;
char compressedPhotoHeader[i_photoHeaderSize];
size = dataBuffer.read(compressedPhotoHeader, i_photoHeaderSize);
if (size != i_photoHeaderSize)
return false;
QByteArray t_photoHeaderBytes = QByteArray::fromRawData(compressedPhotoHeader, i_photoHeaderSize);
t_photoHeaderBytes = qUncompress(t_photoHeaderBytes);
p_photoString = QString::fromUtf8(t_photoHeaderBytes);
char checksum[4];
size = dataBuffer.read(checksum, 4);
if (size != 4)
return false;
p_headerSum = charToUInt32LE(checksum);
char jpegBuffer[4];
size = dataBuffer.read(jpegBuffer, 4);
if (size != 4)
return false;
p_jpegBuffer = charToUInt32LE(jpegBuffer);
char photoSize[4];
size = dataBuffer.peek(photoSize, 4);
if (size != 4)
return false;
quint32 i_photoSize = charToUInt32BE(photoSize) + 4;
char compressedPhoto[i_photoSize];
size = dataBuffer.read(compressedPhoto, i_photoSize);
if (size != i_photoSize)
return false;
QByteArray t_photoData = QByteArray::fromRawData(compressedPhoto, i_photoSize);
p_photoData = qUncompress(t_photoData);
p_photoSize = p_photoData.size();
char jsonOffset[4];
size = dataBuffer.read(jsonOffset, 4);
if (size != 4)
return false;
p_jsonOffset = charToUInt32LE(jsonOffset);
char jsonSize[4];
size = dataBuffer.peek(jsonSize, 4);
if (size != 4)
return false;
quint32 i_jsonSize = charToUInt32BE(jsonSize) + 4;
char compressedJson[i_jsonSize];
size = dataBuffer.read(compressedJson, i_jsonSize);
if (size != i_jsonSize)
return false;
QByteArray t_jsonBytes = QByteArray::fromRawData(compressedJson, i_jsonSize);
t_jsonBytes = qUncompress(t_jsonBytes);
QJsonDocument t_jsonDocument = QJsonDocument::fromJson(t_jsonBytes);
if (t_jsonDocument.isNull())
return false;
p_jsonObject = t_jsonDocument.object();
char titleOffset[4];
size = dataBuffer.read(titleOffset, 4);
if (size != 4)
return false;
p_titlOffset = charToUInt32LE(titleOffset);
char titlSize[4];
size = dataBuffer.peek(titlSize, 4);
if (size != 4)
return false;
quint32 i_titlSize = charToUInt32BE(titlSize) + 4;
char compressedTitl[i_titlSize];
size = dataBuffer.read(compressedTitl, i_titlSize);
if (size != i_titlSize)
return false;
QByteArray t_titlBytes = QByteArray::fromRawData(compressedTitl, i_titlSize);
t_titlBytes = qUncompress(t_titlBytes);
p_titleString = QString::fromUtf8(t_titlBytes);
char descOffset[4];
size = dataBuffer.read(descOffset, 4);
if (size != 4)
return false;
p_descOffset = charToUInt32LE(descOffset);
char descSize[4];
size = dataBuffer.peek(descSize, 4);
if (size != 4)
return false;
quint32 i_descSize = charToUInt32BE(descSize) + 4;
char compressedDesc[i_descSize];
size = dataBuffer.read(compressedDesc, i_descSize);
if (size != i_descSize)
return false;
QByteArray t_descBytes = QByteArray::fromRawData(compressedDesc, i_descSize);
t_descBytes = qUncompress(t_descBytes);
p_descriptionString = QString::fromUtf8(t_descBytes);
char endOfFile[4];
size = dataBuffer.read(endOfFile, 4);
if (size != 4)
return false;
p_endOfFile = charToUInt32LE(endOfFile);
p_fileData.clear();
p_isLoaded = true;
return true;
}
else if (format == ExportFormat::G5E2P) {
p_fileData = dataBuffer.readAll();
p_inputMode = 0;
return load();
}
else {
return false;
}
}
else {
return false;
}
}
void RagePhoto::clear()
{
@ -236,7 +371,7 @@ void RagePhoto::clear()
p_photoData.clear();
p_photoString.clear();
p_titleString.clear();
p_headerCRC = 0;
p_headerSum = 0;
p_isLoaded = false;
}
@ -293,7 +428,12 @@ RagePhoto* RagePhoto::loadFile(const QString &filePath)
return ragePhoto;
}
quint32 RagePhoto::charToUInt32(char *x)
quint32 RagePhoto::charToUInt32BE(char *x)
{
return (((unsigned char)x[0] << 24) | ((unsigned char)x[1] << 16) | ((unsigned char)x[2] << 8) | ((unsigned char)x[3]));
}
quint32 RagePhoto::charToUInt32LE(char *x)
{
return (((unsigned char)x[3] << 24) | ((unsigned char)x[2] << 16) | ((unsigned char)x[1] << 8) | ((unsigned char)x[0]));
}

View file

@ -27,10 +27,18 @@ class RagePhoto : public QObject
{
Q_OBJECT
public:
enum ExportFormat {
G5E2P = 0x01000032U,
G5E2S = 0x02000032U,
G5E3P = 0x01000033U,
G5E3S = 0x02000033U,
};
enum PhotoFormat {
G5EX = 0x45354700U,
GTA5 = 0x01000000U,
RDR2 = 0x04000000U,
};
explicit RagePhoto(const QByteArray &data);
explicit RagePhoto(const QString &filePath = QString());
explicit RagePhoto(QIODevice *device);
bool isLoaded();
@ -48,8 +56,10 @@ public:
static RagePhoto* loadFile(const QString &filePath);
private:
inline quint32 charToUInt32(char *x);
inline quint32 charToUInt32BE(char *x);
inline quint32 charToUInt32LE(char *x);
QJsonObject p_jsonObject;
QByteArray p_fileData;
QByteArray p_photoData;
QIODevice *p_ioDevice;
QString p_descriptionString;
@ -58,7 +68,7 @@ private:
QString p_titleString;
quint32 p_descOffset;
quint32 p_endOfFile;
quint32 p_headerCRC;
quint32 p_headerSum;
quint32 p_jpegBuffer;
quint32 p_jsonOffset;
quint32 p_photoSize;