packet.h

Go to the documentation of this file.

/*
 * This file is part of OpenTTD.
 * OpenTTD is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, version 2.
 * OpenTTD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with OpenTTD. If not, see <http://www.gnu.org/licenses/>.
 */
 
#ifndef NETWORK_CORE_PACKET_H
#define NETWORK_CORE_PACKET_H
 
#include "os_abstraction.h"
#include "config.h"
#include "core.h"
#include "../../string_type.h"
 
typedef uint16_t PacketSize; 
typedef uint8_t  PacketType; 
 
struct Packet {
  static constexpr size_t EncodedLengthOfPacketSize() { return sizeof(PacketSize); }
  static constexpr size_t EncodedLengthOfPacketType() { return sizeof(PacketType); }
private:
  PacketSize pos;
  std::vector<uint8_t> buffer;
  size_t limit;
 
  NetworkSocketHandler *cs;
 
public:
  Packet(NetworkSocketHandler *cs, size_t limit, size_t initial_read_size = EncodedLengthOfPacketSize());
  Packet(NetworkSocketHandler *cs, PacketType type, size_t limit = COMPAT_MTU);
 
  /* Sending/writing of packets */
  void PrepareToSend();
 
  bool   CanWriteToPacket(size_t bytes_to_write);
  void   Send_bool  (bool   data);
  void   Send_uint8 (uint8_t  data);
  void   Send_uint16(uint16_t data);
  void   Send_uint32(uint32_t data);
  void   Send_uint64(uint64_t data);
  void   Send_string(const std::string_view data);
  void   Send_buffer(const std::vector<uint8_t> &data);
  std::span<const uint8_t> Send_bytes(const std::span<const uint8_t> span);
 
  /* Reading/receiving of packets */
  bool HasPacketSizeData() const;
  bool ParsePacketSize();
  size_t Size() const;
  [[nodiscard]] bool PrepareToRead();
  PacketType GetPacketType() const;
 
  bool   CanReadFromPacket(size_t bytes_to_read, bool close_connection = false);
  bool   Recv_bool  ();
  uint8_t  Recv_uint8 ();
  uint16_t Recv_uint16();
  uint32_t Recv_uint32();
  uint64_t Recv_uint64();
  std::vector<uint8_t> Recv_buffer();
  size_t Recv_bytes(std::span<uint8_t> span);
  std::string Recv_string(size_t length, StringValidationSettings settings = SVS_REPLACE_WITH_QUESTION_MARK);
 
  size_t RemainingBytesToTransfer() const;
 
  template <
    typename A = size_t, 
    typename F,          
    typename D,          
    typename ... Args>   
  ssize_t TransferOutWithLimit(F transfer_function, size_t limit, D destination, Args&& ... args)
  {
    size_t amount = std::min(this->RemainingBytesToTransfer(), limit);
    if (amount == 0) return 0;
 
    assert(this->pos < this->buffer.size());
    assert(this->pos + amount <= this->buffer.size());
    /* Making buffer a char means casting a lot in the Recv/Send functions. */
    const char *output_buffer = reinterpret_cast<const char*>(this->buffer.data() + this->pos);
    ssize_t bytes = transfer_function(destination, output_buffer, static_cast<A>(amount), std::forward<Args>(args)...);
    if (bytes > 0) this->pos += bytes;
    return bytes;
  }
 
  template <typename A = size_t, typename F, typename D, typename ... Args>
  ssize_t TransferOut(F transfer_function, D destination, Args&& ... args)
  {
    return TransferOutWithLimit<A>(transfer_function, std::numeric_limits<size_t>::max(), destination, std::forward<Args>(args)...);
  }
 
  template <typename A = size_t, typename F, typename S, typename ... Args>
  ssize_t TransferIn(F transfer_function, S source, Args&& ... args)
  {
    size_t amount = this->RemainingBytesToTransfer();
    if (amount == 0) return 0;
 
    assert(this->pos < this->buffer.size());
    assert(this->pos + amount <= this->buffer.size());
    /* Making buffer a char means casting a lot in the Recv/Send functions. */
    char *input_buffer = reinterpret_cast<char*>(this->buffer.data() + this->pos);
    ssize_t bytes = transfer_function(source, input_buffer, static_cast<A>(amount), std::forward<Args>(args)...);
    if (bytes > 0) this->pos += bytes;
    return bytes;
  }
};
 
#endif /* NETWORK_CORE_PACKET_H */