C / C++ / MFC

#include <windows.h>
#include <stdio.h>

LPCSTR FormatSize(QWORD nSize) {
	static char szSize[16];
	if (nSize < 1024) {
		sprintf(szSize, "%uB", nSize);
	} else {
		char szScales[] = "KMGTPEZY";
		int nScale = 0;
		for (; nScale < sizeof(szScales); ++nScale) {
			if (nSize < 1024 * 1024) {
				sprintf(szSize, "%6.2f%cB", nSize / 1024.0f, szScales[nScale]);
				break;
			}
			nSize = (nSize + 512) / 1024;
		}
		if (nScale == sizeof(szScales)) {
			strcpy(szSize, "Unknown Size");
		}
	}
	return szSize;
}

void PrintDrives(char *szDrives) {
	DWORD nDrives = GetLogicalDrives();
	char szDrive[] = "A:\\";
	char szDriveName[MAX_PATH + 1];
	int nPrintedDrives = 0;
	while (nDrives != 0) {
		if (nDrives & 1) {
			szDrives[nPrintedDrives] = szDrive[0];
			GetVolumeInformation(szDrive, szDriveName, sizeof(szDriveName), NULL, NULL, NULL, NULL, 0);
			QWORD nAvaliable;
			QWORD nTotal;
			QWORD nFree;
			if (!GetDiskFreeSpaceEx(szDrive, (PULARGE_INTEGER)&nAvaliable, (PULARGE_INTEGER)&nTotal, (PULARGE_INTEGER)&nFree)) {
				nAvaliable = 0;
				nTotal = 0;
				nFree = 0;
			}
			if (*szDriveName == 0) {
				if (nTotal > 0) {
					printf("%s (%s)\n", szDrive, FormatSize(nTotal));
				} else {
					printf("%s (Unknown Size)\n", szDrive);
				}
			} else {
				if (nTotal > 0) {
					printf("%s (%s) [%s]\n", szDrive, FormatSize(nTotal), szDriveName);
				} else {
					printf("%s (Unknown Size) [%s]\n", szDrive, szDriveName);
				}
			}
		}
		nDrives >>= 1;
		++szDrive[0];
	}
}

int main(int argc, char *argv[]) {
	char szDrives[27];
	szDrives[26] = 0;
	PrintDrives(szDrives);
	return 0;
}

#define CM_KILL_THREAD WM_USER + 0 //can go up to WM_USER + 0x7BFF

class CTest_Thread : public CWinThread {
	//other member functions and variables
	protected:
		virtual BOOL PreTranslateMessage(MSG *pMsg);
};

BOOL CTest_Thread::PreTranslateMessage(MSG *pMsg) {
	if (pMsg->message == CM_KILL_THREAD) {
		ExitThread(0); //0 is the return code, if you need to send a message back to your application
		return TRUE; //I have procesed the message, nothing else needs to
	}
	return FALSE; //get something else to process the message
}

Th->PostThreadMessage(CM_KILL_THREAD, 0, 0);

Andrew Brock wrote:

ExitThread(0); //0 is the return code, if you need to send a message back to your application
return TRUE; //I have procesed the message, nothing else needs to

BOOL CTest_Thread::PreTranslateMessage(MSG* pMsg)
{
	if (pMsg->message == ERR_KILL_THREAD) 
	{
		ExitThread(0); //0 is the return code, if you need to send a message back to your application
		return TRUE; //I have procesed the message, nothing else needs to
	}
	return FALSE; //get something else to process the message
	
	//return CWinThread::PreTranslateMessage(pMsg);
}

CCriticalSection sec_s;
CSingleLock lock(&sec_s);

// Lock
lock.Lock();

//do something

// un Lock
lock.Unlock();

Th->PostThreadMessage(CM_KILL_THREAD, 0, 0);
switch (WaitForSingleObject(Th->m_hThread, 5000)) { //5000 = 5sec, this can be INFINITE, but if the thread never exits, then this will hang for ever
	case WAIT_OBJECT_0:
		TRACE("Thread terminated.\n");
		break;

	case WAIT_TIMEOUT:
		TRACE("Wait for terminate timed out.\n");
		//You may consider using TerminateThread here if you REALLY REALLY want the thread gone
		break;

	default:
		TRACE("WaitForSingleObject failed with error %d.\n", GetLastError());
		break;
}
//Now that the thread is gone, we can continue, say get the exit code that was passed to ExitThread()
DWORD nRetCode;
GetExitCodeThread(hRemoteThread, &nRetCode);

DWORD KillThreadAndWait(CTest_Thread *pThread, DWORD nTimeoutMs = 5000) {
	HANDLE hThread = pThread->m_hThread;
	pThread->PostThreadMessage(CM_KILL_THREAD, 0, 0);
	switch (WaitForSingleObject(hThread, nTimeoutMs)) {
		case WAIT_OBJECT_0:
			TRACE("Thread terminated.\n");
			break;

		case WAIT_TIMEOUT:
			TRACE("Wait for terminate timed out.\n");
			return 0x80000000; //This can be anything that you can identify an error as.

		default:
			TRACE("WaitForSingleObject failed with error %d.\n", GetLastError());
			return 0x80000001; //This can be anything that you can identify an error as.
	}
	//Now that the thread is gone, we can continue, say get the exit code that was passed to ExitThread()
	DWORD nRetCode;
	GetExitCodeThread(hThread, &nRetCode);
	return nRetCode;
}

CTest_Thread *pThread;
pThread = (CTest_Thread *)AfxBeginThread(RUNTIME_CLASS(CTest_Thread), THREAD_PRIORITY_NORMAL, 0, CREATE_SUSPENDED, NULL);
if (pThread != NULL) {			
	pThread->ResumeThread();
}
TRACE("Thread exited with code 0x%08X\n", KillThreadAndWait(pThread, 5000));

BOOL CTest_Thread::PreTranslateMessage(MSG *pMsg) {
	if (pMsg->message == CM_KILL_THREAD) {
		//CORRECTION: MSDN says to use AfxEndThread when using AfxBeginThread to terminate the thread
		AfxEndThread(0); //0 is the return code, if you need to send a message back to your application
		return TRUE; //I have procesed the message, nothing else needs to
	}
	return FALSE; //get something else to process the message
}

Mike Certini wrote:

What is happening is that after I select two menu selections or two loops of the while statement the program skips over the scanf statement.
Can someone help me with this issue?