src/engine/shared/packer.h - master - dieoska/ddnet

src/engine/shared/packer.h

81 строка2 KB

Robert Müller

Use separate packer with larger buffer only for Teehistorian

21 ноя 2024, 21:52

21 ноя 2024, 21:52c83d79e

/* (c) Magnus Auvinen. See licence.txt in the root of the distribution for more information. */
/* If you are missing that file, acquire a complete release at teeworlds.com.                */
#ifndef ENGINE_SHARED_PACKER_H
#define ENGINE_SHARED_PACKER_H

#include <cstddef>

/**
 * Abstract packer implementation. Subclasses must supply the buffer.
 */
class CAbstractPacker
{
private:
	unsigned char *const m_pBuffer;
	const size_t m_BufferSize;
	unsigned char *m_pCurrent;
	unsigned char *m_pEnd;
	bool m_Error;

protected:
	CAbstractPacker(unsigned char *pBuffer, size_t Size);

public:
	void Reset();
	void AddInt(int i);
	void AddString(const char *pStr, int Limit = 0, bool AllowTruncation = true);
	void AddRaw(const void *pData, int Size);

	int Size() const { return (int)(m_pCurrent - m_pBuffer); }
	const unsigned char *Data() const { return m_pBuffer; }
	bool Error() const { return m_Error; }
};

/**
 * Default packer with buffer for networking.
 */
class CPacker : public CAbstractPacker
{
public:
	enum
	{
		PACKER_BUFFER_SIZE = 1024 * 2
	};
	CPacker() :
		CAbstractPacker(m_aBuffer, sizeof(m_aBuffer))
	{
	}

private:
	unsigned char m_aBuffer[PACKER_BUFFER_SIZE];
};

class CUnpacker
{
	const unsigned char *m_pStart;
	const unsigned char *m_pCurrent;
	const unsigned char *m_pEnd;
	bool m_Error;

public:
	enum
	{
		SANITIZE = 1,
		SANITIZE_CC = 2,
		SKIP_START_WHITESPACES = 4
	};

	void Reset(const void *pData, int Size);
	int GetInt();
	int GetIntOrDefault(int Default);
	int GetUncompressedInt();
	int GetUncompressedIntOrDefault(int Default);
	const char *GetString(int SanitizeType = SANITIZE);
	const unsigned char *GetRaw(int Size);
	bool Error() const { return m_Error; }

	int CompleteSize() const { return m_pEnd - m_pStart; }
	const unsigned char *CompleteData() const { return m_pStart; }
};

#endif