2020-12-11 11:38:02 +00:00
|
|
|
#include <string>
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
#include "utils.h"
|
|
|
|
#include "crypto.h"
|
2020-12-12 14:36:38 +00:00
|
|
|
#include "test.h"
|
2020-12-11 11:38:02 +00:00
|
|
|
|
2020-12-12 14:36:38 +00:00
|
|
|
using namespace horcrux;
|
2020-12-12 12:24:54 +00:00
|
|
|
TEST(CryptoTest, encrypt1){
|
2020-12-12 23:56:18 +00:00
|
|
|
const std::vector<unsigned char> plaintext(test1_str.begin(), test1_str.end());
|
|
|
|
size_t output_len = plaintext.size() + 16;
|
|
|
|
std::vector<unsigned char> output(output_len);
|
|
|
|
const std::vector<unsigned char> key(test1_key.begin(), test1_key.end());
|
|
|
|
const std::vector<unsigned char> iv(test1_iv.begin(), test1_iv.end());
|
2020-12-11 11:38:02 +00:00
|
|
|
|
2020-12-12 23:56:18 +00:00
|
|
|
AES256_CBC a;
|
|
|
|
a.encrypt(key, iv, plaintext, output);
|
2020-12-11 11:38:02 +00:00
|
|
|
|
2020-12-12 23:56:18 +00:00
|
|
|
EXPECT_EQ(test1_enc, output);
|
2020-12-11 11:38:02 +00:00
|
|
|
}
|
2020-12-12 12:24:54 +00:00
|
|
|
|
|
|
|
TEST(CryptoTest, encrypt2){
|
2020-12-12 23:56:18 +00:00
|
|
|
const std::vector<unsigned char> plaintext(test1_str.begin(), test1_str.end());
|
|
|
|
//size_t output_len = plaintext.size() + 16;
|
|
|
|
//std::vector<unsigned char> output(output_len);
|
|
|
|
const std::vector<unsigned char> key(test1_key.begin(), test1_key.end());
|
|
|
|
const std::vector<unsigned char> iv(test1_iv.begin(), test1_iv.end());
|
|
|
|
|
|
|
|
AES256_CBC a;
|
|
|
|
auto output = a.encrypt(key, iv, plaintext);
|
|
|
|
auto temp = iv;
|
|
|
|
temp.insert(temp.end(), test1_enc.begin(), test1_enc.end());
|
|
|
|
|
|
|
|
EXPECT_EQ(temp.size(), output.size());
|
|
|
|
EXPECT_EQ(temp, output);
|
2020-12-12 12:24:54 +00:00
|
|
|
}
|
|
|
|
|
2020-12-12 13:12:42 +00:00
|
|
|
TEST(CryptoTest, encrypt3){
|
2020-12-12 23:56:18 +00:00
|
|
|
const std::vector<unsigned char> plaintext(test1_str.begin(), test1_str.end());
|
|
|
|
//size_t output_len = plaintext.size() + 16;
|
|
|
|
//std::vector<unsigned char> output(output_len);
|
|
|
|
const std::vector<unsigned char> key(test1_key.begin(), test1_key.end());
|
|
|
|
const std::vector<unsigned char> iv(test1_iv.begin(), test1_iv.end());
|
|
|
|
|
|
|
|
AES256_CBC a(key);
|
|
|
|
auto output = a.encrypt(plaintext);
|
|
|
|
auto temp = iv;
|
|
|
|
temp.insert(temp.end(), test1_enc.begin(), test1_enc.end());
|
|
|
|
|
|
|
|
EXPECT_EQ(temp.size(), output.size());
|
2020-12-12 13:12:42 +00:00
|
|
|
}
|
|
|
|
|
2020-12-12 12:24:54 +00:00
|
|
|
TEST(CryptoTest, decrypt1){
|
2020-12-12 23:56:18 +00:00
|
|
|
const std::vector<unsigned char> plaintext(test1_str.begin(), test1_str.end());
|
|
|
|
size_t output_len = test1_enc.size();
|
|
|
|
std::vector<unsigned char> output(output_len);
|
|
|
|
const std::vector<unsigned char> key(test1_key.begin(), test1_key.end());
|
|
|
|
const std::vector<unsigned char> iv(test1_iv.begin(), test1_iv.end());
|
2020-12-12 12:24:54 +00:00
|
|
|
|
2020-12-12 23:56:18 +00:00
|
|
|
AES256_CBC a;
|
|
|
|
a.decrypt(key, iv, test1_enc, output);
|
2020-12-12 12:24:54 +00:00
|
|
|
|
2020-12-12 23:56:18 +00:00
|
|
|
EXPECT_EQ(plaintext, output);
|
2020-12-12 12:24:54 +00:00
|
|
|
}
|
|
|
|
TEST(CryptoTest, decrypt2){
|
2020-12-12 23:56:18 +00:00
|
|
|
const std::vector<unsigned char> plaintext(test1_str.begin(), test1_str.end());
|
|
|
|
size_t output_len = test1_enc.size();
|
|
|
|
const std::vector<unsigned char> key(test1_key.begin(), test1_key.end());
|
|
|
|
const std::vector<unsigned char> iv(test1_iv.begin(), test1_iv.end());
|
|
|
|
// constructs encrypted input (iv + encrypted_data)
|
|
|
|
auto input = std::vector<unsigned char>(iv);
|
|
|
|
input.insert(input.end(),test1_enc.begin(), test1_enc.end());
|
|
|
|
|
|
|
|
AES256_CBC a;
|
|
|
|
auto output = a.decrypt(key, input);
|
|
|
|
|
|
|
|
EXPECT_EQ(plaintext, output);
|
2020-12-12 13:12:42 +00:00
|
|
|
}
|
|
|
|
TEST(CryptoTest, decrypt3){
|
2020-12-12 23:56:18 +00:00
|
|
|
const std::vector<unsigned char> plaintext(test1_str.begin(), test1_str.end());
|
|
|
|
size_t output_len = test1_enc.size();
|
|
|
|
const std::vector<unsigned char> key(test1_key.begin(), test1_key.end());
|
|
|
|
const std::vector<unsigned char> iv(test1_iv.begin(), test1_iv.end());
|
|
|
|
// constructs encrypted input (iv + encrypted_data)
|
|
|
|
auto input = std::vector<unsigned char>(iv);
|
|
|
|
input.insert(input.end(),test1_enc.begin(), test1_enc.end());
|
|
|
|
|
|
|
|
AES256_CBC a(key);
|
|
|
|
auto output = a.decrypt(input);
|
|
|
|
|
|
|
|
EXPECT_EQ(plaintext, output);
|
2020-12-12 12:24:54 +00:00
|
|
|
}
|
2020-12-12 13:12:42 +00:00
|
|
|
TEST(CryptoTest, endToEnd){
|
2020-12-12 23:56:18 +00:00
|
|
|
const std::vector<unsigned char> plaintext(test1_str.begin(), test1_str.end());
|
|
|
|
AES256_CBC a;
|
|
|
|
auto crypto = a.encrypt(plaintext);
|
|
|
|
EXPECT_GT(crypto.size(), plaintext.size());
|
|
|
|
auto decrypted = a.decrypt(crypto);
|
|
|
|
EXPECT_EQ(plaintext, decrypted);
|
2020-12-12 13:12:42 +00:00
|
|
|
}
|