Windows核心編程學習九：利用內核對象進行線程同步

注：源碼爲學習《Windows核心編程》的一些嘗試，非原創。若能有助於一二訪客，幸甚。

1.程序框架

 

#include "Queue.h"
#include <tchar.h>
#include <windowsx.h>
#include <StrSafe.h>
#include <process.h>
#include "resource.h"

/************************************************************************/

#define chHANDLE_DLGMSG(hWnd, message, fn)                 \
   case (message): return (SetDlgMsgResult(hWnd, uMsg,     \
      HANDLE_##message((hWnd), (wParam), (lParam), (fn))))


// This macro function calls the C runtime's _beginthreadex function. 
// The C runtime library doesn't want to have any reliance on Windows' data 
// types such as HANDLE. This means that a Windows programmer needs to cast
// values when using _beginthreadex. Since this is terribly inconvenient, 
// I created this macro to perform the casting.
typedef unsigned (__stdcall *PTHREAD_START) (void *);

#define chBEGINTHREADEX(psa, cbStackSize, pfnStartAddr, \
   pvParam, dwCreateFlags, pdwThreadId)                 \
        ((HANDLE)_beginthreadex(                        \
         (void *)        (psa),                         \
         (unsigned)      (cbStackSize),                 \
         (PTHREAD_START) (pfnStartAddr),                \
         (void *)        (pvParam),                     \
         (unsigned)      (dwCreateFlags),               \
         (unsigned *)    (pdwThreadId)))


// Sets the dialog box icons
inline void chSETDLGICONS(HWND hWnd, int idi) 
{
   SendMessage(hWnd, WM_SETICON, ICON_BIG,  (LPARAM) 
      LoadIcon((HINSTANCE) GetWindowLongPtr(hWnd, GWLP_HINSTANCE), MAKEINTRESOURCE(idi)));
   SendMessage(hWnd, WM_SETICON, ICON_SMALL, (LPARAM) 
      LoadIcon((HINSTANCE) GetWindowLongPtr(hWnd, GWLP_HINSTANCE), MAKEINTRESOURCE(idi)));
}

 

BOOL Dlg_OnInitDialog(HWND hWnd, HWND hWndFocus, LPARAM lParam)
{
	chSETDLGICONS(hWnd, IDI_QUEUE);

	return TRUE;
}



void Dlg_OnCommand(HWND hWnd, int id, HWND hWndCtl, UINT codeNotify)
{
	switch (id)
	{
	case IDCANCEL:
		EndDialog(hWnd, id);
		break;
	}
}


INT_PTR WINAPI Dlg_Proc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
	switch (uMsg)
	{
		chHANDLE_DLGMSG(hWnd, WM_INITDIALOG, Dlg_OnInitDialog);
		chHANDLE_DLGMSG(hWnd, WM_COMMAND,	 Dlg_OnCommand);
	}

	return FALSE;
}

/*************************************************************************/

int WINAPI _tWinMain(HINSTANCE hInstExe, HINSTANCE, PTSTR pszCmdLine, int)
{
	DialogBox(hInstExe, MAKEINTRESOURCE(IDD_QUEUE), NULL, Dlg_Proc);

	return 0;
}

 

2.線程安全隊列的實現

 

 

/*
 * File:	CQueue.h
 * Time:	2013-07-10
 * 描述：	學習《Windows核心編程》
 */

#ifndef _CQUEUE_H_
#define _CQUEUE_H_

#include <windows.h>

class CQueue
{
public:
	struct ELEMENT {
		int m_nThreadNum;				// 線程號
		int m_nRequestNum;				// 請求號
	};
	typedef ELEMENT* PELEMENT;

private:
	PELEMENT	m_pElements;			// 隊列元素數組
	int			m_nMaxElements;			// 數組長度
	HANDLE		m_h[2];					// 兩個內核對象，一個互斥量一個信號量
	HANDLE		&m_hmtxQ;				// 互斥量對象的引用
	HANDLE		&m_hsemNumElements;		// 信號量對象的引用

public:
	CQueue(int nMaxElements);
	~CQueue();
	
	BOOL Append(PELEMENT pElement, DWORD dwMilliseconds);
	BOOL Remove(PELEMENT pElement, DWORD dwMilliseconds);
};


#endif

 

/************************************************************************/

CQueue::CQueue(int nMaxElements) : m_hmtxQ(m_h[0]), m_hsemNumElements(m_h[1])
{
	// 初始化結構數組
	m_pElements = (PELEMENT)HeapAlloc(GetProcessHeap(), 0, sizeof(ELEMENT) * nMaxElements);

	// 初始化數組長度
	m_nMaxElements = nMaxElements;

	// 建立互斥量和信號量內核對象
	m_hmtxQ = CreateMutex(NULL, FALSE, NULL);
	m_hsemNumElements = CreateSemaphore(NULL, 0, nMaxElements, NULL);
}


CQueue::~CQueue()
{
	// 清理內核對象和內存
	CloseHandle(m_hsemNumElements);
	CloseHandle(m_hmtxQ);
	HeapFree(GetProcessHeap(), 0, m_pElements);
}

BOOL CQueue::Append(CQueue::PELEMENT pElement, DWORD dwTimeout)
{
	BOOL fOk = FALSE;

	// 等待互斥量內核對象
	DWORD dw = WaitForSingleObject(m_hmtxQ, dwTimeout);

	// 返回WAIT_OBJECT_0表示獲得了隊列的獨佔訪問權
	if (dw == WAIT_OBJECT_0) {
		LONG lPrevCount;

		// 嘗試向隊列添加新元素
		fOk = ReleaseSemaphore(m_hsemNumElements, 1, &lPrevCount);

		// fOk爲TRUE表示隊列沒滿，能夠添加新元素
		if (fOk) {
			m_pElements[lPrevCount] = *pElement;
		}
		else {
			SetLastError(ERROR_DATABASE_FULL);
		}

		// 釋放互斥量，容許其餘線程訪問隊列
		ReleaseMutex(m_hmtxQ);
	}
	else {
		SetLastError(ERROR_TIMEOUT);
	}

	return fOk;
}

BOOL CQueue::Remove(CQueue::PELEMENT pElement, DWORD dwTimeout)
{
	// 確保對隊列具備獨佔訪問權，且要求隊列中有元素可取
	// 等待信號量成功的反作用使它的計數減一，因此不須要顯示調用ReleaseSemaphore()
	BOOL fOk = (WaitForMultipleObjects(_countof(m_h), m_h, TRUE, dwTimeout) == WAIT_OBJECT_0);
	
	if (fOk)
	{
		// 獲取元素
		*pElement = m_pElements[0];

		// 取出索引爲0 的元素，把數組中剩餘元素向前挪一個位置
		MoveMemory(&m_pElements[0], &m_pElements[1], sizeof(ELEMENT) * (m_nMaxElements - 1));

		// 釋放互斥量，容許其餘線程訪問隊列
		ReleaseMutex(m_hmtxQ);
	}
	else
	{
		SetLastError(ERROR_TIMEOUT);
	}

	return fOk;
}

 

 

3.客戶端線程

 

 

CQueue				g_q(10);
volatile LONG		g_fShutdown = FALSE;
HWND				g_hWnd;

// Handles to all reader/writer threads
HANDLE g_hThreads[MAXIMUM_WAIT_OBJECTS];

// Number of reader/writer threads  
int    g_nNumThreads = 0;

/************************************************************************/

DWORD WINAPI ClientThread(PVOID pvParam)
{
	int nThreadNum = PtrToUlong(pvParam);			// 線程號
	HWND hWndLB = GetDlgItem(g_hWnd, IDC_CLIENT);

	int nRequestNum = 0;
	while (1 != InterlockedCompareExchange(&g_fShutdown, 0, 0))
	{
		nRequestNum++;
		
		TCHAR sz[1024];
		CQueue::ELEMENT e = { nThreadNum, nRequestNum };

		// 嘗試添加元素到隊列
		if (g_q.Append(&e, 200)) {
			StringCchPrintf(sz, _countof(sz), TEXT("客戶線程%d 添加元素%d"), nThreadNum, nRequestNum);
		}
		else {
			StringCchPrintf(sz, _countof(sz), TEXT("客戶線程%d 添加元素%d失敗(%s)"), nThreadNum, nRequestNum,
				(GetLastError() == ERROR_TIMEOUT) ? TEXT("超時") : TEXT("隊列已滿"));
		}

		ListBox_SetCurSel(hWndLB, ListBox_AddString(hWndLB, sz));
		Sleep(2500);
	}

	return 0;
}

 

BOOL Dlg_OnInitDialog(HWND hWnd, HWND hWndFocus, LPARAM lParam)
{
	chSETDLGICONS(hWnd, IDI_QUEUE);

	g_hWnd = hWnd;

	DWORD dwThreadID;

	// 建立客戶端線程
	for (int i = 0; i < 4; i++)
		g_hThreads[g_nNumThreads++] = chBEGINTHREADEX(NULL, 0, ClientThread, (PVOID)(INT_PTR) i, 0, &dwThreadID);


	return TRUE;
}

 

4.服務器端線程

 

DWORD WINAPI ServerThread(PVOID pvParam)
{
	int nThreadNum = PtrToUlong(pvParam);
	HWND hWndLB = GetDlgItem(g_hWnd, IDC_SERVER);

	while (1 != InterlockedCompareExchange(&g_fShutdown, 0, 0)) {
		TCHAR sz[1024];
		CQueue::ELEMENT e;

		// 嘗試移除元素
		if (g_q.Remove(&e, 5000)) {
			StringCchPrintf(sz, _countof(sz), TEXT("服務器端線程%d移除客戶端線程%d放入的元素%d"), nThreadNum, e.m_nThreadNum, e.m_nRequestNum);
			Sleep(2000 * e.m_nThreadNum);
		}
		else {
			StringCchPrintf(sz, _countof(sz), TEXT("服務器端線程%d 沒有元素可取"), nThreadNum);
		}

		ListBox_SetCurSel(hWndLB, ListBox_AddString(hWndLB, sz));
	}

	return 0;
}

 

BOOL Dlg_OnInitDialog(HWND hWnd, HWND hWndFocus, LPARAM lParam)
{
	chSETDLGICONS(hWnd, IDI_QUEUE);

	g_hWnd = hWnd;

	DWORD dwThreadID;

	// 建立客戶端線程
	for (int i = 0; i < 4; i++)
		g_hThreads[g_nNumThreads++] = chBEGINTHREADEX(NULL, 0, ClientThread, (PVOID)(INT_PTR) i, 0, &dwThreadID);

	// 建立讀者線程
	for (int i = 0; i < 2; i++)
		g_hThreads[g_nNumThreads++] = chBEGINTHREADEX(NULL, 0, ServerThread, (PVOID)(INT_PTR) i, 0, &dwThreadID);

	return TRUE;
}

 

int WINAPI _tWinMain(HINSTANCE hInstExe, HINSTANCE, PTSTR pszCmdLine, int)
{
	DialogBox(hInstExe, MAKEINTRESOURCE(IDD_QUEUE), NULL, Dlg_Proc);


	InterlockedExchange(&g_fShutdown, TRUE);


	WaitForMultipleObjects(g_nNumThreads, g_hThreads, TRUE, INFINITE);
	while (g_nNumThreads--)
		CloseHandle(g_hThreads[g_nNumThreads]);


	return 0;
}