summaryrefslogtreecommitdiffstats
path: root/devtools/server/actors/heap-snapshot-file.js
blob: d78fcc1a38597a7da1df82dbf71f20cf54165be6 (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
74
75
76
77
78
79
80
81
82
83
/* 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/. */

"use strict";

const protocol = require("devtools/shared/protocol");
const Services = require("Services");

const {
  heapSnapshotFileSpec,
} = require("devtools/shared/specs/heap-snapshot-file");

loader.lazyRequireGetter(
  this,
  "DevToolsUtils",
  "devtools/shared/DevToolsUtils"
);
loader.lazyRequireGetter(this, "OS", "resource://gre/modules/osfile.jsm", true);
loader.lazyRequireGetter(
  this,
  "HeapSnapshotFileUtils",
  "devtools/shared/heapsnapshot/HeapSnapshotFileUtils"
);

/**
 * The HeapSnapshotFileActor handles transferring heap snapshot files from the
 * server to the client. This has to be a global actor in the parent process
 * because child processes are sandboxed and do not have access to the file
 * system.
 */
exports.HeapSnapshotFileActor = protocol.ActorClassWithSpec(
  heapSnapshotFileSpec,
  {
    initialize: function(conn, parent) {
      if (
        Services.appinfo.processType !== Services.appinfo.PROCESS_TYPE_DEFAULT
      ) {
        const err = new Error(
          "Attempt to create a HeapSnapshotFileActor in a " +
            "child process! The HeapSnapshotFileActor *MUST* " +
            "be in the parent process!"
        );
        DevToolsUtils.reportException(
          "HeapSnapshotFileActor.prototype.initialize",
          err
        );
        return;
      }

      protocol.Actor.prototype.initialize.call(this, conn, parent);
    },

    /**
     * @see MemoryFront.prototype.transferHeapSnapshot
     */
    async transferHeapSnapshot(snapshotId) {
      const snapshotFilePath = HeapSnapshotFileUtils.getHeapSnapshotTempFilePath(
        snapshotId
      );
      if (!snapshotFilePath) {
        throw new Error(`No heap snapshot with id: ${snapshotId}`);
      }

      const streamPromise = DevToolsUtils.openFileStream(snapshotFilePath);

      const { size } = await OS.File.stat(snapshotFilePath);
      const bulkPromise = this.conn.startBulkSend({
        actor: this.actorID,
        type: "heap-snapshot",
        length: size,
      });

      const [bulk, stream] = await Promise.all([bulkPromise, streamPromise]);

      try {
        await bulk.copyFrom(stream);
      } finally {
        stream.close();
      }
    },
  }
);