上课笔记set_new_handler的实现

#include<iostream>
//#include<new.h>
#include<malloc.h>
using namespace std;

void out_of_memory()
{
    cout<<"Out Of Memory!"<<endl;
    //exit(1);
//    return -1;
};

typedef void(* new_handler)();
void (* cur_handler)()=0;
void(* my_set_new_handler(void(* p)()))()
{
    void (* old)()=cur_handler;
    cur_handler=p;
    return old;
}
void *operator new(size_t size)
{
    if(size==0)
    size=1;
    while(1)
    {
        char* p=(char *)malloc(size*64);
        if(p!=NULL)
            return p;
        void (* current)()=my_set_new_handler(0);
        my_set_new_handler(current);
        if(current!=NULL)
            (*current)();
        else
            throw std::bad_alloc();
    }
}

void main()
{        
    my_set_new_handler(out_of_memory);
    int *p = new int[10479149179]; //回调方式  operator new
    //int *p = new int[10];
    if(p == NULL)
    {
        cout<<"Out Of Memery."<<endl;
        exit(1);
    }
    cout<<"OK."<<endl;
}