blob: d672373b3cf2f2bcfe5d3a4ac9287cbb7cb1b0d3 [file] [log] [blame]
// Copyright 2020 The Pigweed Authors
//
// Licensed under the Apache License, Version 2.0 (the "License"); you may not
// use this file except in compliance with the License. You may obtain a copy of
// the License at
//
// https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
// License for the specific language governing permissions and limitations under
// the License.
// clang-format off
#include "pw_rpc/internal/log_config.h" // PW_LOG_* macros must be first.
#include "pw_rpc/internal/channel.h"
// clang-format on
#include "pw_bytes/span.h"
#include "pw_log/log.h"
#include "pw_protobuf/decoder.h"
#include "pw_rpc/internal/config.h"
namespace pw::rpc {
namespace {
// TODO(b/234876617): Dynamically allocate this buffer if
// PW_RPC_DYNAMIC_ALLOCATION is enabled.
std::array<std::byte, cfg::kEncodingBufferSizeBytes> encoding_buffer
PW_GUARDED_BY(internal::rpc_lock());
static_assert(MaxSafePayloadSize() > 0,
"pw_rpc's encode buffer is too small to fit any data");
} // namespace
Result<uint32_t> ExtractChannelId(ConstByteSpan packet) {
protobuf::Decoder decoder(packet);
while (decoder.Next().ok()) {
if (static_cast<internal::RpcPacket::Fields>(decoder.FieldNumber()) !=
internal::RpcPacket::Fields::CHANNEL_ID) {
continue;
}
uint32_t channel_id;
PW_TRY(decoder.ReadUint32(&channel_id));
return channel_id;
}
return Status::DataLoss();
}
namespace internal {
ByteSpan GetPayloadBuffer() PW_EXCLUSIVE_LOCKS_REQUIRED(rpc_lock()) {
return ByteSpan(encoding_buffer)
.subspan(Packet::kMinEncodedSizeWithoutPayload);
}
Status Channel::Send(const Packet& packet) {
Result encoded = packet.Encode(encoding_buffer);
if (!encoded.ok()) {
PW_LOG_ERROR(
"Failed to encode RPC packet type %u to channel %u buffer, status %u",
static_cast<unsigned>(packet.type()),
static_cast<unsigned>(id()),
encoded.status().code());
return Status::Internal();
}
Status sent = output().Send(encoded.value());
if (!sent.ok()) {
PW_LOG_DEBUG("Channel %u failed to send packet with status %u",
static_cast<unsigned>(id()),
sent.code());
return Status::Unknown();
}
return OkStatus();
}
} // namespace internal
} // namespace pw::rpc