| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef NET_DNS_MAPPED_HOST_RESOLVER_H_ | 5 #ifndef NET_DNS_MAPPED_HOST_RESOLVER_H_ |
| 6 #define NET_DNS_MAPPED_HOST_RESOLVER_H_ | 6 #define NET_DNS_MAPPED_HOST_RESOLVER_H_ |
| 7 | 7 |
| 8 #include <string> | 8 #include <string> |
| 9 | 9 |
| 10 #include "base/memory/scoped_ptr.h" | 10 #include "base/memory/scoped_ptr.h" |
| 11 #include "net/base/host_mapping_rules.h" | 11 #include "net/base/host_mapping_rules.h" |
| 12 #include "net/base/net_export.h" | 12 #include "net/base/net_export.h" |
| 13 #include "net/dns/host_resolver.h" | 13 #include "net/dns/host_resolver.h" |
| 14 | 14 |
| 15 namespace net { | 15 namespace net { |
| 16 | 16 |
| 17 // This class wraps an existing HostResolver instance, but modifies the | 17 // This class wraps an existing HostResolver instance, but modifies the |
| 18 // request before passing it off to |impl|. This is different from | 18 // request before passing it off to |impl|. This is different from |
| 19 // MockHostResolver which does the remapping at the HostResolverProc | 19 // MockHostResolver which does the remapping at the HostResolverProc |
| 20 // layer, so it is able to preserve the effectiveness of the cache. | 20 // layer, so it is able to preserve the effectiveness of the cache. |
| 21 class NET_EXPORT MappedHostResolver : public HostResolver { | 21 class NET_EXPORT MappedHostResolver : public HostResolver { |
| 22 public: | 22 public: |
| 23 // Creates a MappedHostResolver that forwards all of its requests through | 23 // Creates a MappedHostResolver that forwards all of its requests through |
| 24 // |impl|. | 24 // |impl|. |
| 25 explicit MappedHostResolver(scoped_ptr<HostResolver> impl); | 25 explicit MappedHostResolver(scoped_ptr<HostResolver> impl); |
| 26 virtual ~MappedHostResolver(); | 26 ~MappedHostResolver() override; |
| 27 | 27 |
| 28 // Adds a rule to this mapper. The format of the rule can be one of: | 28 // Adds a rule to this mapper. The format of the rule can be one of: |
| 29 // | 29 // |
| 30 // "MAP" <hostname_pattern> <replacement_host> [":" <replacement_port>] | 30 // "MAP" <hostname_pattern> <replacement_host> [":" <replacement_port>] |
| 31 // "EXCLUDE" <hostname_pattern> | 31 // "EXCLUDE" <hostname_pattern> |
| 32 // | 32 // |
| 33 // The <replacement_host> can be either a hostname, or an IP address literal, | 33 // The <replacement_host> can be either a hostname, or an IP address literal, |
| 34 // or "~NOTFOUND". If it is "~NOTFOUND" then all matched hostnames will fail | 34 // or "~NOTFOUND". If it is "~NOTFOUND" then all matched hostnames will fail |
| 35 // to be resolved with ERR_NAME_NOT_RESOLVED. | 35 // to be resolved with ERR_NAME_NOT_RESOLVED. |
| 36 // | 36 // |
| 37 // Returns true if the rule was successfully parsed and added. | 37 // Returns true if the rule was successfully parsed and added. |
| 38 bool AddRuleFromString(const std::string& rule_string) { | 38 bool AddRuleFromString(const std::string& rule_string) { |
| 39 return rules_.AddRuleFromString(rule_string); | 39 return rules_.AddRuleFromString(rule_string); |
| 40 } | 40 } |
| 41 | 41 |
| 42 // Takes a comma separated list of rules, and assigns them to this resolver. | 42 // Takes a comma separated list of rules, and assigns them to this resolver. |
| 43 void SetRulesFromString(const std::string& rules_string) { | 43 void SetRulesFromString(const std::string& rules_string) { |
| 44 rules_.SetRulesFromString(rules_string); | 44 rules_.SetRulesFromString(rules_string); |
| 45 } | 45 } |
| 46 | 46 |
| 47 // HostResolver methods: | 47 // HostResolver methods: |
| 48 virtual int Resolve(const RequestInfo& info, | 48 int Resolve(const RequestInfo& info, |
| 49 RequestPriority priority, | 49 RequestPriority priority, |
| 50 AddressList* addresses, | 50 AddressList* addresses, |
| 51 const CompletionCallback& callback, | 51 const CompletionCallback& callback, |
| 52 RequestHandle* out_req, | 52 RequestHandle* out_req, |
| 53 const BoundNetLog& net_log) override; | 53 const BoundNetLog& net_log) override; |
| 54 virtual int ResolveFromCache(const RequestInfo& info, | 54 int ResolveFromCache(const RequestInfo& info, |
| 55 AddressList* addresses, | 55 AddressList* addresses, |
| 56 const BoundNetLog& net_log) override; | 56 const BoundNetLog& net_log) override; |
| 57 virtual void CancelRequest(RequestHandle req) override; | 57 void CancelRequest(RequestHandle req) override; |
| 58 virtual void SetDnsClientEnabled(bool enabled) override; | 58 void SetDnsClientEnabled(bool enabled) override; |
| 59 virtual HostCache* GetHostCache() override; | 59 HostCache* GetHostCache() override; |
| 60 virtual base::Value* GetDnsConfigAsValue() const override; | 60 base::Value* GetDnsConfigAsValue() const override; |
| 61 | 61 |
| 62 private: | 62 private: |
| 63 // Modify the request |info| according to |rules_|. Returns either OK or | 63 // Modify the request |info| according to |rules_|. Returns either OK or |
| 64 // the network error code that the hostname's resolution mapped to. | 64 // the network error code that the hostname's resolution mapped to. |
| 65 int ApplyRules(RequestInfo* info) const; | 65 int ApplyRules(RequestInfo* info) const; |
| 66 | 66 |
| 67 scoped_ptr<HostResolver> impl_; | 67 scoped_ptr<HostResolver> impl_; |
| 68 | 68 |
| 69 HostMappingRules rules_; | 69 HostMappingRules rules_; |
| 70 }; | 70 }; |
| 71 | 71 |
| 72 } // namespace net | 72 } // namespace net |
| 73 | 73 |
| 74 #endif // NET_DNS_MAPPED_HOST_RESOLVER_H_ | 74 #endif // NET_DNS_MAPPED_HOST_RESOLVER_H_ |
| OLD | NEW |