EQStream abstraction layer

common/net/packet.h

@@ -0,0 +1,131 @@
+#pragma once
+
+#include <cstdint>
+#include <string>
+#include <stdexcept>
+#include <cstring>
+#include "../util/memory_stream.h"
+#include <cereal/cereal.hpp>
+#include <cereal/archives/binary.hpp>
+
+namespace EQ {
+	namespace Net {
+		class Packet
+		{
+		public:
+			Packet() { }
+			virtual ~Packet() { }
+
+			virtual const void *Data() const = 0;
+			virtual void *Data() = 0;
+			virtual size_t Length() const = 0;
+			virtual size_t Length() = 0;
+			virtual bool Clear() = 0;
+			virtual bool Resize(size_t new_size) = 0;
+			virtual void Reserve(size_t new_size) = 0;
+
+			template<typename T>
+			T GetSerialize(size_t offset) const
+			{
+				if (T::size() > (Length() - offset)) {
+					throw std::out_of_range("Packet::GetSerialize(), packet not large enough to cast to type.");
+				}
+			
+				T ret;
+				Util::MemoryStreamReader reader(((char*)Data() + offset), Length());
+				cereal::BinaryInputArchive input(reader);
+				input(ret);
+				return ret;
+			}
+
+			template<typename T>
+			void PutSerialize(size_t offset, const T &value) {
+				std::stringstream buffer(std::ios::in | std::ios::out | std::ios::binary);
+				cereal::BinaryOutputArchive output(buffer);
+				output(value);
+			
+				auto str = buffer.str();
+				if (Length() < offset + str.length()) {
+					if (!Resize(offset + str.length())) {
+						throw std::out_of_range("Packet::PutSerialize(), could not resize packet and would of written past the end.");
+					}
+				}
+			
+				memcpy((char*)Data() + offset, &str[0], str.length());
+			}
+
+			void PutInt8(size_t offset, int8_t value);
+			void PutInt16(size_t offset, int16_t value);
+			void PutInt32(size_t offset, int32_t value);
+			void PutInt64(size_t offset, int64_t value);
+			void PutUInt8(size_t offset, uint8_t value);
+			void PutUInt16(size_t offset, uint16_t value);
+			void PutUInt32(size_t offset, uint32_t value);
+			void PutUInt64(size_t offset, uint64_t value);
+			void PutFloat(size_t offset, float value);
+			void PutDouble(size_t offset, double value);
+			void PutString(size_t offset, const std::string &str);
+			void PutCString(size_t offset, const char *str);
+			void PutPacket(size_t offset, const Packet &p);
+			void PutData(size_t offset, void *data, size_t length);
+			
+			int8_t GetInt8(size_t offset) const;
+			int16_t GetInt16(size_t offset) const;
+			int32_t GetInt32(size_t offset) const;
+			int64_t GetInt64(size_t offset) const;
+			uint8_t GetUInt8(size_t offset) const;
+			uint16_t GetUInt16(size_t offset) const;
+			uint32_t GetUInt32(size_t offset) const;
+			uint64_t GetUInt64(size_t offset) const;
+			float GetFloat(size_t offset) const;
+			double GetDouble(size_t offset) const;
+			std::string GetString(size_t offset, size_t length) const;
+			std::string GetCString(size_t offset) const;
+			
+			std::string ToString() const;
+			std::string ToString(size_t line_length) const;
+		};
+
+		class ReadOnlyPacket : public Packet
+		{
+		public:
+			ReadOnlyPacket(void *data, size_t size) { m_data = data; m_data_length = size; }
+			virtual ~ReadOnlyPacket() { }
+			ReadOnlyPacket(const ReadOnlyPacket &o) { m_data = o.m_data; m_data_length = o.m_data_length; }
+			ReadOnlyPacket& operator=(const ReadOnlyPacket &o) { m_data = o.m_data; m_data_length = o.m_data_length; return *this; }
+			ReadOnlyPacket(ReadOnlyPacket &&o) { m_data = o.m_data; m_data_length = o.m_data_length; }
+
+			virtual const void *Data() const { return m_data; }
+			virtual void *Data() { return m_data; }
+			virtual size_t Length() const { return m_data_length; }
+			virtual size_t Length() { return m_data_length; }
+			virtual bool Clear() { return false; }
+			virtual bool Resize(size_t new_size) { return false; }
+			virtual void Reserve(size_t new_size) { }
+
+		protected:
+			void *m_data;
+			size_t m_data_length;
+		};
+
+		class WritablePacket : public Packet
+		{
+		public:
+			WritablePacket() { }
+			virtual ~WritablePacket() { }
+			WritablePacket(WritablePacket &&o) { m_data = std::move(o.m_data); }
+			WritablePacket(const WritablePacket &o) { m_data = o.m_data; }
+			WritablePacket& operator=(const WritablePacket &o) { m_data = o.m_data; return *this; }
+
+			virtual const void *Data() const { return &m_data[0]; }
+			virtual void *Data() { return &m_data[0]; }
+			virtual size_t Length() const { return m_data.size(); }
+			virtual size_t Length() { return m_data.size(); }
+			virtual bool Clear() { m_data.clear(); return true; }
+			virtual bool Resize(size_t new_size) { m_data.resize(new_size); return true; }
+			virtual void Reserve(size_t new_size) { m_data.reserve(new_size); }
+		protected:
+			std::vector<char> m_data;
+		};
+	}
+}