15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#ifndef NET_CERT_MOCK_CERT_VERIFIER_H_ 6c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#define NET_CERT_MOCK_CERT_VERIFIER_H_ 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <list> 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "net/cert/cert_verifier.h" 11c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "net/cert/cert_verify_result.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace net { 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class MockCertVerifier : public CertVerifier { 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Creates a new MockCertVerifier. By default, any call to Verify() will 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // result in the cert status being flagged as CERT_STATUS_INVALID and return 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // an ERR_CERT_INVALID network error code. This behaviour can be overridden 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // by calling set_default_result() to change the default return value for 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Verify() or by calling one of the AddResult*() methods to specifically 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // handle a certificate or certificate and host. 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockCertVerifier(); 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~MockCertVerifier(); 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // CertVerifier implementation 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int Verify(X509Certificate* cert, 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& hostname, 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int flags, 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CRLSet* crl_set, 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CertVerifyResult* verify_result, 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback, 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RequestHandle* out_req, 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const BoundNetLog& net_log) OVERRIDE; 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void CancelRequest(RequestHandle req) OVERRIDE; 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Sets the default return value for Verify() for certificates/hosts that do 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // not have explicit results added via the AddResult*() methods. 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void set_default_result(int default_result) { 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) default_result_ = default_result; 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Adds a rule that will cause any call to Verify() for |cert| to return rv, 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // copying |verify_result| into the verified result. 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Note: Only the primary certificate of |cert| is checked. Any intermediate 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // certificates will be ignored. 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void AddResultForCert(X509Certificate* cert, 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CertVerifyResult& verify_result, 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int rv); 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Same as AddResultForCert(), but further restricts it to only return for 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // hostnames that match |host_pattern|. 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void AddResultForCertAndHost(X509Certificate* cert, 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& host_pattern, 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CertVerifyResult& verify_result, 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int rv); 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) struct Rule; 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) typedef std::list<Rule> RuleList; 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int default_result_; 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RuleList rules_; 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace net 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 69c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#endif // NET_CERT_MOCK_CERT_VERIFIER_H_ 70