fuchsia/cast_streaming/test/cast_message_port_sender_impl.cc

// Copyright 2020 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 "fuchsia/cast_streaming/test/cast_message_port_sender_impl.h"

#include "base/json/json_reader.h"
#include "base/json/json_writer.h"
#include "base/logging.h"
#include "base/values.h"
#include "fuchsia/cast_streaming/message_serialization.h"
#include "third_party/openscreen/src/platform/base/error.h"

namespace cast_streaming {

CastMessagePortSenderImpl::CastMessagePortSenderImpl(
    std::unique_ptr<cast_api_bindings::MessagePort> message_port,
    base::OnceClosure on_close)
    : message_port_(std::move(message_port)), on_close_(std::move(on_close)) {
  VLOG(1) << __func__;
  message_port_->SetReceiver(this);
}

CastMessagePortSenderImpl::~CastMessagePortSenderImpl() = default;

void CastMessagePortSenderImpl::MaybeClose() {
  if (message_port_) {
    message_port_.reset();
  }
  if (client_) {
    client_->OnError(
        openscreen::Error(openscreen::Error::Code::kCastV2CastSocketError));
    if (on_close_) {
      std::move(on_close_).Run();
    }
  }
}

void CastMessagePortSenderImpl::SetClient(
    openscreen::cast::MessagePort::Client* client,
    std::string client_sender_id) {
  VLOG(2) << __func__;
  CHECK(client);
  CHECK(!client_);
  client_ = client;
}

void CastMessagePortSenderImpl::ResetClient() {
  client_ = nullptr;
  MaybeClose();
}

void CastMessagePortSenderImpl::PostMessage(
    const std::string& sender_id,
    const std::string& message_namespace,
    const std::string& message) {
  VLOG(3) << __func__;
  if (!message_port_) {
    return;
  }

  VLOG(3) << "Received Open Screen message. SenderId: " << sender_id
          << ". Namespace: " << message_namespace << ". Message: " << message;
  message_port_->PostMessage(
      SerializeCastMessage(sender_id, message_namespace, message));
}

bool CastMessagePortSenderImpl::OnMessage(
    base::StringPiece message,
    std::vector<std::unique_ptr<cast_api_bindings::MessagePort>> ports) {
  VLOG(3) << __func__;

  // If |client_| was cleared, |message_port_| should have been reset.
  CHECK(client_);

  if (!ports.empty()) {
    // We should never receive any ports for Cast Streaming.
    MaybeClose();
    return false;
  }

  std::string sender_id;
  std::string message_namespace;
  std::string str_message;
  if (!DeserializeCastMessage(message, &sender_id, &message_namespace,
                              &str_message)) {
    LOG(ERROR) << "Received bad message. " << message;
    client_->OnError(
        openscreen::Error(openscreen::Error::Code::kCastV2InvalidMessage));
    return false;
  }
  VLOG(3) << "Received Cast message. SenderId: " << sender_id
          << ". Namespace: " << message_namespace
          << ". Message: " << str_message;

  client_->OnMessage(sender_id, message_namespace, str_message);
  return true;
}

void CastMessagePortSenderImpl::OnPipeError() {
  VLOG(3) << __func__;
  MaybeClose();
}

}  // namespace cast_streaming