1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #ifndef NET_BASE_HTTPS_PROBER_H_ 6 #define NET_BASE_HTTPS_PROBER_H_ 7 8 #include <map> 9 #include <set> 10 #include <string> 11 12 #include "base/singleton.h" 13 #include "base/task.h" 14 #include "net/url_request/url_request.h" 15 16 class URLRequestContext; 17 18 namespace net { 19 20 // This should be scoped inside HTTPSProber, but VC cannot compile 21 // HTTPProber::Delegate when HTTPSProber also inherits from 22 // URLRequest::Delegate. 23 class HTTPSProberDelegate { 24 public: 25 virtual void ProbeComplete(bool result) = 0; 26 }; 27 28 // HTTPSProber is a singleton object that manages HTTPS probes. A HTTPS probe 29 // determines if we can connect to a given host over HTTPS. It's used when 30 // transparently upgrading from HTTP to HTTPS (for example, for SPDY). 31 class HTTPSProber : public URLRequest::Delegate { 32 public: 33 HTTPSProber() { } 34 35 // HaveProbed returns true if the given host is known to have been probed 36 // since the browser was last started. 37 bool HaveProbed(const std::string& host) const; 38 39 // InFlight returns true iff a probe for the given host is currently active. 40 bool InFlight(const std::string& host) const; 41 42 // ProbeHost starts a new probe for the given host. If the host is known to 43 // have been probed since the browser was started, false is returned and no 44 // other action is taken. If a probe to the given host in currently inflight, 45 // false will be returned, and no other action is taken. Otherwise, a new 46 // probe is started, true is returned and the Delegate will be called with the 47 // results (true means a successful handshake). 48 bool ProbeHost(const std::string& host, URLRequestContext* ctx, 49 HTTPSProberDelegate* delegate); 50 51 // Implementation of URLRequest::Delegate 52 void OnAuthRequired(URLRequest* request, 53 net::AuthChallengeInfo* auth_info); 54 void OnSSLCertificateError(URLRequest* request, 55 int cert_error, 56 net::X509Certificate* cert); 57 void OnResponseStarted(URLRequest* request); 58 void OnReadCompleted(URLRequest* request, int bytes_read); 59 60 private: 61 void Success(URLRequest* request); 62 void Failure(URLRequest* request); 63 void DoCallback(URLRequest* request, bool result); 64 65 std::map<std::string, HTTPSProberDelegate*> inflight_probes_; 66 std::set<std::string> probed_; 67 68 friend struct DefaultSingletonTraits<HTTPSProber>; 69 DISALLOW_EVIL_CONSTRUCTORS(HTTPSProber); 70 }; 71 72 } // namespace net 73 #endif 74
