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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
/* 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/. */
import { XPCOMUtils } from "resource://gre/modules/XPCOMUtils.sys.mjs";
const lazy = {};
ChromeUtils.defineESModuleGetters(lazy, {
AsyncShutdown: "resource://gre/modules/AsyncShutdown.sys.mjs",
UnsupportedError: "chrome://remote/content/cdp/Error.sys.mjs",
});
XPCOMUtils.defineLazyModuleGetters(lazy, {
OS: "resource://gre/modules/osfile.jsm",
});
export class StreamRegistry {
constructor() {
// handle => stream
this.streams = new Map();
// Register an async shutdown blocker to ensure all open IO streams are
// closed, and remaining temporary files removed. Needs to happen before
// OS.File has been shutdown.
lazy.AsyncShutdown.profileBeforeChange.addBlocker(
"Remote Agent: Clean-up of open streams",
async () => {
await this.destructor();
}
);
}
async destructor() {
for (const stream of this.streams.values()) {
await this._discard(stream);
}
this.streams.clear();
}
async _discard(stream) {
if (stream instanceof lazy.OS.File) {
let fileInfo;
// Also remove the temporary file
try {
fileInfo = await stream.stat();
stream.close();
await lazy.OS.File.remove(fileInfo.path, { ignoreAbsent: true });
} catch (e) {
console.error(`Failed to remove ${fileInfo?.path}: ${e.message}`);
}
}
}
/**
* Add a new stream to the registry.
*
* @param {OS.File} stream
* Instance of the stream to add.
*
* @return {string}
* Stream handle (uuid)
*/
add(stream) {
let handle;
if (stream instanceof lazy.OS.File) {
handle = Services.uuid
.generateUUID()
.toString()
.slice(1, -1);
} else {
// Bug 1602731 - Implement support for blob
throw new lazy.UnsupportedError(`Unknown stream type for ${stream}`);
}
this.streams.set(handle, stream);
return handle;
}
/**
* Get a stream from the registry.
*
* @param {string} handle
* Handle of the stream to retrieve.
*
* @return {OS.File}
* Requested stream
*/
get(handle) {
const stream = this.streams.get(handle);
if (!stream) {
throw new TypeError(`Invalid stream handle`);
}
return stream;
}
/**
* Remove a stream from the registry.
*
* @param {string} handle
* Handle of the stream to remove.
*
* @return {boolean}
* true if successfully removed
*/
async remove(handle) {
const stream = this.get(handle);
await this._discard(stream);
return this.streams.delete(handle);
}
}
|
