Question

我在以下代码中了解有关警卫标志的麻烦。

// User-defined operator new.  
    void *operator new( size_t stAllocateBlock ) {  
       static int fInOpNew = 0;   // Guard flag.  

       if ( fLogMemory && !fInOpNew ) {  
          fInOpNew = 1;  
          clog << "Memory block " << ++cBlocksAllocated  
              << " allocated for " << stAllocateBlock  
              << " bytes\n";  
          fInOpNew = 0;  
       }  
       return malloc( stAllocateBlock );  
    }

任何人都有帮助理解这一点？

由于

编辑：感谢您清除我的疑问，我现在明白了。我从这里得到了这段代码：https://msdn.microsoft.com/en-us/library/kftdy56f.aspx，他们也使用相同的守卫进行删除。原因是一样的吗？

请检查我的代码，并分享您的建议以改进代码。另外，如何用多个线程保护代码？我真的很感激这一点，它将清楚我对线程安全代码的理解，以及内存管理。

由于

我的代码版本：

/*
 * Memory leak detector
 */

#include <iostream> // for cout, clog
#include <new>      // for size_t

using std::clog;
using std::cout;


/* Logging toggle, 0x0 - No, 0xFF - Yes */
char    logMemory;

/* Number of Memory blocks allocated */
int     memoryBlocksAllocated;

void* AllocateMemory(size_t size)
{
    /*
     * Guard flag for protecting from reentrancy (recursion),
     * as clog also make use of new operator for
     * allocating memory to the buffer.
     */

    static char InOperationNew = 0x0;

    if ( logMemory && !InOperationNew ) {
        InOperationNew = 0xFF;
        clog << "Memory block allocated " << ++memoryBlocksAllocated
        <<" for requested size " << size << " bytes\n";
        InOperationNew = 0x0;
    }

    return malloc(size);
}

void ReleaseMemory(void *ptr)
{
    /*
     * Guard flag for protecting from reentrancy (recursion),
     * as clog also make use of new operator for
     * allocating memory to the buffer.
     */

    static char InOperationDelete = 0x0;


    if ( logMemory && !InOperationDelete ) {
        InOperationDelete = 0xFF;
        clog << "Memory block deallocated " << memoryBlocksAllocated-- << "\n";
        InOperationDelete = 0x0;
    }

    free(ptr);
}

/* User defined(overriden) global scope new operator */
void* operator new (size_t size)
{
    return AllocateMemory(size);
}

/* User defined(overriden) global scope new[] array operator */
void* operator new[] (size_t size)
{
    return AllocateMemory(size);
}


/* User defined(overriden) global scope delete operator */
void operator delete (void *ptr)
{
    ReleaseMemory(ptr);
}

/* User defined(overriden) global scope delete[] array operator */
void operator delete[] (void *ptr)
{
    ReleaseMemory(ptr);
}

int main()
{
    logMemory = 0xFF;   // Enable logging

    // Allocate and destroy an array of objects
    char *objArray = new char[2];
    if ( objArray )
    {
        delete[] objArray;
    }
    else {
        cout << "Memory allocation failed\n";
        return -1;
    }

    // Allocate and destroy an object
    char *obj = new char;
    if ( obj )
    {
        delete obj;
    }
    else {
        cout << "Memory allocation failed\n";
        return -1;
    }

    logMemory = 0x0;    // Disable logging

    return 0;
}

Answer 1

据我所知，这是一种避免多线程破坏日志流的尝试。

假设两个线程同时调用该函数。正在打印的消息将混合（取决于运行时使用的缓冲机制）。

为了避免这种情况，程序员创建了一个static标志。函数本地的静态变量在所有调用中共享。

目的是当一个调用正在记录消息时，它将首先将标志设置为1.这将不允许其他线程进入。（看看！fInOpNew）

完成打印后，它会将标志设置回0。

虽然这种机制在避免竞争条件下无法奏效。由于两个线程可以进入if条件，然后可以设置标志。

程序员需要使用比较和交换等原子基元来确保对关键部分的保护。

使用静态变量也不安全。如果两个线程同时写入它，它可以得到一个不等于0或1的值。

编辑：正如Toby Speight所建议的那样，避免再入境的旗帜的另一个目的。调用＆lt;＆lt; clog上的operator也是一个可以使用new运算符的函数（可以是相同的new运算符）。这将导致此函数的无限递归。该标志确保在＆lt;＆lt;＆lt;＆lt;＆lt;＆lt;操作员呼叫新的。

程序是单线程的假设不会带来竞争条件。

处理内存分配时的Guard标志

1 个答案: