1 files changed, 143 insertions, 0 deletions

diff --git a/src/save/PCSave.cpp b/src/save/PCSave.cpp
new file mode 100644
index 00000000..02bd08ad
--- /dev/null
+++ b/src/save/PCSave.cpp
@@ -0,0 +1,143 @@
+#include "common.h"
+#include "patcher.h"
+#include "FileMgr.h"
+#include "GenericGameStorage.h"
+#include "Messages.h"
+#include "PCSave.h"
+#include "Text.h"
+
+const char* _psGetUserFilesFolder();
+
+void
+C_PcSave::SetSaveDirectory(const char *path)
+{
+	sprintf(DefaultPCSaveFileName, "%s\\%s", path, "GTA3sf");
+}
+
+bool
+C_PcSave::DeleteSlot(int32 slot)
+{
+	char FileName[200];
+
+	PcSaveHelper.nErrorCode = SAVESTATUS_SUCCESSFUL;
+	sprintf(FileName, "%s%i.b", DefaultPCSaveFileName, slot + 1);
+	DeleteFile(FileName);
+	SlotSaveDate[slot][0] = '\0';
+	return true;
+}
+
+bool
+C_PcSave::SaveSlot(int32 slot)
+{
+	MakeValidSaveName(slot);
+	PcSaveHelper.nErrorCode = SAVESTATUS_SUCCESSFUL;
+	_psGetUserFilesFolder();
+	int file = CFileMgr::OpenFile(ValidSaveName, "wb");
+	if (file != 0) {
+		DoGameSpecificStuffBeforeSave();
+		if (GenericSave(file)) {
+			if (CFileMgr::CloseFile(file) != 0)
+				nErrorCode = SAVESTATUS_ERR_SAVE_CLOSE;
+			return true;
+		}
+
+		return false;
+	}
+	PcSaveHelper.nErrorCode = SAVESTATUS_ERR_SAVE_CREATE;
+	return false;
+}
+
+bool
+C_PcSave::PcClassSaveRoutine(int32 file, uint8 *data, uint32 size)
+{
+	CFileMgr::Write(file, (const char*)&size, sizeof(size));
+	if (CFileMgr::GetErrorReadWrite(file)) {
+		nErrorCode = SAVESTATUS_ERR_SAVE_WRITE;
+		strncpy(SaveFileNameJustSaved, ValidSaveName, 259);
+		return false;
+	}
+
+	CFileMgr::Write(file, (const char*)data, align4bytes(size));
+	CheckSum += ((uint8*)&size)[0];
+	CheckSum += ((uint8*)&size)[1];
+	CheckSum += ((uint8*)&size)[2];
+	CheckSum += ((uint8*)&size)[3];
+	for (int i = 0; i < align4bytes(size); i++) {
+		CheckSum += *data++;
+	}
+	if (CFileMgr::GetErrorReadWrite(file)) {
+		nErrorCode = SAVESTATUS_ERR_SAVE_WRITE;
+		strncpy(SaveFileNameJustSaved, ValidSaveName, 259);
+		return false;
+	}
+
+	return true;
+}
+
+void
+C_PcSave::PopulateSlotInfo()
+{
+	for (int i = 0; i < SLOT_COUNT; i++) {
+		Slots[i + 1] = SLOT_EMPTY;
+		SlotFileName[i][0] = '\0';
+		SlotSaveDate[i][0] = '\0';
+	}
+	for (int i = 0; i < SLOT_COUNT; i++) {
+		char savename[52];
+		struct {
+			int size;
+			wchar FileName[24];
+			_SYSTEMTIME SaveDateTime;
+		} header;
+		sprintf(savename, "%s%i%s", DefaultPCSaveFileName, i + 1, ".b");
+		int file = CFileMgr::OpenFile(savename, "rb");
+		if (file != 0) {
+			CFileMgr::Read(file, (char*)&header, sizeof(header));
+			if (strncmp((char*)&header, TopLineEmptyFile, sizeof(TopLineEmptyFile)-1) != 0) {
+				Slots[i + 1] = SLOT_OK;
+				memcpy(SlotFileName[i], &header.FileName, sizeof(header.FileName));
+				
+				SlotFileName[i][24] = '\0';
+			}
+			CFileMgr::CloseFile(file);
+		}
+		if (Slots[i + 1] == SLOT_OK) {
+			if (CheckDataNotCorrupt(i, savename)) {
+				_SYSTEMTIME st;
+				memcpy(&st, &header.SaveDateTime, sizeof(_SYSTEMTIME));
+				const char *month;
+				switch (st.wMonth)
+				{
+				case 1: month = "JAN"; break;
+				case 2: month = "FEB"; break;
+				case 3: month = "MAR"; break;
+				case 4: month = "APR"; break;
+				case 5: month = "MAY"; break;
+				case 6: month = "JUN"; break;
+				case 7: month = "JUL"; break;
+				case 8: month = "AUG"; break;
+				case 9: month = "SEP"; break;
+				case 10: month = "OCT"; break;
+				case 11: month = "NOV"; break;
+				case 12: month = "DEC"; break;
+				default: assert(0);
+				}
+				char date[70];
+				sprintf(date, "%02d %s %04d %02d:%02d:%02d", st.wDay, UnicodeToAsciiForSaveLoad(TheText.Get(month)), st.wYear, st.wHour, st.wMinute, st.wSecond);
+				AsciiToUnicode(date, SlotSaveDate[i]);
+
+			} else {
+				CMessages::InsertNumberInString(TheText.Get("FEC_SLC"), i + 1, -1, -1, -1, -1, -1, SlotFileName[i]);
+				Slots[i + 1] = SLOT_CORRUPTED;
+			}
+		}
+	}
+}
+
+STARTPATCHES
+	InjectHook(0x591EA0, C_PcSave::SetSaveDirectory, PATCH_JUMP);
+	InjectHook(0x5922F0, &C_PcSave::DeleteSlot, PATCH_JUMP);
+	InjectHook(0x591EC0, &C_PcSave::SaveSlot, PATCH_JUMP);
+	InjectHook(0x591F80, &C_PcSave::PcClassSaveRoutine, PATCH_JUMP);
+	InjectHook(0x592090, &C_PcSave::PopulateSlotInfo, PATCH_JUMP);
+ENDPATCHES
\ No newline at end of file