| 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 |
| 11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | 11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 12 * GNU General Public License for more details. | 12 * GNU General Public License for more details. |
| 13 * | 13 * |
| 14 * You should have received a copy of the GNU General Public License | 14 * You should have received a copy of the GNU General Public License |
| 15 * along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. | 15 * along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. |
| 16 */ | 16 */ |
| 17 | 17 |
| 18 #include <map> | 18 #include <map> |
| 19 #include <AdblockPlus/JsValue.h> | 19 #include <AdblockPlus/IWebRequest.h> |
| 20 #include <AdblockPlus/WebRequest.h> | |
| 21 | 20 |
| 22 #include "JsContext.h" | 21 #include "JsContext.h" |
| 23 #include "Thread.h" | |
| 24 #include "Utils.h" | 22 #include "Utils.h" |
| 25 #include "WebRequestJsObject.h" | 23 #include "WebRequestJsObject.h" |
| 24 #include <thread> // TODO: remove with removing of JsEngine::webRequestLegacy |
| 25 |
| 26 using namespace AdblockPlus; |
| 27 |
| 28 void JsEngine::ScheduleWebRequest(const v8::Arguments& arguments) |
| 29 { |
| 30 AdblockPlus::JsEnginePtr jsEngine = AdblockPlus::JsEngine::FromArguments(argum
ents); |
| 31 AdblockPlus::JsValueList converted = jsEngine->ConvertArguments(arguments); |
| 32 if (converted.size() != 3u) |
| 33 throw std::runtime_error("GET requires exactly 3 arguments"); |
| 34 |
| 35 auto url = converted[0].AsString(); |
| 36 if (!url.length()) |
| 37 throw std::runtime_error("Invalid string passed as first argument to GET"); |
| 38 |
| 39 AdblockPlus::HeaderList headers; |
| 40 { |
| 41 const AdblockPlus::JsValue& headersObj = converted[1]; |
| 42 if (!headersObj.IsObject()) |
| 43 throw std::runtime_error("Second argument to GET must be an object"); |
| 44 |
| 45 std::vector<std::string> properties = headersObj.GetOwnPropertyNames(); |
| 46 for (const auto& header : properties) |
| 47 { |
| 48 std::string headerValue = headersObj.GetProperty(header).AsString(); |
| 49 if (header.length() && headerValue.length()) |
| 50 headers.push_back(std::pair<std::string, std::string>(header, headerValu
e)); |
| 51 } |
| 52 } |
| 53 |
| 54 if (!converted[2].IsFunction()) |
| 55 throw std::runtime_error("Third argument to GET must be a function"); |
| 56 |
| 57 auto paramsID = jsEngine->StoreJsValues(converted); |
| 58 std::weak_ptr<JsEngine> weakJsEngine = jsEngine; |
| 59 auto getCallback = [weakJsEngine, paramsID](const ServerResponse& response) |
| 60 { |
| 61 auto jsEngine = weakJsEngine.lock(); |
| 62 if (!jsEngine) |
| 63 return; |
| 64 auto webRequestParams = jsEngine->TakeJsValues(paramsID); |
| 65 |
| 66 AdblockPlus::JsContext context(*jsEngine); |
| 67 |
| 68 auto resultObject = jsEngine->NewObject(); |
| 69 resultObject.SetProperty("status", response.status); |
| 70 resultObject.SetProperty("responseStatus", response.responseStatus); |
| 71 resultObject.SetProperty("responseText", response.responseText); |
| 72 |
| 73 auto headersObject = jsEngine->NewObject(); |
| 74 for (const auto& header : response.responseHeaders) |
| 75 { |
| 76 headersObject.SetProperty(header.first, header.second); |
| 77 } |
| 78 resultObject.SetProperty("responseHeaders", headersObject); |
| 79 |
| 80 webRequestParams[2].Call(resultObject); |
| 81 }; |
| 82 |
| 83 |
| 84 if (jsEngine->webRequestLegacy) |
| 85 { |
| 86 std::thread([jsEngine, url, headers, getCallback] |
| 87 { |
| 88 getCallback(jsEngine->webRequestLegacy->GET(url, headers)); |
| 89 }).detach(); |
| 90 return; |
| 91 } |
| 92 |
| 93 jsEngine->webRequest->GET(url, headers, getCallback); |
| 94 } |
| 26 | 95 |
| 27 namespace | 96 namespace |
| 28 { | 97 { |
| 29 class WebRequestThread : public AdblockPlus::Thread | |
| 30 { | |
| 31 public: | |
| 32 WebRequestThread(const AdblockPlus::JsEnginePtr& jsEngine, const AdblockPlus
::JsValueList& arguments) | |
| 33 : Thread(true), jsEngine(jsEngine), url(arguments[0].AsString()), | |
| 34 callback(arguments[2]) | |
| 35 { | |
| 36 if (!url.length()) | |
| 37 throw std::runtime_error("Invalid string passed as first argument to GET
"); | |
| 38 | |
| 39 { | |
| 40 const AdblockPlus::JsValue& headersObj = arguments[1]; | |
| 41 if (!headersObj.IsObject()) | |
| 42 throw std::runtime_error("Second argument to GET must be an object"); | |
| 43 | |
| 44 std::vector<std::string> properties = headersObj.GetOwnPropertyNames(); | |
| 45 for (const auto& header : properties) | |
| 46 { | |
| 47 std::string headerValue = headersObj.GetProperty(header).AsString(); | |
| 48 if (header.length() && headerValue.length()) | |
| 49 headers.push_back(std::pair<std::string, std::string>(header, header
Value)); | |
| 50 } | |
| 51 } | |
| 52 | |
| 53 if (!callback.IsFunction()) | |
| 54 throw std::runtime_error("Third argument to GET must be a function"); | |
| 55 } | |
| 56 | |
| 57 void Run() | |
| 58 { | |
| 59 AdblockPlus::ServerResponse result = jsEngine->GetWebRequest()->GET(url, h
eaders); | |
| 60 | |
| 61 AdblockPlus::JsContext context(*jsEngine); | |
| 62 | |
| 63 auto resultObject = jsEngine->NewObject(); | |
| 64 resultObject.SetProperty("status", result.status); | |
| 65 resultObject.SetProperty("responseStatus", result.responseStatus); | |
| 66 resultObject.SetProperty("responseText", result.responseText); | |
| 67 | |
| 68 auto headersObject = jsEngine->NewObject(); | |
| 69 for (const auto& header : result.responseHeaders) | |
| 70 { | |
| 71 headersObject.SetProperty(header.first, header.second); | |
| 72 } | |
| 73 resultObject.SetProperty("responseHeaders", headersObject); | |
| 74 | |
| 75 AdblockPlus::JsValueList params; | |
| 76 params.push_back(resultObject); | |
| 77 callback.Call(params); | |
| 78 } | |
| 79 | |
| 80 private: | |
| 81 AdblockPlus::JsEnginePtr jsEngine; | |
| 82 std::string url; | |
| 83 AdblockPlus::HeaderList headers; | |
| 84 AdblockPlus::JsValue callback; | |
| 85 }; | |
| 86 | |
| 87 v8::Handle<v8::Value> GETCallback(const v8::Arguments& arguments) | 98 v8::Handle<v8::Value> GETCallback(const v8::Arguments& arguments) |
| 88 { | 99 { |
| 89 WebRequestThread* thread; | |
| 90 try | 100 try |
| 91 { | 101 { |
| 92 AdblockPlus::JsEnginePtr jsEngine = AdblockPlus::JsEngine::FromArguments(a
rguments); | 102 AdblockPlus::JsEngine::ScheduleWebRequest(arguments); |
| 93 AdblockPlus::JsValueList converted = jsEngine->ConvertArguments(arguments)
; | 103 } catch (const std::exception& e) |
| 94 if (converted.size() != 3u) | |
| 95 throw std::runtime_error("GET requires exactly 3 arguments"); | |
| 96 thread = new WebRequestThread(jsEngine, converted); | |
| 97 } | |
| 98 catch (const std::exception& e) | |
| 99 { | 104 { |
| 100 using AdblockPlus::Utils::ToV8String; | 105 using AdblockPlus::Utils::ToV8String; |
| 101 v8::Isolate* isolate = arguments.GetIsolate(); | 106 v8::Isolate* isolate = arguments.GetIsolate(); |
| 102 return v8::ThrowException(ToV8String(isolate, e.what())); | 107 return v8::ThrowException(ToV8String(isolate, e.what())); |
| 103 } | 108 } |
| 104 thread->Start(); | 109 |
| 105 return v8::Undefined(); | 110 return v8::Undefined(); |
| 106 } | 111 } |
| 107 } | 112 } |
| 108 | 113 |
| 109 AdblockPlus::JsValue& AdblockPlus::WebRequestJsObject::Setup( | 114 AdblockPlus::JsValue& AdblockPlus::WebRequestJsObject::Setup( |
| 110 AdblockPlus::JsEngine& jsEngine, AdblockPlus::JsValue& obj) | 115 AdblockPlus::JsEngine& jsEngine, AdblockPlus::JsValue& obj) |
| 111 { | 116 { |
| 112 obj.SetProperty("GET", jsEngine.NewCallback(::GETCallback)); | 117 obj.SetProperty("GET", jsEngine.NewCallback(::GETCallback)); |
| 113 return obj; | 118 return obj; |
| 114 } | 119 } |
| OLD | NEW |