components/webdata/encryptor/encryptor_unittest.cc

   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 = 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