123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368 |
- require "nixio.util"
- local nixio = require "nixio"
- local ltn12 = require "luci.ltn12"
- local util = require "luci.util"
- local table = require "table"
- local http = require "luci.http.protocol"
- local date = require "luci.http.protocol.date"
- local type, pairs, ipairs, tonumber = type, pairs, ipairs, tonumber
- local unpack = unpack
- module "luci.httpclient"
- function chunksource(sock, buffer)
- buffer = buffer or ""
- return function()
- local output
- local _, endp, count = buffer:find("^([0-9a-fA-F]+);?.-\r\n")
- while not count and #buffer <= 1024 do
- local newblock, code = sock:recv(1024 - #buffer)
- if not newblock then
- return nil, code
- end
- buffer = buffer .. newblock
- _, endp, count = buffer:find("^([0-9a-fA-F]+);?.-\r\n")
- end
- count = tonumber(count, 16)
- if not count then
- return nil, -1, "invalid encoding"
- elseif count == 0 then
- return nil
- elseif count + 2 <= #buffer - endp then
- output = buffer:sub(endp+1, endp+count)
- buffer = buffer:sub(endp+count+3)
- return output
- else
- output = buffer:sub(endp+1, endp+count)
- buffer = ""
- if count - #output > 0 then
- local remain, code = sock:recvall(count-#output)
- if not remain then
- return nil, code
- end
- output = output .. remain
- count, code = sock:recvall(2)
- else
- count, code = sock:recvall(count+2-#buffer+endp)
- end
- if not count then
- return nil, code
- end
- return output
- end
- end
- end
- function request_to_buffer(uri, options)
- local source, code, msg = request_to_source(uri, options)
- local output = {}
-
- if not source then
- return nil, code, msg
- end
-
- source, code = ltn12.pump.all(source, (ltn12.sink.table(output)))
-
- if not source then
- return nil, code
- end
-
- return table.concat(output)
- end
- function request_to_source(uri, options)
- local status, response, buffer, sock = request_raw(uri, options)
- if not status then
- return status, response, buffer
- elseif status ~= 200 and status ~= 206 then
- return nil, status, buffer
- end
-
- if response.headers["Transfer-Encoding"] == "chunked" then
- return chunksource(sock, buffer)
- else
- return ltn12.source.cat(ltn12.source.string(buffer), sock:blocksource())
- end
- end
- function request_raw(uri, options)
- options = options or {}
- local pr, auth, host, port, path
- if options.params then
- uri = uri .. '?' .. http.urlencode_params(options.params)
- end
- if uri:find("%[") then
- if uri:find("@") then
- pr, auth, host, port, path = uri:match("(%w+)://(.+)@(%b[]):?([0-9]*)(.*)")
- host = host:sub(2,-2)
- else
- pr, host, port, path = uri:match("(%w+)://(%b[]):?([0-9]*)(.*)")
- host = host:sub(2,-2)
- end
- else
- if uri:find("@") then
- pr, auth, host, port, path =
- uri:match("(%w+)://(.+)@([%w-.]+):?([0-9]*)(.*)")
- else
- pr, host, port, path = uri:match("(%w+)://([%w-.]+):?([0-9]*)(.*)")
- end
- end
- if not host then
- return nil, -1, "unable to parse URI"
- end
-
- if pr ~= "http" and pr ~= "https" then
- return nil, -2, "protocol not supported"
- end
-
- port = #port > 0 and port or (pr == "https" and 443 or 80)
- path = #path > 0 and path or "/"
-
- options.depth = options.depth or 10
- local headers = options.headers or {}
- local protocol = options.protocol or "HTTP/1.1"
- headers["User-Agent"] = headers["User-Agent"] or "LuCI httpclient 0.1"
-
- if headers.Connection == nil then
- headers.Connection = "close"
- end
-
- if auth and not headers.Authorization then
- headers.Authorization = "Basic " .. nixio.bin.b64encode(auth)
- end
- local sock, code, msg = nixio.connect(host, port)
- if not sock then
- return nil, code, msg
- end
-
- sock:setsockopt("socket", "sndtimeo", options.sndtimeo or 15)
- sock:setsockopt("socket", "rcvtimeo", options.rcvtimeo or 15)
-
- if pr == "https" then
- local tls = options.tls_context or nixio.tls()
- sock = tls:create(sock)
- local stat, code, error = sock:connect()
- if not stat then
- return stat, code, error
- end
- end
-
- if protocol == "HTTP/1.1" then
- headers.Host = headers.Host or host
- end
-
- if type(options.body) == "table" then
- options.body = http.urlencode_params(options.body)
- end
- if type(options.body) == "string" then
- headers["Content-Length"] = headers["Content-Length"] or #options.body
- headers["Content-Type"] = headers["Content-Type"] or
- "application/x-www-form-urlencoded"
- options.method = options.method or "POST"
- end
-
- if type(options.body) == "function" then
- options.method = options.method or "POST"
- end
- if options.cookies then
- local cookiedata = {}
- for _, c in ipairs(options.cookies) do
- local cdo = c.flags.domain
- local cpa = c.flags.path
- if (cdo == host or cdo == "."..host or host:sub(-#cdo) == cdo)
- and (cpa == path or cpa == "/" or cpa .. "/" == path:sub(#cpa+1))
- and (not c.flags.secure or pr == "https")
- then
- cookiedata[#cookiedata+1] = c.key .. "=" .. c.value
- end
- end
- if headers["Cookie"] then
- headers["Cookie"] = headers["Cookie"] .. "; " .. table.concat(cookiedata, "; ")
- else
- headers["Cookie"] = table.concat(cookiedata, "; ")
- end
- end
-
- local message = {(options.method or "GET") .. " " .. path .. " " .. protocol}
-
- for k, v in pairs(headers) do
- if type(v) == "string" or type(v) == "number" then
- message[#message+1] = k .. ": " .. v
- elseif type(v) == "table" then
- for i, j in ipairs(v) do
- message[#message+1] = k .. ": " .. j
- end
- end
- end
- message[#message+1] = ""
- message[#message+1] = ""
-
-
- sock:sendall(table.concat(message, "\r\n"))
-
- if type(options.body) == "string" then
- sock:sendall(options.body)
- elseif type(options.body) == "function" then
- local res = {options.body(sock)}
- if not res[1] then
- sock:close()
- return unpack(res)
- end
- end
-
-
- local linesrc = sock:linesource()
- local line, code, error = linesrc()
-
- if not line then
- sock:close()
- return nil, code, error
- end
-
- local protocol, status, msg = line:match("^([%w./]+) ([0-9]+) (.*)")
-
- if not protocol then
- sock:close()
- return nil, -3, "invalid response magic: " .. line
- end
-
- local response = {
- status = line, headers = {}, code = 0, cookies = {}, uri = uri
- }
-
- line = linesrc()
- while line and line ~= "" do
- local key, val = line:match("^([%w-]+)%s?:%s?(.*)")
- if key and key ~= "Status" then
- if type(response.headers[key]) == "string" then
- response.headers[key] = {response.headers[key], val}
- elseif type(response.headers[key]) == "table" then
- response.headers[key][#response.headers[key]+1] = val
- else
- response.headers[key] = val
- end
- end
- line = linesrc()
- end
-
- if not line then
- sock:close()
- return nil, -4, "protocol error"
- end
-
-
- if response.headers["Set-Cookie"] then
- local cookies = response.headers["Set-Cookie"]
- for _, c in ipairs(type(cookies) == "table" and cookies or {cookies}) do
- local cobj = cookie_parse(c)
- cobj.flags.path = cobj.flags.path or path:match("(/.*)/?[^/]*")
- if not cobj.flags.domain or cobj.flags.domain == "" then
- cobj.flags.domain = host
- response.cookies[#response.cookies+1] = cobj
- else
- local hprt, cprt = {}, {}
-
-
- for part in host:gmatch("[^.]*") do
- table.insert(hprt, 1, part)
- end
- for part in cobj.flags.domain:gmatch("[^.]*") do
- table.insert(cprt, 1, part)
- end
-
- local valid = true
- for i, part in ipairs(cprt) do
-
- if hprt[i] ~= part and #part ~= 0 then
- valid = false
- break
-
- elseif hprt[i] ~= part and #part == 0 then
- if i ~= #cprt or (#hprt ~= i and #hprt+1 ~= i) then
- valid = false
- break
- end
- end
- end
-
- if valid and #cprt > 1 and #cprt[2] > 0 then
- response.cookies[#response.cookies+1] = cobj
- end
- end
- end
- end
-
-
- response.code = tonumber(status)
- 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 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
- if options.headers then
- options.headers.Host = nil
- end
- sock:close()
-
- return request_raw(nuri, options)
- end
- end
-
- return response.code, response, linesrc(true)..sock:readall(), sock
- end
- function cookie_parse(cookiestr)
- local key, val, flags = cookiestr:match("%s?([^=;]+)=?([^;]*)(.*)")
- if not key then
- return nil
- end
- local cookie = {key = key, value = val, flags = {}}
- for fkey, fval in flags:gmatch(";%s?([^=;]+)=?([^;]*)") do
- fkey = fkey:lower()
- if fkey == "expires" then
- fval = date.to_unix(fval:gsub("%-", " "))
- end
- cookie.flags[fkey] = fval
- end
- return cookie
- end
- function cookie_create(cookie)
- local cookiedata = {cookie.key .. "=" .. cookie.value}
- for k, v in pairs(cookie.flags) do
- if k == "expires" then
- v = date.to_http(v):gsub(", (%w+) (%w+) (%w+) ", ", %1-%2-%3 ")
- end
- cookiedata[#cookiedata+1] = k .. ((#v > 0) and ("=" .. v) or "")
- end
- return table.concat(cookiedata, "; ")
- end
|