From 3df00cd3f57ef12ab0d4078045b88585a69c2072 Mon Sep 17 00:00:00 2001
From: Lioncash <mathew1800@gmail.com>
Date: Tue, 24 Jun 2014 16:41:12 -0400
Subject: [PATCH] Fix dragging and dropping savestates in the render window

---
 Source/Core/DolphinWX/Frame.cpp | 26 +++++++++++++++++++++++---
 Source/Core/DolphinWX/Frame.h   |  1 +
 2 files changed, 24 insertions(+), 3 deletions(-)

diff --git a/Source/Core/DolphinWX/Frame.cpp b/Source/Core/DolphinWX/Frame.cpp
index 64cb5a2568..5284eefd57 100644
--- a/Source/Core/DolphinWX/Frame.cpp
+++ b/Source/Core/DolphinWX/Frame.cpp
@@ -15,6 +15,7 @@
 #endif
 
 #include <cstddef>
+#include <fstream>
 #include <string>
 #include <utility>
 #include <vector>
@@ -152,6 +153,7 @@ void CRenderFrame::OnDropFiles(wxDropFilesEvent& event)
 		return;
 
 	wxFileName file = event.GetFiles()[0];
+	const std::string filepath = WxStrToStr(file.GetFullPath());
 
 	if (file.GetExt() == "dtm")
 	{
@@ -165,19 +167,37 @@ void CRenderFrame::OnDropFiles(wxDropFilesEvent& event)
 			main_frame->GetMenuBar()->FindItem(IDM_RECORDREADONLY)->Check(true);
 		}
 
-		if (Movie::PlayInput(WxStrToStr(file.GetFullPath())))
+		if (Movie::PlayInput(filepath))
 			main_frame->BootGame("");
 	}
 	else if (!Core::IsRunning())
 	{
-		main_frame->BootGame(WxStrToStr(file.GetFullPath()));
+		main_frame->BootGame(filepath);
+	}
+	else if (IsValidSavestateDropped(filepath) && Core::IsRunning())
+	{
+		State::LoadAs(filepath);
 	}
 	else
 	{
-		DVDInterface::ChangeDisc(WxStrToStr(file.GetFullPath()));
+		DVDInterface::ChangeDisc(filepath);
 	}
 }
 
+bool CRenderFrame::IsValidSavestateDropped(const std::string& filepath)
+{
+	const int game_id_length = 6;
+	std::ifstream file(filepath, std::ios::in | std::ios::binary);
+
+	if (!file)
+		return false;
+
+	std::string internal_game_id(game_id_length, ' ');
+	file.read(&internal_game_id[0], game_id_length);
+
+	return internal_game_id == SConfig::GetInstance().m_LocalCoreStartupParameter.GetUniqueID();
+}
+
 #ifdef _WIN32
 WXLRESULT CRenderFrame::MSWWindowProc(WXUINT nMsg, WXWPARAM wParam, WXLPARAM lParam)
 {
diff --git a/Source/Core/DolphinWX/Frame.h b/Source/Core/DolphinWX/Frame.h
index f8a260206a..32292c1cb4 100644
--- a/Source/Core/DolphinWX/Frame.h
+++ b/Source/Core/DolphinWX/Frame.h
@@ -80,6 +80,7 @@ class CRenderFrame : public wxFrame
 
 	private:
 		void OnDropFiles(wxDropFilesEvent& event);
+		static bool IsValidSavestateDropped(const std::string& filepath);
 		#ifdef _WIN32
 			// Receive WndProc messages
 			WXLRESULT MSWWindowProc(WXUINT nMsg, WXWPARAM wParam, WXLPARAM lParam);