chore: initial commit (extracted from Launchers monorepo)
Plugin: ns7zip v2.0.0 Architectures: x86-ansi, x86-unicode, amd64-unicode License: LGPL-2.1-or-later
This commit is contained in:
Simone
@@ -0,0 +1,347 @@
|
||||
// ExtractCallback.h
|
||||
|
||||
#ifndef ZIP7_INC_EXTRACT_CALLBACK_H
|
||||
#define ZIP7_INC_EXTRACT_CALLBACK_H
|
||||
|
||||
#include "../../../../C/Alloc.h"
|
||||
|
||||
#include "../../../Common/MyCom.h"
|
||||
#include "../../../Common/StringConvert.h"
|
||||
|
||||
#ifndef Z7_SFX
|
||||
#include "../Agent/IFolderArchive.h"
|
||||
#endif
|
||||
|
||||
#include "../Common/ArchiveExtractCallback.h"
|
||||
#include "../Common/ArchiveOpenCallback.h"
|
||||
|
||||
#ifndef Z7_NO_CRYPTO
|
||||
#include "../../IPassword.h"
|
||||
#endif
|
||||
|
||||
#ifndef Z7_SFX
|
||||
#include "IFolder.h"
|
||||
#endif
|
||||
|
||||
#include "ProgressDialog2.h"
|
||||
|
||||
#ifndef Z7_SFX
|
||||
|
||||
class CGrowBuf
|
||||
{
|
||||
Byte *_items;
|
||||
size_t _size;
|
||||
|
||||
Z7_CLASS_NO_COPY(CGrowBuf)
|
||||
|
||||
public:
|
||||
void Free()
|
||||
{
|
||||
MyFree(_items);
|
||||
_items = NULL;
|
||||
_size = 0;
|
||||
}
|
||||
|
||||
// newSize >= keepSize
|
||||
bool ReAlloc_KeepData(size_t newSize, size_t keepSize)
|
||||
{
|
||||
void *buf = NULL;
|
||||
if (newSize)
|
||||
{
|
||||
buf = MyAlloc(newSize);
|
||||
if (!buf)
|
||||
return false;
|
||||
}
|
||||
if (keepSize)
|
||||
memcpy(buf, _items, keepSize);
|
||||
MyFree(_items);
|
||||
_items = (Byte *)buf;
|
||||
_size = newSize;
|
||||
return true;
|
||||
}
|
||||
|
||||
CGrowBuf(): _items(NULL), _size(0) {}
|
||||
~CGrowBuf() { MyFree(_items); }
|
||||
|
||||
operator Byte *() { return _items; }
|
||||
operator const Byte *() const { return _items; }
|
||||
size_t Size() const { return _size; }
|
||||
};
|
||||
|
||||
|
||||
struct CVirtFile
|
||||
{
|
||||
CGrowBuf Data;
|
||||
|
||||
UInt64 ExpectedSize; // size from props request. 0 if unknown
|
||||
size_t WrittenSize; // size of written data in (Data) buffer
|
||||
// use (WrittenSize) only if (CVirtFileSystem::_newVirtFileStream_IsReadyToWrite == false)
|
||||
UString BaseName; // original name of file inside archive,
|
||||
// It's not path. So any path separators
|
||||
// should be treated as part of name (or as incorrect chars)
|
||||
UString AltStreamName;
|
||||
|
||||
bool CTime_Defined;
|
||||
bool ATime_Defined;
|
||||
bool MTime_Defined;
|
||||
bool Attrib_Defined;
|
||||
|
||||
// bool IsDir;
|
||||
bool IsAltStream;
|
||||
bool ColonWasUsed;
|
||||
DWORD Attrib;
|
||||
|
||||
FILETIME CTime;
|
||||
FILETIME ATime;
|
||||
FILETIME MTime;
|
||||
|
||||
CVirtFile():
|
||||
CTime_Defined(false),
|
||||
ATime_Defined(false),
|
||||
MTime_Defined(false),
|
||||
Attrib_Defined(false),
|
||||
// IsDir(false),
|
||||
IsAltStream(false),
|
||||
ColonWasUsed(false)
|
||||
{}
|
||||
};
|
||||
|
||||
|
||||
/*
|
||||
We use CVirtFileSystem only for single file extraction:
|
||||
It supports the following cases and names:
|
||||
- "fileName" : single file
|
||||
- "fileName" item (main base file) and additional "fileName:altStream" items
|
||||
- "altStream" : single item without "fileName:" prefix.
|
||||
If file is flushed to disk, it uses Get_Correct_FsFile_Name(name).
|
||||
*/
|
||||
|
||||
Z7_CLASS_IMP_NOQIB_1(
|
||||
CVirtFileSystem,
|
||||
ISequentialOutStream
|
||||
)
|
||||
unsigned _numFlushed;
|
||||
public:
|
||||
bool IsAltStreamFile; // in:
|
||||
// = true, if extracting file is alt stream without "fileName:" prefix.
|
||||
// = false, if extracting file is normal file, but additional
|
||||
// alt streams "fileName:altStream" items are possible.
|
||||
private:
|
||||
bool _newVirtFileStream_IsReadyToWrite; // it can non real file (if can't open alt stream)
|
||||
bool _needWriteToRealFile; // we need real writing to open file.
|
||||
bool _wasSwitchedToFsMode;
|
||||
bool _altStream_NeedRestore_Attrib_bool;
|
||||
DWORD _altStream_NeedRestore_AttribVal;
|
||||
|
||||
CMyComPtr2<ISequentialOutStream, COutFileStream> _outFileStream;
|
||||
public:
|
||||
CObjectVector<CVirtFile> Files;
|
||||
size_t MaxTotalAllocSize; // remain size, including Files.Back()
|
||||
FString DirPrefix; // files will be flushed to this FS directory.
|
||||
UString FileName; // name of file that will be extracted.
|
||||
// it can be name of alt stream without "fileName:" prefix, if (IsAltStreamFile == trye).
|
||||
// we use that name to detect altStream part in "FileName:altStream".
|
||||
CByteBuffer ZoneBuf;
|
||||
int Index_of_MainExtractedFile_in_Files; // out: index in Files. == -1, if expected file was not extracted
|
||||
int Index_of_ZoneBuf_AltStream_in_Files; // out: index in Files. == -1, if no zonbuf alt stream
|
||||
|
||||
|
||||
CVirtFileSystem()
|
||||
{
|
||||
_numFlushed = 0;
|
||||
IsAltStreamFile = false;
|
||||
_newVirtFileStream_IsReadyToWrite = false;
|
||||
_needWriteToRealFile = false;
|
||||
_wasSwitchedToFsMode = false;
|
||||
_altStream_NeedRestore_Attrib_bool = false;
|
||||
MaxTotalAllocSize = (size_t)0 - 1;
|
||||
Index_of_MainExtractedFile_in_Files = -1;
|
||||
Index_of_ZoneBuf_AltStream_in_Files = -1;
|
||||
}
|
||||
|
||||
bool WasStreamFlushedToFS() const { return _wasSwitchedToFsMode; }
|
||||
|
||||
HRESULT CloseMemFile()
|
||||
{
|
||||
if (_wasSwitchedToFsMode)
|
||||
return FlushToDisk(true); // closeLast
|
||||
CVirtFile &file = Files.Back();
|
||||
if (file.Data.Size() != file.WrittenSize)
|
||||
file.Data.ReAlloc_KeepData(file.WrittenSize, file.WrittenSize);
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
HRESULT FlushToDisk(bool closeLast);
|
||||
};
|
||||
|
||||
#endif
|
||||
|
||||
|
||||
|
||||
class CExtractCallbackImp Z7_final:
|
||||
public IFolderArchiveExtractCallback,
|
||||
/* IExtractCallbackUI:
|
||||
before v23.00 : it included IFolderArchiveExtractCallback
|
||||
since v23.00 : it doesn't include IFolderArchiveExtractCallback
|
||||
*/
|
||||
public IExtractCallbackUI, // NON-COM interface since 23.00
|
||||
public IOpenCallbackUI, // NON-COM interface
|
||||
public IFolderArchiveExtractCallback2,
|
||||
#ifndef Z7_SFX
|
||||
public IFolderOperationsExtractCallback,
|
||||
public IFolderExtractToStreamCallback,
|
||||
public ICompressProgressInfo,
|
||||
public IArchiveRequestMemoryUseCallback,
|
||||
#endif
|
||||
#ifndef Z7_NO_CRYPTO
|
||||
public ICryptoGetTextPassword,
|
||||
#endif
|
||||
public CMyUnknownImp
|
||||
{
|
||||
Z7_COM_QI_BEGIN2(IFolderArchiveExtractCallback)
|
||||
Z7_COM_QI_ENTRY(IFolderArchiveExtractCallback2)
|
||||
#ifndef Z7_SFX
|
||||
Z7_COM_QI_ENTRY(IFolderOperationsExtractCallback)
|
||||
Z7_COM_QI_ENTRY(IFolderExtractToStreamCallback)
|
||||
Z7_COM_QI_ENTRY(ICompressProgressInfo)
|
||||
Z7_COM_QI_ENTRY(IArchiveRequestMemoryUseCallback)
|
||||
#endif
|
||||
#ifndef Z7_NO_CRYPTO
|
||||
Z7_COM_QI_ENTRY(ICryptoGetTextPassword)
|
||||
#endif
|
||||
Z7_COM_QI_END
|
||||
Z7_COM_ADDREF_RELEASE
|
||||
|
||||
Z7_IFACE_IMP(IExtractCallbackUI)
|
||||
Z7_IFACE_IMP(IOpenCallbackUI)
|
||||
Z7_IFACE_COM7_IMP(IProgress)
|
||||
Z7_IFACE_COM7_IMP(IFolderArchiveExtractCallback)
|
||||
Z7_IFACE_COM7_IMP(IFolderArchiveExtractCallback2)
|
||||
#ifndef Z7_SFX
|
||||
Z7_IFACE_COM7_IMP(IFolderOperationsExtractCallback)
|
||||
Z7_IFACE_COM7_IMP(IFolderExtractToStreamCallback)
|
||||
Z7_IFACE_COM7_IMP(ICompressProgressInfo)
|
||||
Z7_IFACE_COM7_IMP(IArchiveRequestMemoryUseCallback)
|
||||
#endif
|
||||
#ifndef Z7_NO_CRYPTO
|
||||
Z7_IFACE_COM7_IMP(ICryptoGetTextPassword)
|
||||
#endif
|
||||
|
||||
bool _needWriteArchivePath;
|
||||
bool _isFolder;
|
||||
bool _totalFiles_Defined;
|
||||
bool _totalBytes_Defined;
|
||||
public:
|
||||
bool MultiArcMode;
|
||||
bool ProcessAltStreams;
|
||||
bool StreamMode; // set to true, if you want the callee to call GetStream7()
|
||||
bool ThereAreMessageErrors;
|
||||
bool Src_Is_IO_FS_Folder;
|
||||
|
||||
#ifndef Z7_NO_CRYPTO
|
||||
bool PasswordIsDefined;
|
||||
bool PasswordWasAsked;
|
||||
#endif
|
||||
|
||||
private:
|
||||
#ifndef Z7_SFX
|
||||
bool _needUpdateStat;
|
||||
bool _newVirtFileWasAdded;
|
||||
bool _isAltStream;
|
||||
// bool _extractMode;
|
||||
// bool _testMode;
|
||||
bool _hashStream_WasUsed;
|
||||
bool _curSize_Defined;
|
||||
bool NeedAddFile;
|
||||
|
||||
bool _remember;
|
||||
bool _skipArc;
|
||||
#endif
|
||||
|
||||
public:
|
||||
bool YesToAll;
|
||||
bool TestMode;
|
||||
|
||||
UInt32 NumArchiveErrors;
|
||||
NExtract::NOverwriteMode::EEnum OverwriteMode;
|
||||
|
||||
private:
|
||||
UString _currentArchivePath;
|
||||
UString _currentFilePath;
|
||||
UString _filePath; // virtual path than will be sent via IFolderExtractToStreamCallback
|
||||
|
||||
#ifndef Z7_SFX
|
||||
UInt64 _curSize;
|
||||
CMyComPtr2<ISequentialOutStream, COutStreamWithHash> _hashStream;
|
||||
IHashCalc *_hashCalc; // it's for stat in Test operation
|
||||
#endif
|
||||
|
||||
public:
|
||||
CProgressDialog *ProgressDialog;
|
||||
|
||||
#ifndef Z7_SFX
|
||||
CVirtFileSystem *VirtFileSystemSpec;
|
||||
CMyComPtr<ISequentialOutStream> VirtFileSystem;
|
||||
UInt64 NumFolders;
|
||||
UInt64 NumFiles;
|
||||
#endif
|
||||
|
||||
#ifndef Z7_NO_CRYPTO
|
||||
UString Password;
|
||||
#endif
|
||||
|
||||
UString _lang_Extracting;
|
||||
UString _lang_Testing;
|
||||
UString _lang_Skipping;
|
||||
UString _lang_Reading;
|
||||
UString _lang_Empty;
|
||||
|
||||
CExtractCallbackImp():
|
||||
_totalFiles_Defined(false)
|
||||
, _totalBytes_Defined(false)
|
||||
, MultiArcMode(false)
|
||||
, ProcessAltStreams(true)
|
||||
, StreamMode(false)
|
||||
, ThereAreMessageErrors(false)
|
||||
, Src_Is_IO_FS_Folder(false)
|
||||
#ifndef Z7_NO_CRYPTO
|
||||
, PasswordIsDefined(false)
|
||||
, PasswordWasAsked(false)
|
||||
#endif
|
||||
#ifndef Z7_SFX
|
||||
, _remember(false)
|
||||
, _skipArc(false)
|
||||
#endif
|
||||
, YesToAll(false)
|
||||
, TestMode(false)
|
||||
, OverwriteMode(NExtract::NOverwriteMode::kAsk)
|
||||
#ifndef Z7_SFX
|
||||
, _hashCalc(NULL)
|
||||
#endif
|
||||
{}
|
||||
|
||||
~CExtractCallbackImp();
|
||||
void Init();
|
||||
|
||||
HRESULT SetCurrentFilePath2(const wchar_t *filePath);
|
||||
void AddError_Message(LPCWSTR message);
|
||||
void AddError_Message_ShowArcPath(LPCWSTR message);
|
||||
HRESULT MessageError(const char *message, const FString &path);
|
||||
void Add_ArchiveName_Error();
|
||||
|
||||
#ifndef Z7_SFX
|
||||
void SetHashCalc(IHashCalc *hashCalc) { _hashCalc = hashCalc; }
|
||||
|
||||
void SetHashMethods(IHashCalc *hash)
|
||||
{
|
||||
if (!hash)
|
||||
return;
|
||||
_hashStream.Create_if_Empty();
|
||||
_hashStream->_hash = hash;
|
||||
}
|
||||
#endif
|
||||
|
||||
bool IsOK() const { return NumArchiveErrors == 0 && !ThereAreMessageErrors; }
|
||||
};
|
||||
|
||||
#endif
|
||||
Reference in New Issue
Block a user