summaryrefslogtreecommitdiffstats
path: root/devtools/shared/heapsnapshot/tests/gtest/DoesCrossCompartmentBoundaries.cpp
blob: bc517d6d939e533940288061737cfc85a5a386ad (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
/* -*-  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));
  });