michael/lean2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add tests for memory.cpp

Browse source 
Signed-off-by: Leonardo de Moura <leonardo@microsoft.com>
This commit is contained in:
Leonardo de Moura 2013-09-20 22:37:13 -07:00
parent 7ac94ee976
commit d29ec9ab6f
3 changed files with 40 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
src/tests/util/CMakeLists.txt
View file

@ -31,3 +31,6 @@ add_test(pdeque ${CMAKE_CURRENT_BINARY_DIR}/pdeque)
add_executable(pvector pvector.cpp)
target_link_libraries(pvector ${EXTRA_LIBS})
add_test(pvector ${CMAKE_CURRENT_BINARY_DIR}/pvector)
add_executable(memory memory.cpp)
target_link_libraries(memory ${EXTRA_LIBS})
add_test(memory ${CMAKE_CURRENT_BINARY_DIR}/memory)

35
src/tests/util/memory.cpp Normal file
View file

@ -0,0 +1,35 @@
/*
Copyright (c) 2013 Microsoft Corporation. All rights reserved.
Released under Apache 2.0 license as described in the file LICENSE.
Author: Leonardo de Moura
*/
#include <iostream>
#include "util/test.h"
#include "util/memory.cpp"
static void tst1() {
std::cout << "Initial: " << lean::get_allocated_memory() << "\n";
size_t old_mem = lean::get_allocated_memory();
int N = 5;
int * a = static_cast<int*>(lean::malloc(N * sizeof(N)));
lean_assert(lean::get_allocated_memory() >= old_mem + N * sizeof(N));
for (int i = 0; i < N; i++) {
a[i] = i;
}
a = static_cast<int*>(lean::realloc(a, N * 2 * sizeof(N)));
lean_assert(lean::get_allocated_memory() >= old_mem + N * 2 * sizeof(N));
std::cout << "Total: " << static_cast<size_t>(lean::get_allocated_memory()) << "\n";
std::cout << "Thread: " << static_cast<size_t>(lean::get_thread_allocated_memory()) << "\n";
lean_assert(lean::get_allocated_memory() == static_cast<size_t>(lean::get_thread_allocated_memory()));
for (int i = 0; i < N; i++) {
lean_assert(a[i] == i);
}
lean::free(a);
lean_assert(old_mem == lean::get_allocated_memory());
}
int main() {
tst1();
return lean::has_violations() ? 1 : 0;
}

4
src/util/memory.cpp
View file

@ -88,8 +88,8 @@ void * save_alloc_size(void * ptr, size_t sz) {
}
}
inline size_t malloc_size(void * ptr) { return static_cast<size_t*>(ptr)[-1]; }
inline void * malloc_core(size_t sz) { return save_alloc_size(::malloc(sz + sizeof(size_t)), sz); }
inline void * realloc_core(void * ptr, size_t sz) { return save_alloc_size(::realloc(static_cast<size_t*>(ptr) - 1, sz + sizeof(size_t)), sz); }
inline void * malloc_core(size_t sz) { return save_alloc_size(::malloc(sz + sizeof(sz)), sz); }
inline void * realloc_core(void * ptr, size_t sz) { return save_alloc_size(::realloc(static_cast<size_t*>(ptr) - 1, sz + sizeof(sz)), sz); }
inline void free_core(void * ptr) { ::free(static_cast<size_t*>(ptr) - 1); }
}
#endif