mirror of
https://github.com/bjc/prosody.git
synced 2025-04-03 21:27:38 +03:00
util.*: Remove use of module() function, make all module functions local and return them in a table at the end
This commit is contained in:
Kim Alvefur
parent
b49513cdeb
commit
eaa823a597
35 changed files with 435 additions and 302 deletions
|
|
@ -169,7 +169,4 @@ for method, f in pairs(array_base) do
|
|||
end
|
||||
end
|
||||
|
||||
_G.array = array;
|
||||
module("array");
|
||||
|
||||
return array;
|
||||
|
|
|
|||
|
|
@ -12,9 +12,9 @@ local sha1 = require "util.hashes".sha1;
|
|||
local t_insert, t_sort, t_concat = table.insert, table.sort, table.concat;
|
||||
local ipairs = ipairs;
|
||||
|
||||
module "caps"
|
||||
local _ENV = nil;
|
||||
|
||||
function calculate_hash(disco_info)
|
||||
local function calculate_hash(disco_info)
|
||||
local identities, features, extensions = {}, {}, {};
|
||||
for _, tag in ipairs(disco_info) do
|
||||
if tag.name == "identity" then
|
||||
|
|
@ -58,4 +58,6 @@ function calculate_hash(disco_info)
|
|||
return ver, S;
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
calculate_hash = calculate_hash;
|
||||
};
|
||||
|
|
|
|||
|
|
@ -13,14 +13,14 @@ local t_concat = table.concat;
|
|||
local st = require "util.stanza";
|
||||
local jid_prep = require "util.jid".prep;
|
||||
|
||||
module "dataforms"
|
||||
local _ENV = nil;
|
||||
|
||||
local xmlns_forms = 'jabber:x:data';
|
||||
|
||||
local form_t = {};
|
||||
local form_mt = { __index = form_t };
|
||||
|
||||
function new(layout)
|
||||
local function new(layout)
|
||||
return setmetatable(layout, form_mt);
|
||||
end
|
||||
|
||||
|
|
@ -238,7 +238,9 @@ field_readers["hidden"] =
|
|||
return field_tag:get_child_text("value");
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
new = new;
|
||||
};
|
||||
|
||||
|
||||
--[=[
|
||||
|
|
|
|||
|
|
@ -43,7 +43,7 @@ pcall(function()
|
|||
fallocate = pposix.fallocate or fallocate;
|
||||
end);
|
||||
|
||||
module "datamanager"
|
||||
local _ENV = nil;
|
||||
|
||||
---- utils -----
|
||||
local encode, decode;
|
||||
|
|
@ -74,7 +74,7 @@ local callbacks = {};
|
|||
|
||||
------- API -------------
|
||||
|
||||
function set_data_path(path)
|
||||
local function set_data_path(path)
|
||||
log("debug", "Setting data path to: %s", path);
|
||||
data_path = path;
|
||||
end
|
||||
|
|
@ -87,14 +87,14 @@ local function callback(username, host, datastore, data)
|
|||
|
||||
return username, host, datastore, data;
|
||||
end
|
||||
function add_callback(func)
|
||||
local function add_callback(func)
|
||||
if not callbacks[func] then -- Would you really want to set the same callback more than once?
|
||||
callbacks[func] = true;
|
||||
callbacks[#callbacks+1] = func;
|
||||
return true;
|
||||
end
|
||||
end
|
||||
function remove_callback(func)
|
||||
local function remove_callback(func)
|
||||
if callbacks[func] then
|
||||
for i, f in ipairs(callbacks) do
|
||||
if f == func then
|
||||
|
|
@ -106,7 +106,7 @@ function remove_callback(func)
|
|||
end
|
||||
end
|
||||
|
||||
function getpath(username, host, datastore, ext, create)
|
||||
local function getpath(username, host, datastore, ext, create)
|
||||
ext = ext or "dat";
|
||||
host = (host and encode(host)) or "_global";
|
||||
username = username and encode(username);
|
||||
|
|
@ -119,7 +119,7 @@ function getpath(username, host, datastore, ext, create)
|
|||
end
|
||||
end
|
||||
|
||||
function load(username, host, datastore)
|
||||
local function load(username, host, datastore)
|
||||
local data, ret = envloadfile(getpath(username, host, datastore), {});
|
||||
if not data then
|
||||
local mode = lfs.attributes(getpath(username, host, datastore), "mode");
|
||||
|
|
@ -175,7 +175,7 @@ if prosody and prosody.platform ~= "posix" then
|
|||
end
|
||||
end
|
||||
|
||||
function store(username, host, datastore, data)
|
||||
local function store(username, host, datastore, data)
|
||||
if not data then
|
||||
data = {};
|
||||
end
|
||||
|
|
@ -209,7 +209,7 @@ function store(username, host, datastore, data)
|
|||
return true;
|
||||
end
|
||||
|
||||
function list_append(username, host, datastore, data)
|
||||
local function list_append(username, host, datastore, data)
|
||||
if not data then return; end
|
||||
if callback(username, host, datastore) == false then return true; end
|
||||
-- save the datastore
|
||||
|
|
@ -235,7 +235,7 @@ function list_append(username, host, datastore, data)
|
|||
return true;
|
||||
end
|
||||
|
||||
function list_store(username, host, datastore, data)
|
||||
local function list_store(username, host, datastore, data)
|
||||
if not data then
|
||||
data = {};
|
||||
end
|
||||
|
|
@ -259,7 +259,7 @@ function list_store(username, host, datastore, data)
|
|||
return true;
|
||||
end
|
||||
|
||||
function list_load(username, host, datastore)
|
||||
local function list_load(username, host, datastore)
|
||||
local items = {};
|
||||
local data, ret = envloadfile(getpath(username, host, datastore, "list"), {item = function(i) t_insert(items, i); end});
|
||||
if not data then
|
||||
|
|
@ -287,7 +287,7 @@ local type_map = {
|
|||
list = "list";
|
||||
}
|
||||
|
||||
function users(host, store, typ)
|
||||
local function users(host, store, typ)
|
||||
typ = type_map[typ or "keyval"];
|
||||
local store_dir = format("%s/%s/%s", data_path, encode(host), store);
|
||||
|
||||
|
|
@ -306,7 +306,7 @@ function users(host, store, typ)
|
|||
end, state;
|
||||
end
|
||||
|
||||
function stores(username, host, typ)
|
||||
local function stores(username, host, typ)
|
||||
typ = type_map[typ or "keyval"];
|
||||
local store_dir = format("%s/%s/", data_path, encode(host));
|
||||
|
||||
|
|
@ -346,7 +346,7 @@ local function do_remove(path)
|
|||
return true
|
||||
end
|
||||
|
||||
function purge(username, host)
|
||||
local function purge(username, host)
|
||||
local host_dir = format("%s/%s/", data_path, encode(host));
|
||||
local ok, iter, state, var = pcall(lfs.dir, host_dir);
|
||||
if not ok then
|
||||
|
|
@ -366,6 +366,19 @@ function purge(username, host)
|
|||
return #errs == 0, t_concat(errs, ", ");
|
||||
end
|
||||
|
||||
_M.path_decode = decode;
|
||||
_M.path_encode = encode;
|
||||
return _M;
|
||||
return {
|
||||
set_data_path = set_data_path;
|
||||
add_callback = add_callback;
|
||||
remove_callback = remove_callback;
|
||||
getpath = getpath;
|
||||
load = load;
|
||||
store = store;
|
||||
list_append = list_append;
|
||||
list_store = list_store;
|
||||
list_load = list_load;
|
||||
users = users;
|
||||
stores = stores;
|
||||
purge = purge;
|
||||
path_decode = decode;
|
||||
path_encode = encode;
|
||||
};
|
||||
|
|
|
|||
|
|
@ -15,25 +15,25 @@ local os_difftime = os.difftime;
|
|||
local error = error;
|
||||
local tonumber = tonumber;
|
||||
|
||||
module "datetime"
|
||||
local _ENV = nil;
|
||||
|
||||
function date(t)
|
||||
local function date(t)
|
||||
return os_date("!%Y-%m-%d", t);
|
||||
end
|
||||
|
||||
function datetime(t)
|
||||
local function datetime(t)
|
||||
return os_date("!%Y-%m-%dT%H:%M:%SZ", t);
|
||||
end
|
||||
|
||||
function time(t)
|
||||
local function time(t)
|
||||
return os_date("!%H:%M:%S", t);
|
||||
end
|
||||
|
||||
function legacy(t)
|
||||
local function legacy(t)
|
||||
return os_date("!%Y%m%dT%H:%M:%S", t);
|
||||
end
|
||||
|
||||
function parse(s)
|
||||
local function parse(s)
|
||||
if s then
|
||||
local year, month, day, hour, min, sec, tzd;
|
||||
year, month, day, hour, min, sec, tzd = s:match("^(%d%d%d%d)%-?(%d%d)%-?(%d%d)T(%d%d):(%d%d):(%d%d)%.?%d*([Z+%-]?.*)$");
|
||||
|
|
@ -54,4 +54,10 @@ function parse(s)
|
|||
end
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
date = date;
|
||||
datetime = datetime;
|
||||
time = time;
|
||||
legacy = legacy;
|
||||
parse = parse;
|
||||
};
|
||||
|
|
|
|||
|
|
@ -13,7 +13,7 @@ local termcolours = require "util.termcolours";
|
|||
local getstring = termcolours.getstring;
|
||||
local styles;
|
||||
do
|
||||
_ = termcolours.getstyle;
|
||||
local _ = termcolours.getstyle;
|
||||
styles = {
|
||||
boundary_padding = _("bright");
|
||||
filename = _("bright", "blue");
|
||||
|
|
@ -22,9 +22,8 @@ do
|
|||
location = _("yellow");
|
||||
};
|
||||
end
|
||||
module("debugx", package.seeall);
|
||||
|
||||
function get_locals_table(thread, level)
|
||||
local function get_locals_table(thread, level)
|
||||
local locals = {};
|
||||
for local_num = 1, math.huge do
|
||||
local name, value;
|
||||
|
|
@ -39,7 +38,7 @@ function get_locals_table(thread, level)
|
|||
return locals;
|
||||
end
|
||||
|
||||
function get_upvalues_table(func)
|
||||
local function get_upvalues_table(func)
|
||||
local upvalues = {};
|
||||
if func then
|
||||
for upvalue_num = 1, math.huge do
|
||||
|
|
@ -51,7 +50,7 @@ function get_upvalues_table(func)
|
|||
return upvalues;
|
||||
end
|
||||
|
||||
function string_from_var_table(var_table, max_line_len, indent_str)
|
||||
local function string_from_var_table(var_table, max_line_len, indent_str)
|
||||
local var_string = {};
|
||||
local col_pos = 0;
|
||||
max_line_len = max_line_len or math.huge;
|
||||
|
|
@ -87,7 +86,7 @@ function string_from_var_table(var_table, max_line_len, indent_str)
|
|||
end
|
||||
end
|
||||
|
||||
function get_traceback_table(thread, start_level)
|
||||
local function get_traceback_table(thread, start_level)
|
||||
local levels = {};
|
||||
for level = start_level, math.huge do
|
||||
local info;
|
||||
|
|
@ -108,20 +107,12 @@ function get_traceback_table(thread, start_level)
|
|||
return levels;
|
||||
end
|
||||
|
||||
function traceback(...)
|
||||
local ok, ret = pcall(_traceback, ...);
|
||||
if not ok then
|
||||
return "Error in error handling: "..ret;
|
||||
end
|
||||
return ret;
|
||||
end
|
||||
|
||||
local function build_source_boundary_marker(last_source_desc)
|
||||
local padding = string.rep("-", math.floor(((optimal_line_length - 6) - #last_source_desc)/2));
|
||||
return getstring(styles.boundary_padding, "v"..padding).." "..getstring(styles.filename, last_source_desc).." "..getstring(styles.boundary_padding, padding..(#last_source_desc%2==0 and "-v" or "v "));
|
||||
end
|
||||
|
||||
function _traceback(thread, message, level)
|
||||
local function _traceback(thread, message, level)
|
||||
|
||||
-- Lua manual says: debug.traceback ([thread,] [message [, level]])
|
||||
-- I fathom this to mean one of:
|
||||
|
|
@ -192,8 +183,23 @@ function _traceback(thread, message, level)
|
|||
return message.."stack traceback:\n"..table.concat(lines, "\n");
|
||||
end
|
||||
|
||||
function use()
|
||||
local function traceback(...)
|
||||
local ok, ret = pcall(_traceback, ...);
|
||||
if not ok then
|
||||
return "Error in error handling: "..ret;
|
||||
end
|
||||
return ret;
|
||||
end
|
||||
|
||||
local function use()
|
||||
debug.traceback = traceback;
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
get_locals_table = get_locals_table;
|
||||
get_upvalues_table = get_upvalues_table;
|
||||
string_from_var_table = string_from_var_table;
|
||||
get_traceback_table = get_traceback_table;
|
||||
traceback = traceback;
|
||||
use = use;
|
||||
};
|
||||
|
|
|
|||
|
|
@ -6,16 +6,14 @@
|
|||
-- COPYING file in the source package for more information.
|
||||
--
|
||||
|
||||
module("dependencies", package.seeall)
|
||||
|
||||
function softreq(...) local ok, lib = pcall(require, ...); if ok then return lib; else return nil, lib; end end
|
||||
local function softreq(...) local ok, lib = pcall(require, ...); if ok then return lib; else return nil, lib; end end
|
||||
|
||||
-- Required to be able to find packages installed with luarocks
|
||||
if not softreq "luarocks.loader" then -- LuaRocks 2.x
|
||||
softreq "luarocks.require"; -- LuaRocks <1.x
|
||||
end
|
||||
|
||||
function missingdep(name, sources, msg)
|
||||
local function missingdep(name, sources, msg)
|
||||
print("");
|
||||
print("**************************");
|
||||
print("Prosody was unable to find "..tostring(name));
|
||||
|
|
@ -48,7 +46,7 @@ package.preload["util.ztact"] = function ()
|
|||
end
|
||||
end;
|
||||
|
||||
function check_dependencies()
|
||||
local function check_dependencies()
|
||||
if _VERSION ~= "Lua 5.1" then
|
||||
print "***********************************"
|
||||
print("Unsupported Lua version: ".._VERSION);
|
||||
|
|
@ -137,13 +135,15 @@ function check_dependencies()
|
|||
return not fatal;
|
||||
end
|
||||
|
||||
function log_warnings()
|
||||
local function log_warnings()
|
||||
local ssl = softreq"ssl";
|
||||
if ssl then
|
||||
local major, minor, veryminor, patched = ssl._VERSION:match("(%d+)%.(%d+)%.?(%d*)(M?)");
|
||||
if not major or ((tonumber(major) == 0 and (tonumber(minor) or 0) <= 3 and (tonumber(veryminor) or 0) <= 2) and patched ~= "M") then
|
||||
log("error", "This version of LuaSec contains a known bug that causes disconnects, see http://prosody.im/doc/depends");
|
||||
end
|
||||
end
|
||||
local lxp = softreq"lxp";
|
||||
if lxp then
|
||||
if not pcall(lxp.new, { StartDoctypeDecl = false }) then
|
||||
log("error", "The version of LuaExpat on your system leaves Prosody "
|
||||
|
|
@ -162,4 +162,9 @@ function log_warnings()
|
|||
end
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
softreq = softreq;
|
||||
missingdep = missingdep;
|
||||
check_dependencies = check_dependencies;
|
||||
log_warnings = log_warnings;
|
||||
};
|
||||
|
|
|
|||
|
|
@ -14,9 +14,9 @@ local t_sort = table.sort;
|
|||
local setmetatable = setmetatable;
|
||||
local next = next;
|
||||
|
||||
module "events"
|
||||
local _ENV = nil;
|
||||
|
||||
function new()
|
||||
local function new()
|
||||
local handlers = {};
|
||||
local global_wrappers;
|
||||
local wrappers = {};
|
||||
|
|
@ -151,4 +151,6 @@ function new()
|
|||
};
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
new = new;
|
||||
};
|
||||
|
|
|
|||
|
|
@ -8,11 +8,11 @@
|
|||
|
||||
local t_insert, t_remove = table.insert, table.remove;
|
||||
|
||||
module "filters"
|
||||
local _ENV = nil;
|
||||
|
||||
local new_filter_hooks = {};
|
||||
|
||||
function initialize(session)
|
||||
local function initialize(session)
|
||||
if not session.filters then
|
||||
local filters = {};
|
||||
session.filters = filters;
|
||||
|
|
@ -36,7 +36,7 @@ function initialize(session)
|
|||
return session.filter;
|
||||
end
|
||||
|
||||
function add_filter(session, type, callback, priority)
|
||||
local function add_filter(session, type, callback, priority)
|
||||
if not session.filters then
|
||||
initialize(session);
|
||||
end
|
||||
|
|
@ -60,7 +60,7 @@ function add_filter(session, type, callback, priority)
|
|||
filter_list[callback] = priority;
|
||||
end
|
||||
|
||||
function remove_filter(session, type, callback)
|
||||
local function remove_filter(session, type, callback)
|
||||
if not session.filters then return; end
|
||||
local filter_list = session.filters[type];
|
||||
if filter_list and filter_list[callback] then
|
||||
|
|
@ -74,11 +74,11 @@ function remove_filter(session, type, callback)
|
|||
end
|
||||
end
|
||||
|
||||
function add_filter_hook(callback)
|
||||
local function add_filter_hook(callback)
|
||||
t_insert(new_filter_hooks, callback);
|
||||
end
|
||||
|
||||
function remove_filter_hook(callback)
|
||||
local function remove_filter_hook(callback)
|
||||
for i=1,#new_filter_hooks do
|
||||
if new_filter_hooks[i] == callback then
|
||||
t_remove(new_filter_hooks, i);
|
||||
|
|
@ -86,4 +86,10 @@ function remove_filter_hook(callback)
|
|||
end
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
initialize = initialize;
|
||||
add_filter = add_filter;
|
||||
remove_filter = remove_filter;
|
||||
add_filter_hook = add_filter_hook;
|
||||
remove_filter_hook = remove_filter_hook;
|
||||
};
|
||||
|
|
|
|||
|
|
@ -8,21 +8,19 @@
|
|||
|
||||
local debug = require "util.debug";
|
||||
|
||||
module("helpers", package.seeall);
|
||||
|
||||
-- Helper functions for debugging
|
||||
|
||||
local log = require "util.logger".init("util.debug");
|
||||
|
||||
function log_host_events(host)
|
||||
local function log_host_events(host)
|
||||
return log_events(prosody.hosts[host].events, host);
|
||||
end
|
||||
|
||||
function revert_log_host_events(host)
|
||||
local function revert_log_host_events(host)
|
||||
return revert_log_events(prosody.hosts[host].events);
|
||||
end
|
||||
|
||||
function log_events(events, name, logger)
|
||||
local function log_events(events, name, logger)
|
||||
local f = events.fire_event;
|
||||
if not f then
|
||||
error("Object does not appear to be a util.events object");
|
||||
|
|
@ -37,11 +35,11 @@ function log_events(events, name, logger)
|
|||
return events;
|
||||
end
|
||||
|
||||
function revert_log_events(events)
|
||||
local function revert_log_events(events)
|
||||
events.fire_event, events[events.fire_event] = events[events.fire_event], nil; -- :))
|
||||
end
|
||||
|
||||
function show_events(events, specific_event)
|
||||
local function show_events(events, specific_event)
|
||||
local event_handlers = events._handlers;
|
||||
local events_array = {};
|
||||
local event_handler_arrays = {};
|
||||
|
|
@ -70,7 +68,7 @@ function show_events(events, specific_event)
|
|||
return table.concat(events_array, "\n");
|
||||
end
|
||||
|
||||
function get_upvalue(f, get_name)
|
||||
local function get_upvalue(f, get_name)
|
||||
local i, name, value = 0;
|
||||
repeat
|
||||
i = i + 1;
|
||||
|
|
@ -79,4 +77,11 @@ function get_upvalue(f, get_name)
|
|||
return value;
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
log_host_events = log_host_events;
|
||||
revert_log_host_events = revert_log_host_events;
|
||||
log_events = log_events;
|
||||
revert_log_events = revert_log_events;
|
||||
show_events = show_events;
|
||||
get_upvalue = get_upvalue;
|
||||
};
|
||||
|
|
|
|||
42
util/jid.lua
42
util/jid.lua
|
|
@ -23,9 +23,9 @@ local escapes = {
|
|||
local unescapes = {};
|
||||
for k,v in pairs(escapes) do unescapes[v] = k; end
|
||||
|
||||
module "jid"
|
||||
local _ENV = nil;
|
||||
|
||||
local function _split(jid)
|
||||
local function split(jid)
|
||||
if not jid then return; end
|
||||
local node, nodepos = match(jid, "^([^@/]+)@()");
|
||||
local host, hostpos = match(jid, "^([^@/]+)()", nodepos)
|
||||
|
|
@ -34,14 +34,13 @@ local function _split(jid)
|
|||
if (not host) or ((not resource) and #jid >= hostpos) then return nil, nil, nil; end
|
||||
return node, host, resource;
|
||||
end
|
||||
split = _split;
|
||||
|
||||
function bare(jid)
|
||||
local function bare(jid)
|
||||
return jid and match(jid, "^[^/]+");
|
||||
end
|
||||
|
||||
local function _prepped_split(jid)
|
||||
local node, host, resource = _split(jid);
|
||||
local function prepped_split(jid)
|
||||
local node, host, resource = split(jid);
|
||||
if host then
|
||||
if sub(host, -1, -1) == "." then -- Strip empty root label
|
||||
host = sub(host, 1, -2);
|
||||
|
|
@ -59,9 +58,8 @@ local function _prepped_split(jid)
|
|||
return node, host, resource;
|
||||
end
|
||||
end
|
||||
prepped_split = _prepped_split;
|
||||
|
||||
local function _join(node, host, resource)
|
||||
local function join(node, host, resource)
|
||||
if not host then return end
|
||||
if node and resource then
|
||||
return node.."@"..host.."/"..resource;
|
||||
|
|
@ -72,18 +70,17 @@ local function _join(node, host, resource)
|
|||
end
|
||||
return host;
|
||||
end
|
||||
join = _join;
|
||||
|
||||
function prep(jid)
|
||||
local node, host, resource = _prepped_split(jid);
|
||||
return _join(node, host, resource);
|
||||
local function prep(jid)
|
||||
local node, host, resource = prepped_split(jid);
|
||||
return join(node, host, resource);
|
||||
end
|
||||
|
||||
function compare(jid, acl)
|
||||
local function compare(jid, acl)
|
||||
-- compare jid to single acl rule
|
||||
-- TODO compare to table of rules?
|
||||
local jid_node, jid_host, jid_resource = _split(jid);
|
||||
local acl_node, acl_host, acl_resource = _split(acl);
|
||||
local jid_node, jid_host, jid_resource = split(jid);
|
||||
local acl_node, acl_host, acl_resource = split(acl);
|
||||
if ((acl_node ~= nil and acl_node == jid_node) or acl_node == nil) and
|
||||
((acl_host ~= nil and acl_host == jid_host) or acl_host == nil) and
|
||||
((acl_resource ~= nil and acl_resource == jid_resource) or acl_resource == nil) then
|
||||
|
|
@ -92,7 +89,16 @@ function compare(jid, acl)
|
|||
return false
|
||||
end
|
||||
|
||||
function escape(s) return s and (s:gsub(".", escapes)); end
|
||||
function unescape(s) return s and (s:gsub("\\%x%x", unescapes)); end
|
||||
local function escape(s) return s and (s:gsub(".", escapes)); end
|
||||
local function unescape(s) return s and (s:gsub("\\%x%x", unescapes)); end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
split = split;
|
||||
bare = bare;
|
||||
prepped_split = prepped_split;
|
||||
join = join;
|
||||
prep = prep;
|
||||
compare = compare;
|
||||
escape = escape;
|
||||
unescape = unescape;
|
||||
};
|
||||
|
|
|
|||
|
|
@ -11,13 +11,13 @@ local pcall = pcall;
|
|||
local find = string.find;
|
||||
local ipairs, pairs, setmetatable = ipairs, pairs, setmetatable;
|
||||
|
||||
module "logger"
|
||||
local _ENV = nil;
|
||||
|
||||
local level_sinks = {};
|
||||
|
||||
local make_logger;
|
||||
|
||||
function init(name)
|
||||
local function init(name)
|
||||
local log_debug = make_logger(name, "debug");
|
||||
local log_info = make_logger(name, "info");
|
||||
local log_warn = make_logger(name, "warn");
|
||||
|
|
@ -52,7 +52,7 @@ function make_logger(source_name, level)
|
|||
return logger;
|
||||
end
|
||||
|
||||
function reset()
|
||||
local function reset()
|
||||
for level, handler_list in pairs(level_sinks) do
|
||||
-- Clear all handlers for this level
|
||||
for i = 1, #handler_list do
|
||||
|
|
@ -61,7 +61,7 @@ function reset()
|
|||
end
|
||||
end
|
||||
|
||||
function add_level_sink(level, sink_function)
|
||||
local function add_level_sink(level, sink_function)
|
||||
if not level_sinks[level] then
|
||||
level_sinks[level] = { sink_function };
|
||||
else
|
||||
|
|
@ -69,6 +69,10 @@ function add_level_sink(level, sink_function)
|
|||
end
|
||||
end
|
||||
|
||||
_M.new = make_logger;
|
||||
|
||||
return _M;
|
||||
return {
|
||||
init = init;
|
||||
make_logger = make_logger;
|
||||
reset = reset;
|
||||
add_level_sink = add_level_sink;
|
||||
new = make_logger;
|
||||
};
|
||||
|
|
|
|||
|
|
@ -10,7 +10,7 @@ local select = select;
|
|||
local t_insert = table.insert;
|
||||
local unpack, pairs, next, type = unpack, pairs, next, type;
|
||||
|
||||
module "multitable"
|
||||
local _ENV = nil;
|
||||
|
||||
local function get(self, ...)
|
||||
local t = self.data;
|
||||
|
|
@ -126,7 +126,7 @@ local function search_add(self, results, ...)
|
|||
return results;
|
||||
end
|
||||
|
||||
function iter(self, ...)
|
||||
local function iter(self, ...)
|
||||
local query = { ... };
|
||||
local maxdepth = select("#", ...);
|
||||
local stack = { self.data };
|
||||
|
|
@ -161,7 +161,7 @@ function iter(self, ...)
|
|||
return it, self;
|
||||
end
|
||||
|
||||
function new()
|
||||
local function new()
|
||||
return {
|
||||
data = {};
|
||||
get = get;
|
||||
|
|
@ -174,4 +174,7 @@ function new()
|
|||
};
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
iter = iter;
|
||||
new = new;
|
||||
};
|
||||
|
|
|
|||
|
|
@ -17,9 +17,7 @@ end
|
|||
local io_open = io.open;
|
||||
local envload = require "util.envload".envload;
|
||||
|
||||
module "pluginloader"
|
||||
|
||||
function load_file(names)
|
||||
local function load_file(names)
|
||||
local file, err, path;
|
||||
for i=1,#plugin_dir do
|
||||
for j=1,#names do
|
||||
|
|
@ -35,7 +33,7 @@ function load_file(names)
|
|||
return file, err;
|
||||
end
|
||||
|
||||
function load_resource(plugin, resource)
|
||||
local function load_resource(plugin, resource)
|
||||
resource = resource or "mod_"..plugin..".lua";
|
||||
|
||||
local names = {
|
||||
|
|
@ -48,7 +46,7 @@ function load_resource(plugin, resource)
|
|||
return load_file(names);
|
||||
end
|
||||
|
||||
function load_code(plugin, resource, env)
|
||||
local function load_code(plugin, resource, env)
|
||||
local content, err = load_resource(plugin, resource);
|
||||
if not content then return content, err; end
|
||||
local path = err;
|
||||
|
|
@ -57,4 +55,8 @@ function load_code(plugin, resource, env)
|
|||
return f, path;
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
load_file = load_file;
|
||||
load_resource = load_resource;
|
||||
load_code = load_code;
|
||||
};
|
||||
|
|
|
|||
|
|
@ -29,25 +29,19 @@ local CFG_SOURCEDIR = _G.CFG_SOURCEDIR;
|
|||
local _G = _G;
|
||||
local prosody = prosody;
|
||||
|
||||
module "prosodyctl"
|
||||
|
||||
-- UI helpers
|
||||
function show_message(msg, ...)
|
||||
local function show_message(msg, ...)
|
||||
print(msg:format(...));
|
||||
end
|
||||
|
||||
function show_warning(msg, ...)
|
||||
print(msg:format(...));
|
||||
end
|
||||
|
||||
function show_usage(usage, desc)
|
||||
local function show_usage(usage, desc)
|
||||
print("Usage: ".._G.arg[0].." "..usage);
|
||||
if desc then
|
||||
print(" "..desc);
|
||||
end
|
||||
end
|
||||
|
||||
function getchar(n)
|
||||
local function getchar(n)
|
||||
local stty_ret = os.execute("stty raw -echo 2>/dev/null");
|
||||
local ok, char;
|
||||
if stty_ret == 0 then
|
||||
|
|
@ -64,14 +58,14 @@ function getchar(n)
|
|||
end
|
||||
end
|
||||
|
||||
function getline()
|
||||
local function getline()
|
||||
local ok, line = pcall(io.read, "*l");
|
||||
if ok then
|
||||
return line;
|
||||
end
|
||||
end
|
||||
|
||||
function getpass()
|
||||
local function getpass()
|
||||
local stty_ret = os.execute("stty -echo 2>/dev/null");
|
||||
if stty_ret ~= 0 then
|
||||
io.write("\027[08m"); -- ANSI 'hidden' text attribute
|
||||
|
|
@ -88,7 +82,7 @@ function getpass()
|
|||
end
|
||||
end
|
||||
|
||||
function show_yesno(prompt)
|
||||
local function show_yesno(prompt)
|
||||
io.write(prompt, " ");
|
||||
local choice = getchar():lower();
|
||||
io.write("\n");
|
||||
|
|
@ -99,7 +93,7 @@ function show_yesno(prompt)
|
|||
return (choice == "y");
|
||||
end
|
||||
|
||||
function read_password()
|
||||
local function read_password()
|
||||
local password;
|
||||
while true do
|
||||
io.write("Enter new password: ");
|
||||
|
|
@ -120,7 +114,7 @@ function read_password()
|
|||
return password;
|
||||
end
|
||||
|
||||
function show_prompt(prompt)
|
||||
local function show_prompt(prompt)
|
||||
io.write(prompt, " ");
|
||||
local line = getline();
|
||||
line = line and line:gsub("\n$","");
|
||||
|
|
@ -128,7 +122,7 @@ function show_prompt(prompt)
|
|||
end
|
||||
|
||||
-- Server control
|
||||
function adduser(params)
|
||||
local function adduser(params)
|
||||
local user, host, password = nodeprep(params.user), nameprep(params.host), params.password;
|
||||
if not user then
|
||||
return false, "invalid-username";
|
||||
|
|
@ -149,12 +143,12 @@ function adduser(params)
|
|||
|
||||
local ok, errmsg = usermanager.create_user(user, password, host);
|
||||
if not ok then
|
||||
return false, errmsg;
|
||||
return false, errmsg or "creating-user-failed";
|
||||
end
|
||||
return true;
|
||||
end
|
||||
|
||||
function user_exists(params)
|
||||
local function user_exists(params)
|
||||
local user, host, password = nodeprep(params.user), nameprep(params.host), params.password;
|
||||
|
||||
storagemanager.initialize_host(host);
|
||||
|
|
@ -166,16 +160,16 @@ function user_exists(params)
|
|||
return usermanager.user_exists(user, host);
|
||||
end
|
||||
|
||||
function passwd(params)
|
||||
if not _M.user_exists(params) then
|
||||
local function passwd(params)
|
||||
if not user_exists(params) then
|
||||
return false, "no-such-user";
|
||||
end
|
||||
|
||||
return _M.adduser(params);
|
||||
return adduser(params);
|
||||
end
|
||||
|
||||
function deluser(params)
|
||||
if not _M.user_exists(params) then
|
||||
local function deluser(params)
|
||||
if not user_exists(params) then
|
||||
return false, "no-such-user";
|
||||
end
|
||||
local user, host = nodeprep(params.user), nameprep(params.host);
|
||||
|
|
@ -183,7 +177,7 @@ function deluser(params)
|
|||
return usermanager.delete_user(user, host);
|
||||
end
|
||||
|
||||
function getpid()
|
||||
local function getpid()
|
||||
local pidfile = config.get("*", "pidfile");
|
||||
if not pidfile then
|
||||
return false, "no-pidfile";
|
||||
|
|
@ -219,8 +213,8 @@ function getpid()
|
|||
return true, pid;
|
||||
end
|
||||
|
||||
function isrunning()
|
||||
local ok, pid, err = _M.getpid();
|
||||
local function isrunning()
|
||||
local ok, pid, err = getpid();
|
||||
if not ok then
|
||||
if pid == "pidfile-read-failed" or pid == "pidfile-not-locked" then
|
||||
-- Report as not running, since we can't open the pidfile
|
||||
|
|
@ -232,8 +226,8 @@ function isrunning()
|
|||
return true, signal.kill(pid, 0) == 0;
|
||||
end
|
||||
|
||||
function start()
|
||||
local ok, ret = _M.isrunning();
|
||||
local function start()
|
||||
local ok, ret = isrunning();
|
||||
if not ok then
|
||||
return ok, ret;
|
||||
end
|
||||
|
|
@ -248,8 +242,8 @@ function start()
|
|||
return true;
|
||||
end
|
||||
|
||||
function stop()
|
||||
local ok, ret = _M.isrunning();
|
||||
local function stop()
|
||||
local ok, ret = isrunning();
|
||||
if not ok then
|
||||
return ok, ret;
|
||||
end
|
||||
|
|
@ -257,15 +251,15 @@ function stop()
|
|||
return false, "not-running";
|
||||
end
|
||||
|
||||
local ok, pid = _M.getpid()
|
||||
local ok, pid = getpid()
|
||||
if not ok then return false, pid; end
|
||||
|
||||
signal.kill(pid, signal.SIGTERM);
|
||||
return true;
|
||||
end
|
||||
|
||||
function reload()
|
||||
local ok, ret = _M.isrunning();
|
||||
local function reload()
|
||||
local ok, ret = isrunning();
|
||||
if not ok then
|
||||
return ok, ret;
|
||||
end
|
||||
|
|
@ -273,11 +267,30 @@ function reload()
|
|||
return false, "not-running";
|
||||
end
|
||||
|
||||
local ok, pid = _M.getpid()
|
||||
local ok, pid = getpid()
|
||||
if not ok then return false, pid; end
|
||||
|
||||
signal.kill(pid, signal.SIGHUP);
|
||||
return true;
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
show_message = show_message;
|
||||
show_warning = show_message;
|
||||
show_usage = show_usage;
|
||||
getchar = getchar;
|
||||
getline = getline;
|
||||
getpass = getpass;
|
||||
show_yesno = show_yesno;
|
||||
read_password = read_password;
|
||||
show_prompt = show_prompt;
|
||||
adduser = adduser;
|
||||
user_exists = user_exists;
|
||||
passwd = passwd;
|
||||
deluser = deluser;
|
||||
getpid = getpid;
|
||||
isrunning = isrunning;
|
||||
start = start;
|
||||
stop = stop;
|
||||
reload = reload;
|
||||
};
|
||||
|
|
|
|||
|
|
@ -1,8 +1,6 @@
|
|||
local events = require "util.events";
|
||||
local t_remove = table.remove;
|
||||
|
||||
module("pubsub", package.seeall);
|
||||
|
||||
local service = {};
|
||||
local service_mt = { __index = service };
|
||||
|
||||
|
|
@ -15,7 +13,7 @@ local default_node_config = { __index = {
|
|||
["pubsub#max_items"] = "20";
|
||||
} };
|
||||
|
||||
function new(config)
|
||||
local function new(config)
|
||||
config = config or {};
|
||||
return setmetatable({
|
||||
config = setmetatable(config, default_config);
|
||||
|
|
@ -442,4 +440,6 @@ function service:set_node_config(node, actor, new_config)
|
|||
return true;
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
new = new;
|
||||
};
|
||||
|
|
|
|||
|
|
@ -19,7 +19,7 @@ local setmetatable = setmetatable;
|
|||
local assert = assert;
|
||||
local require = require;
|
||||
|
||||
module "sasl"
|
||||
local _ENV = nil;
|
||||
|
||||
--[[
|
||||
Authentication Backend Prototypes:
|
||||
|
|
@ -47,7 +47,7 @@ local backend_mechanism = {};
|
|||
local mechanism_channelbindings = {};
|
||||
|
||||
-- register a new SASL mechanims
|
||||
function registerMechanism(name, backends, f, cb_backends)
|
||||
local function registerMechanism(name, backends, f, cb_backends)
|
||||
assert(type(name) == "string", "Parameter name MUST be a string.");
|
||||
assert(type(backends) == "string" or type(backends) == "table", "Parameter backends MUST be either a string or a table.");
|
||||
assert(type(f) == "function", "Parameter f MUST be a function.");
|
||||
|
|
@ -66,7 +66,7 @@ function registerMechanism(name, backends, f, cb_backends)
|
|||
end
|
||||
|
||||
-- create a new SASL object which can be used to authenticate clients
|
||||
function new(realm, profile)
|
||||
local function new(realm, profile)
|
||||
local mechanisms = profile.mechanisms;
|
||||
if not mechanisms then
|
||||
mechanisms = {};
|
||||
|
|
@ -138,4 +138,7 @@ require "util.sasl.anonymous" .init(registerMechanism);
|
|||
require "util.sasl.scram" .init(registerMechanism);
|
||||
require "util.sasl.external" .init(registerMechanism);
|
||||
|
||||
return _M;
|
||||
return {
|
||||
registerMechanism = registerMechanism;
|
||||
new = new;
|
||||
};
|
||||
|
|
|
|||
|
|
@ -16,7 +16,7 @@ local s_match = string.match;
|
|||
local log = require "util.logger".init("sasl");
|
||||
local generate_uuid = require "util.uuid".generate;
|
||||
|
||||
module "sasl.anonymous"
|
||||
local _ENV = nil;
|
||||
|
||||
--=========================
|
||||
--SASL ANONYMOUS according to RFC 4505
|
||||
|
|
@ -39,8 +39,10 @@ local function anonymous(self, message)
|
|||
return "success"
|
||||
end
|
||||
|
||||
function init(registerMechanism)
|
||||
local function init(registerMechanism)
|
||||
registerMechanism("ANONYMOUS", {"anonymous"}, anonymous);
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
init = init;
|
||||
}
|
||||
|
|
|
|||
|
|
@ -25,7 +25,7 @@ local log = require "util.logger".init("sasl");
|
|||
local generate_uuid = require "util.uuid".generate;
|
||||
local nodeprep = require "util.encodings".stringprep.nodeprep;
|
||||
|
||||
module "sasl.digest-md5"
|
||||
local _ENV = nil;
|
||||
|
||||
--=========================
|
||||
--SASL DIGEST-MD5 according to RFC 2831
|
||||
|
|
@ -241,8 +241,10 @@ local function digest(self, message)
|
|||
end
|
||||
end
|
||||
|
||||
function init(registerMechanism)
|
||||
local function init(registerMechanism)
|
||||
registerMechanism("DIGEST-MD5", {"plain"}, digest);
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
init = init;
|
||||
}
|
||||
|
|
|
|||
|
|
@ -1,6 +1,6 @@
|
|||
local saslprep = require "util.encodings".stringprep.saslprep;
|
||||
|
||||
module "sasl.external"
|
||||
local _ENV = nil;
|
||||
|
||||
local function external(self, message)
|
||||
message = saslprep(message);
|
||||
|
|
@ -18,8 +18,10 @@ local function external(self, message)
|
|||
return "success";
|
||||
end
|
||||
|
||||
function init(registerMechanism)
|
||||
local function init(registerMechanism)
|
||||
registerMechanism("EXTERNAL", {"external"}, external);
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
init = init;
|
||||
}
|
||||
|
|
|
|||
|
|
@ -16,7 +16,7 @@ local saslprep = require "util.encodings".stringprep.saslprep;
|
|||
local nodeprep = require "util.encodings".stringprep.nodeprep;
|
||||
local log = require "util.logger".init("sasl");
|
||||
|
||||
module "sasl.plain"
|
||||
local _ENV = nil;
|
||||
|
||||
-- ================================
|
||||
-- SASL PLAIN according to RFC 4616
|
||||
|
|
@ -82,8 +82,10 @@ local function plain(self, message)
|
|||
return "success";
|
||||
end
|
||||
|
||||
function init(registerMechanism)
|
||||
local function init(registerMechanism)
|
||||
registerMechanism("PLAIN", {"plain", "plain_test"}, plain);
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
init = init;
|
||||
}
|
||||
|
|
|
|||
|
|
@ -60,7 +60,7 @@ local sasl_errstring = {
|
|||
};
|
||||
setmetatable(sasl_errstring, { __index = function() return "undefined error!" end });
|
||||
|
||||
module "sasl_cyrus"
|
||||
local _ENV = nil;
|
||||
|
||||
local method = {};
|
||||
method.__index = method;
|
||||
|
|
@ -82,7 +82,7 @@ end
|
|||
-- For GSSAPI, this determines the hostname in the service ticket (after
|
||||
-- reverse DNS canonicalization, only if [libdefaults] rdns = true which
|
||||
-- is the default).
|
||||
function new(realm, service_name, app_name, host_fqdn)
|
||||
local function new(realm, service_name, app_name, host_fqdn)
|
||||
|
||||
init(app_name or service_name);
|
||||
|
||||
|
|
@ -163,4 +163,6 @@ function method:process(message)
|
|||
end
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
new = new;
|
||||
};
|
||||
|
|
|
|||
|
|
@ -20,7 +20,7 @@ local debug_traceback = debug.traceback;
|
|||
local log = require "util.logger".init("serialization");
|
||||
local envload = require"util.envload".envload;
|
||||
|
||||
module "serialization"
|
||||
local _ENV = nil;
|
||||
|
||||
local indent = function(i)
|
||||
return string_rep("\t", i);
|
||||
|
|
@ -71,16 +71,16 @@ local function _simplesave(o, ind, t, func)
|
|||
end
|
||||
end
|
||||
|
||||
function append(t, o)
|
||||
local function append(t, o)
|
||||
_simplesave(o, 1, t, t.write or t_insert);
|
||||
return t;
|
||||
end
|
||||
|
||||
function serialize(o)
|
||||
local function serialize(o)
|
||||
return t_concat(append({}, o));
|
||||
end
|
||||
|
||||
function deserialize(str)
|
||||
local function deserialize(str)
|
||||
if type(str) ~= "string" then return nil; end
|
||||
str = "return "..str;
|
||||
local f, err = envload(str, "@data", {});
|
||||
|
|
@ -90,4 +90,8 @@ function deserialize(str)
|
|||
return ret;
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
append = append;
|
||||
serialize = serialize;
|
||||
deserialize = deserialize;
|
||||
};
|
||||
|
|
|
|||
101
util/set.lua
101
util/set.lua
|
|
@ -10,59 +10,19 @@ local ipairs, pairs, setmetatable, next, tostring =
|
|||
ipairs, pairs, setmetatable, next, tostring;
|
||||
local t_concat = table.concat;
|
||||
|
||||
module "set"
|
||||
local _ENV = nil;
|
||||
|
||||
local set_mt = {};
|
||||
function set_mt.__call(set, _, k)
|
||||
return next(set._items, k);
|
||||
end
|
||||
function set_mt.__add(set1, set2)
|
||||
return _M.union(set1, set2);
|
||||
end
|
||||
function set_mt.__sub(set1, set2)
|
||||
return _M.difference(set1, set2);
|
||||
end
|
||||
function set_mt.__div(set, func)
|
||||
local new_set = _M.new();
|
||||
local items, new_items = set._items, new_set._items;
|
||||
for item in pairs(items) do
|
||||
local new_item = func(item);
|
||||
if new_item ~= nil then
|
||||
new_items[new_item] = true;
|
||||
end
|
||||
end
|
||||
return new_set;
|
||||
end
|
||||
function set_mt.__eq(set1, set2)
|
||||
set1, set2 = set1._items, set2._items;
|
||||
for item in pairs(set1) do
|
||||
if not set2[item] then
|
||||
return false;
|
||||
end
|
||||
end
|
||||
|
||||
for item in pairs(set2) do
|
||||
if not set1[item] then
|
||||
return false;
|
||||
end
|
||||
end
|
||||
|
||||
return true;
|
||||
end
|
||||
function set_mt.__tostring(set)
|
||||
local s, items = { }, set._items;
|
||||
for item in pairs(items) do
|
||||
s[#s+1] = tostring(item);
|
||||
end
|
||||
return t_concat(s, ", ");
|
||||
end
|
||||
|
||||
local items_mt = {};
|
||||
function items_mt.__call(items, _, k)
|
||||
return next(items, k);
|
||||
end
|
||||
|
||||
function new(list)
|
||||
local function new(list)
|
||||
local items = setmetatable({}, items_mt);
|
||||
local set = { _items = items };
|
||||
|
||||
|
|
@ -116,7 +76,7 @@ function new(list)
|
|||
return setmetatable(set, set_mt);
|
||||
end
|
||||
|
||||
function union(set1, set2)
|
||||
local function union(set1, set2)
|
||||
local set = new();
|
||||
local items = set._items;
|
||||
|
||||
|
|
@ -131,7 +91,7 @@ function union(set1, set2)
|
|||
return set;
|
||||
end
|
||||
|
||||
function difference(set1, set2)
|
||||
local function difference(set1, set2)
|
||||
local set = new();
|
||||
local items = set._items;
|
||||
|
||||
|
|
@ -142,7 +102,7 @@ function difference(set1, set2)
|
|||
return set;
|
||||
end
|
||||
|
||||
function intersection(set1, set2)
|
||||
local function intersection(set1, set2)
|
||||
local set = new();
|
||||
local items = set._items;
|
||||
|
||||
|
|
@ -155,8 +115,55 @@ function intersection(set1, set2)
|
|||
return set;
|
||||
end
|
||||
|
||||
function xor(set1, set2)
|
||||
local function xor(set1, set2)
|
||||
return union(set1, set2) - intersection(set1, set2);
|
||||
end
|
||||
|
||||
return _M;
|
||||
function set_mt.__add(set1, set2)
|
||||
return union(set1, set2);
|
||||
end
|
||||
function set_mt.__sub(set1, set2)
|
||||
return difference(set1, set2);
|
||||
end
|
||||
function set_mt.__div(set, func)
|
||||
local new_set = new();
|
||||
local items, new_items = set._items, new_set._items;
|
||||
for item in pairs(items) do
|
||||
local new_item = func(item);
|
||||
if new_item ~= nil then
|
||||
new_items[new_item] = true;
|
||||
end
|
||||
end
|
||||
return new_set;
|
||||
end
|
||||
function set_mt.__eq(set1, set2)
|
||||
set1, set2 = set1._items, set2._items;
|
||||
for item in pairs(set1) do
|
||||
if not set2[item] then
|
||||
return false;
|
||||
end
|
||||
end
|
||||
|
||||
for item in pairs(set2) do
|
||||
if not set1[item] then
|
||||
return false;
|
||||
end
|
||||
end
|
||||
|
||||
return true;
|
||||
end
|
||||
function set_mt.__tostring(set)
|
||||
local s, items = { }, set._items;
|
||||
for item in pairs(items) do
|
||||
s[#s+1] = tostring(item);
|
||||
end
|
||||
return t_concat(s, ", ");
|
||||
end
|
||||
|
||||
return {
|
||||
new = new;
|
||||
union = union;
|
||||
difference = difference;
|
||||
intersection = intersection;
|
||||
xor = xor;
|
||||
};
|
||||
|
|
|
|||
37
util/sql.lua
37
util/sql.lua
|
|
@ -13,7 +13,7 @@ local DBI = require "DBI";
|
|||
DBI.Drivers();
|
||||
local build_url = require "socket.url".build;
|
||||
|
||||
module("sql")
|
||||
local _ENV = nil;
|
||||
|
||||
local column_mt = {};
|
||||
local table_mt = {};
|
||||
|
|
@ -21,17 +21,17 @@ local query_mt = {};
|
|||
--local op_mt = {};
|
||||
local index_mt = {};
|
||||
|
||||
function is_column(x) return getmetatable(x)==column_mt; end
|
||||
function is_index(x) return getmetatable(x)==index_mt; end
|
||||
function is_table(x) return getmetatable(x)==table_mt; end
|
||||
function is_query(x) return getmetatable(x)==query_mt; end
|
||||
function Integer(n) return "Integer()" end
|
||||
function String(n) return "String()" end
|
||||
local function is_column(x) return getmetatable(x)==column_mt; end
|
||||
local function is_index(x) return getmetatable(x)==index_mt; end
|
||||
local function is_table(x) return getmetatable(x)==table_mt; end
|
||||
local function is_query(x) return getmetatable(x)==query_mt; end
|
||||
local function Integer(n) return "Integer()" end
|
||||
local function String(n) return "String()" end
|
||||
|
||||
function Column(definition)
|
||||
local function Column(definition)
|
||||
return setmetatable(definition, column_mt);
|
||||
end
|
||||
function Table(definition)
|
||||
local function Table(definition)
|
||||
local c = {}
|
||||
for i,col in ipairs(definition) do
|
||||
if is_column(col) then
|
||||
|
|
@ -42,7 +42,7 @@ function Table(definition)
|
|||
end
|
||||
return setmetatable({ __table__ = definition, c = c, name = definition.name }, table_mt);
|
||||
end
|
||||
function Index(definition)
|
||||
local function Index(definition)
|
||||
return setmetatable(definition, index_mt);
|
||||
end
|
||||
|
||||
|
|
@ -302,7 +302,7 @@ function engine:set_encoding() -- to UTF-8
|
|||
end
|
||||
local engine_mt = { __index = engine };
|
||||
|
||||
function db2uri(params)
|
||||
local function db2uri(params)
|
||||
return build_url{
|
||||
scheme = params.driver,
|
||||
user = params.username,
|
||||
|
|
@ -313,8 +313,19 @@ function db2uri(params)
|
|||
};
|
||||
end
|
||||
|
||||
function create_engine(self, params, onconnect)
|
||||
local function create_engine(self, params, onconnect)
|
||||
return setmetatable({ url = db2uri(params), params = params, onconnect = onconnect }, engine_mt);
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
is_column = is_column;
|
||||
is_index = is_index;
|
||||
is_table = is_table;
|
||||
is_query = is_query;
|
||||
Integer = Integer;
|
||||
String = String;
|
||||
Column = Column;
|
||||
Table = Table;
|
||||
Index = Index;
|
||||
create_engine = create_engine;
|
||||
};
|
||||
|
|
|
|||
|
|
@ -1,3 +1,11 @@
|
|||
local type = type;
|
||||
local pairs = pairs;
|
||||
local rawset = rawset;
|
||||
local t_concat = table.concat;
|
||||
local t_insert = table.insert;
|
||||
local setmetatable = setmetatable;
|
||||
|
||||
local _ENV = nil;
|
||||
|
||||
local handlers = { };
|
||||
local finalisers = { };
|
||||
|
|
@ -34,7 +42,7 @@ finalisers.verifyext = finalisers.options;
|
|||
|
||||
function finalisers.ciphers(a)
|
||||
if type(a) == "table" then
|
||||
return table.concat(a, ":");
|
||||
return t_concat(a, ":");
|
||||
end
|
||||
return a;
|
||||
end
|
||||
|
|
@ -47,7 +55,7 @@ local function protocol(a)
|
|||
if min_protocol then
|
||||
a.protocol = "sslv23";
|
||||
for i = 1, min_protocol do
|
||||
table.insert(a.options, "no_"..protocols[i]);
|
||||
t_insert(a.options, "no_"..protocols[i]);
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
|
|||
|
|
@ -35,13 +35,12 @@ end
|
|||
|
||||
local xmlns_stanzas = "urn:ietf:params:xml:ns:xmpp-stanzas";
|
||||
|
||||
module "stanza"
|
||||
local _ENV = nil;
|
||||
|
||||
stanza_mt = { __type = "stanza" };
|
||||
local stanza_mt = { __type = "stanza" };
|
||||
stanza_mt.__index = stanza_mt;
|
||||
local stanza_mt = stanza_mt;
|
||||
|
||||
function stanza(name, attr)
|
||||
local function stanza(name, attr)
|
||||
local stanza = { name = name, attr = attr or {}, tags = {} };
|
||||
return setmetatable(stanza, stanza_mt);
|
||||
end
|
||||
|
|
@ -200,12 +199,8 @@ function stanza_mt:find(path)
|
|||
end
|
||||
|
||||
|
||||
local xml_escape
|
||||
do
|
||||
local escape_table = { ["'"] = "'", ["\""] = """, ["<"] = "<", [">"] = ">", ["&"] = "&" };
|
||||
function xml_escape(str) return (s_gsub(str, "['&<>\"]", escape_table)); end
|
||||
_M.xml_escape = xml_escape;
|
||||
end
|
||||
local escape_table = { ["'"] = "'", ["\""] = """, ["<"] = "<", [">"] = ">", ["&"] = "&" };
|
||||
local function xml_escape(str) return (s_gsub(str, "['&<>\"]", escape_table)); end
|
||||
|
||||
local function _dostring(t, buf, self, xml_escape, parentns)
|
||||
local nsid = 0;
|
||||
|
|
@ -280,15 +275,13 @@ function stanza_mt.get_error(stanza)
|
|||
return type, condition or "undefined-condition", text;
|
||||
end
|
||||
|
||||
do
|
||||
local id = 0;
|
||||
function new_id()
|
||||
id = id + 1;
|
||||
return "lx"..id;
|
||||
end
|
||||
local id = 0;
|
||||
local function new_id()
|
||||
id = id + 1;
|
||||
return "lx"..id;
|
||||
end
|
||||
|
||||
function preserialize(stanza)
|
||||
local function preserialize(stanza)
|
||||
local s = { name = stanza.name, attr = stanza.attr };
|
||||
for _, child in ipairs(stanza) do
|
||||
if type(child) == "table" then
|
||||
|
|
@ -300,7 +293,7 @@ function preserialize(stanza)
|
|||
return s;
|
||||
end
|
||||
|
||||
function deserialize(stanza)
|
||||
local function deserialize(stanza)
|
||||
-- Set metatable
|
||||
if stanza then
|
||||
local attr = stanza.attr;
|
||||
|
|
@ -337,51 +330,48 @@ function deserialize(stanza)
|
|||
return stanza;
|
||||
end
|
||||
|
||||
local function _clone(stanza)
|
||||
local function clone(stanza)
|
||||
local attr, tags = {}, {};
|
||||
for k,v in pairs(stanza.attr) do attr[k] = v; end
|
||||
local new = { name = stanza.name, attr = attr, tags = tags };
|
||||
for i=1,#stanza do
|
||||
local child = stanza[i];
|
||||
if child.name then
|
||||
child = _clone(child);
|
||||
child = clone(child);
|
||||
t_insert(tags, child);
|
||||
end
|
||||
t_insert(new, child);
|
||||
end
|
||||
return setmetatable(new, stanza_mt);
|
||||
end
|
||||
clone = _clone;
|
||||
|
||||
function message(attr, body)
|
||||
local function message(attr, body)
|
||||
if not body then
|
||||
return stanza("message", attr);
|
||||
else
|
||||
return stanza("message", attr):tag("body"):text(body):up();
|
||||
end
|
||||
end
|
||||
function iq(attr)
|
||||
local function iq(attr)
|
||||
if attr and not attr.id then attr.id = new_id(); end
|
||||
return stanza("iq", attr or { id = new_id() });
|
||||
end
|
||||
|
||||
function reply(orig)
|
||||
local function reply(orig)
|
||||
return stanza(orig.name, orig.attr and { to = orig.attr.from, from = orig.attr.to, id = orig.attr.id, type = ((orig.name == "iq" and "result") or orig.attr.type) });
|
||||
end
|
||||
|
||||
do
|
||||
local xmpp_stanzas_attr = { xmlns = xmlns_stanzas };
|
||||
function error_reply(orig, type, condition, message)
|
||||
local t = reply(orig);
|
||||
t.attr.type = "error";
|
||||
t:tag("error", {type = type}) --COMPAT: Some day xmlns:stanzas goes here
|
||||
:tag(condition, xmpp_stanzas_attr):up();
|
||||
if (message) then t:tag("text", xmpp_stanzas_attr):text(message):up(); end
|
||||
return t; -- stanza ready for adding app-specific errors
|
||||
end
|
||||
local xmpp_stanzas_attr = { xmlns = xmlns_stanzas };
|
||||
local function error_reply(orig, type, condition, message)
|
||||
local t = reply(orig);
|
||||
t.attr.type = "error";
|
||||
t:tag("error", {type = type}) --COMPAT: Some day xmlns:stanzas goes here
|
||||
:tag(condition, xmpp_stanzas_attr):up();
|
||||
if (message) then t:tag("text", xmpp_stanzas_attr):text(message):up(); end
|
||||
return t; -- stanza ready for adding app-specific errors
|
||||
end
|
||||
|
||||
function presence(attr)
|
||||
local function presence(attr)
|
||||
return stanza("presence", attr);
|
||||
end
|
||||
|
||||
|
|
@ -425,4 +415,16 @@ else
|
|||
stanza_mt.pretty_top_tag = stanza_mt.top_tag;
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
stanza_mt = stanza_mt;
|
||||
stanza = stanza;
|
||||
new_id = new_id;
|
||||
preserialize = preserialize;
|
||||
deserialize = deserialize;
|
||||
clone = clone;
|
||||
message = message;
|
||||
iq = iq;
|
||||
reply = reply;
|
||||
error_reply = error_reply;
|
||||
presence = presence;
|
||||
};
|
||||
|
|
|
|||
|
|
@ -9,7 +9,7 @@ local debug = debug;
|
|||
local t_remove = table.remove;
|
||||
local parse_xml = require "util.xml".parse;
|
||||
|
||||
module("template")
|
||||
local _ENV = nil;
|
||||
|
||||
local function trim_xml(stanza)
|
||||
for i=#stanza,1,-1 do
|
||||
|
|
|
|||
|
|
@ -19,7 +19,7 @@ if os.getenv("WINDIR") then
|
|||
end
|
||||
local orig_color = windows and windows.get_consolecolor and windows.get_consolecolor();
|
||||
|
||||
module "termcolours"
|
||||
local _ENV = nil;
|
||||
|
||||
local stylemap = {
|
||||
reset = 0; bright = 1, dim = 2, underscore = 4, blink = 5, reverse = 7, hidden = 8;
|
||||
|
|
@ -45,7 +45,7 @@ local cssmap = {
|
|||
};
|
||||
|
||||
local fmt_string = char(0x1B).."[%sm%s"..char(0x1B).."[0m";
|
||||
function getstring(style, text)
|
||||
local function getstring(style, text)
|
||||
if style then
|
||||
return format(fmt_string, style, text);
|
||||
else
|
||||
|
|
@ -53,7 +53,7 @@ function getstring(style, text)
|
|||
end
|
||||
end
|
||||
|
||||
function getstyle(...)
|
||||
local function getstyle(...)
|
||||
local styles, result = { ... }, {};
|
||||
for i, style in ipairs(styles) do
|
||||
style = stylemap[style];
|
||||
|
|
@ -65,7 +65,7 @@ function getstyle(...)
|
|||
end
|
||||
|
||||
local last = "0";
|
||||
function setstyle(style)
|
||||
local function setstyle(style)
|
||||
style = style or "0";
|
||||
if style ~= last then
|
||||
io_write("\27["..style.."m");
|
||||
|
|
@ -95,8 +95,13 @@ local function ansi2css(ansi_codes)
|
|||
return "</span><span style='"..t_concat(css, ";").."'>";
|
||||
end
|
||||
|
||||
function tohtml(input)
|
||||
local function tohtml(input)
|
||||
return input:gsub("\027%[(.-)m", ansi2css);
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
getstring = getstring;
|
||||
getstyle = getstyle;
|
||||
setstyle = setstyle;
|
||||
tohtml = tohtml;
|
||||
};
|
||||
|
|
|
|||
|
|
@ -3,7 +3,7 @@ local gettime = require "socket".gettime;
|
|||
local setmetatable = setmetatable;
|
||||
local floor = math.floor;
|
||||
|
||||
module "throttle"
|
||||
local _ENV = nil;
|
||||
|
||||
local throttle = {};
|
||||
local throttle_mt = { __index = throttle };
|
||||
|
|
@ -39,8 +39,10 @@ function throttle:poll(cost, split)
|
|||
end
|
||||
end
|
||||
|
||||
function create(max, period)
|
||||
local function create(max, period)
|
||||
return setmetatable({ rate = max / period, max = max, t = 0, balance = max }, throttle_mt);
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
create = create;
|
||||
};
|
||||
|
|
|
|||
|
|
@ -17,7 +17,7 @@ local type = type;
|
|||
local data = {};
|
||||
local new_data = {};
|
||||
|
||||
module "timer"
|
||||
local _ENV = nil;
|
||||
|
||||
local _add_task;
|
||||
if not server.event then
|
||||
|
|
@ -78,6 +78,6 @@ else
|
|||
end
|
||||
end
|
||||
|
||||
add_task = _add_task;
|
||||
|
||||
return _M;
|
||||
return {
|
||||
add_task = _add_task;
|
||||
};
|
||||
|
|
|
|||
|
|
@ -2,12 +2,12 @@ local timer = require "util.timer";
|
|||
local setmetatable = setmetatable;
|
||||
local os_time = os.time;
|
||||
|
||||
module "watchdog"
|
||||
local _ENV = nil;
|
||||
|
||||
local watchdog_methods = {};
|
||||
local watchdog_mt = { __index = watchdog_methods };
|
||||
|
||||
function new(timeout, callback)
|
||||
local function new(timeout, callback)
|
||||
local watchdog = setmetatable({ timeout = timeout, last_reset = os_time(), callback = callback }, watchdog_mt);
|
||||
timer.add_task(timeout+1, function (current_time)
|
||||
local last_reset = watchdog.last_reset;
|
||||
|
|
@ -31,4 +31,6 @@ function watchdog_methods:cancel()
|
|||
self.last_reset = nil;
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
new = new;
|
||||
};
|
||||
|
|
|
|||
|
|
@ -24,7 +24,7 @@ local base64 = require "util.encodings".base64;
|
|||
local log = require "util.logger".init("x509");
|
||||
local s_format = string.format;
|
||||
|
||||
module "x509"
|
||||
local _ENV = nil;
|
||||
|
||||
local oid_commonname = "2.5.4.3"; -- [LDAP] 2.3
|
||||
local oid_subjectaltname = "2.5.29.17"; -- [PKIX] 4.2.1.6
|
||||
|
|
@ -147,7 +147,7 @@ local function compare_srvname(host, service, asserted_names)
|
|||
return false
|
||||
end
|
||||
|
||||
function verify_identity(host, service, cert)
|
||||
local function verify_identity(host, service, cert)
|
||||
if cert.setencode then
|
||||
cert:setencode("utf8");
|
||||
end
|
||||
|
|
@ -218,7 +218,7 @@ end
|
|||
local pat = "%-%-%-%-%-BEGIN ([A-Z ]+)%-%-%-%-%-\r?\n"..
|
||||
"([0-9A-Za-z+/=\r\n]*)\r?\n%-%-%-%-%-END %1%-%-%-%-%-";
|
||||
|
||||
function pem2der(pem)
|
||||
local function pem2der(pem)
|
||||
local typ, data = pem:match(pat);
|
||||
if typ and data then
|
||||
return base64.decode(data), typ;
|
||||
|
|
@ -228,10 +228,14 @@ end
|
|||
local wrap = ('.'):rep(64);
|
||||
local envelope = "-----BEGIN %s-----\n%s\n-----END %s-----\n"
|
||||
|
||||
function der2pem(data, typ)
|
||||
local function der2pem(data, typ)
|
||||
typ = typ and typ:upper() or "CERTIFICATE";
|
||||
data = base64.encode(data);
|
||||
return s_format(envelope, typ, data:gsub(wrap, '%0\n', (#data-1)/64), typ);
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
verify_identity = verify_identity;
|
||||
pem2der = pem2der;
|
||||
der2pem = der2pem;
|
||||
};
|
||||
|
|
|
|||
|
|
@ -2,7 +2,7 @@
|
|||
local st = require "util.stanza";
|
||||
local lxp = require "lxp";
|
||||
|
||||
module("xml")
|
||||
local _ENV = nil;
|
||||
|
||||
local parse_xml = (function()
|
||||
local ns_prefixes = {
|
||||
|
|
@ -54,5 +54,6 @@ local parse_xml = (function()
|
|||
end;
|
||||
end)();
|
||||
|
||||
parse = parse_xml;
|
||||
return _M;
|
||||
return {
|
||||
parse = parse_xml;
|
||||
};
|
||||
|
|
|
|||
|
|
@ -24,7 +24,7 @@ local lxp_supports_bytecount = not not lxp.new({}).getcurrentbytecount;
|
|||
|
||||
local default_stanza_size_limit = 1024*1024*10; -- 10MB
|
||||
|
||||
module "xmppstream"
|
||||
local _ENV = nil;
|
||||
|
||||
local new_parser = lxp.new;
|
||||
|
||||
|
|
@ -40,12 +40,9 @@ local xmlns_streams = "http://etherx.jabber.org/streams";
|
|||
local ns_separator = "\1";
|
||||
local ns_pattern = "^([^"..ns_separator.."]*)"..ns_separator.."?(.*)$";
|
||||
|
||||
_M.ns_separator = ns_separator;
|
||||
_M.ns_pattern = ns_pattern;
|
||||
|
||||
local function dummy_cb() end
|
||||
|
||||
function new_sax_handlers(session, stream_callbacks, cb_handleprogress)
|
||||
local function new_sax_handlers(session, stream_callbacks, cb_handleprogress)
|
||||
local xml_handlers = {};
|
||||
|
||||
local cb_streamopened = stream_callbacks.streamopened;
|
||||
|
|
@ -224,7 +221,7 @@ function new_sax_handlers(session, stream_callbacks, cb_handleprogress)
|
|||
return xml_handlers, { reset = reset, set_session = set_session };
|
||||
end
|
||||
|
||||
function new(session, stream_callbacks, stanza_size_limit)
|
||||
local function new(session, stream_callbacks, stanza_size_limit)
|
||||
-- Used to track parser progress (e.g. to enforce size limits)
|
||||
local n_outstanding_bytes = 0;
|
||||
local handle_progress;
|
||||
|
|
@ -281,4 +278,9 @@ function new(session, stream_callbacks, stanza_size_limit)
|
|||
};
|
||||
end
|
||||
|
||||
return _M;
|
||||
return {
|
||||
ns_separator = ns_separator;
|
||||
ns_pattern = ns_pattern;
|
||||
new_sax_handlers = new_sax_handlers;
|
||||
new = new;
|
||||
};
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue