| // 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/internal.h> |
| #include <lib/fidl/llcpp/coding.h> |
| #include <lib/fidl/llcpp/message.h> |
| #include <lib/fidl/trace.h> |
| #include <zircon/assert.h> |
| |
| #include <cstring> |
| #include <string> |
| |
| #ifdef __Fuchsia__ |
| #include <lib/fidl/llcpp/client_base.h> |
| #include <lib/fidl/llcpp/server.h> |
| #include <zircon/syscalls.h> |
| #endif // __Fuchsia__ |
| |
| namespace fidl { |
| |
| OutgoingMessage OutgoingMessage::FromEncodedCMessage(const fidl_outgoing_msg_t* c_msg) { |
| return OutgoingMessage(c_msg); |
| } |
| |
| OutgoingMessage::OutgoingMessage(const fidl_outgoing_msg_t* c_msg) |
| : fidl::Result(fidl::Result::Ok()) { |
| ZX_ASSERT(c_msg); |
| switch (c_msg->type) { |
| case FIDL_OUTGOING_MSG_TYPE_IOVEC: { |
| message_ = *c_msg; |
| iovec_capacity_ = c_msg->iovec.num_iovecs; |
| handle_capacity_ = c_msg->iovec.num_handles; |
| break; |
| } |
| case FIDL_OUTGOING_MSG_TYPE_BYTE: { |
| backing_buffer_ = reinterpret_cast<uint8_t*>(c_msg->byte.bytes); |
| backing_buffer_capacity_ = c_msg->byte.num_bytes; |
| converted_byte_message_iovec_ = { |
| .buffer = backing_buffer_, |
| .capacity = backing_buffer_capacity_, |
| .reserved = 0, |
| }; |
| message_ = { |
| .type = FIDL_OUTGOING_MSG_TYPE_IOVEC, |
| .iovec = |
| { |
| .iovecs = &converted_byte_message_iovec_, |
| .num_iovecs = 1, |
| .handles = c_msg->byte.handles, |
| .num_handles = c_msg->byte.num_handles, |
| }, |
| }; |
| iovec_capacity_ = 1; |
| handle_capacity_ = c_msg->byte.num_handles; |
| break; |
| } |
| default: |
| ZX_PANIC("unhandled FIDL outgoing message type"); |
| } |
| } |
| |
| OutgoingMessage::OutgoingMessage(ConstructorArgs args) |
| : fidl::Result(fidl::Result::Ok()), |
| message_({ |
| .type = FIDL_OUTGOING_MSG_TYPE_IOVEC, |
| .iovec = |
| {.iovecs = args.iovecs, .num_iovecs = 0, .handles = args.handles, .num_handles = 0}, |
| }), |
| iovec_capacity_(args.iovec_capacity), |
| handle_capacity_(args.handle_capacity), |
| backing_buffer_capacity_(args.backing_buffer_capacity), |
| backing_buffer_(args.backing_buffer) {} |
| |
| OutgoingMessage::~OutgoingMessage() { |
| #ifdef __Fuchsia__ |
| if (handle_actual() > 0) { |
| FidlHandleDispositionCloseMany(handles(), handle_actual()); |
| } |
| #else |
| ZX_ASSERT(handle_actual() == 0); |
| #endif |
| } |
| |
| bool OutgoingMessage::BytesMatch(const OutgoingMessage& other) const { |
| uint32_t iovec_index = 0, other_iovec_index = 0; |
| uint32_t byte_index = 0, other_byte_index = 0; |
| while (iovec_index < iovec_actual() && other_iovec_index < other.iovec_actual()) { |
| zx_channel_iovec_t cur_iovec = iovecs()[iovec_index]; |
| zx_channel_iovec_t other_cur_iovec = other.iovecs()[other_iovec_index]; |
| const uint8_t* cur_bytes = reinterpret_cast<const uint8_t*>(cur_iovec.buffer); |
| const uint8_t* other_cur_bytes = reinterpret_cast<const uint8_t*>(other_cur_iovec.buffer); |
| |
| uint32_t cmp_len = |
| std::min(cur_iovec.capacity - byte_index, other_cur_iovec.capacity - other_byte_index); |
| if (memcmp(&cur_bytes[byte_index], &other_cur_bytes[other_byte_index], cmp_len) != 0) { |
| return false; |
| } |
| |
| byte_index += cmp_len; |
| if (byte_index == cur_iovec.capacity) { |
| iovec_index++; |
| byte_index = 0; |
| } |
| other_byte_index += cmp_len; |
| if (other_byte_index == other_cur_iovec.capacity) { |
| other_iovec_index++; |
| other_byte_index = 0; |
| } |
| } |
| return iovec_index == iovec_actual() && other_iovec_index == other.iovec_actual() && |
| byte_index == 0 && other_byte_index == 0; |
| } |
| |
| void OutgoingMessage::EncodeImpl(const fidl_type_t* message_type, void* data) { |
| if (!ok()) { |
| return; |
| } |
| uint32_t num_iovecs_actual; |
| uint32_t num_handles_actual; |
| zx_status_t status; |
| status = |
| fidl::internal::EncodeIovecEtc(message_type, data, iovecs(), iovec_capacity(), handles(), |
| handle_capacity(), backing_buffer(), backing_buffer_capacity(), |
| &num_iovecs_actual, &num_handles_actual, error_address()); |
| if (status != ZX_OK) { |
| SetResult(fidl::Result::EncodeError(status, *error_address())); |
| return; |
| } |
| iovec_message().num_iovecs = num_iovecs_actual; |
| iovec_message().num_handles = num_handles_actual; |
| } |
| |
| #ifdef __Fuchsia__ |
| void OutgoingMessage::WriteImpl(zx_handle_t channel) { |
| if (!ok()) { |
| return; |
| } |
| zx_status_t status = zx_channel_write_etc(channel, ZX_CHANNEL_WRITE_USE_IOVEC, iovecs(), |
| iovec_actual(), handles(), handle_actual()); |
| ReleaseHandles(); |
| if (status != ZX_OK) { |
| SetResult(fidl::Result::TransportError(status)); |
| } |
| } |
| |
| fidl::Result OutgoingMessage::Write(::fidl::internal::ClientBase* client, |
| ::fidl::internal::ResponseContext* context) { |
| if (auto channel = client->GetChannel()) { |
| Write(channel->handle()); |
| if (!ok()) { |
| client->ForgetAsyncTxn(context); |
| context->OnError(fidl::Result(*this)); |
| } |
| } else { |
| SetResult(fidl::Result::Unbound()); |
| client->ForgetAsyncTxn(context); |
| context->OnCanceled(); |
| } |
| return fidl::Result(*this); |
| } |
| |
| void OutgoingMessage::CallImpl(const fidl_type_t* response_type, zx_handle_t channel, |
| uint8_t* result_bytes, uint32_t result_capacity, |
| zx_time_t deadline) { |
| if (status() != ZX_OK) { |
| return; |
| } |
| zx_handle_info_t result_handles[ZX_CHANNEL_MAX_MSG_HANDLES]; |
| uint32_t actual_num_bytes = 0u; |
| uint32_t actual_num_handles = 0u; |
| zx_channel_call_etc_args_t args = {.wr_bytes = iovecs(), |
| .wr_handles = handles(), |
| .rd_bytes = result_bytes, |
| .rd_handles = result_handles, |
| .wr_num_bytes = iovec_actual(), |
| .wr_num_handles = handle_actual(), |
| .rd_num_bytes = result_capacity, |
| .rd_num_handles = ZX_CHANNEL_MAX_MSG_HANDLES}; |
| |
| zx_status_t status; |
| status = zx_channel_call_etc(channel, ZX_CHANNEL_WRITE_USE_IOVEC, deadline, &args, |
| &actual_num_bytes, &actual_num_handles); |
| ReleaseHandles(); |
| if (status != ZX_OK) { |
| SetResult(fidl::Result::TransportError(status)); |
| return; |
| } |
| status = fidl_decode_etc(response_type, result_bytes, actual_num_bytes, result_handles, |
| actual_num_handles, error_address()); |
| if (status != ZX_OK) { |
| SetResult(fidl::Result::DecodeError(status, *error_address())); |
| return; |
| } |
| } |
| #endif |
| |
| OutgoingMessage::CopiedBytes::CopiedBytes(const OutgoingMessage& msg) { |
| uint32_t byte_count = 0; |
| for (uint32_t i = 0; i < msg.iovec_actual(); ++i) { |
| byte_count += msg.iovecs()[i].capacity; |
| } |
| bytes_.reserve(byte_count); |
| for (uint32_t i = 0; i < msg.iovec_actual(); ++i) { |
| zx_channel_iovec_t iovec = msg.iovecs()[i]; |
| const uint8_t* buf_bytes = reinterpret_cast<const uint8_t*>(iovec.buffer); |
| bytes_.insert(bytes_.end(), buf_bytes, buf_bytes + iovec.capacity); |
| } |
| } |
| |
| IncomingMessage::IncomingMessage(uint8_t* bytes, uint32_t byte_actual, zx_handle_info_t* handles, |
| uint32_t handle_actual) |
| : IncomingMessage(bytes, byte_actual, handles, handle_actual, kSkipMessageHeaderValidation) { |
| Validate(); |
| } |
| |
| IncomingMessage IncomingMessage::FromEncodedCMessage(const fidl_incoming_msg_t* c_msg) { |
| return IncomingMessage(reinterpret_cast<uint8_t*>(c_msg->bytes), c_msg->num_bytes, c_msg->handles, |
| c_msg->num_handles); |
| } |
| |
| IncomingMessage::IncomingMessage(uint8_t* bytes, uint32_t byte_actual, zx_handle_info_t* handles, |
| uint32_t handle_actual, SkipMessageHeaderValidationTag) |
| : fidl::Result(fidl::Result::Ok()), |
| message_{ |
| .bytes = bytes, |
| .handles = handles, |
| .num_bytes = byte_actual, |
| .num_handles = handle_actual, |
| } {} |
| |
| IncomingMessage::IncomingMessage(const fidl::Result& failure) : fidl::Result(failure), message_{} { |
| ZX_DEBUG_ASSERT(failure.status() != ZX_OK); |
| } |
| |
| IncomingMessage::~IncomingMessage() { std::move(*this).CloseHandles(); } |
| |
| fidl_incoming_msg_t IncomingMessage::ReleaseToEncodedCMessage() && { |
| ZX_DEBUG_ASSERT(status() == ZX_OK); |
| fidl_incoming_msg_t result = message_; |
| ReleaseHandles(); |
| return result; |
| } |
| |
| void IncomingMessage::CloseHandles() && { |
| #ifdef __Fuchsia__ |
| if (handle_actual() > 0) { |
| FidlHandleInfoCloseMany(handles(), handle_actual()); |
| } |
| #else |
| ZX_ASSERT(handle_actual() == 0); |
| #endif |
| ReleaseHandles(); |
| } |
| |
| void IncomingMessage::Decode(const fidl_type_t* message_type) { |
| ZX_DEBUG_ASSERT(status() == ZX_OK); |
| fidl_trace(WillLLCPPDecode, message_type, bytes(), byte_actual(), handle_actual()); |
| zx_status_t status = fidl_decode_msg(message_type, &message_, error_address()); |
| fidl_trace(DidLLCPPDecode); |
| // Now the caller is responsible for the handles contained in `bytes()`. |
| ReleaseHandles(); |
| if (status != ZX_OK) { |
| SetResult(fidl::Result::DecodeError(status, *error_address())); |
| } |
| } |
| |
| void IncomingMessage::Validate() { |
| if (byte_actual() < sizeof(fidl_message_header_t)) { |
| return SetResult(fidl::Result::UnexpectedMessage(ZX_ERR_INVALID_ARGS, |
| ::fidl::internal::kErrorInvalidHeader)); |
| } |
| |
| auto* hdr = header(); |
| zx_status_t status = fidl_validate_txn_header(hdr); |
| if (status != ZX_OK) { |
| return SetResult( |
| fidl::Result::UnexpectedMessage(status, ::fidl::internal::kErrorInvalidHeader)); |
| } |
| |
| // See https://fuchsia.dev/fuchsia-src/contribute/governance/rfcs/0053_epitaphs?hl=en#wire_format |
| if (unlikely(maybe_epitaph())) { |
| if (hdr->txid != 0) { |
| return SetResult(fidl::Result::UnexpectedMessage(ZX_ERR_INVALID_ARGS, |
| ::fidl::internal::kErrorInvalidHeader)); |
| } |
| } |
| } |
| |
| #ifdef __Fuchsia__ |
| |
| IncomingMessage ChannelReadEtc(zx_handle_t channel, uint32_t options, |
| fidl::BufferSpan bytes_storage, |
| cpp20::span<zx_handle_info_t> handles_storage) { |
| uint32_t num_bytes, num_handles; |
| zx_status_t status = |
| zx_channel_read_etc(channel, options, bytes_storage.data, handles_storage.data(), |
| bytes_storage.capacity, handles_storage.size(), &num_bytes, &num_handles); |
| if (status != ZX_OK) { |
| return IncomingMessage(fidl::Result::TransportError(status)); |
| } |
| return IncomingMessage(bytes_storage.data, num_bytes, handles_storage.data(), num_handles); |
| } |
| |
| #endif // __Fuchsia__ |
| |
| OutgoingToIncomingMessage::OutgoingToIncomingMessage(OutgoingMessage& input) |
| : incoming_message_(ConversionImpl(input, buf_bytes_, buf_handles_)) {} |
| |
| [[nodiscard]] std::string OutgoingToIncomingMessage::FormatDescription() const { |
| return incoming_message_.FormatDescription(); |
| } |
| |
| IncomingMessage OutgoingToIncomingMessage::ConversionImpl( |
| OutgoingMessage& input, OutgoingMessage::CopiedBytes& buf_bytes, |
| std::unique_ptr<zx_handle_info_t[]>& buf_handles) { |
| zx_handle_disposition_t* handles = input.handles(); |
| uint32_t num_handles = input.handle_actual(); |
| input.ReleaseHandles(); |
| |
| if (num_handles > ZX_CHANNEL_MAX_MSG_HANDLES) { |
| FidlHandleDispositionCloseMany(handles, num_handles); |
| return fidl::IncomingMessage(fidl::Result::EncodeError(ZX_ERR_OUT_OF_RANGE)); |
| } |
| |
| auto converted_handles = std::make_unique<zx_handle_info_t[]>(ZX_CHANNEL_MAX_MSG_HANDLES); |
| zx_status_t status = |
| FidlHandleDispositionsToHandleInfos(handles, converted_handles.get(), num_handles); |
| if (status != ZX_OK) { |
| return fidl::IncomingMessage(fidl::Result::EncodeError(status)); |
| } |
| buf_handles = std::move(converted_handles); |
| |
| buf_bytes = input.CopyBytes(); |
| if (buf_bytes.size() > ZX_CHANNEL_MAX_MSG_BYTES) { |
| FidlHandleDispositionCloseMany(handles, num_handles); |
| return fidl::IncomingMessage(fidl::Result::EncodeError(status)); |
| } |
| |
| return fidl::IncomingMessage(buf_bytes.data(), buf_bytes.size(), buf_handles.get(), num_handles, |
| fidl::IncomingMessage::kSkipMessageHeaderValidation); |
| } |
| |
| } // namespace fidl |
