|  | // Copyright 2014 The Chromium Authors. All rights reserved. | 
|  | // Use of this source code is governed by a BSD-style license that can be | 
|  | // found in the LICENSE file. | 
|  |  | 
|  | #include "components/update_client/update_checker.h" | 
|  |  | 
|  | #include <memory> | 
|  |  | 
|  | #include "base/bind.h" | 
|  | #include "base/bind_helpers.h" | 
|  | #include "base/files/file_util.h" | 
|  | #include "base/macros.h" | 
|  | #include "base/memory/ref_counted.h" | 
|  | #include "base/path_service.h" | 
|  | #include "base/run_loop.h" | 
|  | #include "base/threading/thread_task_runner_handle.h" | 
|  | #include "base/version.h" | 
|  | #include "components/prefs/testing_pref_service.h" | 
|  | #include "components/update_client/crx_update_item.h" | 
|  | #include "components/update_client/persisted_data.h" | 
|  | #include "components/update_client/test_configurator.h" | 
|  | #include "components/update_client/url_request_post_interceptor.h" | 
|  | #include "net/url_request/url_request_test_util.h" | 
|  | #include "testing/gtest/include/gtest/gtest.h" | 
|  | #include "url/gurl.h" | 
|  |  | 
|  | using std::string; | 
|  |  | 
|  | namespace update_client { | 
|  |  | 
|  | namespace { | 
|  |  | 
|  | base::FilePath test_file(const char* file) { | 
|  | base::FilePath path; | 
|  | PathService::Get(base::DIR_SOURCE_ROOT, &path); | 
|  | return path.AppendASCII("components") | 
|  | .AppendASCII("test") | 
|  | .AppendASCII("data") | 
|  | .AppendASCII("update_client") | 
|  | .AppendASCII(file); | 
|  | } | 
|  |  | 
|  | }  // namespace | 
|  |  | 
|  | class UpdateCheckerTest : public testing::Test { | 
|  | public: | 
|  | UpdateCheckerTest(); | 
|  | ~UpdateCheckerTest() override; | 
|  |  | 
|  | // Overrides from testing::Test. | 
|  | void SetUp() override; | 
|  | void TearDown() override; | 
|  |  | 
|  | void UpdateCheckComplete(int error, | 
|  | const UpdateResponse::Results& results, | 
|  | int retry_after_sec); | 
|  |  | 
|  | protected: | 
|  | void Quit(); | 
|  | void RunThreads(); | 
|  | void RunThreadsUntilIdle(); | 
|  |  | 
|  | CrxUpdateItem BuildCrxUpdateItem(); | 
|  |  | 
|  | scoped_refptr<TestConfigurator> config_; | 
|  | std::unique_ptr<TestingPrefServiceSimple> pref_; | 
|  | std::unique_ptr<PersistedData> metadata_; | 
|  |  | 
|  | std::unique_ptr<UpdateChecker> update_checker_; | 
|  |  | 
|  | std::unique_ptr<InterceptorFactory> interceptor_factory_; | 
|  | URLRequestPostInterceptor* post_interceptor_;  // Owned by the factory. | 
|  |  | 
|  | int error_; | 
|  | UpdateResponse::Results results_; | 
|  |  | 
|  | private: | 
|  | base::MessageLoopForIO loop_; | 
|  | base::Closure quit_closure_; | 
|  |  | 
|  | DISALLOW_COPY_AND_ASSIGN(UpdateCheckerTest); | 
|  | }; | 
|  |  | 
|  | UpdateCheckerTest::UpdateCheckerTest() : post_interceptor_(NULL), error_(0) { | 
|  | } | 
|  |  | 
|  | UpdateCheckerTest::~UpdateCheckerTest() { | 
|  | } | 
|  |  | 
|  | void UpdateCheckerTest::SetUp() { | 
|  | config_ = new TestConfigurator(base::ThreadTaskRunnerHandle::Get(), | 
|  | base::ThreadTaskRunnerHandle::Get()); | 
|  | pref_.reset(new TestingPrefServiceSimple()); | 
|  | PersistedData::RegisterPrefs(pref_->registry()); | 
|  | metadata_.reset(new PersistedData(pref_.get())); | 
|  | interceptor_factory_.reset( | 
|  | new InterceptorFactory(base::ThreadTaskRunnerHandle::Get())); | 
|  | post_interceptor_ = interceptor_factory_->CreateInterceptor(); | 
|  | EXPECT_TRUE(post_interceptor_); | 
|  |  | 
|  | update_checker_.reset(); | 
|  |  | 
|  | error_ = 0; | 
|  | results_ = UpdateResponse::Results(); | 
|  | } | 
|  |  | 
|  | void UpdateCheckerTest::TearDown() { | 
|  | update_checker_.reset(); | 
|  |  | 
|  | post_interceptor_ = NULL; | 
|  | interceptor_factory_.reset(); | 
|  |  | 
|  | config_ = nullptr; | 
|  |  | 
|  | // The PostInterceptor requires the message loop to run to destruct correctly. | 
|  | // TODO(sorin): This is fragile and should be fixed. | 
|  | RunThreadsUntilIdle(); | 
|  | } | 
|  |  | 
|  | void UpdateCheckerTest::RunThreads() { | 
|  | base::RunLoop runloop; | 
|  | quit_closure_ = runloop.QuitClosure(); | 
|  | runloop.Run(); | 
|  |  | 
|  | // Since some tests need to drain currently enqueued tasks such as network | 
|  | // intercepts on the IO thread, run the threads until they are | 
|  | // idle. The component updater service won't loop again until the loop count | 
|  | // is set and the service is started. | 
|  | RunThreadsUntilIdle(); | 
|  | } | 
|  |  | 
|  | void UpdateCheckerTest::RunThreadsUntilIdle() { | 
|  | base::RunLoop().RunUntilIdle(); | 
|  | } | 
|  |  | 
|  | void UpdateCheckerTest::Quit() { | 
|  | if (!quit_closure_.is_null()) | 
|  | quit_closure_.Run(); | 
|  | } | 
|  |  | 
|  | void UpdateCheckerTest::UpdateCheckComplete( | 
|  | int error, | 
|  | const UpdateResponse::Results& results, | 
|  | int retry_after_sec) { | 
|  | error_ = error; | 
|  | results_ = results; | 
|  | Quit(); | 
|  | } | 
|  |  | 
|  | CrxUpdateItem UpdateCheckerTest::BuildCrxUpdateItem() { | 
|  | CrxComponent crx_component; | 
|  | crx_component.name = "test_jebg"; | 
|  | crx_component.pk_hash.assign(jebg_hash, jebg_hash + arraysize(jebg_hash)); | 
|  | crx_component.installer = NULL; | 
|  | crx_component.version = base::Version("0.9"); | 
|  | crx_component.fingerprint = "fp1"; | 
|  |  | 
|  | CrxUpdateItem crx_update_item; | 
|  | crx_update_item.state = CrxUpdateItem::State::kNew; | 
|  | crx_update_item.id = "jebgalgnebhfojomionfpkfelancnnkf"; | 
|  | crx_update_item.component = crx_component; | 
|  |  | 
|  | return crx_update_item; | 
|  | } | 
|  |  | 
|  | TEST_F(UpdateCheckerTest, UpdateCheckSuccess) { | 
|  | EXPECT_TRUE(post_interceptor_->ExpectRequest( | 
|  | new PartialMatch("updatecheck"), test_file("updatecheck_reply_1.xml"))); | 
|  |  | 
|  | update_checker_ = UpdateChecker::Create(config_, metadata_.get()); | 
|  |  | 
|  | CrxUpdateItem item(BuildCrxUpdateItem()); | 
|  | item.component.installer_attributes["ap"] = "some_ap"; | 
|  | std::vector<CrxUpdateItem*> items_to_check; | 
|  | items_to_check.push_back(&item); | 
|  |  | 
|  | update_checker_->CheckForUpdates( | 
|  | items_to_check, "extra=\"params\"", | 
|  | base::Bind(&UpdateCheckerTest::UpdateCheckComplete, | 
|  | base::Unretained(this))); | 
|  |  | 
|  | RunThreads(); | 
|  |  | 
|  | EXPECT_EQ(1, post_interceptor_->GetHitCount()) | 
|  | << post_interceptor_->GetRequestsAsString(); | 
|  | ASSERT_EQ(1, post_interceptor_->GetCount()) | 
|  | << post_interceptor_->GetRequestsAsString(); | 
|  |  | 
|  | // Sanity check the request. | 
|  | EXPECT_NE(string::npos, post_interceptor_->GetRequests()[0].find( | 
|  | "request protocol=\"3.0\" extra=\"params\"")); | 
|  | // The request must not contain any "dlpref" in the default case. | 
|  | EXPECT_EQ(string::npos, | 
|  | post_interceptor_->GetRequests()[0].find(" dlpref=\"")); | 
|  | EXPECT_NE( | 
|  | string::npos, | 
|  | post_interceptor_->GetRequests()[0].find( | 
|  | "<app appid=\"jebgalgnebhfojomionfpkfelancnnkf\" version=\"0.9\" " | 
|  | "brand=\"TEST\" ap=\"some_ap\"><updatecheck /><ping rd=\"-2\" ")); | 
|  | EXPECT_NE(string::npos, | 
|  | post_interceptor_->GetRequests()[0].find( | 
|  | "<packages><package fp=\"fp1\"/></packages></app>")); | 
|  |  | 
|  | EXPECT_NE(string::npos, | 
|  | post_interceptor_->GetRequests()[0].find("<hw physmemory=")); | 
|  |  | 
|  | // Sanity check the arguments of the callback after parsing. | 
|  | EXPECT_EQ(0, error_); | 
|  | EXPECT_EQ(1ul, results_.list.size()); | 
|  | EXPECT_STREQ("jebgalgnebhfojomionfpkfelancnnkf", | 
|  | results_.list[0].extension_id.c_str()); | 
|  | EXPECT_STREQ("1.0", results_.list[0].manifest.version.c_str()); | 
|  | } | 
|  |  | 
|  | // Tests that an invalid "ap" is not serialized. | 
|  | TEST_F(UpdateCheckerTest, UpdateCheckInvalidAp) { | 
|  | EXPECT_TRUE(post_interceptor_->ExpectRequest( | 
|  | new PartialMatch("updatecheck"), test_file("updatecheck_reply_1.xml"))); | 
|  |  | 
|  | update_checker_ = UpdateChecker::Create(config_, metadata_.get()); | 
|  |  | 
|  | CrxUpdateItem item(BuildCrxUpdateItem()); | 
|  | // Make "ap" too long. | 
|  | item.component.installer_attributes["ap"] = std::string(257, 'a'); | 
|  | std::vector<CrxUpdateItem*> items_to_check; | 
|  | items_to_check.push_back(&item); | 
|  |  | 
|  | update_checker_->CheckForUpdates( | 
|  | items_to_check, "", base::Bind(&UpdateCheckerTest::UpdateCheckComplete, | 
|  | base::Unretained(this))); | 
|  |  | 
|  | RunThreads(); | 
|  |  | 
|  | EXPECT_NE( | 
|  | string::npos, | 
|  | post_interceptor_->GetRequests()[0].find( | 
|  | "app appid=\"jebgalgnebhfojomionfpkfelancnnkf\" version=\"0.9\" " | 
|  | "brand=\"TEST\"><updatecheck /><ping rd=\"-2\" ")); | 
|  | EXPECT_NE(string::npos, | 
|  | post_interceptor_->GetRequests()[0].find( | 
|  | "<packages><package fp=\"fp1\"/></packages></app>")); | 
|  | } | 
|  |  | 
|  | TEST_F(UpdateCheckerTest, UpdateCheckSuccessNoBrand) { | 
|  | EXPECT_TRUE(post_interceptor_->ExpectRequest( | 
|  | new PartialMatch("updatecheck"), test_file("updatecheck_reply_1.xml"))); | 
|  |  | 
|  | config_->SetBrand("TOOLONG");   // Sets an invalid brand code. | 
|  | update_checker_ = UpdateChecker::Create(config_, metadata_.get()); | 
|  |  | 
|  | CrxUpdateItem item(BuildCrxUpdateItem()); | 
|  | std::vector<CrxUpdateItem*> items_to_check; | 
|  | items_to_check.push_back(&item); | 
|  |  | 
|  | update_checker_->CheckForUpdates( | 
|  | items_to_check, "", base::Bind(&UpdateCheckerTest::UpdateCheckComplete, | 
|  | base::Unretained(this))); | 
|  |  | 
|  | RunThreads(); | 
|  |  | 
|  | EXPECT_NE( | 
|  | string::npos, | 
|  | post_interceptor_->GetRequests()[0].find( | 
|  | "<app appid=\"jebgalgnebhfojomionfpkfelancnnkf\" version=\"0.9\">" | 
|  | "<updatecheck /><ping rd=\"-2\" ")); | 
|  | EXPECT_NE(string::npos, | 
|  | post_interceptor_->GetRequests()[0].find( | 
|  | "<packages><package fp=\"fp1\"/></packages></app>")); | 
|  | } | 
|  |  | 
|  | // Simulates a 403 server response error. | 
|  | TEST_F(UpdateCheckerTest, UpdateCheckError) { | 
|  | EXPECT_TRUE( | 
|  | post_interceptor_->ExpectRequest(new PartialMatch("updatecheck"), 403)); | 
|  |  | 
|  | update_checker_ = UpdateChecker::Create(config_, metadata_.get()); | 
|  |  | 
|  | CrxUpdateItem item(BuildCrxUpdateItem()); | 
|  | std::vector<CrxUpdateItem*> items_to_check; | 
|  | items_to_check.push_back(&item); | 
|  |  | 
|  | update_checker_->CheckForUpdates( | 
|  | items_to_check, "", base::Bind(&UpdateCheckerTest::UpdateCheckComplete, | 
|  | base::Unretained(this))); | 
|  | RunThreads(); | 
|  |  | 
|  | EXPECT_EQ(1, post_interceptor_->GetHitCount()) | 
|  | << post_interceptor_->GetRequestsAsString(); | 
|  | EXPECT_EQ(1, post_interceptor_->GetCount()) | 
|  | << post_interceptor_->GetRequestsAsString(); | 
|  |  | 
|  | EXPECT_EQ(403, error_); | 
|  | EXPECT_EQ(0ul, results_.list.size()); | 
|  | } | 
|  |  | 
|  | TEST_F(UpdateCheckerTest, UpdateCheckDownloadPreference) { | 
|  | EXPECT_TRUE(post_interceptor_->ExpectRequest( | 
|  | new PartialMatch("updatecheck"), test_file("updatecheck_reply_1.xml"))); | 
|  |  | 
|  | config_->SetDownloadPreference(string("cacheable")); | 
|  |  | 
|  | update_checker_ = UpdateChecker::Create(config_, metadata_.get()); | 
|  |  | 
|  | CrxUpdateItem item(BuildCrxUpdateItem()); | 
|  | std::vector<CrxUpdateItem*> items_to_check; | 
|  | items_to_check.push_back(&item); | 
|  |  | 
|  | update_checker_->CheckForUpdates( | 
|  | items_to_check, "extra=\"params\"", | 
|  | base::Bind(&UpdateCheckerTest::UpdateCheckComplete, | 
|  | base::Unretained(this))); | 
|  |  | 
|  | RunThreads(); | 
|  |  | 
|  | // The request must contain dlpref="cacheable". | 
|  | EXPECT_NE(string::npos, | 
|  | post_interceptor_->GetRequests()[0].find(" dlpref=\"cacheable\"")); | 
|  | } | 
|  |  | 
|  | // This test is checking that an update check signed with CUP fails, since there | 
|  | // is currently no entity that can respond with a valid signed response. | 
|  | // A proper CUP test requires network mocks, which are not available now. | 
|  | TEST_F(UpdateCheckerTest, UpdateCheckCupError) { | 
|  | EXPECT_TRUE(post_interceptor_->ExpectRequest( | 
|  | new PartialMatch("updatecheck"), test_file("updatecheck_reply_1.xml"))); | 
|  |  | 
|  | config_->SetUseCupSigning(true); | 
|  | update_checker_ = UpdateChecker::Create(config_, metadata_.get()); | 
|  |  | 
|  | CrxUpdateItem item(BuildCrxUpdateItem()); | 
|  | std::vector<CrxUpdateItem*> items_to_check; | 
|  | items_to_check.push_back(&item); | 
|  |  | 
|  | update_checker_->CheckForUpdates( | 
|  | items_to_check, "", base::Bind(&UpdateCheckerTest::UpdateCheckComplete, | 
|  | base::Unretained(this))); | 
|  |  | 
|  | RunThreads(); | 
|  |  | 
|  | EXPECT_EQ(1, post_interceptor_->GetHitCount()) | 
|  | << post_interceptor_->GetRequestsAsString(); | 
|  | ASSERT_EQ(1, post_interceptor_->GetCount()) | 
|  | << post_interceptor_->GetRequestsAsString(); | 
|  |  | 
|  | // Sanity check the request. | 
|  | EXPECT_NE( | 
|  | string::npos, | 
|  | post_interceptor_->GetRequests()[0].find( | 
|  | "<app appid=\"jebgalgnebhfojomionfpkfelancnnkf\" version=\"0.9\" " | 
|  | "brand=\"TEST\"><updatecheck /><ping rd=\"-2\" ")); | 
|  | EXPECT_NE(string::npos, | 
|  | post_interceptor_->GetRequests()[0].find( | 
|  | "<packages><package fp=\"fp1\"/></packages></app>")); | 
|  |  | 
|  | // Expect an error since the response is not trusted. | 
|  | EXPECT_EQ(-10000, error_); | 
|  | EXPECT_EQ(0ul, results_.list.size()); | 
|  | } | 
|  |  | 
|  | // Tests that the UpdateCheckers will not make an update check for a | 
|  | // component that requires encryption when the update check URL is unsecure. | 
|  | TEST_F(UpdateCheckerTest, UpdateCheckRequiresEncryptionError) { | 
|  | config_->SetUpdateCheckUrl(GURL("http:\\foo\bar")); | 
|  |  | 
|  | update_checker_ = UpdateChecker::Create(config_, metadata_.get()); | 
|  |  | 
|  | CrxUpdateItem item(BuildCrxUpdateItem()); | 
|  | item.component.requires_network_encryption = true; | 
|  | std::vector<CrxUpdateItem*> items_to_check; | 
|  | items_to_check.push_back(&item); | 
|  |  | 
|  | update_checker_->CheckForUpdates( | 
|  | items_to_check, "", base::Bind(&UpdateCheckerTest::UpdateCheckComplete, | 
|  | base::Unretained(this))); | 
|  | RunThreads(); | 
|  |  | 
|  | EXPECT_EQ(-1, error_); | 
|  | EXPECT_EQ(0u, results_.list.size()); | 
|  | } | 
|  |  | 
|  | // Tests that the PersistedData will get correctly update and reserialize | 
|  | // the elapsed_days value. | 
|  | TEST_F(UpdateCheckerTest, UpdateCheckDateLastRollCall) { | 
|  | EXPECT_TRUE(post_interceptor_->ExpectRequest( | 
|  | new PartialMatch("updatecheck"), test_file("updatecheck_reply_4.xml"))); | 
|  | EXPECT_TRUE(post_interceptor_->ExpectRequest( | 
|  | new PartialMatch("updatecheck"), test_file("updatecheck_reply_4.xml"))); | 
|  |  | 
|  | update_checker_ = UpdateChecker::Create(config_, metadata_.get()); | 
|  |  | 
|  | CrxUpdateItem item(BuildCrxUpdateItem()); | 
|  | std::vector<CrxUpdateItem*> items_to_check; | 
|  | items_to_check.push_back(&item); | 
|  |  | 
|  | // Do two update-checks. | 
|  | update_checker_->CheckForUpdates( | 
|  | items_to_check, "extra=\"params\"", | 
|  | base::Bind(&UpdateCheckerTest::UpdateCheckComplete, | 
|  | base::Unretained(this))); | 
|  | RunThreads(); | 
|  | update_checker_ = UpdateChecker::Create(config_, metadata_.get()); | 
|  | update_checker_->CheckForUpdates( | 
|  | items_to_check, "extra=\"params\"", | 
|  | base::Bind(&UpdateCheckerTest::UpdateCheckComplete, | 
|  | base::Unretained(this))); | 
|  | RunThreads(); | 
|  |  | 
|  | EXPECT_EQ(2, post_interceptor_->GetHitCount()) | 
|  | << post_interceptor_->GetRequestsAsString(); | 
|  | ASSERT_EQ(2, post_interceptor_->GetCount()) | 
|  | << post_interceptor_->GetRequestsAsString(); | 
|  | EXPECT_NE(string::npos, post_interceptor_->GetRequests()[0].find( | 
|  | "<ping rd=\"-2\" ping_freshness=")); | 
|  | EXPECT_NE(string::npos, post_interceptor_->GetRequests()[1].find( | 
|  | "<ping rd=\"3383\" ping_freshness=")); | 
|  | } | 
|  |  | 
|  | }  // namespace update_client |