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

Improve AES CTR test.

Browse Source 
GitOrigin-RevId: a65b34e408f972e737489222d9f3e0604582fa3d
This commit is contained in:
levlam 2020-09-27 22:07:42 +03:00
parent 6278121549
commit 013db25897
1 changed files with 18 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
tdutils/test/crypto.cpp
View File

@ -71,17 +71,32 @@ TEST(Crypto, AesCtrState) {
td::AesCtrState state; td::AesCtrState state;
state.init(as_slice(key), as_slice(iv)); state.init(as_slice(key), as_slice(iv));
td::string t(length, '\0'); td::string t(length, '\0');
state.encrypt(s, t); std::size_t pos = 0;
for (auto str : td::rand_split(td::string(length, '\0'))) {
auto len = str.size();
state.encrypt(td::Slice(s).substr(pos, len), td::MutableSlice(t).substr(pos, len));
pos += len;
}
ASSERT_EQ(answers1[i], td::crc32(t)); ASSERT_EQ(answers1[i], td::crc32(t));
state.init(as_slice(key), as_slice(iv)); state.init(as_slice(key), as_slice(iv));
state.decrypt(t, t); pos = 0;
for (auto str : td::rand_split(td::string(length, '\0'))) {
auto len = str.size();
state.decrypt(td::Slice(t).substr(pos, len), td::MutableSlice(t).substr(pos, len));
pos += len;
}
ASSERT_STREQ(td::base64_encode(s), td::base64_encode(t)); ASSERT_STREQ(td::base64_encode(s), td::base64_encode(t));
for (auto &c : iv.raw) { for (auto &c : iv.raw) {
c = 0xFF; c = 0xFF;
} }
state.init(as_slice(key), as_slice(iv)); state.init(as_slice(key), as_slice(iv));
state.encrypt(s, t); pos = 0;
for (auto str : td::rand_split(td::string(length, '\0'))) {
auto len = str.size();
state.encrypt(td::Slice(s).substr(pos, len), td::MutableSlice(t).substr(pos, len));
pos += len;
}
ASSERT_EQ(answers2[i], td::crc32(t)); ASSERT_EQ(answers2[i], td::crc32(t));
i++; i++;