blob: b84162adad16110f8470fb71619abc625ef2c75e [file] [log] [blame]
// Copyright (c) 2011 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <base/file_util.h>
#include <base/memory/scoped_temp_dir.h>
#include <gtest/gtest.h>
#include "shill/crypto_des_cbc.h"
#include "shill/crypto_provider.h"
#include "shill/crypto_rot47.h"
#include "shill/glib.h"
using std::string;
using testing::Test;
namespace shill {
namespace {
const char kTestKey[] = "12345678";
const char kTestIV[] = "abcdefgh";
const char kKeyMatterFile[] = "key-matter-file";
const char kEmptyText[] = "";
const char kPlainText[] = "This is a test!";
const char kROT47Text[] = "rot47:%9:D :D 2 E6DEP";
const char kDESCBCText[] = "des-cbc:02:bKlHDISdHMFc0teQd4mAVrXgwlSj6iA+";
} // namespace {}
class CryptoProviderTest : public Test {
public:
CryptoProviderTest() : provider_(&glib_) {}
protected:
FilePath InitKeyMatterFile(const FilePath &dir);
GLib glib_; // Use actual GLib for testing.
CryptoProvider provider_;
};
FilePath CryptoProviderTest::InitKeyMatterFile(const FilePath &dir) {
FilePath path = dir.Append(kKeyMatterFile);
string matter = string(kTestIV) + kTestKey;
file_util::WriteFile(path, matter.data(), matter.size());
return path;
}
TEST_F(CryptoProviderTest, Init) {
EXPECT_EQ(CryptoProvider::kKeyMatterFile, provider_.key_matter_file_.value());
provider_.set_key_matter_file(FilePath("/some/non/existent/file"));
provider_.Init();
ASSERT_EQ(1, provider_.cryptos_.size());
EXPECT_EQ(CryptoROT47::kID, provider_.cryptos_[0]->GetID());
ScopedTempDir temp_dir;
ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
provider_.set_key_matter_file(InitKeyMatterFile(temp_dir.path()));
provider_.Init();
ASSERT_EQ(2, provider_.cryptos_.size());
EXPECT_EQ(CryptoDESCBC::kID, provider_.cryptos_[0]->GetID());
EXPECT_EQ(CryptoROT47::kID, provider_.cryptos_[1]->GetID());
provider_.set_key_matter_file(FilePath("/other/missing/file"));
provider_.Init();
ASSERT_EQ(1, provider_.cryptos_.size());
EXPECT_EQ(CryptoROT47::kID, provider_.cryptos_[0]->GetID());
}
TEST_F(CryptoProviderTest, Encrypt) {
EXPECT_EQ(kPlainText, provider_.Encrypt(kPlainText));
EXPECT_EQ(kEmptyText, provider_.Encrypt(kEmptyText));
provider_.Init();
EXPECT_EQ(kROT47Text, provider_.Encrypt(kPlainText));
ScopedTempDir temp_dir;
ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
provider_.set_key_matter_file(InitKeyMatterFile(temp_dir.path()));
provider_.Init();
EXPECT_EQ(kDESCBCText, provider_.Encrypt(kPlainText));
}
TEST_F(CryptoProviderTest, Decrypt) {
EXPECT_EQ(kPlainText, provider_.Decrypt(kPlainText));
EXPECT_EQ(kEmptyText, provider_.Decrypt(kEmptyText));
ScopedTempDir temp_dir;
ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
provider_.set_key_matter_file(InitKeyMatterFile(temp_dir.path()));
provider_.Init();
EXPECT_EQ(kPlainText, provider_.Decrypt(kROT47Text));
EXPECT_EQ(kPlainText, provider_.Decrypt(kDESCBCText));
EXPECT_EQ(kPlainText, provider_.Decrypt(kPlainText));
EXPECT_EQ(kEmptyText, provider_.Decrypt(kEmptyText));
}
} // namespace shill