1
0
Fork 0
mirror of https://github.com/NixOS/nix synced 2025-06-28 17:51:15 +02:00

* Refactoring to support SHA-1.

This commit is contained in:
Eelco Dolstra 2005-01-13 15:44:44 +00:00
parent d46b4262dc
commit 73992371a3
6 changed files with 36 additions and 18 deletions

View file

@ -8,14 +8,18 @@ extern "C" {
#include "archive.hh"
Hash::Hash()
Hash::Hash(HashType type)
{
memset(hash, 0, sizeof(hash));
this->type = type;
if (type == htMD5) hashSize = md5HashSize;
else if (type == htSHA1) hashSize = sha1HashSize;
memset(hash, 0, hashSize);
}
bool Hash::operator == (const Hash & h2) const
{
if (hashSize != h2.hashSize) return false;
for (unsigned int i = 0; i < hashSize; i++)
if (hash[i] != h2.hash[i]) return false;
return true;
@ -52,10 +56,10 @@ Hash::operator string() const
Hash parseHash(const string & s)
{
Hash hash;
if (s.length() != Hash::hashSize * 2)
Hash hash(htMD5);
if (s.length() != hash.hashSize * 2)
throw Error(format("invalid hash `%1%'") % s);
for (unsigned int i = 0; i < Hash::hashSize; i++) {
for (unsigned int i = 0; i < hash.hashSize; i++) {
string s2(s, i * 2, 2);
if (!isxdigit(s2[0]) || !isxdigit(s2[1]))
throw Error(format("invalid hash `%1%'") % s);
@ -83,7 +87,7 @@ bool isHash(const string & s)
Hash hashString(const string & s)
{
Hash hash;
Hash hash(htMD5);
md5_buffer(s.c_str(), s.length(), hash.hash);
return hash;
}
@ -91,7 +95,7 @@ Hash hashString(const string & s)
Hash hashFile(const Path & path)
{
Hash hash;
Hash hash(htMD5);
FILE * file = fopen(path.c_str(), "rb");
if (!file)
throw SysError(format("file `%1%' does not exist") % path);
@ -115,7 +119,7 @@ struct HashSink : DumpSink
Hash hashPath(const Path & path)
{
Hash hash;
Hash hash(htMD5);
HashSink sink;
md5_init_ctx(&sink.ctx);
dumpPath(path, sink);

View file

@ -8,13 +8,23 @@
using namespace std;
typedef enum { htMD5, htSHA1 } HashType;
const int md5HashSize = 16;
const int sha1HashSize = 20;
struct Hash
{
static const unsigned int hashSize = 16;
unsigned char hash[hashSize];
static const unsigned int maxHashSize = 20;
unsigned int hashSize;
unsigned char hash[maxHashSize];
HashType type;
/* Create a zeroed hash object. */
Hash();
Hash(HashType type);
/* Check whether two hash are equal. */
bool operator == (const Hash & h2) const;