Home | History | Annotate | Download | only in encryptor 
      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 "components/webdata/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 = 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 
     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
    144