chrome/browser/ssl/ssl_client_auth_requestor_mock.cc

// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/ssl/ssl_client_auth_requestor_mock.h"

#include <utility>

#include "base/functional/callback.h"
#include "content/public/browser/client_certificate_delegate.h"
#include "net/cert/x509_certificate.h"
#include "net/ssl/ssl_cert_request_info.h"
#include "net/ssl/ssl_private_key.h"

namespace {

class FakeClientCertificateDelegate
    : public content::ClientCertificateDelegate {
 public:
  FakeClientCertificateDelegate(SSLClientAuthRequestorMock* requestor,
                                base::OnceClosure done_callback)
      : requestor_(requestor), done_callback_(std::move(done_callback)) {}

  FakeClientCertificateDelegate(const FakeClientCertificateDelegate&) = delete;
  FakeClientCertificateDelegate& operator=(
      const FakeClientCertificateDelegate&) = delete;

  ~FakeClientCertificateDelegate() override {
    if (requestor_) {
      requestor_->CancelCertificateSelection();
      std::move(done_callback_).Run();
    }
  }

  // content::ClientCertificateDelegate implementation:
  void ContinueWithCertificate(scoped_refptr<net::X509Certificate> cert,
                               scoped_refptr<net::SSLPrivateKey> key) override {
    requestor_->CertificateSelected(cert.get(), key.get());
    requestor_ = nullptr;
    std::move(done_callback_).Run();
  }

 private:
  scoped_refptr<SSLClientAuthRequestorMock> requestor_;
  base::OnceClosure done_callback_;
};

}  // namespace

SSLClientAuthRequestorMock::SSLClientAuthRequestorMock(
    const scoped_refptr<net::SSLCertRequestInfo>& cert_request_info)
    : cert_request_info_(cert_request_info) {
}

SSLClientAuthRequestorMock::~SSLClientAuthRequestorMock() {}

std::unique_ptr<content::ClientCertificateDelegate>
SSLClientAuthRequestorMock::CreateDelegate() {
  return std::make_unique<FakeClientCertificateDelegate>(
      this, run_loop_.QuitClosure());
}

void SSLClientAuthRequestorMock::WaitForCompletion() {
  run_loop_.Run();
}