--- /dev/null 2019-01-28 17:48:06.000000000 +0800 +++ new/src/share/vm/jfr/utilities/jfrAllocation.cpp 2019-01-28 17:48:06.000000000 +0800 @@ -0,0 +1,139 @@ +/* + * Copyright (c) 2014, 2019, Oracle and/or its affiliates. All rights reserved. + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. + * + * This code is free software; you can redistribute it and/or modify it + * under the terms of the GNU General Public License version 2 only, as + * published by the Free Software Foundation. + * + * This code is distributed in the hope that it will be useful, but WITHOUT + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License + * version 2 for more details (a copy is included in the LICENSE file that + * accompanied this code). + * + * You should have received a copy of the GNU General Public License version + * 2 along with this work; if not, write to the Free Software Foundation, + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. + * + * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA + * or visit www.oracle.com if you need additional information or have any + * questions. + * + */ + +#include "precompiled.hpp" +#include "jfr/recorder/jfrRecorder.hpp" +#include "jfr/utilities/jfrAllocation.hpp" +#include "jfr/utilities/jfrLog.hpp" +#include "memory/allocation.inline.hpp" +#include "runtime/atomic.hpp" +#include "runtime/orderAccess.inline.hpp" +#include "runtime/vm_version.hpp" +#include "utilities/debug.hpp" +#include "utilities/macros.hpp" +#include "utilities/nativeCallStack.hpp" + +#ifdef ASSERT +static jlong atomic_add_jlong(jlong value, jlong volatile* const dest) { + assert(VM_Version::supports_cx8(), "unsupported"); + jlong compare_value; + jlong exchange_value; + do { + compare_value = OrderAccess::load_acquire(dest); + exchange_value = compare_value + value; + } while (Atomic::cmpxchg(exchange_value, dest, compare_value) != compare_value); + return exchange_value; +} + +// debug statistics +static volatile jlong _allocated_bytes = 0; +static volatile jlong _deallocated_bytes = 0; +static volatile jlong _live_set_bytes = 0; + +static void add(size_t alloc_size) { + if (!JfrRecorder::is_created()) { + const jlong total_allocated = atomic_add_jlong(alloc_size, &_allocated_bytes); + const jlong current_live_set = atomic_add_jlong(alloc_size, &_live_set_bytes); + log_trace(jfr, system)("Allocation: [" SIZE_FORMAT "] bytes", alloc_size); + log_trace(jfr, system)("Total alloc [" JLONG_FORMAT "] bytes", total_allocated); + log_trace(jfr, system)("Liveset: [" JLONG_FORMAT "] bytes", current_live_set); + } +} + +static void subtract(size_t dealloc_size) { + if (!JfrRecorder::is_created()) { + const size_t total_deallocated = atomic_add_jlong(dealloc_size, &_deallocated_bytes); + const size_t current_live_set = atomic_add_jlong(dealloc_size * -1, &_live_set_bytes); + log_trace(jfr, system)("Deallocation: [" SIZE_FORMAT "] bytes", dealloc_size); + log_trace(jfr, system)("Total dealloc [" JLONG_FORMAT "] bytes", total_deallocated); + log_trace(jfr, system)("Liveset: [" JLONG_FORMAT "] bytes", current_live_set); + } +} + +static void hook_memory_deallocation(size_t dealloc_size) { + subtract(dealloc_size); +} +#endif // ASSERT + +static void hook_memory_allocation(const char* allocation, size_t alloc_size) { + if (NULL == allocation) { + if (!JfrRecorder::is_created()) { + log_warning(jfr, system)("Memory allocation failed for size [" SIZE_FORMAT "] bytes", alloc_size); + return; + } else { + // after critical startup, fail as by default + vm_exit_out_of_memory(alloc_size, OOM_MALLOC_ERROR, "AllocateHeap"); + } + } + debug_only(add(alloc_size)); +} + +void JfrCHeapObj::on_memory_allocation(const void* allocation, size_t size) { + hook_memory_allocation((const char*)allocation, size); +} + +void* JfrCHeapObj::operator new(size_t size) throw() { + return operator new(size, std::nothrow); +} + +void* JfrCHeapObj::operator new (size_t size, const std::nothrow_t& nothrow_constant) throw() { + void* const memory = CHeapObj::operator new(size, nothrow_constant, CALLER_PC); + hook_memory_allocation((const char*)memory, size); + return memory; +} + +void* JfrCHeapObj::operator new [](size_t size) throw() { + return operator new[](size, std::nothrow); +} + +void* JfrCHeapObj::operator new [](size_t size, const std::nothrow_t& nothrow_constant) throw() { + void* const memory = CHeapObj::operator new[](size, nothrow_constant, CALLER_PC); + hook_memory_allocation((const char*)memory, size); + return memory; +} + +void JfrCHeapObj::operator delete(void* p, size_t size) { + debug_only(hook_memory_deallocation(size);) + CHeapObj::operator delete(p); +} + +void JfrCHeapObj::operator delete[](void* p, size_t size) { + debug_only(hook_memory_deallocation(size);) + CHeapObj::operator delete[](p); +} + +char* JfrCHeapObj::realloc_array(char* old, size_t size) { + char* const memory = ReallocateHeap(old, size, mtTracing, AllocFailStrategy::RETURN_NULL); + hook_memory_allocation(memory, size); + return memory; +} + +void JfrCHeapObj::free(void* p, size_t size) { + debug_only(hook_memory_deallocation(size);) + FreeHeap(p); +} + +char* JfrCHeapObj::allocate_array_noinline(size_t elements, size_t element_size) { + return AllocateHeap(elements * element_size, mtTracing, CALLER_PC, AllocFailStrategy::RETURN_NULL); +}