124 lines
4.5 KiB
C++
124 lines
4.5 KiB
C++
/*============================================================================
|
|
CMake - Cross Platform Makefile Generator
|
|
Copyright 2000-2009 Kitware, Inc., Insight Software Consortium
|
|
|
|
Distributed under the OSI-approved BSD License (the "License");
|
|
see accompanying file Copyright.txt for details.
|
|
|
|
This software is distributed WITHOUT ANY WARRANTY; without even the
|
|
implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
|
|
See the License for more information.
|
|
============================================================================*/
|
|
#include "cmCryptoHash.h"
|
|
|
|
#include "cm_sha2.h"
|
|
#include <cmsys/FStream.hxx>
|
|
#include <cmsys/MD5.h>
|
|
|
|
cmsys::auto_ptr<cmCryptoHash> cmCryptoHash::New(const char* algo)
|
|
{
|
|
if (strcmp(algo, "MD5") == 0) {
|
|
return cmsys::auto_ptr<cmCryptoHash>(new cmCryptoHashMD5);
|
|
} else if (strcmp(algo, "SHA1") == 0) {
|
|
return cmsys::auto_ptr<cmCryptoHash>(new cmCryptoHashSHA1);
|
|
} else if (strcmp(algo, "SHA224") == 0) {
|
|
return cmsys::auto_ptr<cmCryptoHash>(new cmCryptoHashSHA224);
|
|
} else if (strcmp(algo, "SHA256") == 0) {
|
|
return cmsys::auto_ptr<cmCryptoHash>(new cmCryptoHashSHA256);
|
|
} else if (strcmp(algo, "SHA384") == 0) {
|
|
return cmsys::auto_ptr<cmCryptoHash>(new cmCryptoHashSHA384);
|
|
} else if (strcmp(algo, "SHA512") == 0) {
|
|
return cmsys::auto_ptr<cmCryptoHash>(new cmCryptoHashSHA512);
|
|
} else {
|
|
return cmsys::auto_ptr<cmCryptoHash>(CM_NULLPTR);
|
|
}
|
|
}
|
|
|
|
std::string cmCryptoHash::HashString(const std::string& input)
|
|
{
|
|
this->Initialize();
|
|
this->Append(reinterpret_cast<unsigned char const*>(input.c_str()),
|
|
static_cast<int>(input.size()));
|
|
return this->Finalize();
|
|
}
|
|
|
|
std::string cmCryptoHash::HashFile(const std::string& file)
|
|
{
|
|
cmsys::ifstream fin(file.c_str(), std::ios::in | std::ios::binary);
|
|
if (!fin) {
|
|
return "";
|
|
}
|
|
|
|
this->Initialize();
|
|
|
|
// Should be efficient enough on most system:
|
|
cm_sha2_uint64_t buffer[512];
|
|
char* buffer_c = reinterpret_cast<char*>(buffer);
|
|
unsigned char const* buffer_uc =
|
|
reinterpret_cast<unsigned char const*>(buffer);
|
|
// This copy loop is very sensitive on certain platforms with
|
|
// slightly broken stream libraries (like HPUX). Normally, it is
|
|
// incorrect to not check the error condition on the fin.read()
|
|
// before using the data, but the fin.gcount() will be zero if an
|
|
// error occurred. Therefore, the loop should be safe everywhere.
|
|
while (fin) {
|
|
fin.read(buffer_c, sizeof(buffer));
|
|
if (int gcount = static_cast<int>(fin.gcount())) {
|
|
this->Append(buffer_uc, gcount);
|
|
}
|
|
}
|
|
if (fin.eof()) {
|
|
return this->Finalize();
|
|
}
|
|
return "";
|
|
}
|
|
|
|
cmCryptoHashMD5::cmCryptoHashMD5()
|
|
: MD5(cmsysMD5_New())
|
|
{
|
|
}
|
|
|
|
cmCryptoHashMD5::~cmCryptoHashMD5()
|
|
{
|
|
cmsysMD5_Delete(this->MD5);
|
|
}
|
|
|
|
void cmCryptoHashMD5::Initialize()
|
|
{
|
|
cmsysMD5_Initialize(this->MD5);
|
|
}
|
|
|
|
void cmCryptoHashMD5::Append(unsigned char const* buf, int sz)
|
|
{
|
|
cmsysMD5_Append(this->MD5, buf, sz);
|
|
}
|
|
|
|
std::string cmCryptoHashMD5::Finalize()
|
|
{
|
|
char md5out[32];
|
|
cmsysMD5_FinalizeHex(this->MD5, md5out);
|
|
return std::string(md5out, 32);
|
|
}
|
|
|
|
#define cmCryptoHash_SHA_CLASS_IMPL(SHA) \
|
|
cmCryptoHash##SHA::cmCryptoHash##SHA() \
|
|
: SHA(new SHA_CTX) \
|
|
{ \
|
|
} \
|
|
cmCryptoHash##SHA::~cmCryptoHash##SHA() { delete this->SHA; } \
|
|
void cmCryptoHash##SHA::Initialize() { SHA##_Init(this->SHA); } \
|
|
void cmCryptoHash##SHA::Append(unsigned char const* buf, int sz) \
|
|
{ \
|
|
SHA##_Update(this->SHA, buf, sz); \
|
|
} \
|
|
std::string cmCryptoHash##SHA::Finalize() \
|
|
{ \
|
|
char out[SHA##_DIGEST_STRING_LENGTH]; \
|
|
SHA##_End(this->SHA, out); \
|
|
return std::string(out, SHA##_DIGEST_STRING_LENGTH - 1); \
|
|
}
|
|
|
|
cmCryptoHash_SHA_CLASS_IMPL(SHA1) cmCryptoHash_SHA_CLASS_IMPL(SHA224)
|
|
cmCryptoHash_SHA_CLASS_IMPL(SHA256) cmCryptoHash_SHA_CLASS_IMPL(SHA384)
|
|
cmCryptoHash_SHA_CLASS_IMPL(SHA512)
|