| // Copyright 2010 Google Inc. |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| // ======================================================================== |
| |
| #include <windows.h> |
| #include <atlpath.h> |
| #include <atlsecurity.h> |
| #include <atlstr.h> |
| #include <vector> |
| #include "omaha/base/error.h" |
| #include "omaha/base/omaha_version.h" |
| #include "omaha/base/reg_key.h" |
| #include "omaha/base/string.h" |
| #include "omaha/base/utils.h" |
| #include "omaha/base/vistautil.h" |
| #include "omaha/client/help_url_builder.h" |
| #include "omaha/common/config_manager.h" |
| #include "omaha/common/goopdate_utils.h" |
| #include "omaha/net/http_client.h" |
| #include "omaha/testing/unit_test.h" |
| |
| namespace omaha { |
| |
| namespace { |
| |
| #define APP_GUID _T("{B7BAF788-9D64-49c3-AFDC-B336AB12F332}") |
| #define APP_GUID2 _T("{6D2DF75B-11F0-41CA-9874-79DE4568527C}") |
| const TCHAR* const kAppGuid = APP_GUID; |
| const TCHAR* const kAppGuid2 = APP_GUID2; |
| |
| const TCHAR kStringAlmostTooLongForUrl[] = |
| _T("000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000"); // NOLINT |
| |
| // Verifies that one of the expected OS strings was found in the url. |
| // Returns the position along with the length of the OS string. |
| int VerifyOSInUrl(const CString& url, int* length) { |
| ASSERT1(length); |
| *length = 0; |
| |
| // The strings are in descending version order to avoid breaking on a |
| // substring of the version we are looking for. |
| // TODO(omaha): This is a maintenance problem. Consider eliminating the |
| // "&sp=" at the very least. |
| const TCHAR* kExpectedOsStrings[] = {_T("6.1&sp=Service%20Pack%201"), |
| _T("6.1&sp="), |
| _T("6.0&sp=Service%20Pack%201"), |
| _T("6.0&sp="), |
| _T("5.2&sp=Service%20Pack%202"), |
| _T("5.2&sp=Service%20Pack%201"), |
| _T("5.1&sp=Service%20Pack%203"), |
| _T("5.1&sp=Service%20Pack%202"), |
| }; |
| |
| bool found = false; |
| int this_pos = 0; |
| |
| for (int i = 0; i < arraysize(kExpectedOsStrings); ++i) { |
| this_pos = url.Find(kExpectedOsStrings[i]); |
| if (-1 != this_pos) { |
| found = true; |
| *length = _tcslen(kExpectedOsStrings[i]); |
| break; |
| } |
| } |
| |
| EXPECT_TRUE(found); |
| return this_pos; |
| } |
| |
| } // namespace |
| |
| class HelpUrlBuilderTest : public testing::Test { |
| protected: |
| HRESULT BuildHttpGetString( |
| const CString& base_url, |
| const std::vector<HelpUrlBuilder::AppResult>& app_results, |
| const CString& goopdate_version, |
| bool is_machine, |
| const CString& language, |
| const GUID& iid, |
| const CString& brand_code, |
| const CString& source_id, |
| CString* get_request) const { |
| HelpUrlBuilder url_builder(is_machine, language, iid, brand_code); |
| return url_builder.BuildHttpGetString(base_url, |
| app_results, |
| goopdate_version, |
| source_id, |
| get_request); |
| } |
| |
| HelpUrlBuilderTest() : hive_override_key_name_(kRegistryHiveOverrideRoot), |
| module_version_(GetVersion()) { |
| } |
| |
| virtual void SetUp() { |
| RegKey::DeleteKey(hive_override_key_name_, true); |
| OverrideRegistryHives(hive_override_key_name_); |
| InitializeVersion(kFakeVersion); |
| } |
| |
| virtual void TearDown() { |
| InitializeVersion(module_version_); |
| RestoreRegistryHives(); |
| ASSERT_SUCCEEDED(RegKey::DeleteKey(hive_override_key_name_, true)); |
| } |
| |
| CString hive_override_key_name_; |
| const ULONGLONG module_version_; |
| |
| static const ULONGLONG kFakeVersion = 0x0005000600070008; |
| }; |
| |
| TEST_F(HelpUrlBuilderTest, BuildHttpGetString_MachineNoTestSource) { |
| CString expected_str_before_os( |
| _T("http://www.google.com/hello.py?code=123&hl=en&") |
| _T("app.0=%7BB7BAF788-9D64-49c3-AFDC-B336AB12F332%7D&") |
| _T("ec.0=0xa&ex.0=22&") |
| _T("guver=1.0.51.0&m=1&os=")); |
| CString expected_str_after_os( |
| _T("&iid=%7B0F973A20-C484-462B-952C-5D9A459E3326%7D") // Upper case 'B'. |
| _T("&brand=GoOG&source=click")); |
| bool expected_test_source = false; |
| |
| #if defined(DEBUG) || !OFFICIAL_BUILD |
| // TestSource is always set for these builds. It may be set for opt official |
| // builds but this is not guaranteed. |
| expected_str_after_os.Append(_T("&testsource=")); |
| expected_test_source = true; |
| #endif |
| |
| CString url_req; |
| std::vector<HelpUrlBuilder::AppResult> app_results; |
| app_results.push_back(HelpUrlBuilder::AppResult(kAppGuid, 10, 22)); |
| EXPECT_SUCCEEDED(BuildHttpGetString( |
| _T("http://www.google.com/hello.py?code=123&"), |
| app_results, |
| _T("1.0.51.0"), |
| true, |
| _T("en"), |
| StringToGuid(_T("{0F973A20-C484-462b-952C-5D9A459E3326}")), |
| _T("GoOG"), |
| _T("click"), |
| &url_req)); |
| |
| EXPECT_EQ(-1, url_req.FindOneOf(_T("{}"))); |
| |
| EXPECT_LE(expected_str_before_os.GetLength(), url_req.GetLength()); |
| EXPECT_EQ(0, url_req.Find(expected_str_before_os)) << |
| _T("Expected: ") << expected_str_before_os.GetString() << std::endl << |
| _T("At beginning of: ") << url_req.GetString(); |
| int os_fragment_len = 0; |
| EXPECT_EQ(expected_str_before_os.GetLength(), |
| VerifyOSInUrl(url_req, &os_fragment_len)) << |
| _T("Expected OS string not found in: ") << url_req.GetString(); |
| |
| EXPECT_EQ(expected_str_before_os.GetLength() + os_fragment_len, |
| url_req.Find(expected_str_after_os)) << |
| _T("Expected: ") << expected_str_after_os.GetString() << std::endl << |
| _T("At end of: ") << url_req.GetString(); |
| |
| if (expected_test_source) { |
| CString expected_testsource_str = |
| ConfigManager::Instance()->GetTestSource(); |
| int expected_testsource_start = expected_str_before_os.GetLength() + |
| os_fragment_len + |
| expected_str_after_os.GetLength(); |
| EXPECT_EQ(expected_testsource_start, url_req.Find(expected_testsource_str)); |
| EXPECT_EQ(expected_testsource_start + expected_testsource_str.GetLength(), |
| url_req.GetLength()); |
| } else { |
| EXPECT_EQ(expected_str_before_os.GetLength() + |
| os_fragment_len + |
| expected_str_after_os.GetLength(), |
| url_req.GetLength()); |
| |
| EXPECT_EQ(-1, url_req.Find(_T("testsource"))); |
| } |
| } |
| |
| TEST_F(HelpUrlBuilderTest, BuildHttpGetString_UserWithTestSource) { |
| ASSERT_SUCCEEDED(RegKey::SetValue(MACHINE_REG_UPDATE_DEV, |
| kRegValueTestSource, |
| _T("dev"))); |
| |
| const CString expected_str_before_os( |
| _T("http://www.google.com/hello.py?hl=de&") |
| _T("app.0=%7BB7BAF788-9D64-49c3-AFDC-B336AB12F332%7D&") |
| _T("ec.0=0xffffffff&ex.0=99&") |
| _T("guver=foo%20bar&m=0&os=")); |
| const CString expected_str_after_os( |
| _T("&iid=%7B0F973A20-C484-462B-952C-5D9A459E3326%7D") // Upper case 'B'. |
| _T("&brand=GGLE&source=clack") |
| _T("&testsource=")); |
| |
| CString url_req; |
| std::vector<HelpUrlBuilder::AppResult> app_results; |
| app_results.push_back(HelpUrlBuilder::AppResult(kAppGuid, 0xffffffff, 99)); |
| EXPECT_SUCCEEDED(BuildHttpGetString( |
| _T("http://www.google.com/hello.py?"), |
| app_results, |
| _T("foo bar"), |
| false, |
| _T("de"), |
| StringToGuid(_T("{0F973A20-C484-462b-952C-5D9A459E3326}")), |
| _T("GGLE"), |
| _T("clack"), |
| &url_req)); |
| EXPECT_LE(expected_str_before_os.GetLength(), url_req.GetLength()); |
| EXPECT_EQ(0, url_req.Find(expected_str_before_os)); |
| |
| int os_fragment_len = 0; |
| EXPECT_EQ(expected_str_before_os.GetLength(), |
| VerifyOSInUrl(url_req, &os_fragment_len)) << |
| _T("Expected: ") << expected_str_before_os.GetString() << std::endl << |
| _T("At beginning of: ") << url_req.GetString(); |
| |
| EXPECT_EQ(expected_str_before_os.GetLength() + os_fragment_len, |
| url_req.Find(expected_str_after_os)) << |
| _T("Expected OS string not found in: ") << url_req.GetString(); |
| |
| const CString expected_testsource_str = _T("dev"); |
| |
| int expected_testsource_start = expected_str_before_os.GetLength() + |
| os_fragment_len + |
| expected_str_after_os.GetLength(); |
| EXPECT_EQ(expected_testsource_start, url_req.Find(expected_testsource_str)); |
| EXPECT_EQ(expected_testsource_start + expected_testsource_str.GetLength(), |
| url_req.GetLength()); |
| } |
| |
| // IID and brand code are emtpy if not present. |
| TEST_F(HelpUrlBuilderTest, BuildHttpGetString_NoIidOrBrandCode) { |
| const CString expected_str_before_os( |
| _T("http://www.google.com/hello.py?hl=en&") |
| _T("app.0=%7BB7BAF788-9D64-49c3-AFDC-B336AB12F332%7D&") |
| _T("ec.0=0xffffffff&ex.0=99&") |
| _T("guver=foo%20bar&m=1&os=")); |
| const CString expected_str_after_os(_T("&iid=&brand=&source=cluck")); |
| |
| CString url_req; |
| std::vector<HelpUrlBuilder::AppResult> app_results; |
| app_results.push_back(HelpUrlBuilder::AppResult( |
| _T("{B7BAF788-9D64-49c3-AFDC-B336AB12F332}"), 0xffffffff, 99)); |
| EXPECT_SUCCEEDED(BuildHttpGetString( |
| _T("http://www.google.com/hello.py?"), |
| app_results, |
| _T("foo bar"), |
| true, |
| _T("en"), |
| GUID_NULL, |
| _T(""), |
| _T("cluck"), |
| &url_req)); |
| |
| EXPECT_EQ(0, url_req.Find(expected_str_before_os)) << |
| _T("Expected: ") << expected_str_before_os.GetString() << std::endl << |
| _T("At beginning of: ") << url_req.GetString(); |
| |
| EXPECT_LT(0, url_req.Find(expected_str_after_os)); |
| |
| CString expected_test_src; |
| #if defined(DEBUG) || !OFFICIAL_BUILD |
| expected_test_src = _T("&testsource=auto"); |
| #endif |
| const CString expected_iid_str(_T("&iid=&brand=&source=cluck")); |
| EXPECT_EQ(url_req.GetLength() - |
| expected_iid_str.GetLength() - |
| expected_test_src.GetLength(), |
| url_req.Find(expected_iid_str)); |
| } |
| |
| TEST_F(HelpUrlBuilderTest, BuildHttpGetString_UrlTooLong) { |
| EXPECT_LT(INTERNET_MAX_URL_LENGTH, arraysize(kStringAlmostTooLongForUrl) + 5); |
| |
| ExpectAsserts expect_asserts; // BuildHttpGetString asserts on URL length. |
| CString url_req; |
| std::vector<HelpUrlBuilder::AppResult> app_results; |
| app_results.push_back(HelpUrlBuilder::AppResult( |
| _T("{B7BAF788-9D64-49c3-AFDC-B336AB12F332}"), 0xffffffff, 99)); |
| EXPECT_EQ(E_FAIL, BuildHttpGetString( |
| _T("http://www.google.com/hello.py?"), |
| app_results, |
| _T("foo bar"), |
| true, |
| _T("en"), |
| GUID_NULL, |
| _T(""), |
| kStringAlmostTooLongForUrl, |
| &url_req)); |
| } |
| |
| TEST_F(HelpUrlBuilderTest, BuildHttpGetString_MultipleApps) { |
| CString expected_str_before_os( |
| _T("http://www.google.com/hello.py?code=123&hl=en&") |
| _T("app.0=%7BB7BAF788-9D64-49c3-AFDC-B336AB12F332%7D&") |
| _T("ec.0=0x80000001&ex.0=1000&") |
| _T("app.1=%7B6D2DF75B-11F0-41CA-9874-79DE4568527C%7D&") |
| _T("ec.1=0x0&ex.1=0&") |
| _T("guver=1.0.51.22&m=1&os=")); |
| CString expected_str_after_os( |
| _T("&iid=%7B0F973A20-C484-462B-952C-5D9A459E3326%7D") // Upper case 'B'. |
| _T("&brand=TEST&source=click")); |
| bool expected_test_source = false; |
| |
| #if defined(DEBUG) || !OFFICIAL_BUILD |
| // TestSource is always set for these builds. It may be set for opt official |
| // builds but this is not guaranteed. |
| expected_str_after_os.Append(_T("&testsource=")); |
| expected_test_source = true; |
| #endif |
| |
| CString url_req; |
| std::vector<HelpUrlBuilder::AppResult> app_results; |
| app_results.push_back(HelpUrlBuilder::AppResult(kAppGuid, 0x80000001, 1000)); |
| app_results.push_back(HelpUrlBuilder::AppResult(kAppGuid2, 0, 0)); |
| EXPECT_SUCCEEDED(BuildHttpGetString( |
| _T("http://www.google.com/hello.py?code=123&"), |
| app_results, |
| _T("1.0.51.22"), |
| true, |
| _T("en"), |
| StringToGuid(_T("{0F973A20-C484-462b-952C-5D9A459E3326}")), |
| _T("TEST"), |
| _T("click"), |
| &url_req)); |
| |
| EXPECT_EQ(-1, url_req.FindOneOf(_T("{}"))); |
| |
| EXPECT_LE(expected_str_before_os.GetLength(), url_req.GetLength()); |
| EXPECT_EQ(0, url_req.Find(expected_str_before_os)) << |
| _T("Expected: ") << expected_str_before_os.GetString() << std::endl << |
| _T("At beginning of: ") << url_req.GetString(); |
| int os_fragment_len = 0; |
| EXPECT_EQ(expected_str_before_os.GetLength(), |
| VerifyOSInUrl(url_req, &os_fragment_len)) << |
| _T("Expected OS string not found in: ") << url_req.GetString(); |
| |
| EXPECT_EQ(expected_str_before_os.GetLength() + os_fragment_len, |
| url_req.Find(expected_str_after_os)) << |
| _T("Expected: ") << expected_str_after_os.GetString() << std::endl << |
| _T("At end of: ") << url_req.GetString(); |
| |
| if (expected_test_source) { |
| CString expected_testsource_str = |
| ConfigManager::Instance()->GetTestSource(); |
| int expected_testsource_start = expected_str_before_os.GetLength() + |
| os_fragment_len + |
| expected_str_after_os.GetLength(); |
| EXPECT_EQ(expected_testsource_start, url_req.Find(expected_testsource_str)); |
| EXPECT_EQ(expected_testsource_start + expected_testsource_str.GetLength(), |
| url_req.GetLength()); |
| } else { |
| EXPECT_EQ(expected_str_before_os.GetLength() + |
| os_fragment_len + |
| expected_str_after_os.GetLength(), |
| url_req.GetLength()); |
| |
| EXPECT_EQ(-1, url_req.Find(_T("testsource"))); |
| } |
| } |
| |
| // Machine ID must be set or it will be randomly generated in some cases. |
| TEST_F(HelpUrlBuilderTest, BuildGetHelpUrl_User) { |
| // The URL has a begin, middle which is OS-specific and not checked, and end. |
| const CString kExpetedUrlBegin = |
| _T("http://www.google.com/support/installer/?hl=en-GB&") |
| _T("app.0=%7Btest-user-app-id%7D&ec.0=0x80004005&ex.0=-2147418113&") |
| _T("guver=5.6.7.8&m=0&os="); |
| const CString kExpectedUrlAfterOs = _T("iid=&brand=&source=gethelp") |
| #if defined(DEBUG) || !OFFICIAL_BUILD |
| // TestSource is always set for these builds. |
| _T("&testsource="); |
| #else |
| // TestSource never set for other builds because registry is overridden. |
| ; // NOLINT |
| #endif |
| |
| CString url; |
| HelpUrlBuilder url_builder(false, _T("en-GB"), GUID_NULL, _T("")); |
| std::vector<HelpUrlBuilder::AppResult> app_results; |
| app_results.push_back( |
| HelpUrlBuilder::AppResult(_T("{test-user-app-id}"), |
| E_FAIL, |
| static_cast<DWORD>(E_UNEXPECTED))); |
| EXPECT_SUCCEEDED(url_builder.BuildUrl(app_results, &url)); |
| |
| EXPECT_STREQ(kExpetedUrlBegin, url.Left(kExpetedUrlBegin.GetLength())); |
| EXPECT_NE(-1, url.Find(kExpectedUrlAfterOs)) |
| << kExpectedUrlAfterOs.GetString() << std::endl |
| << _T(" not found in ") << std::endl << url.GetString(); |
| } |
| |
| TEST_F(HelpUrlBuilderTest, BuildGetHelpUrl_Machine) { |
| // The URL has a begin, middle which is OS-specific and not checked, and end. |
| const CString kExpetedUrlBegin = |
| _T("http://www.google.com/support/installer/?hl=en-GB&") |
| _T("app.0=%7Btest-machine-app-id%7D&ec.0=0x80004004&ex.0=99&") |
| _T("guver=5.6.7.8&m=1&os="); |
| const CString kExpectedUrlAfterOs = |
| _T("iid=%7B326ADA1D-06AA-4C16-8101-5FC3FEBC852A%7D&") // Upper case 'C'. |
| _T("brand=GOOG&source=gethelp") |
| #if defined(DEBUG) || !OFFICIAL_BUILD |
| // TestSource is always set for these builds. |
| _T("&testsource="); |
| #else |
| // TestSource never set for other builds because registry is overridden. |
| ; // NOLINT |
| #endif |
| |
| const GUID kIid = StringToGuid(_T("{326ADA1D-06AA-4c16-8101-5FC3FEBC852A}")); |
| CString url; |
| HelpUrlBuilder url_builder(true, _T("en-GB"), kIid, _T("GOOG")); |
| std::vector<HelpUrlBuilder::AppResult> app_results; |
| app_results.push_back(HelpUrlBuilder::AppResult(_T("{test-machine-app-id}"), |
| E_ABORT, |
| 99)); |
| EXPECT_SUCCEEDED(url_builder.BuildUrl(app_results, &url)); |
| |
| EXPECT_STREQ(kExpetedUrlBegin, url.Left(kExpetedUrlBegin.GetLength())); |
| EXPECT_NE(-1, url.Find(kExpectedUrlAfterOs)) |
| << kExpectedUrlAfterOs.GetString() << std::endl |
| << _T(" not found in ") << std::endl << url.GetString(); |
| } |
| |
| // Makes BuildHttpGetString fail by making the URL too long. |
| // The call succeeds, but the url is empty. |
| TEST_F(HelpUrlBuilderTest, BuildGetHelpUrl_BuildFails) { |
| EXPECT_LT(INTERNET_MAX_URL_LENGTH, arraysize(kStringAlmostTooLongForUrl) + 5); |
| |
| ExpectAsserts expect_asserts; // BuildHttpGetString asserts on URL length. |
| CString url; |
| HelpUrlBuilder url_builder(false, _T("en-GB"), GUID_NULL, _T("")); |
| std::vector<HelpUrlBuilder::AppResult> app_results; |
| app_results.push_back( |
| HelpUrlBuilder::AppResult(kStringAlmostTooLongForUrl, |
| E_FAIL, |
| static_cast<DWORD>(E_UNEXPECTED))); |
| EXPECT_EQ(E_FAIL, url_builder.BuildUrl(app_results, &url)); |
| EXPECT_TRUE(url.IsEmpty()); |
| } |
| |
| } // namespace omaha |