| OLD | NEW |
| 1 #include <algorithm> | 1 #include <algorithm> |
| 2 #include <cctype> | 2 #include <cctype> |
| 3 #include <functional> | 3 #include <functional> |
| 4 | 4 |
| 5 #include <AdblockPlus.h> | 5 #include <AdblockPlus.h> |
| 6 | 6 |
| 7 using namespace AdblockPlus; | 7 using namespace AdblockPlus; |
| 8 | 8 |
| 9 #if !FILTER_ENGINE_STUBS | 9 #if !FILTER_ENGINE_STUBS |
| 10 extern const char* jsSources[]; | 10 extern const char* jsSources[]; |
| (...skipping 127 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 138 { | 138 { |
| 139 #if FILTER_ENGINE_STUBS | 139 #if FILTER_ENGINE_STUBS |
| 140 for (std::vector<FilterPtr>::iterator it = filterEngine.listedFilters.begin(); | 140 for (std::vector<FilterPtr>::iterator it = filterEngine.listedFilters.begin(); |
| 141 it != filterEngine.listedFilters.end(); ++it) | 141 it != filterEngine.listedFilters.end(); ++it) |
| 142 { | 142 { |
| 143 if (it->get() == this) | 143 if (it->get() == this) |
| 144 return true; | 144 return true; |
| 145 } | 145 } |
| 146 return false; | 146 return false; |
| 147 #else | 147 #else |
| 148 JsValuePtr func = jsEngine.Evaluate("API.isListedFilter"); | 148 JsValuePtr func = jsEngine->Evaluate("API.isListedFilter"); |
| 149 JsValueList params; | 149 JsValueList params; |
| 150 params.push_back(shared_from_this()); | 150 params.push_back(shared_from_this()); |
| 151 return func->Call(params)->AsBool(); | 151 return func->Call(params)->AsBool(); |
| 152 #endif | 152 #endif |
| 153 } | 153 } |
| 154 | 154 |
| 155 void Filter::AddToList() | 155 void Filter::AddToList() |
| 156 { | 156 { |
| 157 #if FILTER_ENGINE_STUBS | 157 #if FILTER_ENGINE_STUBS |
| 158 if (!IsListed()) | 158 if (!IsListed()) |
| 159 filterEngine.listedFilters.push_back(shared_from_this()); | 159 filterEngine.listedFilters.push_back(shared_from_this()); |
| 160 #else | 160 #else |
| 161 JsValuePtr func = jsEngine.Evaluate("API.addFilterToList"); | 161 JsValuePtr func = jsEngine->Evaluate("API.addFilterToList"); |
| 162 JsValueList params; | 162 JsValueList params; |
| 163 params.push_back(shared_from_this()); | 163 params.push_back(shared_from_this()); |
| 164 func->Call(params); | 164 func->Call(params); |
| 165 #endif | 165 #endif |
| 166 } | 166 } |
| 167 | 167 |
| 168 void Filter::RemoveFromList() | 168 void Filter::RemoveFromList() |
| 169 { | 169 { |
| 170 #if FILTER_ENGINE_STUBS | 170 #if FILTER_ENGINE_STUBS |
| 171 for (std::vector<FilterPtr>::iterator it = filterEngine.listedFilters.begin(); | 171 for (std::vector<FilterPtr>::iterator it = filterEngine.listedFilters.begin(); |
| 172 it != filterEngine.listedFilters.end();) | 172 it != filterEngine.listedFilters.end();) |
| 173 { | 173 { |
| 174 if (it->get() == this) | 174 if (it->get() == this) |
| 175 it = filterEngine.listedFilters.erase(it); | 175 it = filterEngine.listedFilters.erase(it); |
| 176 else | 176 else |
| 177 it++; | 177 it++; |
| 178 } | 178 } |
| 179 #else | 179 #else |
| 180 JsValuePtr func = jsEngine.Evaluate("API.removeFilterFromList"); | 180 JsValuePtr func = jsEngine->Evaluate("API.removeFilterFromList"); |
| 181 JsValueList params; | 181 JsValueList params; |
| 182 params.push_back(shared_from_this()); | 182 params.push_back(shared_from_this()); |
| 183 func->Call(params); | 183 func->Call(params); |
| 184 #endif | 184 #endif |
| 185 } | 185 } |
| 186 | 186 |
| 187 bool Filter::operator==(const Filter& filter) const | 187 bool Filter::operator==(const Filter& filter) const |
| 188 { | 188 { |
| 189 return GetProperty("text", "") == filter.GetProperty("text", ""); | 189 return GetProperty("text", "") == filter.GetProperty("text", ""); |
| 190 } | 190 } |
| (...skipping 15 matching lines...) Expand all Loading... |
| 206 { | 206 { |
| 207 #if FILTER_ENGINE_STUBS | 207 #if FILTER_ENGINE_STUBS |
| 208 for (std::vector<SubscriptionPtr>::iterator it = filterEngine.listedSubscripti
ons.begin(); | 208 for (std::vector<SubscriptionPtr>::iterator it = filterEngine.listedSubscripti
ons.begin(); |
| 209 it != filterEngine.listedSubscriptions.end(); ++it) | 209 it != filterEngine.listedSubscriptions.end(); ++it) |
| 210 { | 210 { |
| 211 if (it->get() == this) | 211 if (it->get() == this) |
| 212 return true; | 212 return true; |
| 213 } | 213 } |
| 214 return false; | 214 return false; |
| 215 #else | 215 #else |
| 216 JsValuePtr func = jsEngine.Evaluate("API.isListedFilter"); | 216 JsValuePtr func = jsEngine->Evaluate("API.isListedFilter"); |
| 217 JsValueList params; | 217 JsValueList params; |
| 218 params.push_back(shared_from_this()); | 218 params.push_back(shared_from_this()); |
| 219 return func->Call(params)->AsBool(); | 219 return func->Call(params)->AsBool(); |
| 220 #endif | 220 #endif |
| 221 } | 221 } |
| 222 | 222 |
| 223 void Subscription::AddToList() | 223 void Subscription::AddToList() |
| 224 { | 224 { |
| 225 #if FILTER_ENGINE_STUBS | 225 #if FILTER_ENGINE_STUBS |
| 226 if (!IsListed()) | 226 if (!IsListed()) |
| 227 filterEngine.listedSubscriptions.push_back(shared_from_this()); | 227 filterEngine.listedSubscriptions.push_back(shared_from_this()); |
| 228 #else | 228 #else |
| 229 JsValuePtr func = jsEngine.Evaluate("API.addSubscriptionToList"); | 229 JsValuePtr func = jsEngine->Evaluate("API.addSubscriptionToList"); |
| 230 JsValueList params; | 230 JsValueList params; |
| 231 params.push_back(shared_from_this()); | 231 params.push_back(shared_from_this()); |
| 232 func->Call(params); | 232 func->Call(params); |
| 233 #endif | 233 #endif |
| 234 } | 234 } |
| 235 | 235 |
| 236 void Subscription::RemoveFromList() | 236 void Subscription::RemoveFromList() |
| 237 { | 237 { |
| 238 #if FILTER_ENGINE_STUBS | 238 #if FILTER_ENGINE_STUBS |
| 239 for (std::vector<SubscriptionPtr>::iterator it = filterEngine.listedSubscripti
ons.begin(); | 239 for (std::vector<SubscriptionPtr>::iterator it = filterEngine.listedSubscripti
ons.begin(); |
| 240 it != filterEngine.listedSubscriptions.end();) | 240 it != filterEngine.listedSubscriptions.end();) |
| 241 { | 241 { |
| 242 if (it->get() == this) | 242 if (it->get() == this) |
| 243 it = filterEngine.listedSubscriptions.erase(it); | 243 it = filterEngine.listedSubscriptions.erase(it); |
| 244 else | 244 else |
| 245 it++; | 245 it++; |
| 246 } | 246 } |
| 247 #else | 247 #else |
| 248 JsValuePtr func = jsEngine.Evaluate("API.removeSubscriptionFromList"); | 248 JsValuePtr func = jsEngine->Evaluate("API.removeSubscriptionFromList"); |
| 249 JsValueList params; | 249 JsValueList params; |
| 250 params.push_back(shared_from_this()); | 250 params.push_back(shared_from_this()); |
| 251 func->Call(params); | 251 func->Call(params); |
| 252 #endif | 252 #endif |
| 253 } | 253 } |
| 254 | 254 |
| 255 void Subscription::UpdateFilters() | 255 void Subscription::UpdateFilters() |
| 256 { | 256 { |
| 257 #if !FILTER_ENGINE_STUBS | 257 #if !FILTER_ENGINE_STUBS |
| 258 JsValuePtr func = jsEngine.Evaluate("API.updateSubscription"); | 258 JsValuePtr func = jsEngine->Evaluate("API.updateSubscription"); |
| 259 JsValueList params; | 259 JsValueList params; |
| 260 params.push_back(shared_from_this()); | 260 params.push_back(shared_from_this()); |
| 261 func->Call(params); | 261 func->Call(params); |
| 262 #endif | 262 #endif |
| 263 } | 263 } |
| 264 | 264 |
| 265 bool Subscription::IsUpdating() | 265 bool Subscription::IsUpdating() |
| 266 { | 266 { |
| 267 #if FILTER_ENGINE_STUBS | 267 #if FILTER_ENGINE_STUBS |
| 268 return false; | 268 return false; |
| 269 #else | 269 #else |
| 270 JsValuePtr func = jsEngine.Evaluate("API.isSubscriptionUpdating"); | 270 JsValuePtr func = jsEngine->Evaluate("API.isSubscriptionUpdating"); |
| 271 JsValueList params; | 271 JsValueList params; |
| 272 params.push_back(shared_from_this()); | 272 params.push_back(shared_from_this()); |
| 273 JsValuePtr result = func->Call(params); | 273 JsValuePtr result = func->Call(params); |
| 274 return result->AsBool(); | 274 return result->AsBool(); |
| 275 #endif | 275 #endif |
| 276 } | 276 } |
| 277 | 277 |
| 278 bool Subscription::operator==(const Subscription& subscription) const | 278 bool Subscription::operator==(const Subscription& subscription) const |
| 279 { | 279 { |
| 280 return GetProperty("url", "") == subscription.GetProperty("url", ""); | 280 return GetProperty("url", "") == subscription.GetProperty("url", ""); |
| 281 } | 281 } |
| 282 | 282 |
| 283 FilterEngine::FilterEngine(JsEngine& jsEngine) : jsEngine(jsEngine) | 283 FilterEngine::FilterEngine(JsEnginePtr jsEngine) : jsEngine(jsEngine) |
| 284 { | 284 { |
| 285 #if !FILTER_ENGINE_STUBS | 285 #if !FILTER_ENGINE_STUBS |
| 286 for (int i = 0; jsSources[i] && jsSources[i + 1]; i += 2) | 286 for (int i = 0; jsSources[i] && jsSources[i + 1]; i += 2) |
| 287 jsEngine.Evaluate(jsSources[i + 1], jsSources[i]); | 287 jsEngine->Evaluate(jsSources[i + 1], jsSources[i]); |
| 288 #endif | 288 #endif |
| 289 } | 289 } |
| 290 | 290 |
| 291 FilterPtr FilterEngine::GetFilter(const std::string& text) | 291 FilterPtr FilterEngine::GetFilter(const std::string& text) |
| 292 { | 292 { |
| 293 #if FILTER_ENGINE_STUBS | 293 #if FILTER_ENGINE_STUBS |
| 294 // Via http://stackoverflow.com/questions/216823/whats-the-best-way-to-trim-st
dstring | 294 // Via http://stackoverflow.com/questions/216823/whats-the-best-way-to-trim-st
dstring |
| 295 std::string trimmed(text); | 295 std::string trimmed(text); |
| 296 trimmed.erase(trimmed.begin(), std::find_if(trimmed.begin(), trimmed.end(), st
d::not1(std::ptr_fun<int, int>(std::isspace)))); | 296 trimmed.erase(trimmed.begin(), std::find_if(trimmed.begin(), trimmed.end(), st
d::not1(std::ptr_fun<int, int>(std::isspace)))); |
| 297 trimmed.erase(std::find_if(trimmed.rbegin(), trimmed.rend(), std::not1(std::pt
r_fun<int, int>(std::isspace))).base(), trimmed.end()); | 297 trimmed.erase(std::find_if(trimmed.rbegin(), trimmed.rend(), std::not1(std::pt
r_fun<int, int>(std::isspace))).base(), trimmed.end()); |
| 298 | 298 |
| 299 std::map<std::string, FilterPtr>::const_iterator it = knownFilters.find(trimme
d); | 299 std::map<std::string, FilterPtr>::const_iterator it = knownFilters.find(trimme
d); |
| 300 if (it != knownFilters.end()) | 300 if (it != knownFilters.end()) |
| 301 return it->second; | 301 return it->second; |
| 302 | 302 |
| 303 FilterPtr result(new Filter(*this, trimmed)); | 303 FilterPtr result(new Filter(*this, trimmed)); |
| 304 knownFilters[trimmed] = result; | 304 knownFilters[trimmed] = result; |
| 305 return result; | 305 return result; |
| 306 #else | 306 #else |
| 307 JsValuePtr func = jsEngine.Evaluate("API.getFilterFromText"); | 307 JsValuePtr func = jsEngine->Evaluate("API.getFilterFromText"); |
| 308 JsValueList params; | 308 JsValueList params; |
| 309 params.push_back(jsEngine.NewValue(text)); | 309 params.push_back(jsEngine->NewValue(text)); |
| 310 return FilterPtr(new Filter(func->Call(params))); | 310 return FilterPtr(new Filter(func->Call(params))); |
| 311 #endif | 311 #endif |
| 312 } | 312 } |
| 313 | 313 |
| 314 SubscriptionPtr FilterEngine::GetSubscription(const std::string& url) | 314 SubscriptionPtr FilterEngine::GetSubscription(const std::string& url) |
| 315 { | 315 { |
| 316 #if FILTER_ENGINE_STUBS | 316 #if FILTER_ENGINE_STUBS |
| 317 std::map<std::string, SubscriptionPtr>::const_iterator it = knownSubscriptions
.find(url); | 317 std::map<std::string, SubscriptionPtr>::const_iterator it = knownSubscriptions
.find(url); |
| 318 if (it != knownSubscriptions.end()) | 318 if (it != knownSubscriptions.end()) |
| 319 return it->second; | 319 return it->second; |
| 320 | 320 |
| 321 SubscriptionPtr result(new Subscription(*this, url)); | 321 SubscriptionPtr result(new Subscription(*this, url)); |
| 322 knownSubscriptions[url] = result; | 322 knownSubscriptions[url] = result; |
| 323 return result; | 323 return result; |
| 324 #else | 324 #else |
| 325 JsValuePtr func = jsEngine.Evaluate("API.getSubscriptionFromUrl"); | 325 JsValuePtr func = jsEngine->Evaluate("API.getSubscriptionFromUrl"); |
| 326 JsValueList params; | 326 JsValueList params; |
| 327 params.push_back(jsEngine.NewValue(url)); | 327 params.push_back(jsEngine->NewValue(url)); |
| 328 return SubscriptionPtr(new Subscription(func->Call(params))); | 328 return SubscriptionPtr(new Subscription(func->Call(params))); |
| 329 #endif | 329 #endif |
| 330 } | 330 } |
| 331 | 331 |
| 332 const std::vector<FilterPtr> FilterEngine::GetListedFilters() const | 332 const std::vector<FilterPtr> FilterEngine::GetListedFilters() const |
| 333 { | 333 { |
| 334 #if FILTER_ENGINE_STUBS | 334 #if FILTER_ENGINE_STUBS |
| 335 return listedFilters; | 335 return listedFilters; |
| 336 #else | 336 #else |
| 337 JsValuePtr func = jsEngine.Evaluate("API.getListedFilters"); | 337 JsValuePtr func = jsEngine->Evaluate("API.getListedFilters"); |
| 338 JsValueList values = func->Call()->AsList(); | 338 JsValueList values = func->Call()->AsList(); |
| 339 std::vector<FilterPtr> result; | 339 std::vector<FilterPtr> result; |
| 340 for (JsValueList::iterator it = values.begin(); it != values.end(); it++) | 340 for (JsValueList::iterator it = values.begin(); it != values.end(); it++) |
| 341 result.push_back(FilterPtr(new Filter(*it))); | 341 result.push_back(FilterPtr(new Filter(*it))); |
| 342 return result; | 342 return result; |
| 343 #endif | 343 #endif |
| 344 } | 344 } |
| 345 | 345 |
| 346 const std::vector<SubscriptionPtr> FilterEngine::GetListedSubscriptions() const | 346 const std::vector<SubscriptionPtr> FilterEngine::GetListedSubscriptions() const |
| 347 { | 347 { |
| 348 #if FILTER_ENGINE_STUBS | 348 #if FILTER_ENGINE_STUBS |
| 349 return listedSubscriptions; | 349 return listedSubscriptions; |
| 350 #else | 350 #else |
| 351 JsValuePtr func = jsEngine.Evaluate("API.getListedSubscriptions"); | 351 JsValuePtr func = jsEngine->Evaluate("API.getListedSubscriptions"); |
| 352 JsValueList values = func->Call()->AsList(); | 352 JsValueList values = func->Call()->AsList(); |
| 353 std::vector<SubscriptionPtr> result; | 353 std::vector<SubscriptionPtr> result; |
| 354 for (JsValueList::iterator it = values.begin(); it != values.end(); it++) | 354 for (JsValueList::iterator it = values.begin(); it != values.end(); it++) |
| 355 result.push_back(SubscriptionPtr(new Subscription(*it))); | 355 result.push_back(SubscriptionPtr(new Subscription(*it))); |
| 356 return result; | 356 return result; |
| 357 #endif | 357 #endif |
| 358 } | 358 } |
| 359 | 359 |
| 360 void FilterEngine::FetchAvailableSubscriptions(SubscriptionsCallback callback) | 360 void FilterEngine::FetchAvailableSubscriptions(SubscriptionsCallback callback) |
| 361 { | 361 { |
| (...skipping 20 matching lines...) Expand all Loading... |
| 382 { | 382 { |
| 383 #if FILTER_ENGINE_STUBS | 383 #if FILTER_ENGINE_STUBS |
| 384 //For test on http://simple-adblock.com/faq/testing-your-adblocker/ | 384 //For test on http://simple-adblock.com/faq/testing-your-adblocker/ |
| 385 if (url.find("adbanner.gif") != std::string::npos) | 385 if (url.find("adbanner.gif") != std::string::npos) |
| 386 return GetFilter("adbanner.gif"); | 386 return GetFilter("adbanner.gif"); |
| 387 else if (url.find("notbanner.gif") != std::string::npos) | 387 else if (url.find("notbanner.gif") != std::string::npos) |
| 388 return GetFilter("@@notbanner.gif"); | 388 return GetFilter("@@notbanner.gif"); |
| 389 else | 389 else |
| 390 return AdblockPlus::FilterPtr(); | 390 return AdblockPlus::FilterPtr(); |
| 391 #else | 391 #else |
| 392 JsValuePtr func = jsEngine.Evaluate("API.checkFilterMatch"); | 392 JsValuePtr func = jsEngine->Evaluate("API.checkFilterMatch"); |
| 393 JsValueList params; | 393 JsValueList params; |
| 394 params.push_back(jsEngine.NewValue(url)); | 394 params.push_back(jsEngine->NewValue(url)); |
| 395 params.push_back(jsEngine.NewValue(contentType)); | 395 params.push_back(jsEngine->NewValue(contentType)); |
| 396 params.push_back(jsEngine.NewValue(documentUrl)); | 396 params.push_back(jsEngine->NewValue(documentUrl)); |
| 397 JsValuePtr result = func->Call(params); | 397 JsValuePtr result = func->Call(params); |
| 398 if (!result->IsNull()) | 398 if (!result->IsNull()) |
| 399 return FilterPtr(new Filter(result)); | 399 return FilterPtr(new Filter(result)); |
| 400 else | 400 else |
| 401 return FilterPtr(); | 401 return FilterPtr(); |
| 402 #endif | 402 #endif |
| 403 } | 403 } |
| 404 | 404 |
| 405 std::vector<std::string> FilterEngine::GetElementHidingSelectors(const std::stri
ng& domain) const | 405 std::vector<std::string> FilterEngine::GetElementHidingSelectors(const std::stri
ng& domain) const |
| 406 { | 406 { |
| 407 #if FILTER_ENGINE_STUBS | 407 #if FILTER_ENGINE_STUBS |
| 408 std::vector<std::string> selectors; | 408 std::vector<std::string> selectors; |
| 409 selectors.push_back("#ad"); | 409 selectors.push_back("#ad"); |
| 410 selectors.push_back(".ad"); | 410 selectors.push_back(".ad"); |
| 411 //For test on http://simple-adblock.com/faq/testing-your-adblocker/ | 411 //For test on http://simple-adblock.com/faq/testing-your-adblocker/ |
| 412 if (domain == "simple-adblock.com") | 412 if (domain == "simple-adblock.com") |
| 413 selectors.push_back(".ad_300x250"); | 413 selectors.push_back(".ad_300x250"); |
| 414 return selectors; | 414 return selectors; |
| 415 #else | 415 #else |
| 416 JsValuePtr func = jsEngine.Evaluate("API.getElementHidingSelectors"); | 416 JsValuePtr func = jsEngine->Evaluate("API.getElementHidingSelectors"); |
| 417 JsValueList params; | 417 JsValueList params; |
| 418 params.push_back(jsEngine.NewValue(domain)); | 418 params.push_back(jsEngine->NewValue(domain)); |
| 419 JsValueList result = func->Call(params)->AsList(); | 419 JsValueList result = func->Call(params)->AsList(); |
| 420 std::vector<std::string> selectors; | 420 std::vector<std::string> selectors; |
| 421 for (JsValueList::iterator it = result.begin(); it != result.end(); ++it) | 421 for (JsValueList::iterator it = result.begin(); it != result.end(); ++it) |
| 422 selectors.push_back((*it)->AsString()); | 422 selectors.push_back((*it)->AsString()); |
| 423 return selectors; | 423 return selectors; |
| 424 #endif | 424 #endif |
| 425 } | 425 } |
| OLD | NEW |