311 lines
7.3 KiB
Lua
311 lines
7.3 KiB
Lua
|
|
|
|
|
|
|
|
|
|
|
|
require"socket"
|
|
require"utils"
|
|
module(..., package.seeall)
|
|
|
|
local function response(client,cbFnc,result,prompt,head,body)
|
|
if not result then log.error("http.response",result,prompt) end
|
|
if cbFnc then cbFnc(result,prompt,head,body) end
|
|
if client then client:close() end
|
|
end
|
|
|
|
local function receive(client,timeout,cbFnc,result,prompt,head,body)
|
|
local res,data = client:recv(timeout)
|
|
if not res then
|
|
response(client,cbFnc,result,prompt or "receive timeout",head,body)
|
|
end
|
|
return res,data
|
|
end
|
|
|
|
local function getFileBase64Len(s)
|
|
if s then return (io.fileSize(s)+2)/3*4 end
|
|
end
|
|
|
|
local function taskClient(method,protocal,auth,host,port,path,cert,head,body,timeout,cbFnc,rcvFilePath,tCoreExtPara)
|
|
log.info("http path",path)
|
|
while not socket.isReady() do
|
|
if not sys.waitUntil("IP_READY_IND",timeout) then return response(nil,cbFnc,false,"network not ready") end
|
|
end
|
|
|
|
|
|
local bodyLen = 0
|
|
if body then
|
|
if type(body)=="string" then
|
|
bodyLen = body:len()
|
|
elseif type(body)=="table" then
|
|
for i=1,#body do
|
|
bodyLen = bodyLen + (type(body[i])=="string" and string.len(body[i]) or getFileBase64Len(body[i].file_base64) or io.fileSize(body[i].file))
|
|
end
|
|
end
|
|
end
|
|
|
|
|
|
local heads = head or {}
|
|
if not heads.Host then heads["Host"] = (port ~= 80 and port ~= 443) and (host..":"..port) or host end
|
|
if not heads.Connection then heads["Connection"] = "short" end
|
|
if bodyLen>0 and bodyLen~=tonumber(heads["Content-Length"] or "0") then heads["Content-Length"] = bodyLen end
|
|
if auth~="" and not heads.Authorization then heads["Authorization"] = ("Basic "..crypto.base64_encode(auth,#auth)) end
|
|
local headStr = ""
|
|
for k,v in pairs(heads) do
|
|
headStr = headStr..k..": "..v.."\r\n"
|
|
end
|
|
headStr = headStr.."\r\n"
|
|
|
|
local client = socket.tcp(protocal=="https",cert,tCoreExtPara)
|
|
if not client then return response(nil,cbFnc,false,"create socket error") end
|
|
if not client:connect(host,port,timeout/1000) then
|
|
return response(client,cbFnc,false,"connect fail")
|
|
end
|
|
|
|
|
|
if not client:send(method.." "..path.." HTTP/1.1".."\r\n"..headStr..(type(body)=="string" and body or "")) then
|
|
return response(client,cbFnc,false,"send head fail")
|
|
end
|
|
|
|
|
|
if type(body)=="table" then
|
|
for i=1,#body do
|
|
if type(body[i])=="string" then
|
|
if not client:send(body[i]) then
|
|
return response(client,cbFnc,false,"send body fail")
|
|
end
|
|
else
|
|
local file = io.open(body[i].file or body[i].file_base64,"rb")
|
|
if file then
|
|
while true do
|
|
local dat = file:read(body[i].file and 11200 or 8400)
|
|
if not dat then
|
|
io.close(file)
|
|
break
|
|
end
|
|
if body[i].file_base64 then dat=crypto.base64_encode(dat,#dat) end
|
|
if not client:send(dat) then
|
|
io.close(file)
|
|
return response(client,cbFnc,false,"send file fail")
|
|
end
|
|
end
|
|
else
|
|
return response(client,cbFnc,false,"send file open fail")
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
local rcvCache,rspHead,rspBody,d1,d2,result,data,statusCode,rcvChunked,contentLen = "",{},{}
|
|
|
|
while true do
|
|
result,data = receive(client,timeout,cbFnc,false,nil,rspHead,rcvFilePath or table.concat(rspBody))
|
|
if not result then return end
|
|
rcvCache = rcvCache..data
|
|
d1,d2 = rcvCache:find("\r\n\r\n")
|
|
if d2 then
|
|
|
|
_,d1,statusCode = rcvCache:find("%s(%d+)%s.-\r\n")
|
|
if not statusCode then
|
|
return response(client,cbFnc,false,"parse received status error",rspHead,rcvFilePath or table.concat(rspBody))
|
|
end
|
|
|
|
for k,v in string.gmatch(rcvCache:sub(d1+1,d2-2),"(.-):%s*(.-)\r\n") do
|
|
rspHead[k] = v
|
|
if (string.upper(k)==string.upper("Transfer-Encoding")) and (string.upper(v)==string.upper("chunked")) then rcvChunked = true end
|
|
end
|
|
if not rcvChunked then
|
|
contentLen = tonumber(rspHead["Content-Length"] or "2147483647")
|
|
end
|
|
if method == "HEAD" then
|
|
contentLen = 0
|
|
end
|
|
|
|
rcvCache = rcvCache:sub(d2+1,-1)
|
|
break
|
|
end
|
|
end
|
|
|
|
|
|
if rcvChunked then
|
|
local chunkSize
|
|
|
|
while true do
|
|
|
|
if not chunkSize then
|
|
d1,d2,chunkSize = rcvCache:find("(%x+)\r\n")
|
|
if chunkSize then
|
|
chunkSize = tonumber(chunkSize,16)
|
|
rcvCache = rcvCache:sub(d2+1,-1)
|
|
else
|
|
result,data = receive(client,timeout,cbFnc,false,nil,rspHead,rcvFilePath or table.concat(rspBody))
|
|
if not result then return end
|
|
rcvCache = rcvCache..data
|
|
end
|
|
end
|
|
|
|
|
|
|
|
|
|
if chunkSize then
|
|
if rcvCache:len()<chunkSize+2 then
|
|
result,data = receive(client,timeout,cbFnc,false,nil,rspHead,rcvFilePath or table.concat(rspBody))
|
|
if not result then return end
|
|
rcvCache = rcvCache..data
|
|
else
|
|
if chunkSize>0 then
|
|
local chunkData = rcvCache:sub(1,chunkSize)
|
|
|
|
if type(rcvFilePath)=="string" then
|
|
local file = io.open(rcvFilePath,"a+")
|
|
if not file then return response(client,cbFnc,false,"receive: open file error",rspHead,rcvFilePath or table.concat(rspBody)) end
|
|
if not file:write(chunkData) then response(client,cbFnc,false,"receive: write file error",rspHead,rcvFilePath or table.concat(rspBody)) end
|
|
file:close()
|
|
elseif type(rcvFilePath)=="function" then
|
|
local userResult = rcvFilePath(data,rspHead["Content-Range"] and tonumber((rspHead["Content-Range"]):match("/(%d+)")) or contentLen,statusCode)
|
|
if userResult~=nil then
|
|
return response(client,cbFnc,userResult,userResult and statusCode or "receive: user process error",rspHead)
|
|
end
|
|
else
|
|
table.insert(rspBody,chunkData)
|
|
end
|
|
rcvCache = rcvCache:sub(chunkSize+3,-1)
|
|
chunkSize = nil
|
|
elseif chunkSize==0 then
|
|
return response(client,cbFnc,true,statusCode,rspHead,rcvFilePath or table.concat(rspBody))
|
|
end
|
|
end
|
|
end
|
|
end
|
|
else
|
|
local rmnLen = contentLen
|
|
while true do
|
|
data = rcvCache:len()<=rmnLen and rcvCache or rcvCache:sub(1,rmnLen)
|
|
if type(rcvFilePath)=="string" then
|
|
if data:len()>0 then
|
|
local file = io.open(rcvFilePath,"a+")
|
|
if not file then return response(client,cbFnc,false,"receive: open file error",rspHead,rcvFilePath or table.concat(rspBody)) end
|
|
if not file:write(data) then response(client,cbFnc,false,"receive: write file error",rspHead,rcvFilePath or table.concat(rspBody)) end
|
|
file:close()
|
|
end
|
|
elseif type(rcvFilePath)=="function" then
|
|
local userResult = rcvFilePath(data,rspHead["Content-Range"] and tonumber((rspHead["Content-Range"]):match("/(%d+)")) or contentLen,statusCode)
|
|
if userResult~=nil then
|
|
return response(client,cbFnc,userResult,userResult and statusCode or "receive: user process error",rspHead)
|
|
end
|
|
else
|
|
table.insert(rspBody,data)
|
|
end
|
|
rmnLen = rmnLen-data:len()
|
|
if rmnLen==0 then break end
|
|
result,rcvCache = receive(client,timeout,cbFnc,contentLen==0x7FFFFFFF,contentLen==0x7FFFFFFF and statusCode or nil,rspHead,rcvFilePath or table.concat(rspBody))
|
|
if not result then return end
|
|
end
|
|
return response(client,cbFnc,true,statusCode,rspHead,rcvFilePath or table.concat(rspBody))
|
|
end
|
|
end
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
function request(method,url,cert,head,body,timeout,cbFnc,rcvFileName,tCoreExtPara)
|
|
local protocal,auth,hostName,port,path,d1,d2,offset,rcvFilePath
|
|
d1,d2,protocal = url:find("^(%a+)://")
|
|
if not protocal then protocal = "http" end
|
|
offset = d2 or 0
|
|
|
|
d1,d2,auth = url:find("(.-:.-)@",offset+1)
|
|
offset = d2 or offset
|
|
|
|
if url:match("^[^/]+:(%d+)",offset+1) then
|
|
d1,d2,hostName,port = url:find("^([^/]+):(%d+)",offset+1)
|
|
else
|
|
d1,d2,hostName = url:find("(.-)/",offset+1)
|
|
if hostName then
|
|
d2 = d2-1
|
|
else
|
|
hostName = url:sub(offset+1,-1)
|
|
offset = url:len()
|
|
end
|
|
end
|
|
|
|
if not hostName then return response(nil,cbFnc,false,"Invalid url, can't get host") end
|
|
if port=="" or not port then port = (protocal=="https" and 443 or 80) end
|
|
offset = d2 or offset
|
|
|
|
path = url:sub(offset+1,-1)
|
|
|
|
sys.taskInit(taskClient,method,protocal,auth or "",hostName,port,path=="" and "/" or path,cert,head,body or "",timeout or 30000,cbFnc,rcvFileName,tCoreExtPara)
|
|
if type(rcvFileName) == "string" then
|
|
return rcvFileName
|
|
end
|
|
end
|