dofile("noglobals.lua") local qptest = "qptest.bin" local eqptest = "qptest.bin2" local dqptest = "qptest.bin3" local b64test = "luasocket.exe" local eb64test = "b64test.bin" local db64test = "b64test.bin2" -- from Machado de Assis, "A Mão e a Rosa" local mao = [[ Cursavam estes dois moços a academia de S. Paulo, estando Luís Alves no quarto ano e Estêvão no terceiro. Conheceram-se na academia, e ficaram amigos íntimos, tanto quanto podiam sê-lo dois espíritos diferentes, ou talvez por isso mesmo que o eram. Estêvão, dotado de extrema sensibilidade, e não menor fraqueza de ânimo, afetuoso e bom, não daquela bondade varonil, que é apanágio de uma alma forte, mas dessa outra bondade mole e de cera, que vai à mercê de todas as circunstâncias, tinha, além de tudo isso, o infortúnio de trazer ainda sobre o nariz os óculos cor-de-rosa de suas virginais ilusões. Luís Alves via bem com os olhos da cara. Não era mau rapaz, mas tinha o seu grão de egoísmo, e se não era incapaz de afeições, sabia regê-las, moderá-las, e sobretudo guiá-las ao seu próprio interesse. Entre estes dois homens travara-se amizade íntima, nascida para um na simpatia, para outro no costume. Eram eles os naturais confidentes um do outro, com a diferença que Luís Alves dava menos do que recebia, e, ainda assim, nem tudo o que dava exprimia grande confiança. ]] local fail = function(s) s = s or "failed" assert(nil, s) end local readfile = function(name) local f = io.open(name, "r") if not f then return nil end local s = f:read("*a") f:close() return s end local function transform(input, output, filter) local fi, err = io.open(input, "rb") if not fi then fail(err) end local fo, err = io.open(output, "wb") if not fo then fail(err) end while 1 do local chunk = fi:read(math.random(0, 256)) fo:write(filter(chunk)) if not chunk then break end end fi:close() fo:close() end local function compare(input, output) local original = readfile(input) local recovered = readfile(output) if original ~= recovered then fail("recovering failed") else print("ok") end end local function encode_qptest(mode) local encode = socket.mime.encode("quoted-printable", mode) local split = socket.mime.wrap("quoted-printable") local chain = socket.mime.chain(encode, split) transform(qptest, eqptest, chain) end local function compare_qptest() compare(qptest, dqptest) end local function decode_qptest() local decode = socket.mime.decode("quoted-printable") transform(eqptest, dqptest, decode) end local function create_qptest() local f, err = io.open(qptest, "wb") if not f then fail(err) end -- try all characters for i = 0, 255 do f:write(string.char(i)) end -- try all characters and different line sizes for i = 0, 255 do for j = 0, i do f:write(string.char(i)) end f:write("\r\n") end -- test latin text f:write(mao) -- force soft line breaks and treatment of space/tab in end of line local tab f:write(string.gsub(mao, "(%s)", function(c) if tab then tab = nil return "\t" else tab = 1 return " " end end)) -- test crazy end of line conventions local eol = { "\r\n", "\r", "\n", "\n\r" } local which = 0 f:write(string.gsub(mao, "(\n)", function(c) which = which + 1 if which > 4 then which = 1 end return eol[which] end)) for i = 1, 4 do for j = 1, 4 do f:write(eol[i]) f:write(eol[j]) end end -- try long spaced and tabbed lines f:write("\r\n") for i = 0, 255 do f:write(string.char(9)) end f:write("\r\n") for i = 0, 255 do f:write(' ') end f:write("\r\n") for i = 0, 255 do f:write(string.char(9),' ') end f:write("\r\n") for i = 0, 255 do f:write(' ',string.char(32)) end f:write("\r\n") f:close() end local function cleanup_qptest() os.remove(qptest) os.remove(eqptest) os.remove(dqptest) end local function encode_b64test() local e1 = socket.mime.encode("base64") local e2 = socket.mime.encode("base64") local e3 = socket.mime.encode("base64") local e4 = socket.mime.encode("base64") local sp4 = socket.mime.wrap("character") local sp3 = socket.mime.wrap("character", 59) local sp2 = socket.mime.wrap("character", 30) local sp1 = socket.mime.wrap("character", 27) local chain = socket.mime.chain(e1, sp1, e2, sp2, e3, sp3, e4, sp4) transform(b64test, eb64test, chain) end local function decode_b64test() local d1 = socket.mime.decode("base64") local d2 = socket.mime.decode("base64") local d3 = socket.mime.decode("base64") local d4 = socket.mime.decode("base64") local chain = socket.mime.chain(d1, d2, d3, d4) transform(eb64test, db64test, chain) end local function cleanup_b64test() os.remove(eb64test) os.remove(db64test) end local function compare_b64test() compare(b64test, db64test) end local function padcheck(original, encoded) local e = (socket.mime.b64(original)) local d = (socket.mime.unb64(encoded)) if e ~= encoded then fail("encoding failed") end if d ~= original then fail("decoding failed") end end local function chunkcheck(original, encoded) local len = string.len(original) for i = 0, len do local a = string.sub(original, 1, i) local b = string.sub(original, i+1) local e, r = socket.mime.b64(a, b) local f = (socket.mime.b64(r)) if (e .. f ~= encoded) then fail(e .. f) end end end local function padding_b64test() padcheck("a", "YQ==") padcheck("ab", "YWI=") padcheck("abc", "YWJj") padcheck("abcd", "YWJjZA==") padcheck("abcde", "YWJjZGU=") padcheck("abcdef", "YWJjZGVm") padcheck("abcdefg", "YWJjZGVmZw==") padcheck("abcdefgh", "YWJjZGVmZ2g=") padcheck("abcdefghi", "YWJjZGVmZ2hp") padcheck("abcdefghij", "YWJjZGVmZ2hpag==") chunkcheck("abcdefgh", "YWJjZGVmZ2g=") chunkcheck("abcdefghi", "YWJjZGVmZ2hp") chunkcheck("abcdefghij", "YWJjZGVmZ2hpag==") print("ok") end local t = socket.time() create_qptest() encode_qptest() decode_qptest() compare_qptest() encode_qptest("binary") decode_qptest() compare_qptest() cleanup_qptest() encode_b64test() decode_b64test() compare_b64test() cleanup_b64test() padding_b64test() print(string.format("done in %.2fs", socket.time() - t))