tdlight-team/tdlight
tdlight-team/tdlight
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Simplify XorBytes.

Browse Source 
GitOrigin-RevId: d20ae3c1d44d0a88068b42979685e5b1f9df0193
This commit is contained in:
levlam 2020-06-16 01:13:45 +03:00
parent 44cd25afae
commit e1515226be
2 changed files with 21 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
benchmark/bench_crypto.cpp
View File

@ -346,11 +346,11 @@ class Crc64Bench : public td::Benchmark {
int main() {
td::init_openssl_threads();
td::bench(AesIgeShortBench());
td::bench(AesIgeEncryptBench());
td::bench(AesIgeDecryptBench());
td::bench(AesCtrBench());
td::bench(AesEcbBench());
td::bench(AesIgeShortBench());
td::bench(Pbkdf2Bench());
td::bench(RandBench());

50
tdutils/td/utils/crypto.cpp
View File

@ -75,11 +75,13 @@ struct AesBlock {
}
AesBlock inc() const {
AesBlock res;
AesBlock res = *this;
auto ptr = res.raw();
for (int i = 15; i >= 0; i--) {
if (++ptr[i] != 0) {
break;
if (++ptr[15] == 0) {
for (int i = 14; i >= 0; i--) {
if (++ptr[i] != 0) {
break;
}
}
}
return res;
@ -91,21 +93,23 @@ static_assert(sizeof(AesBlock) == AES_BLOCK_SIZE, "");
class XorBytes {
public:
static void run(const uint8 *a, const uint8 *b, uint8 *c, size_t n) {
XorBytes xorer;
xorer.a = a;
xorer.b = b;
xorer.c = c;
xorer.n = n;
xorer.step<16>();
xorer.step<1>();
static constexpr int BLOCK_SIZE = 16;
auto block_cnt = n / BLOCK_SIZE;
n -= block_cnt * BLOCK_SIZE;
while (block_cnt-- > 0) {
Block<BLOCK_SIZE> a_big = as<Block<BLOCK_SIZE>>(a);
Block<BLOCK_SIZE> b_big = as<Block<BLOCK_SIZE>>(b);
as<Block<BLOCK_SIZE>>(c) = a_big ^ b_big;
a += BLOCK_SIZE;
b += BLOCK_SIZE;
c += BLOCK_SIZE;
}
while (n-- > 0) {
c[n] = a[n] ^ b[n];
}
}
private:
const uint8 *a;
const uint8 *b;
uint8 *c;
size_t n;
template <size_t N>
struct alignas(N) Block {
uint8 data[N];
@ -117,20 +121,6 @@ class XorBytes {
return res;
}
};
template <size_t N>
void step() {
auto cnt = n / N;
n -= cnt * N;
for (size_t i = 0; i < cnt; i++) {
Block<N> a_big = as<Block<N>>(a);
Block<N> b_big = as<Block<N>>(b);
as<Block<N>>(c) = a_big ^ b_big;
a += N;
b += N;
c += N;
}
}
};
struct AesCtrCounterPack {