node-gamedig/protocols/unreal2.js

153 lines
5.4 KiB
JavaScript
Raw Normal View History

2019-01-10 13:03:07 +01:00
const Core = require('./core');
2014-10-29 08:02:03 +01:00
class Unreal2 extends Core {
2017-08-09 12:32:09 +02:00
constructor() {
super();
this.encoding = 'latin1';
}
2019-01-10 13:03:07 +01:00
async run(state) {
let extraInfoReader;
2019-01-10 13:03:07 +01:00
{
const b = await this.sendPacket(0, true);
const reader = this.reader(b);
state.raw.serverid = reader.uint(4);
state.raw.ip = this.readUnrealString(reader);
2019-01-12 12:45:09 +01:00
state.gamePort = reader.uint(4);
2019-01-10 13:03:07 +01:00
state.raw.queryport = reader.uint(4);
state.name = this.readUnrealString(reader, true);
state.map = this.readUnrealString(reader, true);
state.raw.gametype = this.readUnrealString(reader, true);
state.raw.numplayers = reader.uint(4);
state.maxplayers = reader.uint(4);
this.logger.debug(log => {
log("UNREAL2 EXTRA INFO", reader.buffer.slice(reader.i));
});
extraInfoReader = reader;
2019-01-10 13:03:07 +01:00
}
2014-10-29 08:02:03 +01:00
2019-01-10 13:03:07 +01:00
{
const b = await this.sendPacket(1,true);
const reader = this.reader(b);
state.raw.mutators = [];
state.raw.rules = {};
while(!reader.done()) {
const key = this.readUnrealString(reader,true);
const value = this.readUnrealString(reader,true);
this.logger.debug(key+'='+value);
if(key === 'Mutator' || key === 'mutator') {
state.raw.mutators.push(value);
} else if (key || value) {
if (state.raw.rules.hasOwnProperty(key)) {
state.raw.rules[key] += ',' + value;
} else {
state.raw.rules[key] = value;
}
}
2019-01-10 13:03:07 +01:00
}
if('GamePassword' in state.raw.rules)
state.password = state.raw.rules.GamePassword !== 'True';
}
2014-10-29 08:02:03 +01:00
if (state.raw.mutators.includes('KillingFloorMut')
|| state.raw.rules['Num trader weapons']
|| state.raw.rules['Server Version'] === '1065'
) {
// Killing Floor
state.raw.wavecurrent = extraInfoReader.uint(4);
state.raw.wavetotal = extraInfoReader.uint(4);
state.raw.ping = extraInfoReader.uint(4);
state.raw.flags = extraInfoReader.uint(4);
state.raw.skillLevel = this.readUnrealString(extraInfoReader, true);
} else {
state.raw.ping = extraInfoReader.uint(4);
// These fields were added in later revisions of unreal engine
if (extraInfoReader.remaining() >= 8) {
state.raw.flags = extraInfoReader.uint(4);
state.raw.skill = this.readUnrealString(extraInfoReader, true);
}
}
2019-01-10 13:03:07 +01:00
{
const b = await this.sendPacket(2,false);
const reader = this.reader(b);
2014-10-29 08:02:03 +01:00
state.raw.scoreboard = {};
2019-01-10 13:03:07 +01:00
while(!reader.done()) {
const player = {};
player.id = reader.uint(4);
player.name = this.readUnrealString(reader,true);
player.ping = reader.uint(4);
player.score = reader.int(4);
player.statsId = reader.uint(4);
this.logger.debug(player);
2014-10-29 08:02:03 +01:00
if (!player.id) {
state.raw.scoreboard[player.name] = player.score;
} else if (!player.ping) {
state.bots.push(player);
} else {
state.players.push(player);
2019-01-10 13:03:07 +01:00
}
2017-08-09 12:32:09 +02:00
}
2019-01-10 13:03:07 +01:00
}
2017-08-09 12:32:09 +02:00
}
2019-01-10 13:03:07 +01:00
2017-08-09 12:32:09 +02:00
readUnrealString(reader, stripColor) {
let length = reader.uint(1), ucs2 = false;
if(length >= 0x80) {
// This is flagged as a UCS-2 String
2017-08-09 12:32:09 +02:00
length = (length&0x7f)*2;
ucs2 = true;
// For UCS-2 strings, some unreal 2 games randomly insert an extra 0x01 here,
// not included in the length. Skip it if present (hopefully this never happens legitimately)
const peek = reader.uint(1);
if (peek !== 1) reader.skip(-1);
2019-01-09 12:50:30 +01:00
this.debugLog(log => {
log("UCS2 STRING");
log("UCS2 Length: " + length);
log(reader.buffer.slice(reader.i,reader.i+length));
2019-01-09 12:50:30 +01:00
});
}
let out = '';
if (ucs2) {
2017-08-09 12:32:09 +02:00
out = reader.string({encoding:'ucs2',length:length});
this.debugLog("UCS2 String decoded: " + out);
} else if (length > 0) {
out = reader.string();
2017-08-09 12:32:09 +02:00
}
// Sometimes the string has a null at the end (included with the length)
// Strip it if present
if(out.charCodeAt(out.length-1) === 0) {
out = out.substring(0, out.length - 1);
}
2017-08-09 12:32:09 +02:00
if(stripColor) {
out = out.replace(/\x1b...|[\x00-\x1a]/gus,'');
}
2014-10-29 08:02:03 +01:00
2017-08-09 12:32:09 +02:00
return out;
}
2019-01-10 13:03:07 +01:00
async sendPacket(type,required) {
const outbuffer = Buffer.from([0x79,0,0,0,type]);
2014-10-29 08:02:03 +01:00
const packets = [];
2019-01-10 13:03:07 +01:00
return await this.udpSend(outbuffer,(buffer) => {
const reader = this.reader(buffer);
const header = reader.uint(4);
const iType = reader.uint(1);
2017-08-09 12:32:09 +02:00
if(iType !== type) return;
packets.push(reader.rest());
}, () => {
if(!packets.length && required) return;
2019-01-10 13:03:07 +01:00
return Buffer.concat(packets);
2017-08-09 12:32:09 +02:00
});
}
}
module.exports = Unreal2;