net/cert/mock_cert_verifier.h

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