mirrors/prosody
1
0
Fork 0
mirror of https://github.com/bjc/prosody.git synced 2025-04-03 21:27:38 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

mod_http_file_share: Add support for daily upload quotas.

Browse source 
Daily instead of total quotas, should be more efficient to calculate.
Still O(n), but a smaller n. Less affected by total retention period.
This commit is contained in:
Kim Alvefur 2021-01-31 14:43:42 +01:00
parent 8767624de1
commit 55cb283433
1 changed files with 19 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

19
plugins/mod_http_file_share.lua
View file

@ -35,6 +35,7 @@ local file_size_limit = module:get_option_number(module.name .. "_size_limit", 1
local file_types = module:get_option_set(module.name .. "_allowed_file_types", {});
local safe_types = module:get_option_set(module.name .. "_safe_file_types", {"image/*","video/*","audio/*","text/plain"});
local expiry = module:get_option_number(module.name .. "_expires_after", 7 * 86400);
local daily_quota = module:get_option_number(module.name .. "_daily_quota", file_size_limit*10); -- 100 MB / day
local access = module:get_option_set(module.name .. "_access", {});
@ -55,6 +56,7 @@ local upload_errors = errors.init(module.name, namespace, {
extra = {tag = st.stanza("file-too-large", {xmlns = namespace}):tag("max-file-size"):text(tostring(file_size_limit)) };
};
filesizefmt = { type = "modify"; condition = "bad-request"; text = "File size must be positive integer"; };
quota = { type = "wait"; condition = "resource-constraint"; text = "Daily quota reached"; };
});
local upload_cache = cache.new(1024);
@ -66,6 +68,18 @@ local function get_filename(slot, create)
return dm.getpath(slot, module.host, module.name, "bin", create)
end
-- TODO cache
function get_daily_quota(uploader)
local iter, err = uploads:find(nil, {with = uploader; start = os.time() - 86400});
if not iter then return iter, err; end
local total_bytes = 0;
for _, slot in iter do
local size = tonumber(slot.attr.size);
if size then total_bytes = total_bytes + size; end
end
return total_bytes;
end
function may_upload(uploader, filename, filesize, filetype) -- > boolean, error
local uploader_host = jid.host(uploader);
if not ((access:empty() and prosody.hosts[uploader_host]) or access:contains(uploader) or access:contains(uploader_host)) then
@ -84,6 +98,11 @@ function may_upload(uploader, filename, filesize, filetype) -- > boolean, error
return false, upload_errors.new("filesize");
end
local uploader_quota = get_daily_quota(uploader);
if uploader_quota + filesize > daily_quota then
return false, upload_errors.new("quota");
end
if not ( file_types:empty() or file_types:contains(filetype) or file_types:contains(filetype:gsub("/.*", "/*")) ) then
return false, upload_errors.new("filetype");
end