94 lines
3.3 KiB
C++
94 lines
3.3 KiB
C++
|
|
#include "svc_createstringtable.h"
|
|
#include "base/bitfile.h"
|
|
#include "game/sourcecontext.h"
|
|
#include "game/stringtables.h"
|
|
#include "netmath.h"
|
|
#include "netcontants.h"
|
|
|
|
#include "sourcesdk/common.h"
|
|
#include <vector>
|
|
|
|
namespace NetHandlers
|
|
{
|
|
bool SVC_CreateStringTable_BitRead_Internal(BitRead& bitbuf, SourceGameContext& context, NetMsg::SVC_CreateStringTable* data)
|
|
{
|
|
if (bitbuf.PeekUBitLong(8) == ':')
|
|
{
|
|
data->isFileNames = true;
|
|
bitbuf.ReadByte();
|
|
}
|
|
else
|
|
{
|
|
data->isFileNames = false;
|
|
}
|
|
bitbuf.ReadString(data->tableName, sizeof(data->tableName));
|
|
data->maxEntries = bitbuf.ReadWord();
|
|
|
|
const uint32_t numEncodeBits = math::log2(data->maxEntries);
|
|
data->numEntries = bitbuf.ReadUBitLong(numEncodeBits + 1);
|
|
|
|
if (context.protocol > 23)
|
|
{
|
|
data->dataLengthInBits = bitbuf.ReadVarInt32();
|
|
}
|
|
else
|
|
{
|
|
data->dataLengthInBits = bitbuf.ReadUBitLong(NET_MAX_PAYLOAD_BITS_OLD + 3);
|
|
}
|
|
const uint32_t dataLengthInBytes = math::BitsToBytes(data->dataLengthInBits);
|
|
|
|
data->isUserDataFixedSize = bitbuf.ReadOneBit() != 0;
|
|
if (data->isUserDataFixedSize)
|
|
{
|
|
data->userDataSize = bitbuf.ReadUBitLong(12);
|
|
data->userDataSizeBits = bitbuf.ReadUBitLong(4);
|
|
}
|
|
else
|
|
{
|
|
data->userDataSize = 0;
|
|
data->userDataSizeBits = 0;
|
|
}
|
|
if (context.protocol > 14)
|
|
{
|
|
data->compressedData = bitbuf.ReadOneBit() != 0;
|
|
}
|
|
else
|
|
{
|
|
data->compressedData = false;
|
|
}
|
|
|
|
data->data.reset(new uint8_t[dataLengthInBytes]);
|
|
bitbuf.ReadBits(data->data.get(), data->dataLengthInBits);
|
|
|
|
StringTable *table = context.stringTables->GetStringTable(data->tableName, true);
|
|
table->maxEntries = data->maxEntries;
|
|
table->isUserDataFixedSize = data->isUserDataFixedSize;
|
|
table->userDataSize = data->userDataSize;
|
|
table->userDataSizeBits = data->userDataSizeBits;
|
|
table->entryBits = math::log2(table->maxEntries);
|
|
|
|
if (data->compressedData)
|
|
{
|
|
bf_read bitbuf2(data->data.get(), dataLengthInBytes, data->dataLengthInBits);
|
|
const uint32_t decompressedNumBytes = bitbuf2.ReadUBitLong(32);
|
|
const uint32_t compressedNumBytes = bitbuf2.ReadUBitLong(32);
|
|
std::unique_ptr<uint8_t[]> compressedData(new uint8[compressedNumBytes]);
|
|
std::unique_ptr<uint8_t[]> uncompressedData(new uint8[decompressedNumBytes]);
|
|
bitbuf2.ReadBytes(compressedData.get(), compressedNumBytes);
|
|
|
|
uint32_t numWritten = COM_BufferToBufferDecompress(uncompressedData.get(), decompressedNumBytes, compressedData.get(), compressedNumBytes);
|
|
assert(numWritten == decompressedNumBytes);
|
|
bitbuf2 = bf_read(uncompressedData.get(), decompressedNumBytes);
|
|
table->ParseUpdate(bitbuf2, data->numEntries, context);
|
|
}
|
|
else if(dataLengthInBytes)
|
|
{
|
|
bf_read bitbuf2(data->data.get(), dataLengthInBytes, data->dataLengthInBits);
|
|
table->ParseUpdate(bitbuf2, data->numEntries, context);
|
|
}
|
|
|
|
return !bitbuf.IsOverflowed();
|
|
}
|
|
}
|