Rietveld Code Review Tool
Help | Bug tracker | Discussion group | Source code

Side by Side Diff: include/AdblockPlus/ReferrerMapping.h

Issue 5768603836088320: Issue 1564-Fix FilterEngine::Matches for allowing request which is whitelisted in the ascendant node
Patch Set: rename BuildReferrerChain to BuildFrameStructure Created Nov. 25, 2014, 12:40 p.m.
Left:
Right:
Use n/p to move between diff chunks; N/P to move between comments.
Jump to:
View unified diff | Download patch
« no previous file with comments | « no previous file | src/ReferrerMapping.cpp » ('j') | src/ReferrerMapping.cpp » ('J')
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
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
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 #ifndef ADBLOCK_PLUS_REFERRER_MAPPING_H 18 #ifndef ADBLOCK_PLUS_REFERRER_MAPPING_H
19 #define ADBLOCK_PLUS_REFERRER_MAPPING_H 19 #define ADBLOCK_PLUS_REFERRER_MAPPING_H
20 20
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 26
26 namespace AdblockPlus 27 namespace AdblockPlus
27 { 28 {
28 /** 29 /**
29 * Stores a mapping between URLs and their referrers. 30 * Stores a mapping between URLs and their referrers.
30 * 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
31 * (see `BuildReferrerChain()`), which approximates the frame structure, see 32 * (see `BuildFrameStructure()`), it's useful for FilterEngine::Matches().
32 * FilterEngine::Matches().
33 */ 33 */
34 class ReferrerMapping 34 class ReferrerMapping
35 { 35 {
36 public: 36 public:
37 /** 37 /**
38 * Constructor. 38 * Constructor.
39 * @param maxCachedUrls Number of URL mappings to store. The higher the 39 * @param maxCachedUrls Number of URL mappings to store. The higher the
40 * better - clients typically cache requests, and a single cached 40 * better - clients typically cache requests, and a single cached
41 * request will break the referrer chain. 41 * request will break the referrer chain.
42 */ 42 */
43 ReferrerMapping(const int maxCachedUrls = 5000); 43 ReferrerMapping(const int maxCachedUrls = 5000);
44 44
45 /** 45 /**
46 * Records the refferer for a URL. 46 * Records the refferer for a URL.
47 * @param url Request URL. 47 * @param url Request URL.
48 * @param referrer Request referrer. 48 * @param referrer Request referrer.
49 * @param requestType Request type.
49 */ 50 */
50 void Add(const std::string& url, const std::string& referrer); 51 void Add(const std::string& url, const std::string& referrer,
52 FilterEngine::ContentType requestType);
51 53
52 /** 54 /**
53 * Builds a chain of referrers for the supplied URL. 55 * Builds a frame structure for the supplied URL.
54 * This should reconstruct a document's parent frame URLs. 56 * This should reconstruct a document's parent frame URLs.
55 * @param url URL to build the chain for. 57 * @param url URL to build the chain for.
56 * @return List of URLs, starting with `url`. 58 * @return List of URLs, finishing with `url`.
57 */ 59 */
58 std::vector<std::string> BuildReferrerChain(const std::string& url) const; 60 std::vector<std::string> BuildFrameStructure(const std::string& url) const;
59 61
60 private: 62 private:
61 const int maxCachedUrls; 63 const int maxCachedUrls;
62 std::map<std::string, std::string> mapping; 64 struct RequestInfo
65 {
66 RequestInfo(const std::string& referrerArg, FilterEngine::ContentType type Arg)
67 : referrer(referrerArg) , type(typeArg)
68 {
69 }
70 std::string referrer;
71 FilterEngine::ContentType type;
Wladimir Palant 2014/11/25 15:46:37 It is unnecessary to store the actual type here -
sergei 2014/11/26 10:34:43 I would use `isFrame` because it's not necessary t
72 };
73 std::map</*url*/std::string, RequestInfo> mapping;
Wladimir Palant 2014/11/25 15:46:37 How about |using URL = std::string| so that you ca
sergei 2014/11/26 10:34:43 Added typedef for `Url` and `Urls`.
63 std::list<std::string> cachedUrls; 74 std::list<std::string> cachedUrls;
64 }; 75 };
65 } 76 }
66 77
67 #endif 78 #endif
OLDNEW
« no previous file with comments | « no previous file | src/ReferrerMapping.cpp » ('j') | src/ReferrerMapping.cpp » ('J')

Powered by Google App Engine
This is Rietveld