sl@0
|
1 |
// Copyright (c) 2008-2009 Nokia Corporation and/or its subsidiary(-ies).
|
sl@0
|
2 |
// All rights reserved.
|
sl@0
|
3 |
// This component and the accompanying materials are made available
|
sl@0
|
4 |
// under the terms of "Eclipse Public License v1.0"
|
sl@0
|
5 |
// which accompanies this distribution, and is available
|
sl@0
|
6 |
// at the URL "http://www.eclipse.org/legal/epl-v10.html".
|
sl@0
|
7 |
//
|
sl@0
|
8 |
// Initial Contributors:
|
sl@0
|
9 |
// Nokia Corporation - initial contribution.
|
sl@0
|
10 |
//
|
sl@0
|
11 |
// Contributors:
|
sl@0
|
12 |
//
|
sl@0
|
13 |
// Description:
|
sl@0
|
14 |
// Name : new_handler_test.cpp
|
sl@0
|
15 |
// Part of : standard c++ tests.
|
sl@0
|
16 |
//
|
sl@0
|
17 |
//
|
sl@0
|
18 |
|
sl@0
|
19 |
#include <new>
|
sl@0
|
20 |
#include "test_decls.h"
|
sl@0
|
21 |
|
sl@0
|
22 |
int one_huge_chunk = 0xa000;
|
sl@0
|
23 |
int *last_huge_chunk=NULL;
|
sl@0
|
24 |
|
sl@0
|
25 |
int glob_val = 0;
|
sl@0
|
26 |
void foo()
|
sl@0
|
27 |
{
|
sl@0
|
28 |
/*
|
sl@0
|
29 |
* This is the new handler and it frees the last successful alloc so
|
sl@0
|
30 |
* that the allocation in operator new has some free memory.
|
sl@0
|
31 |
*/
|
sl@0
|
32 |
glob_val++;
|
sl@0
|
33 |
delete [] last_huge_chunk;
|
sl@0
|
34 |
|
sl@0
|
35 |
}
|
sl@0
|
36 |
|
sl@0
|
37 |
void bar()
|
sl@0
|
38 |
{
|
sl@0
|
39 |
last_huge_chunk = new int[one_huge_chunk];
|
sl@0
|
40 |
}
|
sl@0
|
41 |
|
sl@0
|
42 |
int main()
|
sl@0
|
43 |
{
|
sl@0
|
44 |
std::new_handler h_old, h_new;
|
sl@0
|
45 |
|
sl@0
|
46 |
try{
|
sl@0
|
47 |
while(1){
|
sl@0
|
48 |
// Keep allocating until we reach OOM condition. At OOM
|
sl@0
|
49 |
// the default new handler throws std::bad_alloc
|
sl@0
|
50 |
bar();
|
sl@0
|
51 |
}
|
sl@0
|
52 |
}catch(std::bad_alloc ba)
|
sl@0
|
53 |
{
|
sl@0
|
54 |
}
|
sl@0
|
55 |
|
sl@0
|
56 |
/*
|
sl@0
|
57 |
* Once the handler is set, failure of 'new' will call the handler to
|
sl@0
|
58 |
* get some free memory...
|
sl@0
|
59 |
*/
|
sl@0
|
60 |
h_new = (std::new_handler)&foo;
|
sl@0
|
61 |
|
sl@0
|
62 |
h_old = std::set_new_handler(h_new);
|
sl@0
|
63 |
|
sl@0
|
64 |
try {
|
sl@0
|
65 |
/*
|
sl@0
|
66 |
* Try once more to see if our handler actually freed up some memory
|
sl@0
|
67 |
*/
|
sl@0
|
68 |
bar();
|
sl@0
|
69 |
}catch(...){
|
sl@0
|
70 |
glob_val = 0;
|
sl@0
|
71 |
}
|
sl@0
|
72 |
|
sl@0
|
73 |
CPP_TESTS_ASSERT_ALLWAYS(glob_val != 0);
|
sl@0
|
74 |
|
sl@0
|
75 |
h_new = std::set_new_handler(h_old);
|
sl@0
|
76 |
return 0;
|
sl@0
|
77 |
}
|