DonLakeFlyer
4 years ago
committed by
Lorenz Meier
13 changed files with 240 additions and 115 deletions
@ -0,0 +1,97 @@ |
|||||||
|
/****************************************************************************
|
||||||
|
* |
||||||
|
* (c) 2009-2020 QGROUNDCONTROL PROJECT <http://www.qgroundcontrol.org>
|
||||||
|
* |
||||||
|
* QGroundControl is licensed according to the terms in the file |
||||||
|
* COPYING.md in the root of the source code directory. |
||||||
|
* |
||||||
|
****************************************************************************/ |
||||||
|
|
||||||
|
#include "QGCLZMA.h" |
||||||
|
|
||||||
|
#include <QFile> |
||||||
|
#include <QDir> |
||||||
|
#include <QtDebug> |
||||||
|
|
||||||
|
#include "lzma.h" |
||||||
|
|
||||||
|
bool QGCLZMA::inflateLZMAFile(const QString& lzmaFilename, const QString& decompressedFilename) |
||||||
|
{ |
||||||
|
bool success = true; |
||||||
|
const int cBuffer = 1024 * 5; |
||||||
|
unsigned char inputBuffer[cBuffer]; |
||||||
|
unsigned char outputBuffer[cBuffer]; |
||||||
|
lzma_stream lzmaStrm = LZMA_STREAM_INIT; |
||||||
|
|
||||||
|
QFile inputFile(lzmaFilename); |
||||||
|
if (!inputFile.open(QIODevice::ReadOnly)) { |
||||||
|
qWarning() << "QGCLZMA::inflateLZMAFile: open input file failed" << lzmaFilename << inputFile.errorString(); |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
QFile outputFile(decompressedFilename); |
||||||
|
if (!outputFile.open(QIODevice::WriteOnly | QIODevice::Truncate)) { |
||||||
|
qWarning() << "QGCLZMA::inflateLZMAFile: open input file failed" << outputFile.fileName() << outputFile.errorString(); |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
// UINT64_MAX - used as much memory as needed to decode
|
||||||
|
int lzmaRetCode = lzma_alone_decoder(&lzmaStrm, UINT64_MAX); |
||||||
|
if (lzmaRetCode != LZMA_OK) { |
||||||
|
qWarning() << "QGCLZMA::inflateLZMAFile: lzma_alone_decoder failed:" << lzmaRetCode; |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
// When LZMA_CONCATENATED flag was used when initializing the decoder,
|
||||||
|
// we need to tell lzma_code() when there will be no more input.
|
||||||
|
// This is done by setting action to LZMA_FINISH instead of LZMA_RUN
|
||||||
|
// in the same way as it is done when encoding.
|
||||||
|
//
|
||||||
|
// When LZMA_CONCATENATED isn't used, there is no need to use
|
||||||
|
// LZMA_FINISH to tell when all the input has been read, but it
|
||||||
|
// is still OK to use it if you want. When LZMA_CONCATENATED isn't
|
||||||
|
// used, the decoder will stop after the first .xz stream. In that
|
||||||
|
// case some unused data may be left in lzmaStrm.next_in.
|
||||||
|
lzma_action action = LZMA_RUN; |
||||||
|
|
||||||
|
lzmaStrm.next_in = nullptr; |
||||||
|
lzmaStrm.avail_in = 0; |
||||||
|
lzmaStrm.next_out = nullptr; |
||||||
|
lzmaStrm.avail_out = cBuffer; |
||||||
|
|
||||||
|
do { |
||||||
|
lzmaStrm.avail_in = static_cast<unsigned>(inputFile.read((char*)inputBuffer, cBuffer)); |
||||||
|
if (lzmaStrm.avail_in == 0) { |
||||||
|
break; |
||||||
|
} |
||||||
|
lzmaStrm.next_in = inputBuffer; |
||||||
|
|
||||||
|
do { |
||||||
|
lzmaStrm.avail_out = cBuffer; |
||||||
|
lzmaStrm.next_out = outputBuffer; |
||||||
|
|
||||||
|
lzmaRetCode = lzma_code(&lzmaStrm, action); |
||||||
|
if (lzmaRetCode != LZMA_OK && lzmaRetCode != LZMA_STREAM_END) { |
||||||
|
qWarning() << "QGCLZMA::inflateLZMAFile: inflate failed:" << lzmaRetCode; |
||||||
|
goto Error; |
||||||
|
} |
||||||
|
|
||||||
|
unsigned cBytesInflated = cBuffer - lzmaStrm.avail_out; |
||||||
|
qint64 cBytesWritten = outputFile.write((char*)outputBuffer, static_cast<int>(cBytesInflated)); |
||||||
|
if (cBytesWritten != cBytesInflated) { |
||||||
|
qWarning() << "QGCLZMA::inflateLZMAFile: output file write failed:" << outputFile.fileName() << outputFile.errorString(); |
||||||
|
goto Error; |
||||||
|
|
||||||
|
} |
||||||
|
} while (lzmaStrm.avail_out == 0); |
||||||
|
} while (lzmaRetCode != LZMA_STREAM_END); |
||||||
|
|
||||||
|
Out: |
||||||
|
lzma_end(&lzmaStrm); |
||||||
|
|
||||||
|
return success; |
||||||
|
|
||||||
|
Error: |
||||||
|
success = false; |
||||||
|
goto Out; |
||||||
|
} |
||||||
@ -0,0 +1,21 @@ |
|||||||
|
/****************************************************************************
|
||||||
|
* |
||||||
|
* (c) 2009-2020 QGROUNDCONTROL PROJECT <http://www.qgroundcontrol.org>
|
||||||
|
* |
||||||
|
* QGroundControl is licensed according to the terms in the file |
||||||
|
* COPYING.md in the root of the source code directory. |
||||||
|
* |
||||||
|
****************************************************************************/ |
||||||
|
|
||||||
|
#pragma once |
||||||
|
|
||||||
|
#include <QString> |
||||||
|
|
||||||
|
class QGCLZMA |
||||||
|
{ |
||||||
|
public: |
||||||
|
/// Decompresses the specified file to the specified directory
|
||||||
|
/// @param lzmaFilename Fully qualified path to lzma file
|
||||||
|
/// @param decompressedFilename Fully qualified path to for file to decompress to
|
||||||
|
static bool inflateLZMAFile(const QString& lzmaFilename, const QString& decompressedFilename); |
||||||
|
}; |
||||||
Binary file not shown.
Binary file not shown.
Loading…
Reference in new issue