// DeadLockTest.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include "DeadLockTest.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif

// The one and only application object

CWinApp theApp;

using namespace std;

CRITICAL_SECTION cs1;
CRITICAL_SECTION cs2;
CRITICAL_SECTION csprint;

//初始化关键代码段
void InitMyCriticalSection();
//删除关键代码段
void DeleteMyCriticalSection();
//打印信息
void PrintString(const CString& strInfo);

DWORD WINAPI Thread1(LPVOID lpParameter);
DWORD WINAPI Thread2(LPVOID lpParameter);

int _tmain(int argc, TCHAR* argv[], TCHAR* envp[])
{
    int nRetCode = 0;

    // initialize MFC and print and error on failure
    if (!AfxWinInit(::GetModuleHandle(NULL), NULL, ::GetCommandLine(), 0))
    {
        // TODO: change error code to suit your needs
        _tprintf(_T("Fatal Error: MFC initialization failed\n"));
        nRetCode = 1;

        return nRetCode;
    }

    //初始化关键代码段
    InitMyCriticalSection();

    //创建线程
    HANDLE hThread1 = CreateThread(NULL, 0, Thread1, NULL, 0, NULL);
    HANDLE hThread2 = CreateThread(NULL, 0, Thread2, NULL, 0, NULL);

    //等待线程结束
    WaitForSingleObject(hThread1, INFINITE);
    WaitForSingleObject(hThread2, INFINITE);

    //关闭线程句柄
    CloseHandle(hThread1);
    CloseHandle(hThread2);

    //释放关键代码段
    DeleteMyCriticalSection();

    return nRetCode;
}

void InitMyCriticalSection()
{
    InitializeCriticalSection(&cs1);
    InitializeCriticalSection(&cs2);
    InitializeCriticalSection(&csprint);
}

void DeleteMyCriticalSection()
{
    DeleteCriticalSection(&cs1);
    DeleteCriticalSection(&cs2);
    DeleteCriticalSection(&csprint);
}

DWORD WINAPI Thread1(LPVOID lpParameter)
{
    for (int i = 0; i < 5; i++)
    {
        EnterCriticalSection(&cs1);
        Sleep(500);
        EnterCriticalSection(&cs2);

        PrintString(_T("Thread1"));

        LeaveCriticalSection(&cs2);
        LeaveCriticalSection(&cs1);
    }

    return 1;
}

DWORD WINAPI Thread2(LPVOID lpParameter)
{
    for (int i = 0; i < 5; i++)
    {
        EnterCriticalSection(&cs2);
        Sleep(500);
        EnterCriticalSection(&cs1);

        PrintString(_T("Thread2"));

        LeaveCriticalSection(&cs1);
        LeaveCriticalSection(&cs2);
    }

    return 1;
}

void PrintString(const CString& strInfo)
{
    EnterCriticalSection(&csprint);
    wcout<<(const TCHAR*)strInfo<<endl;
    LeaveCriticalSection(&csprint);
}