mirror of
https://github.com/PrismarineJS/node-minecraft-protocol.git
synced 2024-11-29 18:55:40 -05:00
transform client.js into an es6 class
This commit is contained in:
Romain Beaumont
parent
2c93a4e796
commit
4e9d8e06c9
1 changed files with 145 additions and 145 deletions
290
src/client.js
290
src/client.js
|
|
@ -1,161 +1,161 @@
|
|||
var EventEmitter = require('events').EventEmitter
|
||||
, util = require('util')
|
||||
, debug = require('./debug')
|
||||
, serializer = require('./transforms/serializer')
|
||||
, compression = require('./transforms/compression')
|
||||
, framing = require('./transforms/framing')
|
||||
, crypto = require('crypto')
|
||||
, states = serializer.states
|
||||
;
|
||||
var EventEmitter = require('events').EventEmitter;
|
||||
var debug = require('./debug');
|
||||
var serializer = require('./transforms/serializer');
|
||||
var compression = require('./transforms/compression');
|
||||
var framing = require('./transforms/framing');
|
||||
var crypto = require('crypto');
|
||||
var states = serializer.states;
|
||||
var version = require('./version');
|
||||
var packets = require('minecraft-data')(version.majorVersion).protocol.states;
|
||||
var readPackets = require("./packets").readPackets;
|
||||
var packetIndexes = readPackets(packets, states);
|
||||
|
||||
module.exports = Client;
|
||||
|
||||
function Client(isServer) {
|
||||
EventEmitter.call(this);
|
||||
class Client extends EventEmitter
|
||||
{
|
||||
constructor(isServer) {
|
||||
super();
|
||||
|
||||
var socket;
|
||||
this.packetsToParse = {};
|
||||
var socket;
|
||||
this.packetsToParse = {};
|
||||
|
||||
this.serializer = serializer.createSerializer({ isServer });
|
||||
this.compressor = null;
|
||||
this.framer = framing.createFramer();
|
||||
this.cipher = null;
|
||||
this.serializer = serializer.createSerializer({ isServer });
|
||||
this.compressor = null;
|
||||
this.framer = framing.createFramer();
|
||||
this.cipher = null;
|
||||
|
||||
this.decipher = null;
|
||||
this.splitter = framing.createSplitter();
|
||||
this.decompressor = null;
|
||||
this.deserializer = serializer.createDeserializer({ isServer, packetsToParse: this.packetsToParse });
|
||||
this.decipher = null;
|
||||
this.splitter = framing.createSplitter();
|
||||
this.decompressor = null;
|
||||
this.deserializer = serializer.createDeserializer({ isServer, packetsToParse: this.packetsToParse });
|
||||
|
||||
this._state = states.HANDSHAKING;
|
||||
Object.defineProperty(this, "state", {
|
||||
get: function() {
|
||||
return this.serializer.protocolState;
|
||||
},
|
||||
set: function(newProperty) {
|
||||
var oldProperty = this.serializer.protocolState;
|
||||
this.serializer.protocolState = newProperty;
|
||||
this.deserializer.protocolState = newProperty;
|
||||
this.emit('state', newProperty, oldProperty);
|
||||
this._state = states.HANDSHAKING;
|
||||
Object.defineProperty(this, "state", {
|
||||
get: function() {
|
||||
return this.serializer.protocolState;
|
||||
},
|
||||
set: function(newProperty) {
|
||||
var oldProperty = this.serializer.protocolState;
|
||||
this.serializer.protocolState = newProperty;
|
||||
this.deserializer.protocolState = newProperty;
|
||||
this.emit('state', newProperty, oldProperty);
|
||||
}
|
||||
});
|
||||
Object.defineProperty(this, "compressionThreshold", {
|
||||
get: () => this.compressor == null ? -2 : this.compressor.compressionThreshold,
|
||||
set: (threshold) => this.setCompressionThreshold(threshold)
|
||||
});
|
||||
|
||||
this.isServer = !!isServer;
|
||||
|
||||
this.on('newListener', function(event, listener) {
|
||||
var direction = this.isServer ? 'toServer' : 'toClient';
|
||||
if(packetIndexes.packetStates[direction].hasOwnProperty(event) || event === "packet") {
|
||||
if(typeof this.packetsToParse[event] === "undefined") this.packetsToParse[event] = 1;
|
||||
else this.packetsToParse[event] += 1;
|
||||
}
|
||||
});
|
||||
this.on('removeListener', function(event, listener) {
|
||||
var direction = this.isServer ? 'toServer' : 'toClient';
|
||||
if(packetIndexes.packetStates[direction].hasOwnProperty(event) || event === "packet") {
|
||||
this.packetsToParse[event] -= 1;
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
setSocket(socket) {
|
||||
var ended = false;
|
||||
|
||||
// TODO : A lot of other things needs to be done.
|
||||
var endSocket = () => {
|
||||
if(ended) return;
|
||||
ended = true;
|
||||
this.socket.removeListener('close', endSocket);
|
||||
this.socket.removeListener('end', endSocket);
|
||||
this.socket.removeListener('timeout', endSocket);
|
||||
this.emit('end', this._endReason);
|
||||
};
|
||||
|
||||
var onFatalError = (err) => {
|
||||
this.emit('error', err);
|
||||
endSocket();
|
||||
};
|
||||
|
||||
var onError = (err) => this.emit('error', err);
|
||||
|
||||
this.socket = socket;
|
||||
|
||||
if(this.socket.setNoDelay)
|
||||
this.socket.setNoDelay(true);
|
||||
|
||||
this.socket.on('connect', () => this.emit('connect'));
|
||||
|
||||
this.socket.on('error', onFatalError);
|
||||
this.socket.on('close', endSocket);
|
||||
this.socket.on('end', endSocket);
|
||||
this.socket.on('timeout', endSocket);
|
||||
this.serializer.on('error', onError);
|
||||
this.deserializer.on('error', onError);
|
||||
this.framer.on('error', onError);
|
||||
this.splitter.on('error', onError);
|
||||
|
||||
this.socket.pipe(this.splitter).pipe(this.deserializer);
|
||||
this.serializer.pipe(this.framer).pipe(this.socket);
|
||||
|
||||
this.deserializer.on('data', (parsed) => {
|
||||
this.emit('packet', parsed.data, parsed.metadata);
|
||||
this.emit(parsed.metadata.name, parsed.data, parsed.metadata);
|
||||
this.emit('raw.' + parsed.metadata.name, parsed.buffer, parsed.metadata);
|
||||
this.emit('raw', parsed.buffer, parsed.metadata);
|
||||
});
|
||||
}
|
||||
|
||||
end(reason) {
|
||||
this._endReason = reason;
|
||||
this.socket.end();
|
||||
}
|
||||
|
||||
setEncryption(sharedSecret) {
|
||||
if (this.cipher != null)
|
||||
throw new Error("Set encryption twice !");
|
||||
this.cipher = crypto.createCipheriv('aes-128-cfb8', sharedSecret, sharedSecret);
|
||||
this.cipher.on('error', (err) => this.emit('error', err));
|
||||
this.framer.unpipe(this.socket);
|
||||
this.framer.pipe(this.cipher).pipe(this.socket);
|
||||
this.decipher = crypto.createDecipheriv('aes-128-cfb8', sharedSecret, sharedSecret);
|
||||
this.decipher.on('error', (err) => this.emit('error', err));
|
||||
this.socket.unpipe(this.splitter);
|
||||
this.socket.pipe(this.decipher).pipe(this.splitter);
|
||||
}
|
||||
|
||||
setCompressionThreshold(threshold) {
|
||||
if (this.compressor == null) {
|
||||
this.compressor = compression.createCompressor(threshold);
|
||||
this.compressor.on('error', (err) => this.emit('error', err));
|
||||
this.serializer.unpipe(this.framer);
|
||||
this.serializer.pipe(this.compressor).pipe(this.framer);
|
||||
this.decompressor = compression.createDecompressor(threshold);
|
||||
this.decompressor.on('error', (err) => this.emit('error', err));
|
||||
this.splitter.unpipe(this.deserializer);
|
||||
this.splitter.pipe(this.decompressor).pipe(this.deserializer);
|
||||
} else {
|
||||
this.decompressor.threshold = threshold;
|
||||
this.compressor.threshold = threshold;
|
||||
}
|
||||
});
|
||||
Object.defineProperty(this, "compressionThreshold", {
|
||||
get: () => this.compressor == null ? -2 : this.compressor.compressionThreshold,
|
||||
set: (threshold) => this.setCompressionThreshold(threshold)
|
||||
});
|
||||
}
|
||||
|
||||
this.isServer = !!isServer;
|
||||
write(packetName, params) {
|
||||
debug("writing packet " + this.state + "." + packetName);
|
||||
debug(params);
|
||||
this.serializer.write({ packetName, params });
|
||||
}
|
||||
|
||||
this.on('newListener', function(event, listener) {
|
||||
var direction = this.isServer ? 'toServer' : 'toClient';
|
||||
if(packetIndexes.packetStates[direction].hasOwnProperty(event) || event === "packet") {
|
||||
if(typeof this.packetsToParse[event] === "undefined") this.packetsToParse[event] = 1;
|
||||
else this.packetsToParse[event] += 1;
|
||||
}
|
||||
});
|
||||
this.on('removeListener', function(event, listener) {
|
||||
var direction = this.isServer ? 'toServer' : 'toClient';
|
||||
if(packetIndexes.packetStates[direction].hasOwnProperty(event) || event === "packet") {
|
||||
this.packetsToParse[event] -= 1;
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
util.inherits(Client, EventEmitter);
|
||||
|
||||
Client.prototype.setSocket = function(socket) {
|
||||
var ended = false;
|
||||
|
||||
// TODO : A lot of other things needs to be done.
|
||||
var endSocket = () => {
|
||||
if(ended) return;
|
||||
ended = true;
|
||||
this.socket.removeListener('close', endSocket);
|
||||
this.socket.removeListener('end', endSocket);
|
||||
this.socket.removeListener('timeout', endSocket);
|
||||
this.emit('end', this._endReason);
|
||||
};
|
||||
|
||||
var onFatalError = (err) => {
|
||||
this.emit('error', err);
|
||||
endSocket();
|
||||
};
|
||||
|
||||
var onError = (err) => this.emit('error', err);
|
||||
|
||||
this.socket = socket;
|
||||
|
||||
if(this.socket.setNoDelay)
|
||||
this.socket.setNoDelay(true);
|
||||
|
||||
this.socket.on('connect', () => this.emit('connect'));
|
||||
|
||||
this.socket.on('error', onFatalError);
|
||||
this.socket.on('close', endSocket);
|
||||
this.socket.on('end', endSocket);
|
||||
this.socket.on('timeout', endSocket);
|
||||
this.serializer.on('error', onError);
|
||||
this.deserializer.on('error', onError);
|
||||
this.framer.on('error', onError);
|
||||
this.splitter.on('error', onError);
|
||||
|
||||
this.socket.pipe(this.splitter).pipe(this.deserializer);
|
||||
this.serializer.pipe(this.framer).pipe(this.socket);
|
||||
|
||||
this.deserializer.on('data', (parsed) => {
|
||||
this.emit('packet', parsed.data, parsed.metadata);
|
||||
this.emit(parsed.metadata.name, parsed.data, parsed.metadata);
|
||||
this.emit('raw.' + parsed.metadata.name, parsed.buffer, parsed.metadata);
|
||||
this.emit('raw', parsed.buffer, parsed.metadata);
|
||||
});
|
||||
};
|
||||
|
||||
Client.prototype.end = function(reason) {
|
||||
this._endReason = reason;
|
||||
this.socket.end();
|
||||
};
|
||||
|
||||
Client.prototype.setEncryption = function(sharedSecret) {
|
||||
if (this.cipher != null)
|
||||
throw new Error("Set encryption twice !");
|
||||
this.cipher = crypto.createCipheriv('aes-128-cfb8', sharedSecret, sharedSecret);
|
||||
this.cipher.on('error', (err) => this.emit('error', err));
|
||||
this.framer.unpipe(this.socket);
|
||||
this.framer.pipe(this.cipher).pipe(this.socket);
|
||||
this.decipher = crypto.createDecipheriv('aes-128-cfb8', sharedSecret, sharedSecret);
|
||||
this.decipher.on('error', (err) => this.emit('error', err));
|
||||
this.socket.unpipe(this.splitter);
|
||||
this.socket.pipe(this.decipher).pipe(this.splitter);
|
||||
}
|
||||
|
||||
Client.prototype.setCompressionThreshold = function(threshold) {
|
||||
if (this.compressor == null) {
|
||||
this.compressor = compression.createCompressor(threshold);
|
||||
this.compressor.on('error', (err) => this.emit('error', err));
|
||||
this.serializer.unpipe(this.framer);
|
||||
this.serializer.pipe(this.compressor).pipe(this.framer);
|
||||
this.decompressor = compression.createDecompressor(threshold);
|
||||
this.decompressor.on('error', (err) => this.emit('error', err));
|
||||
this.splitter.unpipe(this.deserializer);
|
||||
this.splitter.pipe(this.decompressor).pipe(this.deserializer);
|
||||
} else {
|
||||
this.decompressor.threshold = threshold;
|
||||
this.compressor.threshold = threshold;
|
||||
writeRaw(buffer) {
|
||||
if (this.compressor === null)
|
||||
this.framer.write(buffer);
|
||||
else
|
||||
this.compressor.write(buffer);
|
||||
}
|
||||
}
|
||||
|
||||
Client.prototype.write = function(packetName, params) {
|
||||
debug("writing packet " + this.state + "." + packetName);
|
||||
debug(params);
|
||||
this.serializer.write({ packetName, params });
|
||||
};
|
||||
|
||||
Client.prototype.writeRaw = function(buffer) {
|
||||
if (this.compressor === null)
|
||||
this.framer.write(buffer);
|
||||
else
|
||||
this.compressor.write(buffer);
|
||||
};
|
||||
module.exports = Client;
|
||||
|
|
|
|||
Loading…
Reference in a new issue