SHA1_MD5_Object.cpp [plain text]
#include "SHA1_MD5_Object.h"
#include <stdexcept>
#include <string.h>
void MD5Object::digestInit()
{
mIsDone = false;
MD5Init(&mCtx);
}
void MD5Object::digestUpdate(
const void *data,
size_t len)
{
if(mIsDone) {
throw std::runtime_error("MD5 digestUpdate after final");
}
MD5Update(&mCtx, (unsigned char *)data, len);
}
void MD5Object::digestFinal(
void *digest)
{
if(mIsDone) {
throw std::runtime_error("MD5 digestFinal after final");
}
MD5Final(&mCtx, (unsigned char *)digest);
mIsDone = true;
}
DigestObject *MD5Object::digestClone() const
{
return new MD5Object(*this);
}
UInt32 MD5Object::digestSizeInBytes() const
{
return MD5_DIGEST_SIZE;
}
void SHA1Object::digestInit()
{
mIsDone = false;
shsInit(&mCtx);
mBufferCount = 0;
}
void SHA1Object::digestUpdate(
const void *data,
size_t len)
{
size_t cnt;
uint8 *uData = (uint8 *)data;
if(mIsDone) {
throw std::runtime_error("SHA1 digestUpdate after final");
}
if (mBufferCount + len < SHS_BLOCKSIZE) {
memcpy(mBuffer + mBufferCount, uData, len);
mBufferCount += len;
return;
}
if (mBufferCount > 0) {
cnt = SHS_BLOCKSIZE - mBufferCount;
memcpy(mBuffer + mBufferCount, uData, cnt);
shsUpdate(&mCtx, mBuffer, SHS_BLOCKSIZE);
uData += cnt;
len -= cnt;
}
UInt32 blocks = len / SHS_BLOCKSIZE;
if(blocks) {
cnt = blocks * SHS_BLOCKSIZE;
shsUpdate(&mCtx, uData, cnt);
uData += cnt;
len -= cnt;
}
mBufferCount = len;
if (len > 0) {
memcpy(mBuffer, uData, len);
}
}
void SHA1Object::digestFinal(
void *digest)
{
if(mIsDone) {
throw std::runtime_error("SHA1 digestFinal after final");
}
if (mBufferCount > 0) {
shsUpdate(&mCtx, mBuffer, mBufferCount);
}
shsFinal(&mCtx);
memcpy(digest, mCtx.digest, SHS_DIGESTSIZE);
mIsDone = true;
}
DigestObject *SHA1Object::digestClone() const
{
return new SHA1Object(*this);
}
UInt32 SHA1Object::digestSizeInBytes() const
{
return SHS_DIGESTSIZE;
}