mirror of
https://github.com/dolphin-emu/dolphin.git
synced 2025-02-01 03:32:58 +00:00
bf70026728
CARDUCode, GBAUCode, and INITUCode previously didn't have an implementation of it. In practice it's unlikely that this caused an issue, since these uCodes are only active for a few frames at most, but now that GBAUCode doesn't have global state, we can implement it there. I also implemented it for CARDUCode, although our CARDUCode implementation does not have all states handled yet - this is simply future-proofing so that when the card uCode is properly implemented, the save state version does not need to be bumped. INITUCode does not have any state to save, though.
777 lines
22 KiB
C++
777 lines
22 KiB
C++
// Copyright 2008 Dolphin Emulator Project
|
|
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
#include "Core/State.h"
|
|
|
|
#include <lzo/lzo1x.h>
|
|
#include <map>
|
|
#include <mutex>
|
|
#include <string>
|
|
#include <thread>
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
#include <fmt/format.h>
|
|
|
|
#include "Common/ChunkFile.h"
|
|
#include "Common/CommonTypes.h"
|
|
#include "Common/Event.h"
|
|
#include "Common/FileUtil.h"
|
|
#include "Common/IOFile.h"
|
|
#include "Common/MsgHandler.h"
|
|
#include "Common/ScopeGuard.h"
|
|
#include "Common/Thread.h"
|
|
#include "Common/Timer.h"
|
|
#include "Common/Version.h"
|
|
|
|
#include "Core/ConfigManager.h"
|
|
#include "Core/Core.h"
|
|
#include "Core/CoreTiming.h"
|
|
#include "Core/GeckoCode.h"
|
|
#include "Core/HW/HW.h"
|
|
#include "Core/HW/Memmap.h"
|
|
#include "Core/HW/Wiimote.h"
|
|
#include "Core/Host.h"
|
|
#include "Core/Movie.h"
|
|
#include "Core/NetPlayClient.h"
|
|
#include "Core/PowerPC/PowerPC.h"
|
|
|
|
#include "VideoCommon/FrameDump.h"
|
|
#include "VideoCommon/OnScreenDisplay.h"
|
|
#include "VideoCommon/VideoBackendBase.h"
|
|
|
|
namespace State
|
|
{
|
|
#if defined(__LZO_STRICT_16BIT)
|
|
static const u32 IN_LEN = 8 * 1024u;
|
|
#elif defined(LZO_ARCH_I086) && !defined(LZO_HAVE_MM_HUGE_ARRAY)
|
|
static const u32 IN_LEN = 60 * 1024u;
|
|
#else
|
|
static const u32 IN_LEN = 128 * 1024u;
|
|
#endif
|
|
|
|
static const u32 OUT_LEN = IN_LEN + (IN_LEN / 16) + 64 + 3;
|
|
|
|
static unsigned char __LZO_MMODEL out[OUT_LEN];
|
|
|
|
#define HEAP_ALLOC(var, size) \
|
|
lzo_align_t __LZO_MMODEL var[((size) + (sizeof(lzo_align_t) - 1)) / sizeof(lzo_align_t)]
|
|
|
|
static HEAP_ALLOC(wrkmem, LZO1X_1_MEM_COMPRESS);
|
|
|
|
static AfterLoadCallbackFunc s_on_after_load_callback;
|
|
|
|
// Temporary undo state buffer
|
|
static std::vector<u8> g_undo_load_buffer;
|
|
static std::vector<u8> g_current_buffer;
|
|
static std::mutex s_load_or_save_in_progress_mutex;
|
|
|
|
static std::mutex g_cs_undo_load_buffer;
|
|
static std::mutex g_cs_current_buffer;
|
|
static Common::Event g_compressAndDumpStateSyncEvent;
|
|
|
|
static std::recursive_mutex g_save_thread_mutex;
|
|
static std::thread g_save_thread;
|
|
|
|
// Don't forget to increase this after doing changes on the savestate system
|
|
constexpr u32 STATE_VERSION = 148; // Last changed in PR 10768
|
|
|
|
// Maps savestate versions to Dolphin versions.
|
|
// Versions after 42 don't need to be added to this list,
|
|
// because they save the exact Dolphin version to savestates.
|
|
static const std::map<u32, std::pair<std::string, std::string>> s_old_versions = {
|
|
// The 16 -> 17 change modified the size of StateHeader,
|
|
// so versions older than that can't even be decompressed anymore
|
|
{17, {"3.5-1311", "3.5-1364"}}, {18, {"3.5-1366", "3.5-1371"}}, {19, {"3.5-1372", "3.5-1408"}},
|
|
{20, {"3.5-1409", "4.0-704"}}, {21, {"4.0-705", "4.0-889"}}, {22, {"4.0-905", "4.0-1871"}},
|
|
{23, {"4.0-1873", "4.0-1900"}}, {24, {"4.0-1902", "4.0-1919"}}, {25, {"4.0-1921", "4.0-1936"}},
|
|
{26, {"4.0-1939", "4.0-1959"}}, {27, {"4.0-1961", "4.0-2018"}}, {28, {"4.0-2020", "4.0-2291"}},
|
|
{29, {"4.0-2293", "4.0-2360"}}, {30, {"4.0-2362", "4.0-2628"}}, {31, {"4.0-2632", "4.0-3331"}},
|
|
{32, {"4.0-3334", "4.0-3340"}}, {33, {"4.0-3342", "4.0-3373"}}, {34, {"4.0-3376", "4.0-3402"}},
|
|
{35, {"4.0-3409", "4.0-3603"}}, {36, {"4.0-3610", "4.0-4480"}}, {37, {"4.0-4484", "4.0-4943"}},
|
|
{38, {"4.0-4963", "4.0-5267"}}, {39, {"4.0-5279", "4.0-5525"}}, {40, {"4.0-5531", "4.0-5809"}},
|
|
{41, {"4.0-5811", "4.0-5923"}}, {42, {"4.0-5925", "4.0-5946"}}};
|
|
|
|
enum
|
|
{
|
|
STATE_NONE = 0,
|
|
STATE_SAVE = 1,
|
|
STATE_LOAD = 2,
|
|
};
|
|
|
|
static bool s_use_compression = true;
|
|
|
|
void EnableCompression(bool compression)
|
|
{
|
|
s_use_compression = compression;
|
|
}
|
|
|
|
// Returns true if state version matches current Dolphin state version, false otherwise.
|
|
static bool DoStateVersion(PointerWrap& p, std::string* version_created_by)
|
|
{
|
|
u32 version = STATE_VERSION;
|
|
{
|
|
static const u32 COOKIE_BASE = 0xBAADBABE;
|
|
u32 cookie = version + COOKIE_BASE;
|
|
p.Do(cookie);
|
|
version = cookie - COOKIE_BASE;
|
|
}
|
|
|
|
*version_created_by = Common::GetScmRevStr();
|
|
if (version > 42)
|
|
p.Do(*version_created_by);
|
|
else
|
|
version_created_by->clear();
|
|
|
|
if (version != STATE_VERSION)
|
|
{
|
|
if (version_created_by->empty() && s_old_versions.count(version))
|
|
{
|
|
// The savestate is from an old version that doesn't
|
|
// save the Dolphin version number to savestates, but
|
|
// by looking up the savestate version number, it is possible
|
|
// to know approximately which Dolphin version was used.
|
|
|
|
std::pair<std::string, std::string> version_range = s_old_versions.find(version)->second;
|
|
std::string oldest_version = version_range.first;
|
|
std::string newest_version = version_range.second;
|
|
|
|
*version_created_by = "Dolphin " + oldest_version + " - " + newest_version;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
p.DoMarker("Version");
|
|
return true;
|
|
}
|
|
|
|
static void DoState(PointerWrap& p)
|
|
{
|
|
std::string version_created_by;
|
|
if (!DoStateVersion(p, &version_created_by))
|
|
{
|
|
const std::string message =
|
|
version_created_by.empty() ?
|
|
"This savestate was created using an incompatible version of Dolphin" :
|
|
"This savestate was created using the incompatible version " + version_created_by;
|
|
Core::DisplayMessage(message, OSD::Duration::NORMAL);
|
|
p.SetMeasureMode();
|
|
return;
|
|
}
|
|
|
|
bool is_wii = SConfig::GetInstance().bWii || SConfig::GetInstance().m_is_mios;
|
|
const bool is_wii_currently = is_wii;
|
|
p.Do(is_wii);
|
|
if (is_wii != is_wii_currently)
|
|
{
|
|
OSD::AddMessage(fmt::format("Cannot load a savestate created under {} mode in {} mode",
|
|
is_wii ? "Wii" : "GC", is_wii_currently ? "Wii" : "GC"),
|
|
OSD::Duration::NORMAL, OSD::Color::RED);
|
|
p.SetMeasureMode();
|
|
return;
|
|
}
|
|
|
|
// Check to make sure the emulated memory sizes are the same as the savestate
|
|
u32 state_mem1_size = Memory::GetRamSizeReal();
|
|
u32 state_mem2_size = Memory::GetExRamSizeReal();
|
|
p.Do(state_mem1_size);
|
|
p.Do(state_mem2_size);
|
|
if (state_mem1_size != Memory::GetRamSizeReal() || state_mem2_size != Memory::GetExRamSizeReal())
|
|
{
|
|
OSD::AddMessage(fmt::format("Memory size mismatch!\n"
|
|
"Current | MEM1 {:08X} ({:3}MB) MEM2 {:08X} ({:3}MB)\n"
|
|
"State | MEM1 {:08X} ({:3}MB) MEM2 {:08X} ({:3}MB)",
|
|
Memory::GetRamSizeReal(), Memory::GetRamSizeReal() / 0x100000U,
|
|
Memory::GetExRamSizeReal(), Memory::GetExRamSizeReal() / 0x100000U,
|
|
state_mem1_size, state_mem1_size / 0x100000U, state_mem2_size,
|
|
state_mem2_size / 0x100000U));
|
|
p.SetMeasureMode();
|
|
return;
|
|
}
|
|
|
|
// Movie must be done before the video backend, because the window is redrawn in the video backend
|
|
// state load, and the frame number must be up-to-date.
|
|
Movie::DoState(p);
|
|
p.DoMarker("Movie");
|
|
|
|
// Begin with video backend, so that it gets a chance to clear its caches and writeback modified
|
|
// things to RAM
|
|
g_video_backend->DoState(p);
|
|
p.DoMarker("video_backend");
|
|
|
|
PowerPC::DoState(p);
|
|
p.DoMarker("PowerPC");
|
|
// CoreTiming needs to be restored before restoring Hardware because
|
|
// the controller code might need to schedule an event if the controller has changed.
|
|
CoreTiming::DoState(p);
|
|
p.DoMarker("CoreTiming");
|
|
HW::DoState(p);
|
|
p.DoMarker("HW");
|
|
if (SConfig::GetInstance().bWii)
|
|
Wiimote::DoState(p);
|
|
p.DoMarker("Wiimote");
|
|
Gecko::DoState(p);
|
|
p.DoMarker("Gecko");
|
|
}
|
|
|
|
void LoadFromBuffer(std::vector<u8>& buffer)
|
|
{
|
|
if (NetPlay::IsNetPlayRunning())
|
|
{
|
|
OSD::AddMessage("Loading savestates is disabled in Netplay to prevent desyncs");
|
|
return;
|
|
}
|
|
|
|
Core::RunOnCPUThread(
|
|
[&] {
|
|
u8* ptr = buffer.data();
|
|
PointerWrap p(&ptr, buffer.size(), PointerWrap::Mode::Read);
|
|
DoState(p);
|
|
},
|
|
true);
|
|
}
|
|
|
|
void SaveToBuffer(std::vector<u8>& buffer)
|
|
{
|
|
Core::RunOnCPUThread(
|
|
[&] {
|
|
u8* ptr = nullptr;
|
|
PointerWrap p_measure(&ptr, 0, PointerWrap::Mode::Measure);
|
|
|
|
DoState(p_measure);
|
|
const size_t buffer_size = reinterpret_cast<size_t>(ptr);
|
|
buffer.resize(buffer_size);
|
|
|
|
ptr = buffer.data();
|
|
PointerWrap p(&ptr, buffer_size, PointerWrap::Mode::Write);
|
|
DoState(p);
|
|
},
|
|
true);
|
|
}
|
|
|
|
// return state number not in map
|
|
static int GetEmptySlot(std::map<double, int> m)
|
|
{
|
|
for (int i = 1; i <= (int)NUM_STATES; i++)
|
|
{
|
|
bool found = false;
|
|
for (auto& p : m)
|
|
{
|
|
if (p.second == i)
|
|
{
|
|
found = true;
|
|
break;
|
|
}
|
|
}
|
|
if (!found)
|
|
return i;
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
// Arbitrarily chosen value (38 years) that is subtracted in GetSystemTimeAsDouble()
|
|
// to increase sub-second precision of the resulting double timestamp
|
|
static constexpr int DOUBLE_TIME_OFFSET = (38 * 365 * 24 * 60 * 60);
|
|
|
|
static double GetSystemTimeAsDouble()
|
|
{
|
|
// FYI: std::chrono::system_clock epoch is not required to be 1970 until c++20.
|
|
// We will however assume time_t IS unix time.
|
|
using Clock = std::chrono::system_clock;
|
|
|
|
// TODO: Use this on switch to c++20:
|
|
// const auto since_epoch = Clock::now().time_since_epoch();
|
|
const auto unix_epoch = Clock::from_time_t({});
|
|
const auto since_epoch = Clock::now() - unix_epoch;
|
|
|
|
const auto since_double_time_epoch = since_epoch - std::chrono::seconds(DOUBLE_TIME_OFFSET);
|
|
return std::chrono::duration_cast<std::chrono::duration<double>>(since_double_time_epoch).count();
|
|
}
|
|
|
|
static std::string SystemTimeAsDoubleToString(double time)
|
|
{
|
|
// revert adjustments from GetSystemTimeAsDouble() to get a normal Unix timestamp again
|
|
time_t seconds = (time_t)time + DOUBLE_TIME_OFFSET;
|
|
tm* localTime = localtime(&seconds);
|
|
|
|
#ifdef _WIN32
|
|
wchar_t tmp[32] = {};
|
|
wcsftime(tmp, std::size(tmp), L"%x %X", localTime);
|
|
return WStringToUTF8(tmp);
|
|
#else
|
|
char tmp[32] = {};
|
|
strftime(tmp, sizeof(tmp), "%x %X", localTime);
|
|
return tmp;
|
|
#endif
|
|
}
|
|
|
|
static std::string MakeStateFilename(int number);
|
|
|
|
// read state timestamps
|
|
static std::map<double, int> GetSavedStates()
|
|
{
|
|
StateHeader header;
|
|
std::map<double, int> m;
|
|
for (int i = 1; i <= (int)NUM_STATES; i++)
|
|
{
|
|
std::string filename = MakeStateFilename(i);
|
|
if (File::Exists(filename))
|
|
{
|
|
if (ReadHeader(filename, header))
|
|
{
|
|
double d = GetSystemTimeAsDouble() - header.time;
|
|
|
|
// increase time until unique value is obtained
|
|
while (m.find(d) != m.end())
|
|
d += .001;
|
|
|
|
m.emplace(d, i);
|
|
}
|
|
}
|
|
}
|
|
return m;
|
|
}
|
|
|
|
struct CompressAndDumpState_args
|
|
{
|
|
std::vector<u8>* buffer_vector = nullptr;
|
|
std::mutex* buffer_mutex = nullptr;
|
|
std::string filename;
|
|
bool wait = false;
|
|
};
|
|
|
|
static void CompressAndDumpState(CompressAndDumpState_args save_args)
|
|
{
|
|
std::lock_guard lk(*save_args.buffer_mutex);
|
|
|
|
// ScopeGuard is used here to ensure that g_compressAndDumpStateSyncEvent.Set()
|
|
// will be called and that it will happen after the IOFile is closed.
|
|
// Both ScopeGuard's and IOFile's finalization occur at respective object destruction time.
|
|
// As Local (stack) objects are destructed in the reverse order of construction and "ScopeGuard
|
|
// on_exit"
|
|
// is created before the "IOFile f", it is guaranteed that the file will be finalized before
|
|
// the ScopeGuard's finalization (i.e. "g_compressAndDumpStateSyncEvent.Set()" call).
|
|
Common::ScopeGuard on_exit([]() { g_compressAndDumpStateSyncEvent.Set(); });
|
|
// If it is not required to wait, we call finalizer early (and it won't be called again at
|
|
// destruction).
|
|
if (!save_args.wait)
|
|
on_exit.Exit();
|
|
|
|
const u8* const buffer_data = &(*(save_args.buffer_vector))[0];
|
|
const size_t buffer_size = (save_args.buffer_vector)->size();
|
|
std::string& filename = save_args.filename;
|
|
|
|
// For easy debugging
|
|
Common::SetCurrentThreadName("SaveState thread");
|
|
|
|
// Moving to last overwritten save-state
|
|
if (File::Exists(filename))
|
|
{
|
|
if (File::Exists(File::GetUserPath(D_STATESAVES_IDX) + "lastState.sav"))
|
|
File::Delete((File::GetUserPath(D_STATESAVES_IDX) + "lastState.sav"));
|
|
if (File::Exists(File::GetUserPath(D_STATESAVES_IDX) + "lastState.sav.dtm"))
|
|
File::Delete((File::GetUserPath(D_STATESAVES_IDX) + "lastState.sav.dtm"));
|
|
|
|
if (!File::Rename(filename, File::GetUserPath(D_STATESAVES_IDX) + "lastState.sav"))
|
|
Core::DisplayMessage("Failed to move previous state to state undo backup", 1000);
|
|
else if (File::Exists(filename + ".dtm"))
|
|
File::Rename(filename + ".dtm", File::GetUserPath(D_STATESAVES_IDX) + "lastState.sav.dtm");
|
|
}
|
|
|
|
if ((Movie::IsMovieActive()) && !Movie::IsJustStartingRecordingInputFromSaveState())
|
|
Movie::SaveRecording(filename + ".dtm");
|
|
else if (!Movie::IsMovieActive())
|
|
File::Delete(filename + ".dtm");
|
|
|
|
File::IOFile f(filename, "wb");
|
|
if (!f)
|
|
{
|
|
Core::DisplayMessage("Could not save state", 2000);
|
|
return;
|
|
}
|
|
|
|
// Setting up the header
|
|
StateHeader header{};
|
|
SConfig::GetInstance().GetGameID().copy(header.gameID, std::size(header.gameID));
|
|
header.size = s_use_compression ? (u32)buffer_size : 0;
|
|
header.time = GetSystemTimeAsDouble();
|
|
|
|
f.WriteArray(&header, 1);
|
|
|
|
if (header.size != 0) // non-zero header size means the state is compressed
|
|
{
|
|
lzo_uint i = 0;
|
|
while (true)
|
|
{
|
|
lzo_uint32 cur_len = 0;
|
|
lzo_uint out_len = 0;
|
|
|
|
if ((i + IN_LEN) >= buffer_size)
|
|
{
|
|
cur_len = (lzo_uint32)(buffer_size - i);
|
|
}
|
|
else
|
|
{
|
|
cur_len = IN_LEN;
|
|
}
|
|
|
|
if (lzo1x_1_compress(buffer_data + i, cur_len, out, &out_len, wrkmem) != LZO_E_OK)
|
|
PanicAlertFmtT("Internal LZO Error - compression failed");
|
|
|
|
// The size of the data to write is 'out_len'
|
|
f.WriteArray((lzo_uint32*)&out_len, 1);
|
|
f.WriteBytes(out, out_len);
|
|
|
|
if (cur_len != IN_LEN)
|
|
break;
|
|
|
|
i += cur_len;
|
|
}
|
|
}
|
|
else // uncompressed
|
|
{
|
|
f.WriteBytes(buffer_data, buffer_size);
|
|
}
|
|
|
|
Core::DisplayMessage(fmt::format("Saved State to {}", filename), 2000);
|
|
Host_UpdateMainFrame();
|
|
}
|
|
|
|
void SaveAs(const std::string& filename, bool wait)
|
|
{
|
|
std::unique_lock lk(s_load_or_save_in_progress_mutex, std::try_to_lock);
|
|
if (!lk)
|
|
return;
|
|
|
|
Core::RunOnCPUThread(
|
|
[&] {
|
|
// Measure the size of the buffer.
|
|
u8* ptr = nullptr;
|
|
PointerWrap p_measure(&ptr, 0, PointerWrap::Mode::Measure);
|
|
DoState(p_measure);
|
|
const size_t buffer_size = reinterpret_cast<size_t>(ptr);
|
|
|
|
// Then actually do the write.
|
|
bool is_write_mode;
|
|
{
|
|
std::lock_guard lk2(g_cs_current_buffer);
|
|
g_current_buffer.resize(buffer_size);
|
|
ptr = g_current_buffer.data();
|
|
PointerWrap p(&ptr, buffer_size, PointerWrap::Mode::Write);
|
|
DoState(p);
|
|
is_write_mode = p.IsWriteMode();
|
|
}
|
|
|
|
if (is_write_mode)
|
|
{
|
|
Core::DisplayMessage("Saving State...", 1000);
|
|
|
|
CompressAndDumpState_args save_args;
|
|
save_args.buffer_vector = &g_current_buffer;
|
|
save_args.buffer_mutex = &g_cs_current_buffer;
|
|
save_args.filename = filename;
|
|
save_args.wait = wait;
|
|
|
|
{
|
|
std::lock_guard lk3(g_save_thread_mutex);
|
|
Flush();
|
|
g_save_thread = std::thread(CompressAndDumpState, save_args);
|
|
}
|
|
|
|
g_compressAndDumpStateSyncEvent.Wait();
|
|
}
|
|
else
|
|
{
|
|
// someone aborted the save by changing the mode?
|
|
Core::DisplayMessage("Unable to save: Internal DoState Error", 4000);
|
|
}
|
|
},
|
|
true);
|
|
}
|
|
|
|
bool ReadHeader(const std::string& filename, StateHeader& header)
|
|
{
|
|
Flush();
|
|
File::IOFile f(filename, "rb");
|
|
return f.ReadArray(&header, 1);
|
|
}
|
|
|
|
std::string GetInfoStringOfSlot(int slot, bool translate)
|
|
{
|
|
std::string filename = MakeStateFilename(slot);
|
|
if (!File::Exists(filename))
|
|
return translate ? Common::GetStringT("Empty") : "Empty";
|
|
|
|
State::StateHeader header;
|
|
if (!ReadHeader(filename, header))
|
|
return translate ? Common::GetStringT("Unknown") : "Unknown";
|
|
|
|
return SystemTimeAsDoubleToString(header.time);
|
|
}
|
|
|
|
u64 GetUnixTimeOfSlot(int slot)
|
|
{
|
|
State::StateHeader header;
|
|
if (!ReadHeader(MakeStateFilename(slot), header))
|
|
return 0;
|
|
|
|
constexpr u64 MS_PER_SEC = 1000;
|
|
return static_cast<u64>(header.time * MS_PER_SEC) + (DOUBLE_TIME_OFFSET * MS_PER_SEC);
|
|
}
|
|
|
|
static void LoadFileStateData(const std::string& filename, std::vector<u8>& ret_data)
|
|
{
|
|
Flush();
|
|
File::IOFile f(filename, "rb");
|
|
|
|
StateHeader header;
|
|
if (!f.ReadArray(&header, 1))
|
|
{
|
|
Core::DisplayMessage("State not found", 2000);
|
|
return;
|
|
}
|
|
|
|
if (strncmp(SConfig::GetInstance().GetGameID().c_str(), header.gameID, 6))
|
|
{
|
|
Core::DisplayMessage(fmt::format("State belongs to a different game (ID {})",
|
|
std::string_view{header.gameID, std::size(header.gameID)}),
|
|
2000);
|
|
return;
|
|
}
|
|
|
|
std::vector<u8> buffer;
|
|
|
|
if (header.size != 0) // non-zero size means the state is compressed
|
|
{
|
|
Core::DisplayMessage("Decompressing State...", 500);
|
|
|
|
buffer.resize(header.size);
|
|
|
|
lzo_uint i = 0;
|
|
while (true)
|
|
{
|
|
lzo_uint32 cur_len = 0; // number of bytes to read
|
|
lzo_uint new_len = 0; // number of bytes to write
|
|
|
|
if (!f.ReadArray(&cur_len, 1))
|
|
break;
|
|
|
|
f.ReadBytes(out, cur_len);
|
|
const int res = lzo1x_decompress(out, cur_len, &buffer[i], &new_len, nullptr);
|
|
if (res != LZO_E_OK)
|
|
{
|
|
// This doesn't seem to happen anymore.
|
|
PanicAlertFmtT("Internal LZO Error - decompression failed ({0}) ({1}, {2}) \n"
|
|
"Try loading the state again",
|
|
res, i, new_len);
|
|
return;
|
|
}
|
|
|
|
i += new_len;
|
|
}
|
|
}
|
|
else // uncompressed
|
|
{
|
|
const auto size = static_cast<size_t>(f.GetSize() - sizeof(StateHeader));
|
|
buffer.resize(size);
|
|
|
|
if (!f.ReadBytes(&buffer[0], size))
|
|
{
|
|
PanicAlertFmt("Error reading bytes: {0}", size);
|
|
return;
|
|
}
|
|
}
|
|
|
|
// all good
|
|
ret_data.swap(buffer);
|
|
}
|
|
|
|
void LoadAs(const std::string& filename)
|
|
{
|
|
if (!Core::IsRunning())
|
|
return;
|
|
|
|
if (NetPlay::IsNetPlayRunning())
|
|
{
|
|
OSD::AddMessage("Loading savestates is disabled in Netplay to prevent desyncs");
|
|
return;
|
|
}
|
|
|
|
std::unique_lock lk(s_load_or_save_in_progress_mutex, std::try_to_lock);
|
|
if (!lk)
|
|
return;
|
|
|
|
Core::RunOnCPUThread(
|
|
[&] {
|
|
// Save temp buffer for undo load state
|
|
if (!Movie::IsJustStartingRecordingInputFromSaveState())
|
|
{
|
|
std::lock_guard lk2(g_cs_undo_load_buffer);
|
|
SaveToBuffer(g_undo_load_buffer);
|
|
if (Movie::IsMovieActive())
|
|
Movie::SaveRecording(File::GetUserPath(D_STATESAVES_IDX) + "undo.dtm");
|
|
else if (File::Exists(File::GetUserPath(D_STATESAVES_IDX) + "undo.dtm"))
|
|
File::Delete(File::GetUserPath(D_STATESAVES_IDX) + "undo.dtm");
|
|
}
|
|
|
|
bool loaded = false;
|
|
bool loadedSuccessfully = false;
|
|
|
|
// brackets here are so buffer gets freed ASAP
|
|
{
|
|
std::vector<u8> buffer;
|
|
LoadFileStateData(filename, buffer);
|
|
|
|
if (!buffer.empty())
|
|
{
|
|
u8* ptr = buffer.data();
|
|
PointerWrap p(&ptr, buffer.size(), PointerWrap::Mode::Read);
|
|
DoState(p);
|
|
loaded = true;
|
|
loadedSuccessfully = p.IsReadMode();
|
|
}
|
|
}
|
|
|
|
if (loaded)
|
|
{
|
|
if (loadedSuccessfully)
|
|
{
|
|
Core::DisplayMessage(fmt::format("Loaded state from {}", filename), 2000);
|
|
if (File::Exists(filename + ".dtm"))
|
|
Movie::LoadInput(filename + ".dtm");
|
|
else if (!Movie::IsJustStartingRecordingInputFromSaveState() &&
|
|
!Movie::IsJustStartingPlayingInputFromSaveState())
|
|
Movie::EndPlayInput(false);
|
|
}
|
|
else
|
|
{
|
|
Core::DisplayMessage("The savestate could not be loaded", OSD::Duration::NORMAL);
|
|
|
|
// since we could be in an inconsistent state now (and might crash or whatever), undo.
|
|
UndoLoadState();
|
|
}
|
|
}
|
|
|
|
if (s_on_after_load_callback)
|
|
s_on_after_load_callback();
|
|
},
|
|
true);
|
|
}
|
|
|
|
void SetOnAfterLoadCallback(AfterLoadCallbackFunc callback)
|
|
{
|
|
s_on_after_load_callback = std::move(callback);
|
|
}
|
|
|
|
void Init()
|
|
{
|
|
if (lzo_init() != LZO_E_OK)
|
|
PanicAlertFmtT("Internal LZO Error - lzo_init() failed");
|
|
}
|
|
|
|
void Shutdown()
|
|
{
|
|
Flush();
|
|
|
|
// swapping with an empty vector, rather than clear()ing
|
|
// this gives a better guarantee to free the allocated memory right NOW (as opposed to, actually,
|
|
// never)
|
|
{
|
|
std::lock_guard lk(g_cs_current_buffer);
|
|
std::vector<u8>().swap(g_current_buffer);
|
|
}
|
|
|
|
{
|
|
std::lock_guard lk(g_cs_undo_load_buffer);
|
|
std::vector<u8>().swap(g_undo_load_buffer);
|
|
}
|
|
}
|
|
|
|
static std::string MakeStateFilename(int number)
|
|
{
|
|
return fmt::format("{}{}.s{:02d}", File::GetUserPath(D_STATESAVES_IDX),
|
|
SConfig::GetInstance().GetGameID(), number);
|
|
}
|
|
|
|
void Save(int slot, bool wait)
|
|
{
|
|
SaveAs(MakeStateFilename(slot), wait);
|
|
}
|
|
|
|
void Load(int slot)
|
|
{
|
|
LoadAs(MakeStateFilename(slot));
|
|
}
|
|
|
|
void LoadLastSaved(int i)
|
|
{
|
|
std::map<double, int> savedStates = GetSavedStates();
|
|
|
|
if (i > (int)savedStates.size())
|
|
Core::DisplayMessage("State doesn't exist", 2000);
|
|
else
|
|
{
|
|
std::map<double, int>::iterator it = savedStates.begin();
|
|
std::advance(it, i - 1);
|
|
Load(it->second);
|
|
}
|
|
}
|
|
|
|
// must wait for state to be written because it must know if all slots are taken
|
|
void SaveFirstSaved()
|
|
{
|
|
std::map<double, int> savedStates = GetSavedStates();
|
|
|
|
// save to an empty slot
|
|
if (savedStates.size() < NUM_STATES)
|
|
Save(GetEmptySlot(savedStates), true);
|
|
// overwrite the oldest state
|
|
else
|
|
{
|
|
std::map<double, int>::iterator it = savedStates.begin();
|
|
std::advance(it, savedStates.size() - 1);
|
|
Save(it->second, true);
|
|
}
|
|
}
|
|
|
|
void Flush()
|
|
{
|
|
std::lock_guard lk(g_save_thread_mutex);
|
|
|
|
// If already saving state, wait for it to finish
|
|
if (g_save_thread.joinable())
|
|
g_save_thread.join();
|
|
}
|
|
|
|
// Load the last state before loading the state
|
|
void UndoLoadState()
|
|
{
|
|
std::lock_guard lk(g_cs_undo_load_buffer);
|
|
if (!g_undo_load_buffer.empty())
|
|
{
|
|
if (File::Exists(File::GetUserPath(D_STATESAVES_IDX) + "undo.dtm") || (!Movie::IsMovieActive()))
|
|
{
|
|
LoadFromBuffer(g_undo_load_buffer);
|
|
if (Movie::IsMovieActive())
|
|
Movie::LoadInput(File::GetUserPath(D_STATESAVES_IDX) + "undo.dtm");
|
|
}
|
|
else
|
|
{
|
|
PanicAlertFmtT("No undo.dtm found, aborting undo load state to prevent movie desyncs");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
PanicAlertFmtT("There is nothing to undo!");
|
|
}
|
|
}
|
|
|
|
// Load the state that the last save state overwritten on
|
|
void UndoSaveState()
|
|
{
|
|
LoadAs(File::GetUserPath(D_STATESAVES_IDX) + "lastState.sav");
|
|
}
|
|
|
|
} // namespace State
|