diff options
| -rw-r--r-- | libs/httpclient/luasrc/httpclient.lua | 15 |
1 files changed, 10 insertions, 5 deletions
diff --git a/libs/httpclient/luasrc/httpclient.lua b/libs/httpclient/luasrc/httpclient.lua index e75cc0373..4f9559310 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 |