inga-lovinde
/
TrustPilotChallenge
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

FillPhraseSet optimizations

Browse Source
master
Inga 🏳‍🌈 7 years ago
parent bb22805cbc
commit 2d1dcc132c
1 changed files with 89 additions and 33 deletions
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Show Stats Download Patch File Download Diff File
  1. 122
      dotnet/WhiteRabbit.UnmanagedBridge/WhiteRabbit.UnmanagedBridge.cpp

122
dotnet/WhiteRabbit.UnmanagedBridge/WhiteRabbit.UnmanagedBridge.cpp
Unescape View File

@ -26,47 +26,103 @@ void WhiteRabbitUnmanagedBridge::MD5Unmanaged::ComputeMD5(unsigned __int32 * inp
#endif #endif
} }
void WhiteRabbitUnmanagedBridge::MD5Unmanaged::FillPhraseSet(__int64* bufferPointer, __int64* allWordsPointer, __int32* wordIndexes, unsigned __int64* permutationsPointer, int permutationOffset, int numberOfCharacters, int numberOfWords) template<int phraseOffset>
struct Filler
{ {
__int64* longBuffer = (__int64*)bufferPointer; static void FillPhraseSetLength(unsigned __int32* uintBuffer, const unsigned __int32 lengthInBits)
{
*uintBuffer = lengthInBits;
Filler<phraseOffset + 1>::FillPhraseSetLength(uintBuffer + 8, lengthInBits);
}
};
auto currentPermutationPointer = permutationsPointer + permutationOffset; template<>
for (auto i = 0; i < PHRASES_PER_SET; i++, currentPermutationPointer++) struct Filler<PHRASES_PER_SET>
{
static void FillPhraseSetLength(const unsigned __int32* uintBuffer, const unsigned __int32 lengthInBits)
{ {
auto permutation = *currentPermutationPointer; }
if (permutation == 0) };
{
continue;
}
auto cumulativeWordOffsetX4 = 0; void WhiteRabbitUnmanagedBridge::MD5Unmanaged::FillPhraseSet(__int64* bufferPointer, __int64* allWordsPointer, __int32* wordIndexes, unsigned __int64* permutationsPointer, int permutationOffset, int numberOfCharacters, int numberOfWords)
for (auto j = 0; j < numberOfWords; j++) {
{ unsigned __int64 permutations[PHRASES_PER_SET];
auto currentWord = allWordsPointer + wordIndexes[permutation & 15] * 128; __int32 cumulativeWordOffsets[PHRASES_PER_SET];
permutation = permutation >> 4; __int32 permutationOffsetInBytes = permutationOffset * sizeof(*permutations);
longBuffer[0] |= currentWord[cumulativeWordOffsetX4 + 0]; #define INIT_DATA(phraseNumber) \
longBuffer[1] |= currentWord[cumulativeWordOffsetX4 + 1]; permutations[phraseNumber] = *(unsigned __int64*)(((char*)permutationsPointer) + permutationOffsetInBytes + phraseNumber * sizeof(*permutations)); \
longBuffer[2] ^= currentWord[cumulativeWordOffsetX4 + 2]; cumulativeWordOffsets[phraseNumber] = 0;
longBuffer[3] ^= currentWord[cumulativeWordOffsetX4 + 3];
cumulativeWordOffsetX4 += (__int32)currentWord[127];
}
longBuffer += 4; INIT_DATA(0);
} INIT_DATA(1);
INIT_DATA(2);
INIT_DATA(3);
INIT_DATA(4);
INIT_DATA(5);
INIT_DATA(6);
INIT_DATA(7);
INIT_DATA(8);
INIT_DATA(9);
INIT_DATA(10);
INIT_DATA(11);
INIT_DATA(12);
INIT_DATA(13);
INIT_DATA(14);
INIT_DATA(15);
auto length = numberOfCharacters + numberOfWords - 1; for (auto j = 0; j < numberOfWords; j++)
unsigned char* byteBuffer = ((unsigned char*)bufferPointer) + length;
for (auto i = 0; i < PHRASES_PER_SET; i++)
{ {
*byteBuffer = 128; for (auto i = 0; i < PHRASES_PER_SET; i++)
byteBuffer += 32; {
auto currentWord = allWordsPointer + wordIndexes[permutations[i] & 15] * 128;
permutations[i] = permutations[i] >> 4;
bufferPointer[i * 4 + 0] |= currentWord[cumulativeWordOffsets[i] + 0];
bufferPointer[i * 4 + 1] |= currentWord[cumulativeWordOffsets[i] + 1];
bufferPointer[i * 4 + 2] |= currentWord[cumulativeWordOffsets[i] + 2];
bufferPointer[i * 4 + 3] |= currentWord[cumulativeWordOffsets[i] + 3];
cumulativeWordOffsets[i] += (__int32)currentWord[127];
}
} }
auto length = numberOfCharacters + numberOfWords - 1;
auto lengthInBits = (unsigned __int32)(length << 3); auto lengthInBits = (unsigned __int32)(length << 3);
unsigned int* uintBuffer = ((unsigned __int32*)bufferPointer) + 7;
for (auto i = 0; i < PHRASES_PER_SET; i++) #define FILL_PHRASE_LAST_BYTE(phraseNumber, byteBuffer) ((unsigned char*)bufferPointer)[length + phraseNumber * 32] = 128;
{ #define FILL_PHRASE_SET_LENGTH(phraseNumber, uintBuffer, lengthInBits) ((unsigned __int32*)bufferPointer)[7 + phraseNumber * 8] = lengthInBits;
*uintBuffer = lengthInBits;
uintBuffer += 8; FILL_PHRASE_LAST_BYTE(0, byteBuffer);
} FILL_PHRASE_LAST_BYTE(1, byteBuffer);
FILL_PHRASE_LAST_BYTE(2, byteBuffer);
FILL_PHRASE_LAST_BYTE(3, byteBuffer);
FILL_PHRASE_SET_LENGTH(0, uintBuffer, lengthInBits);
FILL_PHRASE_SET_LENGTH(1, uintBuffer, lengthInBits);
FILL_PHRASE_SET_LENGTH(2, uintBuffer, lengthInBits);
FILL_PHRASE_SET_LENGTH(3, uintBuffer, lengthInBits);
FILL_PHRASE_LAST_BYTE(4, byteBuffer);
FILL_PHRASE_LAST_BYTE(5, byteBuffer);
FILL_PHRASE_LAST_BYTE(6, byteBuffer);
FILL_PHRASE_LAST_BYTE(7, byteBuffer);
FILL_PHRASE_SET_LENGTH(4, uintBuffer, lengthInBits);
FILL_PHRASE_SET_LENGTH(5, uintBuffer, lengthInBits);
FILL_PHRASE_SET_LENGTH(6, uintBuffer, lengthInBits);
FILL_PHRASE_SET_LENGTH(7, uintBuffer, lengthInBits);
FILL_PHRASE_LAST_BYTE(8, byteBuffer);
FILL_PHRASE_LAST_BYTE(9, byteBuffer);
FILL_PHRASE_LAST_BYTE(10, byteBuffer);
FILL_PHRASE_LAST_BYTE(11, byteBuffer);
FILL_PHRASE_SET_LENGTH(8, uintBuffer, lengthInBits);
FILL_PHRASE_SET_LENGTH(9, uintBuffer, lengthInBits);
FILL_PHRASE_SET_LENGTH(10, uintBuffer, lengthInBits);
FILL_PHRASE_SET_LENGTH(11, uintBuffer, lengthInBits);
FILL_PHRASE_LAST_BYTE(12, byteBuffer);
FILL_PHRASE_LAST_BYTE(13, byteBuffer);
FILL_PHRASE_LAST_BYTE(14, byteBuffer);
FILL_PHRASE_LAST_BYTE(15, byteBuffer);
FILL_PHRASE_SET_LENGTH(12, uintBuffer, lengthInBits);
FILL_PHRASE_SET_LENGTH(13, uintBuffer, lengthInBits);
FILL_PHRASE_SET_LENGTH(14, uintBuffer, lengthInBits);
FILL_PHRASE_SET_LENGTH(15, uintBuffer, lengthInBits);
} }

Write Preview
Loading…
Cancel
Save