evil-dns.js

var dns = require('dns'),
	net = require('net'),
	dnsLookup = dns.lookup,
	domains = [];

/**
 * Override core DNS lookup function
 **/

dns.lookup = function(domain, options, callback) {
	var i;

	if (arguments.length === 2) {
		callback = options;
		options = {};
	} 

    var family = (typeof(options) === 'object') ? options.family : options;
    if (family) {
		family = +family;
		if (family !== 4 && family !== 6) {
			throw new Error('invalid argument: `family` must be 4 or 6');
		}
	}

	for (i = 0; i < domains.length; i++) {
		var entry = domains[i];
		if (domain.match(entry.domain)) {
			if (!family || family === entry.family) {
				return callback(null, entry.ip, entry.family);
			}			
		}
	}

	return dnsLookup.call(this, domain, options, callback);
};

/**
 * Add a domain to the override list
 *
 * @param domain String or RegExp matching domain
 * @param ip String IPv4 or IPv6
 **/

function add(domain, ip) {
	var entry = { ip: ip };

	if (net.isIPv4(entry.ip)) {
		entry.family = 4;
	}
	else if (net.isIPv6(entry.ip)) {
		entry.family = 6;
	}
	else {
		throw new Error('Invalid ip: ' + entry.ip);
	}

	if (domain instanceof RegExp) {
		entry.source = domain;
		entry.domain = domain;
	}
	else {
		entry.source = domain;
		entry.domain = createRegex(domain);
	}

	domains.push(entry);
}

/**
 * Remove a domain from the override list
 * 
 * @param domain String or RegExp
 * @param ip String optional, if not set all domains equivalent domain will be removed
 **/
function remove(domain, ip) {
	var i;

	for (i = 0; i < domains.length; i++) {
		if (domain instanceof RegExp) {
			if (domains[i].source  instanceof RegExp
				&& domains[i].source.source === domain.source
				&& (!ip || ip === domains[i].ip)) {

				domains.splice(i, 1);
				i--;
			}
		}
		else {
			if (domains[i].source === domain && (!ip || ip === domains[i].ip)) {
				domains.splice(i, 1);
				i--;
			}
		}
		
	}
}

/**
 * Remove all domains from the override list
 **/
function clear() {
	domains = [];
}

function createRegex(val) {
	var parts = val.split('*'),
		i;

	for (i = 0; i < parts.length; i++) {
		parts[i] = regexEscape(parts[i]);
	}

	val = parts.join('.*');
	val = '^' + val + '$';

	return new RegExp(val, 'i');
}


function regexEscape(val) {
	return val.replace(/[-[\]{}()*+?.,\\^$|#\s]/g, "\\$&");
}

module.exports = {
	add: add,
	remove: remove,
	clear: clear,
	domains: domains
};