qpsmtpd/sme9/0029-Run-hook_reset_transaction-after-all-connections.patch

From 5c3bd220aaccd49ebbcbfb4324ac4c789d116273 Mon Sep 17 00:00:00 2001
From: jaredj <jaredj@nmgi.com>
Date: Sat, 28 Feb 2009 15:13:46 +0800
Subject: Run hook_reset_transaction after all connections

This makes hook_reset_transaction a bit more useful by running it on
disconnections.  It also ensures that hook_disconnect runs when the
client disconnects without QUITing.
---
 lib/Qpsmtpd/SMTP.pm              |    1 +
 lib/Qpsmtpd/TcpServer.pm         |    4 ++++
 lib/Qpsmtpd/TcpServer/Prefork.pm |    5 +++++
 3 files changed, 10 insertions(+), 0 deletions(-)

diff --git a/lib/Qpsmtpd/SMTP.pm b/lib/Qpsmtpd/SMTP.pm
index f669055..0735f0c 100644
--- a/lib/Qpsmtpd/SMTP.pm
+++ b/lib/Qpsmtpd/SMTP.pm
@@ -576,6 +576,7 @@ sub quit_respond {
 sub disconnect {
   my $self = shift;
   $self->run_hooks("disconnect");
+  $self->connection->notes(disconnected => 1);
   $self->reset_transaction;
 }
 
diff --git a/lib/Qpsmtpd/TcpServer.pm b/lib/Qpsmtpd/TcpServer.pm
index 07d8d16..de3504f 100644
--- a/lib/Qpsmtpd/TcpServer.pm
+++ b/lib/Qpsmtpd/TcpServer.pm
@@ -115,6 +115,10 @@ sub read_input {
     alarm $timeout;
   }
   alarm(0);
+  return if $self->connection->notes('disconnected');
+  $self->reset_transaction;
+  $self->run_hooks('disconnect');
+  $self->connection->notes(disconnected => 1);
 }
 
 sub respond {
diff --git a/lib/Qpsmtpd/TcpServer/Prefork.pm b/lib/Qpsmtpd/TcpServer/Prefork.pm
index 28f60dc..2728cea 100644
--- a/lib/Qpsmtpd/TcpServer/Prefork.pm
+++ b/lib/Qpsmtpd/TcpServer/Prefork.pm
@@ -35,6 +35,11 @@ sub read_input {
         or $self->respond(502, "command unrecognized: '$_'");
       alarm $timeout;
     }
+    unless ($self->connection->notes('disconnected')) {
+      $self->reset_transaction;
+      $self->run_hooks('disconnect');
+      $self->connection->notes(disconnected => 1);
+    }
   };
   if ($@ =~ /^disconnect_tcpserver/) {
        die "disconnect_tcpserver";
-- 
1.7.2.2

1	From 5c3bd220aaccd49ebbcbfb4324ac4c789d116273 Mon Sep 17 00:00:00 2001
2	From: jaredj <jaredj@nmgi.com>
3	Date: Sat, 28 Feb 2009 15:13:46 +0800
4	Subject: Run hook_reset_transaction after all connections
5
6	This makes hook_reset_transaction a bit more useful by running it on
7	disconnections. It also ensures that hook_disconnect runs when the
8	client disconnects without QUITing.
9	---
10	lib/Qpsmtpd/SMTP.pm \| 1 +
11	lib/Qpsmtpd/TcpServer.pm \| 4 ++++
12	lib/Qpsmtpd/TcpServer/Prefork.pm \| 5 +++++
13	3 files changed, 10 insertions(+), 0 deletions(-)
14
15	diff --git a/lib/Qpsmtpd/SMTP.pm b/lib/Qpsmtpd/SMTP.pm
16	index f669055..0735f0c 100644
17	--- a/lib/Qpsmtpd/SMTP.pm
18	+++ b/lib/Qpsmtpd/SMTP.pm
19	@@ -576,6 +576,7 @@ sub quit_respond {
20	sub disconnect {
21	my $self = shift;
22	$self->run_hooks("disconnect");
23	+ $self->connection->notes(disconnected => 1);
24	$self->reset_transaction;
25	}
26
27	diff --git a/lib/Qpsmtpd/TcpServer.pm b/lib/Qpsmtpd/TcpServer.pm
28	index 07d8d16..de3504f 100644
29	--- a/lib/Qpsmtpd/TcpServer.pm
30	+++ b/lib/Qpsmtpd/TcpServer.pm
31	@@ -115,6 +115,10 @@ sub read_input {
32	alarm $timeout;
33	}
34	alarm(0);
35	+ return if $self->connection->notes('disconnected');
36	+ $self->reset_transaction;
37	+ $self->run_hooks('disconnect');
38	+ $self->connection->notes(disconnected => 1);
39	}
40
41	sub respond {
42	diff --git a/lib/Qpsmtpd/TcpServer/Prefork.pm b/lib/Qpsmtpd/TcpServer/Prefork.pm
43	index 28f60dc..2728cea 100644
44	--- a/lib/Qpsmtpd/TcpServer/Prefork.pm
45	+++ b/lib/Qpsmtpd/TcpServer/Prefork.pm
46	@@ -35,6 +35,11 @@ sub read_input {
47	or $self->respond(502, "command unrecognized: '$_'");
48	alarm $timeout;
49	}
50	+ unless ($self->connection->notes('disconnected')) {
51	+ $self->reset_transaction;
52	+ $self->run_hooks('disconnect');
53	+ $self->connection->notes(disconnected => 1);
54	+ }
55	};
56	if ($@ =~ /^disconnect_tcpserver/) {
57	die "disconnect_tcpserver";
58	--
59	1.7.2.2
60