cmclark00/retro-imager
1
0
Fork 0
mirror of https://github.com/cmclark00/retro-imager.git synced 2025-05-18 16:05:21 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Local .xz files: parse uncompressed size for better progress reports

Browse source
This commit is contained in:
Floris Bos 2023-09-25 21:13:22 +02:00
parent 269e43ced5
commit e8931b7b8f
4 changed files with 47 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

44
src/imagewriter.cpp
View file

@ -14,6 +14,7 @@
#include "wlancredentials.h"
#include <archive.h>
#include <archive_entry.h>
#include <lzma.h>
#include <random>
#include <QFileInfo>
#include <QQmlApplicationEngine>
@ -242,6 +243,8 @@ void ImageWriter::startWrite()
_extrLen = _downloadLen;
else if (lowercaseurl.endsWith(".zip"))
_parseCompressedFile();
else if (lowercaseurl.endsWith(".xz"))
_parseXZFile();
}
if (_devLen && _extrLen > _devLen)
@ -648,6 +651,47 @@ void ImageWriter::_parseCompressedFile()
qDebug() << "Parsed .zip file containing" << numFiles << "files, uncompressed size:" << _extrLen;
}
void ImageWriter::_parseXZFile()
{
QFile f(_src.toLocalFile());
lzma_stream_flags opts = { 0 };
_extrLen = 0;
if (f.size() > LZMA_STREAM_HEADER_SIZE && f.open(f.ReadOnly))
{
f.seek(f.size()-LZMA_STREAM_HEADER_SIZE);
QByteArray footer = f.read(LZMA_STREAM_HEADER_SIZE);
lzma_ret ret = lzma_stream_footer_decode(&opts, (const uint8_t *) footer.constData());
if (ret == LZMA_OK && opts.backward_size < 1000000 && opts.backward_size < f.size()-LZMA_STREAM_HEADER_SIZE)
{
f.seek(f.size()-LZMA_STREAM_HEADER_SIZE-opts.backward_size);
QByteArray buf = f.read(opts.backward_size+LZMA_STREAM_HEADER_SIZE);
lzma_index *idx;
uint64_t memlimit = UINT64_MAX;
size_t pos = 0;
ret = lzma_index_buffer_decode(&idx, &memlimit, NULL, (const uint8_t *) buf.constData(), &pos, buf.size());
if (ret == LZMA_OK)
{
_extrLen = lzma_index_uncompressed_size(idx);
qDebug() << "Parsed .xz file. Uncompressed size:" << _extrLen;
}
else
{
qDebug() << "Unable to parse index of .xz file";
}
lzma_index_end(idx, NULL);
}
else
{
qDebug() << "Unable to parse footer of .xz file";
}
f.close();
}
}
bool ImageWriter::isOnline()
{
return _online || !_embeddedMode;