Home | History | Annotate | Download | only in Common 
      1 // HashCalc.h
      2 
      3 #ifndef __HASH_CALC_H
      4 #define __HASH_CALC_H
      5 
      6 #include "../../../Common/Wildcard.h"
      7 
      8 #include "../../Common/CreateCoder.h"
      9 #include "../../Common/MethodProps.h"
     10 
     11 #include "Property.h"
     12 
     13 const unsigned k_HashCalc_DigestSize_Max = 64;
     14 
     15 const unsigned k_HashCalc_NumGroups = 4;
     16 
     17 enum
     18 {
     19   k_HashCalc_Index_Current,
     20   k_HashCalc_Index_DataSum,
     21   k_HashCalc_Index_NamesSum,
     22   k_HashCalc_Index_StreamsSum
     23 };
     24 
     25 struct CHasherState
     26 {
     27   CMyComPtr<IHasher> Hasher;
     28   UString Name;
     29   UInt32 DigestSize;
     30   Byte Digests[k_HashCalc_NumGroups][k_HashCalc_DigestSize_Max];
     31 };
     32 
     33 struct IHashCalc
     34 {
     35   virtual void InitForNewFile() = 0;
     36   virtual void Update(const void *data, UInt32 size) = 0;
     37   virtual void SetSize(UInt64 size) = 0;
     38   virtual void Final(bool isDir, bool isAltStream, const UString &path) = 0;
     39 };
     40 
     41 struct CHashBundle: public IHashCalc
     42 {
     43   CObjectVector<CHasherState> Hashers;
     44 
     45   UInt64 NumFiles;
     46   UInt64 NumDirs;
     47   UInt64 NumAltStreams;
     48   UInt64 FilesSize;
     49   UInt64 AltStreamsSize;
     50   UInt64 NumErrors;
     51 
     52   UInt64 CurSize;
     53 
     54   HRESULT SetMethods(DECL_EXTERNAL_CODECS_LOC_VARS const UStringVector &methods);
     55 
     56   void Init()
     57   {
     58     NumFiles = NumDirs = NumAltStreams = FilesSize = AltStreamsSize = NumErrors = 0;
     59   }
     60 
     61   void InitForNewFile();
     62   void Update(const void *data, UInt32 size);
     63   void SetSize(UInt64 size);
     64   void Final(bool isDir, bool isAltStream, const UString &path);
     65 };
     66 
     67 #define INTERFACE_IHashCallbackUI(x) \
     68   virtual HRESULT StartScanning() x; \
     69   virtual HRESULT ScanProgress(UInt64 numFolders, UInt64 numFiles, UInt64 totalSize, const wchar_t *path, bool isDir) x; \
     70   virtual HRESULT CanNotFindError(const wchar_t *name, DWORD systemError) x; \
     71   virtual HRESULT FinishScanning() x; \
     72   virtual HRESULT SetNumFiles(UInt64 numFiles) x; \
     73   virtual HRESULT SetTotal(UInt64 size) x; \
     74   virtual HRESULT SetCompleted(const UInt64 *completeValue) x; \
     75   virtual HRESULT CheckBreak() x; \
     76   virtual HRESULT BeforeFirstFile(const CHashBundle &hb) x; \
     77   virtual HRESULT GetStream(const wchar_t *name, bool isFolder) x; \
     78   virtual HRESULT OpenFileError(const wchar_t *name, DWORD systemError) x; \
     79   virtual HRESULT SetOperationResult(UInt64 fileSize, const CHashBundle &hb, bool showHash) x; \
     80   virtual HRESULT AfterLastFile(const CHashBundle &hb) x; \
     81 
     82 struct IHashCallbackUI
     83 {
     84   INTERFACE_IHashCallbackUI(=0)
     85 };
     86 
     87 struct CHashOptions
     88 {
     89   UStringVector Methods;
     90   bool OpenShareForWrite;
     91   bool StdInMode;
     92   bool AltStreamsMode;
     93   NWildcard::ECensorPathMode PathMode;
     94 
     95   CHashOptions(): StdInMode(false), OpenShareForWrite(false), AltStreamsMode(false), PathMode(NWildcard::k_RelatPath) {};
     96 };
     97 
     98 HRESULT HashCalc(
     99     DECL_EXTERNAL_CODECS_LOC_VARS
    100     const NWildcard::CCensor &censor,
    101     const CHashOptions &options,
    102     UString &errorInfo,
    103     IHashCallbackUI *callback);
    104 
    105 void AddHashHexToString(char *dest, const Byte *data, UInt32 size);
    106 
    107 #endif
    108