components/openscreen_platform/message_port_tls_connection.cc

// Copyright 2021 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 "components/openscreen_platform/message_port_tls_connection.h"

#include "third_party/openscreen/src/platform/api/task_runner.h"
#include "third_party/openscreen/src/platform/base/error.h"

namespace openscreen_platform {

MessagePortTlsConnection::MessagePortTlsConnection(
    std::unique_ptr<cast_api_bindings::MessagePort> message_port,
    openscreen::TaskRunner* task_runner)
    : message_port_(std::move(message_port)), task_runner_(task_runner) {
  DCHECK(message_port_);
  DCHECK(task_runner_);

  message_port_->SetReceiver(this);
}

MessagePortTlsConnection::~MessagePortTlsConnection() = default;

// TlsConnection overrides.
void MessagePortTlsConnection::SetClient(TlsConnection::Client* client) {
  DCHECK(task_runner_->IsRunningOnTaskRunner());
  client_ = client;
}

bool MessagePortTlsConnection::Send(const void* data, size_t len) {
  return message_port_->PostMessage(
      base::StringPiece(static_cast<const char*>(data), len));
}

openscreen::IPEndpoint MessagePortTlsConnection::GetRemoteEndpoint() const {
  return openscreen::IPEndpoint{openscreen::IPAddress::kV6LoopbackAddress()};
}

bool MessagePortTlsConnection::OnMessage(
    base::StringPiece message,
    std::vector<std::unique_ptr<cast_api_bindings::MessagePort>> ports) {
  DCHECK(ports.empty());

  if (client_) {
    if (!task_runner_->IsRunningOnTaskRunner()) {
      task_runner_->PostTask([ptr = AsWeakPtr(), m = std::move(message)]() {
        if (ptr) {
          ptr->OnMessage(
              std::move(m),
              std::vector<std::unique_ptr<cast_api_bindings::MessagePort>>());
        }
      });

      return true;
    }

    client_->OnRead(this, std::vector<uint8_t>(message.begin(), message.end()));
  }

  return true;
}

void MessagePortTlsConnection::OnPipeError() {
  if (client_) {
    if (!task_runner_->IsRunningOnTaskRunner()) {
      task_runner_->PostTask([ptr = AsWeakPtr()]() {
        if (ptr) {
          ptr->OnPipeError();
        }
      });
      return;
    }

    client_->OnError(
        this, openscreen::Error(openscreen::Error::Code::kSocketFailure));
  }
}

}  // namespace openscreen_platform