fixes bug 89113 "Proxy: http downloads might not display file size"

r=bbaetz, sr=mscott

netwerk/protocol/http/src/nsHttpResponseHead.cpp

@@ -31,6 +31,19 @@
 // nsHttpResponseHead <public>
 //-----------------------------------------------------------------------------
 
+void
+nsHttpResponseHead::SetContentLength(PRInt32 len)
+{
+    mContentLength = len;
+    if (len < 0)
+        SetHeader(nsHttp::Content_Length, nsnull);
+    else {
+        nsCAutoString buf;
+        buf.AppendInt(len);
+        SetHeader(nsHttp::Content_Length, buf.get());
+    }
+}
+
 void
 nsHttpResponseHead::Flatten(nsACString &buf, PRBool pruneTransients)
 {

netwerk/protocol/http/src/nsHttpResponseHead.h

@@ -55,7 +55,7 @@ public:
     void     ClearHeaders()                         { mHeaders.Clear(); }
 
     void     SetContentType(const char *s) { mContentType.Adopt(s ? nsCRT::strdup(s) : 0); }
-    void     SetContentLength(PRInt32 len) { mContentLength = len; }
+    void     SetContentLength(PRInt32);
 
     // write out the response status line and headers as a single text block,
     // optionally pruning out transient headers (ie. headers that only make

netwerk/protocol/http/src/nsHttpTransaction.cpp

@@ -389,6 +389,19 @@ nsHttpTransaction::HandleContentStart()
             return NS_BASE_STREAM_WOULD_BLOCK;
         }
 
+        // check if this is a no-content response
+        switch (mResponseHead->Status()) {
+        case 204:
+        case 205:
+        case 304:
+            mNoContent = PR_TRUE;
+            LOG(("this response should not contain a body.\n"));
+            break;
+        }
+
+        if (mNoContent)
+            mContentLength = 0;
+        else {
             // grab the content-length from the response headers
             mContentLength = mResponseHead->ContentLength();
 
@@ -412,39 +425,12 @@ nsHttpTransaction::HandleContentStart()
                     mConnection->DontReuse();
                 }
             }
-        else if (mContentLength == -1) {
-            // check if this is a no-content response
-            switch (mResponseHead->Status()) {
-            case 204:
-            case 205:
-            case 304:
-                mNoContent = PR_TRUE;
-                LOG(("looks like this response should not contain a body.\n"));
-                break;
-            default:
+#if defined(PR_LOGGING)
+            else if (mContentLength == -1)
                 LOG(("waiting for the server to close the connection.\n"));
-                break;
+#endif
         }
     }
-        else if (mResponseHead->Version() < NS_HTTP_VERSION_1_1 &&
-                 !mConnection->IsKeepAlive()) {
-            // HTTP/1.0 servers have been known to send erroneous Content-Length
-            // headers.  So, unless the connection is keep-alive, we'll ignore
-            // the Content-Length header altogether.
-            LOG(("ignoring possibly erroneous content-length header\n"));
-            // eliminate any references to this content length value, so our
-            // consumers don't get confused.
-            mContentLength = -1;
-            mResponseHead->SetContentLength(-1);
-            // if a client is really interested in the content length,
-            // even though we think it is invalid, them them hut for it
-            // in the raw headers.
-            //          mResponseHead->SetHeader(nsHttp::Content_Length, nsnull);
-
-        }
-        if (mNoContent)
-            mContentLength = 0;
-    }
 
     LOG(("nsHttpTransaction [this=%x] sending OnStartRequest\n", this));
     mFiredOnStart = PR_TRUE;
@@ -483,9 +469,22 @@ nsHttpTransaction::HandleContent(char *buf,
         rv = mChunkedDecoder->HandleChunkedContent(buf, count, countRead);
         if (NS_FAILED(rv)) return rv;
     }
-    else if (mContentLength >= 0)
-        // when a content length has been specified...
+    else if (mContentLength >= 0) {
+        // HTTP/1.0 servers have been known to send erroneous Content-Length
+        // headers. So, unless the connection is keep-alive, we must make
+        // allowances for a possibly invalid Content-Length header. Thus, if
+        // NOT keep-alive, we simply accept everything in |buf|.
+        if (mConnection->IsKeepAlive())
             *countRead = PR_MIN(count, mContentLength - mContentRead);
+        else {
+            *countRead = count;
+            // mContentLength might need to be increased...
+            if (*countRead + mContentRead > (PRUint32) mContentLength) {
+                mContentLength = *countRead + mContentRead;
+                mResponseHead->SetContentLength(mContentLength);
+            }
+        }
+    }
     else
         // when we are just waiting for the server to close the connection...
         *countRead = count;