防止可重入：一个有趣的类和一个很有用的宏！

这几天在研究Todolist源码，看到一简短的类，很很有意思，现把源码列下，并附上简短说明：

 

 

// macro to help prevent re-entrancy in functions //定义一个宏AF_NOREENTRANT，防止函数中可重入 #define AF_NOREENTRANT / static BOOL bAFInHere = FALSE; / if (bAFInHere) / return; / CAutoFlag af(bAFInHere, TRUE); class CAutoFlag { public: CAutoFlag(BOOL& bFlag, BOOL bState); //bFlag为引用参数 virtual ~CAutoFlag(); protected: BOOL& m_bFlag; //引用内部成员 BOOL m_bInitialState; };   

 

      // // 构造时m_bFlag初始化为bFlag的引用，设置m_bFlag(bFlag)的值为bState， // CAutoFlag::CAutoFlag(BOOL& bFlag, BOOL bState) : m_bFlag(bFlag) { m_bInitialState = bFlag; m_bFlag = bState; //此时m_bFlag初始化为bFlag的引用 } CAutoFlag::~CAutoFlag() { m_bFlag = m_bInitialState; //析构时bFlag恢复原值。 } 下面我们看一下函数及宏的具体应用： void CToDoListWnd::OnTimerAutoSave() { AF_NOREENTRANT // 防止OnTimerAutoSave同时被多个地方调用。 // don't save if the user is editing a task label if (!GetToDoCtrl().IsTreeLabelEditing()) SaveAll(FALSE, FALSE, FALSE, FALSE); } int CCheckComboBox::GetCurSel() const { CAutoFlag af(m_bDrawing, FALSE); //把m_bDrawing设为FALSE，函数执行完后自动恢复原值。 return CComboBox::GetCurSel(); }

 

 

注：

重入即表示重复进入，首先它意味着这个函数可以被中断，其次意味着它除了使用自己栈上的变量以外不依赖于任何环境（包括static），这样的函数就是purecode（纯代码）可重入，可以允许有该函数的多个副本在运行，由于它们使用的是分离的栈，所以不会互相干扰。如果确实需要访问全局变量（包括static），一定要注意实施互斥手段。可重入函数在并行运行环境中非常重要，但是一般要为访问全局变量付出一些性能代价。

 

在多线程（或多进程）程序设计中，“重入”是指多个线程或进程“同时”访问某个代码段（通常指某一函数）。因此，“可重入”就是指这个代码段支持这样的操作而不会产生任何问题。无状态的代码段肯定是可重入的，比如求两个数的平方的函数，但如果代码段共享某些资源（比如共享变量），则要达到可重入，必须使用一些同步和或互斥机制。

 

在单任务的系统中同样有函数重入的问题，因为c/c＋＋函数支持中断和回调，为此编译器在函数调用时将返回值的地址做为一个参数压栈；这就意味这，在你的函数执行的任何时候中断都有可能发生，所以你在写代码的时候要加以小心，尤其是在设计中断服务程序的时候。如果你想支持重入，最好写成纯代码（与任何全局变量无关），不得已用到全局变量，要对临界区加锁；如果你不想重入就关中断，并保证你的函数是唯一实例运行。