zgu@7074: /*
zgu@7074:  * Copyright (c) 2013, 2014, Oracle and/or its affiliates. All rights reserved.
zgu@7074:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
zgu@7074:  *
zgu@7074:  * This code is free software; you can redistribute it and/or modify it
zgu@7074:  * under the terms of the GNU General Public License version 2 only, as
zgu@7074:  * published by the Free Software Foundation.
zgu@7074:  *
zgu@7074:  * This code is distributed in the hope that it will be useful, but WITHOUT
zgu@7074:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
zgu@7074:  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
zgu@7074:  * version 2 for more details (a copy is included in the LICENSE file that
zgu@7074:  * accompanied this code).
zgu@7074:  *
zgu@7074:  * You should have received a copy of the GNU General Public License version
zgu@7074:  * 2 along with this work; if not, write to the Free Software Foundation,
zgu@7074:  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
zgu@7074:  *
zgu@7074:  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
zgu@7074:  * or visit www.oracle.com if you need additional information or have any
zgu@7074:  * questions.
zgu@7074:  *
zgu@7074:  */
zgu@7074: #include "precompiled.hpp"
zgu@7074: 
zgu@7074: #include "runtime/threadCritical.hpp"
zgu@7074: #include "services/virtualMemoryTracker.hpp"
zgu@7074: 
zgu@7074: size_t VirtualMemorySummary::_snapshot[CALC_OBJ_SIZE_IN_TYPE(VirtualMemorySnapshot, size_t)];
zgu@7074: 
zgu@7074: void VirtualMemorySummary::initialize() {
zgu@7074:   assert(sizeof(_snapshot) >= sizeof(VirtualMemorySnapshot), "Sanity Check");
zgu@7074:   // Use placement operator new to initialize static data area.
zgu@7074:   ::new ((void*)_snapshot) VirtualMemorySnapshot();
zgu@7074: }
zgu@7074: 
zgu@7074: SortedLinkedList<ReservedMemoryRegion, compare_reserved_region_base> VirtualMemoryTracker::_reserved_regions;
zgu@7074: 
zgu@7074: int compare_committed_region(const CommittedMemoryRegion& r1, const CommittedMemoryRegion& r2) {
zgu@7074:   return r1.compare(r2);
zgu@7074: }
zgu@7074: 
zgu@7074: int compare_reserved_region_base(const ReservedMemoryRegion& r1, const ReservedMemoryRegion& r2) {
zgu@7074:   return r1.compare(r2);
zgu@7074: }
zgu@7074: 
zgu@7074: bool ReservedMemoryRegion::add_committed_region(address addr, size_t size, const NativeCallStack& stack) {
zgu@7074:   assert(addr != NULL, "Invalid address");
zgu@7074:   assert(size > 0, "Invalid size");
zgu@7074:   assert(contain_region(addr, size), "Not contain this region");
zgu@7074: 
zgu@7074:   if (all_committed()) return true;
zgu@7074: 
zgu@7074:   CommittedMemoryRegion committed_rgn(addr, size, stack);
zgu@7074:   LinkedListNode<CommittedMemoryRegion>* node = _committed_regions.find_node(committed_rgn);
zgu@7074:   if (node != NULL) {
zgu@7074:     CommittedMemoryRegion* rgn = node->data();
zgu@7074:     if (rgn->same_region(addr, size)) {
zgu@7074:       return true;
zgu@7074:     }
zgu@7074: 
zgu@7074:     if (rgn->adjacent_to(addr, size)) {
zgu@7074:       // check if the next region covers this committed region,
zgu@7074:       // the regions may not be merged due to different call stacks
zgu@7074:       LinkedListNode<CommittedMemoryRegion>* next =
zgu@7074:         node->next();
zgu@7074:       if (next != NULL && next->data()->contain_region(addr, size)) {
zgu@7074:         if (next->data()->same_region(addr, size)) {
zgu@7074:           next->data()->set_call_stack(stack);
zgu@7074:         }
zgu@7074:         return true;
zgu@7074:       }
zgu@7074:       if (rgn->call_stack()->equals(stack)) {
zgu@7074:         VirtualMemorySummary::record_uncommitted_memory(rgn->size(), flag());
zgu@7074:         // the two adjacent regions have the same call stack, merge them
zgu@7074:         rgn->expand_region(addr, size);
zgu@7074:         VirtualMemorySummary::record_committed_memory(rgn->size(), flag());
zgu@7074:         return true;
zgu@7074:       }
zgu@7074:       VirtualMemorySummary::record_committed_memory(size, flag());
zgu@7074:       if (rgn->base() > addr) {
zgu@7074:         return _committed_regions.insert_before(committed_rgn, node) != NULL;
zgu@7074:       } else {
zgu@7074:         return _committed_regions.insert_after(committed_rgn, node) != NULL;
zgu@7074:       }
zgu@7074:     }
zgu@7074:     assert(rgn->contain_region(addr, size), "Must cover this region");
zgu@7074:     return true;
zgu@7074:   } else {
zgu@7074:     // New committed region
zgu@7074:     VirtualMemorySummary::record_committed_memory(size, flag());
zgu@7074:     return add_committed_region(committed_rgn);
zgu@7074:   }
zgu@7074: }
zgu@7074: 
zgu@7074: void ReservedMemoryRegion::set_all_committed(bool b) {
zgu@7074:   if (all_committed() != b) {
zgu@7074:     _all_committed = b;
zgu@7074:     if (b) {
zgu@7074:       VirtualMemorySummary::record_committed_memory(size(), flag());
zgu@7074:     }
zgu@7074:   }
zgu@7074: }
zgu@7074: 
zgu@7074: bool ReservedMemoryRegion::remove_uncommitted_region(LinkedListNode<CommittedMemoryRegion>* node,
zgu@7074:   address addr, size_t size) {
zgu@7074:   assert(addr != NULL, "Invalid address");
zgu@7074:   assert(size > 0, "Invalid size");
zgu@7074: 
zgu@7074:   CommittedMemoryRegion* rgn = node->data();
zgu@7074:   assert(rgn->contain_region(addr, size), "Has to be contained");
zgu@7074:   assert(!rgn->same_region(addr, size), "Can not be the same region");
zgu@7074: 
zgu@7074:   if (rgn->base() == addr ||
zgu@7074:       rgn->end() == addr + size) {
zgu@7074:     rgn->exclude_region(addr, size);
zgu@7074:     return true;
zgu@7074:   } else {
zgu@7074:     // split this region
zgu@7074:     address top =rgn->end();
zgu@7074:     // use this region for lower part
zgu@7074:     size_t exclude_size = rgn->end() - addr;
zgu@7074:     rgn->exclude_region(addr, exclude_size);
zgu@7074: 
zgu@7074:     // higher part
zgu@7074:     address high_base = addr + size;
zgu@7074:     size_t  high_size = top - high_base;
zgu@7074: 
zgu@7074:     CommittedMemoryRegion high_rgn(high_base, high_size, *rgn->call_stack());
zgu@7074:     LinkedListNode<CommittedMemoryRegion>* high_node = _committed_regions.add(high_rgn);
zgu@7074:     assert(high_node == NULL || node->next() == high_node, "Should be right after");
zgu@7074:     return (high_node != NULL);
zgu@7074:   }
zgu@7074: 
zgu@7074:   return false;
zgu@7074: }
zgu@7074: 
zgu@7074: bool ReservedMemoryRegion::remove_uncommitted_region(address addr, size_t sz) {
zgu@7074:   // uncommit stack guard pages
zgu@7074:   if (flag() == mtThreadStack && !same_region(addr, sz)) {
zgu@7074:     return true;
zgu@7074:   }
zgu@7074: 
zgu@7074:   assert(addr != NULL, "Invalid address");
zgu@7074:   assert(sz > 0, "Invalid size");
zgu@7074: 
zgu@7074:   if (all_committed()) {
zgu@7074:     assert(_committed_regions.is_empty(), "Sanity check");
zgu@7074:     assert(contain_region(addr, sz), "Reserved region does not contain this region");
zgu@7074:     set_all_committed(false);
zgu@7074:     VirtualMemorySummary::record_uncommitted_memory(sz, flag());
zgu@7074:     if (same_region(addr, sz)) {
zgu@7074:       return true;
zgu@7074:     } else {
zgu@7074:       CommittedMemoryRegion rgn(base(), size(), *call_stack());
zgu@7074:       if (rgn.base() == addr || rgn.end() == (addr + sz)) {
zgu@7074:         rgn.exclude_region(addr, sz);
zgu@7074:         return add_committed_region(rgn);
zgu@7074:       } else {
zgu@7074:         // split this region
zgu@7074:         // top of the whole region
zgu@7074:         address top =rgn.end();
zgu@7074:         // use this region for lower part
zgu@7074:         size_t exclude_size = rgn.end() - addr;
zgu@7074:         rgn.exclude_region(addr, exclude_size);
zgu@7074:         if (add_committed_region(rgn)) {
zgu@7074:           // higher part
zgu@7074:           address high_base = addr + sz;
zgu@7074:           size_t  high_size = top - high_base;
zgu@7074:           CommittedMemoryRegion high_rgn(high_base, high_size, emptyStack);
zgu@7074:           return add_committed_region(high_rgn);
zgu@7074:         } else {
zgu@7074:           return false;
zgu@7074:         }
zgu@7074:       }
zgu@7074:     }
zgu@7074:   } else {
zgu@7074:     // we have to walk whole list to remove the committed regions in
zgu@7074:     // specified range
zgu@7074:     LinkedListNode<CommittedMemoryRegion>* head =
zgu@7074:       _committed_regions.head();
zgu@7074:     LinkedListNode<CommittedMemoryRegion>* prev = NULL;
zgu@7074:     VirtualMemoryRegion uncommitted_rgn(addr, sz);
zgu@7074: 
zgu@7074:     while (head != NULL && !uncommitted_rgn.is_empty()) {
zgu@7074:       CommittedMemoryRegion* crgn = head->data();
zgu@7074:       // this committed region overlaps to region to uncommit
zgu@7074:       if (crgn->overlap_region(uncommitted_rgn.base(), uncommitted_rgn.size())) {
zgu@7074:         if (crgn->same_region(uncommitted_rgn.base(), uncommitted_rgn.size())) {
zgu@7074:           // find matched region, remove the node will do
zgu@7074:           VirtualMemorySummary::record_uncommitted_memory(uncommitted_rgn.size(), flag());
zgu@7074:           _committed_regions.remove_after(prev);
zgu@7074:           return true;
zgu@7074:         } else if (crgn->contain_region(uncommitted_rgn.base(), uncommitted_rgn.size())) {
zgu@7074:           // this committed region contains whole uncommitted region
zgu@7074:           VirtualMemorySummary::record_uncommitted_memory(uncommitted_rgn.size(), flag());
zgu@7074:           return remove_uncommitted_region(head, uncommitted_rgn.base(), uncommitted_rgn.size());
zgu@7074:         } else if (uncommitted_rgn.contain_region(crgn->base(), crgn->size())) {
zgu@7074:           // this committed region has been uncommitted
zgu@7074:           size_t exclude_size = crgn->end() - uncommitted_rgn.base();
zgu@7074:           uncommitted_rgn.exclude_region(uncommitted_rgn.base(), exclude_size);
zgu@7074:           VirtualMemorySummary::record_uncommitted_memory(crgn->size(), flag());
zgu@7074:           LinkedListNode<CommittedMemoryRegion>* tmp = head;
zgu@7074:           head = head->next();
zgu@7074:           _committed_regions.remove_after(prev);
zgu@7074:           continue;
zgu@7074:         } else if (crgn->contain_address(uncommitted_rgn.base())) {
zgu@7074:           size_t toUncommitted = crgn->end() - uncommitted_rgn.base();
zgu@7074:           crgn->exclude_region(uncommitted_rgn.base(), toUncommitted);
zgu@7074:           uncommitted_rgn.exclude_region(uncommitted_rgn.base(), toUncommitted);
zgu@7074:           VirtualMemorySummary::record_uncommitted_memory(toUncommitted, flag());
zgu@7074:         } else if (uncommitted_rgn.contain_address(crgn->base())) {
zgu@7074:           size_t toUncommitted = uncommitted_rgn.end() - crgn->base();
zgu@7074:           crgn->exclude_region(crgn->base(), toUncommitted);
zgu@7074:           uncommitted_rgn.exclude_region(uncommitted_rgn.end() - toUncommitted,
zgu@7074:             toUncommitted);
zgu@7074:           VirtualMemorySummary::record_uncommitted_memory(toUncommitted, flag());
zgu@7074:         }
zgu@7074:       }
zgu@7074:       prev = head;
zgu@7074:       head = head->next();
zgu@7074:     }
zgu@7074:   }
zgu@7074: 
zgu@7074:   return true;
zgu@7074: }
zgu@7074: 
zgu@7074: void ReservedMemoryRegion::move_committed_regions(address addr, ReservedMemoryRegion& rgn) {
zgu@7074:   assert(addr != NULL, "Invalid address");
zgu@7074: 
zgu@7074:   // split committed regions
zgu@7074:   LinkedListNode<CommittedMemoryRegion>* head =
zgu@7074:     _committed_regions.head();
zgu@7074:   LinkedListNode<CommittedMemoryRegion>* prev = NULL;
zgu@7074: 
zgu@7074:   while (head != NULL) {
zgu@7074:     if (head->data()->base() >= addr) {
zgu@7074:       break;
zgu@7074:     }
zgu@7074:     prev = head;
zgu@7074:     head = head->next();
zgu@7074:   }
zgu@7074: 
zgu@7074:   if (head != NULL) {
zgu@7074:     if (prev != NULL) {
zgu@7074:       prev->set_next(head->next());
zgu@7074:     } else {
zgu@7074:       _committed_regions.set_head(NULL);
zgu@7074:     }
zgu@7074:   }
zgu@7074: 
zgu@7074:   rgn._committed_regions.set_head(head);
zgu@7074: }
zgu@7074: 
zgu@7074: size_t ReservedMemoryRegion::committed_size() const {
zgu@7074:   if (all_committed()) {
zgu@7074:     return size();
zgu@7074:   } else {
zgu@7074:     size_t committed = 0;
zgu@7074:     LinkedListNode<CommittedMemoryRegion>* head =
zgu@7074:       _committed_regions.head();
zgu@7074:     while (head != NULL) {
zgu@7074:       committed += head->data()->size();
zgu@7074:       head = head->next();
zgu@7074:     }
zgu@7074:     return committed;
zgu@7074:   }
zgu@7074: }
zgu@7074: 
zgu@7074: void ReservedMemoryRegion::set_flag(MEMFLAGS f) {
zgu@7074:   assert((flag() == mtNone || flag() == f), "Overwrite memory type");
zgu@7074:   if (flag() != f) {
zgu@7074:     VirtualMemorySummary::move_reserved_memory(flag(), f, size());
zgu@7074:     VirtualMemorySummary::move_committed_memory(flag(), f, committed_size());
zgu@7074:     _flag = f;
zgu@7074:   }
zgu@7074: }
zgu@7074: 
zgu@7074: bool VirtualMemoryTracker::initialize(NMT_TrackingLevel level) {
zgu@7074:   if (level >= NMT_summary) {
zgu@7074:     VirtualMemorySummary::initialize();
zgu@7074:   }
zgu@7074:   return true;
zgu@7074: }
zgu@7074: 
zgu@7074: bool VirtualMemoryTracker::add_reserved_region(address base_addr, size_t size,
zgu@7074:    const NativeCallStack& stack, MEMFLAGS flag, bool all_committed) {
zgu@7074:   assert(base_addr != NULL, "Invalid address");
zgu@7074:   assert(size > 0, "Invalid size");
zgu@7074: 
zgu@7074:   ReservedMemoryRegion  rgn(base_addr, size, stack, flag);
zgu@7074:   ReservedMemoryRegion* reserved_rgn = _reserved_regions.find(rgn);
zgu@7074:   LinkedListNode<ReservedMemoryRegion>* node;
zgu@7074:   if (reserved_rgn == NULL) {
zgu@7074:     VirtualMemorySummary::record_reserved_memory(size, flag);
zgu@7074:     node = _reserved_regions.add(rgn);
zgu@7074:     if (node != NULL) {
zgu@7074:       node->data()->set_all_committed(all_committed);
zgu@7074:       return true;
zgu@7074:     } else {
zgu@7074:       return false;
zgu@7074:     }
zgu@7074:   } else {
zgu@7074:     if (reserved_rgn->same_region(base_addr, size)) {
zgu@7074:       reserved_rgn->set_call_stack(stack);
zgu@7074:       reserved_rgn->set_flag(flag);
zgu@7074:       return true;
zgu@7074:     } else if (reserved_rgn->adjacent_to(base_addr, size)) {
zgu@7074:       VirtualMemorySummary::record_reserved_memory(size, flag);
zgu@7074:       reserved_rgn->expand_region(base_addr, size);
zgu@7074:       reserved_rgn->set_call_stack(stack);
zgu@7074:       return true;
zgu@7074:     } else {
zgu@7074:       // Overlapped reservation.
zgu@7074:       // It can happen when the regions are thread stacks, as JNI
zgu@7074:       // thread does not detach from VM before exits, and leads to
zgu@7074:       // leak JavaThread object
zgu@7074:       if (reserved_rgn->flag() == mtThreadStack) {
zgu@7074:         guarantee(!CheckJNICalls, "Attached JNI thread exited without being detached");
zgu@7074:         // Overwrite with new region
zgu@7074: 
zgu@7074:         // Release old region
zgu@7074:         VirtualMemorySummary::record_uncommitted_memory(reserved_rgn->committed_size(), reserved_rgn->flag());
zgu@7074:         VirtualMemorySummary::record_released_memory(reserved_rgn->size(), reserved_rgn->flag());
zgu@7074: 
zgu@7074:         // Add new region
zgu@7074:         VirtualMemorySummary::record_reserved_memory(rgn.size(), flag);
zgu@7074: 
zgu@7074:         *reserved_rgn = rgn;
zgu@7074:         return true;
zgu@7074:       } else {
zgu@7074:         ShouldNotReachHere();
zgu@7074:         return false;
zgu@7074:       }
zgu@7074:     }
zgu@7074:   }
zgu@7074: }
zgu@7074: 
zgu@7074: void VirtualMemoryTracker::set_reserved_region_type(address addr, MEMFLAGS flag) {
zgu@7074:   assert(addr != NULL, "Invalid address");
zgu@7074: 
zgu@7074:   ReservedMemoryRegion   rgn(addr, 1);
zgu@7074:   ReservedMemoryRegion*  reserved_rgn = _reserved_regions.find(rgn);
zgu@7074:   if (reserved_rgn != NULL) {
zgu@7074:     assert(reserved_rgn->contain_address(addr), "Containment");
zgu@7074:     if (reserved_rgn->flag() != flag) {
zgu@7074:       assert(reserved_rgn->flag() == mtNone, "Overwrite memory type");
zgu@7074:       reserved_rgn->set_flag(flag);
zgu@7074:     }
zgu@7074:   }
zgu@7074: }
zgu@7074: 
zgu@7074: bool VirtualMemoryTracker::add_committed_region(address addr, size_t size,
zgu@7074:   const NativeCallStack& stack) {
zgu@7074:   assert(addr != NULL, "Invalid address");
zgu@7074:   assert(size > 0, "Invalid size");
zgu@7074:   ReservedMemoryRegion  rgn(addr, size);
zgu@7074:   ReservedMemoryRegion* reserved_rgn = _reserved_regions.find(rgn);
zgu@7074: 
zgu@7074:   assert(reserved_rgn != NULL, "No reserved region");
zgu@7074:   assert(reserved_rgn->contain_region(addr, size), "Not completely contained");
zgu@7074:   return reserved_rgn->add_committed_region(addr, size, stack);
zgu@7074: }
zgu@7074: 
zgu@7074: bool VirtualMemoryTracker::remove_uncommitted_region(address addr, size_t size) {
zgu@7074:   assert(addr != NULL, "Invalid address");
zgu@7074:   assert(size > 0, "Invalid size");
zgu@7074:   ReservedMemoryRegion  rgn(addr, size);
zgu@7074:   ReservedMemoryRegion* reserved_rgn = _reserved_regions.find(rgn);
zgu@7074:   assert(reserved_rgn != NULL, "No reserved region");
zgu@7074:   assert(reserved_rgn->contain_region(addr, size), "Not completely contained");
zgu@7074:   return reserved_rgn->remove_uncommitted_region(addr, size);
zgu@7074: }
zgu@7074: 
zgu@7074: bool VirtualMemoryTracker::remove_released_region(address addr, size_t size) {
zgu@7074:   assert(addr != NULL, "Invalid address");
zgu@7074:   assert(size > 0, "Invalid size");
zgu@7074: 
zgu@7074:   ReservedMemoryRegion  rgn(addr, size);
zgu@7074:   ReservedMemoryRegion* reserved_rgn = _reserved_regions.find(rgn);
zgu@7074: 
zgu@7074:   assert(reserved_rgn != NULL, "No reserved region");
zgu@7074: 
zgu@7074:   // uncommit regions within the released region
zgu@7074:   if (!reserved_rgn->remove_uncommitted_region(addr, size)) {
zgu@7074:     return false;
zgu@7074:   }
zgu@7074: 
zgu@7074: 
zgu@7074:   VirtualMemorySummary::record_released_memory(size, reserved_rgn->flag());
zgu@7074: 
zgu@7074:   if (reserved_rgn->same_region(addr, size)) {
zgu@7074:     return _reserved_regions.remove(rgn);
zgu@7074:   } else {
zgu@7074:     assert(reserved_rgn->contain_region(addr, size), "Not completely contained");
zgu@7074:     if (reserved_rgn->base() == addr ||
zgu@7074:         reserved_rgn->end() == addr + size) {
zgu@7074:         reserved_rgn->exclude_region(addr, size);
zgu@7074:       return true;
zgu@7074:     } else {
zgu@7074:       address top = reserved_rgn->end();
zgu@7074:       address high_base = addr + size;
zgu@7074:       ReservedMemoryRegion high_rgn(high_base, top - high_base,
zgu@7074:         *reserved_rgn->call_stack(), reserved_rgn->flag());
zgu@7074: 
zgu@7074:       // use original region for lower region
zgu@7074:       reserved_rgn->exclude_region(addr, top - addr);
zgu@7074:       LinkedListNode<ReservedMemoryRegion>* new_rgn = _reserved_regions.add(high_rgn);
zgu@7074:       if (new_rgn == NULL) {
zgu@7074:         return false;
zgu@7074:       } else {
zgu@7074:         reserved_rgn->move_committed_regions(addr, *new_rgn->data());
zgu@7074:         return true;
zgu@7074:       }
zgu@7074:     }
zgu@7074:   }
zgu@7074: }
zgu@7074: 
zgu@7074: 
zgu@7074: bool VirtualMemoryTracker::walk_virtual_memory(VirtualMemoryWalker* walker) {
zgu@7074:   ThreadCritical tc;
zgu@7074:   LinkedListNode<ReservedMemoryRegion>* head = _reserved_regions.head();
zgu@7074:   while (head != NULL) {
zgu@7074:     const ReservedMemoryRegion* rgn = head->peek();
zgu@7074:     if (!walker->do_allocation_site(rgn)) {
zgu@7074:       return false;
zgu@7074:     }
zgu@7074:     head = head->next();
zgu@7074:   }
zgu@7074:   return true;
zgu@7074: }
zgu@7074: 
zgu@7074: // Transition virtual memory tracking level.
zgu@7074: bool VirtualMemoryTracker::transition(NMT_TrackingLevel from, NMT_TrackingLevel to) {
zgu@7074:   if (from == NMT_minimal) {
zgu@7074:     assert(to == NMT_summary || to == NMT_detail, "Just check");
zgu@7074:     VirtualMemorySummary::reset();
zgu@7074:   } else if (to == NMT_minimal) {
zgu@7074:     assert(from == NMT_summary || from == NMT_detail, "Just check");
zgu@7074:     // Clean up virtual memory tracking data structures.
zgu@7074:     ThreadCritical tc;
zgu@7074:     _reserved_regions.clear();
zgu@7074:   }
zgu@7074: 
zgu@7074:   return true;
zgu@7074: }
zgu@7074: 
zgu@7074: