1 // Sha3Reg.cpp 2 3 #include "StdAfx.h" 4 5 #include "../../C/Sha3.h" 6 7 #include "../Common/MyBuffer2.h" 8 #include "../Common/MyCom.h" 9 10 #include "../7zip/Common/RegisterCodec.h" 11 12 Z7_CLASS_IMP_COM_1( 13 CSha3Hasher 14 , IHasher 15 ) 16 unsigned _digestSize; 17 bool _isShake; 18 CAlignedBuffer1 _buf; 19 public: 20 Byte _mtDummy[1 << 7]; 21 22 CSha3 *Sha() { return (CSha3 *)(void *)(Byte *)_buf; } 23 public: 24 CSha3Hasher(unsigned digestSize, bool isShake, unsigned blockSize): 25 _digestSize(digestSize), 26 _isShake(isShake), 27 _buf(sizeof(CSha3)) 28 { 29 CSha3 *p = Sha(); 30 Sha3_SET_blockSize(p, blockSize) 31 Sha3_Init(Sha()); 32 } 33 }; 34 35 Z7_COM7F_IMF2(void, CSha3Hasher::Init()) 36 { 37 Sha3_Init(Sha()); 38 } 39 40 Z7_COM7F_IMF2(void, CSha3Hasher::Update(const void *data, UInt32 size)) 41 { 42 Sha3_Update(Sha(), (const Byte *)data, size); 43 } 44 45 Z7_COM7F_IMF2(void, CSha3Hasher::Final(Byte *digest)) 46 { 47 Sha3_Final(Sha(), digest, _digestSize, _isShake); 48 } 49 50 Z7_COM7F_IMF2(UInt32, CSha3Hasher::GetDigestSize()) 51 { 52 return (UInt32)_digestSize; 53 } 54 55 56 #define REGISTER_SHA3_HASHER_2(cls, id, name, digestSize, isShake, digestSize_for_blockSize) \ 57 namespace N ## cls { \ 58 static IHasher *CreateHasherSpec() \ 59 { return new CSha3Hasher(digestSize / 8, isShake, \ 60 SHA3_BLOCK_SIZE_FROM_DIGEST_SIZE(digestSize_for_blockSize / 8)); } \ 61 static const CHasherInfo g_HasherInfo = { CreateHasherSpec, id, name, digestSize }; \ 62 struct REGISTER_HASHER_NAME(cls) { REGISTER_HASHER_NAME(cls)() { RegisterHasher(&g_HasherInfo); }}; \ 63 static REGISTER_HASHER_NAME(cls) g_RegisterHasher; } 64 65 #define REGISTER_SHA3_HASHER( cls, id, name, size, isShake) \ 66 REGISTER_SHA3_HASHER_2(cls, id, name, size, isShake, size) 67 68 // REGISTER_SHA3_HASHER (Sha3_224_Hasher, 0x230, "SHA3-224", 224, false) 69 REGISTER_SHA3_HASHER (Sha3_256_Hasher, 0x231, "SHA3-256", 256, false) 70 // REGISTER_SHA3_HASHER (Sha3_386_Hasher, 0x232, "SHA3-384", 384, false) 71 // REGISTER_SHA3_HASHER (Sha3_512_Hasher, 0x233, "SHA3-512", 512, false) 72 // REGISTER_SHA3_HASHER (Shake128_Hasher, 0x240, "SHAKE128", 128, true) 73 // REGISTER_SHA3_HASHER (Shake256_Hasher, 0x241, "SHAKE256", 256, true) 74 // REGISTER_SHA3_HASHER_2 (Shake128_512_Hasher, 0x248, "SHAKE128-256", 256, true, 128) // -1344 (max) 75 // REGISTER_SHA3_HASHER_2 (Shake256_512_Hasher, 0x249, "SHAKE256-512", 512, true, 256) // -1088 (max) 76 // Shake supports different digestSize values for same blockSize 77