99 lines
2.1 KiB
C++
99 lines
2.1 KiB
C++
// CrcReg.cpp
|
|
|
|
#include "StdAfx.h"
|
|
|
|
#include "../../C/7zCrc.h"
|
|
#include "../../C/CpuArch.h"
|
|
|
|
#include "../Common/MyCom.h"
|
|
|
|
#include "../7zip/Common/RegisterCodec.h"
|
|
|
|
EXTERN_C_BEGIN
|
|
|
|
typedef UInt32 (MY_FAST_CALL *CRC_FUNC)(UInt32 v, const void *data, size_t size, const UInt32 *table);
|
|
|
|
UInt32 MY_FAST_CALL CrcUpdateT1(UInt32 v, const void *data, size_t size, const UInt32 *table);
|
|
|
|
extern CRC_FUNC g_CrcUpdate;
|
|
extern CRC_FUNC g_CrcUpdateT8;
|
|
extern CRC_FUNC g_CrcUpdateT4;
|
|
|
|
EXTERN_C_END
|
|
|
|
class CCrcHasher:
|
|
public IHasher,
|
|
public ICompressSetCoderProperties,
|
|
public CMyUnknownImp
|
|
{
|
|
UInt32 _crc;
|
|
CRC_FUNC _updateFunc;
|
|
Byte mtDummy[1 << 7];
|
|
|
|
bool SetFunctions(UInt32 tSize);
|
|
public:
|
|
CCrcHasher(): _crc(CRC_INIT_VAL) { SetFunctions(0); }
|
|
|
|
MY_UNKNOWN_IMP2(IHasher, ICompressSetCoderProperties)
|
|
INTERFACE_IHasher(;)
|
|
STDMETHOD(SetCoderProperties)(const PROPID *propIDs, const PROPVARIANT *props, UInt32 numProps);
|
|
};
|
|
|
|
bool CCrcHasher::SetFunctions(UInt32 tSize)
|
|
{
|
|
_updateFunc = g_CrcUpdate;
|
|
|
|
if (tSize == 1)
|
|
_updateFunc = CrcUpdateT1;
|
|
else if (tSize == 4)
|
|
{
|
|
if (g_CrcUpdateT4)
|
|
_updateFunc = g_CrcUpdateT4;
|
|
else
|
|
return false;
|
|
}
|
|
else if (tSize == 8)
|
|
{
|
|
if (g_CrcUpdateT8)
|
|
_updateFunc = g_CrcUpdateT8;
|
|
else
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
STDMETHODIMP CCrcHasher::SetCoderProperties(const PROPID *propIDs, const PROPVARIANT *coderProps, UInt32 numProps)
|
|
{
|
|
for (UInt32 i = 0; i < numProps; i++)
|
|
{
|
|
const PROPVARIANT &prop = coderProps[i];
|
|
if (propIDs[i] == NCoderPropID::kDefaultProp)
|
|
{
|
|
if (prop.vt != VT_UI4)
|
|
return E_INVALIDARG;
|
|
if (!SetFunctions(prop.ulVal))
|
|
return E_NOTIMPL;
|
|
}
|
|
}
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP_(void) CCrcHasher::Init() throw()
|
|
{
|
|
_crc = CRC_INIT_VAL;
|
|
}
|
|
|
|
STDMETHODIMP_(void) CCrcHasher::Update(const void *data, UInt32 size) throw()
|
|
{
|
|
_crc = _updateFunc(_crc, data, size, g_CrcTable);
|
|
}
|
|
|
|
STDMETHODIMP_(void) CCrcHasher::Final(Byte *digest) throw()
|
|
{
|
|
UInt32 val = CRC_GET_DIGEST(_crc);
|
|
SetUi32(digest, val);
|
|
}
|
|
|
|
REGISTER_HASHER(CCrcHasher, 0x1, "CRC32", 4)
|