ddnet/src/engine/shared/packer.cpp

165 lines
2.5 KiB
C++
Raw Normal View History

2010-11-20 10:37:14 +00:00
/* (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. */
2009-10-27 14:38:53 +00:00
#include <base/system.h>
2010-05-29 07:25:38 +00:00
#include "packer.h"
#include "compression.h"
#include "config.h"
2009-10-27 14:38:53 +00:00
void CPacker::Reset()
{
m_Error = 0;
m_pCurrent = m_aBuffer;
m_pEnd = m_pCurrent + PACKER_BUFFER_SIZE;
}
void CPacker::AddInt(int i)
{
if(m_Error)
return;
2010-05-29 07:25:38 +00:00
// make sure that we have space enough
2009-10-27 14:38:53 +00:00
if(m_pEnd - m_pCurrent < 6)
{
dbg_break();
m_Error = 1;
}
else
2010-05-29 07:25:38 +00:00
m_pCurrent = CVariableInt::Pack(m_pCurrent, i);
2009-10-27 14:38:53 +00:00
}
void CPacker::AddString(const char *pStr, int Limit)
{
if(m_Error)
return;
2010-05-29 07:25:38 +00:00
//
2009-10-27 14:38:53 +00:00
if(Limit > 0)
{
while(*pStr && Limit != 0)
{
*m_pCurrent++ = *pStr++;
Limit--;
2009-10-27 14:38:53 +00:00
if(m_pCurrent >= m_pEnd)
{
m_Error = 1;
break;
}
}
*m_pCurrent++ = 0;
}
else
{
while(*pStr)
{
*m_pCurrent++ = *pStr++;
if(m_pCurrent >= m_pEnd)
{
m_Error = 1;
break;
}
}
*m_pCurrent++ = 0;
}
}
2010-05-29 07:25:38 +00:00
void CPacker::AddRaw(const void *pData, int Size)
2009-10-27 14:38:53 +00:00
{
if(m_Error)
return;
2009-10-27 14:38:53 +00:00
if(m_pCurrent+Size >= m_pEnd)
{
m_Error = 1;
return;
}
2010-05-29 07:25:38 +00:00
const unsigned char *pSrc = (const unsigned char *)pData;
2009-10-27 14:38:53 +00:00
while(Size)
{
2010-05-29 07:25:38 +00:00
*m_pCurrent++ = *pSrc++;
2009-10-27 14:38:53 +00:00
Size--;
}
}
2010-05-29 07:25:38 +00:00
void CUnpacker::Reset(const void *pData, int Size)
2009-10-27 14:38:53 +00:00
{
m_Error = 0;
2010-05-29 07:25:38 +00:00
m_pStart = (const unsigned char *)pData;
2009-10-27 14:38:53 +00:00
m_pEnd = m_pStart + Size;
m_pCurrent = m_pStart;
}
int CUnpacker::GetInt()
{
if(m_Error)
return 0;
2009-10-27 14:38:53 +00:00
if(m_pCurrent >= m_pEnd)
{
m_Error = 1;
return 0;
}
2009-10-27 14:38:53 +00:00
int i;
2010-05-29 07:25:38 +00:00
m_pCurrent = CVariableInt::Unpack(m_pCurrent, &i);
2009-10-27 14:38:53 +00:00
if(m_pCurrent > m_pEnd)
{
m_Error = 1;
return 0;
}
return i;
}
const char *CUnpacker::GetString(int SanitizeType)
2009-10-27 14:38:53 +00:00
{
if(m_Error)
2009-10-27 14:38:53 +00:00
return "";
if(m_pCurrent >= m_pEnd)
{
m_Error = 1;
return "";
}
2009-10-27 14:38:53 +00:00
char *pPtr = (char *)m_pCurrent;
2010-05-29 07:25:38 +00:00
while(*m_pCurrent) // skip the string
2009-10-27 14:38:53 +00:00
{
m_pCurrent++;
if(m_pCurrent == m_pEnd)
{
2016-05-03 22:24:54 +00:00
m_Error = 1;
2009-10-27 14:38:53 +00:00
return "";
}
}
m_pCurrent++;
2010-05-29 07:25:38 +00:00
// sanitize all strings
if(SanitizeType&SANITIZE)
str_sanitize(pPtr);
else if(SanitizeType&SANITIZE_CC)
str_sanitize_cc(pPtr);
2013-04-01 18:30:58 +00:00
return SanitizeType&SKIP_START_WHITESPACES ? str_utf8_skip_whitespaces(pPtr) : pPtr;
2009-10-27 14:38:53 +00:00
}
const unsigned char *CUnpacker::GetRaw(int Size)
{
const unsigned char *pPtr = m_pCurrent;
if(m_Error)
return 0;
2010-05-29 07:25:38 +00:00
// check for nasty sizes
2009-10-27 14:38:53 +00:00
if(Size < 0 || m_pCurrent+Size > m_pEnd)
{
m_Error = 1;
return 0;
}
2010-05-29 07:25:38 +00:00
// "unpack" the data
2009-10-27 14:38:53 +00:00
m_pCurrent += Size;
return pPtr;
}