/*---------------------------------------------------------------------------*/
//       Author : hiyohiyo
//         Mail : hiyohiyo@crystalmark.info
//          Web : https://crystalmark.info/
//      License : The MIT License
/*---------------------------------------------------------------------------*/

#include "stdafx.h"
#include <afxmt.h>

#include "DiskMark.h"
#include "DiskMarkDlg.h"
#include "DiskBench.h"
#include "GetFileVersion.h"
#include "GetOsInfo.h"

#include <winioctl.h>
#include <mmsystem.h>
#pragma comment(lib,"winmm.lib")

#pragma warning(disable : 4996)

static CString TestFilePath;
static CString TestFileDir;
static CString DiskSpdExe;

static HANDLE hFile;
static int DiskTestCount;
static UINT64 DiskTestSize;
static int SequentialSize1;
static int SequentialSize2;
static int RandomSize1;
static int RandomSize2;
static int RandomSize3;
static int Affinity;
static BOOL MixMode;
static int MixRatio;

static void ShowErrorMessage(CString message);
static void Interval(UINT time, LPVOID dlg);

static BOOL Init(LPVOID dlg);
static void DiskSpd(LPVOID dlg, DISK_SPD_CMD cmd);

static UINT Exit(LPVOID dlg);
static void CALLBACK TimerProc(HWND hwnd, UINT uMsg, UINT_PTR idEvent, DWORD dwTime);
static volatile BOOL WaitFlag;

#define DISK_SPD_EXE_32           L"CdmResource\\diskspd\\diskspd32.exe"
#define DISK_SPD_EXE_64           L"CdmResource\\diskspd\\diskspd64.exe"
#define DISK_SPD_EXE_32_LEGACY    L"CdmResource\\diskspd\\diskspd32L.exe"
#define DISK_SPD_EXE_64_LEGACY    L"CdmResource\\diskspd\\diskspd64L.exe"
#define DISK_SPD_EXE_ARM32        L"CdmResource\\diskspd\\diskspdA32.exe"
#define DISK_SPD_EXE_ARM64        L"CdmResource\\diskspd\\diskspdA64.exe"

PROCESS_INFORMATION pi;

int ExecAndWait(TCHAR *pszCmd, BOOL bNoWindow, double *latency)
{
	DWORD Code = 0;
	BOOL bSuccess;
	STARTUPINFO si;

	memset(&si, 0, sizeof(STARTUPINFO));
	si.cb = sizeof(STARTUPINFO);

	if (bNoWindow) {
		si.dwFlags = STARTF_USESHOWWINDOW;
		si.wShowWindow = SW_HIDE;
	}

	CString name;
	name.Format(L"CrystalDiskMark%08X", GetCurrentProcessId());
	auto size = 8;

	HANDLE hSharedMemory = CreateFileMapping(INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE, NULL, size, name.GetString());
	if (hSharedMemory != NULL)
	{
		auto pMemory = (double*)MapViewOfFile(hSharedMemory, FILE_MAP_ALL_ACCESS, NULL, NULL, size);
		if (pMemory != NULL)
		{
			bSuccess = CreateProcess(NULL, pszCmd, NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi);
			if (bSuccess != TRUE)
			{
				UnmapViewOfFile(pMemory);
				CloseHandle(hSharedMemory);
				return 0;
			}

			WaitForInputIdle(pi.hProcess, INFINITE);
			WaitForSingleObject(pi.hProcess, INFINITE);
			GetExitCodeProcess(pi.hProcess, &Code);

			CloseHandle(pi.hThread);
			CloseHandle(pi.hProcess);

			pi.hProcess = NULL;

			*latency = (double)*pMemory * 1000; // milli sec to micro sec

			UnmapViewOfFile(pMemory);
			CloseHandle(hSharedMemory);
		}
	}

	return Code;
}


void ShowErrorMessage(CString message)
{
	DWORD lastErrorCode = GetLastError();
	CString errorMessage;
	LPVOID lpMessageBuffer;
	FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM, NULL, lastErrorCode, 
		MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPTSTR) &lpMessageBuffer, 0, NULL);
	errorMessage.Format(_T("0x%08X:%s"), lastErrorCode, (LPTSTR) lpMessageBuffer);

	AfxMessageBox(message + _T("\r\n") + errorMessage);
	LocalFree( lpMessageBuffer );
}

VOID Interval(LPVOID dlg)
{
	int intervalTime = ((CDiskMarkDlg*) dlg)->m_IntervalTime;
	CString title;

	for (int i = 0; i < intervalTime; i++)
	{
		if (!((CDiskMarkDlg*) dlg)->m_DiskBenchStatus)
		{
			return;
		}
		title.Format(L"Interval Time %d/%d sec", i, intervalTime);
		::PostMessage(((CDiskMarkDlg*) dlg)->GetSafeHwnd(), WM_USER_UPDATE_MESSAGE, (WPARAM) &title, 0);
		Sleep(1000);
	}
}

UINT ExecDiskBenchAll(LPVOID dlg)
{
	if(Init(dlg))
	{
		DiskSpd(dlg, TEST_SEQUENTIAL_READ_1);
		Interval(dlg);
		DiskSpd(dlg, TEST_SEQUENTIAL_READ_2);
		Interval(dlg);
		DiskSpd(dlg, TEST_RANDOM_READ_1);
		Interval(dlg);
		DiskSpd(dlg, TEST_RANDOM_READ_2);
		Interval(dlg);
		DiskSpd(dlg, TEST_SEQUENTIAL_WRITE_1);
		Interval(dlg);
		DiskSpd(dlg, TEST_SEQUENTIAL_WRITE_2);
		Interval(dlg);
		DiskSpd(dlg, TEST_RANDOM_WRITE_1);
		Interval(dlg);
		DiskSpd(dlg, TEST_RANDOM_WRITE_2);
#ifdef MIX_MODE
		if (MixMode)
		{
			Interval(dlg);
			DiskSpd(dlg, TEST_SEQUENTIAL_MIX_1);
			Interval(dlg);
			DiskSpd(dlg, TEST_SEQUENTIAL_MIX_2);
			Interval(dlg);
			DiskSpd(dlg, TEST_RANDOM_MIX_1);
			Interval(dlg);
			DiskSpd(dlg, TEST_RANDOM_MIX_2);
		}
#endif
	}

	return Exit(dlg);
}

UINT ExecDiskBenchAllPeak(LPVOID dlg)
{
	if (Init(dlg))
	{
		DiskSpd(dlg, TEST_SEQUENTIAL_READ_1);
		Interval(dlg);
		DiskSpd(dlg, TEST_RANDOM_READ_1);
		Interval(dlg);
		DiskSpd(dlg, TEST_SEQUENTIAL_WRITE_1);
		Interval(dlg);
		DiskSpd(dlg, TEST_RANDOM_WRITE_1);

#ifdef MIX_MODE
		if (MixMode)
		{
			Interval(dlg);
			DiskSpd(dlg, TEST_SEQUENTIAL_MIX_1);
			Interval(dlg);
			DiskSpd(dlg, TEST_RANDOM_MIX_1);
		}
#endif
	}

	return Exit(dlg);
}

UINT ExecDiskBenchAllReal(LPVOID dlg)
{
	if (Init(dlg))
	{
		DiskSpd(dlg, TEST_SEQUENTIAL_READ_REAL);
		Interval(dlg);
		DiskSpd(dlg, TEST_RANDOM_READ_REAL);
		Interval(dlg);
		DiskSpd(dlg, TEST_SEQUENTIAL_WRITE_REAL);
		Interval(dlg);
		DiskSpd(dlg, TEST_RANDOM_WRITE_REAL);

#ifdef MIX_MODE
		if (MixMode)
		{
			Interval(dlg);
			DiskSpd(dlg, TEST_SEQUENTIAL_MIX_REAL);
			Interval(dlg);
			DiskSpd(dlg, TEST_RANDOM_MIX_REAL);
		}
#endif
	}

	return Exit(dlg);
}

UINT ExecDiskBenchSequential1(LPVOID dlg)
{
	if (Init(dlg))
	{
		DiskSpd(dlg, TEST_SEQUENTIAL_READ_1);
		Interval(dlg);
		DiskSpd(dlg, TEST_SEQUENTIAL_WRITE_1);
#ifdef MIX_MODE
		if (MixMode)
		{
			Interval(dlg);
			DiskSpd(dlg, TEST_SEQUENTIAL_MIX_1);
		}
#endif
	}
	return Exit(dlg);
}

UINT ExecDiskBenchSequential2(LPVOID dlg)
{

	if (Init(dlg))
	{
		DiskSpd(dlg, TEST_SEQUENTIAL_READ_2);
		Interval(dlg);
		DiskSpd(dlg, TEST_SEQUENTIAL_WRITE_2);
#ifdef MIX_MODE
		if (MixMode)
		{
			Interval(dlg);
			DiskSpd(dlg, TEST_SEQUENTIAL_MIX_2);
		}
#endif
	}
	return Exit(dlg);
}

UINT ExecDiskBenchRandom1(LPVOID dlg)
{
	if (Init(dlg))
	{
		DiskSpd(dlg, TEST_RANDOM_READ_1);
		Interval(dlg);
		DiskSpd(dlg, TEST_RANDOM_WRITE_1);
#ifdef MIX_MODE
		if (MixMode)
		{
			Interval(dlg);
			DiskSpd(dlg, TEST_RANDOM_MIX_1);
		}
#endif
	}
	return Exit(dlg);
}
UINT ExecDiskBenchRandom2(LPVOID dlg)
{
	if (Init(dlg))
	{
		DiskSpd(dlg, TEST_RANDOM_READ_2);
		Interval(dlg);
		DiskSpd(dlg, TEST_RANDOM_WRITE_2);
#ifdef MIX_MODE
		if (MixMode)
		{
			Interval(dlg);
			DiskSpd(dlg, TEST_RANDOM_MIX_2);
		}
#endif
	}
	return Exit(dlg);
}

UINT ExecDiskBenchSequentialReal(LPVOID dlg)
{

	if (Init(dlg))
	{
		DiskSpd(dlg, TEST_SEQUENTIAL_READ_REAL);
		Interval(dlg);
		DiskSpd(dlg, TEST_SEQUENTIAL_WRITE_REAL);
#ifdef MIX_MODE
		if (MixMode)
		{
			Interval(dlg);
			DiskSpd(dlg, TEST_SEQUENTIAL_MIX_REAL);
		}
#endif
	}
	return Exit(dlg);
}

UINT ExecDiskBenchRandomReal(LPVOID dlg)
{
	if (Init(dlg))
	{
		DiskSpd(dlg, TEST_RANDOM_READ_REAL);
		Interval(dlg);
		DiskSpd(dlg, TEST_RANDOM_WRITE_REAL);
#ifdef MIX_MODE
		if (MixMode)
		{
			Interval(dlg);
			DiskSpd(dlg, TEST_RANDOM_MIX_REAL);
		}
#endif
	}
	return Exit(dlg);
}

BOOL Init(void* dlg)
{
	BOOL FlagArc;
	BOOL result;
	static CString cstr;
	TCHAR drive;

	ULARGE_INTEGER freeBytesAvailableToCaller;
	ULARGE_INTEGER totalNumberOfBytes;
	ULARGE_INTEGER totalNumberOfFreeBytes;

	// Init m_Ini
	TCHAR *ptrEnd;
	TCHAR temp[MAX_PATH];
	::GetModuleFileName(NULL, temp, MAX_PATH);
	if ((ptrEnd = _tcsrchr(temp, '\\')) != NULL)
	{
		*ptrEnd = '\0';
	}

	pi.hProcess = NULL;

#ifdef _M_ARM
	DiskSpdExe.Format(L"%s\\%s", temp, DISK_SPD_EXE_ARM32);
#elif _M_ARM64
	DiskSpdExe.Format(L"%s\\%s", temp, DISK_SPD_EXE_ARM64);
#elif _M_X64
	if(Is7orLater())
	{
		DiskSpdExe.Format(L"%s\\%s", temp, DISK_SPD_EXE_64);
	}
	else
	{
		DiskSpdExe.Format(L"%s\\%s", temp, DISK_SPD_EXE_64_LEGACY);
	}
#else
	if (Is7orLater())
	{
		DiskSpdExe.Format(L"%s\\%s", temp, DISK_SPD_EXE_32);
	}
	else
	{
		DiskSpdExe.Format(L"%s\\%s", temp, DISK_SPD_EXE_32_LEGACY);
	}
#endif

	if (! IsFileExist(DiskSpdExe))
	{
		AfxMessageBox(((CDiskMarkDlg*) dlg)->m_MesDiskSpdNotFound);
		((CDiskMarkDlg*) dlg)->m_DiskBenchStatus = FALSE;
		return FALSE;
	}
	DiskTestCount = ((CDiskMarkDlg*) dlg)->m_IndexTestCount + 1;
	
	CString testSize = ((CDiskMarkDlg*)dlg)->m_ValueTestSize;
	if (testSize.Find(L"M") == -1) // GiB
	{
		DiskTestSize = (UINT64)_tstoi(testSize) * 1024;
	}
	else // MiB
	{
		DiskTestSize = (UINT64)_tstoi(testSize);
	}

	SequentialSize1 = ((CDiskMarkDlg*)dlg)->m_SequentialSize1;
	SequentialSize2 = ((CDiskMarkDlg*)dlg)->m_SequentialSize2;
	RandomSize1 = ((CDiskMarkDlg*)dlg)->m_RandomSize1;
	RandomSize2 = ((CDiskMarkDlg*)dlg)->m_RandomSize2;
	RandomSize3 = ((CDiskMarkDlg*)dlg)->m_RandomSize3;

	Affinity = ((CDiskMarkDlg*)dlg)->m_Affinity;
	MixMode = ((CDiskMarkDlg*)dlg)->m_MixMode;
	MixRatio = ((CDiskMarkDlg*)dlg)->m_MixRatio;

	CString RootPath;
	if(((CDiskMarkDlg*)dlg)->m_MaxIndexTestDrive != ((CDiskMarkDlg*)dlg)->m_IndexTestDrive)
	{

		drive = ((CDiskMarkDlg*)dlg)->m_ValueTestDrive.GetAt(0);
		cstr.Format(_T("%C:"), drive);
		GetDiskFreeSpaceEx(cstr, &freeBytesAvailableToCaller, &totalNumberOfBytes, &totalNumberOfFreeBytes);
		if (totalNumberOfBytes.QuadPart < ((ULONGLONG)8 * 1024 * 1024 * 1024)) // < 8 GB
		{
			((CDiskMarkDlg*)dlg)->m_TestDriveInfo.Format(_T("%C: %.1f%% (%.1f/%.1f MiB)"), drive,
				(double)(totalNumberOfBytes.QuadPart - totalNumberOfFreeBytes.QuadPart) / (double)totalNumberOfBytes.QuadPart * 100,
				(totalNumberOfBytes.QuadPart - totalNumberOfFreeBytes.QuadPart) / 1024 / 1024.0,
				totalNumberOfBytes.QuadPart / 1024 / 1024.0);
		}
		else
		{
			((CDiskMarkDlg*)dlg)->m_TestDriveInfo.Format(_T("%C: %.1f%% (%.1f/%.1f GiB)"), drive,
				(double)(totalNumberOfBytes.QuadPart - totalNumberOfFreeBytes.QuadPart) / (double)totalNumberOfBytes.QuadPart * 100,
				(totalNumberOfBytes.QuadPart - totalNumberOfFreeBytes.QuadPart) / 1024 / 1024 / 1024.0,
				totalNumberOfBytes.QuadPart / 1024 / 1024 / 1024.0);
		}
		RootPath.Format(_T("%c:\\"), drive);
	}
	else
	{
		RootPath = ((CDiskMarkDlg*)dlg)->m_TestTargetPath;
		RootPath += L"\\";
	}

	TestFileDir.Format(_T("%sCrystalDiskMark%08X"), (LPTSTR)RootPath.GetString(), timeGetTime());
	CreateDirectory(TestFileDir, NULL);
	TestFilePath.Format(_T("%s\\CrystalDiskMark%08X.tmp"), (LPTSTR)TestFileDir.GetString(), timeGetTime());

	DWORD FileSystemFlags;
	GetVolumeInformation(RootPath, NULL, NULL, NULL, NULL, &FileSystemFlags, NULL, NULL);
	if(FileSystemFlags & FS_VOL_IS_COMPRESSED)
	{
		FlagArc = TRUE;
	}
	else
	{
		FlagArc = FALSE;
	}

// Check Disk Capacity //
	OSVERSIONINFO osVersionInfo;
	osVersionInfo.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
	GetVersionEx(&osVersionInfo);

	ULARGE_INTEGER FreeBytesAvailableToCaller, TotalNumberOfBytes, TotalNumberOfFreeBytes;
	GetDiskFreeSpaceEx(RootPath, &FreeBytesAvailableToCaller, &TotalNumberOfBytes, &TotalNumberOfFreeBytes);
	if(DiskTestSize > TotalNumberOfFreeBytes.QuadPart / 1024 / 1024 )
	{
		AfxMessageBox(((CDiskMarkDlg*)dlg)->m_MesDiskCapacityError);
		((CDiskMarkDlg*)dlg)->m_DiskBenchStatus = FALSE;
		return FALSE;
	}

	CString title;
	title.Format(L"Preparing... Create Test File");
	::PostMessage(((CDiskMarkDlg*)dlg)->GetSafeHwnd(), WM_USER_UPDATE_MESSAGE, (WPARAM)& title, 0);

// Preapare Test File
	hFile = ::CreateFile(TestFilePath, GENERIC_READ | GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,
			FILE_ATTRIBUTE_NORMAL|FILE_FLAG_NO_BUFFERING|FILE_FLAG_SEQUENTIAL_SCAN, NULL);

	if(hFile == INVALID_HANDLE_VALUE)
	{
		AfxMessageBox(((CDiskMarkDlg*)dlg)->m_MesDiskCreateFileError);
		((CDiskMarkDlg*)dlg)->m_DiskBenchStatus = FALSE;
		return FALSE;
	}

// COMPRESSION_FORMAT_NONE
	USHORT lpInBuffer = COMPRESSION_FORMAT_NONE;
	DWORD lpBytesReturned = 0;
	DeviceIoControl(hFile, FSCTL_SET_COMPRESSION, (LPVOID) &lpInBuffer,
				sizeof(USHORT), NULL, 0, (LPDWORD)&lpBytesReturned, NULL);


	// Fill Test Data
	char* buf = NULL;
	int BufSize;
	int Loop;
	int i;
	DWORD writesize;
	BufSize = 1024 * 1024;
	Loop = (int)DiskTestSize;

	buf = (char*) VirtualAlloc(NULL, BufSize, MEM_RESERVE | MEM_COMMIT, PAGE_READWRITE);
	if (buf == NULL)
	{
		AfxMessageBox(_T("Failed VirtualAlloc()."));
		((CDiskMarkDlg*) dlg)->m_DiskBenchStatus = FALSE;
		return FALSE;
	}

	if (((CDiskMarkDlg*) dlg)->m_TestData == TEST_DATA_ALL0X00)
	{
		for (i = 0; i < BufSize; i++)
		{
			buf[i] = 0;
		}
	}
	else
	{
		// Compatible with DiskSpd
		for (i = 0; i < BufSize; i++)
		{
			buf[i] = (char) (rand() % 256);
		}
	}

	for (i = 0; i < Loop; i++)
	{
		if (((CDiskMarkDlg*) dlg)->m_DiskBenchStatus)
		{
			result = WriteFile(hFile, buf, BufSize, &writesize, NULL);
		}
		else
		{
			CloseHandle(hFile);
			VirtualFree(buf, 0, MEM_RELEASE);
			((CDiskMarkDlg*) dlg)->m_DiskBenchStatus = FALSE;
			return FALSE;
		}
	}
	VirtualFree(buf, 0, MEM_RELEASE);
	CloseHandle(hFile);

	return TRUE;
}

void CALLBACK TimerProc(HWND hwnd, UINT uMsg, UINT_PTR idEvent, DWORD dwTime)
{
	if(idEvent == TIMER_ID)
	{
		WaitFlag = FALSE;
		KillTimer(hwnd, idEvent);
	}
}

UINT Exit(void* dlg)
{
	DeleteFile(TestFilePath);
	RemoveDirectory(TestFileDir);

	static CString cstr;
	cstr = _T("");

	if(((CDiskMarkDlg*)dlg)->m_TestData == TEST_DATA_ALL0X00)
	{
		cstr = ALL_0X00_0FILL;
	}
	else
	{
		cstr = _T("");
	}

	::PostMessage(((CDiskMarkDlg*)dlg)->GetSafeHwnd(), WM_USER_UPDATE_MESSAGE, NULL, (LPARAM)&cstr);
	::PostMessage(((CDiskMarkDlg*)dlg)->GetSafeHwnd(), WM_USER_EXIT_BENCHMARK, 0, 0);

	((CDiskMarkDlg*)dlg)->m_DiskBenchStatus = FALSE;
	((CDiskMarkDlg*)dlg)->m_WinThread = NULL;

	return 0;
}

void DiskSpd(void* dlg, DISK_SPD_CMD cmd)
{
	static CString cstr;
	double *maxScore = NULL;
	double *minLatency = NULL;
	CString command;
	CString title;
	CString qt;
	CString option;
	CString bufOption;

	int duration = 5;

	int j;

	if (!((CDiskMarkDlg*) dlg)->m_DiskBenchStatus)
	{
		return;
	}


	if (((CDiskMarkDlg*) dlg)->m_TestData == TEST_DATA_ALL0X00)
	{
		bufOption += L" -Z";
	}
	else
	{
		switch (cmd)
		{
		case TEST_SEQUENTIAL_WRITE_1:
		case TEST_SEQUENTIAL_MIX_1:
			cstr.Format(L" -Z%dM", SequentialSize1);
			bufOption += cstr;
			break;
		case TEST_SEQUENTIAL_WRITE_2:
		case TEST_SEQUENTIAL_MIX_2:
			cstr.Format(L" -Z%dM", SequentialSize2);
			bufOption += cstr;
			break;
		case TEST_RANDOM_WRITE_1:
		case TEST_RANDOM_MIX_1:
			cstr.Format(L" -Z%dK", RandomSize1);
			bufOption += cstr;
			break;
		case TEST_RANDOM_WRITE_2:
		case TEST_RANDOM_MIX_2:
			cstr.Format(L" -Z%dK", RandomSize2);
			bufOption += cstr;
			break;
		case TEST_RANDOM_WRITE_REAL:
		case TEST_RANDOM_MIX_REAL:
			bufOption += L" -Z4K";
			break;
		case TEST_SEQUENTIAL_WRITE_REAL:
		case TEST_SEQUENTIAL_MIX_REAL:
			bufOption += L" -Z1M";
			break;
		}
	}

	switch (cmd)
	{
	/*
	case TEST_CREATE_FILE:
		title = L"Preparing...";
		::PostMessage(((CDiskMarkDlg*) dlg)->GetSafeHwnd(), WM_USER_UPDATE_MESSAGE, (WPARAM) &title, 0);

		option.Format(L"-c%dM", (int)DiskTestSize);
		option += bufOption;
		command.Format(_T("\"%s\" %s \"%s\""), DiskSpdExe.GetString(), option, TestFilePath);
		ExecAndWait((TCHAR*) (command.GetString()), TRUE, NULL);

		return;
		break;
	*/
	case TEST_SEQUENTIAL_READ_1:
		title.Format(L"Sequential Read");
		qt.Format(L"[T=%d]", ((CDiskMarkDlg*) dlg)->m_SequentialThreads1);
		option.Format(L"-b%dM -d%d -o%d -t%d -W0 -S -w0", SequentialSize1, duration, ((CDiskMarkDlg*) dlg)->m_SequentialQueues1, ((CDiskMarkDlg*) dlg)->m_SequentialThreads1);
		maxScore = &(((CDiskMarkDlg*) dlg)->m_SequentialReadScore1);
		minLatency = &(((CDiskMarkDlg*)dlg)->m_SequentialReadLatency1);
		break;
	case TEST_SEQUENTIAL_WRITE_1:
		title.Format(L"Sequential Write");
		qt.Format(L"[T=%d]", ((CDiskMarkDlg*) dlg)->m_SequentialThreads1);
		option.Format(L"-b%dM -d%d -o%d -t%d -W0 -S -w100", SequentialSize1, duration, ((CDiskMarkDlg*) dlg)->m_SequentialQueues1, ((CDiskMarkDlg*) dlg)->m_SequentialThreads1);
		option += bufOption;
		maxScore = &(((CDiskMarkDlg*) dlg)->m_SequentialWriteScore1);
		minLatency = &(((CDiskMarkDlg*)dlg)->m_SequentialWriteLatency1);
		break;
#ifdef MIX_MODE
	case TEST_SEQUENTIAL_MIX_1:
		title.Format(L"Sequential Mix");
		qt.Format(L"[T=%d]", ((CDiskMarkDlg*)dlg)->m_SequentialThreads1);
		option.Format(L"-b%dM -d%d -o%d -t%d -W0 -S -w%d", SequentialSize1, duration, ((CDiskMarkDlg*)dlg)->m_SequentialQueues1, ((CDiskMarkDlg*)dlg)->m_SequentialThreads1, MixRatio);
		option += bufOption;
		maxScore = &(((CDiskMarkDlg*)dlg)->m_SequentialMixScore1);
		minLatency = &(((CDiskMarkDlg*)dlg)->m_SequentialMixLatency1);
		break;
#endif
	case TEST_SEQUENTIAL_READ_2:
		title.Format(L"Sequential Read");
		qt.Format(L"[T=%d]", ((CDiskMarkDlg*)dlg)->m_SequentialThreads2);
		option.Format(L"-b%dM -d%d -o%d -t%d -W0 -S -w0", SequentialSize2, duration, ((CDiskMarkDlg*)dlg)->m_SequentialQueues2, ((CDiskMarkDlg*)dlg)->m_SequentialThreads2);
		maxScore = &(((CDiskMarkDlg*)dlg)->m_SequentialReadScore2);
		minLatency = &(((CDiskMarkDlg*)dlg)->m_SequentialReadLatency2);
		break;
	case TEST_SEQUENTIAL_WRITE_2:
		title.Format(L"Sequential Write");
		qt.Format(L"[T=%d]", ((CDiskMarkDlg*)dlg)->m_SequentialThreads2);
		option.Format(L"-b%dM -d%d -o%d -t%d -W0 -S -w100", SequentialSize2, duration, ((CDiskMarkDlg*)dlg)->m_SequentialQueues2, ((CDiskMarkDlg*)dlg)->m_SequentialThreads2);
		option += bufOption;
		maxScore = &(((CDiskMarkDlg*)dlg)->m_SequentialWriteScore2);
		minLatency = &(((CDiskMarkDlg*)dlg)->m_SequentialWriteLatency2);
		break;
#ifdef MIX_MODE
	case TEST_SEQUENTIAL_MIX_2:
		title.Format(L"Sequential Mix");
		qt.Format(L"[T=%d]", ((CDiskMarkDlg*)dlg)->m_SequentialThreads2);
		option.Format(L"-b%dM -d%d -o%d -t%d -W0 -S -w%d", SequentialSize2, duration, ((CDiskMarkDlg*)dlg)->m_SequentialQueues2, ((CDiskMarkDlg*)dlg)->m_SequentialThreads2, MixRatio);
		option += bufOption;
		maxScore = &(((CDiskMarkDlg*)dlg)->m_SequentialMixScore2);
		minLatency = &(((CDiskMarkDlg*)dlg)->m_SequentialMixLatency2);
		break;
#endif
	case TEST_SEQUENTIAL_READ_REAL:
		title.Format(L"Sequential Read");
		qt.Format(L"[T=1]");
		option.Format(L"-b1M -d%d -o1 -t1 -W0 -S -w0", duration);
		maxScore = &(((CDiskMarkDlg*)dlg)->m_SequentialReadScore1);
		minLatency = &(((CDiskMarkDlg*)dlg)->m_SequentialReadLatency1);
		break;
	case TEST_SEQUENTIAL_WRITE_REAL:
		title.Format(L"Sequential Write");
		qt.Format(L"[T=1]");
		option.Format(L"-b1M -d%d -o1 -t1 -W0 -S -w100", duration);
		option += bufOption;
		maxScore = &(((CDiskMarkDlg*)dlg)->m_SequentialWriteScore1);
		minLatency = &(((CDiskMarkDlg*)dlg)->m_SequentialWriteLatency1);
		break;
#ifdef MIX_MODE
	case TEST_SEQUENTIAL_MIX_REAL:
		title.Format(L"Sequential Mix");
		qt.Format(L"[T=1]");
		option.Format(L"-b1M -d%d -o1 -t1 -W0 -S -w%d", duration, MixRatio);
		option += bufOption;
		maxScore = &(((CDiskMarkDlg*)dlg)->m_SequentialMixScore1);
		minLatency = &(((CDiskMarkDlg*)dlg)->m_SequentialMixLatency1);
		break;
#endif
	case TEST_RANDOM_READ_1:
		title.Format(L"Random Read");
		qt.Format(L"[Q=%d/T=%d]", ((CDiskMarkDlg*) dlg)->m_RandomQueues1, ((CDiskMarkDlg*) dlg)->m_RandomThreads1);
		option.Format(L"-b%dK -d%d -o%d -t%d -W0 -r -S -w0", RandomSize1, duration, ((CDiskMarkDlg*) dlg)->m_RandomQueues1, ((CDiskMarkDlg*) dlg)->m_RandomThreads1);
		maxScore = &(((CDiskMarkDlg*) dlg)->m_RandomReadScore1);
		minLatency = &(((CDiskMarkDlg*)dlg)->m_RandomReadLatency1);
		break;
	case TEST_RANDOM_WRITE_1:
		title.Format(L"Random Write");
		qt.Format(L"[Q=%d/T=%d]", ((CDiskMarkDlg*) dlg)->m_RandomQueues1, ((CDiskMarkDlg*) dlg)->m_RandomThreads1);
		option.Format(L"-b%dK -d%d -o%d -t%d -W0 -r -S -w100", RandomSize1, duration, ((CDiskMarkDlg*) dlg)->m_RandomQueues1, ((CDiskMarkDlg*) dlg)->m_RandomThreads1);
		option += bufOption;
		maxScore = &(((CDiskMarkDlg*) dlg)->m_RandomWriteScore1);
		minLatency = &(((CDiskMarkDlg*)dlg)->m_RandomWriteLatency1);
		break;
#ifdef MIX_MODE
	case TEST_RANDOM_MIX_1:
		title.Format(L"Random Mix");
		qt.Format(L"[Q=%d/T=%d]", ((CDiskMarkDlg*)dlg)->m_RandomQueues1, ((CDiskMarkDlg*)dlg)->m_RandomThreads1);
		option.Format(L"-b%dK -d%d -o%d -t%d -W0 -r -S -w%d", RandomSize1, duration, ((CDiskMarkDlg*)dlg)->m_RandomQueues1, ((CDiskMarkDlg*)dlg)->m_RandomThreads1, MixRatio);
		maxScore = &(((CDiskMarkDlg*)dlg)->m_RandomMixScore1);
		minLatency = &(((CDiskMarkDlg*)dlg)->m_RandomMixLatency1);
		break;
#endif
	case TEST_RANDOM_READ_2:
		title.Format(L"Random Read");
		qt.Format(L"[Q=%d/T=%d]", ((CDiskMarkDlg*)dlg)->m_RandomQueues2, ((CDiskMarkDlg*)dlg)->m_RandomThreads2);
		option.Format(L"-b%dK -d%d -o%d -t%d -W0 -r -S -w0", RandomSize2, duration, ((CDiskMarkDlg*)dlg)->m_RandomQueues2, ((CDiskMarkDlg*)dlg)->m_RandomThreads2);
		maxScore = &(((CDiskMarkDlg*)dlg)->m_RandomReadScore2);
		minLatency = &(((CDiskMarkDlg*)dlg)->m_RandomReadLatency2);
		break;
	case TEST_RANDOM_WRITE_2:
		title.Format(L"Random Write");
		qt.Format(L"[Q=%d/T=%d]", ((CDiskMarkDlg*)dlg)->m_RandomQueues2, ((CDiskMarkDlg*)dlg)->m_RandomThreads2);
		option.Format(L"-b%dK -d%d -o%d -t%d -W0 -r -S -w100", RandomSize2, duration, ((CDiskMarkDlg*)dlg)->m_RandomQueues2, ((CDiskMarkDlg*)dlg)->m_RandomThreads2);
		option += bufOption;
		maxScore = &(((CDiskMarkDlg*)dlg)->m_RandomWriteScore2);
		minLatency = &(((CDiskMarkDlg*)dlg)->m_RandomWriteLatency2);
		break;
#ifdef MIX_MODE
	case TEST_RANDOM_MIX_2:
		title.Format(L"Random Mix");
		qt.Format(L"[Q=%d/T=%d]", ((CDiskMarkDlg*)dlg)->m_RandomQueues2, ((CDiskMarkDlg*)dlg)->m_RandomThreads2);
		option.Format(L"-b%dK -d%d -o%d -t%d -W0 -r -S -w%d", RandomSize2, duration, ((CDiskMarkDlg*)dlg)->m_RandomQueues2, ((CDiskMarkDlg*)dlg)->m_RandomThreads2, MixRatio);
		option += bufOption;
		maxScore = &(((CDiskMarkDlg*)dlg)->m_RandomMixScore2);
		minLatency = &(((CDiskMarkDlg*)dlg)->m_RandomMixLatency2);
		break;
#endif
	case TEST_RANDOM_READ_REAL:
		title.Format(L"Random Read");
		qt.Format(L"[Q=1/T=1]");
		option.Format(L"-b4K -d%d -o1 -t1 -W0 -r -S -w0", duration);
		maxScore = &(((CDiskMarkDlg*)dlg)->m_RandomReadScore1);
		minLatency = &(((CDiskMarkDlg*)dlg)->m_RandomReadLatency1);
		break;
	case TEST_RANDOM_WRITE_REAL:
		title.Format(L"Random Write");
		qt.Format(L"[Q=1/T=1]");
		option.Format(L"-b4K -d%d -o1 -t1 -W0 -r -S -w100", duration);
		option += bufOption;
		maxScore = &(((CDiskMarkDlg*)dlg)->m_RandomWriteScore1);
		minLatency = &(((CDiskMarkDlg*)dlg)->m_RandomWriteLatency1);
		break;
#ifdef MIX_MODE
	case TEST_RANDOM_MIX_REAL:
		title.Format(L"Random Mix");
		qt.Format(L"[Q=1/T=1]");
		option.Format(L"-b4K -d%d -o1 -t1 -W0 -r -S -w%d", duration, MixRatio);
		maxScore = &(((CDiskMarkDlg*)dlg)->m_RandomMixScore1);
		minLatency = &(((CDiskMarkDlg*)dlg)->m_RandomMixLatency1);
		break;
#endif
	}

	if (Affinity == 0)
	{
	//	AfxMessageBox(L"Affinity OFF/-n");
		option += L" -n";
	}
	else
	{
	//	AfxMessageBox(L"Affinity ON/-ag");
		option += L" -ag";
	}

	double score = 0.0;
	double latency = 0.0;

	if (maxScore == NULL || minLatency == NULL)
	{
		return ;
	}
	*maxScore = 0.0;
	*minLatency = -1.0;

	for (j = 0; j <= DiskTestCount; j++)
	{
		if (j == 0)
		{
			cstr.Format(L"Preparing... %s", title.GetString());
		}
		else
		{
			cstr.Format(L"%s [%d/%d]", title.GetString(), j, DiskTestCount);
		}
		::PostMessage(((CDiskMarkDlg*) dlg)->GetSafeHwnd(), WM_USER_UPDATE_MESSAGE, (WPARAM) &cstr, 0);
		

		command.Format(_T("\"%s\" %s -A%d -L \"%s\""), (LPTSTR)DiskSpdExe.GetString(), (LPTSTR)option.GetString(), GetCurrentProcessId(), (LPTSTR)TestFilePath.GetString());

		score = ExecAndWait((TCHAR*) (command.GetString()), TRUE, &latency) / 10 / 1000.0;

		if (j > 0 && score > *maxScore)
		{
			*maxScore = score;
			::PostMessage(((CDiskMarkDlg*) dlg)->GetSafeHwnd(), WM_USER_UPDATE_SCORE, 0, 0);
		}

		if (j > 0 && score > 0.0 && (latency < *minLatency || *minLatency < 0))
		{
			*minLatency = latency;
			::PostMessage(((CDiskMarkDlg*)dlg)->GetSafeHwnd(), WM_USER_UPDATE_SCORE, 0, 0);
		}

		if (!((CDiskMarkDlg*) dlg)->m_DiskBenchStatus)
		{
			return;
		}
	}
	::PostMessage(((CDiskMarkDlg*) dlg)->GetSafeHwnd(), WM_USER_UPDATE_SCORE, 0, 0);
}
