74 lines
2.6 KiB
C++
74 lines
2.6 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2; -*- */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
// Test that heap snapshots cross compartment boundaries when expected.
|
|
|
|
#include "DevTools.h"
|
|
|
|
DEF_TEST(DoesCrossCompartmentBoundaries, {
|
|
// Create a new global to get a new compartment.
|
|
JS::CompartmentOptions options;
|
|
JS::RootedObject newGlobal(cx, JS_NewGlobalObject(cx,
|
|
getGlobalClass(),
|
|
nullptr,
|
|
JS::FireOnNewGlobalHook,
|
|
options));
|
|
ASSERT_TRUE(newGlobal);
|
|
JSCompartment* newCompartment = nullptr;
|
|
{
|
|
JSAutoCompartment ac(cx, newGlobal);
|
|
ASSERT_TRUE(JS_InitStandardClasses(cx, newGlobal));
|
|
newCompartment = js::GetContextCompartment(cx);
|
|
}
|
|
ASSERT_TRUE(newCompartment);
|
|
ASSERT_NE(newCompartment, compartment);
|
|
|
|
// Our set of target compartments is both the old and new compartments.
|
|
JS::CompartmentSet targetCompartments;
|
|
ASSERT_TRUE(targetCompartments.init());
|
|
ASSERT_TRUE(targetCompartments.put(compartment));
|
|
ASSERT_TRUE(targetCompartments.put(newCompartment));
|
|
|
|
FakeNode nodeA;
|
|
FakeNode nodeB;
|
|
FakeNode nodeC;
|
|
FakeNode nodeD;
|
|
|
|
nodeA.compartment = compartment;
|
|
nodeB.compartment = nullptr;
|
|
nodeC.compartment = newCompartment;
|
|
nodeD.compartment = nullptr;
|
|
|
|
AddEdge(nodeA, nodeB);
|
|
AddEdge(nodeA, nodeC);
|
|
AddEdge(nodeB, nodeD);
|
|
|
|
::testing::NiceMock<MockWriter> writer;
|
|
|
|
// Should serialize nodeA, because it is in one of our target compartments.
|
|
ExpectWriteNode(writer, nodeA);
|
|
|
|
// Should serialize nodeB, because it doesn't belong to a compartment and is
|
|
// therefore assumed to be shared.
|
|
ExpectWriteNode(writer, nodeB);
|
|
|
|
// Should also serialize nodeC, which is in our target compartments, but a
|
|
// different compartment than A.
|
|
ExpectWriteNode(writer, nodeC);
|
|
|
|
// Should serialize nodeD because it's reachable via B and both nodes B and D
|
|
// don't belong to a specific compartment.
|
|
ExpectWriteNode(writer, nodeD);
|
|
|
|
JS::AutoCheckCannotGC noGC(cx);
|
|
|
|
ASSERT_TRUE(WriteHeapGraph(cx,
|
|
JS::ubi::Node(&nodeA),
|
|
writer,
|
|
/* wantNames = */ false,
|
|
&targetCompartments,
|
|
noGC));
|
|
});
|