prosody
615a0774e4cc
tests/test_util_ip.lua

Find changesets by keywords by author, files, the commit message, revision number or hash, or revset expression.

Software / code / prosody

File

tests/test_util_ip.lua @ 5877:615a0774e4cc

util.timer: Updated to use util.indexedbheap to provide a more complete API. Timers can now be stopped or rescheduled. Callbacks are now pcall'd. Adding/removing timers from within timer callbacks works better. Optional parameter can be passed when creating timer which gets passed to callback, eliminating the need for closures in various timer uses. Timers are now much more lightweight.
author Waqas Hussain <waqas20@gmail.com>
date Wed, 30 Oct 2013 17:44:42 -0400
parent 5776:bd0ff8ae98a8
line wrap: on
line source


function match(match, _M)
	local _ = _M.new_ip;
	local ip = _"10.20.30.40";
	assert_equal(match(ip, _"10.0.0.0", 8), true);
	assert_equal(match(ip, _"10.0.0.0", 16), false);
	assert_equal(match(ip, _"10.0.0.0", 24), false);
	assert_equal(match(ip, _"10.0.0.0", 32), false);

	assert_equal(match(ip, _"10.20.0.0", 8), true);
	assert_equal(match(ip, _"10.20.0.0", 16), true);
	assert_equal(match(ip, _"10.20.0.0", 24), false);
	assert_equal(match(ip, _"10.20.0.0", 32), false);

	assert_equal(match(ip, _"0.0.0.0", 32), false);
	assert_equal(match(ip, _"0.0.0.0", 0), true);
	assert_equal(match(ip, _"0.0.0.0"), false);

	assert_equal(match(ip, _"10.0.0.0", 255), false, "excessive number of bits");
	assert_equal(match(ip, _"10.0.0.0", -8), true, "negative number of bits");
	assert_equal(match(ip, _"10.0.0.0", -32), true, "negative number of bits");
	assert_equal(match(ip, _"10.0.0.0", 0), true, "zero bits");
	assert_equal(match(ip, _"10.0.0.0"), false, "no specified number of bits (differing ip)");
	assert_equal(match(ip, _"10.20.30.40"), true, "no specified number of bits (same ip)");

	assert_equal(match(_"127.0.0.1", _"127.0.0.1"), true, "simple ip");

	assert_equal(match(_"8.8.8.8", _"8.8.0.0", 16), true);
	assert_equal(match(_"8.8.4.4", _"8.8.0.0", 16), true);
end

function parse_cidr(parse_cidr, _M)
	local new_ip = _M.new_ip;

	assert_equal(new_ip"0.0.0.0", new_ip"0.0.0.0")

	local function assert_cidr(cidr, ip, bits)
		local parsed_ip, parsed_bits = parse_cidr(cidr);
		assert_equal(new_ip(ip), parsed_ip, cidr.." parsed ip is "..ip);
		assert_equal(bits, parsed_bits, cidr.." parsed bits is "..tostring(bits));
	end
	assert_cidr("0.0.0.0", "0.0.0.0", nil);
	assert_cidr("127.0.0.1", "127.0.0.1", nil);
	assert_cidr("127.0.0.1/0", "127.0.0.1", 0);
	assert_cidr("127.0.0.1/8", "127.0.0.1", 8);
	assert_cidr("127.0.0.1/32", "127.0.0.1", 32);
	assert_cidr("127.0.0.1/256", "127.0.0.1", 256);
	assert_cidr("::/48", "::", 48);
end

function new_ip(new_ip)
	local v4, v6 = "IPv4", "IPv6";
	local function assert_proto(s, proto)
		local ip = new_ip(s);
		if proto then
			assert_equal(ip and ip.proto, proto, "protocol is correct for "..("%q"):format(s));
		else
			assert_equal(ip, nil, "address is invalid");
		end
	end
	assert_proto("127.0.0.1", v4);
	assert_proto("::1", v6);
	assert_proto("", nil);
	assert_proto("abc", nil);
	assert_proto("   ", nil);
end

function commonPrefixLength(cpl, _M)
	local new_ip = _M.new_ip;
	local function assert_cpl6(a, b, len, v4)
		local ipa, ipb = new_ip(a), new_ip(b);
		if v4 then len = len+96; end
		assert_equal(cpl(ipa, ipb), len, "common prefix length of "..a.." and "..b.." is "..len);
		assert_equal(cpl(ipb, ipa), len, "common prefix length of "..b.." and "..a.." is "..len);
	end
	local function assert_cpl4(a, b, len)
		return assert_cpl6(a, b, len, "IPv4");
	end
	assert_cpl4("0.0.0.0", "0.0.0.0", 32);
	assert_cpl4("255.255.255.255", "0.0.0.0", 0);
	assert_cpl4("255.255.255.255", "255.255.0.0", 16);
	assert_cpl4("255.255.255.255", "255.255.255.255", 32);
	assert_cpl4("255.255.255.255", "255.255.255.255", 32);

	assert_cpl6("::1", "::1", 128);
	assert_cpl6("abcd::1", "abcd::1", 128);
	assert_cpl6("abcd::abcd", "abcd::", 112);
	assert_cpl6("abcd::abcd", "abcd::abcd:abcd", 96);
end