Debugging fixes for features recently added.

2 files changed, 54 insertions, 28 deletions

diff --git a/common/smtppass.c b/common/smtppass.c
index 57bdf27..2604a85 100644
--- a/common/smtppass.c
+++ b/common/smtppass.c
@@ -176,7 +176,7 @@ static int connect_out(spctx_t* ctx);
 static int read_server_response(spctx_t* ctx);
 static int parse_config_file(const char* configfile);
 static char* parse_address(char* line);
-static int is_successful_rsp(const char* line);
+static const char* get_successful_rsp(const char* line);
 
 /* Used externally in some cases */
 int sp_parse_option(const char* name, const char* option);
@@ -649,8 +649,8 @@ static void* thread_main(void* arg)
         fd = thread->fd;
     sp_unlock();
 
-    ctx = init_thread(fd);
-    if(!ctx)
+    /* Sometimes we get to this point and then quit is noted */
+    if(sp_is_quit() || (ctx = init_thread(fd)) == NULL)
     {
         /* Special case. We don't have a context so clean up descriptor */
         close(fd);
@@ -776,6 +776,7 @@ static int connect_out(spctx_t* ctx)
 static int smtp_passthru(spctx_t* ctx)
 {
     char* t;
+    const char* p;
     int r, ret = 0;
     unsigned int mask;
     int neterror = 0;
@@ -789,7 +790,7 @@ static int smtp_passthru(spctx_t* ctx)
     #define C_LINE  ctx->client.line
     #define S_LINE  ctx->server.line
 
-    for(;;)
+    while(!sp_is_quit())
     {
         mask = spio_select(ctx, &(ctx->client), &(ctx->server), NULL);
 
@@ -962,7 +963,7 @@ static int smtp_passthru(spctx_t* ctx)
                 }
             }
 
-            if(is_successful_rsp(S_LINE))
+            if((p = get_successful_rsp(S_LINE)) != NULL)
             {
                 /*
                  * Filter out any EHLO responses that we can't or don't want
@@ -970,14 +971,13 @@ static int smtp_passthru(spctx_t* ctx)
                  */
                 if(is_first_word(C_LINE, EHLO_CMD, KL(EHLO_CMD)))
                 {
-                    char* p = S_LINE + r;
                     if(is_first_word(p, ESMTP_PIPELINE, KL(ESMTP_PIPELINE)) ||
                        is_first_word(p, ESMTP_TLS, KL(ESMTP_TLS)) ||
                        is_first_word(p, ESMTP_CHUNK, KL(ESMTP_CHUNK)) ||
                        is_first_word(p, ESMTP_BINARY, KL(ESMTP_BINARY)) ||
                        is_first_word(p, ESMTP_CHECK, KL(ESMTP_CHECK)))
                     {
-                        sp_messagex(ctx, LOG_DEBUG, "filtered ESMTP feature: %s", trim_space(p));
+                        sp_messagex(ctx, LOG_DEBUG, "filtered ESMTP feature: %s", trim_space((char*)p));
                         continue;
                     }
                 }
@@ -1008,6 +1008,9 @@ static int smtp_passthru(spctx_t* ctx)
                         /* Recipients are separated by lines */
                         if(r != 0)
                             strcat(ctx->recipients, "\n");
+                        else
+                            ctx->recipients[0] = 0;
+
                         strcat(ctx->recipients, t);
                     }
                 }
@@ -1060,7 +1063,7 @@ static char* parse_address(char* line)
     return trim_end(line);
 }
 
-static int is_successful_rsp(const char* line)
+static const char* get_successful_rsp(const char* line)
 {
     /*
      * We check for both '250 xxx' type replies
@@ -1071,9 +1074,9 @@ static int is_successful_rsp(const char* line)
 
     if(line[0] == '2' && isdigit(line[1]) && isdigit(line[2]) &&
        (line[3] == ' ' || line[3] == '-'))
-        return 1;
+        return line + 4;
 
-    return 0;
+    return NULL;
 }
 
 void sp_add_log(spctx_t* ctx, char* prefix, char* line)
diff --git a/common/spio.c b/common/spio.c
index 2d42eac..8cd315b 100644
--- a/common/spio.c
+++ b/common/spio.c
@@ -164,6 +164,7 @@ unsigned int spio_select(spctx_t* ctx, ...)
     fd_set mask;
     spio_t* io;
     int ret = 0;
+    int have = 0;
     int i = 0;
     va_list ap;
 
@@ -174,16 +175,20 @@ unsigned int spio_select(spctx_t* ctx, ...)
 
     while((io = va_arg(ap, spio_t*)) != NULL)
     {
-        /* We can't handle more than 31 args */
-        if(i > (sizeof(int) * 8) - 2)
-            break;
+        if(spio_valid(io))
+        {
+            /* We can't handle more than 31 args */
+            if(i > (sizeof(int) * 8) - 2)
+                break;
 
-        /* Check if the buffer has something in it */
-        if(HAS_EXTRA(io))
-            ret |= (1 << i);
+            /* Check if the buffer has something in it */
+            if(HAS_EXTRA(io))
+                ret |= (1 << i);
 
-        /* Mark for select */
-        FD_SET(io->fd, &mask);
+            /* Mark for select */
+            FD_SET(io->fd, &mask);
+            have = 1;
+        }
 
         i++;
     }
@@ -194,17 +199,35 @@ unsigned int spio_select(spctx_t* ctx, ...)
     if(ret != 0)
         return ret;
 
-    /* Otherwise wait on more data */
-    switch(select(FD_SETSIZE, &mask, NULL, NULL,
-           (struct timeval*)&(g_state.timeout)))
-    {
-    case 0:
-        sp_messagex(ctx, LOG_ERR, "network operation timed out");
-        return ~0;
-    case -1:
-        sp_message(ctx, LOG_ERR, "couldn't select on sockets");
+    /* No valid file descriptors */
+    if(!have)
         return ~0;
-    };
+
+    for(;;)
+    {
+        /* Otherwise wait on more data */
+        switch(select(FD_SETSIZE, &mask, NULL, NULL,
+               (struct timeval*)&(g_state.timeout)))
+        {
+        case 0:
+            sp_messagex(ctx, LOG_ERR, "network operation timed out");
+            return ~0;
+
+        case -1:
+            if(errno == EINTR)
+            {
+                if(!sp_is_quit())
+                    continue;
+            }
+
+            else
+                sp_message(ctx, LOG_ERR, "couldn't select on sockets");
+
+            return ~0;
+        };
+
+        break;
+    }
 
     /* See what came in */
     i = 0;