X-Git-Url: https://git.enpas.org/?a=blobdiff_plain;f=util%2Fhttpstream.lua;h=4b5060a1a6835e1303a463af5d2a9d0d8474c636;hb=0ef23c673d1d620c9c1cc3e8aed43add63634426;hp=fac38d81f2a5135f848a06832b25f5e7ff16f63a;hpb=10f205698b90a2b5c6da7479cf03365f55941e03;p=prosody.git

diff --git a/util/httpstream.lua b/util/httpstream.lua
index fac38d81..4b5060a1 100644
--- a/util/httpstream.lua
+++ b/util/httpstream.lua
@@ -7,7 +7,7 @@ coroutine.resume(deadroutine);
 
 module("httpstream")
 
-local function parser(success_cb, parser_type)
+local function parser(success_cb, parser_type, options_cb)
 	local data = coroutine.yield();
 	local function readline()
 		local pos = data:find("\r\n", nil, true);
@@ -67,20 +67,39 @@ local function parser(success_cb, parser_type)
 			-- read status line
 			local status_line = readline();
 			local httpversion, status_code, reason_phrase = status_line:match("^HTTP/(%S+)%s+(%d%d%d)%s+(.*)$");
-			if not httpversion then coroutine.yield("invalid-status-line"); end
+			status_code = tonumber(status_code);
+			if not status_code then coroutine.yield("invalid-status-line"); end
 			local headers = readheaders();
 			
 			-- read body
+			local have_body = not
+				 ( (options_cb and options_cb().method == "HEAD")
+				or (status_code == 204 or status_code == 304 or status_code == 301)
+				or (status_code >= 100 and status_code < 200) );
+			
 			local body;
-			local len = tonumber(headers["content-length"]);
-			if len then -- TODO check for invalid len
-				body = readlength(len);
-			else -- read to end
-				repeat
-					local newdata = coroutine.yield();
-					data = data..newdata;
-				until newdata == "";
-				body, data = data, "";
+			if have_body then
+				local len = tonumber(headers["content-length"]);
+				if headers["transfer-encoding"] == "chunked" then
+					body = "";
+					while true do
+						local chunk_size = readline():match("^%x+");
+						if not chunk_size then coroutine.yield("invalid-chunk-size"); end
+						chunk_size = tonumber(chunk_size, 16)
+						if chunk_size == 0 then break; end
+						body = body..readlength(chunk_size);
+						if readline() ~= "" then coroutine.yield("invalid-chunk-ending"); end
+					end
+					local trailers = readheaders();
+				elseif len then -- TODO check for invalid len
+					body = readlength(len);
+				else -- read to end
+					repeat
+						local newdata = coroutine.yield();
+						data = data..newdata;
+					until newdata == "";
+					body, data = data, "";
+				end
 			end
 			
 			success_cb({
@@ -93,9 +112,9 @@ local function parser(success_cb, parser_type)
 	else coroutine.yield("unknown-parser-type"); end
 end
 
-function new(success_cb, error_cb, parser_type)
+function new(success_cb, error_cb, parser_type, options_cb)
 	local co = coroutine.create(parser);
-	coroutine.resume(co, success_cb, parser_type)
+	coroutine.resume(co, success_cb, parser_type, options_cb)
 	return {
 		feed = function(self, data)
 			if not data then