Testsuite: case for TLS client tls-on-connect

[users/heiko/exim.git] / test / src / client.c

diff --git a/test/src/client.c b/test/src/client.c
index 64b8203bf5bbf997c5f85e94ded9220ff098271f..9190af0689295a1bde2a926d9118828d2a44b2b3 100644 (file)
--- a/test/src/client.c
+++ b/test/src/client.c
@@ -297,7 +297,7 @@ if (rc <= 0)
   return 0;
   }
 
   return 0;
   }
 

-printf("SSL connection using %s\n", SSL_get_cipher (*ssl));
+/* printf("SSL connection using %s\n", SSL_get_cipher (*ssl)); */

 return 1;
 }
 
 return 1;
 }
 


@@ -569,63 +569,78 @@ while (fgets(CS outbuffer, sizeof(outbuffer), f) != NULL)
 nextinput:
     if (*inptr == 0)   /* Refill input buffer */
       {
 nextinput:
     if (*inptr == 0)   /* Refill input buffer */
       {

+      unsigned char *inbufferp = inbuffer;
+

       alarm(timeout);
       alarm(timeout);

-      if (srv->tls_active)
-        {
+      for (;;)
+       {
+       if (srv->tls_active)
+         {

 #ifdef HAVE_OPENSSL
 #ifdef HAVE_OPENSSL

-       int error;
-       DEBUG { printf("call SSL_read\n"); fflush(stdout); }
-        rc = SSL_read(srv->ssl, inbuffer, bsiz - 1);
-       DEBUG { printf("SSL_read: %d\n", rc); fflush(stdout); }
-       if (rc <= 0)
-          switch (error = SSL_get_error(srv->ssl, rc))
-           {
-           case SSL_ERROR_ZERO_RETURN:
-             break;
-           case SSL_ERROR_SYSCALL:
-             printf("%s\n", ERR_error_string(ERR_get_error(), NULL));
-             rc = -1;
-             break;
-           case SSL_ERROR_SSL:
-             printf("%s\nTLS terminated\n", ERR_error_string(ERR_get_error(), NULL));
-             SSL_shutdown(srv->ssl);
-             SSL_free(srv->ssl);
-             srv->tls_active = FALSE;
-             { /* OpenSSL leaves it in restartsys mode */
-             struct sigaction act = {.sa_handler = sigalrm_handler_flag, .sa_flags = 0};
-             sigalrm_seen = 1;
-             sigaction(SIGALRM, &act, NULL);
+         int error;
+         DEBUG { printf("call SSL_read\n"); fflush(stdout); }
+         rc = SSL_read(srv->ssl, inbufferp, bsiz - (inbufferp - inbuffer) - 1);
+         DEBUG { printf("SSL_read: %d\n", rc); fflush(stdout); }
+         if (rc <= 0)
+           switch (error = SSL_get_error(srv->ssl, rc))
+             {
+             case SSL_ERROR_ZERO_RETURN:
+               break;
+             case SSL_ERROR_SYSCALL:
+               printf("%s\n", ERR_error_string(ERR_get_error(), NULL));
+               rc = -1;
+               break;
+             case SSL_ERROR_SSL:
+               printf("%s\nTLS terminated\n", ERR_error_string(ERR_get_error(), NULL));
+               SSL_shutdown(srv->ssl);
+               SSL_free(srv->ssl);
+               srv->tls_active = FALSE;
+               {       /* OpenSSL leaves it in restartsys mode */
+               struct sigaction act = {.sa_handler = sigalrm_handler_flag, .sa_flags = 0};
+               sigalrm_seen = 1;
+               sigaction(SIGALRM, &act, NULL);
+               }
+               *inptr = 0;
+               DEBUG { printf("go round\n"); fflush(stdout); }
+               goto nextinput;
+             default:
+               printf("SSL error code %d\n", error);

              }
              }

-             *inptr = 0;
-             DEBUG { printf("go round\n"); fflush(stdout); }
-             goto nextinput;
-           default:
-             printf("SSL error code %d\n", error);
-           }

 #endif
 #ifdef HAVE_GNUTLS
 #endif
 #ifdef HAVE_GNUTLS

-      retry1:
-       DEBUG { printf("call gnutls_record_recv\n"); fflush(stdout); }
-        rc = gnutls_record_recv(tls_session, CS inbuffer, bsiz - 1);
-       if (rc < 0)
+       retry1:
+         DEBUG { printf("call gnutls_record_recv\n"); fflush(stdout); }
+         rc = gnutls_record_recv(tls_session, CS inbufferp, bsiz - (inbufferp - inbuffer) - 1);
+         if (rc < 0)
+           {
+           DEBUG { printf("gnutls_record_recv: %s\n", gnutls_strerror(rc)); fflush(stdout); }
+           if (rc == GNUTLS_E_INTERRUPTED || rc == GNUTLS_E_AGAIN)
+             goto retry1;
+           printf("%s\n", gnutls_strerror(rc));
+           srv->tls_active = FALSE;
+           *inptr = 0;
+           DEBUG { printf("go round\n"); fflush(stdout); }
+           goto nextinput;
+           }
+         DEBUG { printf("gnutls_record_recv: %d\n", rc); fflush(stdout); }
+#endif
+         }
+       else

          {
          {

-         DEBUG { printf("gnutls_record_recv: %s\n", gnutls_strerror(rc)); fflush(stdout); }
-         if (rc == GNUTLS_E_INTERRUPTED || rc == GNUTLS_E_AGAIN)
-           goto retry1;
-         printf("%s\n", gnutls_strerror(rc));
-         srv->tls_active = FALSE;
-         *inptr = 0;
-         DEBUG { printf("go round\n"); fflush(stdout); }
-         goto nextinput;
+         DEBUG { printf("call read\n"); fflush(stdout); }
+         rc = read(srv->sock, inbufferp, bsiz - (inbufferp - inbuffer) - 1);
+         DEBUG { printf("read: %d\n", rc); fflush(stdout); }

          }
          }

-       DEBUG { printf("gnutls_record_recv: %d\n", rc); fflush(stdout); }
-#endif
-        }
-      else
-       {
-       DEBUG { printf("call read\n"); fflush(stdout); }
-       rc = read(srv->sock, inbuffer, bsiz);
-       DEBUG { printf("read: %d\n", rc); fflush(stdout); }
+
+         if (rc > 0) inbufferp[rc] = '\0';
+         if (rc <= 0 || strchr(inbufferp, '\n')) break;
+         inbufferp += rc;
+         if (inbufferp >= inbuffer + bsiz)
+           {
+           printf("Input buffer overrun, need more than %d bytes input buffer\n", bsiz);
+           exit(73);
+           }
+         DEBUG { printf("read more\n"); }

        }
       alarm(0);
 
        }
       alarm(0);
 


@@ -657,10 +672,7 @@ nextinput:
         exit(74);
         }
       else
         exit(74);
         }
       else

-        {
-        inbuffer[rc] = 0;

         inptr = inbuffer;
         inptr = inbuffer;

-        }

       }
     DEBUG { printf("read: '%s'\n", inptr); fflush(stdout); }
 
       }
     DEBUG { printf("read: '%s'\n", inptr); fflush(stdout); }
 


@@ -977,7 +989,7 @@ struct sockaddr_in6 s_in6;
 
 srv_ctx srv;
 
 
 srv_ctx srv;
 

-unsigned char inbuffer[10240];
+unsigned char inbuffer[100 * 1024];

 unsigned char *inptr = inbuffer;
 
 *inptr = 0;   /* Buffer empty */
 unsigned char *inptr = inbuffer;
 
 *inptr = 0;   /* Buffer empty */


@@ -1167,14 +1179,30 @@ if (interface != NULL)
 #if HAVE_IPV6
 if (host_af == AF_INET6)
   {
 #if HAVE_IPV6
 if (host_af == AF_INET6)
   {

+# ifdef HAVE_GETADDRINFO
+  struct addrinfo hints, *res;
+  memset(&hints, 0, sizeof(hints));
+  hints.ai_family = AF_INET6;
+  hints.ai_socktype = SOCK_STREAM;
+  hints.ai_flags = AI_NUMERICHOST;
+  if ((rc = getaddrinfo(address, NULL, &hints, &res)) != 0 || res == NULL)
+    {
+    printf("unable to parse \"%s\" as an IP address: %s\n", address,
+      rc == 0 ? "NULL result returned" : gai_strerror(rc));
+    exit(86);
+    }
+  memcpy(&s_in6, res->ai_addr, res->ai_addrlen);
+  freeaddrinfo(res);
+# else

   memset(&s_in6, 0, sizeof(s_in6));
   s_in6.sin6_family = AF_INET6;
   memset(&s_in6, 0, sizeof(s_in6));
   s_in6.sin6_family = AF_INET6;

-  s_in6.sin6_port = htons(port);

   if (inet_pton(host_af, address, &s_in6.sin6_addr) != 1)
     {
     printf("Unable to parse \"%s\"", address);
     exit(86);
     }
   if (inet_pton(host_af, address, &s_in6.sin6_addr) != 1)
     {
     printf("Unable to parse \"%s\"", address);
     exit(86);
     }

+# endif
+  s_in6.sin6_port = htons(port);

   }
 else
 #endif
   }
 else
 #endif