| // Copyright 2014 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "android_webview/browser/aw_contents_client_bridge.h" |
| |
| #include <memory> |
| |
| #include "android_webview/test/android_webview_unittests_jni/MockAwContentsClientBridge_jni.h" |
| #include "base/android/jni_android.h" |
| #include "base/android/jni_array.h" |
| #include "base/android/scoped_java_ref.h" |
| #include "base/bind.h" |
| #include "base/macros.h" |
| #include "base/memory/ptr_util.h" |
| #include "base/run_loop.h" |
| #include "content/public/browser/client_certificate_delegate.h" |
| #include "content/public/test/browser_task_environment.h" |
| #include "net/cert/x509_certificate.h" |
| #include "net/ssl/ssl_cert_request_info.h" |
| #include "net/ssl/ssl_private_key.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| using base::android::AttachCurrentThread; |
| using base::android::ScopedJavaLocalRef; |
| using net::SSLCertRequestInfo; |
| using net::SSLClientCertType; |
| using net::SSLPrivateKey; |
| using net::X509Certificate; |
| using testing::NotNull; |
| using testing::Test; |
| |
| namespace android_webview { |
| |
| namespace { |
| |
| // Tests the android_webview contents client bridge. |
| class AwContentsClientBridgeTest : public Test { |
| public: |
| AwContentsClientBridgeTest() {} |
| |
| // Callback method called when a cert is selected. |
| void CertSelected(scoped_refptr<X509Certificate> cert, |
| scoped_refptr<SSLPrivateKey> key); |
| |
| protected: |
| void SetUp() override; |
| void TestCertType(SSLClientCertType type, const std::string& expected_name); |
| // Create the TestBrowserThreads. Just instantiate the member variable. |
| content::BrowserTaskEnvironment task_environment_; |
| base::android::ScopedJavaGlobalRef<jobject> jbridge_; |
| std::unique_ptr<AwContentsClientBridge> bridge_; |
| scoped_refptr<SSLCertRequestInfo> cert_request_info_; |
| scoped_refptr<X509Certificate> selected_cert_; |
| scoped_refptr<SSLPrivateKey> selected_key_; |
| int cert_selected_callbacks_; |
| JNIEnv* env_; |
| }; |
| |
| class TestClientCertificateDelegate |
| : public content::ClientCertificateDelegate { |
| public: |
| explicit TestClientCertificateDelegate(AwContentsClientBridgeTest* test) |
| : test_(test) {} |
| |
| // content::ClientCertificateDelegate. |
| void ContinueWithCertificate(scoped_refptr<net::X509Certificate> cert, |
| scoped_refptr<net::SSLPrivateKey> key) override { |
| test_->CertSelected(std::move(cert), std::move(key)); |
| test_ = nullptr; |
| } |
| |
| private: |
| AwContentsClientBridgeTest* test_; |
| |
| DISALLOW_COPY_AND_ASSIGN(TestClientCertificateDelegate); |
| }; |
| |
| } // namespace |
| |
| void AwContentsClientBridgeTest::SetUp() { |
| env_ = AttachCurrentThread(); |
| ASSERT_THAT(env_, NotNull()); |
| jbridge_.Reset( |
| env_, |
| Java_MockAwContentsClientBridge_getAwContentsClientBridge(env_).obj()); |
| bridge_ = std::make_unique<AwContentsClientBridge>(env_, jbridge_); |
| selected_cert_ = nullptr; |
| cert_selected_callbacks_ = 0; |
| cert_request_info_ = base::MakeRefCounted<net::SSLCertRequestInfo>(); |
| } |
| |
| void AwContentsClientBridgeTest::CertSelected( |
| scoped_refptr<X509Certificate> cert, |
| scoped_refptr<SSLPrivateKey> key) { |
| selected_cert_ = std::move(cert); |
| selected_key_ = std::move(key); |
| cert_selected_callbacks_++; |
| } |
| |
| TEST_F(AwContentsClientBridgeTest, TestClientCertKeyTypesCorrectlyEncoded) { |
| SSLClientCertType cert_types[2] = {net::CLIENT_CERT_RSA_SIGN, |
| net::CLIENT_CERT_ECDSA_SIGN}; |
| std::string expected_names[2] = {"RSA", "ECDSA"}; |
| |
| for (int i = 0; i < 2; i++) { |
| TestCertType(cert_types[i], expected_names[i]); |
| } |
| } |
| |
| void AwContentsClientBridgeTest::TestCertType( |
| SSLClientCertType type, |
| const std::string& expected_name) { |
| cert_request_info_->cert_key_types.clear(); |
| cert_request_info_->cert_key_types.push_back(type); |
| bridge_->SelectClientCertificate( |
| cert_request_info_.get(), |
| std::make_unique<TestClientCertificateDelegate>(this)); |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_EQ(0, cert_selected_callbacks_); |
| ScopedJavaLocalRef<jobjectArray> key_types = |
| Java_MockAwContentsClientBridge_getKeyTypes(env_, jbridge_); |
| std::vector<std::string> vec; |
| base::android::AppendJavaStringArrayToStringVector(env_, key_types, &vec); |
| EXPECT_EQ(1u, vec.size()); |
| EXPECT_EQ(expected_name, vec[0]); |
| } |
| |
| // Verify that ProvideClientCertificateResponse works properly when the client |
| // responds with a null key. |
| TEST_F(AwContentsClientBridgeTest, |
| TestProvideClientCertificateResponseCallsCallbackOnNullKey) { |
| // Call SelectClientCertificate to create a callback id that mock java object |
| // can call on. |
| bridge_->SelectClientCertificate( |
| cert_request_info_.get(), |
| base::WrapUnique(new TestClientCertificateDelegate(this))); |
| bridge_->ProvideClientCertificateResponse( |
| env_, jbridge_, |
| Java_MockAwContentsClientBridge_getRequestId(env_, jbridge_), |
| Java_MockAwContentsClientBridge_createTestCertChain(env_, jbridge_), |
| nullptr); |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_EQ(nullptr, selected_cert_.get()); |
| EXPECT_EQ(nullptr, selected_key_.get()); |
| EXPECT_EQ(1, cert_selected_callbacks_); |
| } |
| |
| // Verify that ProvideClientCertificateResponse calls the callback with |
| // null parameters when private key is not provided. |
| TEST_F(AwContentsClientBridgeTest, |
| TestProvideClientCertificateResponseCallsCallbackOnNullChain) { |
| // Call SelectClientCertificate to create a callback id that mock java object |
| // can call on. |
| bridge_->SelectClientCertificate( |
| cert_request_info_.get(), |
| base::WrapUnique(new TestClientCertificateDelegate(this))); |
| int requestId = Java_MockAwContentsClientBridge_getRequestId(env_, jbridge_); |
| bridge_->ProvideClientCertificateResponse(env_, jbridge_, requestId, nullptr, |
| nullptr); |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_EQ(nullptr, selected_cert_.get()); |
| EXPECT_EQ(nullptr, selected_key_.get()); |
| EXPECT_EQ(1, cert_selected_callbacks_); |
| } |
| |
| } // namespace android_webview |