100 lines
3.6 KiB
C++
100 lines
3.6 KiB
C++
#include <string>
|
|
#include "gtest/gtest.h"
|
|
#include "utils.h"
|
|
#include "crypto.h"
|
|
#include "test.h"
|
|
|
|
using namespace horcrux;
|
|
TEST(CryptoTest, encrypt1){
|
|
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;
|
|
a.encrypt(key, iv, plaintext, output);
|
|
|
|
EXPECT_EQ(test1_enc, output);
|
|
}
|
|
|
|
TEST(CryptoTest, encrypt2){
|
|
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);
|
|
}
|
|
|
|
TEST(CryptoTest, encrypt3){
|
|
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());
|
|
}
|
|
|
|
TEST(CryptoTest, decrypt1){
|
|
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());
|
|
|
|
AES256_CBC a;
|
|
a.decrypt(key, iv, test1_enc, output);
|
|
|
|
EXPECT_EQ(plaintext, output);
|
|
}
|
|
TEST(CryptoTest, decrypt2){
|
|
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);
|
|
}
|
|
TEST(CryptoTest, decrypt3){
|
|
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);
|
|
}
|
|
TEST(CryptoTest, endToEnd){
|
|
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);
|
|
}
|