Added smtp-data encrypt

pulsar-sd · Nov 21, 2024 · f380d3e · f380d3e
1 parent f5eda56
commit f380d3e
Show file tree

Hide file tree

Showing 3 changed files with 45 additions and 18 deletions.
diff --git a/lib/smtp-interface.js b/lib/smtp-interface.js
@@ -388,31 +388,58 @@ class SMTPInterface {
             envelope.tls = session.tlsOptions;
         }
 
+        let data = stream;
+
         plugins.handler.runHooks('smtp:data', [envelope, session], err => {
             if (err) {
                 stream.destroy();
                 return setImmediate(() => callback(err));
             }
 
-            let sizeLimiter = new SizeLimiter({
-                maxSize: this.options.maxSize
-            });
+            if(!envelope.encrypt) {
+                let sizeLimiter = new SizeLimiter({
+                    maxSize: this.options.maxSize
+                });
+
+                stream.on('error', err => sizeLimiter.emit('error', err));
+                stream.pipe(sizeLimiter);
+
+                this.maildrop.add(envelope, sizeLimiter, (err, message) => {
+                    if (err) {
+                        if (err.name === 'SMTPResponse') {
+                            if (err.responseCode) {
+                                return callback(err);
+                            }
+                            return callback(null, err.message);
+                        }
+                        return callback(err);
+                    }
+                    callback(null, message);
+                });
+            } else {
+                let customStream = envelope.encrypt(stream, result => result);
+                stream = customStream;
 
-            stream.on('error', err => sizeLimiter.emit('error', err));
-            stream.pipe(sizeLimiter);
+                let sizeLimiter = new SizeLimiter({
+                    maxSize: this.options.maxSize
+                });
 
-            this.maildrop.add(envelope, sizeLimiter, (err, message) => {
-                if (err) {
-                    if (err.name === 'SMTPResponse') {
-                        if (err.responseCode) {
-                            return callback(err);
+                stream.on('error', err => sizeLimiter.emit('error', err));
+                stream.pipe(sizeLimiter);
+
+                this.maildrop.add(envelope, sizeLimiter, (err, message) => {
+                    if (err) {
+                        if (err.name === 'SMTPResponse') {
+                            if (err.responseCode) {
+                                return callback(err);
+                            }
+                            return callback(null, err.message);
                         }
-                        return callback(null, err.message);
+                        return callback(err);
                     }
-                    return callback(err);
-                }
-                callback(null, message);
-            });
+                    callback(null, message);
+                });
+            }
         });
     }
 }

diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,7 +1,7 @@
 {
     "name": "@pulsar-sd/zone-mta",
     "private": false,
-    "version": "0.1.5",
+    "version": "0.1.6",
     "description": "Tiny outbound MTA",
     "main": "app.js",
     "scripts": {