./cibn/ 0000755 0000000 0000000 00000000000 12226741277 010636 5 ustar root root ./cibn/tool.lua 0000644 0000000 0000000 00000003717 12224673535 012325 0 ustar root root
do
local function splittail(tsrc, tsubstr, tlist)
local t_strat = 0;
local t_end = 0;
local tail = '';
local first = '';
t_start, t_end = string.find(tsrc, tsubstr);
first = string.sub(tsrc, 0, t_start - 1);
tail = string.sub(tsrc, t_end + 1);
table.insert(tlist, first);
t_start = nil;
t_end = nil;
t_start, t_end = string.find(tail, tsubstr);
if t_start ~= nil then
splittail(tail, tsubstr, tlist);
else
table.insert(tlist, tail);
return tlist;
end
end
local function split(str, substr)
local pos_start = nil;
local pos_end = nil;
local list = {};
local tail = '';
pos_start, pos_end = string.find(str, substr);
if pos_start == nil then
return nil;
else
first = string.sub(str, 0, pos_start - 1);
table.insert(list, first);
tail = string.sub(str, tonumber(pos_end) + 1);
pos_start = nil;
pos_end = nil;
pos_start, pos_end = string.find(tail, substr);
if pos_start == nil then
table.insert(list, tail);
return list;
else
splittail(tail, substr, list);
return list;
end
end
end
local function m8bin(x)
result = '';
while x > 0
do
mod = x % 2;
x, y = math.modf(x / 2);
result = tostring(mod) .. result;
end
prepos = (8 - string.len(result));
while prepos > 0
do
result = '0' .. result;
prepos = prepos - 1;
end
return result;
end
local function ip2bin(ip)
slist = split(ip, '%.');
if slist ~= nil then
ip1 = m8bin(tonumber(slist[1]));
ip2 = m8bin(tonumber(slist[2]));
ip3 = m8bin(tonumber(slist[3]));
ip4 = m8bin(tonumber(slist[4]));
res = ip1 .. ip2 .. ip3 .. ip4;
return res;
else
return nil;
end
end
tool = {
splittail = splittail,
split = split,
m8bin = m8bin,
ip2bin = ip2bin,
}
end
./cibn/dictionary.lua 0000644 0000000 0000000 00000001111 12222163554 013470 0 ustar root root do
local function dict()
local self = {length = 0, dd = {}};
local set = function(k, v)
self.dd[k] = v;
self.length = self.length + 1;
end
local del = function(k)
self.dd[k] = nil;
if self.length > 0 then
self.length = self.length - 1;
end
end
local get = function(k)
return self.dd[k];
end
local len = function()
return self.length;
end
return
{
set = set,
del = del,
get = get,
len = len
}
end
dictionary = {
dict = dict,
}
end
./cibn/ring.lua 0000644 0000000 0000000 00000004463 12226657475 012315 0 ustar root root require 'luabit.bit';
require 'luabit.hex';
do
local function shift(s)
s1 = tonumber((string.sub(s, 1, 2)), 16);
s2 = bit.blshift(tonumber((string.sub(s, 3, 4)), 16), 8);
s3 = bit.blshift(tonumber((string.sub(s, 5, 6)), 16), 16);
s4 = bit.blshift(tonumber((string.sub(s, 7, 8)), 16), 24);
sa = bit.bor(bit.bor(bit.bor(s1, s2), s3), s4);
return sa;
end
local function bitwise(m)
s1 = shift(string.sub(m, 1, 8));
s2 = shift(string.sub(m, 9, 16));
s3 = shift(string.sub(m, 17, 24));
s4 = shift(string.sub(m, 25, 32));
sa = s1 + s2 + s3 + s4;
return sa;
end
local function str2point(str)
md5 = ngx.md5(str);
point = bitwise(md5);
return point;
end
-- sps = aring
-- sp = point
local function search(sp, sps, pstart, pend, n)
n = n + 1;
local mid, mod = math.modf((pstart + pend) / 2);
if n > 20 then
ngx.log(ngx.ERR, 'the point ' .. sp .. ' cycle index exceed 20');
ngx.exit(509);
end
if sps[mid] >= sp and sps[mid - 1] < sp then
return sps[mid];
elseif sps[mid + 1] >= sp and sps[mid] < sp then
return sps[mid + 1];
elseif sps[mid] > sp then
pend = mid;
return search(sp, sps, pstart, pend, n);
elseif sps[mid] < sp then
pstart = mid;
return search(sp, sps, pstart, pend, n);
end
end
-- fps = points
-- frs = rings
-- fnode = node name
local function findhost(fp, fps, ffls, fsgls, fls, frs, fnode)
local fflag = ffls.get(fnode);
if fflag == 's' then
--ngx.say('single
');
return fsgls.get(fnode);
elseif fflag == 'h' then
local ps = fps.get(fnode);
local fend = table.getn(ps);
if fp <= ps[1] or ps[fend] < fp then
return frs.get(fnode).get(ps[1]);
else
tp = search(fp, ps, 1, fend, 0);
if tp ~= nil then
return frs.get(fnode).get(tp);
else
ngx.log(ngx.ERR, 'search ring error ' .. fp .. '
');
end
end
elseif fflag == 'l' then
floops = fls.get(fnode);
total = table.getn(floops);
math.randomseed(os.time());
rp = math.random(1, total);
return floops[rp];
end
end
ring = {
shift = shift,
bitwise = bitwise,
str2point = str2point,
findhost = findhost,
}
end
./cibn/base.lua 0000644 0000000 0000000 00000021362 12226734705 012255 0 ustar root root require 'cibn.tool';
require 'cibn.dictionary';
do
local function readconts(filepath)
local contents = {};
for ipline in io.lines(filepath)
do
if ipline ~= nil then
contents = {next = contents , value = ipline};
else
break;
end
end
return contents;
end
local function init_ip2zone(conts)
local tzone = dictionary.dict();
while conts
do
if conts.value == nil then
break;
end
iplist = tool.split(tostring(conts.value), ';');
if iplist ~= nil then
ipzone = iplist[1];
area = string.sub(iplist[2], 2);
ips = tool.split(tostring(ipzone), '/');
if ips ~= nil then
ip = ips[1];
mask = ips[2];
bin = tool.ip2bin(ip);
if bin ~= nil then
binkey = string.sub(bin, 1, tonumber(mask));
tzone.set(binkey, area);
end
else
ngx.say('ipzone = ' .. ipzone .. '
');
break;
end
else
ngx.log(ngx.ERR, 'the config of ip is error!');
os.exit(1);
end
if conts.next.value ~= nil then
conts = conts.next;
else
break;
end
end
return tzone;
end
local function search_zone(dd, bip)
local lip = bip;
local area = dd.get(lip);
if area then
return area;
else
nextlen = #lip - 1;
if nextlen >= 1 then
lip = string.sub(lip, 1, nextlen);
return search_zone(dd, lip);
else
return 'default';
end
end
end
local function init_isp2node(ispconts)
local tnode = dictionary.dict();
local tflag = dictionary.dict();
while ispconts
do
if ispconts.value == nil then
break;
end
ins = tool.split(tostring(ispconts.value) , ':');
if ins ~= nil then
isp = ins[1];
istart, iend = string.find(isp, '#');
if istart ~= nil then
isp = string.sub(isp, 1, istart - 1);
flag = string.upper(string.sub(isp, iend));
if flag == nil then
flag = 'P';
end
tflag.set(isp, flag);
else
flag = 'P';
tflag.set(isp, flag);
end
nodelist = ins[2];
if string.sub(nodelist, -1) == ';' then
nodelist = string.sub(nodelist, 1, -2);
end
tnode.set(isp, nodelist);
else
ngx.log(ngx.ERR, 'the config of ip is error!');
os.exit(1);
end
if ispconts.next.value ~= nil then
ispconts = ispconts.next;
else
break;
end
end
return tflag, tnode;
end
local function search_percent_node(ds, tisp)
local nodes = ds.get(tisp);
local nnodelist = {};
local nn = dictionary.dict();
local tp = 0;
if nodes ~= nil then
nstart, nend = string.find(nodes, ';');
if nstart ~= nil then
nlist = tool.split(nodes, ';');
for i = 1, table.getn(nlist)
do
nnstart, nnend = string.find(nlist[i], '#');
if nnstart ~= nil then
nnode = string.sub(nlist[i], 1, nnstart - 1);
npercent = tonumber(string.sub(nlist[i], nnend + 1));
if nnode == nil then
ngx.log(ngx.ERR, 'the config file of isp2node is error. at ' .. nlist[i]);
ngx.exit(509);
elseif npercent == nil then
npercent = 1;
end
else
nnode = nlist[i];
npercent = 1;
end
pzone = tostring(tp) .. '-' .. tostring(tp + npercent);
nn.set(nnode, pzone);
tp = tp + npercent;
table.insert(nnodelist, nnode);
end
--ngx.say('tp = ' .. tp .. '
');
math.randomseed(os.time());
rp = math.random(1, tp);
--ngx.say('rp = ' .. rp .. '
');
for i = 1, table.getn(nnodelist)
do
pz = nn.get(nnodelist[i]);
ps = tool.split(pz, '-');
if tonumber(ps[1]) < rp and rp <= tonumber(ps[2]) then
node = nnodelist[i];
break;
end
end
return node;
else
nnstart, nnend = string.find(nodes, '#');
if nnstart ~= nil then
node = string.sub(nodes, nnstart - 1);
else
node = nodes;
end
end
return node;
else
return 'default';
end
end
local function init_slist(slconts)
local sloop = dictionary.dict();
local single = dictionary.dict();
local sflag = dictionary.dict();
local spoint = dictionary.dict();
local sring = dictionary.dict();
while slconts
do
if slconts.value == nil then
break;
end
local aring = dictionary.dict();
local apoint = {};
servers = tool.split(tostring(slconts.value) , '#');
if servers ~= nil then
node = servers[1];
sslist = tostring(servers[2]);
if string.sub(sslist, -1) == ';' then
sslist = string.sub(sslist, 1, #sslist - 1);
end
f_start, f_end = string.find(sslist, ':');
if f_start ~= nil then
ss = tool.split(sslist, ':');
if ss[1] == nil then
ss[1] = 'l';
end
if ss[2] == nil then
ngx.log(ngx.ERR, 'the config of serverlist is error! at ' .. sslist);
os.exit(1);
end
flag = ss[1];
sslist = ss[2];
sflag.set(node, flag);
else
ngx.log(ngx.ERR, 'the config of serverlist is error! search servers without flag.');
os.exit(1);
end
t_start, t_end = string.find(sslist, ';');
if t_start ~= nil then
if flag == 'h' then
hosts = tool.split(sslist, ';');
for k,v in ipairs(hosts)
do
for i = 1, 500
do
seed = string.format('%s-%03d', v, i);
point = ring.str2point(seed);
aring.set(point, v);
table.insert(apoint, point);
end
end
table.sort(apoint);
sring.set(node, aring);
spoint.set(node, apoint);
elseif flag == 'l' then
hosts = tool.split(sslist, ';');
sloop.set(node, hosts);
end
else
flag = 's';
sflag.set(node, flag);
single.set(node, sslist);
end
end
if slconts.next.value ~= nil then
slconts = slconts.next;
else
break;
end
end
return sflag, spoint, sring, single, sloop;
end
local function init_spec(conts)
local domain2node = dictionary.dict();
while conts
do
if conts.value == nil then
break;
end
tstart, tend = string.find(conts.value, ':');
if tstart == nil then
ngx.log(ngx.ERR, 'the config file of specdomain is error. at ' .. conts.value);
os.exit(1);
end
ninfo = tool.split(conts.value, ':');
if ninfo ~= nil then
tnode = ninfo[1];
dstart, dend = string.find(ninfo[2], '#');
if dstart == nil then
ngx.log(ngx.ERR, 'the config file of specdomain is error. at ' .. ninfo[2]);
os.exit(1);
end
d2n = tool.split(ninfo[2], '#');
domain2node.set(string.lower(d2n[1]), string.lower(d2n[2]));
end
if conts.next.value ~= nil then
conts = conts.next;
else
break;
end
end
if domain2node.get('default') == nil then
ngx.log(ngx.ERR, 'the config file of specdomain is error. the default domain must be set.');
os.exit(1);
end
return tnode, domain2node;
end
base = {
readconts = readconts,
init_ip2zone = init_ip2zone,
search_zone = search_zone,
init_isp2node = init_isp2node,
search_percent_node = search_percent_node,
init_slist = init_slist,
init_spec = init_spec,
}
end