1 files changed, 55 insertions, 2 deletions

diff --git a/qmail-smtpd.c b/qmail-smtpd.c
index 95993eb..6b4262f 100644
--- a/qmail-smtpd.c
+++ b/qmail-smtpd.c
@@ -1,3 +1,4 @@
+#include <arpa/inet.h>
 #include "sig.h"
 #include "readwrite.h"
 #include "stralloc.h"
@@ -594,7 +595,7 @@ void mailfrom_parms(arg) char *arg;
            if (case_starts(mfparms.s,"SMTPUTF8")) smtputf8 = 1;
 #endif
            if (case_starts(mfparms.s,"SIZE=")) if (mailfrom_size(mfparms.s+5)) { flagsize = 1; return; }
-           if (case_starts(mfparms.s,"AUTH=")) mailfrom_auth(mfparms.s+5,mfparms.len-5);  
+           if (case_starts(mfparms.s,"AUTH=")) mailfrom_auth(mfparms.s+5,mfparms.len-5);
            if (!stralloc_copys(&mfparms,"")) die_nomem;
         }
         else
@@ -603,6 +604,55 @@ void mailfrom_parms(arg) char *arg;
     }
 }
 
+#define V4MAPPREFIX "::ffff:"
+static const stralloc *client_get_session_id()
+{
+  static stralloc buf = {0}, base64_buf = {0};
+  const char *tmp;
+  struct timeval tv;
+  uint64_t timestamp;
+  unsigned int i;
+  unsigned long port;
+  int family = AF_INET;
+
+  if (base64_buf.s)
+    return &base64_buf;
+
+  if (!stralloc_ready(&buf, 24)) die_nomem();
+
+  /* add lowest 48 bits of the timestamp. this gives us a bit less than
+     9 years until it wraps */
+  gettimeofday(&tv, NULL);
+  timestamp = tv.tv_usec + (long long)tv.tv_sec * 1000ULL*1000ULL;
+  for (i = 0; i < 48; i += 8)
+    buf.s[buf.len++] = (timestamp >> i) & 0xff;
+
+  if ((tmp = getenv("TCPREMOTEPORT")) != NULL && scan_ulong(tmp, &port))
+  {
+    buf.s[buf.len]   = port & 0xff;
+    buf.s[buf.len+1] = (port >> 8) & 0xff;
+  }
+  buf.len += 2;
+
+  family = ((tmp = getenv("PROTO")) && strcmp(tmp, "TCP6") == 0)
+    ? AF_INET6 : AF_INET;
+  if ((tmp = getenv("TCPREMOTEIP")))
+  {
+    if (family == AF_INET6 && !strncmp(tmp, V4MAPPREFIX, strlen(V4MAPPREFIX)))
+    {
+      tmp += strlen(V4MAPPREFIX);
+      family = AF_INET;
+    }
+    (void)inet_pton(family, tmp, buf.s + buf.len);
+    buf.len += (family == AF_INET) ? 4 : 16;
+  }
+
+  if (b64encode(&buf, &base64_buf) < 0) die_nomem();
+  if (!stralloc_0(&base64_buf)) die_nomem();
+  if (!env_put2("SMTPSESSION", base64_buf.s)) die_nomem();
+  return &base64_buf;
+}
+
 void smtp_helo(arg) char *arg;
 {
   if(!spp_helo(arg)) return;
@@ -878,7 +928,10 @@ void smtp_data(arg) char *arg; {
     protocol = utf8proto.s;
   }
 
-  received(&qqt,protocol,local,remoteip,remotehost,remoteinfo,fakehelo);
+  if (flagauth)
+    received_authed(&qqt,protocol,local,remoteinfo,client_get_session_id()->s);
+  else
+    received(&qqt,protocol,local,remoteip,remotehost,remoteinfo,fakehelo);
   qmail_put(&qqt,sppheaders.s,sppheaders.len); /* set in qmail-spp.c */
   spp_rset();
   blast(&hops);