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
|
/* 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";
function WebSocketListener(closure, ws, sentMsg) {
this._closure = closure;
this._ws = ws;
this._sentMsg = sentMsg;
}
WebSocketListener.prototype = {
_closure: null,
_ws: null,
_sentMsg: null,
_received: null,
QueryInterface: ChromeUtils.generateQI(["nsIWebSocketListener"]),
onAcknowledge(aContext, aSize) {},
onBinaryMessageAvailable(aContext, aMsg) {
info("WsListener::onBinaryMessageAvailable");
this._received = aMsg;
this._ws.close(0, null);
},
onMessageAvailable(aContext, aMsg) {},
onServerClose(aContext, aCode, aReason) {},
onWebSocketListenerStart(aContext) {},
onStart(aContext) {
this._ws.sendMsg(this._sentMsg);
},
onStop(aContext, aStatusCode) {
try {
this._closure(aStatusCode, this._received);
this._ws = null;
} catch (ex) {
do_throw("Error in closure function: " + ex);
}
},
};
function makeWebSocketChan() {
let chan = Cc["@mozilla.org/network/protocol;1?name=wss"].createInstance(
Ci.nsIWebSocketChannel
);
chan.initLoadInfo(
null, // aLoadingNode
Services.scriptSecurityManager.getSystemPrincipal(),
null, // aTriggeringPrincipal
Ci.nsILoadInfo.SEC_ALLOW_CROSS_ORIGIN_SEC_CONTEXT_IS_NULL,
Ci.nsIContentPolicy.TYPE_WEBSOCKET
);
return chan;
}
function openWebSocketChannelPromise(chan, url, msg) {
let uri = Services.io.newURI(url);
return new Promise(resolve => {
function finish(status, result) {
resolve([status, result]);
}
chan.asyncOpen(
uri,
url,
{},
0,
new WebSocketListener(finish, chan, msg),
null
);
});
}
|