15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "content/browser/net/sqlite_persistent_cookie_store.h" 6c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind.h" 82a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/compiler_specific.h" 92a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/files/scoped_temp_dir.h" 102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/sequenced_task_runner.h" 11868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/stringprintf.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/synchronization/waitable_event.h" 13d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)#include "base/test/perf_time_logger.h" 142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/test/sequenced_worker_pool_owner.h" 152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/threading/sequenced_worker_pool.h" 165d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "content/public/browser/cookie_crypto_delegate.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/cookies/canonical_cookie.h" 18c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "net/cookies/cookie_constants.h" 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 207dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch#include "url/gurl.h" 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 22c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)namespace content { 23c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 24c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)namespace { 25c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 26c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)const base::FilePath::CharType cookie_filename[] = FILE_PATH_LITERAL("Cookies"); 27c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 28c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} // namespace 29c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class SQLitePersistentCookieStorePerfTest : public testing::Test { 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SQLitePersistentCookieStorePerfTest() 332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) : pool_owner_(new base::SequencedWorkerPoolOwner(1, "Background Pool")), 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) loaded_event_(false, false), 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) key_loaded_event_(false, false) { 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void OnLoaded(const std::vector<net::CanonicalCookie*>& cookies) { 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cookies_ = cookies; 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) loaded_event_.Signal(); 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void OnKeyLoaded(const std::vector<net::CanonicalCookie*>& cookies) { 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cookies_ = cookies; 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) key_loaded_event_.Signal(); 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void Load() { 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) store_->Load(base::Bind(&SQLitePersistentCookieStorePerfTest::OnLoaded, 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Unretained(this))); 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) loaded_event_.Wait(); 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_refptr<base::SequencedTaskRunner> background_task_runner() { 552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return pool_owner_->pool()->GetSequencedTaskRunner( 562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pool_owner_->pool()->GetNamedSequenceToken("background")); 572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_refptr<base::SequencedTaskRunner> client_task_runner() { 602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return pool_owner_->pool()->GetSequencedTaskRunner( 612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pool_owner_->pool()->GetNamedSequenceToken("client")); 622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void SetUp() OVERRIDE { 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) store_ = new SQLitePersistentCookieStore( 67c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) temp_dir_.path().Append(cookie_filename), 682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) client_task_runner(), 692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) background_task_runner(), 705d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) false, NULL, NULL); 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<net::CanonicalCookie*> cookies; 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Load(); 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(0u, cookies_.size()); 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Creates 15000 cookies from 300 eTLD+1s. 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Time t = base::Time::Now(); 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (int domain_num = 0; domain_num < 300; domain_num++) { 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string domain_name(base::StringPrintf(".domain_%d.com", domain_num)); 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GURL gurl("www" + domain_name); 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (int cookie_num = 0; cookie_num < 50; ++cookie_num) { 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) t += base::TimeDelta::FromInternalValue(10); 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) store_->AddCookie( 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) net::CanonicalCookie(gurl, 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::StringPrintf("Cookie_%d", cookie_num), "1", 84c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) domain_name, "/", t, t, t, false, false, 85c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) net::COOKIE_PRIORITY_DEFAULT)); 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Replace the store effectively destroying the current one and forcing it 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // to write its data to disk. 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) store_ = NULL; 912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Shut down the pool, causing deferred (no-op) commits to be discarded. 932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pool_owner_->pool()->Shutdown(); 942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // ~SequencedWorkerPoolOwner blocks on pool shutdown. 952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pool_owner_.reset(new base::SequencedWorkerPoolOwner(1, "pool")); 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) store_ = new SQLitePersistentCookieStore( 98c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) temp_dir_.path().Append(cookie_filename), 99c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) client_task_runner(), 100c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) background_task_runner(), 1015d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) false, NULL, NULL); 1022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 1032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void TearDown() OVERRIDE { 1052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) store_ = NULL; 1062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pool_owner_->pool()->Shutdown(); 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) protected: 1102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_ptr<base::SequencedWorkerPoolOwner> pool_owner_; 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::WaitableEvent loaded_event_; 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::WaitableEvent key_loaded_event_; 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<net::CanonicalCookie*> cookies_; 1142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::ScopedTempDir temp_dir_; 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<SQLitePersistentCookieStore> store_; 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Test the performance of priority load of cookies for a specfic domain key 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(SQLitePersistentCookieStorePerfTest, TestLoadForKeyPerformance) { 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (int domain_num = 0; domain_num < 3; ++domain_num) { 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string domain_name(base::StringPrintf("domain_%d.com", domain_num)); 122d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) base::PerfTimeLogger timer( 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ("Load cookies for the eTLD+1 " + domain_name).c_str()); 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) store_->LoadCookiesForKey(domain_name, 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Bind(&SQLitePersistentCookieStorePerfTest::OnKeyLoaded, 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Unretained(this))); 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) key_loaded_event_.Wait(); 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) timer.Done(); 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(50U, cookies_.size()); 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Test the performance of load 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(SQLitePersistentCookieStorePerfTest, TestLoadPerformance) { 136d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) base::PerfTimeLogger timer("Load all cookies"); 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Load(); 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) timer.Done(); 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(15000U, cookies_.size()); 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 142c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 143c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} // namespace content 144