node-gamedig/lib/GlobalUdpSocket.js

69 lines
2.2 KiB
JavaScript
Raw Normal View History

const dgram = require('dgram');
const HexUtil = require('./HexUtil');
const Logger = require('./Logger');
const util = require('util');
2019-01-12 11:43:36 +01:00
class GlobalUdpSocket {
constructor({port}) {
2019-01-12 11:43:36 +01:00
this.socket = null;
this.callbacks = new Set();
this.debuggingCallbacks = new Set();
2019-01-20 11:21:40 +01:00
this.logger = new Logger();
this.port = port;
2019-01-12 11:43:36 +01:00
}
async _getSocket() {
2019-01-12 11:43:36 +01:00
if (!this.socket) {
const udpSocket = dgram.createSocket({
type: 'udp4',
reuseAddr: true
});
2019-01-12 11:43:36 +01:00
udpSocket.unref();
udpSocket.on('message', (buffer, rinfo) => {
const fromAddress = rinfo.address;
const fromPort = rinfo.port;
2019-01-20 11:21:40 +01:00
this.logger.debug(log => {
log(fromAddress + ':' + fromPort + " <--UDP(" + this.port + ")");
2019-01-20 11:21:40 +01:00
log(HexUtil.debugDump(buffer));
});
2019-01-12 11:43:36 +01:00
for (const cb of this.callbacks) {
cb(fromAddress, fromPort, buffer);
2019-01-12 11:43:36 +01:00
}
});
2019-01-20 11:21:40 +01:00
udpSocket.on('error', e => {
this.logger.debug("UDP ERROR:", e);
2019-01-12 11:43:36 +01:00
});
await util.promisify(udpSocket.bind).bind(udpSocket)(this.port);
this.port = udpSocket.address().port;
this.socket = udpSocket;
2019-01-12 11:43:36 +01:00
}
return this.socket;
}
async send(buffer, address, port, debug) {
const socket = await this._getSocket();
if (debug) {
this.logger._print(log => {
log(address + ':' + port + " UDP(" + this.port + ")-->");
log(HexUtil.debugDump(buffer));
});
}
await util.promisify(socket.send).bind(socket)(buffer,0,buffer.length,port,address);
2019-01-12 11:43:36 +01:00
}
addCallback(callback, debug) {
2019-01-12 11:43:36 +01:00
this.callbacks.add(callback);
if (debug) {
this.debuggingCallbacks.add(callback);
2019-01-20 11:21:40 +01:00
this.logger.debugEnabled = true;
}
2019-01-12 11:43:36 +01:00
}
removeCallback(callback) {
this.callbacks.delete(callback);
this.debuggingCallbacks.delete(callback);
2019-01-20 11:21:40 +01:00
this.logger.debugEnabled = this.debuggingCallbacks.size > 0;
2019-01-12 11:43:36 +01:00
}
}
module.exports = GlobalUdpSocket;