From: Steven Barth <steven@midlink.org>
Date: Thu, 12 Mar 2009 20:14:55 +0000 (+0000)
Subject: httpclient: handle redirects more graceful
X-Git-Tag: 0.9.0~603
X-Git-Url: http://git.lede-project.org./?a=commitdiff_plain;h=2c7a30708d2c31da64ab23fc07f44f94c1e8b8d7;p=project%2Fluci.git

httpclient: handle redirects more graceful
---

diff --git a/libs/httpclient/luasrc/httpclient.lua b/libs/httpclient/luasrc/httpclient.lua
index e75cc0373c..4f95593103 100644
--- a/libs/httpclient/luasrc/httpclient.lua
+++ b/libs/httpclient/luasrc/httpclient.lua
@@ -223,7 +223,9 @@ function request_raw(uri, options)
 		return nil, -3, "invalid response magic: " .. line
 	end
 	
-	local response = {status = line, headers = {}, code = 0, cookies = {}}
+	local response = {
+		status = line, headers = {}, code = 0, cookies = {}, uri = uri
+	}
 	
 	line = linesrc()
 	while line and line ~= "" do
@@ -292,15 +294,18 @@ function request_raw(uri, options)
 	if response.code and options.depth > 0 then
 		if response.code == 301 or response.code == 302 or response.code == 307
 		 and response.headers.Location then
-			local nexturi = response.headers.Location
-			if not nexturi:find("https?://") then
-				nexturi = pr .. "://" .. host .. ":" .. port .. nexturi
+			local nuri = response.headers.Location or response.headers.location
+			if not nuri then
+				return nil, -5, "invalid reference"
+			end
+			if not nuri:find("https?://") then
+				nuri = pr .. "://" .. host .. ":" .. port .. nuri
 			end
 			
 			options.depth = options.depth - 1
 			sock:close()
 			
-			return request_raw(nexturi, options)
+			return request_raw(nuri, options)
 		end
 	end