If MyDocuments isn't available, use zen data folder in largest drive.

1 files changed, 33 insertions, 0 deletions

diff --git a/zenserver/config.cpp b/zenserver/config.cpp
index 3e85daa9e..aa61fa5a8 100644
--- a/zenserver/config.cpp
+++ b/zenserver/config.cpp
@@ -45,6 +45,39 @@ PickDefaultStateDirectory()
 		return myDocumentsDir;
 	}
 
+	int CandidateDriveLetterOffset = -1;
+	ULARGE_INTEGER CandidateDriveSize;
+	CandidateDriveSize.QuadPart = 0L;
+	DWORD LogicalDrives = GetLogicalDrives();
+	char CandidateDriveName[] = "A:\\";
+	for (int DriveLetterOffset = 0; DriveLetterOffset < 32; ++DriveLetterOffset)
+	{
+		if ((LogicalDrives & (1 << DriveLetterOffset)) != 0)
+		{
+			CandidateDriveName[0] = (char)('A' + DriveLetterOffset);
+			if (GetDriveTypeA(CandidateDriveName) == DRIVE_FIXED)
+			{
+				ULARGE_INTEGER FreeBytesAvailableToCaller;
+				ULARGE_INTEGER TotalNumberOfBytes;
+				if (0 != GetDiskFreeSpaceExA(CandidateDriveName, &FreeBytesAvailableToCaller, &TotalNumberOfBytes, nullptr))
+				{
+					if ((FreeBytesAvailableToCaller.QuadPart > 0) && (TotalNumberOfBytes.QuadPart > CandidateDriveSize.QuadPart))
+					{
+						CandidateDriveLetterOffset = DriveLetterOffset;
+						CandidateDriveSize = TotalNumberOfBytes;
+					}
+				}
+			}
+		}
+	}
+
+	if (CandidateDriveLetterOffset >= 0)
+	{
+		char RootZenFolderName[] = "A:\\zen";
+		RootZenFolderName[0] = (char)('A' + CandidateDriveLetterOffset);
+		return RootZenFolderName;
+	}
+
 	return L"";
 }