bitcoincashjs/Peer.js

require('classtool');

function spec(b) {
  var Net = b.Net || require('net');
  var Binary = b.Binary || require('binary');

  function Peer(host, port, services) {
    if ("string" === typeof host) {
      if (host.indexOf(':') && !port) {
        var parts = host.split(':');
        host = parts[0];
        port = parts[1];
      }
      this.host = host;
      this.port = +port || 8333;
    } else if (host instanceof Peer) {
      this.host = host.host;
      this.port = host.port;
    } else if (Buffer.isBuffer(host)) {
      if (host.slice(0, 12).compare(Peer.IPV6_IPV4_PADDING) != 0) {
        throw new Error('IPV6 not supported yet! Cannot instantiate host.');
      }
      this.host = Array.prototype.slice.apply(host.slice(12)).join('.');
      this.port = +port || 8333;
    } else {
      throw new Error('Could not instantiate peer, invalid parameter type: ' +
                      typeof host);
    }

    this.services = (services) ? services : null;
    this.lastSeen = 0;
  };

  Peer.IPV6_IPV4_PADDING = new Buffer([0,0,0,0,0,0,0,0,0,0,255,255]);

  Peer.prototype.createConnection = function () {
    var c = Net.createConnection(this.port, this.host);
    return c;
  };

  Peer.prototype.getHostAsBuffer = function () {
    return new Buffer(this.host.split('.'));
  };

  Peer.prototype.toString = function () {
    return this.host + ":" + this.port;
  };

  Peer.prototype.toBuffer = function () {
    var put = Binary.put();
    put.word32le(this.lastSeen);
    put.word64le(this.services);
    put.put(this.getHostAsBuffer());
    put.word16be(this.port);
    return put.buffer();
  };

  return Peer;
};
module.defineClass(spec);