// Copyright 2018 The Fuchsia 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 <lib/fidl/coding.h>
#include <lib/fidl/cpp/builder.h>
#include <lib/fidl/cpp/message.h>
#include <lib/fidl/internal.h>
#include <lib/fidl/transformer.h>
#include <string.h>

#ifdef __Fuchsia__
#include <zircon/errors.h>
#include <zircon/syscalls.h>
#endif

namespace fidl {

namespace {

constexpr uint32_t kMaxStackAllocSize = 256;

// This is analogous to ClampedMessageSize in traits.h, but does its work at
// runtime instead of at compile time and is only called on v1 wire format types
// in the sending direction.
uint32_t ClampedMessageSize(const FidlCodedStruct& type) {
  // convert these to u64 before summing
  auto primary = static_cast<uint64_t>(type.size);
  auto max_out_of_line = static_cast<uint64_t>(type.max_out_of_line);
  uint64_t total_size = primary + max_out_of_line;
  if (total_size > ZX_CHANNEL_MAX_MSG_BYTES) {
    return ZX_CHANNEL_MAX_MSG_BYTES;
  } else {
    return static_cast<uint32_t>(total_size);
  }
}

// RAII managed heap allocated storage for raw message bytes. Used to hold
// the temporary output of fidl_transform (see ValidateV1Bytes)
struct HeapAllocatedMessage {
  explicit HeapAllocatedMessage(uint32_t size) : data(static_cast<uint8_t*>(malloc(size))) {}
  ~HeapAllocatedMessage() { free(data); }

  uint8_t* data;
};

}  // namespace

const fidl_type_t* get_alt_type(const fidl_type_t* type) {
  switch (type->type_tag) {
    case kFidlTypePrimitive:
    case kFidlTypeEnum:
    case kFidlTypeBits:
    case kFidlTypeString:
    case kFidlTypeHandle:
      return type;
    case kFidlTypeStruct:
      return type->coded_struct.alt_type;
    case kFidlTypeUnion:
      return type->coded_union.alt_type;
    case kFidlTypeXUnion:
      return type->coded_xunion.alt_type;
    case kFidlTypeArray:
      return type->coded_array.alt_type;
    case kFidlTypeVector:
      return type->coded_vector.alt_type;
    default:
      assert(false && "cannot get alt type of a type that lacks an alt type");
      return type;
  }
}

zx_status_t FidlTransformWithCallback(
    fidl_transformation_t transformation, const fidl_type_t* type, const uint8_t* src_bytes,
    uint32_t src_num_bytes, const char** out_error_msg,
    const std::function<zx_status_t(const uint8_t* dst_bytes, uint32_t dst_num_bytes)>& callback) {
  if (type->type_tag != kFidlTypeStruct) {
    return ZX_ERR_INVALID_ARGS;
  }
  auto struct_type = type->coded_struct;
  if (!struct_type.contains_union) {
    return callback(src_bytes, src_num_bytes);
  }

  auto msg_size = ClampedMessageSize(get_alt_type(type)->coded_struct);
  uint32_t dst_num_bytes;
  if (msg_size <= kMaxStackAllocSize) {
    auto dst_bytes = static_cast<uint8_t*>(alloca(msg_size));
    zx_status_t status = fidl_transform(transformation, type, src_bytes, src_num_bytes, dst_bytes,
                                        msg_size, &dst_num_bytes, out_error_msg);
    if (status != ZX_OK) {
      return status;
    }
    return callback(dst_bytes, dst_num_bytes);
  } else {
    HeapAllocatedMessage dst_bytes(msg_size);
    zx_status_t status = fidl_transform(transformation, type, src_bytes, src_num_bytes,
                                        dst_bytes.data, msg_size, &dst_num_bytes, out_error_msg);
    if (status != ZX_OK) {
      return status;
    }
    return callback(dst_bytes.data, dst_num_bytes);
  }
}

Message::Message() = default;

Message::Message(BytePart bytes, HandlePart handles)
    : bytes_(static_cast<BytePart&&>(bytes)), handles_(static_cast<HandlePart&&>(handles)) {}

Message::~Message() {
#ifdef __Fuchsia__
  if (handles_.actual() > 0) {
    zx_handle_close_many(handles_.data(), handles_.actual());
  }
#endif
  ClearHandlesUnsafe();
}

Message::Message(Message&& other)
    : bytes_(static_cast<BytePart&&>(other.bytes_)),
      handles_(static_cast<HandlePart&&>(other.handles_)) {}

Message& Message::operator=(Message&& other) {
  bytes_ = static_cast<BytePart&&>(other.bytes_);
  handles_ = static_cast<HandlePart&&>(other.handles_);
  return *this;
}

zx_status_t Message::Encode(const fidl_type_t* type, const char** error_msg_out) {
  uint32_t actual_handles = 0u;
  zx_status_t status = fidl_encode(type, bytes_.data(), bytes_.actual(), handles_.data(),
                                   handles_.capacity(), &actual_handles, error_msg_out);
  if (status == ZX_OK)
    handles_.set_actual(actual_handles);

  return status;
}

zx_status_t Message::Decode(const fidl_type_t* type, const char** error_msg_out) {
  zx_status_t status = fidl_decode(type, bytes_.data(), bytes_.actual(), handles_.data(),
                                   handles_.actual(), error_msg_out);
  ClearHandlesUnsafe();
  return status;
}

zx_status_t Message::Validate(const fidl_type_t* v1_type, const char** error_msg_out) const {
  return fidl_validate(v1_type, bytes_.data(), bytes_.actual(), handles_.actual(), error_msg_out);
}

#ifdef __Fuchsia__
zx_status_t Message::Read(zx_handle_t channel, uint32_t flags) {
  uint32_t actual_bytes = 0u;
  uint32_t actual_handles = 0u;
  zx_status_t status =
      zx_channel_read(channel, flags, bytes_.data(), handles_.data(), bytes_.capacity(),
                      handles_.capacity(), &actual_bytes, &actual_handles);
  if (status == ZX_OK) {
    bytes_.set_actual(actual_bytes);
    handles_.set_actual(actual_handles);
  }
  if (actual_bytes < sizeof(fidl_message_header_t)) {
    // When reading a message, the size should always greater than the header size.
    return ZX_ERR_INVALID_ARGS;
  }
  return status;
}

zx_status_t Message::Write(zx_handle_t channel, uint32_t flags) {
  zx_status_t status = zx_channel_write(channel, flags, bytes_.data(), bytes_.actual(),
                                        handles_.data(), handles_.actual());
  ClearHandlesUnsafe();
  return status;
}

zx_status_t Message::WriteTransformV1(zx_handle_t channel, uint32_t flags,
                                      const fidl_type_t* old_type) {
  auto src_data = bytes().data();
  auto src_num_bytes = bytes().actual();
  auto callback = [&](const uint8_t* dst_bytes, uint32_t dst_num_bytes) -> zx_status_t {
    zx_status_t status = zx_channel_write(channel, flags, dst_bytes, dst_num_bytes, handles_.data(),
                                          handles_.actual());
    ClearHandlesUnsafe();
    return status;
  };
  return FidlTransformWithCallback(FIDL_TRANSFORMATION_OLD_TO_V1, old_type, src_data, src_num_bytes,
                                   nullptr, callback);
}

zx_status_t Message::Call(zx_handle_t channel, uint32_t flags, zx_time_t deadline,
                          Message* response) {
  zx_channel_call_args_t args;
  args.wr_bytes = bytes_.data();
  args.wr_handles = handles_.data();
  args.rd_bytes = response->bytes_.data();
  args.rd_handles = response->handles_.data();
  args.wr_num_bytes = bytes_.actual();
  args.wr_num_handles = handles_.actual();
  args.rd_num_bytes = response->bytes_.capacity();
  args.rd_num_handles = response->handles_.capacity();
  uint32_t actual_bytes = 0u;
  uint32_t actual_handles = 0u;
  zx_status_t status =
      zx_channel_call(channel, flags, deadline, &args, &actual_bytes, &actual_handles);
  ClearHandlesUnsafe();
  if (status == ZX_OK) {
    response->bytes_.set_actual(actual_bytes);
    response->handles_.set_actual(actual_handles);
  }
  return status;
}
#endif

void Message::ClearHandlesUnsafe() { handles_.set_actual(0u); }

}  // namespace fidl