90 lines
No EOL
2.8 KiB
JavaScript
90 lines
No EOL
2.8 KiB
JavaScript
const EventEmitter = require('events');
|
|
|
|
const constants = require("./constants");
|
|
const parser = require('./Parser');
|
|
const { createLog } = require('./logger');
|
|
|
|
const socketLog = createLog([ 'Wormhole', 'Socket' ]);
|
|
|
|
class Socket extends EventEmitter {
|
|
constructor({ initialState='CONNECTING', socket }) {
|
|
super();
|
|
|
|
this._state = initialState;
|
|
this._socket = socket;
|
|
this._accepted = false;
|
|
this._fateDecided = false; // Wether the decision to accept or reject the socket was made
|
|
|
|
this._socket.on('data', (e) => {
|
|
if (!this._isConnected()) return socketLog('ERROR: Received data, but connection is not fully completed');
|
|
|
|
let packet;
|
|
try {
|
|
packet = this._decodePayload(e.buffer);
|
|
} catch(e) {
|
|
return socketLog('ERROR: Received data, but encountered error while parsing: ', e);
|
|
}
|
|
|
|
if (!packet.FIN) return socketLog('WARN: Message fragmentation is not supported, but got fragmented message');
|
|
|
|
if (packet.Opcode === constants.opcodes.TEXT_FRAME) {
|
|
this.emit('text', packet);
|
|
} else if (packet.Opcode === constants.opcodes.BINARY_FRAME) {
|
|
this.emit('binary', packet);
|
|
} else {
|
|
return socketLog(`WARN: Received unsupported opcode 0x${packet.Opcode.toString(16)}`);
|
|
}
|
|
});
|
|
}
|
|
}
|
|
|
|
Socket.prototype.send = function(payload) {
|
|
if (typeof payload === 'string') {
|
|
this._sendBuffer(Buffer.from(payload));
|
|
} else if (payload instanceof Buffer) {
|
|
this._sendBuffer(payload);
|
|
} else {
|
|
throw new Error(`Unsupported payload type ${typeof payload}, supported types are: string, Buffer`);
|
|
}
|
|
};
|
|
|
|
Socket.prototype._sendBuffer = function(buffer) {
|
|
const frame = new parser.WebsocketFrame();
|
|
frame.FIN = 1;
|
|
frame.RSVx = 0;
|
|
frame.Opcode = 0x01;
|
|
|
|
frame.MASK = 0;
|
|
const length = buffer.byteLength;
|
|
if (length > 125) {
|
|
frame.PayloadLen = 126;
|
|
if (length >= 65535) frame.PayloadLen = 127;
|
|
} else if (length <= 125) {
|
|
frame.PayloadLen = length;
|
|
}
|
|
frame.PayloadLenEx = length;
|
|
frame.PayloadData = buffer;
|
|
|
|
this._socket.write(frame.toBuffer());
|
|
};
|
|
|
|
Socket.prototype._decodePayload = function(payload) {
|
|
return parser.parseWebsocketFrame(new DataView(payload));
|
|
};
|
|
|
|
Socket.prototype._setConnectionState = function(state) {
|
|
this._state = state;
|
|
};
|
|
|
|
Socket.prototype._setAccepted = function(state) {
|
|
if (this._fateDecided) throw new Error('Tried to decide fate (wether socket is accepted or not) more than 1 time');
|
|
|
|
this._fateDecided = true;
|
|
this._accepted = state;
|
|
};
|
|
|
|
Socket.prototype._isConnected = function() {
|
|
return (this._fateDecided && this._isConnected && this._accepted);
|
|
}
|
|
|
|
module.exports = Socket; |