mirror of
https://github.com/XTXMarkets/ternfs.git
synced 2026-01-02 08:59:36 -06:00
110705db8d
Things not done because probably disruptive: * kmod filesystem string * sysctl/debugfs/trace * metrics names * xmon instance names Some of these might be renamed too, but starting with a relatively safe set.
113 lines
3.8 KiB
C++
113 lines
3.8 KiB
C++
// Code from
|
|
// <https://www.intel.com/content/dam/doc/white-paper/advanced-encryption-standard-new-instructions-set-paper.pdf>.
|
|
#include <emmintrin.h>
|
|
#include <wmmintrin.h>
|
|
#include <string.h>
|
|
#include <fcntl.h>
|
|
#include <unistd.h>
|
|
#include <sys/random.h>
|
|
|
|
#include "Crypto.hpp"
|
|
#include "Exception.hpp"
|
|
|
|
|
|
void generateSecretKey(std::array<uint8_t, 16>& key) {
|
|
ssize_t read = getrandom(key.data(), key.size(), 0);
|
|
if (read < 0) {
|
|
throw SYSCALL_EXCEPTION("getrandom");
|
|
}
|
|
if (read != key.size()) {
|
|
// getrandom(2) states that once initialized you can always get up to 256 bytes.
|
|
throw TERN_EXCEPTION("could not read %s random bytes, read %s instead!", key.size(), read);
|
|
}
|
|
}
|
|
|
|
inline __m128i AES_128_ASSIST(__m128i temp1, __m128i temp2) {
|
|
__m128i temp3;
|
|
temp2 = _mm_shuffle_epi32(temp2 ,0xff);
|
|
temp3 = _mm_slli_si128(temp1, 0x4);
|
|
temp1 = _mm_xor_si128(temp1, temp3);
|
|
temp3 = _mm_slli_si128(temp3, 0x4);
|
|
temp1 = _mm_xor_si128(temp1, temp3);
|
|
temp3 = _mm_slli_si128(temp3, 0x4);
|
|
temp1 = _mm_xor_si128(temp1, temp3);
|
|
temp1 = _mm_xor_si128(temp1, temp2);
|
|
return temp1;
|
|
}
|
|
|
|
void expandKey(const std::array<uint8_t, 16>& userkey, AES128Key& key) {
|
|
__m128i temp1, temp2;
|
|
__m128i *Key_Schedule = (__m128i*)&key;
|
|
temp1 = _mm_loadu_si128((__m128i*)userkey.data());
|
|
Key_Schedule[0] = temp1;
|
|
temp2 = _mm_aeskeygenassist_si128(temp1, 0x1);
|
|
temp1 = AES_128_ASSIST(temp1, temp2);
|
|
Key_Schedule[1] = temp1;
|
|
temp2 = _mm_aeskeygenassist_si128(temp1, 0x2);
|
|
temp1 = AES_128_ASSIST(temp1, temp2);
|
|
Key_Schedule[2] = temp1;
|
|
temp2 = _mm_aeskeygenassist_si128(temp1, 0x4);
|
|
temp1 = AES_128_ASSIST(temp1, temp2);
|
|
Key_Schedule[3] = temp1;
|
|
temp2 = _mm_aeskeygenassist_si128(temp1, 0x8);
|
|
temp1 = AES_128_ASSIST(temp1, temp2);
|
|
Key_Schedule[4] = temp1;
|
|
temp2 = _mm_aeskeygenassist_si128(temp1, 0x10);
|
|
temp1 = AES_128_ASSIST(temp1, temp2);
|
|
Key_Schedule[5] = temp1;
|
|
temp2 = _mm_aeskeygenassist_si128(temp1, 0x20);
|
|
temp1 = AES_128_ASSIST(temp1, temp2);
|
|
Key_Schedule[6] = temp1;
|
|
temp2 = _mm_aeskeygenassist_si128(temp1, 0x40);
|
|
temp1 = AES_128_ASSIST(temp1, temp2);
|
|
Key_Schedule[7] = temp1;
|
|
temp2 = _mm_aeskeygenassist_si128(temp1, 0x80);
|
|
temp1 = AES_128_ASSIST(temp1, temp2);
|
|
Key_Schedule[8] = temp1;
|
|
temp2 = _mm_aeskeygenassist_si128(temp1, 0x1b);
|
|
temp1 = AES_128_ASSIST(temp1, temp2);
|
|
Key_Schedule[9] = temp1;
|
|
temp2 = _mm_aeskeygenassist_si128(temp1, 0x36);
|
|
temp1 = AES_128_ASSIST(temp1, temp2);
|
|
Key_Schedule[10] = temp1;
|
|
}
|
|
|
|
std::array<uint8_t, 8> cbcmac(const AES128Key& key, const uint8_t* data, size_t len) {
|
|
// load key
|
|
__m128i xmmKey[11];
|
|
for (int i = 0; i < 11; i++) {
|
|
xmmKey[i] = _mm_load_si128((__m128i*)(&key) + i);
|
|
}
|
|
// CBC MAC step
|
|
__m128i block = _mm_setzero_si128();
|
|
__m128i dataBlock;
|
|
auto step = [&xmmKey, &block, &dataBlock]() {
|
|
// CBC xor
|
|
block = _mm_xor_si128(block, dataBlock);
|
|
// encrypt
|
|
block = _mm_xor_si128(block, xmmKey[0]); // Whitening step (Round 0)
|
|
for (int i = 1; i < 10; i++) {
|
|
block = _mm_aesenc_si128(block, xmmKey[i]); // Round i
|
|
}
|
|
block = _mm_aesenclast_si128(block, xmmKey[10]); // Round 10
|
|
};
|
|
// unpadded load
|
|
size_t i = 0;
|
|
for (; len-i >= 16; i += 16) {
|
|
dataBlock = _mm_loadu_si128((__m128i*)(data+i));
|
|
step();
|
|
}
|
|
// zero-padded load
|
|
ALIGNED(16) uint8_t scratch[16];
|
|
if (len-i > 0) {
|
|
memset(scratch, 0, 16);
|
|
memcpy(scratch, data+i, len-i);
|
|
dataBlock = _mm_load_si128((__m128i*)scratch);
|
|
step();
|
|
}
|
|
// extract MAC
|
|
_mm_store_si128((__m128i*)scratch, block);
|
|
std::array<uint8_t, 8> mac;
|
|
memcpy(mac.data(), scratch, 8);
|
|
return mac;
|
|
} |