mirror of
https://github.com/shadps4-emu/ext-cryptopp.git
synced 2024-11-23 18:09:48 +00:00
467 lines
13 KiB
C++
467 lines
13 KiB
C++
// rdrand.cpp - written and placed in public domain by Jeffrey Walton and Uri Blumenthal.
|
|
|
|
#include "pch.h"
|
|
#include "config.h"
|
|
#include "cryptlib.h"
|
|
#include "secblock.h"
|
|
#include "rdrand.h"
|
|
#include "cpu.h"
|
|
|
|
// This file (and friends) provides both RDRAND and RDSEED. They were added at
|
|
// Crypto++ 5.6.3. At compile time, it uses CRYPTOPP_BOOL_{X86|X32|X64}
|
|
// to select an implementation or "throw NotImplemented". The class does not
|
|
// determine if RDRAND or RDSEED are available at runtime. If not available,
|
|
// then a SIGILL will result. Users of the classes should call HasRDRAND()
|
|
// or HasRDSEED() to determine if a generator is available.
|
|
// The original classes accepted a retry count. Retries were superflous for
|
|
// RDRAND, and RDSEED encountered a failure about 1 in 256 bytes depending
|
|
// on the processor. Retries were removed at Crypto++ 6.0 because
|
|
// GenerateBlock unconditionally retries and always fulfills the request.
|
|
|
|
/////////////////////////////////////////////////////////////////////
|
|
/////////////////////////////////////////////////////////////////////
|
|
|
|
// For Linux, NASM is optional. Run rdrand-nasm.sh, and then make
|
|
// with "USE_NASM" like so: USE_NASM=1 make -j 4. The makefile
|
|
// will add the appropriate defines when building rdrand.cpp,
|
|
// and add the appropriate object file during link.
|
|
|
|
#if 0
|
|
#define NASM_RDRAND_ASM_AVAILABLE 1
|
|
#define NASM_RDSEED_ASM_AVAILABLE 1
|
|
#endif
|
|
|
|
/////////////////////////////////////////////////////////////////////
|
|
/////////////////////////////////////////////////////////////////////
|
|
|
|
#if defined(CRYPTOPP_CPUID_AVAILABLE)
|
|
# if defined(CRYPTOPP_MSC_VERSION)
|
|
# if (CRYPTOPP_MSC_VERSION >= 1700)
|
|
# define ALL_RDRAND_INTRIN_AVAILABLE 1
|
|
# else
|
|
# define MASM_RDRAND_ASM_AVAILABLE 1
|
|
# endif
|
|
# if (CRYPTOPP_MSC_VERSION >= 1800)
|
|
# define ALL_RDSEED_INTRIN_AVAILABLE 1
|
|
# else
|
|
# define MASM_RDSEED_ASM_AVAILABLE 1
|
|
# endif
|
|
# elif defined(CRYPTOPP_LLVM_CLANG_VERSION) || defined(CRYPTOPP_APPLE_CLANG_VERSION)
|
|
# if defined(__RDRND__)
|
|
# define ALL_RDRAND_INTRIN_AVAILABLE 1
|
|
# else
|
|
# define GCC_RDRAND_ASM_AVAILABLE 1
|
|
# endif
|
|
# if defined(__RDSEED__)
|
|
# define ALL_RDSEED_INTRIN_AVAILABLE 1
|
|
# else
|
|
# define GCC_RDSEED_ASM_AVAILABLE 1
|
|
# endif
|
|
# elif defined(__SUNPRO_CC)
|
|
# if defined(__RDRND__) && (__SUNPRO_CC >= 0x5130)
|
|
# define ALL_RDRAND_INTRIN_AVAILABLE 1
|
|
# elif (__SUNPRO_CC >= 0x5100)
|
|
# define GCC_RDRAND_ASM_AVAILABLE 1
|
|
# endif
|
|
# if defined(__RDSEED__) && (__SUNPRO_CC >= 0x5140)
|
|
# define ALL_RDSEED_INTRIN_AVAILABLE 1
|
|
# elif (__SUNPRO_CC >= 0x5100)
|
|
# define GCC_RDSEED_ASM_AVAILABLE 1
|
|
# endif
|
|
# elif defined(CRYPTOPP_GCC_VERSION)
|
|
# if defined(__RDRND__) && (CRYPTOPP_GCC_VERSION >= 40700) && !defined(__OPTIMIZE__)
|
|
# define ALL_RDRAND_INTRIN_AVAILABLE 1
|
|
# else
|
|
# define GCC_RDRAND_ASM_AVAILABLE 1
|
|
# endif
|
|
# if defined(__RDSEED__) && (CRYPTOPP_GCC_VERSION >= 40800) && !defined(__OPTIMIZE__)
|
|
# define ALL_RDSEED_INTRIN_AVAILABLE 1
|
|
# else
|
|
# define GCC_RDSEED_ASM_AVAILABLE 1
|
|
# endif
|
|
# endif
|
|
#endif
|
|
|
|
/////////////////////////////////////////////////////////////////////
|
|
/////////////////////////////////////////////////////////////////////
|
|
|
|
#if (ALL_RDRAND_INTRIN_AVAILABLE || ALL_RDSEED_INTRIN_AVAILABLE)
|
|
# include <immintrin.h> // rdrand, MSC, ICC, GCC, and SunCC
|
|
# if defined(__GNUC__) && (CRYPTOPP_GCC_VERSION >= 40700)
|
|
# include <x86intrin.h> // rdseed for some compilers, like GCC
|
|
# endif
|
|
# if defined(__has_include)
|
|
# if __has_include(<x86intrin.h>)
|
|
# include <x86intrin.h>
|
|
# endif
|
|
# endif
|
|
#endif
|
|
|
|
typedef unsigned char byte;
|
|
|
|
#if MASM_RDRAND_ASM_AVAILABLE
|
|
extern "C" void CRYPTOPP_FASTCALL MASM_RDRAND_GenerateBlock(byte*, size_t);
|
|
#endif
|
|
|
|
#if MASM_RDSEED_ASM_AVAILABLE
|
|
extern "C" void CRYPTOPP_FASTCALL MASM_RDSEED_GenerateBlock(byte*, size_t);
|
|
#endif
|
|
|
|
#if NASM_RDRAND_ASM_AVAILABLE
|
|
extern "C" void NASM_RDRAND_GenerateBlock(byte*, size_t);
|
|
#endif
|
|
|
|
#if NASM_RDSEED_ASM_AVAILABLE
|
|
extern "C" void NASM_RDSEED_GenerateBlock(byte*, size_t);
|
|
#endif
|
|
|
|
/////////////////////////////////////////////////////////////////////
|
|
/////////////////////////////////////////////////////////////////////
|
|
|
|
NAMESPACE_BEGIN(CryptoPP)
|
|
|
|
#if defined(CRYPTOPP_CPUID_AVAILABLE)
|
|
|
|
// Fills 4 bytes
|
|
inline void RDRAND32(void* output)
|
|
{
|
|
#if defined(__SUNPRO_CC)
|
|
__asm__ __volatile__
|
|
(
|
|
"1:\n"
|
|
".byte 0x0f, 0xc7, 0xf0;\n"
|
|
"jnc 1b;\n"
|
|
: "=a" (*reinterpret_cast<word32*>(output))
|
|
: : "cc"
|
|
);
|
|
#elif defined(GCC_RDRAND_ASM_AVAILABLE) && (CRYPTOPP_GCC_VERSION >= 40700)
|
|
__asm__ __volatile__
|
|
(
|
|
INTEL_NOPREFIX
|
|
ASL(1)
|
|
AS1(rdrand eax)
|
|
ASJ(jnc, 1, b)
|
|
ATT_NOPREFIX
|
|
: "=a" (*reinterpret_cast<word32*>(output))
|
|
: : "cc"
|
|
);
|
|
#elif defined(GCC_RDRAND_ASM_AVAILABLE) && (CRYPTOPP_GCC_VERSION >= 30200)
|
|
__asm__ __volatile__
|
|
(
|
|
"1:\n"
|
|
".byte 0x0f, 0xc7, 0xf0;\n"
|
|
"jnc 1b;\n"
|
|
: "=a" (*reinterpret_cast<word32*>(output))
|
|
: : "cc"
|
|
);
|
|
#elif defined(ALL_RDRAND_INTRIN_AVAILABLE)
|
|
while(!_rdrand32_step(reinterpret_cast<word32*>(output))) {}
|
|
#else
|
|
// RDRAND not detected at compile time, or no suitable compiler found
|
|
CRYPTOPP_UNUSED(output);
|
|
throw NotImplemented("RDRAND: failed to find an implementation");
|
|
#endif
|
|
}
|
|
|
|
#if CRYPTOPP_BOOL_X64 || CRYPTOPP_BOOL_X32
|
|
// Fills 8 bytes
|
|
inline void RDRAND64(void* output)
|
|
{
|
|
#if defined(__SUNPRO_CC) && (__SUNPRO_CC >= 0x5100)
|
|
__asm__ __volatile__
|
|
(
|
|
"1:\n"
|
|
".byte 0x48, 0x0f, 0xc7, 0xf0;\n"
|
|
"jnc 1b;\n"
|
|
: "=a" (*reinterpret_cast<word64*>(output))
|
|
: : "cc"
|
|
);
|
|
#elif defined(GCC_RDRAND_ASM_AVAILABLE) && (CRYPTOPP_GCC_VERSION >= 40700)
|
|
__asm__ __volatile__
|
|
(
|
|
INTEL_NOPREFIX
|
|
ASL(1)
|
|
AS1(rdrand rax)
|
|
ASJ(jnc, 1, b)
|
|
ATT_NOPREFIX
|
|
: "=a" (*reinterpret_cast<word64*>(output))
|
|
: : "cc"
|
|
);
|
|
#elif defined(GCC_RDRAND_ASM_AVAILABLE) && (CRYPTOPP_GCC_VERSION >= 30200)
|
|
__asm__ __volatile__
|
|
(
|
|
"1:\n"
|
|
".byte 0x48, 0x0f, 0xc7, 0xf0;\n"
|
|
"jnc 1b;\n"
|
|
: "=a" (*reinterpret_cast<word64*>(output))
|
|
: : "cc"
|
|
);
|
|
#elif defined(ALL_RDRAND_INTRIN_AVAILABLE)
|
|
while(!_rdrand64_step(reinterpret_cast<unsigned long long*>(output))) {}
|
|
#else
|
|
// RDRAND not detected at compile time, or no suitable compiler found
|
|
CRYPTOPP_UNUSED(output);
|
|
throw NotImplemented("RDRAND: failed to find an implementation");
|
|
#endif
|
|
}
|
|
#endif // CRYPTOPP_BOOL_X64, CRYPTOPP_BOOL_X32 and RDRAND64
|
|
|
|
RDRAND::RDRAND()
|
|
{
|
|
if (!HasRDRAND())
|
|
throw RDRAND_Err("HasRDRAND");
|
|
}
|
|
|
|
void RDRAND::GenerateBlock(byte *output, size_t size)
|
|
{
|
|
CRYPTOPP_ASSERT((output && size) || !(output || size));
|
|
if (size == 0) return;
|
|
|
|
#if defined(NASM_RDRAND_ASM_AVAILABLE)
|
|
|
|
NASM_RDRAND_GenerateBlock(output, size);
|
|
|
|
#elif defined(MASM_RDRAND_ASM_AVAILABLE)
|
|
|
|
MASM_RDRAND_GenerateBlock(output, size);
|
|
|
|
#elif CRYPTOPP_BOOL_X64 || CRYPTOPP_BOOL_X32
|
|
size_t i = 0;
|
|
for (i = 0; i < size/8; i++)
|
|
RDRAND64(reinterpret_cast<word64*>(output)+i);
|
|
|
|
output += i*8;
|
|
size -= i*8;
|
|
|
|
if (size)
|
|
{
|
|
word64 val;
|
|
RDRAND64(&val);
|
|
::memcpy(output, &val, size);
|
|
}
|
|
#elif CRYPTOPP_BOOL_X86
|
|
size_t i = 0;
|
|
for (i = 0; i < size/4; i++)
|
|
RDRAND32(reinterpret_cast<word32*>(output)+i);
|
|
|
|
output += i*4;
|
|
size -= i*4;
|
|
|
|
if (size)
|
|
{
|
|
word32 val;
|
|
RDRAND32(&val);
|
|
::memcpy(output, &val, size);
|
|
}
|
|
#else
|
|
// RDRAND not detected at compile time, or no suitable compiler found
|
|
CRYPTOPP_UNUSED(output);
|
|
throw NotImplemented("RDRAND: failed to find a suitable implementation");
|
|
#endif
|
|
}
|
|
|
|
void RDRAND::DiscardBytes(size_t n)
|
|
{
|
|
// RoundUpToMultipleOf is used because a full word is read, and its cheaper
|
|
// to discard full words. There's no sense in dealing with tail bytes.
|
|
FixedSizeSecBlock<word64, 16> discard;
|
|
n = RoundUpToMultipleOf(n, sizeof(word64));
|
|
|
|
size_t count = STDMIN(n, discard.SizeInBytes());
|
|
while (count)
|
|
{
|
|
GenerateBlock(discard.BytePtr(), count);
|
|
n -= count;
|
|
count = STDMIN(n, discard.SizeInBytes());
|
|
}
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////
|
|
/////////////////////////////////////////////////////////////////////
|
|
|
|
// Fills 4 bytes
|
|
inline void RDSEED32(void* output)
|
|
{
|
|
#if defined(__SUNPRO_CC)
|
|
__asm__ __volatile__
|
|
(
|
|
"1:\n"
|
|
".byte 0x0f, 0xc7, 0xf8;\n"
|
|
"jnc 1b;\n"
|
|
: "=a" (*reinterpret_cast<word32*>(output))
|
|
: : "cc"
|
|
);
|
|
#elif defined(GCC_RDSEED_ASM_AVAILABLE) && (CRYPTOPP_GCC_VERSION >= 40800)
|
|
__asm__ __volatile__
|
|
(
|
|
INTEL_NOPREFIX
|
|
ASL(1)
|
|
AS1(rdseed eax)
|
|
ASJ(jnc, 1, b)
|
|
ATT_NOPREFIX
|
|
: "=a" (*reinterpret_cast<word32*>(output))
|
|
: : "cc"
|
|
);
|
|
#elif defined(GCC_RDSEED_ASM_AVAILABLE) && (CRYPTOPP_GCC_VERSION >= 30200)
|
|
__asm__ __volatile__
|
|
(
|
|
"1:\n"
|
|
".byte 0x0f, 0xc7, 0xf8;\n"
|
|
"jnc 1b;\n"
|
|
: "=a" (*reinterpret_cast<word32*>(output))
|
|
: : "cc"
|
|
);
|
|
#elif defined(ALL_RDSEED_INTRIN_AVAILABLE)
|
|
while(!_rdseed32_step(reinterpret_cast<word32*>(output))) {}
|
|
#else
|
|
// RDSEED not detected at compile time, or no suitable compiler found
|
|
CRYPTOPP_UNUSED(output);
|
|
throw NotImplemented("RDSEED: failed to find an implementation");
|
|
#endif
|
|
}
|
|
|
|
#if CRYPTOPP_BOOL_X64 || CRYPTOPP_BOOL_X32
|
|
// Fills 8 bytes
|
|
inline void RDSEED64(void* output)
|
|
{
|
|
#if defined(__SUNPRO_CC) && (__SUNPRO_CC >= 0x5100)
|
|
__asm__ __volatile__
|
|
(
|
|
"1:\n"
|
|
".byte 0x48, 0x0f, 0xc7, 0xf8;\n"
|
|
"jnc 1b;\n"
|
|
: "=a" (*reinterpret_cast<word64*>(output))
|
|
: : "cc"
|
|
);
|
|
#elif defined(GCC_RDSEED_ASM_AVAILABLE) && (CRYPTOPP_GCC_VERSION >= 40800)
|
|
__asm__ __volatile__
|
|
(
|
|
INTEL_NOPREFIX
|
|
ASL(1)
|
|
AS1(rdseed rax)
|
|
ASJ(jnc, 1, b)
|
|
ATT_NOPREFIX
|
|
: "=a" (*reinterpret_cast<word64*>(output))
|
|
: : "cc"
|
|
);
|
|
#elif defined(GCC_RDSEED_ASM_AVAILABLE) && (CRYPTOPP_GCC_VERSION >= 30200)
|
|
__asm__ __volatile__
|
|
(
|
|
"1:\n"
|
|
".byte 0x48, 0x0f, 0xc7, 0xf8;\n"
|
|
"jnc 1b;\n"
|
|
: "=a" (*reinterpret_cast<word64*>(output))
|
|
: : "cc"
|
|
);
|
|
#elif defined(ALL_RDSEED_INTRIN_AVAILABLE)
|
|
while(!_rdseed64_step(reinterpret_cast<unsigned long long*>(output))) {}
|
|
#else
|
|
// RDSEED not detected at compile time, or no suitable compiler found
|
|
CRYPTOPP_UNUSED(output);
|
|
throw NotImplemented("RDSEED: failed to find an implementation");
|
|
#endif
|
|
}
|
|
#endif // CRYPTOPP_BOOL_X64 and RDSEED64
|
|
|
|
RDSEED::RDSEED()
|
|
{
|
|
if (!HasRDSEED())
|
|
throw RDSEED_Err("HasRDSEED");
|
|
}
|
|
|
|
void RDSEED::GenerateBlock(byte *output, size_t size)
|
|
{
|
|
CRYPTOPP_ASSERT((output && size) || !(output || size));
|
|
if (size == 0) return;
|
|
|
|
#if defined(NASM_RDSEED_ASM_AVAILABLE)
|
|
|
|
NASM_RDSEED_GenerateBlock(output, size);
|
|
|
|
#elif defined(MASM_RDSEED_ASM_AVAILABLE)
|
|
|
|
MASM_RDSEED_GenerateBlock(output, size);
|
|
|
|
#elif CRYPTOPP_BOOL_X64 || CRYPTOPP_BOOL_X32
|
|
size_t i = 0;
|
|
for (i = 0; i < size/8; i++)
|
|
RDSEED64(reinterpret_cast<word64*>(output)+i);
|
|
|
|
output += i*8;
|
|
size -= i*8;
|
|
|
|
if (size)
|
|
{
|
|
word64 val;
|
|
RDSEED64(&val);
|
|
::memcpy(output, &val, size);
|
|
}
|
|
#elif CRYPTOPP_BOOL_X86
|
|
size_t i = 0;
|
|
for (i = 0; i < size/4; i++)
|
|
RDSEED32(reinterpret_cast<word32*>(output)+i);
|
|
|
|
output += i*4;
|
|
size -= i*4;
|
|
|
|
if (size)
|
|
{
|
|
word32 val;
|
|
RDSEED32(&val);
|
|
::memcpy(output, &val, size);
|
|
}
|
|
#endif // CRYPTOPP_BOOL_X64, CRYPTOPP_BOOL_X32 and RDSEED64
|
|
}
|
|
|
|
void RDSEED::DiscardBytes(size_t n)
|
|
{
|
|
// RoundUpToMultipleOf is used because a full word is read, and its cheaper
|
|
// to discard full words. There's no sense in dealing with tail bytes.
|
|
FixedSizeSecBlock<word64, 16> discard;
|
|
n = RoundUpToMultipleOf(n, sizeof(word64));
|
|
|
|
size_t count = STDMIN(n, discard.SizeInBytes());
|
|
while (count)
|
|
{
|
|
GenerateBlock(discard.BytePtr(), count);
|
|
n -= count;
|
|
count = STDMIN(n, discard.SizeInBytes());
|
|
}
|
|
}
|
|
|
|
#else // CRYPTOPP_CPUID_AVAILABLE
|
|
|
|
RDRAND::RDRAND()
|
|
{
|
|
throw RDRAND_Err("HasRDRAND");
|
|
}
|
|
|
|
void RDRAND::GenerateBlock(byte *output, size_t size)
|
|
{
|
|
CRYPTOPP_UNUSED(output); CRYPTOPP_UNUSED(size);
|
|
}
|
|
|
|
void RDRAND::DiscardBytes(size_t n)
|
|
{
|
|
CRYPTOPP_UNUSED(n);
|
|
}
|
|
|
|
RDSEED::RDSEED()
|
|
{
|
|
throw RDSEED_Err("HasRDSEED");
|
|
}
|
|
|
|
void RDSEED::GenerateBlock(byte *output, size_t size)
|
|
{
|
|
CRYPTOPP_UNUSED(output); CRYPTOPP_UNUSED(size);
|
|
}
|
|
|
|
void RDSEED::DiscardBytes(size_t n)
|
|
{
|
|
CRYPTOPP_UNUSED(n);
|
|
}
|
|
|
|
#endif
|
|
|
|
NAMESPACE_END
|