re3/src/save/GenericGameStorage.cpp

399 lines
11 KiB
C++
Raw Normal View History

2020-01-13 23:13:42 +00:00
#include "common.h"
#include "main.h"
#include "patcher.h"
#include "Camera.h"
2020-03-05 09:44:42 +00:00
#include "CarGen.h"
#include "Cranes.h"
2020-01-13 23:13:42 +00:00
#include "Clock.h"
2020-01-27 22:16:38 +00:00
#include "Date.h"
2020-01-13 23:13:42 +00:00
#include "FileMgr.h"
#include "GameLogic.h"
2020-03-05 09:44:42 +00:00
#include "Gangs.h"
2020-01-13 23:13:42 +00:00
#include "Garages.h"
#include "GenericGameStorage.h"
#include "Pad.h"
2020-03-05 09:44:42 +00:00
#include "PathFind.h"
2020-01-13 23:13:42 +00:00
#include "PCSave.h"
2020-03-05 09:44:42 +00:00
#include "Phones.h"
#include "Pickups.h"
2020-01-13 23:13:42 +00:00
#include "PlayerPed.h"
#include "Pools.h"
2020-03-05 09:44:42 +00:00
#include "Radar.h"
#include "Restart.h"
2020-01-13 23:13:42 +00:00
#include "Script.h"
2020-03-05 09:44:42 +00:00
#include "Stats.h"
2020-01-13 23:13:42 +00:00
#include "Streaming.h"
#include "Weather.h"
2020-01-13 23:13:42 +00:00
#include "World.h"
2020-03-05 09:44:42 +00:00
#include "Zones.h"
2020-01-13 23:13:42 +00:00
const uint32 SIZE_OF_ONE_GAME_IN_BYTES = 201729;
2020-01-13 23:13:42 +00:00
char (&DefaultPCSaveFileName)[260] = *(char(*)[260])*(uintptr*)0x8E28C0;
char (&ValidSaveName)[260] = *(char(*)[260])*(uintptr*)0x8E2CBC;
char (&LoadFileName)[256] = *(char(*)[256])*(uintptr*)0x9403C4;
wchar (&SlotFileName)[SLOT_COUNT][260] = *(wchar(*)[SLOT_COUNT][260])*(uintptr*)0x6F07C8;
wchar (&SlotSaveDate)[SLOT_COUNT][70] = *(wchar(*)[SLOT_COUNT][70])*(uintptr*)0x72B858;
int &CheckSum = *(int*)0x8E2BE0;
eLevelName &m_LevelToLoad = *(eLevelName*)0x8E29CC;
char SaveFileNameJustSaved[260];
int (&Slots)[SLOT_COUNT+1] = *(int(*)[SLOT_COUNT+1])*(uintptr*)0x72803C;
CDate &CompileDateAndTime = *(CDate*)0x72BCB8;
2020-01-31 08:12:36 +00:00
#define ReadDataFromBufferPointer(buf, to) memcpy(&to, buf, sizeof(to)); buf += align4bytes(sizeof(to));
#define WriteDataToBufferPointer(buf, from) memcpy(buf, &from, sizeof(from)); buf += align4bytes(sizeof(from));
2020-03-05 09:44:42 +00:00
//WRAPPER bool GenericSave(int file) { EAXJMP(0x58F8D0); }
2020-01-13 23:13:42 +00:00
WRAPPER bool GenericLoad() { EAXJMP(0x590A00); }
2020-03-05 09:44:42 +00:00
bool
GenericSave(int file)
{
static const int NameBufferSize = 0x30;
static const int MaxNameLength = 22;
static const char *NameSuffix = "'...\0";
2020-03-05 09:44:42 +00:00
uint8 *buf;
uint8 *tmpbuf;
uint8 *postsize;
uint32 size;
uint32 reserved;
wchar *saveName;
wchar suffix[6];
wchar nameBuf[MaxNameLength] = { 0 };
int nameLen;
CPad *currPad;
CheckSum = 0;
buf = work_buff;
saveName = TheText.Get(CStats::LastMissionPassedName);
if (saveName[0] != '\0') {
AsciiToUnicode(NameSuffix, suffix);
TextCopy(nameBuf, saveName);
nameLen = UnicodeStrlen(nameBuf);
nameBuf[nameLen] = '\0';
if (nameLen > MaxNameLength)
TextCopy(nameBuf + MaxNameLength, suffix);
}
2020-03-05 09:44:42 +00:00
memcpy(buf, saveName, NameBufferSize);
buf += NameBufferSize;
GetLocalTime((LPSYSTEMTIME) buf);
buf += sizeof(SYSTEMTIME);
WriteSaveBuf(buf, SIZE_OF_ONE_GAME_IN_BYTES);
WriteSaveBuf(buf, CGame::currLevel);
WriteSaveBuf(buf, TheCamera.m_matrix.m_matrix.pos.x);
WriteSaveBuf(buf, TheCamera.m_matrix.m_matrix.pos.y);
WriteSaveBuf(buf, TheCamera.m_matrix.m_matrix.pos.z);
WriteSaveBuf(buf, CClock::ms_nMillisecondsPerGameMinute);
WriteSaveBuf(buf, CClock::ms_nLastClockTick);
WriteSaveBuf(buf, CClock::ms_nGameClockHours);
buf += 3;
WriteSaveBuf(buf, CClock::ms_nGameClockMinutes); // TOOD: aligned WriteSaveBuf?
buf += 3;
currPad = CPad::GetPad(0);
WriteSaveBuf(buf, currPad->Mode);
buf += 2;
WriteSaveBuf(buf, CTimer::m_snTimeInMilliseconds);
WriteSaveBuf(buf, CTimer::ms_fTimeScale);
WriteSaveBuf(buf, CTimer::ms_fTimeStep);
WriteSaveBuf(buf, CTimer::ms_fTimeStepNonClipped);
WriteSaveBuf(buf, CTimer::m_FrameCounter);
WriteSaveBuf(buf, 1.0f); // CTimeStep::ms_fTimeStep;
WriteSaveBuf(buf, 1.0f); // CTimeStep::ms_fFramesPerUpdate;
WriteSaveBuf(buf, 1.0f); // CTimeStep::ms_fTimeScale;
WriteSaveBuf(buf, CWeather::OldWeatherType);
buf += 2;
WriteSaveBuf(buf, CWeather::NewWeatherType);
buf += 2;
WriteSaveBuf(buf, CWeather::ForcedWeatherType);
buf += 2;
WriteSaveBuf(buf, CWeather::InterpolationValue);
WriteSaveBuf(buf, CompileDateAndTime.m_nSecond);
WriteSaveBuf(buf, CompileDateAndTime.m_nMinute);
WriteSaveBuf(buf, CompileDateAndTime.m_nHour);
WriteSaveBuf(buf, CompileDateAndTime.m_nDay);
WriteSaveBuf(buf, CompileDateAndTime.m_nMonth);
WriteSaveBuf(buf, CompileDateAndTime.m_nYear);
WriteSaveBuf(buf, CWeather::WeatherTypeInList);
WriteSaveBuf(buf, TheCamera.m_fCarZoomValueScript); // TODO: unconfirmed
WriteSaveBuf(buf, TheCamera.m_fPedZoomValueScript); // TODO: unconfirmed
size = 0;
reserved = 0;
MakeSpaceForSizeInBufferPointer(tmpbuf, buf, postsize);
CTheScripts::SaveAllScripts(buf, &size);
CopySizeAndPreparePointer(tmpbuf, buf, postsize, reserved, size);
if (!PcSaveHelper.PcClassSaveRoutine(file, work_buff, size))
return false;
2020-03-05 09:44:42 +00:00
for (int i = 1; i < 19; i++) {
buf = work_buff;
size = 0;
reserved = 0;
MakeSpaceForSizeInBufferPointer(tmpbuf, buf, postsize);
switch (i) {
case 1: CPools::SavePedPool(buf, &size); break;
case 2: CGarages::Save(buf, &size); break;
case 3: CPools::SaveVehiclePool(buf, &size); break;
case 4: CPools::SaveObjectPool(buf, &size); break;
2020-03-05 09:44:42 +00:00
case 5: ThePaths.Save(buf, &size); break;
case 6: CCranes::Save(buf, &size); break;
2020-03-05 09:44:42 +00:00
case 7: CPickups::Save(buf, &size); break;
case 8: gPhoneInfo.Save(buf, &size); break;
case 9: CRestart::SaveAllRestartPoints(buf, &size); break;
case 10: CRadar::SaveAllRadarBlips(buf, &size); break;
case 11: CTheZones::SaveAllZones(buf, &size); break;
case 12: CGangs::SaveAllGangData(buf, &size); break;
case 13: CTheCarGenerators::SaveAllCarGenerators(buf, &size); break;
case 14: break;
case 15: break;
case 16: break;
case 17: break;
case 18: break;
case 19: break;
}
CopySizeAndPreparePointer(tmpbuf, buf, postsize, reserved, size);
if (!PcSaveHelper.PcClassSaveRoutine(file, work_buff, size))
2020-03-05 09:44:42 +00:00
return false;
}
// TODO: padding
return true;
}
2020-01-13 23:13:42 +00:00
bool
ReadInSizeofSaveFileBuffer(int32 &file, uint32 &size)
{
file = CFileMgr::OpenFile(LoadFileName, "rb");
if (file == 0) {
PcSaveHelper.nErrorCode = SAVESTATUS_ERR_LOAD_OPEN;
return false;
}
2020-01-31 08:12:36 +00:00
CFileMgr::Read(file, (const char*)&size, sizeof(size));
2020-01-13 23:13:42 +00:00
if (CFileMgr::GetErrorReadWrite(file)) {
PcSaveHelper.nErrorCode = SAVESTATUS_ERR_LOAD_READ;
if (!CloseFile(file))
PcSaveHelper.nErrorCode = SAVESTATUS_ERR_LOAD_CLOSE;
return false;
}
return true;
}
bool
ReadDataFromFile(int32 file, uint8 *buf, uint32 size)
{
if (file == 0) {
PcSaveHelper.nErrorCode = SAVESTATUS_ERR_LOAD_OPEN;
return false;
}
size_t read_size = CFileMgr::Read(file, (const char*)buf, size);
if (CFileMgr::GetErrorReadWrite(file) || read_size != size) {
PcSaveHelper.nErrorCode = SAVESTATUS_ERR_LOAD_READ;
if (!CloseFile(file))
PcSaveHelper.nErrorCode = SAVESTATUS_ERR_LOAD_CLOSE;
return false;
}
return true;
}
bool
CloseFile(int32 file)
{
return CFileMgr::CloseFile(file) == 0;
}
void
DoGameSpecificStuffAfterSucessLoad()
{
StillToFadeOut = true;
JustLoadedDontFadeInYet = true;
CTheScripts::Process();
}
2020-01-13 23:13:42 +00:00
bool
CheckSlotDataValid(int32 slot)
{
PcSaveHelper.nErrorCode = SAVESTATUS_SUCCESSFUL;
if (CheckDataNotCorrupt(slot, LoadFileName)) {
CStreaming::DeleteAllRwObjects();
return true;
}
PcSaveHelper.nErrorCode = SAVESTATUS_ERR_DATA_INVALID;
return false;
}
void
MakeSpaceForSizeInBufferPointer(uint8 *&presize, uint8 *&buf, uint8 *&postsize)
{
presize = buf;
2020-01-31 08:12:36 +00:00
buf += sizeof(uint32);
2020-01-13 23:13:42 +00:00
postsize = buf;
}
void
CopySizeAndPreparePointer(uint8 *&buf, uint8 *&postbuf, uint8 *&postbuf2, uint32 &unused, uint32 &size)
{
2020-01-31 08:12:36 +00:00
memcpy(buf, &size, sizeof(size));
2020-01-13 23:13:42 +00:00
size = align4bytes(size);
postbuf2 += size;
postbuf = postbuf2;
}
void
DoGameSpecificStuffBeforeSave()
{
CGameLogic::PassTime(360);
CPlayerPed *ped = FindPlayerPed();
ped->m_fCurrentStamina = ped->m_fMaxStamina;
CGame::TidyUpMemory(true, false);
}
void
MakeValidSaveName(int32 slot)
{
ValidSaveName[0] = '\0';
sprintf(ValidSaveName, "%s%i", DefaultPCSaveFileName, slot + 1);
strncat(ValidSaveName, ".b", 5);
}
wchar *
GetSavedGameDateAndTime(int32 slot)
{
return SlotSaveDate[slot];
}
wchar *
GetNameOfSavedGame(int32 slot)
{
return SlotFileName[slot];
}
bool
CheckDataNotCorrupt(int32 slot, char *name)
{
char filename[100];
int32 blocknum = 0;
eLevelName level = LEVEL_NONE;
CheckSum = 0;
2020-01-31 12:33:46 +00:00
uint32 bytes_processed = 0;
2020-01-13 23:13:42 +00:00
sprintf(filename, "%s%i%s", DefaultPCSaveFileName, slot + 1, ".b");
int file = CFileMgr::OpenFile(filename, "rb");
if (file == 0)
return false;
strcpy(name, filename);
2020-01-31 12:33:46 +00:00
while (SIZE_OF_ONE_GAME_IN_BYTES - sizeof(uint32) > bytes_processed && blocknum < 40) {
2020-01-13 23:13:42 +00:00
int32 blocksize;
2020-01-31 08:12:36 +00:00
if (!ReadDataFromFile(file, (uint8*)&blocksize, sizeof(blocksize))) {
2020-01-13 23:13:42 +00:00
CloseFile(file);
return false;
}
if (blocksize > align4bytes(sizeof(work_buff)))
2020-01-31 08:12:36 +00:00
blocksize = sizeof(work_buff) - sizeof(uint32);
2020-01-13 23:13:42 +00:00
if (!ReadDataFromFile(file, work_buff, align4bytes(blocksize))) {
CloseFile(file);
return false;
}
CheckSum += ((uint8*)&blocksize)[0];
CheckSum += ((uint8*)&blocksize)[1];
CheckSum += ((uint8*)&blocksize)[2];
CheckSum += ((uint8*)&blocksize)[3];
uint8 *_work_buf = work_buff;
for (int i = 0; i < align4bytes(blocksize); i++) {
CheckSum += *_work_buf++;
2020-01-31 12:33:46 +00:00
bytes_processed++;
2020-01-13 23:13:42 +00:00
}
if (blocknum == 0)
2020-01-31 08:12:36 +00:00
memcpy(&level, work_buff+4, sizeof(level));
2020-01-13 23:13:42 +00:00
blocknum++;
}
int32 _checkSum;
2020-01-31 08:12:36 +00:00
if (ReadDataFromFile(file, (uint8*)&_checkSum, sizeof(_checkSum))) {
2020-01-13 23:13:42 +00:00
if (CloseFile(file)) {
if (CheckSum == _checkSum) {
m_LevelToLoad = level;
return true;
}
return false;
}
return false;
}
CloseFile(file);
return false;
}
bool
RestoreForStartLoad()
{
uint8 buf[999];
int file = CFileMgr::OpenFile(LoadFileName, "rb");
if (file == 0) {
PcSaveHelper.nErrorCode = SAVESTATUS_ERR_LOAD_OPEN;
return false;
}
2020-01-31 08:12:36 +00:00
ReadDataFromFile(file, buf, sizeof(buf));
2020-01-13 23:13:42 +00:00
if (CFileMgr::GetErrorReadWrite(file)) {
PcSaveHelper.nErrorCode = SAVESTATUS_ERR_LOAD_READ;
if (!CloseFile(file))
PcSaveHelper.nErrorCode = SAVESTATUS_ERR_LOAD_CLOSE;
return false;
} else {
2020-01-31 08:12:36 +00:00
uint8 *_buf = buf + sizeof(int32) + sizeof(wchar[24]) + sizeof(SYSTEMTIME) + sizeof(SIZE_OF_ONE_GAME_IN_BYTES);
ReadDataFromBufferPointer(_buf, CGame::currLevel);
ReadDataFromBufferPointer(_buf, TheCamera.GetPosition().x);
ReadDataFromBufferPointer(_buf, TheCamera.GetPosition().y);
ReadDataFromBufferPointer(_buf, TheCamera.GetPosition().z);
CStreaming::RemoveUnusedBigBuildings(CGame::currLevel);
2020-01-13 23:13:42 +00:00
CStreaming::RemoveUnusedBuildings(CGame::currLevel);
CCollision::SortOutCollisionAfterLoad();
CStreaming::RequestBigBuildings(CGame::currLevel);
CStreaming::LoadAllRequestedModels(false);
CStreaming::HaveAllBigBuildingsLoaded(CGame::currLevel);
CGame::TidyUpMemory(true, false);
if (CloseFile(file)) {
return true;
} else {
PcSaveHelper.nErrorCode = SAVESTATUS_ERR_LOAD_CLOSE;
return false;
}
}
}
int
align4bytes(int32 size)
{
return (size + 3) & 0xFFFFFFFC;
}
STARTPATCHES
//InjectHook(0x58F8D0, GenericSave, PATCH_JUMP);
//InjectHook(0x590A00, GenericLoad, PATCH_JUMP);
InjectHook(0x591910, ReadInSizeofSaveFileBuffer, PATCH_JUMP);
InjectHook(0x591990, ReadDataFromFile, PATCH_JUMP);
InjectHook(0x591A00, CloseFile, PATCH_JUMP);
InjectHook(0x591A20, DoGameSpecificStuffAfterSucessLoad, PATCH_JUMP);
InjectHook(0x591A40, CheckSlotDataValid, PATCH_JUMP);
InjectHook(0x591A80, MakeSpaceForSizeInBufferPointer, PATCH_JUMP);
InjectHook(0x591AA0, CopySizeAndPreparePointer, PATCH_JUMP);
InjectHook(0x591AE0, DoGameSpecificStuffBeforeSave, PATCH_JUMP);
InjectHook(0x591B10, MakeValidSaveName, PATCH_JUMP);
InjectHook(0x591B50, GetSavedGameDateAndTime, PATCH_JUMP);
InjectHook(0x591B60, GetNameOfSavedGame, PATCH_JUMP);
InjectHook(0x591B70, CheckDataNotCorrupt, PATCH_JUMP);
InjectHook(0x591D60, RestoreForStartLoad, PATCH_JUMP);
InjectHook(0x591E80, align4bytes, PATCH_JUMP);
ENDPATCHES