mirror of
https://github.com/hashcat/hashcat.git
synced 2024-11-29 03:18:30 +00:00
108 lines
2.6 KiB
C++
108 lines
2.6 KiB
C++
#ifndef _RAR_DATAIO_
|
|
#define _RAR_DATAIO_
|
|
|
|
class Archive;
|
|
class CmdAdd;
|
|
class Unpack;
|
|
class ArcFileSearch;
|
|
|
|
#if 0
|
|
// We use external i/o calls for Benchmark command.
|
|
#define COMPRDATAIO_EXTIO
|
|
#endif
|
|
|
|
class ComprDataIO
|
|
{
|
|
private:
|
|
void ShowUnpRead(int64 ArcPos,int64 ArcSize);
|
|
void ShowUnpWrite();
|
|
|
|
|
|
bool UnpackFromMemory;
|
|
size_t UnpackFromMemorySize;
|
|
byte *UnpackFromMemoryAddr;
|
|
|
|
bool UnpackToMemory;
|
|
size_t UnpackToMemorySize;
|
|
byte *UnpackToMemoryAddr;
|
|
|
|
size_t UnpWrSize;
|
|
byte *UnpWrAddr;
|
|
|
|
int64 UnpPackedSize;
|
|
int64 UnpPackedLeft;
|
|
|
|
bool ShowProgress;
|
|
bool TestMode;
|
|
bool SkipUnpCRC;
|
|
bool NoFileHeader;
|
|
|
|
File *SrcFile;
|
|
File *DestFile;
|
|
|
|
CmdAdd *Command;
|
|
|
|
FileHeader *SubHead;
|
|
int64 *SubHeadPos;
|
|
|
|
#ifndef RAR_NOCRYPT
|
|
CryptData *Crypt;
|
|
CryptData *Decrypt;
|
|
#endif
|
|
|
|
|
|
int LastPercent;
|
|
|
|
wchar CurrentCommand;
|
|
|
|
public:
|
|
ComprDataIO();
|
|
~ComprDataIO();
|
|
void Init();
|
|
int UnpRead(byte *Addr,size_t Count);
|
|
void UnpWrite(byte *Addr,size_t Count);
|
|
void EnableShowProgress(bool Show) {ShowProgress=Show;}
|
|
void GetUnpackedData(byte **Data,size_t *Size);
|
|
void SetPackedSizeToRead(int64 Size) {UnpPackedSize=UnpPackedLeft=Size;}
|
|
void SetTestMode(bool Mode) {TestMode=Mode;}
|
|
void SetSkipUnpCRC(bool Skip) {SkipUnpCRC=Skip;}
|
|
void SetNoFileHeader(bool Mode) {NoFileHeader=Mode;}
|
|
void SetFiles(File *SrcFile,File *DestFile);
|
|
void SetCommand(CmdAdd *Cmd) {Command=Cmd;}
|
|
void SetSubHeader(FileHeader *hd,int64 *Pos) {SubHead=hd;SubHeadPos=Pos;}
|
|
void SetEncryption(bool Encrypt,CRYPT_METHOD Method,SecPassword *Password,
|
|
const byte *Salt,const byte *InitV,uint Lg2Cnt,byte *HashKey,byte *PswCheck);
|
|
void InitRijindal(byte *Key,byte *InitV);
|
|
void SetAV15Encryption();
|
|
void SetCmt13Encryption();
|
|
void SetUnpackToMemory(byte *Addr,uint Size);
|
|
void SetUnpackFromMemory(byte *Addr,uint Size);
|
|
void SetCurrentCommand(wchar Cmd) {CurrentCommand=Cmd;}
|
|
void AdjustTotalArcSize(Archive *Arc);
|
|
|
|
|
|
bool PackVolume;
|
|
bool UnpVolume;
|
|
bool NextVolumeMissing;
|
|
int64 CurPackRead,CurPackWrite,CurUnpRead,CurUnpWrite;
|
|
|
|
|
|
// Size of already processed archives.
|
|
// Used to calculate the total operation progress.
|
|
int64 ProcessedArcSize;
|
|
|
|
// Last extracted archive size up to QO or RR block.
|
|
int64 LastArcSize;
|
|
|
|
int64 TotalArcSize;
|
|
|
|
DataHash PackedDataHash; // Packed write and unpack read hash.
|
|
DataHash PackHash; // Pack read hash.
|
|
DataHash UnpHash; // Unpack write hash.
|
|
|
|
bool Encryption;
|
|
bool Decryption;
|
|
};
|
|
|
|
#endif
|