You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
327 lines
12 KiB
327 lines
12 KiB
#include "stdafx.h"
|
|
#include "md5.h"
|
|
|
|
#include "intrin.h"
|
|
|
|
#pragma unmanaged
|
|
|
|
#if AVX2
|
|
|
|
//typedef __m256i MD5Vector;
|
|
struct MD5Vector
|
|
{
|
|
__m256i m_V0;
|
|
__m256i m_V1;
|
|
inline MD5Vector() {}
|
|
inline MD5Vector(__m256i C0, __m256i C1) :m_V0(C0), m_V1(C1) {}
|
|
|
|
__forceinline MD5Vector MXor(MD5Vector R) const
|
|
{
|
|
return MD5Vector(_mm256_xor_si256(m_V0, R.m_V0), _mm256_xor_si256(m_V1, R.m_V1));
|
|
}
|
|
|
|
__forceinline MD5Vector MAnd(MD5Vector R) const
|
|
{
|
|
return MD5Vector(_mm256_and_si256(m_V0, R.m_V0), _mm256_and_si256(m_V1, R.m_V1));
|
|
}
|
|
|
|
__forceinline MD5Vector MAndNot(MD5Vector R) const
|
|
{
|
|
return MD5Vector(_mm256_andnot_si256(m_V0, R.m_V0), _mm256_andnot_si256(m_V1, R.m_V1));
|
|
}
|
|
|
|
__forceinline MD5Vector MOr(MD5Vector R) const
|
|
{
|
|
return MD5Vector(_mm256_or_si256(m_V0, R.m_V0), _mm256_or_si256(m_V1, R.m_V1));
|
|
}
|
|
|
|
__forceinline MD5Vector MAdd(MD5Vector R) const
|
|
{
|
|
return MD5Vector(_mm256_add_epi32(m_V0, R.m_V0), _mm256_add_epi32(m_V1, R.m_V1));
|
|
}
|
|
|
|
__forceinline MD5Vector MShiftLeft(int shift) const
|
|
{
|
|
return MD5Vector(_mm256_slli_epi32(m_V0, shift), _mm256_slli_epi32(m_V1, shift));
|
|
}
|
|
|
|
__forceinline MD5Vector MShiftRight(int shift) const
|
|
{
|
|
return MD5Vector(_mm256_srli_epi32(m_V0, shift), _mm256_srli_epi32(m_V1, shift));
|
|
}
|
|
};
|
|
|
|
#define OP_XOR(a, b) a.MXor(b)
|
|
#define OP_AND(a, b) a.MAnd(b)
|
|
#define OP_ANDNOT(a, b) a.MAndNot(b)
|
|
#define OP_OR(a, b) a.MOr(b)
|
|
#define OP_ADD(a, b) a.MAdd(b)
|
|
#define OP_ROT(a, r) OP_OR(a.MShiftLeft(r), a.MShiftRight(32 - (r)))
|
|
#define OP_BLEND(a, b, x) OP_OR(OP_AND(x, b), OP_ANDNOT(x, a))
|
|
|
|
#define CREATE_VECTOR(a) MD5Vector(_mm256_set1_epi32(a), _mm256_set1_epi32(a))
|
|
#define CREATE_VECTOR_FROM_INPUT(input, offset) MD5Vector(\
|
|
_mm256_set_epi32( \
|
|
input[offset + 7 * 8], \
|
|
input[offset + 6 * 8], \
|
|
input[offset + 5 * 8], \
|
|
input[offset + 4 * 8], \
|
|
input[offset + 3 * 8], \
|
|
input[offset + 2 * 8], \
|
|
input[offset + 1 * 8], \
|
|
input[offset + 0 * 8]), \
|
|
_mm256_set_epi32( \
|
|
input[offset + 15 * 8], \
|
|
input[offset + 14 * 8], \
|
|
input[offset + 13 * 8], \
|
|
input[offset + 12 * 8], \
|
|
input[offset + 11 * 8], \
|
|
input[offset + 10 * 8], \
|
|
input[offset + 9 * 8], \
|
|
input[offset + 8 * 8]) \
|
|
)
|
|
|
|
#define WRITE_TO_OUTPUT(a, output, expected) \
|
|
output[0] = _mm256_movemask_epi8(_mm256_cmpeq_epi32(*expected, _mm256_permute4x64_epi64(a.m_V0, 0 * 0x55))); \
|
|
output[1] = _mm256_movemask_epi8(_mm256_cmpeq_epi32(*expected, _mm256_permute4x64_epi64(a.m_V1, 0 * 0x55))); \
|
|
output[2] = _mm256_movemask_epi8(_mm256_cmpeq_epi32(*expected, _mm256_permute4x64_epi64(a.m_V0, 1 * 0x55))); \
|
|
output[3] = _mm256_movemask_epi8(_mm256_cmpeq_epi32(*expected, _mm256_permute4x64_epi64(a.m_V1, 1 * 0x55))); \
|
|
output[4] = _mm256_movemask_epi8(_mm256_cmpeq_epi32(*expected, _mm256_permute4x64_epi64(a.m_V0, 2 * 0x55))); \
|
|
output[5] = _mm256_movemask_epi8(_mm256_cmpeq_epi32(*expected, _mm256_permute4x64_epi64(a.m_V1, 2 * 0x55))); \
|
|
output[6] = _mm256_movemask_epi8(_mm256_cmpeq_epi32(*expected, _mm256_permute4x64_epi64(a.m_V0, 3 * 0x55))); \
|
|
output[7] = _mm256_movemask_epi8(_mm256_cmpeq_epi32(*expected, _mm256_permute4x64_epi64(a.m_V1, 3 * 0x55))); \
|
|
output[8] = _mm256_movemask_epi8(_mm256_cmpeq_epi8(*((__m256i*)output), _mm256_setzero_si256()));
|
|
|
|
#elif SIMD
|
|
|
|
typedef __m128i MD5Vector;
|
|
|
|
#define OP_XOR(a, b) _mm_xor_si128(a, b)
|
|
#define OP_AND(a, b) _mm_and_si128(a, b)
|
|
#define OP_ANDNOT(a, b) _mm_andnot_si128(a, b)
|
|
#define OP_OR(a, b) _mm_or_si128(a, b)
|
|
#define OP_ADD(a, b) _mm_add_epi32(a, b)
|
|
#define OP_ROT(a, r) OP_OR(_mm_slli_epi32(a, r), _mm_srli_epi32(a, 32 - (r)))
|
|
#define OP_BLEND(a, b, x) OP_OR(OP_AND(x, b), OP_ANDNOT(x, a))
|
|
//#define OP_BLEND(a, b, x) OP_XOR(a, OP_AND(x, OP_XOR(b, a)))
|
|
|
|
#define CREATE_VECTOR(a) _mm_set1_epi32(a)
|
|
#define CREATE_VECTOR_FROM_INPUT(input, offset) _mm_set_epi32( \
|
|
input[offset + 3 * 8], \
|
|
input[offset + 2 * 8], \
|
|
input[offset + 1 * 8], \
|
|
input[offset + 0 * 8])
|
|
|
|
#define WRITE_TO_OUTPUT(a, output) \
|
|
output[7 + 0 * 8] = a.m128i_u32[0]; \
|
|
output[7 + 1 * 8] = a.m128i_u32[1]; \
|
|
output[7 + 2 * 8] = a.m128i_u32[2]; \
|
|
output[7 + 3 * 8] = a.m128i_u32[3];
|
|
|
|
#else
|
|
|
|
typedef unsigned int MD5Vector;
|
|
|
|
#define OP_XOR(a, b) (a) ^ (b)
|
|
#define OP_AND(a, b) (a) & (b)
|
|
#define OP_ANDNOT(a, b) ~(a) & (b)
|
|
#define OP_OR(a, b) (a) | (b)
|
|
#define OP_ADD(a, b) (a) + (b)
|
|
#define OP_ROT(a, r) _rotl(a, r)
|
|
#define OP_BLEND(a, b, x) ((x) & (b)) | (~(x) & (a))
|
|
|
|
#define CREATE_VECTOR(a) a
|
|
#define CREATE_VECTOR_FROM_INPUT(input, offset) (input[offset])
|
|
|
|
#define WRITE_TO_OUTPUT(a, output) \
|
|
output[7] = a;
|
|
#endif
|
|
|
|
#define OP_NEG(a) OP_ANDNOT(a, CREATE_VECTOR(0xffffffff))
|
|
|
|
typedef struct {
|
|
unsigned int K[64];
|
|
unsigned int Init[4];
|
|
} MD5Parameters;
|
|
|
|
static const MD5Parameters Parameters = {
|
|
{
|
|
0xd76aa478,
|
|
0xe8c7b756,
|
|
0x242070db,
|
|
0xc1bdceee,
|
|
0xf57c0faf,
|
|
0x4787c62a,
|
|
0xa8304613,
|
|
0xfd469501,
|
|
0x698098d8,
|
|
0x8b44f7af,
|
|
0xffff5bb1,
|
|
0x895cd7be,
|
|
0x6b901122,
|
|
0xfd987193,
|
|
0xa679438e,
|
|
0x49b40821,
|
|
0xf61e2562,
|
|
0xc040b340,
|
|
0x265e5a51,
|
|
0xe9b6c7aa,
|
|
0xd62f105d,
|
|
0x02441453,
|
|
0xd8a1e681,
|
|
0xe7d3fbc8,
|
|
0x21e1cde6,
|
|
0xc33707d6,
|
|
0xf4d50d87,
|
|
0x455a14ed,
|
|
0xa9e3e905,
|
|
0xfcefa3f8,
|
|
0x676f02d9,
|
|
0x8d2a4c8a,
|
|
0xfffa3942,
|
|
0x8771f681,
|
|
0x6d9d6122,
|
|
0xfde5380c,
|
|
0xa4beea44,
|
|
0x4bdecfa9,
|
|
0xf6bb4b60,
|
|
0xbebfbc70,
|
|
0x289b7ec6,
|
|
0xeaa127fa,
|
|
0xd4ef3085,
|
|
0x04881d05,
|
|
0xd9d4d039,
|
|
0xe6db99e5,
|
|
0x1fa27cf8,
|
|
0xc4ac5665,
|
|
0xf4292244,
|
|
0x432aff97,
|
|
0xab9423a7,
|
|
0xfc93a039,
|
|
0x655b59c3,
|
|
0x8f0ccc92,
|
|
0xffeff47d,
|
|
0x85845dd1,
|
|
0x6fa87e4f,
|
|
0xfe2ce6e0,
|
|
0xa3014314,
|
|
0x4e0811a1,
|
|
0xf7537e82,
|
|
0xbd3af235,
|
|
0x2ad7d2bb,
|
|
0xeb86d391,
|
|
},
|
|
{
|
|
0x67452301,
|
|
0xefcdab89,
|
|
0x98badcfe,
|
|
0x10325476,
|
|
},
|
|
};
|
|
|
|
#define Blend(a, b, x) OP_BLEND(a, b, x)
|
|
#define Xor(a, b, c) OP_XOR(a, OP_XOR(b, c))
|
|
#define I(a, b, c) OP_XOR(a, OP_OR(b, OP_NEG(c)))
|
|
|
|
#define StepOuter(r, a, b, x) \
|
|
a = x; \
|
|
a = OP_ADD(b, OP_ROT(a, r));
|
|
|
|
#define Step1(r, a, b, c, d, k, w) StepOuter(r, a, b, OP_ADD(Blend(d, c, b), OP_ADD(CREATE_VECTOR(k), OP_ADD(a, w))))
|
|
#define Step1E(r, a, b, c, d, k) StepOuter(r, a, b, OP_ADD(Blend(d, c, b), OP_ADD(CREATE_VECTOR(k), a)))
|
|
|
|
#define Step2(r, a, b, c, d, k, w) StepOuter(r, a, c, OP_ADD(Blend(d, c, b), OP_ADD(CREATE_VECTOR(k), OP_ADD(a, w))))
|
|
#define Step2E(r, a, b, c, d, k) StepOuter(r, a, c, OP_ADD(Blend(d, c, b), OP_ADD(CREATE_VECTOR(k), a)))
|
|
|
|
#define Step3(r, a, b, c, d, k, w) StepOuter(r, a, b, OP_ADD(Xor(b, c, d), OP_ADD(CREATE_VECTOR(k), OP_ADD(a, w))))
|
|
#define Step3E(r, a, b, c, d, k) StepOuter(r, a, b, OP_ADD(Xor(b, c, d), OP_ADD(CREATE_VECTOR(k), a)))
|
|
|
|
#define Step4(r, a, b, c, d, k, w) StepOuter(r, a, b, OP_ADD(I(c, b, d), OP_ADD(CREATE_VECTOR(k), OP_ADD(a, w))))
|
|
#define Step4E(r, a, b, c, d, k) StepOuter(r, a, b, OP_ADD(I(c, b, d), OP_ADD(CREATE_VECTOR(k), a)))
|
|
|
|
void md5(unsigned __int32 * input, unsigned __int32 * expected)
|
|
{
|
|
MD5Vector a = CREATE_VECTOR(Parameters.Init[0]);
|
|
MD5Vector b = CREATE_VECTOR(Parameters.Init[1]);
|
|
MD5Vector c = CREATE_VECTOR(Parameters.Init[2]);
|
|
MD5Vector d = CREATE_VECTOR(Parameters.Init[3]);
|
|
|
|
MD5Vector inputVector0 = CREATE_VECTOR_FROM_INPUT(input, 0);
|
|
MD5Vector inputVector1 = CREATE_VECTOR_FROM_INPUT(input, 1);
|
|
MD5Vector inputVector2 = CREATE_VECTOR_FROM_INPUT(input, 2);
|
|
MD5Vector inputVector3 = CREATE_VECTOR_FROM_INPUT(input, 3);
|
|
MD5Vector inputVector4 = CREATE_VECTOR_FROM_INPUT(input, 4);
|
|
MD5Vector inputVector5 = CREATE_VECTOR_FROM_INPUT(input, 5);
|
|
MD5Vector inputVector6 = CREATE_VECTOR_FROM_INPUT(input, 6);
|
|
MD5Vector inputVector7 = CREATE_VECTOR_FROM_INPUT(input, 7);
|
|
|
|
a = Step1 ( 7, a, b, c, d, Parameters.K[ 0], inputVector0);
|
|
d = Step1 (12, d, a, b, c, Parameters.K[ 1], inputVector1);
|
|
c = Step1 (17, c, d, a, b, Parameters.K[ 2], inputVector2);
|
|
b = Step1 (22, b, c, d, a, Parameters.K[ 3], inputVector3);
|
|
a = Step1 ( 7, a, b, c, d, Parameters.K[ 4], inputVector4);
|
|
d = Step1 (12, d, a, b, c, Parameters.K[ 5], inputVector5);
|
|
c = Step1 (17, c, d, a, b, Parameters.K[ 6], inputVector6);
|
|
b = Step1E(22, b, c, d, a, Parameters.K[ 7]);
|
|
a = Step1E( 7, a, b, c, d, Parameters.K[ 8]);
|
|
d = Step1E(12, d, a, b, c, Parameters.K[ 9]);
|
|
c = Step1E(17, c, d, a, b, Parameters.K[10]);
|
|
b = Step1E(22, b, c, d, a, Parameters.K[11]);
|
|
a = Step1E( 7, a, b, c, d, Parameters.K[12]);
|
|
d = Step1E(12, d, a, b, c, Parameters.K[13]);
|
|
c = Step1 (17, c, d, a, b, Parameters.K[14], inputVector7);
|
|
b = Step1E(22, b, c, d, a, Parameters.K[15]);
|
|
|
|
a = Step2 ( 5, a, d, b, c, Parameters.K[16], inputVector1);
|
|
d = Step2 ( 9, d, c, a, b, Parameters.K[17], inputVector6);
|
|
c = Step2E(14, c, b, d, a, Parameters.K[18]);
|
|
b = Step2 (20, b, a, c, d, Parameters.K[19], inputVector0);
|
|
a = Step2 ( 5, a, d, b, c, Parameters.K[20], inputVector5);
|
|
d = Step2E( 9, d, c, a, b, Parameters.K[21]);
|
|
c = Step2E(14, c, b, d, a, Parameters.K[22]);
|
|
b = Step2 (20, b, a, c, d, Parameters.K[23], inputVector4);
|
|
a = Step2E( 5, a, d, b, c, Parameters.K[24]);
|
|
d = Step2 ( 9, d, c, a, b, Parameters.K[25], inputVector7);
|
|
c = Step2 (14, c, b, d, a, Parameters.K[26], inputVector3);
|
|
b = Step2E(20, b, a, c, d, Parameters.K[27]);
|
|
a = Step2E( 5, a, d, b, c, Parameters.K[28]);
|
|
d = Step2 ( 9, d, c, a, b, Parameters.K[29], inputVector2);
|
|
c = Step2E(14, c, b, d, a, Parameters.K[30]);
|
|
b = Step2E(20, b, a, c, d, Parameters.K[31]);
|
|
|
|
a = Step3 ( 4, a, b, c, d, Parameters.K[32], inputVector5);
|
|
d = Step3E(11, d, a, b, c, Parameters.K[33]);
|
|
c = Step3E(16, c, d, a, b, Parameters.K[34]);
|
|
b = Step3 (23, b, c, d, a, Parameters.K[35], inputVector7);
|
|
a = Step3 ( 4, a, b, c, d, Parameters.K[36], inputVector1);
|
|
d = Step3 (11, d, a, b, c, Parameters.K[37], inputVector4);
|
|
c = Step3E(16, c, d, a, b, Parameters.K[38]);
|
|
b = Step3E(23, b, c, d, a, Parameters.K[39]);
|
|
a = Step3E( 4, a, b, c, d, Parameters.K[40]);
|
|
d = Step3 (11, d, a, b, c, Parameters.K[41], inputVector0);
|
|
c = Step3 (16, c, d, a, b, Parameters.K[42], inputVector3);
|
|
b = Step3 (23, b, c, d, a, Parameters.K[43], inputVector6);
|
|
a = Step3E( 4, a, b, c, d, Parameters.K[44]);
|
|
d = Step3E(11, d, a, b, c, Parameters.K[45]);
|
|
c = Step3E(16, c, d, a, b, Parameters.K[46]);
|
|
b = Step3 (23, b, c, d, a, Parameters.K[47], inputVector2);
|
|
|
|
a = Step4 ( 6, a, b, c, d, Parameters.K[48], inputVector0);
|
|
d = Step4E(10, d, a, b, c, Parameters.K[49]);
|
|
c = Step4 (15, c, d, a, b, Parameters.K[50], inputVector7);
|
|
b = Step4 (21, b, c, d, a, Parameters.K[51], inputVector5);
|
|
a = Step4E( 6, a, b, c, d, Parameters.K[52]);
|
|
d = Step4 (10, d, a, b, c, Parameters.K[53], inputVector3);
|
|
c = Step4E(15, c, d, a, b, Parameters.K[54]);
|
|
b = Step4 (21, b, c, d, a, Parameters.K[55], inputVector1);
|
|
a = Step4E( 6, a, b, c, d, Parameters.K[56]);
|
|
d = Step4E(10, d, a, b, c, Parameters.K[57]);
|
|
c = Step4 (15, c, d, a, b, Parameters.K[58], inputVector6);
|
|
b = Step4E(21, b, c, d, a, Parameters.K[59]);
|
|
a = Step4 ( 6, a, b, c, d, Parameters.K[60], inputVector4);
|
|
|
|
a = OP_ADD(CREATE_VECTOR(Parameters.Init[0]), a);
|
|
|
|
WRITE_TO_OUTPUT(a, ((__int32*)input), ((__m256i*)expected));
|
|
}
|
|
#pragma managed
|
|
|