| OLD | NEW | 
|---|
| 1 /* | 1 /* | 
| 2  * This file is part of Adblock Plus <https://adblockplus.org/>, | 2  * This file is part of Adblock Plus <https://adblockplus.org/>, | 
| 3  * Copyright (C) 2006-2017 eyeo GmbH | 3  * Copyright (C) 2006-2017 eyeo GmbH | 
| 4  * | 4  * | 
| 5  * Adblock Plus is free software: you can redistribute it and/or modify | 5  * Adblock Plus is free software: you can redistribute it and/or modify | 
| 6  * it under the terms of the GNU General Public License version 3 as | 6  * it under the terms of the GNU General Public License version 3 as | 
| 7  * published by the Free Software Foundation. | 7  * published by the Free Software Foundation. | 
| 8  * | 8  * | 
| 9  * Adblock Plus is distributed in the hope that it will be useful, | 9  * Adblock Plus is distributed in the hope that it will be useful, | 
| 10  * but WITHOUT ANY WARRANTY; without even the implied warranty of | 10  * but WITHOUT ANY WARRANTY; without even the implied warranty of | 
| (...skipping 26 matching lines...) Expand all  Loading... | 
| 37       } | 37       } | 
| 38       { | 38       { | 
| 39         std::lock_guard<std::mutex> lock(requestHeaderNamesMutex); | 39         std::lock_guard<std::mutex> lock(requestHeaderNamesMutex); | 
| 40         // we currently ignore the result. We should check it actually gets inse
     rted. | 40         // we currently ignore the result. We should check it actually gets inse
     rted. | 
| 41         requestHeaderNames.insert(std::make_pair(url, std::move(headerNames))); | 41         requestHeaderNames.insert(std::make_pair(url, std::move(headerNames))); | 
| 42       } | 42       } | 
| 43 | 43 | 
| 44       AdblockPlus::Sleep(50); | 44       AdblockPlus::Sleep(50); | 
| 45 | 45 | 
| 46       AdblockPlus::ServerResponse result; | 46       AdblockPlus::ServerResponse result; | 
| 47       result.status = NS_OK; | 47       result.status = IWebRequest::NS_OK; | 
| 48       result.responseStatus = 123; | 48       result.responseStatus = 123; | 
| 49       result.responseHeaders.push_back(std::pair<std::string, std::string>("Foo"
     , "Bar")); | 49       result.responseHeaders.push_back(std::pair<std::string, std::string>("Foo"
     , "Bar")); | 
| 50       result.responseText = url + "\n"; | 50       result.responseText = url + "\n"; | 
| 51       if (!requestHeaders.empty()) | 51       if (!requestHeaders.empty()) | 
| 52       { | 52       { | 
| 53         result.responseText += requestHeaders[0].first + "\n" + requestHeaders[0
     ].second; | 53         result.responseText += requestHeaders[0].first + "\n" + requestHeaders[0
     ].second; | 
| 54       } | 54       } | 
| 55       return result; | 55       return result; | 
| 56     } | 56     } | 
| 57 | 57 | 
| (...skipping 27 matching lines...) Expand all  Loading... | 
| 85       BaseJsTest::SetUp(); | 85       BaseJsTest::SetUp(); | 
| 86       webRequest = std::make_shared<T>(); | 86       webRequest = std::make_shared<T>(); | 
| 87       jsEngine->SetWebRequest(webRequest); | 87       jsEngine->SetWebRequest(webRequest); | 
| 88       jsEngine->SetFileSystem(AdblockPlus::FileSystemPtr(new LazyFileSystem())); | 88       jsEngine->SetFileSystem(AdblockPlus::FileSystemPtr(new LazyFileSystem())); | 
| 89     } | 89     } | 
| 90 | 90 | 
| 91     std::shared_ptr<T> webRequest; | 91     std::shared_ptr<T> webRequest; | 
| 92   }; | 92   }; | 
| 93 | 93 | 
| 94   typedef WebRequestTest<MockWebRequest> MockWebRequestTest; | 94   typedef WebRequestTest<MockWebRequest> MockWebRequestTest; | 
| 95   typedef WebRequestTest<AdblockPlus::DefaultWebRequest> DefaultWebRequestTest; | 95   typedef WebRequestTest<AdblockPlus::DefaultWebRequestSync> DefaultWebRequestTe
     st; | 
| 96   typedef WebRequestTest<MockWebRequest> XMLHttpRequestTest; | 96   typedef WebRequestTest<MockWebRequest> XMLHttpRequestTest; | 
| 97 | 97 | 
| 98   // we return the url of the XHR. | 98   // we return the url of the XHR. | 
| 99   std::string ResetTestXHR(const AdblockPlus::JsEnginePtr& jsEngine, const std::
     string& defaultUrl = "") | 99   std::string ResetTestXHR(const AdblockPlus::JsEnginePtr& jsEngine, const std::
     string& defaultUrl = "") | 
| 100   { | 100   { | 
| 101     std::string url = defaultUrl; | 101     std::string url = defaultUrl; | 
| 102     // make up a unique URL if we don't have one. | 102     // make up a unique URL if we don't have one. | 
| 103     if (url == "") | 103     if (url == "") | 
| 104     { | 104     { | 
| 105       url = "https://tests.adblockplus.org/easylist.txt-"; | 105       url = "https://tests.adblockplus.org/easylist.txt-"; | 
| (...skipping 29 matching lines...) Expand all  Loading... | 
| 135   ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET('http://example.com/', nu
     ll, function(){})")); | 135   ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET('http://example.com/', nu
     ll, function(){})")); | 
| 136   ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET('http://example.com/', {}
     , null)")); | 136   ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET('http://example.com/', {}
     , null)")); | 
| 137   ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET('http://example.com/', {}
     , function(){}, 0)")); | 137   ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET('http://example.com/', {}
     , function(){}, 0)")); | 
| 138 } | 138 } | 
| 139 | 139 | 
| 140 TEST_F(MockWebRequestTest, SuccessfulRequest) | 140 TEST_F(MockWebRequestTest, SuccessfulRequest) | 
| 141 { | 141 { | 
| 142   jsEngine->Evaluate("_webRequest.GET('http://example.com/', {X: 'Y'}, function(
     result) {foo = result;} )"); | 142   jsEngine->Evaluate("_webRequest.GET('http://example.com/', {X: 'Y'}, function(
     result) {foo = result;} )"); | 
| 143   ASSERT_TRUE(jsEngine->Evaluate("this.foo").IsUndefined()); | 143   ASSERT_TRUE(jsEngine->Evaluate("this.foo").IsUndefined()); | 
| 144   AdblockPlus::Sleep(200); | 144   AdblockPlus::Sleep(200); | 
| 145   ASSERT_EQ(AdblockPlus::WebRequest::NS_OK, jsEngine->Evaluate("foo.status").AsI
     nt()); | 145   ASSERT_EQ(IWebRequest::NS_OK, jsEngine->Evaluate("foo.status").AsInt()); | 
| 146   ASSERT_EQ(123, jsEngine->Evaluate("foo.responseStatus").AsInt()); | 146   ASSERT_EQ(123, jsEngine->Evaluate("foo.responseStatus").AsInt()); | 
| 147   ASSERT_EQ("http://example.com/\nX\nY", jsEngine->Evaluate("foo.responseText").
     AsString()); | 147   ASSERT_EQ("http://example.com/\nX\nY", jsEngine->Evaluate("foo.responseText").
     AsString()); | 
| 148   ASSERT_EQ("{\"Foo\":\"Bar\"}", jsEngine->Evaluate("JSON.stringify(foo.response
     Headers)").AsString()); | 148   ASSERT_EQ("{\"Foo\":\"Bar\"}", jsEngine->Evaluate("JSON.stringify(foo.response
     Headers)").AsString()); | 
| 149 } | 149 } | 
| 150 | 150 | 
| 151 #if defined(HAVE_CURL) || defined(_WIN32) | 151 #if defined(HAVE_CURL) || defined(_WIN32) | 
| 152 TEST_F(DefaultWebRequestTest, RealWebRequest) | 152 TEST_F(DefaultWebRequestTest, RealWebRequest) | 
| 153 { | 153 { | 
| 154   // This URL should redirect to easylist-downloads.adblockplus.org and we | 154   // This URL should redirect to easylist-downloads.adblockplus.org and we | 
| 155   // should get the actual filter list back. | 155   // should get the actual filter list back. | 
| 156   jsEngine->Evaluate("_webRequest.GET('https://easylist-downloads.adblockplus.or
     g/easylist.txt', {}, function(result) {foo = result;} )"); | 156   jsEngine->Evaluate("_webRequest.GET('https://easylist-downloads.adblockplus.or
     g/easylist.txt', {}, function(result) {foo = result;} )"); | 
| 157   WaitForVariable("this.foo", jsEngine); | 157   WaitForVariable("this.foo", jsEngine); | 
| 158   ASSERT_EQ("text/plain", jsEngine->Evaluate("foo.responseHeaders['content-type'
     ].substr(0, 10)").AsString()); | 158   ASSERT_EQ("text/plain", jsEngine->Evaluate("foo.responseHeaders['content-type'
     ].substr(0, 10)").AsString()); | 
| 159   ASSERT_EQ(AdblockPlus::WebRequest::NS_OK, jsEngine->Evaluate("foo.status").AsI
     nt()); | 159   ASSERT_EQ(IWebRequest::NS_OK, jsEngine->Evaluate("foo.status").AsInt()); | 
| 160   ASSERT_EQ(200, jsEngine->Evaluate("foo.responseStatus").AsInt()); | 160   ASSERT_EQ(200, jsEngine->Evaluate("foo.responseStatus").AsInt()); | 
| 161   ASSERT_EQ("[Adblock Plus ", jsEngine->Evaluate("foo.responseText.substr(0, 14)
     ").AsString()); | 161   ASSERT_EQ("[Adblock Plus ", jsEngine->Evaluate("foo.responseText.substr(0, 14)
     ").AsString()); | 
| 162   ASSERT_EQ("text/plain", jsEngine->Evaluate("foo.responseHeaders['content-type'
     ].substr(0, 10)").AsString()); | 162   ASSERT_EQ("text/plain", jsEngine->Evaluate("foo.responseHeaders['content-type'
     ].substr(0, 10)").AsString()); | 
| 163 #if defined(HAVE_CURL) | 163 #if defined(HAVE_CURL) | 
| 164   ASSERT_EQ("gzip", jsEngine->Evaluate("foo.responseHeaders['content-encoding'].
     substr(0, 4)").AsString()); | 164   ASSERT_EQ("gzip", jsEngine->Evaluate("foo.responseHeaders['content-encoding'].
     substr(0, 4)").AsString()); | 
| 165 #endif | 165 #endif | 
| 166   ASSERT_TRUE(jsEngine->Evaluate("foo.responseHeaders['location']").IsUndefined(
     )); | 166   ASSERT_TRUE(jsEngine->Evaluate("foo.responseHeaders['location']").IsUndefined(
     )); | 
| 167 } | 167 } | 
| 168 | 168 | 
| 169 TEST_F(DefaultWebRequestTest, XMLHttpRequest) | 169 TEST_F(DefaultWebRequestTest, XMLHttpRequest) | 
| 170 { | 170 { | 
| 171   auto filterEngine = AdblockPlus::FilterEngine::Create(jsEngine); | 171   auto filterEngine = AdblockPlus::FilterEngine::Create(jsEngine); | 
| 172 | 172 | 
| 173   ResetTestXHR(jsEngine, "https://easylist-downloads.adblockplus.org/easylist.tx
     t"); | 173   ResetTestXHR(jsEngine, "https://easylist-downloads.adblockplus.org/easylist.tx
     t"); | 
| 174   jsEngine->Evaluate("\ | 174   jsEngine->Evaluate("\ | 
| 175     request.setRequestHeader('X', 'Y');\ | 175     request.setRequestHeader('X', 'Y');\ | 
| 176     request.setRequestHeader('X2', 'Y2');\ | 176     request.setRequestHeader('X2', 'Y2');\ | 
| 177     request.send(null);"); | 177     request.send(null);"); | 
| 178   WaitForVariable("result", jsEngine); | 178   WaitForVariable("result", jsEngine); | 
| 179   ASSERT_EQ(AdblockPlus::WebRequest::NS_OK, jsEngine->Evaluate("request.channel.
     status").AsInt()); | 179   ASSERT_EQ(IWebRequest::NS_OK, jsEngine->Evaluate("request.channel.status").AsI
     nt()); | 
| 180   ASSERT_EQ(200, jsEngine->Evaluate("request.status").AsInt()); | 180   ASSERT_EQ(200, jsEngine->Evaluate("request.status").AsInt()); | 
| 181   ASSERT_EQ("[Adblock Plus ", jsEngine->Evaluate("result.substr(0, 14)").AsStrin
     g()); | 181   ASSERT_EQ("[Adblock Plus ", jsEngine->Evaluate("result.substr(0, 14)").AsStrin
     g()); | 
| 182   ASSERT_EQ("text/plain", jsEngine->Evaluate("request.getResponseHeader('Content
     -Type').substr(0, 10)").AsString()); | 182   ASSERT_EQ("text/plain", jsEngine->Evaluate("request.getResponseHeader('Content
     -Type').substr(0, 10)").AsString()); | 
| 183 #if defined(HAVE_CURL) | 183 #if defined(HAVE_CURL) | 
| 184   ASSERT_EQ("gzip", jsEngine->Evaluate("request.getResponseHeader('Content-Encod
     ing').substr(0, 4)").AsString()); | 184   ASSERT_EQ("gzip", jsEngine->Evaluate("request.getResponseHeader('Content-Encod
     ing').substr(0, 4)").AsString()); | 
| 185 #endif | 185 #endif | 
| 186   ASSERT_TRUE(jsEngine->Evaluate("request.getResponseHeader('Location')").IsNull
     ()); | 186   ASSERT_TRUE(jsEngine->Evaluate("request.getResponseHeader('Location')").IsNull
     ()); | 
| 187 } | 187 } | 
| 188 #else | 188 #else | 
| 189 TEST_F(DefaultWebRequestTest, DummyWebRequest) | 189 TEST_F(DefaultWebRequestTest, DummyWebRequest) | 
| 190 { | 190 { | 
| 191   jsEngine->Evaluate("_webRequest.GET('https://easylist-downloads.adblockplus.or
     g/easylist.txt', {}, function(result) {foo = result;} )"); | 191   jsEngine->Evaluate("_webRequest.GET('https://easylist-downloads.adblockplus.or
     g/easylist.txt', {}, function(result) {foo = result;} )"); | 
| 192   WaitForVariable("this.foo", jsEngine); | 192   WaitForVariable("this.foo", jsEngine); | 
| 193   ASSERT_EQ(AdblockPlus::WebRequest::NS_ERROR_FAILURE, jsEngine->Evaluate("foo.s
     tatus").AsInt()); | 193   ASSERT_EQ(IWebRequest::NS_ERROR_FAILURE, jsEngine->Evaluate("foo.status").AsIn
     t()); | 
| 194   ASSERT_EQ(0, jsEngine->Evaluate("foo.responseStatus").AsInt()); | 194   ASSERT_EQ(0, jsEngine->Evaluate("foo.responseStatus").AsInt()); | 
| 195   ASSERT_EQ("", jsEngine->Evaluate("foo.responseText").AsString()); | 195   ASSERT_EQ("", jsEngine->Evaluate("foo.responseText").AsString()); | 
| 196   ASSERT_EQ("{}", jsEngine->Evaluate("JSON.stringify(foo.responseHeaders)").AsSt
     ring()); | 196   ASSERT_EQ("{}", jsEngine->Evaluate("JSON.stringify(foo.responseHeaders)").AsSt
     ring()); | 
| 197 } | 197 } | 
| 198 | 198 | 
| 199 TEST_F(DefaultWebRequestTest, XMLHttpRequest) | 199 TEST_F(DefaultWebRequestTest, XMLHttpRequest) | 
| 200 { | 200 { | 
| 201   auto filterEngine = AdblockPlus::FilterEngine::Create(jsEngine); | 201   auto filterEngine = AdblockPlus::FilterEngine::Create(jsEngine); | 
| 202 | 202 | 
| 203   ResetTestXHR(jsEngine); | 203   ResetTestXHR(jsEngine); | 
| 204   jsEngine->Evaluate("\ | 204   jsEngine->Evaluate("\ | 
| 205     request.setRequestHeader('X', 'Y');\ | 205     request.setRequestHeader('X', 'Y');\ | 
| 206     request.send(null);"); | 206     request.send(null);"); | 
| 207   WaitForVariable("result", jsEngine); | 207   WaitForVariable("result", jsEngine); | 
| 208   ASSERT_EQ(AdblockPlus::WebRequest::NS_ERROR_FAILURE, jsEngine->Evaluate("reque
     st.channel.status").AsInt()); | 208   ASSERT_EQ(IWebRequest::NS_ERROR_FAILURE, jsEngine->Evaluate("request.channel.s
     tatus").AsInt()); | 
| 209   ASSERT_EQ(0, jsEngine->Evaluate("request.status").AsInt()); | 209   ASSERT_EQ(0, jsEngine->Evaluate("request.status").AsInt()); | 
| 210   ASSERT_EQ("error", jsEngine->Evaluate("result").AsString()); | 210   ASSERT_EQ("error", jsEngine->Evaluate("result").AsString()); | 
| 211   ASSERT_TRUE(jsEngine->Evaluate("request.getResponseHeader('Content-Type')").Is
     Null()); | 211   ASSERT_TRUE(jsEngine->Evaluate("request.getResponseHeader('Content-Type')").Is
     Null()); | 
| 212 } | 212 } | 
| 213 | 213 | 
| 214 #endif | 214 #endif | 
| 215 | 215 | 
| 216 namespace | 216 namespace | 
| 217 { | 217 { | 
| 218   class CatchLogSystem : public AdblockPlus::LogSystem | 218   class CatchLogSystem : public AdblockPlus::LogSystem | 
| (...skipping 121 matching lines...) Expand 10 before | Expand all | Expand 10 after  Loading... | 
| 340   EXPECT_EQ(AdblockPlus::LogSystem::LOG_LEVEL_TRACE, catchLogSystem->lastLogLeve
     l); | 340   EXPECT_EQ(AdblockPlus::LogSystem::LOG_LEVEL_TRACE, catchLogSystem->lastLogLeve
     l); | 
| 341   EXPECT_EQ("", catchLogSystem->lastMessage); | 341   EXPECT_EQ("", catchLogSystem->lastMessage); | 
| 342   WaitForVariable("result", jsEngine); | 342   WaitForVariable("result", jsEngine); | 
| 343   { | 343   { | 
| 344     auto headersRequest = webRequest->headersForRequest(url); | 344     auto headersRequest = webRequest->headersForRequest(url); | 
| 345     EXPECT_TRUE(headersRequest.first); | 345     EXPECT_TRUE(headersRequest.first); | 
| 346     const auto& headers = headersRequest.second; | 346     const auto& headers = headersRequest.second; | 
| 347     EXPECT_FALSE(headers.cend() == headers.find("Security")); | 347     EXPECT_FALSE(headers.cend() == headers.find("Security")); | 
| 348   } | 348   } | 
| 349 } | 349 } | 
| OLD | NEW | 
|---|