| Left: | ||
| Right: |
| OLD | NEW |
|---|---|
| 1 #include "PluginStdAfx.h" | 1 #include "PluginStdAfx.h" |
| 2 | 2 |
| 3 #include "PluginSettings.h" | 3 #include "PluginSettings.h" |
| 4 #include "PluginSystem.h" | 4 #include "PluginSystem.h" |
| 5 #include "PluginFilter.h" | 5 #include "PluginFilter.h" |
| 6 #include "PluginClientFactory.h" | 6 #include "PluginClientFactory.h" |
| 7 #include "PluginMutex.h" | 7 #include "PluginMutex.h" |
| 8 #include "PluginClass.h" | 8 #include "PluginClass.h" |
| 9 | 9 |
| 10 #include "AdblockPlusClient.h" | 10 #include "AdblockPlusClient.h" |
| (...skipping 20 matching lines...) Expand all Loading... | |
| 31 0, 0, 0, 0, 0, 0, &startupInfo, &processInformatio n)) | 31 0, 0, 0, 0, 0, 0, &startupInfo, &processInformatio n)) |
| 32 { | 32 { |
| 33 DWORD error = GetLastError(); | 33 DWORD error = GetLastError(); |
| 34 throw std::runtime_error("Failed to start Adblock Plus Engine"); | 34 throw std::runtime_error("Failed to start Adblock Plus Engine"); |
| 35 } | 35 } |
| 36 | 36 |
| 37 CloseHandle(processInformation.hProcess); | 37 CloseHandle(processInformation.hProcess); |
| 38 CloseHandle(processInformation.hThread); | 38 CloseHandle(processInformation.hThread); |
| 39 } | 39 } |
| 40 | 40 |
| 41 std::auto_ptr<Communication::Pipe> OpenAdblockPlusEnginePipe() | 41 Communication::Pipe* OpenEnginePipe() |
| 42 { | 42 { |
| 43 try | 43 try |
| 44 { | 44 { |
| 45 return std::auto_ptr<Communication::Pipe>(new Communication::Pipe(Communic ation::pipeName, Communication::Pipe::MODE_CONNECT)); | 45 return new Communication::Pipe(Communication::pipeName, Communication::Pip e::MODE_CONNECT); |
| 46 } | 46 } |
| 47 catch (Communication::PipeConnectionError e) | 47 catch (Communication::PipeConnectionError e) |
| 48 { | 48 { |
| 49 SpawnAdblockPlusEngine(); | 49 SpawnAdblockPlusEngine(); |
| 50 | 50 |
| 51 const int step = 100; | 51 const int step = 100; |
| 52 for (int timeout = 10000; timeout > 0; timeout -= step) | 52 for (int timeout = 10000; timeout > 0; timeout -= step) |
| 53 { | 53 { |
| 54 Sleep(step); | 54 Sleep(step); |
| 55 try | 55 try |
| 56 { | 56 { |
| 57 return std::auto_ptr<Communication::Pipe>(new Communication::Pipe(Comm unication::pipeName, Communication::Pipe::MODE_CONNECT)); | 57 return new Communication::Pipe(Communication::pipeName, Communication: :Pipe::MODE_CONNECT); |
| 58 } | 58 } |
| 59 catch (Communication::PipeConnectionError e) | 59 catch (Communication::PipeConnectionError e) |
| 60 { | 60 { |
| 61 } | 61 } |
| 62 } | 62 } |
| 63 throw std::runtime_error("Unable to open Adblock Plus Engine pipe"); | 63 throw std::runtime_error("Unable to open Adblock Plus Engine pipe"); |
| 64 } | 64 } |
| 65 } | 65 } |
| 66 | 66 |
| 67 std::vector<std::wstring> ReadStrings(Communication::InputBuffer& message) | 67 std::vector<std::wstring> ReadStrings(Communication::InputBuffer& message) |
| (...skipping 27 matching lines...) Expand all Loading... | |
| 95 message >> title; | 95 message >> title; |
| 96 description.title = ToUtf16String(title); | 96 description.title = ToUtf16String(title); |
| 97 std::string specialization; | 97 std::string specialization; |
| 98 message >> specialization; | 98 message >> specialization; |
| 99 description.specialization = ToUtf16String(specialization); | 99 description.specialization = ToUtf16String(specialization); |
| 100 message >> description.listed; | 100 message >> description.listed; |
| 101 result.push_back(description); | 101 result.push_back(description); |
| 102 } | 102 } |
| 103 return result; | 103 return result; |
| 104 } | 104 } |
| 105 | |
| 106 bool CallEngine(Communication::OutputBuffer& message, Communication::InputBuff er& inputBuffer = Communication::InputBuffer()) | |
| 107 { | |
| 108 try | |
| 109 { | |
| 110 std::auto_ptr<Communication::Pipe> pipe = OpenAdblockPlusEnginePipe(); | |
| 111 pipe->WriteMessage(message); | |
| 112 inputBuffer = pipe->ReadMessage(); | |
| 113 } | |
| 114 catch (const std::exception& e) | |
| 115 { | |
| 116 DEBUG_GENERAL(e.what()); | |
| 117 return false; | |
| 118 } | |
| 119 return true; | |
| 120 } | |
| 121 | |
| 122 bool CallEngine(Communication::ProcType proc, Communication::InputBuffer& inpu tBuffer = Communication::InputBuffer()) | |
| 123 { | |
| 124 Communication::OutputBuffer message; | |
| 125 message << proc; | |
| 126 return CallEngine(message, inputBuffer); | |
| 127 } | |
| 128 } | 105 } |
| 129 | 106 |
| 130 CAdblockPlusClient* CAdblockPlusClient::s_instance = NULL; | 107 CAdblockPlusClient* CAdblockPlusClient::s_instance = NULL; |
| 131 | 108 |
| 132 CAdblockPlusClient::CAdblockPlusClient() : CPluginClientBase() | 109 CAdblockPlusClient::CAdblockPlusClient() : CPluginClientBase() |
| 133 { | 110 { |
| 134 m_filter = std::auto_ptr<CPluginFilter>(new CPluginFilter()); | 111 m_filter = std::auto_ptr<CPluginFilter>(new CPluginFilter()); |
| 135 } | 112 } |
| 136 | 113 |
| 114 bool CAdblockPlusClient::CallEngine(Communication::OutputBuffer& message, Commun ication::InputBuffer& inputBuffer) | |
| 115 { | |
| 116 try | |
| 117 { | |
| 118 if (!enginePipe) | |
| 119 enginePipe.reset(OpenEnginePipe()); | |
|
Oleksandr
2013/08/08 06:26:31
Might it make sense to have some sort of a thread
Felix Dahlke
2013/08/08 08:32:51
Do we use a single CAdblockPlusClient instance fro
| |
| 120 enginePipe->WriteMessage(message); | |
| 121 inputBuffer = enginePipe->ReadMessage(); | |
| 122 } | |
| 123 catch (const std::exception& e) | |
| 124 { | |
| 125 DEBUG_GENERAL(e.what()); | |
| 126 return false; | |
| 127 } | |
| 128 return true; | |
| 129 } | |
| 130 | |
| 131 bool CAdblockPlusClient::CallEngine(Communication::ProcType proc, Communication: :InputBuffer& inputBuffer) | |
| 132 { | |
| 133 Communication::OutputBuffer message; | |
| 134 message << proc; | |
| 135 return CallEngine(message, inputBuffer); | |
| 136 } | |
| 137 | |
| 137 CAdblockPlusClient::~CAdblockPlusClient() | 138 CAdblockPlusClient::~CAdblockPlusClient() |
| 138 { | 139 { |
| 139 s_instance = NULL; | 140 s_instance = NULL; |
| 140 } | 141 } |
| 141 | 142 |
| 142 | 143 |
| 143 CAdblockPlusClient* CAdblockPlusClient::GetInstance() | 144 CAdblockPlusClient* CAdblockPlusClient::GetInstance() |
| 144 { | 145 { |
| 145 CAdblockPlusClient* instance = NULL; | 146 CAdblockPlusClient* instance = NULL; |
| 146 | 147 |
| (...skipping 278 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 425 | 426 |
| 426 std::wstring CAdblockPlusClient::GetDocumentationLink() | 427 std::wstring CAdblockPlusClient::GetDocumentationLink() |
| 427 { | 428 { |
| 428 Communication::InputBuffer response; | 429 Communication::InputBuffer response; |
| 429 if (!CallEngine(Communication::PROC_GET_DOCUMENTATION_LINK, response)) | 430 if (!CallEngine(Communication::PROC_GET_DOCUMENTATION_LINK, response)) |
| 430 return L""; | 431 return L""; |
| 431 std::wstring docLink; | 432 std::wstring docLink; |
| 432 response >> docLink; | 433 response >> docLink; |
| 433 return docLink; | 434 return docLink; |
| 434 } | 435 } |
| OLD | NEW |