1// Copyright (c) 2011 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/autofill/autofill_ie_toolbar_import_win.h"
6
7#include "base/basictypes.h"
8#include "base/string16.h"
9#include "base/win/registry.h"
10#include "chrome/browser/autofill/autofill_profile.h"
11#include "chrome/browser/autofill/credit_card.h"
12#include "chrome/browser/autofill/field_types.h"
13#include "chrome/browser/sync/util/data_encryption.h"
14#include "testing/gtest/include/gtest/gtest.h"
15
16using base::win::RegKey;
17
18// Defined in autofill_ie_toolbar_import_win.cc. Not exposed in the header file.
19bool ImportCurrentUserProfiles(std::vector<AutofillProfile>* profiles,
20                               std::vector<CreditCard>* credit_cards);
21
22namespace {
23
24const wchar_t kUnitTestRegistrySubKey[] = L"SOFTWARE\\Chromium Unit Tests";
25const wchar_t kUnitTestUserOverrideSubKey[] =
26    L"SOFTWARE\\Chromium Unit Tests\\HKCU Override";
27
28const wchar_t* const kProfileKey =
29    L"Software\\Google\\Google Toolbar\\4.0\\Autofill\\Profiles";
30const wchar_t* const kCreditCardKey =
31    L"Software\\Google\\Google Toolbar\\4.0\\Autofill\\Credit Cards";
32const wchar_t* const kPasswordHashValue = L"password_hash";
33const wchar_t* const kSaltValue = L"salt";
34
35struct ValueDescription {
36  wchar_t const* const value_name;
37  wchar_t const* const value;
38};
39
40ValueDescription profile1[] = {
41  { L"name_first", L"John" },
42  { L"name_middle", L"Herman" },
43  { L"name_last", L"Doe" },
44  { L"email", L"jdoe@test.com" },
45  { L"company_name", L"Testcompany" },
46  { L"phone_home_number", L"555-5555" },
47  { L"phone_home_city_code", L"444" },
48  { L"phone_home_country_code", L"1" },
49};
50
51ValueDescription profile2[] = {
52  { L"name_first", L"Jane" },
53  { L"name_last", L"Doe" },
54  { L"email", L"janedoe@test.com" },
55  { L"company_name", L"Testcompany" },
56  { L"phone_fax_number", L"555-6666" },
57  { L"phone_fax_city_code", L"777" },
58  { L"phone_fax_country_code", L"2" },
59};
60
61ValueDescription credit_card[] = {
62  { L"credit_card_name", L"Tommy Gun" },
63  // "4111111111111111" encrypted:
64  { L"credit_card_number", L"\xE53F\x19AB\xC1BF\xC9EB\xECCC\x9BDA\x8515"
65                           L"\xE14D\x6852\x80A8\x50A3\x4375\xFD9F\x1E07"
66                           L"\x790E\x7336\xB773\xAF33\x93EA\xB846\xEC89"
67                           L"\x265C\xD0E6\x4E23\xB75F\x7983" },
68  { L"credit_card_exp_month", L"11" },
69  { L"credit_card_exp_4_digit_year", L"2011" },
70};
71
72ValueDescription empty_salt = {
73  L"salt", L"\x1\x2\x3\x4\x5\x6\x7\x8\x9\xA\xB\xC\xD\xE\xF\x10\x11\x12\x13\x14"
74};
75
76ValueDescription empty_password = {
77  L"password_hash", L""
78};
79
80ValueDescription protected_salt = {
81  L"salt", L"\x4854\xB906\x9C7C\x50A6\x4376\xFD9D\x1E02"
82};
83
84ValueDescription protected_password = {
85  L"password_hash", L"\x18B7\xE586\x459B\x7457\xA066\x3842\x71DA"
86};
87
88void EncryptAndWrite(RegKey* key, const ValueDescription* value) {
89  string data;
90  size_t data_size = (lstrlen(value->value) + 1) * sizeof(wchar_t);
91  data.resize(data_size);
92  memcpy(&data[0], value->value, data_size);
93
94  std::vector<uint8> encrypted_data = EncryptData(data);
95  EXPECT_EQ(ERROR_SUCCESS, key->WriteValue(value->value_name,
96      &encrypted_data[0], encrypted_data.size(), REG_BINARY));
97}
98
99void CreateSubkey(RegKey* key, wchar_t const* subkey_name,
100                  const ValueDescription* values, size_t values_size) {
101  RegKey subkey;
102  subkey.Create(key->Handle(), subkey_name, KEY_ALL_ACCESS);
103  EXPECT_TRUE(subkey.Valid());
104  for (size_t i = 0; i < values_size; ++i)
105    EncryptAndWrite(&subkey, values + i);
106}
107
108}  // namespace
109
110class AutofillIeToolbarImportTest : public testing::Test {
111 public:
112  AutofillIeToolbarImportTest();
113
114  // testing::Test method overrides:
115  virtual void SetUp();
116  virtual void TearDown();
117
118 private:
119  RegKey temp_hkcu_hive_key_;
120
121  DISALLOW_COPY_AND_ASSIGN(AutofillIeToolbarImportTest);
122};
123
124AutofillIeToolbarImportTest::AutofillIeToolbarImportTest() {
125}
126
127void AutofillIeToolbarImportTest::SetUp() {
128  temp_hkcu_hive_key_.Create(HKEY_CURRENT_USER,
129                             kUnitTestUserOverrideSubKey,
130                             KEY_ALL_ACCESS);
131  EXPECT_TRUE(temp_hkcu_hive_key_.Valid());
132  EXPECT_EQ(ERROR_SUCCESS, RegOverridePredefKey(HKEY_CURRENT_USER,
133                                                temp_hkcu_hive_key_.Handle()));
134}
135
136void AutofillIeToolbarImportTest::TearDown() {
137  EXPECT_EQ(ERROR_SUCCESS, RegOverridePredefKey(HKEY_CURRENT_USER, NULL));
138  temp_hkcu_hive_key_.Close();
139  RegKey key(HKEY_CURRENT_USER, kUnitTestRegistrySubKey, KEY_ALL_ACCESS);
140  key.DeleteKey(L"");
141}
142
143TEST_F(AutofillIeToolbarImportTest, TestAutofillImport) {
144  RegKey profile_key;
145  profile_key.Create(HKEY_CURRENT_USER, kProfileKey, KEY_ALL_ACCESS);
146  EXPECT_TRUE(profile_key.Valid());
147
148  CreateSubkey(&profile_key, L"0", profile1, arraysize(profile1));
149  CreateSubkey(&profile_key, L"1", profile2, arraysize(profile2));
150
151  RegKey cc_key;
152  cc_key.Create(HKEY_CURRENT_USER, kCreditCardKey, KEY_ALL_ACCESS);
153  EXPECT_TRUE(cc_key.Valid());
154  CreateSubkey(&cc_key, L"0", credit_card, arraysize(credit_card));
155  EncryptAndWrite(&cc_key, &empty_password);
156  EncryptAndWrite(&cc_key, &empty_salt);
157
158  profile_key.Close();
159  cc_key.Close();
160
161  std::vector<AutofillProfile> profiles;
162  std::vector<CreditCard> credit_cards;
163  EXPECT_TRUE(ImportCurrentUserProfiles(&profiles, &credit_cards));
164  ASSERT_EQ(profiles.size(), 2);
165  // The profiles are read in reverse order.
166  EXPECT_EQ(profiles[1].GetInfo(NAME_FIRST), profile1[0].value);
167  EXPECT_EQ(profiles[1].GetInfo(NAME_MIDDLE), profile1[1].value);
168  EXPECT_EQ(profiles[1].GetInfo(NAME_LAST), profile1[2].value);
169  EXPECT_EQ(profiles[1].GetInfo(EMAIL_ADDRESS), profile1[3].value);
170  EXPECT_EQ(profiles[1].GetInfo(COMPANY_NAME), profile1[4].value);
171  EXPECT_EQ(profiles[1].GetInfo(PHONE_HOME_COUNTRY_CODE), profile1[7].value);
172  EXPECT_EQ(profiles[1].GetInfo(PHONE_HOME_CITY_CODE), profile1[6].value);
173  EXPECT_EQ(profiles[1].GetInfo(PHONE_HOME_NUMBER), L"5555555");
174  EXPECT_EQ(profiles[1].GetInfo(PHONE_HOME_WHOLE_NUMBER), L"14445555555");
175
176  EXPECT_EQ(profiles[0].GetInfo(NAME_FIRST), profile2[0].value);
177  EXPECT_EQ(profiles[0].GetInfo(NAME_LAST), profile2[1].value);
178  EXPECT_EQ(profiles[0].GetInfo(EMAIL_ADDRESS), profile2[2].value);
179  EXPECT_EQ(profiles[0].GetInfo(COMPANY_NAME), profile2[3].value);
180  EXPECT_EQ(profiles[0].GetInfo(PHONE_FAX_COUNTRY_CODE), profile2[6].value);
181  EXPECT_EQ(profiles[0].GetInfo(PHONE_FAX_CITY_CODE), profile2[5].value);
182  EXPECT_EQ(profiles[0].GetInfo(PHONE_FAX_NUMBER), L"5556666");
183  EXPECT_EQ(profiles[0].GetInfo(PHONE_FAX_WHOLE_NUMBER), L"27775556666");
184
185  ASSERT_EQ(credit_cards.size(), 1);
186  EXPECT_EQ(credit_cards[0].GetInfo(CREDIT_CARD_NAME), credit_card[0].value);
187  EXPECT_EQ(credit_cards[0].GetInfo(CREDIT_CARD_NUMBER), L"4111111111111111");
188  EXPECT_EQ(credit_cards[0].GetInfo(CREDIT_CARD_EXP_MONTH),
189            credit_card[2].value);
190  EXPECT_EQ(credit_cards[0].GetInfo(CREDIT_CARD_EXP_4_DIGIT_YEAR),
191            credit_card[3].value);
192
193  // Mock password encrypted cc.
194  cc_key.Open(HKEY_CURRENT_USER, kCreditCardKey, KEY_ALL_ACCESS);
195  EXPECT_TRUE(cc_key.Valid());
196  EncryptAndWrite(&cc_key, &protected_password);
197  EncryptAndWrite(&cc_key, &protected_salt);
198  cc_key.Close();
199
200  profiles.clear();
201  credit_cards.clear();
202  EXPECT_TRUE(ImportCurrentUserProfiles(&profiles, &credit_cards));
203  // Profiles are not protected.
204  EXPECT_EQ(profiles.size(), 2);
205  // Credit cards are.
206  EXPECT_EQ(credit_cards.size(), 0);
207}
208
209