httpclient: Improved callback support

[project/luci.git] / libs / httpclient / luasrc / httpclient / receiver.lua

diff --git a/libs/httpclient/luasrc/httpclient/receiver.lua b/libs/httpclient/luasrc/httpclient/receiver.lua
index c977d06..55503f1 100644 (file)
--- a/libs/httpclient/luasrc/httpclient/receiver.lua
+++ b/libs/httpclient/luasrc/httpclient/receiver.lua
@@ -203,7 +203,12 @@ function request_to_file(uri, target, options, cbs)
        end
        
        if cbs.on_header then
-               cbs.on_header(file, code, resp)
+               local stat = {cbs.on_header(file, code, resp)}
+               if stat[1] == false then
+                       file:close()
+                       sock:close()
+                       return true, nil, nil, unpack(stat, 2)
+               end
        end
 
        local chunked = resp.headers["Transfer-Encoding"] == "chunked"