blob: 125f59e6a95d35c40b6bbf7e5581ab17b4860da7 [file] [log] [blame]
// Copyright 2024 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.
#define PW_LOG_MODULE_NAME "TRN"
#define PW_LOG_LEVEL PW_TRANSFER_CONFIG_LOG_LEVEL
#include "pw_transfer/client.h"
#include "pw_log/log.h"
namespace pw::transfer {
Result<Client::Handle> Client::Read(
uint32_t resource_id,
stream::Writer& output,
CompletionFunc&& on_completion,
ProtocolVersion protocol_version,
chrono::SystemClock::duration timeout,
chrono::SystemClock::duration initial_chunk_timeout,
uint32_t initial_offset) {
if (on_completion == nullptr ||
protocol_version == ProtocolVersion::kUnknown) {
return Status::InvalidArgument();
}
if (protocol_version < ProtocolVersion::kVersionTwo && initial_offset != 0) {
return Status::InvalidArgument();
}
if (!has_read_stream_) {
rpc::RawClientReaderWriter read_stream =
client_.Read(nullptr, // on_next will be set by the transfer_thread.
[this](Status status) {
OnRpcError(status, internal::TransferType::kReceive);
});
transfer_thread_.SetClientReadStream(
read_stream, [this](ConstByteSpan chunk) {
transfer_thread_.ProcessClientChunk(chunk);
});
has_read_stream_ = true;
}
Handle handle = AssignHandle();
transfer_thread_.StartClientTransfer(internal::TransferType::kReceive,
protocol_version,
resource_id,
handle.id(),
&output,
max_parameters_,
std::move(on_completion),
timeout,
initial_chunk_timeout,
max_retries_,
max_lifetime_retries_,
initial_offset);
return handle;
}
Result<Client::Handle> Client::Write(
uint32_t resource_id,
stream::Reader& input,
CompletionFunc&& on_completion,
ProtocolVersion protocol_version,
chrono::SystemClock::duration timeout,
chrono::SystemClock::duration initial_chunk_timeout,
uint32_t initial_offset) {
if (on_completion == nullptr ||
protocol_version == ProtocolVersion::kUnknown) {
return Status::InvalidArgument();
}
if (protocol_version < ProtocolVersion::kVersionTwo && initial_offset != 0) {
return Status::InvalidArgument();
}
if (!has_write_stream_) {
rpc::RawClientReaderWriter write_stream =
client_.Write(nullptr, // on_next will be set by the transfer thread.
[this](Status status) {
OnRpcError(status, internal::TransferType::kTransmit);
});
transfer_thread_.SetClientWriteStream(
write_stream, [this](ConstByteSpan chunk) {
transfer_thread_.ProcessClientChunk(chunk);
});
has_write_stream_ = true;
}
Handle handle = AssignHandle();
transfer_thread_.StartClientTransfer(internal::TransferType::kTransmit,
protocol_version,
resource_id,
handle.id(),
&input,
max_parameters_,
std::move(on_completion),
timeout,
initial_chunk_timeout,
max_retries_,
max_lifetime_retries_,
initial_offset);
return handle;
}
Client::Handle Client::AssignHandle() {
uint32_t handle_id = next_handle_id_++;
if (handle_id == Handle::kUnassignedHandleId) {
handle_id = next_handle_id_++;
}
return Handle(this, handle_id);
}
void Client::OnRpcError(Status status, internal::TransferType type) {
bool is_write_error = type == internal::TransferType::kTransmit;
PW_LOG_ERROR("Client %s stream terminated with status %d",
is_write_error ? "Write()" : "Read()",
status.code());
if (is_write_error) {
has_write_stream_ = false;
} else {
has_read_stream_ = false;
}
}
} // namespace pw::transfer