Ylianst/MeshCentral
1
0
Fork 0
mirror of https://github.com/Ylianst/MeshCentral.git synced 2024-11-22 12:52:50 +03:00
Code Issues Packages Projects Releases Wiki Activity

Improved handling of HTTP session socket closure (#4369)

Browse Source
This commit is contained in:
Ylian Saint-Hilaire 2022-08-08 16:39:27 -07:00
parent b96df86296
commit 1257c4bc3a
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
apprelays.js
View File

@ -241,6 +241,9 @@ module.exports.CreateWebRelay = function (parent, db, args, domain) {
obj.onconnect = null;
obj.onNextRequest = null;
// Called when we need to close the tunnel because the response stream has closed
function handleResponseClosure() { obj.close(); }
// Process a HTTP request
obj.processRequest = function (req, res) {
if (obj.relayActive == false) { console.log("ERROR: Attempt to use an unconnected tunnel"); return false; }
@ -249,6 +252,9 @@ module.exports.CreateWebRelay = function (parent, db, args, domain) {
// Check if this is a websocket
if (req.headers['upgrade'] == 'websocket') { console.log('Attempt to process a websocket in HTTP tunnel method.'); res.end(); return false; }
// If the response stream is closed, close this tunnel right away
res.socket.on('end', handleResponseClosure);
// Construct the HTTP request
var request = req.method + ' ' + req.url + ' HTTP/' + req.httpVersion + '\r\n';
const blockedHeaders = ['origin', 'cookie', 'upgrade-insecure-requests', 'sec-ch-ua', 'sec-ch-ua-mobile', 'dnt', 'sec-fetch-user', 'sec-ch-ua-platform', 'sec-fetch-site', 'sec-fetch-mode', 'sec-fetch-dest']; // These are headers we do not forward
@ -288,6 +294,9 @@ module.exports.CreateWebRelay = function (parent, db, args, domain) {
// Pause the websocket until we get a tunnel connected
obj.ws._socket.pause();
// If the response stream is closed, close this tunnel right away
res.socket.on('end', function () { obj.close(); });
// Remove the trailing '/.websocket' if needed
var baseurl = req.url, i = req.url.indexOf('?');
if (i > 0) { baseurl = req.url.substring(0, i); }
@ -392,7 +401,7 @@ module.exports.CreateWebRelay = function (parent, db, args, domain) {
}
// Close any pending request
if (obj.res) { obj.res.end(); delete obj.res; }
if (obj.res) { obj.res.socket.removeListener('end', handleResponseClosure); obj.res.end(); delete obj.res; }
if (obj.ws) { obj.ws.close(); delete obj.ws; }
// Event disconnection
@ -689,6 +698,7 @@ module.exports.CreateWebRelay = function (parent, db, args, domain) {
// If we are done, close the response
if (done == true) {
// Close the response
obj.res.socket.removeListener('end', handleResponseClosure);
obj.res.end();
delete obj.res;