From 4c13b99e957d9641a73c61dfccde58f18d4fe39a Mon Sep 17 00:00:00 2001
From: joao <joao@e0a8ed71-7df4-0310-8962-fdc924857419>
Date: Tue, 15 Sep 2009 03:47:46 +0000
Subject: [PATCH] Giving priority to transfer-encoding first than
 content-length, as mentioned in rfc2616, section 4.4. isChunked now checks
 for transfer-encoding: identity instead of transfer-encoding: chunked. If
 transfer encoding is present and it is not identity, chunked encoding is
 considered. Also rfc2616, section 4.4 (item 2)

---
 nselib/http.lua | 29 +++++++++++++++++++----------
 1 file changed, 19 insertions(+), 10 deletions(-)

diff --git a/nselib/http.lua b/nselib/http.lua
index af6556d6b..e50e727dd 100644
--- a/nselib/http.lua
+++ b/nselib/http.lua
@@ -516,11 +516,7 @@ function getNextResult( full_response, method )
   -- If it is a get response, attach body to response
   if method == "get" then
     body_start = body_start + 1 -- fixing body start offset
-    length = getLength( header )
-    if length then
-      length = length + #header
-      body = full_response:sub(body_start, length)
-    elseif isChunked(header) then
+    if isChunked(header) then
       full_response = full_response:sub(body_start)
       local body_delim = ( full_response:match( "\r\n" ) and "\r\n" )  or
                          ( full_response:match( "\n" )   and "\n" ) or nil
@@ -528,12 +524,18 @@ function getNextResult( full_response, method )
       local chunks = {}
       for tmp_size, chunk in get_chunks(full_response, 1, body_delim) do
         chunks[#chunks + 1] = chunk
-		size = tmp_size
+                size = tmp_size
       end
       body = table.concat(chunks)
     else
-      stdnse.print_debug("Didn't find chunked encoding or content-length field, not splitting response")
-      body = full_response:sub(body_start)
+      length = getLength( header )
+      if length then
+        length = length + #header
+        body = full_response:sub(body_start, length)
+      else
+        stdnse.print_debug("Didn't find chunked encoding or content-length field, not splitting response")
+        body = full_response:sub(body_start)
+      end
     end
   end
 
@@ -556,8 +558,15 @@ function isChunked( header )
   local encoding = nil
   for number, line in ipairs( header or {} ) do
     line = line:lower()
-    encoding = line:match("(transfer%-encoding: chunked)")
-    if encoding then return true end
+    encoding = line:match("transfer%-encoding: (.*)")
+    if encoding then
+      print(encoding)
+      if encoding:match("identity") then
+        return false
+      else
+        return true
+      end
+    end
   end
   return false
 end