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

util.format: Ensure metatable __tostring results are also sanitized

Browse source
This commit is contained in:
Kim Alvefur 2021-12-13 16:34:55 +01:00
parent 283042d7c3
commit 58c9115351
3 changed files with 20 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

16
spec/util_format_spec.lua
View file

@ -780,96 +780,112 @@ describe("util.format", function()
describe("to %c", function ()
it("works", function ()
assert.matches("[table: 0[xX]%x+]", format("%c", { }))
assert.equal("[foo \226\144\129\226\144\130\226\144\131 bar]", format("%c", setmetatable({},{__tostring=function ()return "foo \1\2\3 bar"end})))
end);
end);
describe("to %d", function ()
it("works", function ()
assert.matches("[table: 0[xX]%x+]", format("%d", { }))
assert.equal("[foo \226\144\129\226\144\130\226\144\131 bar]", format("%d", setmetatable({},{__tostring=function ()return "foo \1\2\3 bar"end})))
end);
end);
describe("to %i", function ()
it("works", function ()
assert.matches("[table: 0[xX]%x+]", format("%i", { }))
assert.equal("[foo \226\144\129\226\144\130\226\144\131 bar]", format("%i", setmetatable({},{__tostring=function ()return "foo \1\2\3 bar"end})))
end);
end);
describe("to %o", function ()
it("works", function ()
assert.matches("[table: 0[xX]%x+]", format("%o", { }))
assert.equal("[foo \226\144\129\226\144\130\226\144\131 bar]", format("%o", setmetatable({},{__tostring=function ()return "foo \1\2\3 bar"end})))
end);
end);
describe("to %u", function ()
it("works", function ()
assert.matches("[table: 0[xX]%x+]", format("%u", { }))
assert.equal("[foo \226\144\129\226\144\130\226\144\131 bar]", format("%u", setmetatable({},{__tostring=function ()return "foo \1\2\3 bar"end})))
end);
end);
describe("to %x", function ()
it("works", function ()
assert.matches("[table: 0[xX]%x+]", format("%x", { }))
assert.equal("[foo \226\144\129\226\144\130\226\144\131 bar]", format("%x", setmetatable({},{__tostring=function ()return "foo \1\2\3 bar"end})))
end);
end);
describe("to %X", function ()
it("works", function ()
assert.matches("[table: 0[xX]%x+]", format("%X", { }))
assert.equal("[foo \226\144\129\226\144\130\226\144\131 bar]", format("%X", setmetatable({},{__tostring=function ()return "foo \1\2\3 bar"end})))
end);
end);
describe("to %a", function ()
it("works", function ()
assert.matches("[table: 0[xX]%x+]", format("%a", { }))
assert.equal("[foo \226\144\129\226\144\130\226\144\131 bar]", format("%a", setmetatable({},{__tostring=function ()return "foo \1\2\3 bar"end})))
end);
end);
describe("to %A", function ()
it("works", function ()
assert.matches("[table: 0[xX]%x+]", format("%A", { }))
assert.equal("[foo \226\144\129\226\144\130\226\144\131 bar]", format("%A", setmetatable({},{__tostring=function ()return "foo \1\2\3 bar"end})))
end);
end);
describe("to %e", function ()
it("works", function ()
assert.matches("[table: 0[xX]%x+]", format("%e", { }))
assert.equal("[foo \226\144\129\226\144\130\226\144\131 bar]", format("%e", setmetatable({},{__tostring=function ()return "foo \1\2\3 bar"end})))
end);
end);
describe("to %E", function ()
it("works", function ()
assert.matches("[table: 0[xX]%x+]", format("%E", { }))
assert.equal("[foo \226\144\129\226\144\130\226\144\131 bar]", format("%E", setmetatable({},{__tostring=function ()return "foo \1\2\3 bar"end})))
end);
end);
describe("to %f", function ()
it("works", function ()
assert.matches("[table: 0[xX]%x+]", format("%f", { }))
assert.equal("[foo \226\144\129\226\144\130\226\144\131 bar]", format("%f", setmetatable({},{__tostring=function ()return "foo \1\2\3 bar"end})))
end);
end);
describe("to %g", function ()
it("works", function ()
assert.matches("[table: 0[xX]%x+]", format("%g", { }))
assert.equal("[foo \226\144\129\226\144\130\226\144\131 bar]", format("%g", setmetatable({},{__tostring=function ()return "foo \1\2\3 bar"end})))
end);
end);
describe("to %G", function ()
it("works", function ()
assert.matches("[table: 0[xX]%x+]", format("%G", { }))
assert.equal("[foo \226\144\129\226\144\130\226\144\131 bar]", format("%G", setmetatable({},{__tostring=function ()return "foo \1\2\3 bar"end})))
end);
end);
describe("to %q", function ()
it("works", function ()
assert.matches("{ }", format("%q", { }))
assert.equal("{ }", format("%q", setmetatable({},{__tostring=function ()return "foo \1\2\3 bar"end})))
end);
end);
describe("to %s", function ()
it("works", function ()
assert.matches("table: 0[xX]%x+", format("%s", { }))
assert.equal("foo \226\144\129\226\144\130\226\144\131 bar", format("%s", setmetatable({},{__tostring=function ()return "foo \1\2\3 bar"end})))
end);
end);

3
tools/generate_format_spec.lua
View file

@ -18,13 +18,14 @@ local example_values = {
["function"] = { function() end };
-- ["userdata"] = {};
["thread"] = { coroutine.create(function() end) };
["table"] = { {} };
["table"] = { {}, setmetatable({},{__tostring=function ()return "foo \1\2\3 bar"end}) };
};
local example_strings = setmetatable({
["nil"] = { "nil" };
["function"] = { "function() end" };
["number"] = { "97"; "-12345"; "1.5"; "73786976294838206464"; "math.huge"; "2147483647" };
["thread"] = { "coroutine.create(function() end)" };
["table"] = { "{ }", "setmetatable({},{__tostring=function ()return \"foo \\1\\2\\3 bar\"end})" }
}, { __index = function() return {} end });
for _, lua_type in ipairs(types) do
print(string.format("\t\tdescribe(\"%s\", function ()", lua_type));

3
util/format.lua
View file

@ -70,7 +70,8 @@ local function format(formatstring, ...)
-- No UTF-8 or control characters, assumed to be the common case.
return
elseif option == "s" and t ~= "string" then
args[i] = tostring(arg);
arg = tostring(arg);
t = "string";
end
if option ~= "s" and option ~= "q" and option ~= "p" then