node-minecraft-protocol/examples/proxy/proxy.js

var mc = require('../../');

var states = mc.states;
function printHelpAndExit(exitCode) {
  console.log("usage: node proxy.js [<options>...] <target_srv> <user> [<password>]");
  console.log("options:");
  console.log("  --dump name");
  console.log("    print to stdout messages with the specified name.");
  console.log("  --dump-all");
  console.log("    print to stdout all messages, except those specified with -x.");
  console.log("  -x name");
  console.log("    do not print messages with this name.");
  console.log("  name");
  console.log("    a packet name as defined in protocol.json");
  console.log("examples:");
  console.log("  node proxy.js --dump-all -x keep_alive -x update_time -x entity_velocity -x rel_entity_move -x entity_look -x entity_move_look -x entity_teleport -x entity_head_rotation -x position -x localhost Player");
  console.log("    print all messages except for some of the most prolific.");
  console.log("  node examples/proxy.js --dump open_window --dump close_window --dump set_slot --dump window_items --dump craft_progress_bar --dump transaction --dump close_window --dump window_click --dump set_creative_slot --dump enchant_item localhost Player");
  console.log("    print messages relating to inventory management.");

  process.exit(exitCode);
}

if(process.argv.length < 4) {
  console.log("Too few arguments!");
  printHelpAndExit(1);
}

process.argv.forEach(function(val, index, array) {
  if(val == "-h") {
    printHelpAndExit(0);
  }
});

var args = process.argv.slice(2);
var host;
var port = 25565;
var user;
var passwd;

var printAllNames = false;
var printNameWhitelist = {};
var printNameBlacklist = {};
(function() {
  for(var i = 0; i < args.length; i++) {
    var option = args[i];
    if(!/^-/.test(option)) break;
    if(option == "--dump-all") {
      printAllNames = true;
      continue;
    }
    i++;
    var name = args[i];
    if(option == "--dump") {
      printNameWhitelist[name] = "io";
    } else if(option == "-x") {
      printNameBlacklist[name] = "io";
    } else {
      printHelpAndExit(1);
    }
  }
  if(!(i + 2 <= args.length && args.length <= i + 3)) printHelpAndExit(1);
  host = args[i++];
  user = args[i++];
  passwd = args[i++];
})();

if(host.indexOf(':') != -1) {
  port = host.substring(host.indexOf(':') + 1);
  host = host.substring(0, host.indexOf(':'));
}

var srv = mc.createServer({
  'online-mode': false,
  port: 25566,
  keepAlive: false
});
srv.on('login', function(client) {
  var addr = client.socket.remoteAddress;
  console.log('Incoming connection', '(' + addr + ')');
  var endedClient = false;
  var endedTargetClient = false;
  client.on('end', function() {
    endedClient = true;
    console.log('Connection closed by client', '(' + addr + ')');
    if(!endedTargetClient)
      targetClient.end("End");
  });
  client.on('error', function() {
    endedClient = true;
    console.log('Connection error by client', '(' + addr + ')');
    if(!endedTargetClient)
      targetClient.end("Error");
  });
  var targetClient = mc.createClient({
    host: host,
    port: port,
    username: user,
    password: passwd,
    'online-mode': passwd != null ? true : false,
    keepAlive:false
  });
  client.on('packet', function(data, meta) {
    if(targetClient.state == states.PLAY && meta.state == states.PLAY) {
      if(shouldDump(meta.name, "o")) {
        console.log("client->server:",
          client.state + ".0x" + meta.id.toString(16) + " :",
          JSON.stringify(data));
      }
      if(!endedTargetClient)
        targetClient.write(meta.name, data);
    }
  });
  targetClient.on('packet', function(data, meta) {
    if(meta.state == states.PLAY && client.state == states.PLAY) {
      if(shouldDump(meta.name, "i")) {
        console.log("client<-server:",
          targetClient.state + "." + meta.name + " :" +
          JSON.stringify(data));
      }
      if(!endedClient)
        client.write(meta.name, data);
      if (meta.name === 'set_compression' || meta.name === 'compression') // Set compression
        client.compressionThreshold = data.threshold;
    }
  });
  var buffertools = require('buffertools');
  targetClient.on('raw', function(buffer, meta) {
    if(client.state != states.PLAY || meta.state != states.PLAY)
      return;
    var packetData = mc.parsePacketData(buffer, meta.state, false, {"packet": 1}).data;
    var packetBuff = mc.createPacketBuffer(meta.name, meta.state, packetData, true);
    if(buffertools.compare(buffer, packetBuff) != 0) {
      console.log("client<-server: Error in packet " + state + "." + meta.name);
      console.log(buffer.toString('hex'));
      console.log(packetBuff.toString('hex'));
      console.log(buffer.length);
      console.log(packetBuff.length);
    }
    /*if (client.state == states.PLAY && brokenPackets.indexOf(packetId.value) !== -1)
     {
     console.log(`client<-server: raw packet);
     console.log(packetData);
     if (!endedClient)
     client.writeRaw(buffer);
     }*/
  });
  client.on('raw', function(buffer, meta) {
    if(meta.state != states.PLAY || targetClient.state != states.PLAY)
      return;
    var packetData = mc.parsePacketData(buffer, meta.state, true, {"packet": 1}).data;
    var packetBuff = mc.createPacketBuffer(meta.name, meta.state, packetData, false);
    if(buffertools.compare(buffer, packetBuff) != 0) {
      console.log("client->server: Error in packet " + state + "." + meta.name);
      console.log(buffer.toString('hex'));
      console.log(packetBuff.toString('hex'));
      console.log(buffer.length);
      console.log(packetBuff.length);
    }
  });
  targetClient.on('end', function() {
    endedTargetClient = true;
    console.log('Connection closed by server', '(' + addr + ')');
    if(!endedClient)
      client.end("End");
  });
  targetClient.on('error', function(err) {
    endedTargetClient = true;
    console.log('Connection error by server', '(' + addr + ') ',err);
    console.log(err.stack);
    if(!endedClient)
      client.end("Error");
  });
});

function shouldDump(name, direction) {
  if(matches(printNameBlacklist[name])) return false;
  if(printAllNames) return true;
  if(matches(printNameWhitelist[name])) return true;
  return false;
  function matches(result) {
    return result != null && result.indexOf(direction) !== -1;
  }
}