From d5a1c6dfcbe1b1ab7ffe711996f970f1c410b919 Mon Sep 17 00:00:00 2001
From: Ade Lee <alee@redhat.com>
Date: Wed, 4 Jan 2012 00:08:03 -0500
Subject: BZ 771357 - sslget does not work after FEDORA-2011-17400 update,
 breaking FreeIPA install

Modified sslget doIO() function to be able to handle small reads.
---
 pki/base/native-tools/src/sslget/sslget.c | 56 ++++++++++++++++++++++++++-----
 1 file changed, 47 insertions(+), 9 deletions(-)

diff --git a/pki/base/native-tools/src/sslget/sslget.c b/pki/base/native-tools/src/sslget/sslget.c
index 5f4d448d..7288a1c5 100644
--- a/pki/base/native-tools/src/sslget/sslget.c
+++ b/pki/base/native-tools/src/sslget/sslget.c
@@ -345,15 +345,15 @@ do_writes(
 SECStatus
 do_io( PRFileDesc *ssl_sock, int connection)
 {
-    int	    countRead = 0;
+    int countRead = 0;
     PRInt32 rv;
-    char    *buf;
-	int first=1;
+    char *buf;
+    char *buf2;
+    int first=1;
 
     buf = PR_Malloc(RD_BUF_SIZE);
     if (!buf) exit(5);
 	
-
     /* send the http request here. */
 
     rv = do_writes(ssl_sock);
@@ -374,6 +374,8 @@ do_io( PRFileDesc *ssl_sock, int connection)
 	}
 	if (rv < 0) {
 	    errWarn("PR_Read");
+            PR_Free(buf);
+            buf = 0;
 	    exit(1);
 	}
 
@@ -386,12 +388,48 @@ do_io( PRFileDesc *ssl_sock, int connection)
         if (first) {
              first=0;
 	     if (rv < 13) {
-	         errWarn("not enough bytes read in first read");
-                 exit(2);
-             } else {
-                 if ( ! PL_strnstr(buf,"200",13)) {
-                     exit(3);
+                 int ret = 0;
+                 buf2 = PR_Malloc(RD_BUF_SIZE);
+                 if (!buf2) {
+                     PR_Free(buf);
+                     buf = 0;
+                     exit(5);
+                 }
+
+                 for (ret=0; rv < 13 ; rv += ret) {
+                     ret = PR_Read(ssl_sock, buf2, RD_BUF_SIZE - rv); 
+                     if (ret <  0 ) {
+                         errWarn("PR_Read");
+                         PR_Free(buf);
+                         buf = 0;
+                         PR_Free(buf2);
+                         buf2 = 0;
+                         exit(1);
+                     }
+                     if (ret == 0) {
+                         errWarn("not enough bytes read in first read");
+                         PR_Free(buf);
+                         buf = 0;
+                         PR_Free(buf2);
+                         buf2 = 0;
+                         exit(2);
+                     }
+                     countRead +=  ret;
+                     FPRINTF(stderr, "connection %d read %d bytes (%d total).\n",
+                         connection, ret, countRead );
+                     FPRINTF(stderr, "these bytes read:\n");
+                     PR_Write(PR_STDOUT, buf2, ret);
+
+                     PR_snprintf(buf, RD_BUF_SIZE, "%s%s", buf, buf2);
                  }
+                 PR_Free(buf2);
+                 buf2 = 0;
+             }
+
+             if ( ! PL_strnstr(buf,"200",13)) {
+                 PR_Free(buf);
+                 buf = 0;
+                 exit(3);
              }
 	}
     }
-- 
cgit