packetbase.hpp

// GPT
#pragma once
#include <cstdint>
#include <memory>
#include <steamclient/types/generated/SteamLanguageInternal.hpp>
#include <vector>
 
 
namespace Steam::Messaging::Packets {
// Packet interface (read-only, header + payload)
class IPacketMsg {
 public:
  virtual ~IPacketMsg() = default;
  virtual bool IsProto() const = 0;
  virtual Steam::Internal::Enums::EMsg MsgType() const = 0;
  virtual uint64_t TargetJobID() const = 0;
  virtual uint64_t SourceJobID() const = 0;
  virtual const std::vector<uint8_t>& GetData() const = 0;
};
 
// Protobuf-backed packet
class PacketClientMsgProtobuf : public IPacketMsg {
 public:
  Steam::Internal::Enums::EMsg msgType;
  Steam::Internal::MsgHdrProtoBuf header;
  std::vector<uint8_t> payload;
  size_t bodyOffset = 0;
 
  PacketClientMsgProtobuf(Steam::Internal::Enums::EMsg eMsg,
                          const std::vector<uint8_t>& data)
      : msgType(eMsg), payload(data) {
    Steam::Utils::Stream s(payload);
    header.Deserialize(s);
    bodyOffset = s.Position();
  }
 
  bool IsProto() const override { return true; }
  Steam::Internal::Enums::EMsg MsgType() const override { return msgType; }
  uint64_t TargetJobID() const override { return header.proto.jobid_target(); }
  uint64_t SourceJobID() const override { return header.proto.jobid_source(); }
  const std::vector<uint8_t>& GetData() const override { return payload; }
  const size_t GetBodyOffset() { return bodyOffset; }
};
 
class PacketClientMsg : public IPacketMsg {
 public:
  Steam::Internal::Enums::EMsg msgType;
  Steam::Internal::ExtendedClientMsgHdr header;
  std::vector<uint8_t> payload;
  size_t bodyOffset = 0;
 
  PacketClientMsg(Steam::Internal::Enums::EMsg eMsg,
                  const std::vector<uint8_t>& data)
      : msgType(eMsg), payload(data) {
    Steam::Utils::Stream s(payload);
    header.Deserialize(s);
    bodyOffset = s.Position();
  }
 
  bool IsProto() const override { return false; }
  Steam::Internal::Enums::EMsg MsgType() const override { return msgType; }
  uint64_t TargetJobID() const override { return header.targetJobID; }
  uint64_t SourceJobID() const override { return header.sourceJobID; }
  const std::vector<uint8_t>& GetData() const override { return payload; }
};
 
class PacketMsg : public IPacketMsg {
 public:
  Steam::Internal::Enums::EMsg msgType;
  Steam::Internal::MsgHdr header;
  std::vector<uint8_t> payload;
  size_t bodyOffset = 0;
 
  PacketMsg(Steam::Internal::Enums::EMsg eMsg, const std::vector<uint8_t>& data)
      : msgType(eMsg), payload(data) {
    Steam::Utils::Stream s(payload);
    header.Deserialize(s);
    bodyOffset = s.Position();
  }
 
  bool IsProto() const override { return false; }
  Steam::Internal::Enums::EMsg MsgType() const override { return msgType; }
  uint64_t TargetJobID() const override { return header.targetJobID; }
  uint64_t SourceJobID() const override { return header.sourceJobID; }
  const std::vector<uint8_t>& GetData() const override { return payload; }
};
}  // namespace Steam::Messaging::Packets