From 818f9fbaf45bc710bcacc5154365d0db0bedef3b Mon Sep 17 00:00:00 2001
From: meep-eep <meep-eep@8092fc87-c524-0410-9efc-e669fe64eaf9>
Date: Sun, 29 Oct 2006 04:53:36 +0000
Subject: [PATCH] More thorough checksums. Incompatible with earlier builds.

git-svn-id: svn://svn.code.sf.net/p/sc2/code/trunk@2480 8092fc87-c524-0410-9efc-e669fe64eaf9
---
 sc2/src/sc2code/battle.c           |  2 +-
 sc2/src/sc2code/netplay/checksum.c | 49 +++++++++++++++++++++++++++---
 sc2/src/sc2code/netplay/checksum.h | 10 +++++-
 3 files changed, 55 insertions(+), 6 deletions(-)

diff --git a/sc2/src/sc2code/battle.c b/sc2/src/sc2code/battle.c
index d37f4bb97..9f4848d22 100644
--- a/sc2/src/sc2code/battle.c
+++ b/sc2/src/sc2code/battle.c
@@ -258,7 +258,7 @@ DoBattle (BATTLE_STATE *bs)
 		Checksum checksum;
 
 		crc_init(&state);
-		crc_processDispQueue (&state);
+		crc_processState (&state);
 		checksum = (Checksum) crc_finish (&state);
 
 		sendChecksumConnections ((uint32) battleFrameCount,
diff --git a/sc2/src/sc2code/netplay/checksum.c b/sc2/src/sc2code/netplay/checksum.c
index 51d42bb48..78237ec28 100644
--- a/sc2/src/sc2code/netplay/checksum.c
+++ b/sc2/src/sc2code/netplay/checksum.c
@@ -29,6 +29,7 @@
 #include "netconnection.h"
 #include "netmelee.h"
 #include "libs/log.h"
+#include "libs/mathlib.h"
 #include "libs/misc.h"
 
 
@@ -61,7 +62,6 @@ crc_processVELOCITY_DESC(crc_State *state, const VELOCITY_DESC *val) {
 #endif
 }
 
-#if 0
 void
 crc_processPOINT(crc_State *state, const POINT *val) {
 #ifdef DUMP_CRC_OPS
@@ -74,6 +74,7 @@ crc_processPOINT(crc_State *state, const POINT *val) {
 #endif
 }
 
+#if 0
 void
 crc_processSTAMP(crc_State *state, const STAMP *val) {
 #ifdef DUMP_CRC_OPS
@@ -99,6 +100,11 @@ crc_processINTERSECT_CONTROL(crc_State *state, const INTERSECT_CONTROL *val) {
 }
 #endif
 
+void
+crc_processSTATE(crc_State *state, const STATE *val) {
+	crc_processPOINT(state, &val->location);
+}
+
 void
 crc_processELEMENT(crc_State *state, const ELEMENT *val) {
 #ifdef DUMP_CRC_OPS
@@ -124,6 +130,8 @@ crc_processELEMENT(crc_State *state, const ELEMENT *val) {
 
 	crc_processBYTE(state, val->thrust_wait);
 	crc_processVELOCITY_DESC(state, &val->velocity);
+	crc_processSTATE(state, &val->current);
+	crc_processSTATE(state, &val->next);
 #ifdef DUMP_CRC_OPS
 	log_add(log_Debug, "END   crc_processELEMENT().");
 #endif
@@ -136,8 +144,7 @@ crc_processDispQueue(crc_State *state) {
 
 #ifdef DUMP_CRC_OPS
 	size_t i = 0;
-	log_add(log_Debug, "START crc_processDispQueue() (frame %u).",
-			battleFrameCount);
+	log_add(log_Debug, "START crc_processDispQueue().");
 #endif
 	for (element = GetHeadElement(); element != 0; element = nextElement) {
 		ELEMENTPTR elementPtr;
@@ -156,7 +163,41 @@ crc_processDispQueue(crc_State *state) {
 #endif
 	}
 #ifdef DUMP_CRC_OPS
-	log_add(log_Debug, "END   crc_processDispQueue() (frame %u).",
+	log_add(log_Debug, "END   crc_processDispQueue().");
+#endif
+}
+
+void
+crc_processRNG(crc_State *state) {
+	DWORD seed;
+
+#ifdef DUMP_CRC_OPS
+	log_add(log_Debug, "START crc_processRNG().");
+#endif
+
+	seed = TFB_SeedRandom(0);
+			// This modifies the seed too.
+	crc_processDWORD(state, seed);
+	TFB_SeedRandom(seed);
+			// Restore the old seed.
+
+#ifdef DUMP_CRC_OPS
+	log_add(log_Debug, "END   crc_processRNG().");
+#endif
+}
+
+void
+crc_processState(crc_State *state) {
+#ifdef DUMP_CRC_OPS
+	log_add(log_Debug, "--------------------\n"
+			"START crc_processState() (frame %u).", battleFrameCount);
+#endif
+
+	crc_processRNG(state);
+	crc_processDispQueue(state);
+
+#ifdef DUMP_CRC_OPS
+	log_add(log_Debug, "END   crc_processState() (frame %u).",
 			battleFrameCount);
 #endif
 }
diff --git a/sc2/src/sc2code/netplay/checksum.h b/sc2/src/sc2code/netplay/checksum.h
index 128ac76bb..3cbec4377 100644
--- a/sc2/src/sc2code/netplay/checksum.h
+++ b/sc2/src/sc2code/netplay/checksum.h
@@ -50,6 +50,11 @@ crc_processBYTE(crc_State *state, BYTE val) {
 	crc_processUint8(state, (uint8) val);
 }
 
+static inline void
+crc_processDWORD(crc_State *state, DWORD val) {
+	crc_processUint32(state, (uint32) val);
+}
+
 static inline void
 crc_processCOORD(crc_State *state, COORD val) {
 	crc_processUint16(state, (uint16) val);
@@ -64,14 +69,17 @@ crc_processTIME_VALUE(crc_State *state, const TIME_VALUE val) {
 
 void crc_processEXTENT(crc_State *state, const EXTENT *val);
 void crc_processVELOCITY_DESC(crc_State *state, const VELOCITY_DESC *val);
-#if 0
 void crc_processPOINT(crc_State *state, const POINT *val);
+#if 0
 void crc_processSTAMP(crc_State *state, const STAMP *val);
 void crc_processINTERSECT_CONTROL(crc_State *state,
 		const INTERSECT_CONTROL *val);
 #endif
+void crc_processSTATE(crc_State *state, const STATE *val);
 void crc_processELEMENT(crc_State *state, const ELEMENT *val);
 void crc_processDispQueue(crc_State *state);
+void crc_processRNG(crc_State *state);
+void crc_processState(crc_State *state);
 
 
 void initChecksumBuffers(void);