extern "C" void
GenRandom( unsigned char * pData, unsigned long Size )
{
int i;
while( Size > 0 )
{
__asm
{
rdtsc
add random_seed.tsc_lo, eax
add random_seed.tsc_hi, edx
}
GetSystemTime( &random_seed.sys_time);
QueryPerformanceCounter( &random_seed.PerfCounter );
SHA1_Hash((unsigned char*)&random_seed, sizeof(random_seed), random_seed.digest );
for( i = 0; i < 20; i++ )
{
*pData++ = random_seed.digest[i];
if( --Size == 0 )
return;
}
}
}
/*! *********************************************************************************
* \brief Pseudo Random Number Generator (PRNG) implementation
* according to NIST FIPS Publication 186-2, APPENDIX 3
*
* Let x be the signer's private key.
* The following may be used to generate m values of x:
* Step 1. Choose a new, secret value for the seed-key, XKEY.
* Step 2. In hexadecimal notation let
* t = 67452301 EFCDAB89 98BADCFE 10325476 C3D2E1F0.
* This is the initial value for H0 || H1 || H2 || H3 || H4 in the SHS.
* Step 3. For j = 0 to m - 1 do
* a. XSEEDj = optional user input.
* b. XVAL = (XKEY + XSEEDj) mod 2^b
* c. xj = G(t,XVAL) mod q
* d. XKEY = (1 + XKEY + xj) mod 2^b
*
* \param[out] pOut - pointer to the output buffer
* \param[in] outBytes - the number of bytes to be copyed (1-20)
* \param[in] pXSEED - optional user SEED. Should be NULL if not used.
*
* \return The number of bytes copied or -1 if reseed is needed
*
********************************************************************************** */
int16_t RNG_GetPseudoRandomNo(uint8_t* pOut, uint8_t outBytes, uint8_t* pXSEED)
{
uint32_t i;
sha1Context_t ctx;
if(pXSEED)
{
mPRNG_Requests = 1;
}
if (mPRNG_Requests == gRngMaxRequests_d)
{
return -1;
}
mPRNG_Requests++;
/* a. XSEEDj = optional user input. */
if (pXSEED)
{
/* b. XVAL = (XKEY + XSEEDj) mod 2^b */
for (i=0; i<mPRNG_NoOfBytes_c; i++)
{
ctx.buffer[i] = ((uint8_t*)XKEY)[i] + pXSEED[i];
}
}
else
{
for (i=0; i<mPRNG_NoOfBytes_c; i++)
{
ctx.buffer[i] = ((uint8_t*)XKEY)[i];
}
}
/* c. xj = G(t,XVAL) mod q
***************************/
SHA1_Hash(&ctx, ctx.buffer, mPRNG_NoOfBytes_c);
/* d. XKEY = (1 + XKEY + xj) mod 2^b */
XKEY[0] += 1;
for (i=0; i<mPRNG_NoOfLongWords_c; i++)
{
XKEY[i] += ctx.hash[i];
}
/* Check if the length provided exceeds the output data size */
if (outBytes > mPRNG_NoOfBytes_c)
{
outBytes = mPRNG_NoOfBytes_c;
}
/* Copy the generated number */
for (i=0; i < outBytes; i++)
{
pOut[i] = ((uint8_t*)ctx.hash)[i];
}
return outBytes;
}
int wmain(
_In_ int argc,
_In_ wchar_t* argv[])
{
#elif (defined(PLATFORM_LINUX) || defined(PLATFORM_MACX))
int main(int argc, char *argv[])
{
#endif
#if defined(ENABLE_LIBSODIUM)
//Initialization(Part 1)
#if defined(PLATFORM_WIN)
std::wstring FileName, Command;
#elif (defined(PLATFORM_LINUX) || defined(PLATFORM_MACX))
std::string FileName, Command;
#endif
//Read commands.
if (argc < 2)
{
PrintDescription();
return EXIT_SUCCESS;
}
else if (argc == 2) //File name only, use default hash function.
{
//Commands check
Command = argv[1U];
CaseConvert(true, Command);
if (Command == COMMAND_LONG_PRINT_VERSION || Command == COMMAND_SHORT_PRINT_VERSION)
{
fwprintf_s(stderr, L"FileHash ");
fwprintf_s(stderr, FULL_VERSION);
fwprintf_s(stderr, L"\n");
return EXIT_SUCCESS;
}
else if (Command == COMMAND_LONG_HELP || Command == COMMAND_SHORT_HELP || Command == COMMAND_SIGN_HELP)
{
PrintDescription();
return EXIT_SUCCESS;
}
else if (Command == COMMAND_LIB_VERSION)
{
std::wstring LibVersion;
if (MBSToWCSString(SODIUM_VERSION_STRING, strlen(SODIUM_VERSION_STRING), LibVersion))
fwprintf_s(stderr, L"LibSodium version %ls\n", LibVersion.c_str());
return EXIT_SUCCESS;
}
else { //Mark filename.
FileName = argv[1U];
}
}
else if (argc == 3) //File name with hash function command
{
Command = argv[1U];
FileName = argv[2U];
//Commands check
if (Command.length() < 3U)
{
fwprintf_s(stderr, L"Commands error.\n");
return EXIT_FAILURE;
}
else {
CaseConvert(true, Command);
}
//CRC family
if (Command.find(HASH_COMMAND_CRC) == 0)
{
HashFamilyID = HASH_ID_CRC;
if (!ReadCommand_CRC(Command))
return EXIT_FAILURE;
}
//Internet Protocol Checksum
else if (Command == HASH_COMMAND_CHECKSUM)
{
HashFamilyID = HASH_ID_CHECKSUM;
}
//MD2
else if (Command == HASH_COMMAND_MD2)
{
HashFamilyID = HASH_ID_MD2;
}
//MD4 family
else if (Command == HASH_COMMAND_MD4)
{
HashFamilyID = HASH_ID_MD4;
}
else if (Command == HASH_COMMAND_ED2K)
{
HashFamilyID = HASH_ID_ED2K;
}
//MD5
else if (Command == HASH_COMMAND_MD || Command == HASH_COMMAND_MD5)
{
HashFamilyID = HASH_ID_MD5;
}
//SHA-1
else if (Command.find(HASH_COMMAND_SHA1) == 0)
{
HashFamilyID = HASH_ID_SHA1;
}
//SHA-2 family
else if (Command == HASH_COMMAND_SHA2_384 || Command.find(HASH_COMMAND_SHA2_512) == 0 || Command.find(HASH_COMMAND_SHA2) == 0)
{
HashFamilyID = HASH_ID_SHA2;
if (!ReadCommand_SHA2(Command))
return EXIT_FAILURE;
}
//SHA-3 family
else if (Command == HASH_COMMAND_SHA || Command.find(HASH_COMMAND_SHA3) == 0)
{
HashFamilyID = HASH_ID_SHA3;
if (!ReadCommand_SHA3(Command))
return EXIT_FAILURE;
}
//Commands error
else {
fwprintf_s(stderr, L"Commands error.\n");
return EXIT_FAILURE;
}
}
else {
fwprintf_s(stderr, L"Commands error.\n");
return EXIT_FAILURE;
}
Command.clear();
Command.shrink_to_fit();
//Open file.
FILE *Input = nullptr;
#if defined(PLATFORM_WIN)
if (_wfopen_s(&Input, FileName.c_str(), L"rb") != 0 || Input == nullptr)
#elif (defined(PLATFORM_LINUX) || defined(PLATFORM_MACX))
Input = fopen(FileName.c_str(), "rb");
if (Input == nullptr)
#endif
{
fwprintf_s(stderr, L"Open file error.\n");
return EXIT_FAILURE;
}
else {
if (HashFamilyID == HASH_ID_CRC && !CRC_Hash(Input) || //CRC family
HashFamilyID == HASH_ID_CHECKSUM && !Checksum_Hash(Input) || //Internet Protocol Checksum
HashFamilyID == HASH_ID_MD2 && !MD2_Hash(Input) || //MD2
(HashFamilyID == HASH_ID_MD4 || HashFamilyID == HASH_ID_ED2K) && !MD4_Hash(Input) || //MD4 family
HashFamilyID == HASH_ID_MD5 && !MD5_Hash(Input) || //MD5
HashFamilyID == HASH_ID_SHA1 && !SHA1_Hash(Input) || //SHA-1
HashFamilyID == HASH_ID_SHA2 && !SHA2_Hash(Input) || //SHA-2 family
HashFamilyID == HASH_ID_SHA3 && !SHA3_Hash(Input)) //SHA-3 family
{
fclose(Input);
return EXIT_FAILURE;
}
}
fclose(Input);
#else
#if defined(PLATFORM_WIN)
fwprintf_s(stderr, L"LibSodium is disable.\n\n");
system("PAUSE");
#elif (defined(PLATFORM_LINUX) || defined(PLATFORM_MACX))
fwprintf(stderr, L"LibSodium is disable.\n\n");
#endif
#endif
return EXIT_SUCCESS;
}
