WIN 程序员的 Linux 互斥类

{
    //作用域的开始声明局部变量
    auto_entercs ace(&m_cs);

    //互斥区域代码。。。

    //离开作用域的时候自动释放控制权

}

#ifndef _X_HSS_CRITICAL_SECTION_HSS__
#define _X_HSS_CRITICAL_SECTION_HSS__

/**************************************************************************************************用于 win32 和 linux 下的通用的互斥类，如下的使用代码在 win 下和 linux 下使用同样的代码使用互斥

  用法：

  (0) 包含
  #include "auto_entercs.h"

  (1) 定义
  CRITICAL_SECTION m_cs;

  (2) 初始化
  InitializeCriticalSection(&m_cs);

  (3) 进入和离开互斥
  {
    auto_entercs ace(&m_cs);
    ....互斥区域
  }

  (4) 删除
  DeleteCriticalSection(&m_cs);

作者： 黄山松，http://www.cnblogs.com/tomview/

\**************************************************************************************************/
#ifndef WIN32

#include <windows.h>

#else

#include <pthread.h>

#define CRITICAL_SECTION            pthread_mutex_t
#define InitializeCriticalSection(p)            \
        {            pthread_mutexattr_t attr;            pthread_mutexattr_init(&attr);            pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE_NP);            pthread_mutex_init((p), &attr);            }    
#define DeleteCriticalSection(p)    pthread_mutex_destroy(p)
#define EnterCriticalSection(p)        pthread_mutex_lock(p)
#define LeaveCriticalSection(p)        pthread_mutex_unlock(p)

#endif

//auto_entercs 需要在互斥代码区域内声明局部变量，当代码执行出区域的时候，析构函数自动调用LeaveCriticalSection
class auto_entercs
{
public:
    auto_entercs(CRITICAL_SECTION* pcs)
    {
        m_pcs = pcs;
        
        if (m_pcs)
        {
            EnterCriticalSection(m_pcs);
        }
    }

    ~auto_entercs()
    {
        if (m_pcs)
        {
            LeaveCriticalSection(m_pcs);
        }
    }

    CRITICAL_SECTION* m_pcs;
};
 
#endif