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