* src/engine.c: Fix timeout calculations

(struct timeout_ctl): New data type
(UPDATE_TTW,smtp_stream_wait,smtp_wait): Operate on struct timeout_ctl.
All callers updated.
(init_timeout_ctl): New function.


git-svn-id: file:///svnroot/mailfromd/trunk@1517 7a8a7f39-df28-0410-adc6-e0d955640f24

2 files changed, 54 insertions, 33 deletions

diff --git a/ChangeLog b/ChangeLog
index ee51e4c7..c50cbd64 100644
--- a/ChangeLog
+++ b/ChangeLog
@@ -1,2 +1,10 @@
+2007-10-21  Sergey Poznyakoff  <gray@gnu.org.ua>
+
+	* src/engine.c: Fix timeout calculations
+	(struct timeout_ctl): New data type
+	(UPDATE_TTW,smtp_stream_wait,smtp_wait): Operate on struct timeout_ctl.
+	All callers updated.
+	(init_timeout_ctl): New function.
+
 2007-10-12  Sergey Poznyakoff  <gray@gnu.org.ua>
 
diff --git a/src/engine.c b/src/engine.c
index 1801fdec..bb3fa18c 100644
--- a/src/engine.c
+++ b/src/engine.c
@@ -306,27 +306,40 @@ smtp_io_data_destroy(struct smtp_io_data *dat)
 }
 
-#define UPDATE_TTW(t,start) do {                                     \
-	time_t delta = time(NULL) - start;                           \
-	if (t > delta)                                               \
-		t -= delta;                                          \
+struct timeout_ctl {
+  time_t start;
+  time_t timeout;
+};
+
+static void
+init_timeout_ctl(struct timeout_ctl *tctl, time_t timeout)
+{
+  tctl->start = time(NULL);
+  tctl->timeout = timeout;
+}
+
+#define UPDATE_TTW(t) do {                                           \
+        time_t now = time(NULL);                                     \
+	time_t delta = now - (t).start;                              \
+	if ((t).timeout > delta)                                     \
+		(t).timeout -= delta;                                \
 	else                                                         \
-		t = 0;                                               \
+		(t).timeout = 0;                                     \
+        (t).start = now;                                             \
 } while (0)
 
 int
-smtp_stream_wait(mu_stream_t stream, int flags, time_t *timeout)
+smtp_stream_wait(mu_stream_t stream, int flags, struct timeout_ctl *tctl)
 {
 	int rc;
 	int oflags = flags;
 	struct timeval tv;
-	time_t start = time(NULL);
 	
-	while (*timeout) {
-		tv.tv_sec = *timeout;
+	while (tctl->timeout) {
+		tv.tv_sec = tctl->timeout;
 		tv.tv_usec = 0;
 		rc = mu_stream_wait(stream, &oflags, &tv);
 
 		/* Correct the timeout */
-		UPDATE_TTW(*timeout, start);
+		UPDATE_TTW(*tctl);
 		
 		switch (rc) {
@@ -348,7 +361,7 @@ smtp_stream_wait(mu_stream_t stream, int flags, time_t *timeout)
 
 int
-smtp_wait(struct smtp_io_data *dat, int flags, time_t *timeout)
+smtp_wait(struct smtp_io_data *dat, int flags, struct timeout_ctl *tctl)
 {
-	return smtp_stream_wait(dat->stream, flags, timeout);
+	return smtp_stream_wait(dat->stream, flags, tctl);
 }
 
@@ -357,6 +370,7 @@ smtp_send(struct smtp_io_data *dat, char *command)
 {
 	size_t len = strlen(command);
-	time_t timeout = io_timeout;
-	time_t start = time(NULL);
+	struct timeout_ctl tctl;
+
+	init_timeout_ctl (&tctl, io_timeout);
 	
 	dat->reply = NULL; /* Clear reply for logging purposes */
@@ -366,5 +380,5 @@ smtp_send(struct smtp_io_data *dat, char *command)
 		int rc;
 		
-		UPDATE_TTW(timeout, start);
+		UPDATE_TTW(tctl);
 		
 		rc = mu_stream_write(dat->stream, command, len,
@@ -379,6 +393,5 @@ smtp_send(struct smtp_io_data *dat, char *command)
 			command += nb;
 		} else if (rc == EAGAIN) {
-			rc = smtp_wait(dat, MU_STREAM_READY_WR,
-				       &timeout);
+			rc = smtp_wait(dat, MU_STREAM_READY_WR, &tctl);
 			if (rc) {
 				mu_error(_("smtp_wait failed: %s"),
@@ -425,11 +438,11 @@ int
 smtp_recvline(struct smtp_io_data *dat, time_t timeout)
 {
-	time_t start = time(NULL);
-
+	struct timeout_ctl tctl;
+	init_timeout_ctl(&tctl, timeout);
 	for (;;) {
 		char *p;
 		
-		UPDATE_TTW(timeout, start);
-
+		UPDATE_TTW(tctl);
+		
 		if (dat->level == 0) {
 			int rc = mu_stream_read(dat->stream,
@@ -444,6 +457,5 @@ smtp_recvline(struct smtp_io_data *dat, time_t timeout)
 			} else if (rc == EAGAIN) {
 				rc = smtp_wait(dat, 
-					       MU_STREAM_READY_RD, 
-					       &timeout);
+					       MU_STREAM_READY_RD, &tctl);
 				if (rc) {
 					mu_error(_("smtp_wait failed: %s"),
@@ -564,5 +576,5 @@ check_on_host(eval_environ_t env, char *email, char *client_addr,
 	int count = 0;
 	SMFICTX *ctx = env_get_context(env);
-	time_t timeout, start  = time(NULL);
+	struct timeout_ctl tctl;
 	
 	debug2(10, "email = %s, client_addr = %s", email, client_addr);
@@ -579,13 +591,13 @@ check_on_host(eval_environ_t env, char *email, char *client_addr,
 		return mf_temp_failure;
 	}
-
-	timeout = connect_timeout;
+	
+	init_timeout_ctl(&tctl, connect_timeout);
 	while ((rc = mu_stream_open(stream))) {
 		if ((rc == EAGAIN || rc == EINPROGRESS)
-		    && timeout) {
+		    && tctl.timeout) {
 			rc = smtp_stream_wait(stream, MU_STREAM_READY_WR,
-					      &timeout);
+					      &tctl);
 			if (rc == 0) {
-				UPDATE_TTW(timeout, start);
+				UPDATE_TTW(tctl);
 				continue;
 			}
@@ -828,5 +840,5 @@ listens_on (const char *client_addr, int port)
 	mu_stream_t stream;
 	int rc;
-	time_t timeout = connect_timeout, start = time(NULL);
+	struct timeout_ctl tctl;
 	
 	if (port == 0)
@@ -843,10 +855,11 @@ listens_on (const char *client_addr, int port)
 	}
 
+	init_timeout_ctl(&tctl, connect_timeout);
 	while ((rc = mu_stream_open(stream))) {
-		if ((rc == EAGAIN || rc == EINPROGRESS) && timeout) {
+		if ((rc == EAGAIN || rc == EINPROGRESS) && tctl.timeout) {
 			rc = smtp_stream_wait(stream, MU_STREAM_READY_WR,
-                                              &timeout);
+                                              &tctl);
 			if (rc == 0) {
-				UPDATE_TTW(timeout, start);
+				UPDATE_TTW(tctl);
 				continue;
 			}