| 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_CERT_MOCK_CERT_VERIFIER_H_ | 5 #ifndef NET_CERT_MOCK_CERT_VERIFIER_H_ |
| 6 #define NET_CERT_MOCK_CERT_VERIFIER_H_ | 6 #define NET_CERT_MOCK_CERT_VERIFIER_H_ |
| 7 | 7 |
| 8 #include <list> | 8 #include <list> |
| 9 #include <memory> | 9 #include <memory> |
| 10 | 10 |
| 11 #include "net/cert/cert_verifier.h" | 11 #include "net/cert/cert_verifier.h" |
| 12 #include "net/cert/cert_verify_result.h" | 12 #include "net/cert/cert_verify_result.h" |
| 13 | 13 |
| 14 namespace net { | 14 namespace net { |
| 15 | 15 |
| 16 class MockCertVerifier : public CertVerifier { | 16 class MockCertVerifier : public CertVerifier { |
| 17 public: | 17 public: |
| 18 // Creates a new MockCertVerifier. By default, any call to Verify() will | 18 // Creates a new MockCertVerifier. By default, any call to Verify() will |
| 19 // result in the cert status being flagged as CERT_STATUS_INVALID and return | 19 // result in the cert status being flagged as CERT_STATUS_INVALID and return |
| 20 // an ERR_CERT_INVALID network error code. This behaviour can be overridden | 20 // an ERR_CERT_INVALID network error code. This behaviour can be overridden |
| 21 // by calling set_default_result() to change the default return value for | 21 // by calling set_default_result() to change the default return value for |
| 22 // Verify() or by calling one of the AddResult*() methods to specifically | 22 // Verify() or by calling one of the AddResult*() methods to specifically |
| 23 // handle a certificate or certificate and host. | 23 // handle a certificate or certificate and host. |
| 24 MockCertVerifier(); | 24 MockCertVerifier(); |
| 25 | 25 |
| 26 ~MockCertVerifier() override; | 26 ~MockCertVerifier() override; |
| 27 | 27 |
| 28 // CertVerifier implementation | 28 // CertVerifier implementation |
| 29 int Verify(X509Certificate* cert, | 29 int Verify(const CertVerifier::RequestParams& params, |
| 30 const std::string& hostname, | |
| 31 const std::string& ocsp_response, | |
| 32 int flags, | |
| 33 CRLSet* crl_set, | 30 CRLSet* crl_set, |
| 34 CertVerifyResult* verify_result, | 31 CertVerifyResult* verify_result, |
| 35 const CompletionCallback& callback, | 32 const CompletionCallback& callback, |
| 36 std::unique_ptr<Request>* out_req, | 33 std::unique_ptr<Request>* out_req, |
| 37 const BoundNetLog& net_log) override; | 34 const BoundNetLog& net_log) override; |
| 38 | 35 |
| 39 // Sets the default return value for Verify() for certificates/hosts that do | 36 // Sets the default return value for Verify() for certificates/hosts that do |
| 40 // not have explicit results added via the AddResult*() methods. | 37 // not have explicit results added via the AddResult*() methods. |
| 41 void set_default_result(int default_result) { | 38 void set_default_result(int default_result) { |
| 42 default_result_ = default_result; | 39 default_result_ = default_result; |
| 43 } | 40 } |
| 44 | 41 |
| 45 // Adds a rule that will cause any call to Verify() for |cert| to return rv, | 42 // Adds a rule that will cause any call to Verify() for |cert| to return rv, |
| 46 // copying |verify_result| into the verified result. | 43 // copying |verify_result| into the verified result. |
| 47 // Note: Only the primary certificate of |cert| is checked. Any intermediate | 44 // Note: Only the primary certificate of |cert| is checked. Any intermediate |
| 48 // certificates will be ignored. | 45 // certificates will be ignored. |
| 49 void AddResultForCert(X509Certificate* cert, | 46 void AddResultForCert(scoped_refptr<X509Certificate> cert, |
| 50 const CertVerifyResult& verify_result, | 47 const CertVerifyResult& verify_result, |
| 51 int rv); | 48 int rv); |
| 52 | 49 |
| 53 // Same as AddResultForCert(), but further restricts it to only return for | 50 // Same as AddResultForCert(), but further restricts it to only return for |
| 54 // hostnames that match |host_pattern|. | 51 // hostnames that match |host_pattern|. |
| 55 void AddResultForCertAndHost(X509Certificate* cert, | 52 void AddResultForCertAndHost(scoped_refptr<X509Certificate> cert, |
| 56 const std::string& host_pattern, | 53 const std::string& host_pattern, |
| 57 const CertVerifyResult& verify_result, | 54 const CertVerifyResult& verify_result, |
| 58 int rv); | 55 int rv); |
| 59 | 56 |
| 60 private: | 57 private: |
| 61 struct Rule; | 58 struct Rule; |
| 62 typedef std::list<Rule> RuleList; | 59 typedef std::list<Rule> RuleList; |
| 63 | 60 |
| 64 int default_result_; | 61 int default_result_; |
| 65 RuleList rules_; | 62 RuleList rules_; |
| 66 }; | 63 }; |
| 67 | 64 |
| 68 } // namespace net | 65 } // namespace net |
| 69 | 66 |
| 70 #endif // NET_CERT_MOCK_CERT_VERIFIER_H_ | 67 #endif // NET_CERT_MOCK_CERT_VERIFIER_H_ |
| OLD | NEW |