mirror of
https://github.com/hashcat/hashcat.git
synced 2024-11-14 03:39:09 +00:00
1125 lines
36 KiB
C
1125 lines
36 KiB
C
/*
|
|
* Brute force collision tester for 64-bit hashes
|
|
* Part of the xxHash project
|
|
* Copyright (C) 2019-2020 Yann Collet
|
|
*
|
|
* GPL v2 License
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with this program; if not, write to the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*
|
|
* You can contact the author at:
|
|
* - xxHash homepage: https://www.xxhash.com
|
|
* - xxHash source repository: https://github.com/Cyan4973/xxHash
|
|
*/
|
|
|
|
/*
|
|
* The collision tester will generate 24 billion hashes (by default),
|
|
* and count how many collisions were produced by the 64-bit hash algorithm.
|
|
* The optimal amount of collisions for 64-bit is ~18 collisions.
|
|
* A good hash should be close to this figure.
|
|
*
|
|
* This program requires a lot of memory:
|
|
* - Either store hash values directly => 192 GB
|
|
* - Or use a filter:
|
|
* - 32 GB (by default) for the filter itself
|
|
* - + ~14 GB for the list of hashes (depending on the filter's outcome)
|
|
* Due to these memory constraints, it requires a 64-bit system.
|
|
*/
|
|
|
|
|
|
/* === Dependencies === */
|
|
|
|
#include <stdint.h> /* uint64_t */
|
|
#include <stdlib.h> /* malloc, free, qsort, exit */
|
|
#include <string.h> /* memset */
|
|
#include <stdio.h> /* printf, fflush */
|
|
|
|
#undef NDEBUG /* ensure assert is _not_ disabled */
|
|
#include <assert.h>
|
|
|
|
#include "hashes.h" /* UniHash, hashfn, hashfnTable */
|
|
|
|
#include "sort.hh" /* sort64 */
|
|
|
|
|
|
|
|
typedef enum { ht32, ht64, ht128 } Htype_e;
|
|
|
|
/* === Debug === */
|
|
|
|
#define EXIT(...) { printf(__VA_ARGS__); printf("\n"); exit(1); }
|
|
|
|
static void hexRaw(const void* buffer, size_t size)
|
|
{
|
|
const unsigned char* p = (const unsigned char*)buffer;
|
|
for (size_t i=0; i<size; i++) {
|
|
printf("%02X", p[i]);
|
|
}
|
|
}
|
|
|
|
void hexDisp(const void* buffer, size_t size)
|
|
{
|
|
hexRaw(buffer, size);
|
|
printf("\n");
|
|
}
|
|
|
|
static void printHash(const void* table, size_t n, Htype_e htype)
|
|
{
|
|
if ((htype == ht64) || (htype == ht32)){
|
|
uint64_t const h64 = ((const uint64_t*)table)[n];
|
|
hexRaw(&h64, sizeof(h64));
|
|
} else {
|
|
assert(htype == ht128);
|
|
XXH128_hash_t const h128 = ((const XXH128_hash_t*)table)[n];
|
|
hexRaw(&h128, sizeof(h128));
|
|
}
|
|
}
|
|
|
|
/* === Generate Random unique Samples to hash === */
|
|
|
|
/*
|
|
* These functions will generate and update a sample to hash.
|
|
* initSample() will fill a buffer with random bytes,
|
|
* updateSample() will modify one slab in the input buffer.
|
|
* updateSample() guarantees it will produce unique samples,
|
|
* but it needs to know the total number of samples.
|
|
*/
|
|
|
|
|
|
static const uint64_t prime64_1 = 11400714785074694791ULL; /* 0b1001111000110111011110011011000110000101111010111100101010000111 */
|
|
static const uint64_t prime64_2 = 14029467366897019727ULL; /* 0b1100001010110010101011100011110100100111110101001110101101001111 */
|
|
static const uint64_t prime64_3 = 1609587929392839161ULL; /* 0b0001011001010110011001111011000110011110001101110111100111111001 */
|
|
|
|
static uint64_t avalanche64(uint64_t h64)
|
|
{
|
|
h64 ^= h64 >> 33;
|
|
h64 *= prime64_2;
|
|
h64 ^= h64 >> 29;
|
|
h64 *= prime64_3;
|
|
h64 ^= h64 >> 32;
|
|
return h64;
|
|
}
|
|
|
|
static unsigned char randomByte(size_t n)
|
|
{
|
|
uint64_t n64 = avalanche64(n+1);
|
|
n64 *= prime64_1;
|
|
return (unsigned char)(n64 >> 56);
|
|
}
|
|
|
|
typedef enum { sf_slab5, sf_sparse } sf_genMode;
|
|
|
|
|
|
#ifdef SLAB5
|
|
|
|
/*
|
|
* Slab5 sample generation.
|
|
* This algorithm generates unique inputs flipping on average 16 bits per candidate.
|
|
* It is generally much more friendly for most hash algorithms, especially
|
|
* weaker ones, as it shuffles more the input.
|
|
* The algorithm also avoids overfitting the per4 or per8 ingestion patterns.
|
|
*/
|
|
|
|
#define SLAB_SIZE 5
|
|
|
|
typedef struct {
|
|
void* buffer;
|
|
size_t size;
|
|
sf_genMode mode;
|
|
size_t prngSeed;
|
|
uint64_t hnb;
|
|
} sampleFactory;
|
|
|
|
static void init_sampleFactory(sampleFactory* sf, uint64_t htotal)
|
|
{
|
|
uint64_t const minNbSlabs = ((htotal-1) >> 32) + 1;
|
|
uint64_t const minSize = minNbSlabs * SLAB_SIZE;
|
|
if (sf->size < minSize)
|
|
EXIT("sample size must be >= %i bytes for this amount of hashes",
|
|
(int)minSize);
|
|
|
|
unsigned char* const p = (unsigned char*)sf->buffer;
|
|
for (size_t n=0; n < sf->size; n++)
|
|
p[n] = randomByte(n);
|
|
sf->hnb = 0;
|
|
}
|
|
|
|
static sampleFactory*
|
|
create_sampleFactory(size_t size, uint64_t htotal, uint64_t seed)
|
|
{
|
|
sampleFactory* const sf = malloc(sizeof(sampleFactory));
|
|
if (!sf) EXIT("not enough memory");
|
|
void* const buffer = malloc(size);
|
|
if (!buffer) EXIT("not enough memory");
|
|
sf->buffer = buffer;
|
|
sf->size = size;
|
|
sf->mode = sf_slab5;
|
|
sf->prngSeed = seed;
|
|
init_sampleFactory(sf, htotal);
|
|
return sf;
|
|
}
|
|
|
|
static void free_sampleFactory(sampleFactory* sf)
|
|
{
|
|
if (!sf) return;
|
|
free(sf->buffer);
|
|
free(sf);
|
|
}
|
|
|
|
static inline void update_sampleFactory(sampleFactory* sf)
|
|
{
|
|
size_t const nbSlabs = sf->size / SLAB_SIZE;
|
|
size_t const SlabNb = sf->hnb % nbSlabs;
|
|
sf->hnb++;
|
|
|
|
char* const ptr = (char*)sf->buffer;
|
|
size_t const start = (SlabNb * SLAB_SIZE) + 1;
|
|
uint32_t val32;
|
|
memcpy(&val32, ptr+start, sizeof(val32));
|
|
static const uint32_t prime32_5 = 374761393U;
|
|
val32 += prime32_5;
|
|
memcpy(ptr+start, &val32, sizeof(val32));
|
|
}
|
|
|
|
#else
|
|
|
|
/*
|
|
* Sparse sample generation.
|
|
* This is the default pattern generator.
|
|
* It only flips one bit at a time (mostly).
|
|
* Low hamming distance scenario is more difficult for weak hash algorithms.
|
|
* Note that CRC is immune to this scenario, since they are specifically
|
|
* designed to detect low hamming distances.
|
|
* Prefer the Slab5 pattern generator for collisions on CRC algorithms.
|
|
*/
|
|
|
|
#define SPARSE_LEVEL_MAX 15
|
|
|
|
/* Nb of combinations of m bits in a register of n bits */
|
|
static double Cnm(int n, int m)
|
|
{
|
|
assert(n > 0);
|
|
assert(m > 0);
|
|
assert(m <= m);
|
|
double acc = 1;
|
|
for (int i=0; i<m; i++) {
|
|
acc *= n - i;
|
|
acc /= 1 + i;
|
|
}
|
|
return acc;
|
|
}
|
|
|
|
static int enoughCombos(size_t size, uint64_t htotal)
|
|
{
|
|
if (size < 2) return 0; /* ensure no multiplication by negative */
|
|
uint64_t acc = 0;
|
|
uint64_t const srcBits = size * 8; assert(srcBits < INT_MAX);
|
|
int nbBitsSet = 0;
|
|
while (acc < htotal) {
|
|
nbBitsSet++;
|
|
if (nbBitsSet >= SPARSE_LEVEL_MAX) return 0;
|
|
acc += (uint64_t)Cnm((int)srcBits, nbBitsSet);
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
typedef struct {
|
|
void* buffer;
|
|
size_t size;
|
|
sf_genMode mode;
|
|
/* sparse */
|
|
size_t bitIdx[SPARSE_LEVEL_MAX];
|
|
int level;
|
|
size_t maxBitIdx;
|
|
/* slab5 */
|
|
size_t nbSlabs;
|
|
size_t current;
|
|
size_t prngSeed;
|
|
} sampleFactory;
|
|
|
|
static void init_sampleFactory(sampleFactory* sf, uint64_t htotal)
|
|
{
|
|
if (!enoughCombos(sf->size, htotal)) {
|
|
EXIT("sample size must be larger for this amount of hashes");
|
|
}
|
|
|
|
memset(sf->bitIdx, 0, sizeof(sf->bitIdx));
|
|
sf->level = 0;
|
|
|
|
unsigned char* const p = (unsigned char*)sf->buffer;
|
|
for (size_t n=0; n<sf->size; n++)
|
|
p[n] = randomByte(sf->prngSeed + n);
|
|
}
|
|
|
|
static sampleFactory*
|
|
create_sampleFactory(size_t size, uint64_t htotal, uint64_t seed)
|
|
{
|
|
sampleFactory* const sf = malloc(sizeof(sampleFactory));
|
|
if (!sf) EXIT("not enough memory");
|
|
void* const buffer = malloc(size);
|
|
if (!buffer) EXIT("not enough memory");
|
|
sf->buffer = buffer;
|
|
sf->size = size;
|
|
sf->mode = sf_sparse;
|
|
sf->maxBitIdx = size * 8;
|
|
sf->prngSeed = seed;
|
|
init_sampleFactory(sf, htotal);
|
|
return sf;
|
|
}
|
|
|
|
static void free_sampleFactory(sampleFactory* sf)
|
|
{
|
|
if (!sf) return;
|
|
free(sf->buffer);
|
|
free(sf);
|
|
}
|
|
|
|
static void flipbit(void* buffer, uint64_t bitID)
|
|
{
|
|
size_t const pos = bitID >> 3;
|
|
unsigned char const mask = (unsigned char)(1 << (bitID & 7));
|
|
unsigned char* const p = (unsigned char*)buffer;
|
|
p[pos] ^= mask;
|
|
}
|
|
|
|
static int updateBit(void* buffer, size_t* bitIdx, int level, size_t max)
|
|
{
|
|
if (level==0) return 0; /* can't progress further */
|
|
|
|
flipbit(buffer, bitIdx[level]); /* erase previous bits */
|
|
|
|
if (bitIdx[level] < max-1) { /* simple case: go to next bit */
|
|
bitIdx[level]++;
|
|
flipbit(buffer, bitIdx[level]); /* set new bit */
|
|
return 1;
|
|
}
|
|
|
|
/* reached last bit: need to update a bit from lower level */
|
|
if (!updateBit(buffer, bitIdx, level-1, max-1)) return 0;
|
|
bitIdx[level] = bitIdx[level-1] + 1;
|
|
flipbit(buffer, bitIdx[level]); /* set new bit */
|
|
return 1;
|
|
}
|
|
|
|
static inline void update_sampleFactory(sampleFactory* sf)
|
|
{
|
|
if (!updateBit(sf->buffer, sf->bitIdx, sf->level, sf->maxBitIdx)) {
|
|
/* no more room => move to next level */
|
|
sf->level++;
|
|
assert(sf->level < SPARSE_LEVEL_MAX);
|
|
|
|
/* set new bits */
|
|
for (int i=1; i <= sf->level; i++) {
|
|
sf->bitIdx[i] = (size_t)(i-1);
|
|
flipbit(sf->buffer, sf->bitIdx[i]);
|
|
}
|
|
}
|
|
}
|
|
|
|
#endif /* pattern generator selection */
|
|
|
|
|
|
/* === Candidate Filter === */
|
|
|
|
typedef unsigned char Filter;
|
|
|
|
Filter* create_Filter(int bflog)
|
|
{
|
|
assert(bflog < 64 && bflog > 1);
|
|
size_t bfsize = (size_t)1 << bflog;
|
|
Filter* bf = malloc(bfsize);
|
|
assert(((void)"Filter creation failed", bf));
|
|
memset(bf, 0, bfsize);
|
|
return bf;
|
|
}
|
|
|
|
void free_Filter(Filter* bf)
|
|
{
|
|
free(bf);
|
|
}
|
|
|
|
#ifdef FILTER_1_PROBE
|
|
|
|
/*
|
|
* Attach hash to a slot
|
|
* return: Nb of potential collision candidates detected
|
|
* 0: position not yet occupied
|
|
* 2: position previously occupied by a single candidate
|
|
* 1: position already occupied by multiple candidates
|
|
*/
|
|
inline int Filter_insert(Filter* bf, int bflog, uint64_t hash)
|
|
{
|
|
int const slotNb = hash & 3;
|
|
int const shift = slotNb * 2 ;
|
|
|
|
size_t const bfmask = ((size_t)1 << bflog) - 1;
|
|
size_t const pos = (hash >> 2) & bfmask;
|
|
|
|
int const existingCandidates = ((((unsigned char*)bf)[pos]) >> shift) & 3;
|
|
|
|
static const int addCandidates[4] = { 0, 2, 1, 1 };
|
|
static const int nextValue[4] = { 1, 2, 3, 3 };
|
|
|
|
((unsigned char*)bf)[pos] |= (unsigned char)(nextValue[existingCandidates] << shift);
|
|
return addCandidates[existingCandidates];
|
|
}
|
|
|
|
/*
|
|
* Check if provided 64-bit hash is a collision candidate
|
|
* Requires the slot to be occupied by at least 2 candidates.
|
|
* return >0 if hash is a collision candidate
|
|
* 0 otherwise (slot unoccupied, or only one candidate)
|
|
* note: unoccupied slots should not happen in this algorithm,
|
|
* since all hashes are supposed to have been inserted at least once.
|
|
*/
|
|
inline int Filter_check(const Filter* bf, int bflog, uint64_t hash)
|
|
{
|
|
int const slotNb = hash & 3;
|
|
int const shift = slotNb * 2;
|
|
|
|
size_t const bfmask = ((size_t)1 << bflog) - 1;
|
|
size_t const pos = (hash >> 2) & bfmask;
|
|
|
|
return (((const unsigned char*)bf)[pos]) >> (shift+1) & 1;
|
|
}
|
|
|
|
#else
|
|
|
|
/*
|
|
* 2-probes strategy,
|
|
* more efficient at filtering candidates,
|
|
* requires filter size to be > nb of hashes
|
|
*/
|
|
|
|
#define MIN(a,b) ((a) < (b) ? (a) : (b))
|
|
#define MAX(a,b) ((a) > (b) ? (a) : (b))
|
|
|
|
/*
|
|
* Attach hash to 2 slots
|
|
* return: Nb of potential candidates detected
|
|
* 0: position not yet occupied
|
|
* 2: position previously occupied by a single candidate (at most)
|
|
* 1: position already occupied by multiple candidates
|
|
*/
|
|
static inline int Filter_insert(Filter* bf, int bflog, uint64_t hash)
|
|
{
|
|
hash = avalanche64(hash);
|
|
unsigned const slot1 = hash & 255;
|
|
hash >>= 8;
|
|
unsigned const slot2 = hash & 255;
|
|
hash >>= 8;
|
|
|
|
size_t const fclmask = ((size_t)1 << (bflog-6)) - 1;
|
|
size_t const cacheLineNb = hash & fclmask;
|
|
|
|
size_t const pos1 = (cacheLineNb << 6) + (slot1 >> 2);
|
|
unsigned const shift1 = (slot1 & 3) * 2;
|
|
unsigned const ex1 = (bf[pos1] >> shift1) & 3;
|
|
|
|
size_t const pos2 = (cacheLineNb << 6) + (slot2 >> 2);
|
|
unsigned const shift2 = (slot2 & 3) * 2;
|
|
unsigned const ex2 = (bf[pos2] >> shift2) & 3;
|
|
|
|
unsigned const existing = MIN(ex1, ex2);
|
|
|
|
static const int addCandidates[4] = { 0, 2, 1, 1 };
|
|
static const unsigned nextValue[4] = { 1, 2, 3, 3 };
|
|
|
|
bf[pos1] &= (Filter)(~(3 << shift1)); /* erase previous value */
|
|
bf[pos1] |= (Filter)(MAX(ex1, nextValue[existing]) << shift1);
|
|
bf[pos2] |= (Filter)(MAX(ex2, nextValue[existing]) << shift2);
|
|
|
|
return addCandidates[existing];
|
|
}
|
|
|
|
|
|
/*
|
|
* Check if provided 64-bit hash is a collision candidate
|
|
* Requires the slot to be occupied by at least 2 candidates.
|
|
* return >0 if hash is a collision candidate
|
|
* 0 otherwise (slot unoccupied, or only one candidate)
|
|
* note: unoccupied slots should not happen in this algorithm,
|
|
* since all hashes are supposed to have been inserted at least once.
|
|
*/
|
|
static inline int Filter_check(const Filter* bf, int bflog, uint64_t hash)
|
|
{
|
|
hash = avalanche64(hash);
|
|
unsigned const slot1 = hash & 255;
|
|
hash >>= 8;
|
|
unsigned const slot2 = hash & 255;
|
|
hash >>= 8;
|
|
|
|
size_t const fclmask = ((size_t)1 << (bflog-6)) - 1;
|
|
size_t const cacheLineNb = hash & fclmask;
|
|
|
|
size_t const pos1 = (cacheLineNb << 6) + (slot1 >> 2);
|
|
unsigned const shift1 = (slot1 & 3) * 2;
|
|
unsigned const ex1 = (bf[pos1] >> shift1) & 3;
|
|
|
|
size_t const pos2 = (cacheLineNb << 6) + (slot2 >> 2);
|
|
unsigned const shift2 = (slot2 & 3) * 2;
|
|
unsigned const ex2 = (bf[pos2] >> shift2) & 3;
|
|
|
|
return (ex1 >= 2) && (ex2 >= 2);
|
|
}
|
|
|
|
#endif // FILTER_1_PROBE
|
|
|
|
|
|
/* === Display === */
|
|
|
|
#include <time.h> /* clock_t, clock, time_t, time, difftime */
|
|
|
|
void update_indicator(uint64_t v, uint64_t total)
|
|
{
|
|
static clock_t start = 0;
|
|
if (start==0) start = clock();
|
|
clock_t const updateRate = CLOCKS_PER_SEC / 2;
|
|
|
|
clock_t const clockSpan = (clock_t)(clock() - start);
|
|
if (clockSpan > updateRate) {
|
|
start = clock();
|
|
assert(v <= total);
|
|
assert(total > 0);
|
|
double share = ((double)v / (double)total) * 100;
|
|
printf("%6.2f%% (%llu) \r", share, (unsigned long long)v);
|
|
fflush(NULL);
|
|
}
|
|
}
|
|
|
|
/* note: not thread safe */
|
|
const char* displayDelay(double delay_s)
|
|
{
|
|
static char delayString[50];
|
|
memset(delayString, 0, sizeof(delayString));
|
|
|
|
int const mn = ((int)delay_s / 60) % 60;
|
|
int const h = (int)delay_s / 3600;
|
|
int const sec = (int)delay_s % 60;
|
|
|
|
char* p = delayString;
|
|
if (h) sprintf(p, "%i h ", h);
|
|
if (mn || h) {
|
|
p = delayString + strlen(delayString);
|
|
sprintf(p, "%i mn ", mn);
|
|
}
|
|
p = delayString + strlen(delayString);
|
|
sprintf(p, "%is ", sec);
|
|
|
|
return delayString;
|
|
}
|
|
|
|
|
|
/* === Math === */
|
|
|
|
static double power(uint64_t base, int p)
|
|
{
|
|
double value = 1;
|
|
assert(p>=0);
|
|
for (int i=0; i<p; i++) {
|
|
value *= (double)base;
|
|
}
|
|
return value;
|
|
}
|
|
|
|
static double estimateNbCollisions(uint64_t nbH, int nbBits)
|
|
{
|
|
return ((double)nbH * (double)(nbH-1)) / power(2, nbBits+1);
|
|
}
|
|
|
|
static int highestBitSet(uint64_t v)
|
|
{
|
|
assert(v!=0);
|
|
int bitId = 0;
|
|
while (v >>= 1) bitId++;
|
|
return bitId;
|
|
}
|
|
|
|
|
|
/* === Filter and search collisions === */
|
|
|
|
#undef NDEBUG /* ensure assert is not disabled */
|
|
#include <assert.h>
|
|
|
|
/* will recommend 24 billion samples for 64-bit hashes,
|
|
* expecting 18 collisions for a good 64-bit hash */
|
|
#define NB_BITS_MAX 64 /* can't store nor analyze hash wider than 64-bits for the time being */
|
|
uint64_t select_nbh(int nbBits)
|
|
{
|
|
assert(nbBits > 0);
|
|
if (nbBits > NB_BITS_MAX) nbBits = NB_BITS_MAX;
|
|
double targetColls = (double)((128 + 17) - (nbBits * 2));
|
|
uint64_t nbH = 24;
|
|
while (estimateNbCollisions(nbH, nbBits) < targetColls) nbH *= 2;
|
|
return nbH;
|
|
}
|
|
|
|
|
|
typedef struct {
|
|
uint64_t nbCollisions;
|
|
} searchCollisions_results;
|
|
|
|
typedef struct {
|
|
uint64_t nbH;
|
|
uint64_t mask;
|
|
uint64_t maskSelector;
|
|
size_t sampleSize;
|
|
uint64_t prngSeed;
|
|
int filterLog; /* <0 = disable filter; 0 = auto-size; */
|
|
int hashID;
|
|
int display;
|
|
int nbThreads;
|
|
searchCollisions_results* resultPtr;
|
|
} searchCollisions_parameters;
|
|
|
|
#define DISPLAY(...) { if (display) printf(__VA_ARGS__); }
|
|
|
|
static int isEqual(void* hTablePtr, size_t index1, size_t index2, Htype_e htype)
|
|
{
|
|
if ((htype == ht64) || (htype == ht32)) {
|
|
uint64_t const h1 = ((const uint64_t*)hTablePtr)[index1];
|
|
uint64_t const h2 = ((const uint64_t*)hTablePtr)[index2];
|
|
return (h1 == h2);
|
|
} else {
|
|
assert(htype == ht128);
|
|
XXH128_hash_t const h1 = ((const XXH128_hash_t*)hTablePtr)[index1];
|
|
XXH128_hash_t const h2 = ((const XXH128_hash_t*)hTablePtr)[index2];
|
|
return XXH128_isEqual(h1, h2);
|
|
}
|
|
}
|
|
|
|
static int isHighEqual(void* hTablePtr, size_t index1, size_t index2, Htype_e htype, int rShift)
|
|
{
|
|
uint64_t h1, h2;
|
|
if ((htype == ht64) || (htype == ht32)) {
|
|
h1 = ((const uint64_t*)hTablePtr)[index1];
|
|
h2 = ((const uint64_t*)hTablePtr)[index2];
|
|
} else {
|
|
assert(htype == ht128);
|
|
h1 = ((const XXH128_hash_t*)hTablePtr)[index1].high64;
|
|
h2 = ((const XXH128_hash_t*)hTablePtr)[index2].high64;
|
|
assert(rShift >= 64);
|
|
rShift -= 64;
|
|
}
|
|
assert(0 <= rShift && rShift < 64);
|
|
return (h1 >> rShift) == (h2 >> rShift);
|
|
}
|
|
|
|
/* assumption: (htype*)hTablePtr[index] is valid */
|
|
static void addHashCandidate(void* hTablePtr, UniHash h, Htype_e htype, size_t index)
|
|
{
|
|
if ((htype == ht64) || (htype == ht32)) {
|
|
((uint64_t*)hTablePtr)[index] = h.h64;
|
|
} else {
|
|
assert(htype == ht128);
|
|
((XXH128_hash_t*)hTablePtr)[index] = h.h128;
|
|
}
|
|
}
|
|
|
|
static int getNbBits_fromHtype(Htype_e htype) {
|
|
switch(htype) {
|
|
case ht32: return 32;
|
|
case ht64: return 64;
|
|
case ht128:return 128;
|
|
default: EXIT("hash size not supported");
|
|
}
|
|
}
|
|
|
|
static Htype_e getHtype_fromHbits(int nbBits) {
|
|
switch(nbBits) {
|
|
case 32 : return ht32;
|
|
case 64 : return ht64;
|
|
case 128: return ht128;
|
|
default: EXIT("hash size not supported");
|
|
}
|
|
}
|
|
|
|
static size_t search_collisions(
|
|
searchCollisions_parameters param)
|
|
{
|
|
uint64_t totalH = param.nbH;
|
|
const uint64_t hMask = param.mask;
|
|
const uint64_t hSelector = param.maskSelector;
|
|
int bflog = param.filterLog;
|
|
const int filter = (param.filterLog >= 0);
|
|
const size_t sampleSize = param.sampleSize;
|
|
const int hashID = param.hashID;
|
|
const Htype_e htype = getHtype_fromHbits(hashfnTable[hashID].bits);
|
|
const int display = param.display;
|
|
/* init */
|
|
sampleFactory* const sf = create_sampleFactory(sampleSize, totalH, param.prngSeed);
|
|
if (!sf) EXIT("not enough memory");
|
|
|
|
//const char* const hname = hashfnTable[hashID].name;
|
|
hashfn const hfunction = hashfnTable[hashID].fn;
|
|
int const hwidth = hashfnTable[hashID].bits;
|
|
if (totalH == 0) totalH = select_nbh(hwidth);
|
|
if (bflog == 0) bflog = highestBitSet(totalH) + 1; /* auto-size filter */
|
|
uint64_t const bfsize = (1ULL << bflog);
|
|
|
|
|
|
/* === filter hashes (optional) === */
|
|
|
|
Filter* bf = NULL;
|
|
uint64_t nbPresents = totalH;
|
|
|
|
if (filter) {
|
|
time_t const filterTBegin = time(NULL);
|
|
DISPLAY(" Creating filter (%i GB) \n", (int)(bfsize >> 30));
|
|
bf = create_Filter(bflog);
|
|
if (!bf) EXIT("not enough memory for filter");
|
|
|
|
|
|
DISPLAY(" Generate %llu hashes from samples of %u bytes \n",
|
|
(unsigned long long)totalH, (unsigned)sampleSize);
|
|
nbPresents = 0;
|
|
|
|
for (uint64_t n=0; n < totalH; n++) {
|
|
if (display && ((n&0xFFFFF) == 1) )
|
|
update_indicator(n, totalH);
|
|
update_sampleFactory(sf);
|
|
|
|
UniHash const h = hfunction(sf->buffer, sampleSize);
|
|
if ((h.h64 & hMask) != hSelector) continue;
|
|
|
|
nbPresents += (uint64_t)Filter_insert(bf, bflog, h.h64);
|
|
}
|
|
|
|
if (nbPresents==0) {
|
|
DISPLAY(" Analysis completed: No collision detected \n");
|
|
if (param.resultPtr) param.resultPtr->nbCollisions = 0;
|
|
free_Filter(bf);
|
|
free_sampleFactory(sf);
|
|
return 0;
|
|
}
|
|
|
|
{ double const filterDelay = difftime(time(NULL), filterTBegin);
|
|
DISPLAY(" Generation and filter completed in %s, detected up to %llu candidates \n",
|
|
displayDelay(filterDelay), (unsigned long long) nbPresents);
|
|
} }
|
|
|
|
|
|
/* === store hash candidates: duplicates will be present here === */
|
|
|
|
time_t const storeTBegin = time(NULL);
|
|
size_t const hashByteSize = (htype == ht128) ? 16 : 8;
|
|
size_t const tableSize = (nbPresents+1) * hashByteSize;
|
|
assert(tableSize > nbPresents); /* check tableSize calculation overflow */
|
|
DISPLAY(" Storing hash candidates (%i MB) \n", (int)(tableSize >> 20));
|
|
|
|
/* Generate and store hashes */
|
|
void* const hashCandidates = malloc(tableSize);
|
|
if (!hashCandidates) EXIT("not enough memory to store candidates");
|
|
init_sampleFactory(sf, totalH);
|
|
size_t nbCandidates = 0;
|
|
for (uint64_t n=0; n < totalH; n++) {
|
|
if (display && ((n&0xFFFFF) == 1) ) update_indicator(n, totalH);
|
|
update_sampleFactory(sf);
|
|
|
|
UniHash const h = hfunction(sf->buffer, sampleSize);
|
|
if ((h.h64 & hMask) != hSelector) continue;
|
|
|
|
if (filter) {
|
|
if (Filter_check(bf, bflog, h.h64)) {
|
|
assert(nbCandidates < nbPresents);
|
|
addHashCandidate(hashCandidates, h, htype, nbCandidates++);
|
|
}
|
|
} else {
|
|
assert(nbCandidates < nbPresents);
|
|
addHashCandidate(hashCandidates, h, htype, nbCandidates++);
|
|
}
|
|
}
|
|
if (nbCandidates < nbPresents) {
|
|
/* Try to mitigate gnuc_quicksort behavior, by reducing allocated memory,
|
|
* since gnuc_quicksort uses a lot of additional memory for mergesort */
|
|
void* const checkPtr = realloc(hashCandidates, nbCandidates * hashByteSize);
|
|
assert(checkPtr != NULL);
|
|
assert(checkPtr == hashCandidates); /* simplification: since we are reducing the size,
|
|
* we hope to keep the same ptr position.
|
|
* Otherwise, hashCandidates must be mutable. */
|
|
DISPLAY(" List of hashes reduced to %u MB from %u MB (saved %u MB) \n",
|
|
(unsigned)((nbCandidates * hashByteSize) >> 20),
|
|
(unsigned)(tableSize >> 20),
|
|
(unsigned)((tableSize - (nbCandidates * hashByteSize)) >> 20) );
|
|
}
|
|
double const storeTDelay = difftime(time(NULL), storeTBegin);
|
|
DISPLAY(" Stored %llu hash candidates in %s \n",
|
|
(unsigned long long) nbCandidates, displayDelay(storeTDelay));
|
|
free_Filter(bf);
|
|
free_sampleFactory(sf);
|
|
|
|
|
|
/* === step 3: look for duplicates === */
|
|
time_t const sortTBegin = time(NULL);
|
|
DISPLAY(" Sorting candidates... ");
|
|
fflush(NULL);
|
|
if ((htype == ht64) || (htype == ht32)) {
|
|
/*
|
|
* Use C++'s std::sort, as it's faster than C stdlib's qsort, and
|
|
* doesn't suffer from gnuc_libsort's memory expansion
|
|
*/
|
|
sort64(hashCandidates, nbCandidates);
|
|
} else {
|
|
assert(htype == ht128);
|
|
sort128(hashCandidates, nbCandidates); /* sort with custom comparator */
|
|
}
|
|
double const sortTDelay = difftime(time(NULL), sortTBegin);
|
|
DISPLAY(" Completed in %s \n", displayDelay(sortTDelay));
|
|
|
|
/* scan and count duplicates */
|
|
time_t const countBegin = time(NULL);
|
|
DISPLAY(" Looking for duplicates: ");
|
|
fflush(NULL);
|
|
size_t collisions = 0;
|
|
for (size_t n=1; n<nbCandidates; n++) {
|
|
if (isEqual(hashCandidates, n, n-1, htype)) {
|
|
printf("collision: ");
|
|
printHash(hashCandidates, n, htype);
|
|
printf(" / ");
|
|
printHash(hashCandidates, n-1, htype);
|
|
printf(" \n");
|
|
collisions++;
|
|
} }
|
|
|
|
if (!filter /* all candidates */ && display /*single thead*/ ) {
|
|
/* check partial bitfields (high bits) */
|
|
DISPLAY(" \n");
|
|
int const hashBits = getNbBits_fromHtype(htype);
|
|
double worstRatio = 0.;
|
|
int worstNbHBits = 0;
|
|
for (int nbHBits = 1; nbHBits < hashBits; nbHBits++) {
|
|
uint64_t const nbSlots = (uint64_t)1 << nbHBits;
|
|
double const expectedCollisions = estimateNbCollisions(nbCandidates, nbHBits);
|
|
if ( (nbSlots > nbCandidates * 100) /* within range for meaningfull collision analysis results */
|
|
&& (expectedCollisions > 18.0) ) {
|
|
int const rShift = hashBits - nbHBits;
|
|
size_t HBits_collisions = 0;
|
|
for (size_t n=1; n<nbCandidates; n++) {
|
|
if (isHighEqual(hashCandidates, n, n-1, htype, rShift)) {
|
|
HBits_collisions++;
|
|
} }
|
|
double const collisionRatio = (double)HBits_collisions / expectedCollisions;
|
|
if (collisionRatio > 2.0) DISPLAY("WARNING !!! ===> ");
|
|
DISPLAY(" high %i bits: %zu collision (%.1f expected): x%.2f \n",
|
|
nbHBits, HBits_collisions, expectedCollisions, collisionRatio);
|
|
if (collisionRatio > worstRatio) {
|
|
worstNbHBits = nbHBits;
|
|
worstRatio = collisionRatio;
|
|
} } }
|
|
DISPLAY("Worst collision ratio at %i high bits: x%.2f \n",
|
|
worstNbHBits, worstRatio);
|
|
}
|
|
double const countDelay = difftime(time(NULL), countBegin);
|
|
DISPLAY(" Completed in %s \n", displayDelay(countDelay));
|
|
|
|
/* clean and exit */
|
|
free (hashCandidates);
|
|
|
|
#if 0 /* debug */
|
|
for (size_t n=0; n<nbCandidates; n++)
|
|
printf("0x%016llx \n", (unsigned long long)hashCandidates[n]);
|
|
#endif
|
|
|
|
if (param.resultPtr) param.resultPtr->nbCollisions = collisions;
|
|
return collisions;
|
|
}
|
|
|
|
|
|
|
|
#if defined(__MACH__) || defined(__linux__)
|
|
#include <sys/resource.h>
|
|
static size_t getProcessMemUsage(int children)
|
|
{
|
|
struct rusage stats;
|
|
if (getrusage(children ? RUSAGE_CHILDREN : RUSAGE_SELF, &stats) == 0)
|
|
return (size_t)stats.ru_maxrss;
|
|
return 0;
|
|
}
|
|
#else
|
|
static size_t getProcessMemUsage(int ignore) { return 0; }
|
|
#endif
|
|
|
|
void time_collisions(searchCollisions_parameters param)
|
|
{
|
|
uint64_t totalH = param.nbH;
|
|
int hashID = param.hashID;
|
|
int display = param.display;
|
|
|
|
/* init */
|
|
assert(0 <= hashID && hashID < HASH_FN_TOTAL);
|
|
//const char* const hname = hashfnTable[hashID].name;
|
|
int const hwidth = hashfnTable[hashID].bits;
|
|
if (totalH == 0) totalH = select_nbh(hwidth);
|
|
double const targetColls = estimateNbCollisions(totalH, hwidth);
|
|
|
|
/* Start the timer to measure start/end of hashing + collision detection. */
|
|
time_t const programTBegin = time(NULL);
|
|
|
|
/* Generate hashes, and count collisions */
|
|
size_t const collisions = search_collisions(param);
|
|
|
|
/* display results */
|
|
double const programTDelay = difftime(time(NULL), programTBegin);
|
|
size_t const programBytesSelf = getProcessMemUsage(0);
|
|
size_t const programBytesChildren = getProcessMemUsage(1);
|
|
DISPLAY("\n\n");
|
|
DISPLAY("===> Found %llu collisions (x%.2f, %.1f expected) in %s\n",
|
|
(unsigned long long)collisions,
|
|
(double)collisions / targetColls,
|
|
targetColls,
|
|
displayDelay(programTDelay));
|
|
if (programBytesSelf)
|
|
DISPLAY("===> MaxRSS(self) %zuMB, MaxRSS(children) %zuMB\n",
|
|
programBytesSelf>>20,
|
|
programBytesChildren>>20);
|
|
DISPLAY("------------------------------------------ \n");
|
|
}
|
|
|
|
// wrapper for pthread interface
|
|
void MT_searchCollisions(void* payload)
|
|
{
|
|
search_collisions(*(searchCollisions_parameters*)payload);
|
|
}
|
|
|
|
/* === Command Line === */
|
|
|
|
/*!
|
|
* readU64FromChar():
|
|
* Allows and interprets K, KB, KiB, M, MB and MiB suffix.
|
|
* Will also modify `*stringPtr`, advancing it to the position where it stopped reading.
|
|
*/
|
|
static uint64_t readU64FromChar(const char** stringPtr)
|
|
{
|
|
static uint64_t const max = (((uint64_t)(-1)) / 10) - 1;
|
|
uint64_t result = 0;
|
|
while ((**stringPtr >='0') && (**stringPtr <='9')) {
|
|
assert(result < max);
|
|
result *= 10;
|
|
result += (unsigned)(**stringPtr - '0');
|
|
(*stringPtr)++ ;
|
|
}
|
|
if ((**stringPtr=='K') || (**stringPtr=='M') || (**stringPtr=='G')) {
|
|
uint64_t const maxK = ((uint64_t)(-1)) >> 10;
|
|
assert(result < maxK);
|
|
result <<= 10;
|
|
if ((**stringPtr=='M') || (**stringPtr=='G')) {
|
|
assert(result < maxK);
|
|
result <<= 10;
|
|
if (**stringPtr=='G') {
|
|
assert(result < maxK);
|
|
result <<= 10;
|
|
}
|
|
}
|
|
(*stringPtr)++; /* skip `K` or `M` */
|
|
if (**stringPtr=='i') (*stringPtr)++;
|
|
if (**stringPtr=='B') (*stringPtr)++;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
|
|
/**
|
|
* longCommandWArg():
|
|
* Checks if *stringPtr is the same as longCommand.
|
|
* If yes, @return 1 and advances *stringPtr to the position which immediately follows longCommand.
|
|
* @return 0 and doesn't modify *stringPtr otherwise.
|
|
*/
|
|
static int longCommandWArg(const char** stringPtr, const char* longCommand)
|
|
{
|
|
assert(longCommand); assert(stringPtr); assert(*stringPtr);
|
|
size_t const comSize = strlen(longCommand);
|
|
int const result = !strncmp(*stringPtr, longCommand, comSize);
|
|
if (result) *stringPtr += comSize;
|
|
return result;
|
|
}
|
|
|
|
|
|
#include "pool.h"
|
|
|
|
/*
|
|
* As some hashes use different algorithms depending on input size,
|
|
* it can be necessary to test multiple input sizes
|
|
* to paint an accurate picture of collision performance
|
|
*/
|
|
#define SAMPLE_SIZE_DEFAULT 256
|
|
#define HASHFN_ID_DEFAULT 0
|
|
|
|
void help(const char* exeName)
|
|
{
|
|
printf("usage: %s [hashName] [opt] \n\n", exeName);
|
|
printf("list of hashNames:");
|
|
printf("%s ", hashfnTable[0].name);
|
|
for (int i=1; i < HASH_FN_TOTAL; i++) {
|
|
printf(", %s ", hashfnTable[i].name);
|
|
}
|
|
printf(" \n");
|
|
printf("Default hashName is %s\n", hashfnTable[HASHFN_ID_DEFAULT].name);
|
|
|
|
printf(" \n");
|
|
printf("Optional parameters: \n");
|
|
printf(" --nbh=NB Select nb of hashes to generate (%llu by default) \n", (unsigned long long)select_nbh(64));
|
|
printf(" --filter Activates the filter. Slower, but reduces memory usage for the same nb of hashes.\n");
|
|
printf(" --threadlog=NB Use 2^NB threads.\n");
|
|
printf(" --len=MB Set length of the input (%i bytes by default) \n", SAMPLE_SIZE_DEFAULT);
|
|
}
|
|
|
|
int bad_argument(const char* exeName)
|
|
{
|
|
printf("incorrect command: \n");
|
|
help(exeName);
|
|
return 1;
|
|
}
|
|
|
|
|
|
int main(int argc, const char** argv)
|
|
{
|
|
if (sizeof(size_t) < 8) return 1; // cannot work on systems without ability to allocate objects >= 4 GB
|
|
|
|
assert(argc > 0);
|
|
const char* const exeName = argv[0];
|
|
uint64_t totalH = 0; /* auto, based on nbBits */
|
|
int bflog = 0; /* auto */
|
|
int filter = 0; /* disabled */
|
|
size_t sampleSize = SAMPLE_SIZE_DEFAULT;
|
|
int hashID = HASHFN_ID_DEFAULT;
|
|
int threadlog = 0;
|
|
uint64_t prngSeed = 0;
|
|
|
|
int arg_nb;
|
|
for (arg_nb = 1; arg_nb < argc; arg_nb++) {
|
|
const char** arg = argv + arg_nb;
|
|
|
|
if (!strcmp(*arg, "-h")) { help(exeName); return 0; }
|
|
if (longCommandWArg(arg, "-T")) { threadlog = (int)readU64FromChar(arg); continue; }
|
|
|
|
if (!strcmp(*arg, "--filter")) { filter=1; continue; }
|
|
if (!strcmp(*arg, "--no-filter")) { filter=0; continue; }
|
|
|
|
if (longCommandWArg(arg, "--seed")) { prngSeed = readU64FromChar(arg); continue; }
|
|
if (longCommandWArg(arg, "--nbh=")) { totalH = readU64FromChar(arg); continue; }
|
|
if (longCommandWArg(arg, "--filter=")) { filter=1; bflog = (int)readU64FromChar(arg); assert(bflog < 64); continue; }
|
|
if (longCommandWArg(arg, "--filterlog=")) { filter=1; bflog = (int)readU64FromChar(arg); assert(bflog < 64); continue; }
|
|
if (longCommandWArg(arg, "--size=")) { sampleSize = (size_t)readU64FromChar(arg); continue; }
|
|
if (longCommandWArg(arg, "--len=")) { sampleSize = (size_t)readU64FromChar(arg); continue; }
|
|
if (longCommandWArg(arg, "--threadlog=")) { threadlog = (int)readU64FromChar(arg); continue; }
|
|
|
|
/* argument understood as hash name (must be correct) */
|
|
int hnb;
|
|
for (hnb=0; hnb < HASH_FN_TOTAL; hnb++) {
|
|
if (!strcmp(*arg, hashfnTable[hnb].name)) { hashID = hnb; break; }
|
|
}
|
|
if (hnb == HASH_FN_TOTAL) return bad_argument(exeName);
|
|
}
|
|
|
|
/* init */
|
|
const char* const hname = hashfnTable[hashID].name;
|
|
int const hwidth = hashfnTable[hashID].bits;
|
|
if (totalH == 0) totalH = select_nbh(hwidth);
|
|
double const targetColls = estimateNbCollisions(totalH, hwidth);
|
|
if (bflog == 0) bflog = highestBitSet(totalH) + 1; /* auto-size filter */
|
|
if (!filter) bflog = -1; // disable filter
|
|
|
|
if (sizeof(size_t) < 8)
|
|
EXIT("This program has not been validated on architectures other than "
|
|
"64bit \n");
|
|
|
|
printf(" *** Collision tester for 64+ bit hashes *** \n\n");
|
|
printf("Testing %s algorithm (%i-bit) \n", hname, hwidth);
|
|
printf("This program will allocate a lot of memory,\n");
|
|
printf("generate %llu %i-bit hashes from samples of %u bytes, \n",
|
|
(unsigned long long)totalH, hwidth, (unsigned)sampleSize);
|
|
printf("and attempt to produce %.0f collisions. \n\n", targetColls);
|
|
|
|
int const nbThreads = 1 << threadlog;
|
|
if (nbThreads <= 0) EXIT("Invalid --threadlog value.");
|
|
|
|
if (nbThreads == 1) {
|
|
|
|
searchCollisions_parameters params;
|
|
params.nbH = totalH;
|
|
params.mask = 0;
|
|
params.maskSelector = 0;
|
|
params.sampleSize = sampleSize;
|
|
params.filterLog = bflog;
|
|
params.hashID = hashID;
|
|
params.display = 1;
|
|
params.resultPtr = NULL;
|
|
params.prngSeed = prngSeed;
|
|
params.nbThreads = 1;
|
|
time_collisions(params);
|
|
|
|
} else { /* nbThreads > 1 */
|
|
|
|
/* use multithreading */
|
|
if (threadlog >= 30) EXIT("too many threads requested");
|
|
if ((uint64_t)nbThreads > (totalH >> 16))
|
|
EXIT("too many threads requested");
|
|
if (bflog > 0 && threadlog > (bflog-10))
|
|
EXIT("too many threads requested");
|
|
printf("using %i threads ... \n", nbThreads);
|
|
|
|
/* allocation */
|
|
time_t const programTBegin = time(NULL);
|
|
POOL_ctx* const pt = POOL_create((size_t)nbThreads, 1);
|
|
if (!pt) EXIT("not enough memory for threads");
|
|
searchCollisions_results* const MTresults = calloc (sizeof(searchCollisions_results), (size_t)nbThreads);
|
|
if (!MTresults) EXIT("not enough memory");
|
|
searchCollisions_parameters* const MTparams = calloc (sizeof(searchCollisions_parameters), (size_t)nbThreads);
|
|
if (!MTparams) EXIT("not enough memory");
|
|
|
|
/* distribute jobs */
|
|
for (int tnb=0; tnb<nbThreads; tnb++) {
|
|
MTparams[tnb].nbH = totalH;
|
|
MTparams[tnb].mask = (uint64_t)nbThreads - 1;
|
|
MTparams[tnb].sampleSize = sampleSize;
|
|
MTparams[tnb].filterLog = bflog ? bflog - threadlog : 0;
|
|
MTparams[tnb].hashID = hashID;
|
|
MTparams[tnb].display = 0;
|
|
MTparams[tnb].resultPtr = MTresults+tnb;
|
|
MTparams[tnb].prngSeed = prngSeed;
|
|
MTparams[tnb].maskSelector = (uint64_t)tnb;
|
|
POOL_add(pt, MT_searchCollisions, MTparams + tnb);
|
|
}
|
|
POOL_free(pt); /* actually joins and free */
|
|
|
|
/* Gather results */
|
|
uint64_t nbCollisions=0;
|
|
for (int tnb=0; tnb<nbThreads; tnb++) {
|
|
nbCollisions += MTresults[tnb].nbCollisions;
|
|
}
|
|
|
|
double const programTDelay = difftime(time(NULL), programTBegin);
|
|
size_t const programBytesSelf = getProcessMemUsage(0);
|
|
size_t const programBytesChildren = getProcessMemUsage(1);
|
|
printf("\n\n");
|
|
printf("===> Found %llu collisions (x%.2f, %.1f expected) in %s\n",
|
|
(unsigned long long)nbCollisions,
|
|
(double)nbCollisions / targetColls,
|
|
targetColls,
|
|
displayDelay(programTDelay));
|
|
if (programBytesSelf)
|
|
printf("===> MaxRSS(self) %zuMB, MaxRSS(children) %zuMB\n",
|
|
programBytesSelf>>20,
|
|
programBytesChildren>>20);
|
|
printf("------------------------------------------ \n");
|
|
|
|
/* Clean up */
|
|
free(MTparams);
|
|
free(MTresults);
|
|
}
|
|
|
|
return 0;
|
|
}
|