New操作符重载

[toc]

C++中 new 操作符的重载

在C++中，new 运算符可以被重载，以提供自定义的内存分配行为。重载 new 运算符的函数原型如下：

void* operator new(std::size_t size) throw(std::bad_alloc);

此外，还可以重载 new[] 运算符（用于分配数组）和 delete/delete[] 运算符（用于释放内存）。这些重载函数的原型如下：

// new
void* operator new(std::size_t size) throw(std::bad_alloc);
void* operator new[](std::size_t size) throw(std::bad_alloc);

// delete
void operator delete(void* ptr) noexcept;
void operator delete[](void* ptr) noexcept;

注意，从C++11开始，抛出异常的重载版本已被弃用，建议使用不抛出异常的版本。不抛出异常的版本如下：

// new
void* operator new(std::size_t size) noexcept(true);
void* operator new[](std::size_t size) noexcept(true);

// delete
void operator delete(void* ptr) noexcept;
void operator delete[](void* ptr) noexcept;

此外，还可以为类重载 new/delete 运算符，这样就可以为该类的对象提供自定义的内存分配行为。一般在做 windows 驱动开发时候，使用自定义类型的动态内存构建，可以对new/delete 重载，类成员重载函数的原型如下：

class MyClass {
public:
    // new
    static void* operator new(std::size_t size) noexcept(true);
    static void* operator new[](std::size_t size) noexcept(true);

    // delete
    static void operator delete(void* ptr) noexcept;
    static void operator delete[](void* ptr) noexcept;
};

请注意，重载 new/delete 运算符时，应确保正确处理内存分配和释放，以避免内存泄漏和其他问题。

Windows 驱动开发中重写 new/delete

#pragma once

// 基类定义new/delete操作符
class Base
{

public:

    // 重定义 new 操作符.
    static void* __cdecl operator new(size_t size, unsigned tag = 'base') noexcept;

    // 重定义 delete 操作符.
    static void __cdecl operator delete(void* ptr) noexcept;

    // 析构函数.
    virtual ~Base() {}
};

#include "Base.h"
#include 

void* __cdecl Base::operator new(size_t size, unsigned tag) noexcept
{
    DbgPrint("[Base]Operator new run.rn");
    return ExAllocatePool2(POOL_FLAG_PAGED, size, tag);
}

void __cdecl Base::operator delete(void* ptr) noexcept
{
    DbgPrint("[Base]Operator delete run.rn");
    ExFreePool(ptr);
}

#include "Base.h"

class Derived : public Base
{
public:
    void test() {
        DbgPrint("This is a test.");   
    }
}

// 调用代码如下
auto p = new Derived(); // new ('abcd')Derived();
p->test();
delete p;