z-nomp/init.js

var fs = require('fs');
var os = require('os');
var cluster = require('cluster');


var posix = require('posix');
var Stratum = require('stratum-pool');
var PoolLogger = require('./libs/logutils.js');
var BlocknotifyListener = require('./libs/blocknotifyListener.js');

JSON.minify = JSON.minify || require("node-json-minify");



//Try to give process ability to handle 100k concurrent connections
try{
    posix.setrlimit('nofile', { soft: 100000, hard: 100000 });
}
catch(e){
    console.error(e);
}



var loggerInstance = new PoolLogger({
    'default': true,
    'keys': {
        //'client'      : 'warning',
        'system'      : true,
        'submitblock' : true
    }
});

var logDebug   = loggerInstance.logDebug;
var logWarning = loggerInstance.logWarning;
var logError   = loggerInstance.logError;



if (cluster.isMaster){


    var config = JSON.parse(JSON.minify(fs.readFileSync("config.json", {encoding: 'utf8'})));

    //Read all coin profile json files from coins directory and build object where key is name of coin
    var coinProfiles = (function(){
        var profiles = {};
        fs.readdirSync('coins').forEach(function(file){
            var coinProfile = JSON.parse(JSON.minify(fs.readFileSync('coins/' + file, {encoding: 'utf8'})));
            profiles[coinProfile.name.toLowerCase()] = coinProfile;
        });
        return profiles;
    })();


    //Read all pool configs from pool_configs and join them with their coin profile
    var poolConfigs = (function(){
        var configs = [];
        fs.readdirSync('pool_configs').forEach(function(file){
            var poolOptions = JSON.parse(JSON.minify(fs.readFileSync('pool_configs/' + file, {encoding: 'utf8'})));
            if (poolOptions.disabled) return;
            if (!(poolOptions.coin.toLowerCase() in coinProfiles)){
                logError(poolOptions.coin, 'system', 'could not find coin profile');
                return;
            }
            poolOptions.coin = coinProfiles[poolOptions.coin.toLowerCase()];
            configs.push(poolOptions);
        });
        return configs;
    })();


    var serializedConfigs = JSON.stringify(poolConfigs);


    var numForks = (function(){
        if (!config.clustering || !config.clustering.enabled)
            return 1;
        if (config.clustering.forks === 'auto')
            return os.cpus().length;
        if (!config.clustering.forks || isNaN(config.clustering.forks))
            return 1;
        return config.clustering.forks;
    })();

    for (var i = 0; i < numForks; i++) {
        cluster.fork({
            fork: i,
            pools: serializedConfigs
        });
    }

    cluster.on('exit', function(worker, code, signal) {
        console.log('worker fork with PID ' + worker.process.pid + ' died');
    });


    //block notify options
    //setup block notify here and use IPC to tell appropriate pools
    var listener = new BlocknotifyListener(config.blockNotifyListener);
    listener.on('log', function(text){
        logDebug('blocknotify', 'system', text);
    });
    listener.on('hash', function(message){

        var serializedMessage = JSON.stringify({'blocknotify': message.hash});
        Object.keys(cluster.workers).forEach(function(id) {
            cluster.workers[id].send(serializedMessage);
        });

    });
    listener.start();

}

else{

    var poolConfigs = JSON.parse(process.env.pools);
    var fork = process.env.fork;

    var pools = [];

    //Handle blocknotify message from master process sent via IPC
    process.on('message', function(msg) {
        var message = JSON.parse(msg);
        if (message.blocknotify){
            for (var i = 0; i < pools.length; i++){
                if (pools[i].options.coin.name.toLowerCase() === message.coin.toLowerCase()){
                    pools[i].processBlockNotify(message.blockHash)
                    return;
                }
            }
        }
    });




    poolConfigs.forEach(function(poolOptions){

        var logIdentify = poolOptions.coin.name + ' (Fork ' + fork + ')';

        var authorizeFN = function (ip, workerName, password, callback) {
            // Default implementation just returns true
            logDebug(logIdentify, 'client', "Authorize [" + ip + "] " + workerName + ":" + password);
            callback({
                error: null,
                authorized: true,
                disconnect: false
            });
        };


        var pool = Stratum.createPool(poolOptions, authorizeFN);
        pool.on('share', function(isValidShare, isValidBlock, data){

            var shareData = JSON.stringify(data);

            if (data.solution && !isValidBlock)
                logDebug(logIdentify, 'client', 'We thought a block solution was found but it was rejected by the daemon, share data: ' + shareData);
            else if (isValidBlock)
                logDebug(logIdentify, 'client', 'Block found, share data: ' + shareData);
            else if (isValidShare)
                logDebug(logIdentify, 'client', 'Valid share submitted, share data: ' + shareData);
            else
                logDebug(logIdentify, 'client', 'Invalid share submitted, share data: ' + shareData)


        }).on('log', function(severity, logKey, logText) {
                if (severity == 'debug') {
                    logDebug(logIdentify, logKey, logText);
                } else if (severity == 'warning') {
                    logWarning(logIdentify, logKey, logText);
                } else if (severity == 'error') {
                    logError(logIdentify, logKey, logText);
                }
            });
        pool.start();
        pools.push(pool);
    });
}