|
46 | 46 | #include "src/profiler/heap-snapshot-generator-inl.h" |
47 | 47 | #include "test/cctest/cctest.h" |
48 | 48 | #include "test/cctest/collector.h" |
| 49 | +#include "test/cctest/heap/heap-utils.h" |
49 | 50 |
|
50 | 51 | using i::AllocationTraceNode; |
51 | 52 | using i::AllocationTraceTree; |
@@ -1693,6 +1694,154 @@ TEST(HeapSnapshotRetainedObjectInfo) { |
1693 | 1694 | CHECK_EQ(native_group_ccc, GetChildByName(n_CCC, "ccc-group")); |
1694 | 1695 | } |
1695 | 1696 |
|
| 1697 | +namespace { |
| 1698 | + |
| 1699 | +class EmbedderGraphBuilderForNativeSnapshotObjectId final { |
| 1700 | + public: |
| 1701 | + class RegularNode : public v8::EmbedderGraph::Node { |
| 1702 | + public: |
| 1703 | + RegularNode(v8::NativeObject native_object, const char* name, size_t size, |
| 1704 | + Node* wrapper_node) |
| 1705 | + : name_(name), |
| 1706 | + size_(size), |
| 1707 | + native_object_(native_object), |
| 1708 | + wrapper_node_(wrapper_node) {} |
| 1709 | + // v8::EmbedderGraph::Node |
| 1710 | + const char* Name() override { return name_; } |
| 1711 | + size_t SizeInBytes() override { return size_; } |
| 1712 | + Node* WrapperNode() override { return wrapper_node_; } |
| 1713 | + v8::NativeObject GetNativeObject() override { |
| 1714 | + return native_object_ ? native_object_ : this; |
| 1715 | + } |
| 1716 | + |
| 1717 | + private: |
| 1718 | + const char* name_; |
| 1719 | + size_t size_; |
| 1720 | + v8::NativeObject native_object_; |
| 1721 | + Node* wrapper_node_; |
| 1722 | + }; |
| 1723 | + |
| 1724 | + class RootNode : public RegularNode { |
| 1725 | + public: |
| 1726 | + explicit RootNode(const char* name) |
| 1727 | + : RegularNode(nullptr, name, 0, nullptr) {} |
| 1728 | + // v8::EmbedderGraph::EmbedderNode |
| 1729 | + bool IsRootNode() override { return true; } |
| 1730 | + }; |
| 1731 | + |
| 1732 | + struct BuildParameter { |
| 1733 | + v8::Persistent<v8::String>* wrapper; |
| 1734 | + void* native1; |
| 1735 | + void* native2; |
| 1736 | + }; |
| 1737 | + |
| 1738 | + static void BuildEmbedderGraph(v8::Isolate* isolate, v8::EmbedderGraph* graph, |
| 1739 | + void* data) { |
| 1740 | + BuildParameter* parameter = reinterpret_cast<BuildParameter*>(data); |
| 1741 | + v8::Local<v8::String> local_str = |
| 1742 | + v8::Local<v8::String>::New(isolate, *(parameter->wrapper)); |
| 1743 | + auto* v8_node = graph->V8Node(local_str); |
| 1744 | + CHECK(!v8_node->IsEmbedderNode()); |
| 1745 | + auto* root_node = |
| 1746 | + graph->AddNode(std::unique_ptr<RootNode>(new RootNode("root"))); |
| 1747 | + auto* non_merged_node = graph->AddNode(std::unique_ptr<RegularNode>( |
| 1748 | + new RegularNode(parameter->native1, "non-merged", 0, nullptr))); |
| 1749 | + auto* merged_node = graph->AddNode(std::unique_ptr<RegularNode>( |
| 1750 | + new RegularNode(parameter->native2, "merged", 0, v8_node))); |
| 1751 | + graph->AddEdge(root_node, non_merged_node); |
| 1752 | + graph->AddEdge(root_node, merged_node); |
| 1753 | + } |
| 1754 | +}; |
| 1755 | + |
| 1756 | +} // namespace |
| 1757 | + |
| 1758 | +TEST(NativeSnapshotObjectId) { |
| 1759 | + LocalContext env; |
| 1760 | + v8::Isolate* isolate = env->GetIsolate(); |
| 1761 | + v8::HandleScope scope(isolate); |
| 1762 | + v8::HeapProfiler* heap_profiler = isolate->GetHeapProfiler(); |
| 1763 | + |
| 1764 | + v8::Persistent<v8::String> wrapper(isolate, v8_str("wrapper")); |
| 1765 | + int native1; |
| 1766 | + int native2; |
| 1767 | + |
| 1768 | + EmbedderGraphBuilderForNativeSnapshotObjectId::BuildParameter parameter{ |
| 1769 | + &wrapper, &native1, &native2}; |
| 1770 | + heap_profiler->AddBuildEmbedderGraphCallback( |
| 1771 | + EmbedderGraphBuilderForNativeSnapshotObjectId::BuildEmbedderGraph, |
| 1772 | + ¶meter); |
| 1773 | + const v8::HeapSnapshot* snapshot = heap_profiler->TakeHeapSnapshot(); |
| 1774 | + CHECK(ValidateSnapshot(snapshot)); |
| 1775 | + |
| 1776 | + v8::SnapshotObjectId non_merged_id = heap_profiler->GetObjectId(&native1); |
| 1777 | + CHECK_NE(v8::HeapProfiler::kUnknownObjectId, non_merged_id); |
| 1778 | + v8::SnapshotObjectId merged_id = heap_profiler->GetObjectId(&native2); |
| 1779 | + CHECK_NE(v8::HeapProfiler::kUnknownObjectId, merged_id); |
| 1780 | + CHECK_NE(non_merged_id, merged_id); |
| 1781 | + const v8::HeapGraphNode* non_merged_node = |
| 1782 | + snapshot->GetNodeById(non_merged_id); |
| 1783 | + CHECK_NOT_NULL(non_merged_node); |
| 1784 | + const v8::HeapGraphNode* merged_node = snapshot->GetNodeById(merged_id); |
| 1785 | + CHECK_NOT_NULL(merged_node); |
| 1786 | + |
| 1787 | + heap_profiler->ClearObjectIds(); |
| 1788 | + CHECK_EQ(v8::HeapProfiler::kUnknownObjectId, |
| 1789 | + heap_profiler->GetObjectId(&native1)); |
| 1790 | + CHECK_EQ(v8::HeapProfiler::kUnknownObjectId, |
| 1791 | + heap_profiler->GetObjectId(&native2)); |
| 1792 | +} |
| 1793 | + |
| 1794 | +TEST(NativeSnapshotObjectIdMoving) { |
| 1795 | + // Required to allow moving specific objects. |
| 1796 | + i::FLAG_manual_evacuation_candidates_selection = true; |
| 1797 | + |
| 1798 | + LocalContext env; |
| 1799 | + v8::Isolate* isolate = env->GetIsolate(); |
| 1800 | + v8::HandleScope scope(isolate); |
| 1801 | + v8::HeapProfiler* heap_profiler = isolate->GetHeapProfiler(); |
| 1802 | + heap_profiler->StartTrackingHeapObjects(true); |
| 1803 | + |
| 1804 | + v8::Persistent<v8::String> wrapper(isolate, v8_str("wrapper")); |
| 1805 | + int native1; |
| 1806 | + int native2; |
| 1807 | + |
| 1808 | + EmbedderGraphBuilderForNativeSnapshotObjectId::BuildParameter parameter{ |
| 1809 | + &wrapper, &native1, &native2}; |
| 1810 | + heap_profiler->AddBuildEmbedderGraphCallback( |
| 1811 | + EmbedderGraphBuilderForNativeSnapshotObjectId::BuildEmbedderGraph, |
| 1812 | + ¶meter); |
| 1813 | + const v8::HeapSnapshot* snapshot = heap_profiler->TakeHeapSnapshot(); |
| 1814 | + CHECK(ValidateSnapshot(snapshot)); |
| 1815 | + |
| 1816 | + v8::SnapshotObjectId non_merged_id = heap_profiler->GetObjectId(&native1); |
| 1817 | + CHECK_NE(v8::HeapProfiler::kUnknownObjectId, non_merged_id); |
| 1818 | + v8::SnapshotObjectId merged_id = heap_profiler->GetObjectId(&native2); |
| 1819 | + CHECK_NE(v8::HeapProfiler::kUnknownObjectId, merged_id); |
| 1820 | + CHECK_NE(non_merged_id, merged_id); |
| 1821 | + const v8::HeapGraphNode* non_merged_node = |
| 1822 | + snapshot->GetNodeById(non_merged_id); |
| 1823 | + CHECK_NOT_NULL(non_merged_node); |
| 1824 | + const v8::HeapGraphNode* merged_node = snapshot->GetNodeById(merged_id); |
| 1825 | + CHECK_NOT_NULL(merged_node); |
| 1826 | + |
| 1827 | + { |
| 1828 | + v8::HandleScope scope(isolate); |
| 1829 | + auto local = v8::Local<v8::String>::New(isolate, wrapper); |
| 1830 | + i::Handle<i::String> internal = i::Handle<i::String>::cast( |
| 1831 | + v8::Utils::OpenHandle(*v8::Local<v8::String>::Cast(local))); |
| 1832 | + i::heap::ForceEvacuationCandidate(i::Page::FromHeapObject(*internal)); |
| 1833 | + } |
| 1834 | + CcTest::CollectAllGarbage(); |
| 1835 | + |
| 1836 | + non_merged_id = heap_profiler->GetObjectId(&native1); |
| 1837 | + CHECK_NE(v8::HeapProfiler::kUnknownObjectId, non_merged_id); |
| 1838 | + merged_id = heap_profiler->GetObjectId(&native2); |
| 1839 | + CHECK_NE(v8::HeapProfiler::kUnknownObjectId, merged_id); |
| 1840 | + CHECK_NE(non_merged_id, merged_id); |
| 1841 | + |
| 1842 | + heap_profiler->StopTrackingHeapObjects(); |
| 1843 | +} |
| 1844 | + |
1696 | 1845 | TEST(DeleteAllHeapSnapshots) { |
1697 | 1846 | LocalContext env; |
1698 | 1847 | v8::HandleScope scope(env->GetIsolate()); |
|
0 commit comments