torrent_manager.js

var net = require('net');
var TorrentContext = require('./torrent_context');
var WP = require('wire_protocol');

var ctxs = {};
module.exports = {
    get: function(infoHex) {
	var ctx = ctxs.hasOwnProperty(infoHex) ?
	    ctxs[infoHex] :
	    (ctxs[infoHex] = new TorrentContext(module.exports, infoHex));
	ctx.lastGet = Date.now();
	return ctx;
    },

    port: parseInt(process.env.WIRE_PORT || "6881", 10),

    getStats: function() {
	var stats = { torrents: 0,
		      peers: { total: 0, connected: 0 },
		      downloaded: 0,
		      streams: 0
		    };

	for(var infoHex in ctxs)
	    if (ctxs.hasOwnProperty(infoHex)) {
		stats.torrents++;
		var ctxStats = ctxs[infoHex].getStats();
		stats.peers.total += ctxStats.peers.total;
		stats.peers.connected += ctxStats.peers.connected;
		stats.downloaded += ctxStats.downloaded;
		stats.streams += ctxStats.streams;
	    }

	return stats;
    }
};

/**
 * Clean loop
 */
setInterval(function() {
		var now = Date.now();
		for(var infoHex in ctxs)
		    if (ctxs.hasOwnProperty(infoHex)) {
			var ctx = ctxs[infoHex];
			if (ctx.streams.length <= 0 &&
			    ctx.lastGet <= now - 30 * 1000) {

			    ctx.close();
			    delete ctxs[infoHex];
			}
		    }
	    }, 10 * 1000);

/**
 * Wire listening stuff
 */
var peerId = new Buffer("-BS00-              ");
for(var i = 6; i < 20; i++)
    peerId[i] = Math.ceil(Math.random * 255);

function checkInfoHash(infoHash) {
    var infoHex = binToHex(infoHash);
    if (ctxs.hasOwnProperty(infoHex))
	return true;
    else {
	console.log("Refusing connection for " + infoHex);
	return false;
    }
}

net.createServer(function(stream) {
		     var wire = new WP.WireAcceptor(stream, checkInfoHash, peerId);
		     wire.on('established', function(infoHash, peerId) {
				 var infoHex = binToHex(infoHash);
				 // presence checked by checkInfoHash()
				 ctxs[infoHex].addIncomingPeer(stream, wire);
			     });
		 }).listen(module.exports.port, '0.0.0.0');


// ✞ Once upon a time this was dead code ✞
function hexChar(v) {
    if (v >= 0 && v <= 9)
        return v + 48;  // '0'
    else if (v >= 10 && v <= 16)
        return v - 10 + 65;  // 'A'
    else
        throw 'hexChar';
}

function binToHex(bin) {
    var r = new Buffer(bin.length * 2);
    for(var i = 0; i < bin.length; i++) {
        var c = bin[i];
        r[i * 2] = hexChar((c & 0xF0) >> 4);
        r[i * 2 + 1] = hexChar(c & 0xF);
    }
    return r.toString();
}