source/texk/web2c/luatexdir/luasocket/src/smtp.lua

   1 -----------------------------------------------------------------------------
   2 -- SMTP client support for the Lua language.
   3 -- LuaSocket toolkit.
   4 -- Author: Diego Nehab
   5 -----------------------------------------------------------------------------
   6
   7 -----------------------------------------------------------------------------
   8 -- Declare module and import dependencies
   9 -----------------------------------------------------------------------------
  10 local base = _G
  11 local coroutine = require("coroutine")
  12 local string = require("string")
  13 local math = require("math")
  14 local os = require("os")
  15 local socket = require("socket")
  16 local tp = require("socket.tp")
  17 local ltn12 = require("ltn12")
  18 local headers = require("socket.headers")
  19 local mime = require("mime")
  20 module("socket.smtp")
  21
  22 -----------------------------------------------------------------------------
  23 -- Program constants
  24 -----------------------------------------------------------------------------
  25 -- timeout for connection
  26 TIMEOUT = 60
  27 -- default server used to send e-mails
  28 SERVER = "localhost"
  29 -- default port
  30 PORT = 25
  31 -- domain used in HELO command and default sendmail
  32 -- If we are under a CGI, try to get from environment
  33 DOMAIN = os.getenv("SERVER_NAME") or "localhost"
  34 -- default time zone (means we don't know)
  35 ZONE = "-0000"
  36
  37 ---------------------------------------------------------------------------
  38 -- Low level SMTP API
  39 -----------------------------------------------------------------------------
  40 local metat = { __index = {} }
  41
  42 function metat.__index:greet(domain)
  43     self.try(self.tp:check("2.."))
  44     self.try(self.tp:command("EHLO", domain or DOMAIN))
  45     return socket.skip(1, self.try(self.tp:check("2..")))
  46 end
  47
  48 function metat.__index:mail(from)
  49     self.try(self.tp:command("MAIL", "FROM:" .. from))
  50     return self.try(self.tp:check("2.."))
  51 end
  52
  53 function metat.__index:rcpt(to)
  54     self.try(self.tp:command("RCPT", "TO:" .. to))
  55     return self.try(self.tp:check("2.."))
  56 end
  57
  58 function metat.__index:data(src, step)
  59     self.try(self.tp:command("DATA"))
  60     self.try(self.tp:check("3.."))
  61     self.try(self.tp:source(src, step))
  62     self.try(self.tp:send("\r\n.\r\n"))
  63     return self.try(self.tp:check("2.."))
  64 end
  65
  66 function metat.__index:quit()
  67     self.try(self.tp:command("QUIT"))
  68     return self.try(self.tp:check("2.."))
  69 end
  70
  71 function metat.__index:close()
  72     return self.tp:close()
  73 end
  74
  75 function metat.__index:login(user, password)
  76     self.try(self.tp:command("AUTH", "LOGIN"))
  77     self.try(self.tp:check("3.."))
  78     self.try(self.tp:send(mime.b64(user) .. "\r\n"))
  79     self.try(self.tp:check("3.."))
  80     self.try(self.tp:send(mime.b64(password) .. "\r\n"))
  81     return self.try(self.tp:check("2.."))
  82 end
  83
  84 function metat.__index:plain(user, password)
  85     local auth = "PLAIN " .. mime.b64("\0" .. user .. "\0" .. password)
  86     self.try(self.tp:command("AUTH", auth))
  87     return self.try(self.tp:check("2.."))
  88 end
  89
  90 function metat.__index:auth(user, password, ext)
  91     if not user or not password then return 1 end
  92     if string.find(ext, "AUTH[^\n]+LOGIN") then
  93         return self:login(user, password)
  94     elseif string.find(ext, "AUTH[^\n]+PLAIN") then
  95         return self:plain(user, password)
  96     else
  97         self.try(nil, "authentication not supported")
  98     end
  99 end
 100
 101 -- send message or throw an exception
 102 function metat.__index:send(mailt)
 103     self:mail(mailt.from)
 104     if base.type(mailt.rcpt) == "table" then
 105         for i,v in base.ipairs(mailt.rcpt) do
 106             self:rcpt(v)
 107         end
 108     else
 109         self:rcpt(mailt.rcpt)
 110     end
 111     self:data(ltn12.source.chain(mailt.source, mime.stuff()), mailt.step)
 112 end
 113
 114 function open(server, port, create)
 115     local tp = socket.try(tp.connect(server or SERVER, port or PORT,
 116         TIMEOUT, create))
 117     local s = base.setmetatable({tp = tp}, metat)
 118     -- make sure tp is closed if we get an exception
 119     s.try = socket.newtry(function()
 120         s:close()
 121     end)
 122     return s
 123 end
 124
 125 -- convert headers to lowercase
 126 local function lower_headers(headers)
 127     local lower = {}
 128     for i,v in base.pairs(headers or lower) do
 129         lower[string.lower(i)] = v
 130     end
 131     return lower
 132 end
 133
 134 ---------------------------------------------------------------------------
 135 -- Multipart message source
 136 -----------------------------------------------------------------------------
 137 -- returns a hopefully unique mime boundary
 138 local seqno = 0
 139 local function newboundary()
 140     seqno = seqno + 1
 141     return string.format('%s%05d==%05u', os.date('%d%m%Y%H%M%S'),
 142         math.random(0, 99999), seqno)
 143 end
 144
 145 -- send_message forward declaration
 146 local send_message
 147
 148 -- yield the headers all at once, it's faster
 149 local function send_headers(tosend)
 150     local canonic = headers.canonic
 151     local h = "\r\n"
 152     for f,v in base.pairs(tosend) do
 153         h = (canonic[f] or f) .. ': ' .. v .. "\r\n" .. h
 154     end
 155     coroutine.yield(h)
 156 end
 157
 158 -- yield multipart message body from a multipart message table
 159 local function send_multipart(mesgt)
 160     -- make sure we have our boundary and send headers
 161     local bd = newboundary()
 162     local headers = lower_headers(mesgt.headers or {})
 163     headers['content-type'] = headers['content-type'] or 'multipart/mixed'
 164     headers['content-type'] = headers['content-type'] ..
 165         '; boundary="' ..  bd .. '"'
 166     send_headers(headers)
 167     -- send preamble
 168     if mesgt.body.preamble then
 169         coroutine.yield(mesgt.body.preamble)
 170         coroutine.yield("\r\n")
 171     end
 172     -- send each part separated by a boundary
 173     for i, m in base.ipairs(mesgt.body) do
 174         coroutine.yield("\r\n--" .. bd .. "\r\n")
 175         send_message(m)
 176     end
 177     -- send last boundary
 178     coroutine.yield("\r\n--" .. bd .. "--\r\n\r\n")
 179     -- send epilogue
 180     if mesgt.body.epilogue then
 181         coroutine.yield(mesgt.body.epilogue)
 182         coroutine.yield("\r\n")
 183     end
 184 end
 185
 186 -- yield message body from a source
 187 local function send_source(mesgt)
 188     -- make sure we have a content-type
 189     local headers = lower_headers(mesgt.headers or {})
 190     headers['content-type'] = headers['content-type'] or
 191         'text/plain; charset="iso-8859-1"'
 192     send_headers(headers)
 193     -- send body from source
 194     while true do
 195         local chunk, err = mesgt.body()
 196         if err then coroutine.yield(nil, err)
 197         elseif chunk then coroutine.yield(chunk)
 198         else break end
 199     end
 200 end
 201
 202 -- yield message body from a string
 203 local function send_string(mesgt)
 204     -- make sure we have a content-type
 205     local headers = lower_headers(mesgt.headers or {})
 206     headers['content-type'] = headers['content-type'] or
 207         'text/plain; charset="iso-8859-1"'
 208     send_headers(headers)
 209     -- send body from string
 210     coroutine.yield(mesgt.body)
 211 end
 212
 213 -- message source
 214 function send_message(mesgt)
 215     if base.type(mesgt.body) == "table" then send_multipart(mesgt)
 216     elseif base.type(mesgt.body) == "function" then send_source(mesgt)
 217     else send_string(mesgt) end
 218 end
 219
 220 -- set defaul headers
 221 local function adjust_headers(mesgt)
 222     local lower = lower_headers(mesgt.headers)
 223     lower["date"] = lower["date"] or
 224         os.date("!%a, %d %b %Y %H:%M:%S ") .. (mesgt.zone or ZONE)
 225     lower["x-mailer"] = lower["x-mailer"] or socket._VERSION
 226     -- this can't be overriden
 227     lower["mime-version"] = "1.0"
 228     return lower
 229 end
 230
 231 function message(mesgt)
 232     mesgt.headers = adjust_headers(mesgt)
 233     -- create and return message source
 234     local co = coroutine.create(function() send_message(mesgt) end)
 235     return function()
 236         local ret, a, b = coroutine.resume(co)
 237         if ret then return a, b
 238         else return nil, a end
 239     end
 240 end
 241
 242 ---------------------------------------------------------------------------
 243 -- High level SMTP API
 244 -----------------------------------------------------------------------------
 245 send = socket.protect(function(mailt)
 246     local s = open(mailt.server, mailt.port, mailt.create)
 247     local ext = s:greet(mailt.domain)
 248     s:auth(mailt.user, mailt.password, ext)
 249     s:send(mailt)
 250     s:quit()
 251     return s:close()
 252 end)