| OLD | NEW |
| (Empty) |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "components/encryptor/encryptor.h" | |
| 6 | |
| 7 #include <string> | |
| 8 | |
| 9 #include "base/strings/string_util.h" | |
| 10 #include "base/strings/utf_string_conversions.h" | |
| 11 #include "testing/gtest/include/gtest/gtest.h" | |
| 12 | |
| 13 namespace { | |
| 14 | |
| 15 class EncryptorTest : public testing::Test { | |
| 16 public: | |
| 17 EncryptorTest() {} | |
| 18 | |
| 19 virtual void SetUp() { | |
| 20 #if defined(OS_MACOSX) | |
| 21 Encryptor::UseMockKeychain(true); | |
| 22 #endif | |
| 23 } | |
| 24 | |
| 25 private: | |
| 26 DISALLOW_COPY_AND_ASSIGN(EncryptorTest); | |
| 27 }; | |
| 28 | |
| 29 TEST_F(EncryptorTest, String16EncryptionDecryption) { | |
| 30 base::string16 plaintext; | |
| 31 base::string16 result; | |
| 32 std::string utf8_plaintext; | |
| 33 std::string utf8_result; | |
| 34 std::string ciphertext; | |
| 35 | |
| 36 // Test borderline cases (empty strings). | |
| 37 EXPECT_TRUE(Encryptor::EncryptString16(plaintext, &ciphertext)); | |
| 38 EXPECT_TRUE(Encryptor::DecryptString16(ciphertext, &result)); | |
| 39 EXPECT_EQ(plaintext, result); | |
| 40 | |
| 41 // Test a simple string. | |
| 42 plaintext = base::ASCIIToUTF16("hello"); | |
| 43 EXPECT_TRUE(Encryptor::EncryptString16(plaintext, &ciphertext)); | |
| 44 EXPECT_TRUE(Encryptor::DecryptString16(ciphertext, &result)); | |
| 45 EXPECT_EQ(plaintext, result); | |
| 46 | |
| 47 // Test a 16-byte aligned string. This previously hit a boundary error in | |
| 48 // base::Encryptor::Crypt() on Mac. | |
| 49 plaintext = base::ASCIIToUTF16("1234567890123456"); | |
| 50 EXPECT_TRUE(Encryptor::EncryptString16(plaintext, &ciphertext)); | |
| 51 EXPECT_TRUE(Encryptor::DecryptString16(ciphertext, &result)); | |
| 52 EXPECT_EQ(plaintext, result); | |
| 53 | |
| 54 // Test Unicode. | |
| 55 base::char16 wchars[] = { 0xdbeb, 0xdf1b, 0x4e03, 0x6708, 0x8849, | |
| 56 0x661f, 0x671f, 0x56db, 0x597c, 0x4e03, | |
| 57 0x6708, 0x56db, 0x6708, 0xe407, 0xdbaf, | |
| 58 0xdeb5, 0x4ec5, 0x544b, 0x661f, 0x671f, | |
| 59 0x65e5, 0x661f, 0x671f, 0x4e94, 0xd8b1, | |
| 60 0xdce1, 0x7052, 0x5095, 0x7c0b, 0xe586, 0}; | |
| 61 plaintext = wchars; | |
| 62 utf8_plaintext = base::UTF16ToUTF8(plaintext); | |
| 63 EXPECT_EQ(plaintext, base::UTF8ToUTF16(utf8_plaintext)); | |
| 64 EXPECT_TRUE(Encryptor::EncryptString16(plaintext, &ciphertext)); | |
| 65 EXPECT_TRUE(Encryptor::DecryptString16(ciphertext, &result)); | |
| 66 EXPECT_EQ(plaintext, result); | |
| 67 EXPECT_TRUE(Encryptor::DecryptString(ciphertext, &utf8_result)); | |
| 68 EXPECT_EQ(utf8_plaintext, base::UTF16ToUTF8(result)); | |
| 69 | |
| 70 EXPECT_TRUE(Encryptor::EncryptString(utf8_plaintext, &ciphertext)); | |
| 71 EXPECT_TRUE(Encryptor::DecryptString16(ciphertext, &result)); | |
| 72 EXPECT_EQ(plaintext, result); | |
| 73 EXPECT_TRUE(Encryptor::DecryptString(ciphertext, &utf8_result)); | |
| 74 EXPECT_EQ(utf8_plaintext, base::UTF16ToUTF8(result)); | |
| 75 } | |
| 76 | |
| 77 TEST_F(EncryptorTest, EncryptionDecryption) { | |
| 78 std::string plaintext; | |
| 79 std::string result; | |
| 80 std::string ciphertext; | |
| 81 | |
| 82 // Test borderline cases (empty strings). | |
| 83 ASSERT_TRUE(Encryptor::EncryptString(plaintext, &ciphertext)); | |
| 84 ASSERT_TRUE(Encryptor::DecryptString(ciphertext, &result)); | |
| 85 EXPECT_EQ(plaintext, result); | |
| 86 | |
| 87 // Test a simple string. | |
| 88 plaintext = "hello"; | |
| 89 ASSERT_TRUE(Encryptor::EncryptString(plaintext, &ciphertext)); | |
| 90 ASSERT_TRUE(Encryptor::DecryptString(ciphertext, &result)); | |
| 91 EXPECT_EQ(plaintext, result); | |
| 92 | |
| 93 // Make sure it null terminates. | |
| 94 plaintext.assign("hello", 3); | |
| 95 ASSERT_TRUE(Encryptor::EncryptString(plaintext, &ciphertext)); | |
| 96 ASSERT_TRUE(Encryptor::DecryptString(ciphertext, &result)); | |
| 97 EXPECT_EQ(plaintext, "hel"); | |
| 98 } | |
| 99 | |
| 100 TEST_F(EncryptorTest, CypherTextDiffers) { | |
| 101 std::string plaintext; | |
| 102 std::string result; | |
| 103 std::string ciphertext; | |
| 104 | |
| 105 // Test borderline cases (empty strings). | |
| 106 ASSERT_TRUE(Encryptor::EncryptString(plaintext, &ciphertext)); | |
| 107 ASSERT_TRUE(Encryptor::DecryptString(ciphertext, &result)); | |
| 108 // |cyphertext| is empty on the Mac, different on Windows. | |
| 109 EXPECT_TRUE(ciphertext.empty() || plaintext != ciphertext); | |
| 110 EXPECT_EQ(plaintext, result); | |
| 111 | |
| 112 // Test a simple string. | |
| 113 plaintext = "hello"; | |
| 114 ASSERT_TRUE(Encryptor::EncryptString(plaintext, &ciphertext)); | |
| 115 ASSERT_TRUE(Encryptor::DecryptString(ciphertext, &result)); | |
| 116 EXPECT_NE(plaintext, ciphertext); | |
| 117 EXPECT_EQ(plaintext, result); | |
| 118 | |
| 119 // Make sure it null terminates. | |
| 120 plaintext.assign("hello", 3); | |
| 121 ASSERT_TRUE(Encryptor::EncryptString(plaintext, &ciphertext)); | |
| 122 ASSERT_TRUE(Encryptor::DecryptString(ciphertext, &result)); | |
| 123 EXPECT_NE(plaintext, ciphertext); | |
| 124 EXPECT_EQ(result, "hel"); | |
| 125 } | |
| 126 | |
| 127 TEST_F(EncryptorTest, DecryptError) { | |
| 128 std::string plaintext; | |
| 129 std::string result; | |
| 130 std::string ciphertext; | |
| 131 | |
| 132 // Test a simple string, messing with ciphertext prior to decrypting. | |
| 133 plaintext = "hello"; | |
| 134 ASSERT_TRUE(Encryptor::EncryptString(plaintext, &ciphertext)); | |
| 135 EXPECT_NE(plaintext, ciphertext); | |
| 136 ASSERT_LT(4UL, ciphertext.size()); | |
| 137 ciphertext[3] = ciphertext[3] + 1; | |
| 138 EXPECT_FALSE(Encryptor::DecryptString(ciphertext, &result)); | |
| 139 EXPECT_NE(plaintext, result); | |
| 140 EXPECT_TRUE(result.empty()); | |
| 141 } | |
| 142 | |
| 143 } // namespace | |
| OLD | NEW |