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)
55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ppapi/tests/test_tcp_socket_private_trusted.h"
65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ppapi/cpp/private/tcp_socket_private.h"
85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ppapi/cpp/private/x509_certificate_private.h"
95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ppapi/tests/testing_instance.h"
105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ppapi/tests/test_utils.h"
115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)REGISTER_TEST_CASE(TCPSocketPrivateTrusted);
135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TestTCPSocketPrivateTrusted::TestTCPSocketPrivateTrusted(
155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    TestingInstance* instance)
165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    : TestCase(instance) {
175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool TestTCPSocketPrivateTrusted::Init() {
205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!pp::TCPSocketPrivate::IsAvailable())
215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return false;
225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // We need something to connect to, so we connect to the HTTP server whence we
245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // came. Grab the host and port.
255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!EnsureRunningOverHTTP())
265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return false;
275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!GetLocalHostPort(instance_->pp_instance(), &host_, &port_))
295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return false;
305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Get the port for the SSL server.
325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  ssl_port_ = instance_->ssl_server_port();
335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return true;
355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void TestTCPSocketPrivateTrusted::RunTests(const std::string& filter) {
38c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)  RUN_CALLBACK_TEST(TestTCPSocketPrivateTrusted, GetServerCertificate, filter);
395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)std::string TestTCPSocketPrivateTrusted::TestGetServerCertificate() {
425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  pp::TCPSocketPrivate socket(instance_);
43c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)  TestCompletionCallback cb(instance_->pp_instance(), callback_type());
445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
45c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)  cb.WaitForResult(
46c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)      socket.Connect(host_.c_str(), ssl_port_, cb.GetCallback()));
47c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)  CHECK_CALLBACK_BEHAVIOR(cb);
48c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)  ASSERT_EQ(PP_OK, cb.result());
495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
50c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)  cb.WaitForResult(
51c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)      socket.SSLHandshake(host_.c_str(), ssl_port_, cb.GetCallback()));
52c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)  CHECK_CALLBACK_BEHAVIOR(cb);
53c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)  ASSERT_EQ(PP_OK, cb.result());
545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  const pp::X509CertificatePrivate& cert = socket.GetServerCertificate();
565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  ASSERT_EQ(
575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      cert.GetField(PP_X509CERTIFICATE_PRIVATE_ISSUER_COMMON_NAME).AsString(),
58eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch      "Test Root CA");
595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  ASSERT_EQ(
605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      cert.GetField(PP_X509CERTIFICATE_PRIVATE_SUBJECT_COMMON_NAME).AsString(),
615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      "127.0.0.1");
625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  PASS();
635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
64