void print_allocator_stat(const memory::allocator* allocator)
{
const memory::statistics* stat = allocator->get_statistics();
assert(stat != nullptr);
size_t heap_alloc = 0;
size_t heap_free = 0;
for (size_t i = 0; i < stat->get_heap_count(); ++i)
{
const memory::heap_statistics_t* heap = stat->get_heap(i);
assert(heap != nullptr);
if (heap != nullptr)
{
::wprintf(L"heap[%u]: alloc: %u, free: %u\n", i, heap->alloc, heap->free);
heap_alloc += heap->alloc;
heap_free += heap->free;
}
}
size_t lookaside_alloc = 0;
size_t lookaside_free = 0;
for (size_t i = 0; i < stat->get_cache_count(); ++i)
{
const memory::cache_statistics_t* cache = stat->get_cache(i);
assert(cache != nullptr);
if (cache != nullptr)
{
size_t cache_alloc = 0;
size_t cache_free = 0;
for (size_t j = 0; j < cache->count; ++j)
{
const memory::lookaside_statistics_t* lookaside = &cache->lookaside[j];
cache_alloc += lookaside->alloc;
cache_free += lookaside->free;
}
lookaside_alloc += cache_alloc;
lookaside_free += cache_free;
::wprintf(L"cache[%u]: alloc: %u, free: %u\n", i, cache_alloc, cache_free);
}
}
::wprintf(L"\nheap(%u) alloc: %u, free: %u\n", stat->get_heap_count(), heap_alloc, heap_free);
::wprintf(L"lookaside(%u) alloc: %u, free: %u\n", stat->get_cache_count(), lookaside_alloc, lookaside_free);
::wprintf(L"total alloc: %u, total free: %u\n", heap_alloc + lookaside_alloc, heap_free + lookaside_free);
}
int main()
{
#if defined(CRT_LFH)
::wprintf(L"crt lfh\n");
set_lfh();
#elif defined(TBBMALLOC)
::wprintf(L"tbbmalloc\n");
#elif defined(TCMALLOC)
::wprintf(L"tcmalloc\n");
#elif defined(JEMALLOC)
::wprintf(L"jemalloc\n");
set_jemalloc();
#elif defined(MS_CONCURRENCY)
::wprintf(L"ms concurrency\n");
#elif defined(MY_ALLOCATOR)
::wprintf(L"my allocator\n");
if (allocator.create(true, 0) == false)
{
::wprintf(L"allocator.create() failed.\n");
return 0;
}
#endif
// for crt tbb tcmalloc
char* dummy = new char[10];
SecureZeroMemory(dummy, 10);
::wprintf(L"start memory info\n");
print_memory_info();
HANDLE thread_handle[thread_count] = { nullptr, };
ULONGLONG start = ::GetTickCount64();
for (std::size_t i = 0; i < thread_count; ++i)
{
unsigned int id = 0;
thread_handle[i] = reinterpret_cast< HANDLE >(::_beginthreadex(nullptr, 0, memory_alloc_test_thread_func, nullptr, 0, &id));
if (thread_handle[i] == nullptr)
{
::wprintf(L"%llu: _beginthreadex failed.\n", i);
}
}
::WaitForMultipleObjects(thread_count, thread_handle, TRUE, INFINITE);
ULONGLONG run_time = ::GetTickCount64() - start;
::wprintf(L"run time: %llu\n", run_time);
for (std::size_t i = 0; i < thread_count; ++i)
{
if (thread_handle[i] != nullptr)
{
::CloseHandle(thread_handle[i]);
thread_handle[i] = nullptr;
}
}
delete[] dummy;
dummy = nullptr;
::wprintf(L"end memory info\n");
print_memory_info();
#if defined(MY_ALLOCATOR)
print_allocator_stat(allocator.get_statistics());
allocator.destroy();
#elif defined(JEMALLOC)
{
//je_malloc_stats_print(nullptr, nullptr, nullptr);
size_t sz = sizeof(jemalloc_post_allocated);
je_mallctl("stats.active", &jemalloc_post_allocated, &sz, nullptr, 0);
size_t leaked = jemalloc_post_allocated - jemalloc_pre_allocated;
::wprintf(L"\nDone. Leaked: %zd bytes\n", leaked);
bool failed = leaked > 65536; // in case C++ runtime allocated something (e.g. iostream locale or facet)
::wprintf(L"\nTest %s!\n", (failed ? L"FAILED" : L"successful"));
}
#endif
return 0;
}