var async = require('async'), Bzip2 = require('compressjs').Bzip2; module.exports = require('./core').extend({ init: function() { this._super(); this.goldsrc = false; this.options.port = 27015; }, run: function() { var self = this; var challenge; var state = {}; async.series([ function(c) { self.sendPacket( 0x54,false,new Buffer('Source Engine Query\0'), self.goldsrc ? 0x6D : 0x49, function(b) { var reader = self.reader(b); if(self.goldsrc) state.address = reader.string(); else state.protocol = reader.uint(1); state.name = reader.string(); state.map = reader.string(); state.folder = reader.string(); state.game = reader.string(); state.steamappid = reader.uint(2); state.numplayers = reader.uint(1); state.maxplayers = reader.uint(1); if(self.goldsrc) state.protocol = reader.uint(1); else state.numbots = reader.uint(1); state.listentype = String.fromCharCode(reader.uint(1)); state.environment = String.fromCharCode(reader.uint(1)); state.passworded = reader.uint(1); if(self.goldsrc) { state.ismod = reader.uint(1); if(state.ismod) { state.modlink = reader.string(); state.moddownload = reader.string(); reader.skip(1); state.modversion = reader.uint(4); state.modsize = reader.uint(4); state.modtype = reader.uint(1); state.moddll = reader.uint(1); } } state.secure = reader.uint(1); if(self.goldsrc) { state.numbots = reader.uint(1); } else { if(state.folder == 'ship') { state.shipmode = reader.uint(1); state.shipwitnesses = reader.uint(1); state.shipduration = reader.uint(1); } state.version = reader.string(); var extraFlag = reader.uint(1); if(extraFlag & 0x80) state.port = reader.uint(2); if(extraFlag & 0x10) state.steamid = reader.uint(8); if(extraFlag & 0x40) { state.sourcetvport = reader.uint(2); state.sourcetvname = reader.string(); } if(extraFlag & 0x20) state.tags = reader.string(); if(extraFlag & 0x01) state.gameid = reader.uint(8); } c(); } ); }, function(c) { self.sendPacket(0x55,0xffffffff,false,0x41,function(b) { var reader = self.reader(b); challenge = reader.uint(4); c(); }); }, function(c) { self.sendPacket(0x55,challenge,false,0x44,function(b) { var reader = self.reader(b); var num = reader.uint(1); state.players = []; for(var i = 0; i < num; i++) { reader.skip(1); var name = reader.string(); var score = reader.uint(4); var time = reader.float(); state.players.push({ name:name, score:score, time:time }); } c(); }); }, function(c) { self.sendPacket(0x56,challenge,false,0x45,function(b) { var reader = self.reader(b); var num = reader.uint(2); state.rules = []; for(var i = 0; i < num; i++) { var key = reader.string(); var value = reader.string(); state.rules[key] = value; } c(); }); }, function(c) { self.finish(state); } ]); }, sendPacket: function(type,challenge,payload,expect,callback) { var self = this; var challengeLength = challenge === false ? 0 : 4; var payloadLength = payload ? payload.length : 0; var b = new Buffer(5 + challengeLength + payloadLength); b.writeInt32LE(-1, 0); b.writeUInt8(type, 4); if(challengeLength) b.writeUInt32LE(challenge, 5); if(payloadLength) payload.copy(b, 5+challengeLength); function received(payload) { var type = payload.readUInt8(0); if(type != expect) return; callback(payload.slice(1)); return true; } var numPackets = 0; var packets = []; var bzip = false; this.udpSend(b,function(buffer) { var header = buffer.readInt32LE(0); if(header == -1) { // full package return received(buffer.slice(4)); } if(header == -2) { // partial package var uid = buffer.readUInt32LE(4); if(!self.goldsrc && uid & 0x80000000) bzip = true; var id,payload; if(self.goldsrc) { id = buffer.readUInt8(8); numPackets = id & 0x0f; id = id & 0xf0 >> 4; payload = buffer.slice(9); } else { numPackets = buffer.readUInt8(8); id = buffer.readUInt8(9); if(id == 0 && bzip) payload = buffer.slice(20); else payload = buffer.slice(12); } packets[id] = payload; if(!numPackets || Object.keys(packets).length != numPackets) return; // assemble the parts var list = []; for(var i = 0; i < numPackets; i++) { if(!(i in packets)) { self.error('Missing packet #'+i); return true; } list.push(packets[i]); } var assembled = Buffer.concat(list); var payload = assembled.slice(4); if(bzip) payload = Bzip2.uncompressFile(payload); return received(payload); } }); } });