1 files changed, 51 insertions, 0 deletions

diff --git a/qmail-inject.c b/qmail-inject.c
index 43b53fe..59212d4 100644
--- a/qmail-inject.c
+++ b/qmail-inject.c
@@ -1,4 +1,8 @@
+#include <sys/time.h>
+#include <sys/types.h>
 #include <unistd.h>
+#include <stdint.h>
+#include <pwd.h>
 #include "sig.h"
 #include "substdio.h"
 #include "stralloc.h"
@@ -23,6 +27,7 @@
 #include "auto_qmail.h"
 #include "newfield.h"
 #include "constmap.h"
+#include "base64.h"
 
 #define LINELEN 80
 
@@ -515,6 +520,42 @@ void dodefaultreturnpath()
  if (token822_unparse(&defaultreturnpath,&hfrewrite,LINELEN) != 1) die_nomem();
 }
 
+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;
+
+  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;
+
+  /* port */
+  buf.s[buf.len]   = 0;
+  buf.s[buf.len+1] = 0;
+  buf.len += 2;
+
+  buf.s[buf.len]  = 127;
+  buf.s[buf.len+1] = 0;
+  buf.s[buf.len+2] = 0;
+  buf.s[buf.len+3] = 1;
+  buf.len += 4;
+
+  if (b64encode(&buf, &base64_buf) < 0) die_nomem();
+  if (!stralloc_0(&base64_buf)) die_nomem();
+  return &base64_buf;
+}
+
 int flagmft = 0;
 stralloc mft = {0};
 struct constmap mapmft;
@@ -635,6 +676,16 @@ void finishheader()
    finishmft();
   }
 
+ const uid_t uid = geteuid();
+ const struct passwd *pw = getpwuid(uid);
+ if (pw) {
+   puts("X-UD-Smtp-Session: ");
+   puts(pw->pw_name);
+   puts("@");
+   puts(client_get_session_id()->s);
+   puts("\n");
+ }
+
  savedh_print();
 }