// Copyright 2026 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "net/socket/read_multiple_emulator.h"

#include <algorithm>

#include "base/check.h"
#include "base/functional/bind.h"
#include "base/numerics/safe_conversions.h"
#include "net/base/io_buffer.h"
#include "net/socket/datagram_client_socket.h"
#include "net/socket/diff_serv_code_point.h"

namespace net {

ReadMultipleEmulator::ReadMultipleEmulator(DatagramClientSocket* socket)
    : socket_(socket) {
  CHECK(socket_);
}

ReadMultipleEmulator::~ReadMultipleEmulator() = default;

base::expected<DatagramsMetadata, Error> ReadMultipleEmulator::ReadMultiple(
    IOBuffer* buf,
    size_t buf_len,
    size_t maximum_packet_size,
    base::OnceCallback<void(base::expected<DatagramsMetadata, Error>)>
        callback) {
  auto adapted_callback =
      base::BindOnce(&ReadMultipleEmulator::OnReadComplete,
                     weak_factory_.GetWeakPtr(), std::move(callback));

  int rv = socket_->Read(
      buf, base::checked_cast<int>(std::min(buf_len, maximum_packet_size)),
      std::move(adapted_callback));

  if (rv < 0) {
    return base::unexpected(static_cast<Error>(rv));
  }

  // Preserve the socket's per-packet TOS so QUIC can read the ECN codepoint
  // (a bare 0 would make every packet appear Not-ECT). GetLastTos() reflects
  // the datagram just returned by the synchronous Read() above.
  const DscpAndEcn tos = socket_->GetLastTos();
  return DatagramsMetadata{{/*offset=*/0, /*length=*/static_cast<size_t>(rv),
                            /*tos=*/DscpAndEcnToTos(tos.dscp, tos.ecn)}};
}

void ReadMultipleEmulator::OnReadComplete(
    base::OnceCallback<void(base::expected<DatagramsMetadata, Error>)> callback,
    int rv) {
  if (rv < 0) {
    std::move(callback).Run(base::unexpected(static_cast<Error>(rv)));
  } else {
    // See ReadMultiple(): preserve the just-read datagram's TOS for ECN.
    const DscpAndEcn tos = socket_->GetLastTos();
    std::move(callback).Run(
        DatagramsMetadata{{/*offset=*/0, /*length=*/static_cast<size_t>(rv),
                           /*tos=*/DscpAndEcnToTos(tos.dscp, tos.ecn)}});
  }
}

}  // namespace net