geoip-native/geoip.js

A fast, native JavaScript geoip api. This product includes GeoLite data created by MaxMind, available from http://www.maxmind.com

var countries = [],
	midpoints = [],
	numcountries = 0;

var geoip = module.exports = {

    ready: false,

    lookup: function(ip) {

        if(!geoip.ready) {
            return { error: "GeoIP not ready" };
        }

        var ipl = iplong(ip);

        if(ipl == 0) {
            return { error: "Invalid ip address " + ip + " -> " + ipl + " as integer" };
        }

        return find(ipl);
    }
};

function iplong(ip) {

    if(!ip) {
        return 0;
    }

    ip = ip.toString();

    if(isNaN(ip) && ip.indexOf(".") == -1) {
        return 0;
    }

    if(ip.indexOf(".") == -1) {

        try {
            ip = parseFloat(ip);
            return ip < 0 || ip > 4294967296 ? 0 : ip;
        }
        catch(s) {
        }
    }

    var parts = ip.split(".");

    if(parts.length != 4) {
        return 0;
    }

    var ipl = 0;

    for(var i=0; i<4; i++) {
        parts[i] = parseInt(parts[i], 10);

        if(parts[i] < 0 || parts[i] > 255) {
            return 0;
        }

        ipl += parts[3-i] * (Math.pow(256, i));
    }

    return ipl > 4294967296 ? 0 : ipl;
}

/**
 * A qcuick little binary search
 * @param ip the ip we're looking for
 * @return {*}
 */
function find(ipl) {

    var mpi = 0;
    var n = midpoints[0];
    var step;
    var current;
    var next;
    var prev;
    var nn;
    var pn;
    
    while(true) {

        mpi++;
        step = midpoints[mpi];
        current = countries[n];
        nn = n + 1;
        pn = n - 1;

        next = nn < numcountries ? countries[nn] : null;
        prev = pn > -1 ? countries[pn] : null;
        
		// take another step?
        if(step > 0) {

            if(!next || next.ipstart < ipl) {
                n += step;
            } else {
                n -= step;
            }

            continue;
        }

        // we're either current, next or previous depending on which is closest to ipl
        var cd = Math.abs(ipl - current.ipstart);
        var nd = next && next.ipstart< ipl ? ipl - next.ipstart : 1000000000;
        var pd = prev && prev.ipstart < ipl ? ipl - prev.ipstart : 1000000000;

        // current wins
        if(cd < nd && cd < pd) {
            return current;
        }

         // next wins
        if(nd < cd && nd < pd) {
            return next;

        }

        // prev wins
        return prev;
    }
}

/**
* Prepare the data.  This uses the standard free GeoIP CSV database 
* from MaxMind, you should be able to update it at any time by just
* overwriting GeoIPCountryWhois.csv with a new version.
*/
(function() {

    var fs = require("fs");
    var sys = require("sys");
    var stream = fs.createReadStream(__dirname + "/GeoIPCountryWhois.csv");
    var buffer = "";

    stream.addListener("data", function(data) {
        buffer += data.toString().replace(/"/g, "");
    });

    stream.addListener("end", function() {

        var entries = buffer.split("\n");

        for(var i=0; i<entries.length; i++) {
            var entry = entries[i].split(",");
            countries.push({ipstart: parseInt(entry[2]), code: entry[4], name: entry[5]});
        }

        countries.sort(function(a, b) {
            return a.ipstart - b.ipstart;
        });

        var n = Math.floor(countries.length / 2);
        while(n >= 1) {
            n = Math.floor(n / 2);
            midpoints.push(n);
        }

        numcountries = countries.length;
		geoip.ready = true;
    });

}());