1 // Copyright (c) 2010 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 "chrome/browser/password_manager/encryptor.h"
6
7 #include <string>
8
9 #include "base/string_util.h"
10 #include "base/utf_string_conversions.h"
11 #include "testing/gtest/include/gtest/gtest.h"
12
13 namespace {
14
15 class EncryptorTest : public testing::Test {
16 public:
EncryptorTest()17 EncryptorTest() {}
18
SetUp()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
TEST_F(EncryptorTest,String16EncryptionDecryption)29 TEST_F(EncryptorTest, String16EncryptionDecryption) {
30 string16 plaintext;
31 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 = 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 = 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 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 = UTF16ToUTF8(plaintext);
63 EXPECT_EQ(plaintext, 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, 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, UTF16ToUTF8(result));
75 }
76
TEST_F(EncryptorTest,EncryptionDecryption)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
TEST_F(EncryptorTest,CypherTextDiffers)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
TEST_F(EncryptorTest,DecryptError)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
144