smtp.c was eatten by mime.c

8 files changed, 54 insertions, 42 deletions

diff --git a/src/ftp.lua b/src/ftp.lua
index 87a1e4e..f3b8a5e 100644
--- a/src/ftp.lua
+++ b/src/ftp.lua
@@ -14,11 +14,6 @@ local url = require("url")
 local tp = require("tp")
 
 -----------------------------------------------------------------------------
--- Setup namespace
------------------------------------------------------------------------------
-_LOADED["ftp"] = getfenv(1)
-
------------------------------------------------------------------------------
 -- Program constants
 -----------------------------------------------------------------------------
 -- timeout in seconds before the program gives up on a connection
diff --git a/src/http.lua b/src/http.lua
index b341ebb..3bd4d6a 100644
--- a/src/http.lua
+++ b/src/http.lua
@@ -14,11 +14,6 @@ local mime = require("mime")
 local url = require("url")
 
 -----------------------------------------------------------------------------
--- Setup namespace
--------------------------------------------------------------------------------
-_LOADED["http"] = getfenv(1)
-
------------------------------------------------------------------------------
 -- Program constants
 -----------------------------------------------------------------------------
 -- connection timeout in seconds
diff --git a/src/ltn12.lua b/src/ltn12.lua
index 09771d4..04656cb 100644
--- a/src/ltn12.lua
+++ b/src/ltn12.lua
@@ -5,11 +5,6 @@
 -- RCS ID: $Id$
 -----------------------------------------------------------------------------
 
------------------------------------------------------------------------------
--- Setup namespace
------------------------------------------------------------------------------
-_LOADED["ltn12"] = getfenv(1)
-
 filter = {}
 source = {}
 sink = {}
diff --git a/src/mime.c b/src/mime.c
index 7810580..5b1d5e8 100644
--- a/src/mime.c
+++ b/src/mime.c
@@ -28,7 +28,9 @@ static int mime_global_qp(lua_State *L);
 static int mime_global_unqp(lua_State *L);
 static int mime_global_qpwrp(lua_State *L);
 static int mime_global_eol(lua_State *L);
+static int mime_global_dot(lua_State *L);
 
+static size_t dot(int c, size_t state, luaL_Buffer *buffer);
 static void b64setup(UC *b64unbase);
 static size_t b64encode(UC c, UC *input, size_t size, luaL_Buffer *buffer);
 static size_t b64pad(const UC *input, size_t size, luaL_Buffer *buffer);
@@ -43,6 +45,7 @@ static size_t qppad(UC *input, size_t size, luaL_Buffer *buffer);
 
 /* code support functions */
 static luaL_reg func[] = {
+    { "dot", mime_global_dot },
     { "b64", mime_global_b64 },
     { "eol", mime_global_eol },
     { "qp", mime_global_qp },
@@ -659,3 +662,48 @@ static int mime_global_eol(lua_State *L)
     lua_pushnumber(L, ctx);
     return 2;
 }
+
+/*-------------------------------------------------------------------------*\
+* Takes one byte and stuff it if needed. 
+\*-------------------------------------------------------------------------*/
+static size_t dot(int c, size_t state, luaL_Buffer *buffer)
+{
+    luaL_putchar(buffer, c);
+    switch (c) {
+        case '\r': 
+            return 1;
+        case '\n': 
+            return (state == 1)? 2: 0; 
+        case '.':  
+            if (state == 2) 
+                luaL_putchar(buffer, '.');
+        default:
+            return 0;
+    }
+}
+
+/*-------------------------------------------------------------------------*\
+* Incrementally applies smtp stuffing to a string
+* A, n = dot(l, D)
+\*-------------------------------------------------------------------------*/
+static int mime_global_dot(lua_State *L)
+{
+    size_t isize = 0, state = (size_t) luaL_checknumber(L, 1);
+    const char *input = luaL_optlstring(L, 2, NULL, &isize);
+    const char *last = input + isize;
+    luaL_Buffer buffer;
+    /* end-of-input blackhole */
+    if (!input) {
+        lua_pushnil(L);
+        lua_pushnumber(L, 2);
+        return 2;
+    }
+    /* process all input */
+    luaL_buffinit(L, &buffer);
+    while (input < last) 
+        state = dot(*input++, state, &buffer);
+    luaL_pushresult(&buffer);
+    lua_pushnumber(L, state);
+    return 2;
+}
+
diff --git a/src/mime.lua b/src/mime.lua
index 4fade3b..345bd25 100644
--- a/src/mime.lua
+++ b/src/mime.lua
@@ -11,11 +11,6 @@
 local mime = requirelib("mime", "luaopen_mime", getfenv(1))
 local ltn12 = require("ltn12")
 
------------------------------------------------------------------------------
--- Setup namespace 
------------------------------------------------------------------------------
-_LOADED["mime"] = mime
-
 -- encode, decode and wrap algorithm tables
 encodet = {}
 decodet = {}
@@ -73,3 +68,8 @@ wrap = choose(wrapt)
 function normalize(marker)
     return ltn12.filter.cycle(eol, 0, marker)
 end
+
+-- high level stuffing filter
+function stuff()
+    return ltn12.filter.cycle(dot, 2)
+end
diff --git a/src/smtp.lua b/src/smtp.lua
index 2ea6097..1708053 100644
--- a/src/smtp.lua
+++ b/src/smtp.lua
@@ -8,17 +8,11 @@
 -----------------------------------------------------------------------------
 -- Load required modules
 -----------------------------------------------------------------------------
-local smtp = requirelib("smtp", "luaopen_smtp", getfenv(1))
 local socket = require("socket")
 local ltn12 = require("ltn12")
 local mime = require("mime")
 local tp = require("tp")
 
------------------------------------------------------------------------------
--- Setup namespace 
------------------------------------------------------------------------------
-_LOADED["smtp"] = smtp
-
 -- timeout for connection
 TIMEOUT = 60
 -- default server used to send e-mails
@@ -31,11 +25,6 @@ DOMAIN = os.getenv("SERVER_NAME") or "localhost"
 -- default time zone (means we don't know)
 ZONE = "-0000"
 
--- high level stuffing filter
-function stuff()
-    return ltn12.filter.cycle(dot, 2)
-end
-
 ---------------------------------------------------------------------------
 -- Low level SMTP API
 -----------------------------------------------------------------------------
@@ -110,7 +99,7 @@ function metat.__index:send(mailt)
     else
         self:rcpt(mailt.rcpt)
     end
-    self:data(ltn12.source.chain(mailt.source, stuff()), mailt.step)
+    self:data(ltn12.source.chain(mailt.source, mime.stuff()), mailt.step)
 end
 
 function open(server, port)
diff --git a/src/tp.lua b/src/tp.lua
index 731191e..7b581b9 100644
--- a/src/tp.lua
+++ b/src/tp.lua
@@ -12,11 +12,6 @@ local socket = require("socket")
 local ltn12 = require("ltn12")
 
 -----------------------------------------------------------------------------
--- Setup namespace
------------------------------------------------------------------------------
-_LOADED["tp"] = getfenv(1)
-
------------------------------------------------------------------------------
 -- Program constants
 -----------------------------------------------------------------------------
 TIMEOUT = 60
diff --git a/src/url.lua b/src/url.lua
index c708e19..3fd0aa6 100644
--- a/src/url.lua
+++ b/src/url.lua
@@ -6,11 +6,6 @@
 -----------------------------------------------------------------------------
 
 -----------------------------------------------------------------------------
--- Setup namespace
------------------------------------------------------------------------------
-_LOADED["url"] = getfenv(1)
-
------------------------------------------------------------------------------
 -- Encodes a string into its escaped hexadecimal representation
 -- Input 
 --   s: binary string to be encoded