From 6456691a3ca4fdbf5ca8f27375d88403261307be Mon Sep 17 00:00:00 2001
From: staphen <staphen@gmail.com>
Date: Sat, 1 Feb 2025 20:27:04 -0500
Subject: [PATCH] Avoid write mode when opening save files

---
 Source/mpq/mpq_writer.cpp | 29 +++++++++++++++++++----------
 1 file changed, 19 insertions(+), 10 deletions(-)

diff --git a/Source/mpq/mpq_writer.cpp b/Source/mpq/mpq_writer.cpp
index 111e3eb89..99c0bafca 100644
--- a/Source/mpq/mpq_writer.cpp
+++ b/Source/mpq/mpq_writer.cpp
@@ -92,17 +92,26 @@ MpqWriter::MpqWriter(const char *path)
 	LogVerbose("Opening {}", path);
 	std::string error;
 	bool exists = FileExists(path);
-	const char *mode = "wbx";
-	if (exists) {
-		mode = "r+b";
-		if (!GetFileSize(path, &size_)) {
-			error = R"(GetFileSize failed: "{}")";
-			LogError(error, path, std::strerror(errno));
-			goto on_error;
-		}
-		LogVerbose("GetFileSize(\"{}\") = {}", path, size_);
+	if (!exists) {
+		// FileExists() may return false in the case of an error
+		// so we use "ab" instead of "wb" to avoid accidentally
+		// truncating an existing file
+		stream_.Open(path, "ab");
+
+		// However, we cannot actually use a file handle that was
+		// opened in "ab" mode because we need to be able to seek
+		// and write to the middle of the file
+		stream_.Close();
 	}
-	if (!stream_.Open(path, mode)) {
+
+	if (!GetFileSize(path, &size_)) {
+		error = R"(GetFileSize failed: "{}")";
+		LogError(error, path, std::strerror(errno));
+		goto on_error;
+	}
+	LogVerbose("GetFileSize(\"{}\") = {}", path, size_);
+
+	if (!stream_.Open(path, "r+b")) {
 		stream_.Close();
 		error = "Failed to open file";
 		goto on_error;