From bc0823758d3d905244e56196d6071568147cbba3 Mon Sep 17 00:00:00 2001
From: hidnplayr <hidnplayr@kolibrios.org>
Date: Tue, 12 Nov 2013 20:08:02 +0000
Subject: [PATCH] HTTP library: correctly receive content that is not chunked
 and has unknown content length.

git-svn-id: svn://kolibrios.org@4220 a494cfbc-eb01-0410-851d-a64ba20cac60
---
 .../libraries/http/examples/downloader.asm    |  2 +-
 programs/develop/libraries/http/http.asm      | 38 +++++++++++++++++--
 programs/develop/libraries/http/http.inc      |  1 +
 3 files changed, 36 insertions(+), 5 deletions(-)

diff --git a/programs/develop/libraries/http/examples/downloader.asm b/programs/develop/libraries/http/examples/downloader.asm
index e77658f199..02d51a38a9 100644
--- a/programs/develop/libraries/http/examples/downloader.asm
+++ b/programs/develop/libraries/http/examples/downloader.asm
@@ -141,7 +141,7 @@ fail:
 
 save:
         mov     ebp, [identifier]
-        mov     eax, [ebp + http_msg.content_length]
+        mov     eax, [ebp + http_msg.content_received]
         mov     [final_size], eax
         lea     ebx, [ebp + http_msg.data]
         add     ebx, [ebp + http_msg.header_length]
diff --git a/programs/develop/libraries/http/http.asm b/programs/develop/libraries/http/http.asm
index f8f847d89b..ccdf4b20ee 100644
--- a/programs/develop/libraries/http/http.asm
+++ b/programs/develop/libraries/http/http.asm
@@ -447,6 +447,9 @@ proc HTTP_process identifier ;//////////////////////////////////////////////////
                       [ebp + http_msg.buffer_length], MSG_DONTWAIT
         cmp     eax, 0xffffffff
         je      .check_socket
+
+        test    eax, eax
+        jz      .server_closed
         DEBUGF  1, "Received %u bytes\n", eax
 
 ; Update timestamp
@@ -616,6 +619,7 @@ proc HTTP_process identifier ;//////////////////////////////////////////////////
         or      eax, eax
         jz      .no_ram
 
+  .not_chunked:
         mov     eax, [ebp + http_msg.write_ptr]
         sub     eax, [ebp + http_msg.header_length]
         sub     eax, http_msg.data
@@ -629,17 +633,17 @@ proc HTTP_process identifier ;//////////////////////////////////////////////////
 ; Try to find 'transfer-encoding' header.
         stdcall find_header_field, ebp, str_te
         test    eax, eax
-        jz      .invalid_header
+        jz      .not_chunked
 
         mov     ebx, dword[eax]
         or      ebx, 0x20202020
         cmp     ebx, 'chun'
-        jne     .invalid_header
+        jne     .not_chunked
         mov     ebx, dword[eax+4]
         or      ebx, 0x00202020
         and     ebx, 0x00ffffff
         cmp     ebx, 'ked'
-        jne     .invalid_header
+        jne     .not_chunked
 
         or      [ebp + http_msg.flags], FLAG_CHUNKED
         DEBUGF  1, "Transfer type is: chunked\n"
@@ -719,10 +723,21 @@ proc HTTP_process identifier ;//////////////////////////////////////////////////
 ; Check if we got all the data.
   .header_parsed:
         add     [ebp + http_msg.content_received], eax
+        test    [ebp + http_msg.flags], FLAG_CONTENT_LENGTH
+        jz      .need_more_data_and_space
         mov     eax, [ebp + http_msg.content_received]
         cmp     eax, [ebp + http_msg.content_length]
         jae     .got_all_data
-;        jmp     .need_more_data
+        jmp     .need_more_data
+
+  .need_more_data_and_space:
+        mov     eax, [ebp + http_msg.write_ptr]
+        add     eax, BUFFERSIZE
+        sub     eax, ebp
+        invoke  mem.realloc, ebp, eax
+        or      eax, eax
+        jz      .no_ram
+        mov     [ebp + http_msg.buffer_length], BUFFERSIZE
 
   .need_more_data:
         popa
@@ -765,6 +780,19 @@ proc HTTP_process identifier ;//////////////////////////////////////////////////
         or      [ebp + http_msg.flags], FLAG_TIMEOUT_ERROR
         jmp     .disconnect
 
+  .server_closed:
+        DEBUGF  1, "server closed connection, transfer complete?\n"
+        test    [ebp + http_msg.flags], FLAG_GOT_HEADER
+        jz      .server_error
+        test    [ebp + http_msg.flags], FLAG_CONTENT_LENGTH
+        jz      .got_all_data
+
+  .server_error:
+        pop     eax
+        DEBUGF  1, "ERROR: server closed connection unexpectedly\n"
+        or      [ebp + http_msg.flags], FLAG_TRANSFER_FAILED
+        jmp     .disconnect
+
   .invalid_header:
         pop     eax
         DEBUGF  1, "ERROR: invalid header\n"
@@ -1059,6 +1087,8 @@ endp
 ;;////////////////////////////////////////////////////////////////////////////////////////////////;;
 ;;================================================================================================;;
 ;! Internal procedures section                                                                    ;;
+;;                                                                                                ;;
+;; NOTICE: These procedures do not follow stdcall conventions and thus may destroy any register.  ;;
 ;;================================================================================================;;
 ;;////////////////////////////////////////////////////////////////////////////////////////////////;;
 ;;================================================================================================;;
diff --git a/programs/develop/libraries/http/http.inc b/programs/develop/libraries/http/http.inc
index ac79ffdfa8..5e7000f084 100644
--- a/programs/develop/libraries/http/http.inc
+++ b/programs/develop/libraries/http/http.inc
@@ -25,6 +25,7 @@ FLAG_INVALID_HEADER     = 1 shl 16
 FLAG_NO_RAM             = 1 shl 17
 FLAG_SOCKET_ERROR       = 1 shl 18
 FLAG_TIMEOUT_ERROR      = 1 shl 19
+FLAG_TRANSFER_FAILED    = 1 shl 20
 
 struc http_msg {