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