LEFT | RIGHT |
1 /* | 1 /* |
2 * This file is part of Adblock Plus <http://adblockplus.org/>, | 2 * This file is part of Adblock Plus <http://adblockplus.org/>, |
3 * Copyright (C) 2006-2014 Eyeo GmbH | 3 * Copyright (C) 2006-2014 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 10 matching lines...) Expand all Loading... |
21 #include <list> | 21 #include <list> |
22 #include <map> | 22 #include <map> |
23 #include <string> | 23 #include <string> |
24 #include <vector> | 24 #include <vector> |
25 #include "FilterEngine.h" | 25 #include "FilterEngine.h" |
26 | 26 |
27 namespace AdblockPlus | 27 namespace AdblockPlus |
28 { | 28 { |
29 /** | 29 /** |
30 * Stores a mapping between URLs and their referrers. | 30 * Stores a mapping between URLs and their referrers. |
31 * This can be used to build a chain of referrers for any URL | 31 * This can be used to build a frame structure for any URL |
32 * (see `BuildReferrerChain()`), which approximates the frame structure, see | 32 * (see `BuildFrameStructure()`), it's useful for FilterEngine::Matches(). |
33 * FilterEngine::Matches(). | |
34 */ | 33 */ |
35 class ReferrerMapping | 34 class ReferrerMapping |
36 { | 35 { |
37 public: | 36 public: |
| 37 /// The type for URL. |
| 38 typedef std::string Url; |
| 39 |
| 40 /// Contains an ordered list of URLs. |
| 41 typedef std::vector<Url> Urls; |
| 42 |
| 43 /// Indicates whether the URL is a frame or not. |
| 44 enum FrameIndicator |
| 45 { |
| 46 FRAME_INDICATOR_NOT_FRAME = 0, FRAME_INDICATOR_FRAME = 1 |
| 47 }; |
38 /** | 48 /** |
39 * Constructor. | 49 * Constructor. |
40 * @param maxCachedUrls Number of URL mappings to store. The higher the | 50 * @param maxCachedUrls Number of URL mappings to store. The higher the |
41 * better - clients typically cache requests, and a single cached | 51 * better - clients typically cache requests, and a single cached |
42 * request will break the referrer chain. | 52 * request will break the referrer chain. |
43 */ | 53 */ |
44 ReferrerMapping(const int maxCachedUrls = 5000); | 54 ReferrerMapping(const int maxCachedUrls = 5000); |
45 | 55 |
46 /** | 56 /** |
47 * Records the refferer for a URL. | 57 * Records the refferer for a URL. |
48 * @param url Request URL. | 58 * @param url Request URL. |
49 * @param referrer Request referrer. | 59 * @param referrer Request referrer. |
50 * @param requestType Request type. | 60 * @param isFrame Indicates whether the url is a frame. |
51 */ | 61 */ |
52 void Add(const std::string& url, const std::string& referrer, | 62 void Add(const Url& url, const Url& referrer, FrameIndicator isFrame); |
53 FilterEngine::ContentType requestType); | |
54 | 63 |
55 /** | 64 /** |
56 * Builds a chain of referrers for the supplied URL. | 65 * Builds a frame structure for the supplied URL. |
57 * This should reconstruct a document's parent frame URLs. | 66 * This should reconstruct a document's parent frame URLs. |
58 * @param url URL to build the chain for. | 67 * @param url URL to build the chain for, it is referrer as well. |
59 * @return List of URLs, starting with `url`. | 68 * @return List of URLs, finishing with `url`. |
60 */ | 69 */ |
61 std::vector<std::string> BuildReferrerChain(const std::string& url) const; | 70 Urls BuildFrameStructure(const Url& url) const; |
62 | 71 |
63 private: | 72 private: |
64 const int maxCachedUrls; | 73 const int maxCachedUrls; |
65 struct RequestInfo | 74 struct RequestInfo |
66 { | 75 { |
67 RequestInfo(const std::string& referrerArg, FilterEngine::ContentType type
Arg) | 76 explicit RequestInfo(const Url& referrerArg = Url(), |
68 : referrer(referrerArg) , type(typeArg) | 77 FrameIndicator frameIndicatorArg = FrameIndicator::FRAME_INDICATOR_NOT_F
RAME) |
| 78 : referrer(referrerArg), frameIndicator(frameIndicatorArg) |
69 { | 79 { |
70 } | 80 } |
71 std::string referrer; | 81 bool IsFrame() const |
72 FilterEngine::ContentType type; | 82 { |
| 83 return frameIndicator == FrameIndicator::FRAME_INDICATOR_FRAME; |
| 84 } |
| 85 Url referrer; |
| 86 FrameIndicator frameIndicator; |
73 }; | 87 }; |
74 std::map</*url*/std::string, RequestInfo> mapping; | 88 typedef std::map<Url, RequestInfo> ReferrerMap; |
75 std::list<std::string> cachedUrls; | 89 ReferrerMap mapping; |
| 90 std::list<Url> cachedUrls; |
76 }; | 91 }; |
77 } | 92 } |
78 | 93 |
79 #endif | 94 #endif |
LEFT | RIGHT |