luasocket/src/tp.lua

124 lines
3.5 KiB
Lua
Raw Normal View History

-----------------------------------------------------------------------------
-- Unified SMTP/FTP subsystem
-- LuaSocket toolkit.
-- Author: Diego Nehab
-- RCS ID: $Id$
-----------------------------------------------------------------------------
2004-05-30 23:36:22 +02:00
2004-06-04 17:15:45 +02:00
-----------------------------------------------------------------------------
-- Declare module and import dependencies
2004-06-04 17:15:45 +02:00
-----------------------------------------------------------------------------
2004-11-27 08:58:04 +01:00
local base = require("base")
local string = require("string")
2004-06-04 17:15:45 +02:00
local socket = require("socket")
local ltn12 = require("ltn12")
2004-12-23 23:32:12 +01:00
module("socket.tp")
-----------------------------------------------------------------------------
-- Program constants
-----------------------------------------------------------------------------
TIMEOUT = 60
-----------------------------------------------------------------------------
-- Implementation
-----------------------------------------------------------------------------
-- gets server reply (works for SMTP and FTP)
2004-06-18 23:41:44 +02:00
local function get_reply(c)
2004-05-28 08:16:43 +02:00
local code, current, sep
2004-06-18 23:41:44 +02:00
local line, err = c:receive()
local reply = line
if err then return nil, err end
2004-05-28 08:16:43 +02:00
code, sep = socket.skip(2, string.find(line, "^(%d%d%d)(.?)"))
if not code then return nil, "invalid server reply" end
2004-05-28 08:16:43 +02:00
if sep == "-" then -- reply is multiline
repeat
2004-06-18 23:41:44 +02:00
line, err = c:receive()
if err then return nil, err end
2004-05-28 08:16:43 +02:00
current, sep = socket.skip(2, string.find(line, "^(%d%d%d)(.?)"))
reply = reply .. "\n" .. line
-- reply ends with same code
2004-05-28 08:16:43 +02:00
until code == current and sep == " "
end
return code, reply
end
-- metatable for sock object
local metat = { __index = {} }
function metat.__index:check(ok)
2004-06-18 23:41:44 +02:00
local code, reply = get_reply(self.c)
if not code then return nil, reply end
2004-11-27 08:58:04 +01:00
if base.type(ok) ~= "function" then
if base.type(ok) == "table" then
for i, v in base.ipairs(ok) do
if string.find(code, v) then
return base.tonumber(code), reply
end
end
return nil, reply
else
2004-11-27 08:58:04 +01:00
if string.find(code, ok) then return base.tonumber(code), reply
else return nil, reply end
end
2004-11-27 08:58:04 +01:00
else return ok(base.tonumber(code), reply) end
end
function metat.__index:command(cmd, arg)
2004-11-27 08:58:04 +01:00
if arg then
return self.c:send(cmd .. " " .. arg.. "\r\n")
else
return self.c:send(cmd .. "\r\n")
end
end
function metat.__index:sink(snk, pat)
2004-06-18 23:41:44 +02:00
local chunk, err = c:receive(pat)
return snk(chunk, err)
end
function metat.__index:send(data)
2004-06-18 23:41:44 +02:00
return self.c:send(data)
end
function metat.__index:receive(pat)
2004-06-18 23:41:44 +02:00
return self.c:receive(pat)
end
function metat.__index:getfd()
2004-06-18 23:41:44 +02:00
return self.c:getfd()
end
function metat.__index:dirty()
2004-06-18 23:41:44 +02:00
return self.c:dirty()
end
function metat.__index:getcontrol()
2004-06-18 23:41:44 +02:00
return self.c
end
2004-05-26 06:58:32 +02:00
function metat.__index:source(source, step)
2004-06-18 23:41:44 +02:00
local sink = socket.sink("keep-open", self.c)
2004-05-26 06:58:32 +02:00
return ltn12.pump.all(source, sink, step or ltn12.pump.step)
end
2004-06-18 23:41:44 +02:00
-- closes the underlying c
function metat.__index:close()
2004-06-18 23:41:44 +02:00
self.c:close()
return 1
end
2004-06-18 23:41:44 +02:00
-- connect with server and return c object
function connect(host, port, timeout)
local c, e = socket.tcp()
if not c then return nil, e end
c:settimeout(timeout or TIMEOUT)
local r, e = c:connect(host, port)
if not r then
c:close()
return nil, e
end
2004-11-27 08:58:04 +01:00
return base.setmetatable({c = c}, metat)
2004-06-18 23:41:44 +02:00
end
2004-11-27 08:58:04 +01:00
2004-12-23 23:32:12 +01:00
base.getmetatable(_M).__index