Made demreader use demmessage handling funcs

demboyz/demowriter.cpp

@@ -4,6 +4,7 @@
 #include "demofile/demofile.h"
 #include "netmessages/nethandlers.h"
 #include "netmessages/netcontants.h"
+#include "demmessages/demhandlers.h"
 #include "sourcesdk/bitbuf.h"
 #include <cstdio>
 #include <memory>
@@ -51,11 +52,7 @@ void DemoWriter::EndWriting()
 void DemoWriter::StartCommandPacket(CommandPacket& packet)
 {
     m_writer.WriteCmdHeader(packet.cmd, packet.tick);
-    if (packet.cmd == dem_packet || packet.cmd == dem_signon)
+    DemHandlers::DemMsg_FileWrite(packet.cmd, m_writer, packet.data);
-    {
-        m_writer.WriteCmdInfo(*packet.cmdInfo);
-        m_writer.WriteSequenceInfo(packet.sequenceInfo1, packet.sequenceInfo2);
-    }
     m_cmdPacketBuf.Reset();
 }
 

demboyz/demreader.cpp

@@ -6,13 +6,14 @@
 
 #include "netmessages/netmessages.h"
 #include "netmessages/netcontants.h"
+#include "demmessages/demmessages.h"
+#include "demmessages/demhandlers.h"
 #include "sourcesdk/bitbuf.h"
 #include <vector>
 #include <cstdint>
+#include <memory>
 
-static void* netDataStructs[32];
+static void CreateNetMsgStructs(void* netDataStructs[32])
-
-static void CreateNetMsgStructs()
 {
     netDataStructs[0] = new NetMsg::Net_NOP();
     netDataStructs[1] = new NetMsg::Net_Disconnect();
@@ -48,7 +49,7 @@ static void CreateNetMsgStructs()
     netDataStructs[31] = new NetMsg::SVC_GetCvarValue();
 }
 
-static void DestroyNetMsgStructs()
+static void DestroyNetMsgStructs(void* netDataStructs[32])
 {
     delete reinterpret_cast<NetMsg::Net_NOP*>(netDataStructs[0]);
     delete reinterpret_cast<NetMsg::Net_Disconnect*>(netDataStructs[1]);
@@ -84,7 +85,35 @@ static void DestroyNetMsgStructs()
     delete reinterpret_cast<NetMsg::SVC_GetCvarValue*>(netDataStructs[31]);
 }
 
-void ParsePacket(uint8_t* packet, size_t length, SourceGameContext& context, IDemoWriter* writer)
+static void CreateDemMsgStructs(void* demDataStructs[9])
+{
+    demDataStructs[0] = new DemMsg::Dem_Unknown();
+    demDataStructs[1] = new DemMsg::Dem_Packet();
+    demDataStructs[2] = new DemMsg::Dem_Packet();
+    demDataStructs[3] = new DemMsg::Dem_SyncTick();
+    demDataStructs[4] = new DemMsg::Dem_ConsoleCmd();
+    demDataStructs[5] = new DemMsg::Dem_UserCmd();
+    demDataStructs[6] = new DemMsg::Dem_DataTables();
+    demDataStructs[7] = new DemMsg::Dem_Stop();
+    demDataStructs[8] = new DemMsg::Dem_StringTables();
+}
+
+static void DestroyDemMsgStructs(void* demDataStructs[9])
+{
+    delete reinterpret_cast<DemMsg::Dem_Unknown*>(demDataStructs[0]);
+    delete reinterpret_cast<DemMsg::Dem_Packet*>(demDataStructs[1]);
+    delete reinterpret_cast<DemMsg::Dem_Packet*>(demDataStructs[2]);
+    delete reinterpret_cast<DemMsg::Dem_SyncTick*>(demDataStructs[3]);
+    delete reinterpret_cast<DemMsg::Dem_ConsoleCmd*>(demDataStructs[4]);
+    delete reinterpret_cast<DemMsg::Dem_UserCmd*>(demDataStructs[5]);
+    delete reinterpret_cast<DemMsg::Dem_DataTables*>(demDataStructs[6]);
+    delete reinterpret_cast<DemMsg::Dem_Stop*>(demDataStructs[7]);
+    delete reinterpret_cast<DemMsg::Dem_StringTables*>(demDataStructs[8]);
+}
+
+void ParsePacket(uint8_t* packet, size_t length,
+                 SourceGameContext& context, IDemoWriter* writer,
+                 void* netDataStructs[32])
 {
     assert(length <= NET_MAX_PAYLOAD);
     bf_read bitbuf(packet, length);
@@ -98,60 +127,14 @@ void ParsePacket(uint8_t* packet, size_t length, SourceGameContext& context, IDe
     }
 }
 
-void ReadStringTable(bf_read& bitbuf)
-{
-    std::vector<uint8_t> data;
-    size_t numEntries = bitbuf.ReadWord();
-    char entryName[4096];
-    while (numEntries--)
-    {
-        bitbuf.ReadString(entryName, sizeof(entryName));
-        printf("%s ", entryName);
-        if (bitbuf.ReadOneBit() != 0)
-        {
-            size_t numDataBytes = bitbuf.ReadWord();
-            data.resize(numDataBytes);
-            bitbuf.ReadBytes(data.data(), numDataBytes);
-            printf("bytes %i ", numDataBytes);
-        }
-        printf("\n");
-    }
-    if (bitbuf.ReadOneBit() == 0)
-    {
-        return;
-    }
-    while (numEntries--)
-    {
-        bitbuf.ReadString(entryName, sizeof(entryName));
-        printf("%s ", entryName);
-        if (bitbuf.ReadOneBit() != 0)
-        {
-            size_t numDataBytes = bitbuf.ReadWord();
-            data.resize(numDataBytes);
-            bitbuf.ReadBytes(data.data(), numDataBytes);
-            printf("bytes %i ", numDataBytes);
-        }
-        printf("\n");
-    }
-}
-
-void ParseStringTables(uint8_t* stringtables, size_t length, SourceGameContext& context, IDemoWriter* writer)
-{
-    bf_read bitbuf(stringtables, length);
-    size_t numTables = bitbuf.ReadByte();
-    char tablename[256];
-    while (numTables--)
-    {
-        bitbuf.ReadString(tablename, sizeof(tablename));
-        printf("stringtable %s\n", tablename);
-        ReadStringTable(bitbuf);
-    }
-}
-
 void DemoReader::ProcessDem(void* inputFp, IDemoWriter* writer)
 {
-    CreateNetMsgStructs();
+    void* netDataStructs[32];
-    SourceGameContext context;
+    void* demDataStructs[9];
+    CreateNetMsgStructs(netDataStructs);
+    CreateDemMsgStructs(demDataStructs);
+
+    SourceGameContext context = SourceGameContext();
     DemoFileReader reader(reinterpret_cast<FILE*>(inputFp));
     {
         demoheader_t header;
@@ -160,62 +143,24 @@ void DemoReader::ProcessDem(void* inputFp, IDemoWriter* writer)
         context.protocol = header.networkprotocol;
     }
 
-    democmdinfo_t cmdInfo;
     CommandPacket packet;
-    packet.cmdInfo = &cmdInfo;
-    std::unique_ptr<uint8_t[]> buffer(new uint8_t[NET_MAX_PAYLOAD]);
     do
     {
         size_t rawDataSize = 0;
         reader.ReadCmdHeader(packet.cmd, packet.tick);
-        if (packet.cmd == dem_packet || packet.cmd == dem_signon)
+        packet.data = demDataStructs[packet.cmd];
-        {
+        DemHandlers::DemMsg_FileRead(packet.cmd, reader, packet.data);
-            reader.ReadCmdInfo(*packet.cmdInfo);
-            reader.ReadSequenceInfo(packet.sequenceInfo1, packet.sequenceInfo2);
-            assert(packet.sequenceInfo1 == packet.sequenceInfo2);
-        }
 
         writer->StartCommandPacket(packet);
-        switch (packet.cmd)
+        if (packet.cmd == dem_packet || packet.cmd == dem_signon)
         {
-        case dem_signon:
+            Array<uint8_t> buffer = reader.ReadRawData(NET_MAX_PAYLOAD);
-        case dem_packet:
+            ParsePacket(buffer.begin(), buffer.length(), context, writer, netDataStructs);
-            rawDataSize = reader.ReadRawData(buffer.get(), NET_MAX_PAYLOAD);
-            ParsePacket(buffer.get(), rawDataSize, context, writer);
-            break;
-        case dem_synctick:
-            // nothing
-            break;
-        case dem_consolecmd:
-            reader.ReadRawData(nullptr, 1024);
-            break;
-        case dem_usercmd:
-            {
-                int32_t sequenceNum;
-                reader.ReadUserCmd(sequenceNum, buffer.get(), 256);
-            }
-            break;
-        case dem_datatables:
-            // TODO: datatables
-            reader.ReadRawData(nullptr, 64*1024);
-            break;
-        case dem_stop:
-            // TODO assert frame and tick numbers
-            break;
-        case dem_stringtables:
-            {
-                uint8_t* stringtables = reinterpret_cast<uint8_t*>(malloc(MAX_STRINGTABLE_DATA));
-                size_t length = reader.ReadRawData(stringtables, MAX_STRINGTABLE_DATA);
-                ParseStringTables(stringtables, length, context, writer);
-                free(stringtables);
-            }
-            break;
-        default:
-            assert(false);
-            break;
         }
         writer->EndCommandPacket();
     } while (packet.cmd != dem_stop);
     writer->EndWriting();
-    DestroyNetMsgStructs();
+
+    DestroyDemMsgStructs(demDataStructs);
+    DestroyNetMsgStructs(netDataStructs);
 }

demboyz/idemowriter.h

@@ -14,9 +14,7 @@ struct CommandPacket
     unsigned char cmd;
     int32_t tick;
 
-    democmdinfo_t* cmdInfo;
+    void* data;
-    int32_t sequenceInfo1;
-    int32_t sequenceInfo2;
 };
 
 struct NetPacket