Initial commit

Torque/SDK/engine/sim/connectionStringTable.cc

@@ -0,0 +1,172 @@
+//-----------------------------------------------------------------------------
+// Torque Game Engine
+// Copyright (c) 2002 GarageGames.Com
+//-----------------------------------------------------------------------------
+
+#include "platform/platform.h"
+#include "core/dnet.h"
+#include "console/simBase.h"
+#include "sim/netConnection.h"
+#include "core/bitStream.h"
+#include "console/consoleTypes.h"
+
+class NetStringEvent : public NetEvent
+{
+   StringHandle mString;
+   U32 mIndex;
+public:
+   NetStringEvent(U32 index = 0, StringHandle string = StringHandle())
+   {
+      mIndex = index;
+      mString = string;
+   }
+   virtual void pack(NetConnection* /*ps*/, BitStream *bstream)
+   {
+      bstream->writeInt(mIndex, ConnectionStringTable::EntryBitSize);
+      bstream->writeString(mString.getString());
+   }
+   virtual void write(NetConnection* /*ps*/, BitStream *bstream)
+   {
+      bstream->writeInt(mIndex, ConnectionStringTable::EntryBitSize);
+      bstream->writeString(mString.getString());
+   }
+   virtual void unpack(NetConnection* /*con*/, BitStream *bstream)
+   {
+      char buf[256];
+      mIndex = bstream->readInt(ConnectionStringTable::EntryBitSize);
+      bstream->readString(buf);
+      mString = StringHandle(buf);
+   }
+   virtual void notifyDelivered(NetConnection *ps, bool madeit)
+   {
+      if(madeit)
+         ps->confirmStringReceived(mString, mIndex);
+   }
+   virtual void process(NetConnection *connection)
+   {
+      Con::printf("Mapping string: %s to index: %d", mString.getString(), mIndex);
+      connection->mapString(mIndex, mString);
+   }
+#ifdef TORQUE_DEBUG_NET
+   const char *getDebugName()
+   {
+      static char buffer[512];
+      dSprintf(buffer, sizeof(buffer), "%s - \"", getClassName());
+      expandEscape(buffer + dStrlen(buffer), mString.getString());
+      dStrcat(buffer, "\"");
+      return buffer;
+   }
+#endif
+   DECLARE_CONOBJECT(NetStringEvent);
+};
+
+IMPLEMENT_CO_NETEVENT_V1(NetStringEvent);
+
+//--------------------------------------------------------------------
+
+
+ConnectionStringTable::ConnectionStringTable(NetConnection *parent)
+{
+   mParent = parent;
+   for(U32 i = 0; i < EntryCount; i++)
+   {
+      mEntryTable[i].nextHash = NULL;
+      mEntryTable[i].nextLink = &mEntryTable[i+1];
+      mEntryTable[i].prevLink = &mEntryTable[i-1];
+      mEntryTable[i].index = i;
+
+      mHashTable[i] = NULL;
+   }
+   mLRUHead.nextLink = &mEntryTable[0];
+   mEntryTable[0].prevLink = &mLRUHead;
+   mLRUTail.prevLink = &mEntryTable[EntryCount-1];
+   mEntryTable[EntryCount-1].nextLink = &mLRUTail;
+}
+
+U32 ConnectionStringTable::getNetSendId(StringHandle &string)
+{
+   // see if the entry is in the hash table right now
+   U32 hashIndex = string.getIndex() % EntryCount;
+   for(Entry *walk = mHashTable[hashIndex]; walk; walk = walk->nextHash)
+      if(walk->string == string)
+         return walk->index;
+   AssertFatal(0, "Net send id is not in the table.  Error!");
+   return 0;
+}
+
+U32 ConnectionStringTable::checkString(StringHandle &string, bool *isOnOtherSide)
+{
+   // see if the entry is in the hash table right now
+   U32 hashIndex = string.getIndex() % EntryCount;
+   for(Entry *walk = mHashTable[hashIndex]; walk; walk = walk->nextHash)
+   {
+      if(walk->string == string)
+      {
+         pushBack(walk);
+		 if(isOnOtherSide)
+			 *isOnOtherSide = walk->receiveConfirmed;
+         return walk->index;
+      }
+   }
+
+   // not in the hash table, means we have to add it
+   Entry *newEntry;
+
+   // pull the new entry from the LRU list.
+   newEntry = mLRUHead.nextLink;
+   pushBack(newEntry);
+
+   // remove the string from the hash table
+   Entry **hashWalk;
+   for (hashWalk = &mHashTable[newEntry->string.getIndex() % EntryCount]; *hashWalk; hashWalk = &((*hashWalk)->nextHash))
+   {
+      if(*hashWalk == newEntry)
+      {
+         *hashWalk = newEntry->nextHash;
+         break;
+      }
+   }
+
+   newEntry->string = string;
+   newEntry->receiveConfirmed = false;
+   newEntry->nextHash = mHashTable[hashIndex];
+   mHashTable[hashIndex] = newEntry;
+
+   mParent->postNetEvent(new NetStringEvent(newEntry->index, string));
+   if(isOnOtherSide)
+      *isOnOtherSide = false;
+   return newEntry->index;
+}
+
+void ConnectionStringTable::mapString(U32 netId, StringHandle &string)
+{
+   // the netId is sent by the other side... throw it in our mapping table:
+   mRemoteStringTable[netId] = string;
+}
+
+void ConnectionStringTable::readDemoStartBlock(BitStream *stream)
+{
+   // ok, reading the demo start block
+   for(U32 i = 0; i < EntryCount; i++)
+   {
+      if(stream->readFlag())
+      {
+         char buffer[256];
+         stream->readString(buffer);
+         mRemoteStringTable[i] = StringHandle(buffer);
+      }
+   }
+}
+
+void ConnectionStringTable::writeDemoStartBlock(ResizeBitStream *stream)
+{
+   // ok, writing the demo start block
+   for(U32 i = 0; i < EntryCount; i++)
+   {
+      if(stream->writeFlag(mRemoteStringTable[i].isValidString()))
+      {
+         stream->writeString(mRemoteStringTable[i].getString());
+         stream->validate();
+      }
+   }
+}