mirror of
https://github.com/celisej567/mcpe.git
synced 2025-12-31 17:49:17 +03:00
151 lines
3.3 KiB
C++
151 lines
3.3 KiB
C++
/*
|
|
* Copyright (c) 2014, Oculus VR, Inc.
|
|
* All rights reserved.
|
|
*
|
|
* This source code is licensed under the BSD-style license found in the
|
|
* LICENSE file in the root directory of this source tree. An additional grant
|
|
* of patent rights can be found in the PATENTS file in the same directory.
|
|
*
|
|
*/
|
|
|
|
#include "StringTable.h"
|
|
#include <string.h>
|
|
#include "RakAssert.h"
|
|
#include <stdio.h>
|
|
#include "BitStream.h"
|
|
#include "StringCompressor.h"
|
|
using namespace RakNet;
|
|
|
|
StringTable* StringTable::instance=0;
|
|
int StringTable::referenceCount=0;
|
|
|
|
|
|
int RakNet::StrAndBoolComp( char *const &key, const StrAndBool &data )
|
|
{
|
|
return strcmp(key,(const char*)data.str);
|
|
}
|
|
|
|
StringTable::StringTable()
|
|
{
|
|
|
|
}
|
|
|
|
StringTable::~StringTable()
|
|
{
|
|
unsigned i;
|
|
for (i=0; i < orderedStringList.Size(); i++)
|
|
{
|
|
if (orderedStringList[i].b)
|
|
rakFree_Ex(orderedStringList[i].str, _FILE_AND_LINE_ );
|
|
}
|
|
}
|
|
|
|
void StringTable::AddReference(void)
|
|
{
|
|
if (++referenceCount==1)
|
|
{
|
|
instance = RakNet::OP_NEW<StringTable>( _FILE_AND_LINE_ );
|
|
}
|
|
}
|
|
void StringTable::RemoveReference(void)
|
|
{
|
|
RakAssert(referenceCount > 0);
|
|
|
|
if (referenceCount > 0)
|
|
{
|
|
if (--referenceCount==0)
|
|
{
|
|
RakNet::OP_DELETE(instance, _FILE_AND_LINE_);
|
|
instance=0;
|
|
}
|
|
}
|
|
}
|
|
|
|
StringTable* StringTable::Instance(void)
|
|
{
|
|
return instance;
|
|
}
|
|
|
|
void StringTable::AddString(const char *str, bool copyString)
|
|
{
|
|
StrAndBool sab;
|
|
sab.b=copyString;
|
|
if (copyString)
|
|
{
|
|
sab.str = (char*) rakMalloc_Ex( strlen(str)+1, _FILE_AND_LINE_ );
|
|
strcpy(sab.str, str);
|
|
}
|
|
else
|
|
{
|
|
sab.str=(char*)str;
|
|
}
|
|
|
|
// If it asserts inside here you are adding duplicate strings.
|
|
orderedStringList.Insert(sab.str,sab, true, _FILE_AND_LINE_);
|
|
|
|
// If this assert hits you need to increase the range of StringTableType
|
|
RakAssert(orderedStringList.Size() < (StringTableType)-1);
|
|
|
|
}
|
|
void StringTable::EncodeString( const char *input, int maxCharsToWrite, RakNet::BitStream *output )
|
|
{
|
|
unsigned index;
|
|
bool objectExists;
|
|
// This is fast because the list is kept ordered.
|
|
index=orderedStringList.GetIndexFromKey((char*)input, &objectExists);
|
|
if (objectExists)
|
|
{
|
|
output->Write(true);
|
|
output->Write((StringTableType)index);
|
|
}
|
|
else
|
|
{
|
|
LogStringNotFound(input);
|
|
output->Write(false);
|
|
StringCompressor::Instance()->EncodeString(input, maxCharsToWrite, output);
|
|
}
|
|
}
|
|
|
|
bool StringTable::DecodeString( char *output, int maxCharsToWrite, RakNet::BitStream *input )
|
|
{
|
|
bool hasIndex=false;
|
|
RakAssert(maxCharsToWrite>0);
|
|
|
|
if (maxCharsToWrite==0)
|
|
return false;
|
|
if (!input->Read(hasIndex))
|
|
return false;
|
|
if (hasIndex==false)
|
|
{
|
|
StringCompressor::Instance()->DecodeString(output, maxCharsToWrite, input);
|
|
}
|
|
else
|
|
{
|
|
StringTableType index;
|
|
if (!input->Read(index))
|
|
return false;
|
|
if (index >= orderedStringList.Size())
|
|
{
|
|
#ifdef _DEBUG
|
|
// Critical error - got a string index out of range, which means AddString was called more times on the remote system than on this system.
|
|
// All systems must call AddString the same number of types, with the same strings in the same order.
|
|
RakAssert(0);
|
|
#endif
|
|
return false;
|
|
}
|
|
|
|
strncpy(output, orderedStringList[index].str, maxCharsToWrite);
|
|
output[maxCharsToWrite-1]=0;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
void StringTable::LogStringNotFound(const char *strName)
|
|
{
|
|
(void) strName;
|
|
|
|
#ifdef _DEBUG
|
|
RAKNET_DEBUG_PRINTF("Efficiency Warning! Unregistered String %s sent to StringTable.\n", strName);
|
|
#endif
|
|
}
|