Fixed the load-state memory leak, also fixed various other savestate issues. Found (probably) the cause of load-state crashing Dolphin: BPWrites. Tried to put a CriticalSection around BPWritten but it only causes Dolphin to hang when loading a state. Can someone find why it hangs?

git-svn-id: https://dolphin-emu.googlecode.com/svn/trunk@3873 8ced0084-cf51-0410-be5f-012b33b47a6e

Source/Core/Core/Src/State.cpp

@@ -84,6 +84,8 @@ void DoState(PointerWrap &p)
         pm.GetVideo()->DoState(p.GetPPtr(), p.GetMode());
         pm.GetDSP()->DoState(p.GetPPtr(), p.GetMode());
 		pm.GetPad(0)->DoState(p.GetPPtr(), p.GetMode());
+		if(Core::g_CoreStartupParameter.bWii)
+			pm.GetWiimote(0)->DoState(p.GetPPtr(), p.GetMode());
 	PowerPC::DoState(p);
 	HW::DoState(p);
 	CoreTiming::DoState(p);
@@ -124,6 +126,9 @@ void SaveBufferStateCallback(u64 userdata, int cyclesLate)
 	DoState(p);
 	sz = (size_t)ptr;
 
+	if(*cur_buffer)
+		delete[] (*cur_buffer);
+
 	*cur_buffer = new u8[sz];
 	ptr = *cur_buffer;
 	p.SetMode(PointerWrap::MODE_WRITE);
@@ -155,6 +160,7 @@ THREAD_RETURN CompressAndDumpState(void *pArgs)
 	FILE *f = fopen(filename.c_str(), "wb");
 	if(f == NULL) {
 		Core::DisplayMessage("Could not save state", 2000);
+		delete [] buffer;
 		return 0;
 	}
 
@@ -165,9 +171,6 @@ THREAD_RETURN CompressAndDumpState(void *pArgs)
 	fwrite(&header, sizeof(state_header), 1, f);
 
 	if (bCompressed) {
-		if (lzo_init() != LZO_E_OK)
-			PanicAlert("Internal LZO Error - lzo_init() failed");
-		else {
 		lzo_uint cur_len;
 		lzo_uint i = 0;
 
@@ -188,7 +191,7 @@ THREAD_RETURN CompressAndDumpState(void *pArgs)
 				break;
 			i += cur_len;
 		}
-		}
+
 	} else
 		fwrite(buffer, sz, 1, f);
 
@@ -278,11 +281,13 @@ void LoadStateCallback(u64 userdata, int cyclesLate)
 	if (bCompressedState) {
 		Core::DisplayMessage("Decompressing State...", 500);
 
-		if (lzo_init() != LZO_E_OK)
-			PanicAlert("Internal LZO Error - lzo_init() failed");
-		else {
 		lzo_uint i = 0;
 		buffer = new u8[sz];
+		if(!buffer) {
+			PanicAlert("Error allocating buffer");
+			return;
+		}
+
 		
 		for (;;) {
 			if (fread(&cur_len, 1, sizeof(int), f) == 0)
@@ -291,16 +296,17 @@ void LoadStateCallback(u64 userdata, int cyclesLate)
 
 			int res = lzo1x_decompress(out, cur_len, (buffer + i), &new_len, NULL);
 			if(res != LZO_E_OK) {
-					PanicAlert("Internal LZO Error - decompression failed (%d)\n"
-						"Try loading the state again", res);
+				PanicAlert("Internal LZO Error - decompression failed (%d) (%d, %d) \n"
+					"Try loading the state again", res, i, new_len);
 				fclose(f);
+
+				delete[] buffer;
 				return;
 			}
 	
 			// The size of the data to read to our buffer is 'new_len'
 			i += new_len;
 		}
-		}
 	} else {
 		fseek(f, 0, SEEK_END);
 		sz = (int)(ftell(f) - sizeof(int));
@@ -331,6 +337,9 @@ void State_Init()
 	ev_Save = CoreTiming::RegisterEvent("SaveState", &SaveStateCallback);
 	ev_BufferLoad = CoreTiming::RegisterEvent("LoadBufferState", &LoadBufferStateCallback);
 	ev_BufferSave = CoreTiming::RegisterEvent("SaveBufferState", &SaveBufferStateCallback);
+
+	if (lzo_init() != LZO_E_OK)
+		PanicAlert("Internal LZO Error - lzo_init() failed");
 }
 
 void State_Shutdown()

Source/Core/VideoCommon/Src/BPStructs.cpp

@@ -27,15 +27,20 @@
 #include "OpcodeDecoding.h"
 #include "VertexLoader.h"
 #include "VertexShaderManager.h"
+#include "Thread.h"
 
 using namespace BPFunctions;
 
+// FIXME: Hangs load-state, but should fix graphic-heavy games state loading
+//Common::CriticalSection s_bpCritical;
+
 void BPInit()
 {
     memset(&bpmem, 0, sizeof(bpmem));
     bpmem.bpMask = 0xFFFFFF;
 }
 
+
 // ----------------------------------------------------------------------------------------------------------
 // Write to the Bypass Memory (Bypass Raster State Registers)
 /* ------------------
@@ -73,6 +78,9 @@ void BPWritten(const Bypass& bp)
 	//default: break;
 	//}
 
+	// FIXME: Hangs load-state, but should fix graphic-heavy games state loading
+	//s_bpCritical.Enter();
+
 	FlushPipeline();
     ((u32*)&bpmem)[bp.address] = bp.newvalue;
 	
@@ -606,9 +614,10 @@ void BPWritten(const Bypass& bp)
 				WARN_LOG(VIDEO, "Unknown BP opcode: address = 0x%08x value = 0x%08x", bp.address, bp.newvalue);
 				break;
 			}		
-			
 	}
 
+	// FIXME: Hangs load-state, but should fix graphic-heavy games state loading
+	//s_bpCritical.Leave();
 }
 }
 

Source/Plugins/Plugin_DSP_HLE/Src/MailHandler.cpp

@@ -53,6 +53,7 @@ u16 CMailHandler::ReadDSPMailboxLow()
 	if (!m_Mails.empty())
 	{
 		u16 result = m_Mails.front() & 0xFFFF;
+
         m_Mails.pop();
 
 		Update();