X-Git-Url: https://diplodocus.org/git/nmh/blobdiff_plain/c2655c2b98253e03c42eeeee1e5c60fe7107c5a4..dd6bfc058da415e30995eeea116c65861671e2aa:/mts/smtp/smtp.c

diff --git a/mts/smtp/smtp.c b/mts/smtp/smtp.c
index cc4a27cb..0887e1cd 100644
--- a/mts/smtp/smtp.c
+++ b/mts/smtp/smtp.c
@@ -5,12 +5,13 @@
  * complete copyright information.
  */
 
-#include <h/mh.h>
+#include "h/mh.h"
+#include "sbr/error.h"
 #include "smtp.h"
-#include <h/mts.h>
-#include <h/signals.h>
-#include <h/utils.h>
-#include <h/netsec.h>
+#include "h/mts.h"
+#include "h/signals.h"
+#include "h/utils.h"
+#include "h/netsec.h"
 
 #include <sys/socket.h>
 #include "sbr/base64.h"
@@ -529,10 +530,14 @@ sm_waend (void)
 int
 sm_wtxt (char *buffer, int len)
 {
-    int result;
+    int result, snoopstate;
+
+    if ((snoopstate = netsec_get_snoop(nsc)))
+	netsec_set_snoop(nsc, 0);
 
     result = sm_wstream (buffer, len);
 
+    netsec_set_snoop(nsc, snoopstate);
     return result == NOTOK ? RP_BHST : RP_OK;
 }
 
@@ -540,9 +545,30 @@ sm_wtxt (char *buffer, int len)
 int
 sm_wtend (void)
 {
+    int snoopstate;
+
+    if ((snoopstate = netsec_get_snoop(nsc)))
+	netsec_set_snoop(nsc, 0);
+
     if (sm_wstream(NULL, 0) == NOTOK)
 	return RP_BHST;
 
+    /*
+     * Because snoop output now happens at flush time, if we are using snoop
+     * we force an extra flush here.  While this introduces some extra network
+     * traffic, we're only doing it when snoop is in effect so I think it's
+     * reasonable.
+     */
+
+    if (snoopstate) {
+	char *errstr;
+	if (netsec_flush(nsc, &errstr) != OK) {
+	    sm_nerror(errstr);
+	    return RP_BHST;
+	}
+	netsec_set_snoop(nsc, snoopstate);
+    }
+
     switch (smtalk (SM_DOT + 3 * sm_addrs, ".")) {
 	case 250: 
 	case 251: 
@@ -760,7 +786,7 @@ smhear (void)
     char **ehlo = EHLOkeys, *buffer;
 
     if (doingEHLO) {
-	static int at_least_once = 0;
+	static bool at_least_once;
 
 	if (at_least_once) {
 	    char *ep;
@@ -770,7 +796,7 @@ smhear (void)
 		free (ep);
 	    }
 	} else {
-	    at_least_once = 1;
+	    at_least_once = true;
 	}
 
 	ehlo = EHLOkeys;
@@ -975,15 +1001,19 @@ sm_sasl_callback(enum sasl_message_type mtype, unsigned const char *indata,
 	    snoopoffset = 6 + strlen(mech);
 	    rc = netsec_printf(nsc, errstr, "AUTH %s %s\r\n", mech, b64data);
 	    free(b64data);
-	    netsec_set_snoop_callback(nsc, NULL, NULL);
 	} else {
 	    rc = netsec_printf(nsc, errstr, "AUTH %s\r\n", mech);
 	}
 
-	if (rc != OK)
+	if (rc != OK) {
+	    netsec_set_snoop_callback(nsc, NULL, NULL);
 	    return NOTOK;
+	}
+
+	rc = netsec_flush(nsc, errstr);
+	netsec_set_snoop_callback(nsc, NULL, NULL);
 
-	if (netsec_flush(nsc, errstr) != OK)
+	if (rc != OK)
 	    return NOTOK;
 
 	break;
@@ -1037,14 +1067,16 @@ sm_sasl_callback(enum sasl_message_type mtype, unsigned const char *indata,
 	    writeBase64raw(indata, indatalen, b64data);
 	    netsec_set_snoop_callback(nsc, netsec_b64_snoop_decoder, NULL);
 	    rc = netsec_printf(nsc, errstr, "%s\r\n", b64data);
-	    netsec_set_snoop_callback(nsc, NULL, NULL);
 	    free(b64data);
 	}
 
 	if (rc != OK)
 	    return NOTOK;
 
-	if (netsec_flush(nsc, errstr) != OK)
+	rc = netsec_flush(nsc, errstr);
+	netsec_set_snoop_callback(nsc, NULL, NULL);
+
+	if (rc != OK)
 	    return NOTOK;
 	break;