dany/luasocket
1
0
Fork 0
mirror of https://github.com/lunarmodules/luasocket.git synced 2024-12-27 04:48:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

Updated for Lua 4.1-w3.

Browse Source 
Dealing with 100 response codes.
This commit is contained in:
Diego Nehab 2002-07-08 21:01:18 +00:00
parent 9d7f43768a
commit 60e7bf48b0
1 changed files with 55 additions and 54 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

109
src/http.lua
View File

@ -1,6 +1,6 @@
----------------------------------------------------------------------------- -----------------------------------------------------------------------------
-- HTTP/1.1 client support for the Lua language. -- HTTP/1.1 client support for the Lua language.
-- LuaSocket 1.4 toolkit. -- LuaSocket 1.5 toolkit.
-- Author: Diego Nehab -- Author: Diego Nehab
-- Date: 26/12/2000 -- Date: 26/12/2000
-- Conforming to: RFC 2616, LTN7 -- Conforming to: RFC 2616, LTN7
@ -18,17 +18,10 @@ Public.TIMEOUT = 60
-- default port for document retrieval -- default port for document retrieval
Public.PORT = 80 Public.PORT = 80
-- user agent field sent in request -- user agent field sent in request
Public.USERAGENT = "LuaSocket 1.4" Public.USERAGENT = "LuaSocket 1.5"
-- block size used in transfers -- block size used in transfers
Public.BLOCKSIZE = 8192 Public.BLOCKSIZE = 8192
-----------------------------------------------------------------------------
-- Required libraries
-----------------------------------------------------------------------------
dofile "concat.lua"
dofile "url.lua"
dofile "code.lua"
----------------------------------------------------------------------------- -----------------------------------------------------------------------------
-- Tries to get a pattern from the server and closes socket on error -- Tries to get a pattern from the server and closes socket on error
-- sock: socket connected to the server -- sock: socket connected to the server
@ -84,8 +77,8 @@ end
----------------------------------------------------------------------------- -----------------------------------------------------------------------------
function Private.receive_status(sock) function Private.receive_status(sock)
local line, err local line, err
line, err = %Private.try_receive(sock) line, err = Private.try_receive(sock)
if not err then return %Private.get_statuscode(line), line if not err then return Private.get_statuscode(line), line
else return nil, nil, err end else return nil, nil, err end
end end
@ -104,7 +97,7 @@ function Private.receive_headers(sock, headers)
local line, err local line, err
local name, value, _ local name, value, _
-- get first line -- get first line
line, err = %Private.try_receive(sock) line, err = Private.try_receive(sock)
if err then return nil, err end if err then return nil, err end
-- headers go until a blank line is found -- headers go until a blank line is found
while line ~= "" do while line ~= "" do
@ -116,12 +109,12 @@ function Private.receive_headers(sock, headers)
end end
name = strlower(name) name = strlower(name)
-- get next line (value might be folded) -- get next line (value might be folded)
line, err = %Private.try_receive(sock) line, err = Private.try_receive(sock)
if err then return nil, err end if err then return nil, err end
-- unfold any folded values -- unfold any folded values
while not err and strfind(line, "^%s") do while not err and strfind(line, "^%s") do
value = value .. line value = value .. line
line, err = %Private.try_receive(sock) line, err = Private.try_receive(sock)
if err then return nil, err end if err then return nil, err end
end end
-- save pair in table -- save pair in table
@ -144,7 +137,7 @@ function Private.receivebody_bychunks(sock, headers, receive_cb)
local chunk, size, line, err, go, uerr, _ local chunk, size, line, err, go, uerr, _
while 1 do while 1 do
-- get chunk size, skip extention -- get chunk size, skip extention
line, err = %Private.try_receive(sock) line, err = Private.try_receive(sock)
if err then if err then
local go, uerr = receive_cb(nil, err) local go, uerr = receive_cb(nil, err)
return uerr or err return uerr or err
@ -159,7 +152,7 @@ function Private.receivebody_bychunks(sock, headers, receive_cb)
-- was it the last chunk? -- was it the last chunk?
if size <= 0 then break end if size <= 0 then break end
-- get chunk -- get chunk
chunk, err = %Private.try_receive(sock, size) chunk, err = Private.try_receive(sock, size)
if err then if err then
go, uerr = receive_cb(nil, err) go, uerr = receive_cb(nil, err)
return uerr or err return uerr or err
@ -171,7 +164,7 @@ function Private.receivebody_bychunks(sock, headers, receive_cb)
return uerr or "aborted by callback" return uerr or "aborted by callback"
end end
-- skip CRLF on end of chunk -- skip CRLF on end of chunk
_, err = %Private.try_receive(sock) _, err = Private.try_receive(sock)
if err then if err then
go, uerr = receive_cb(nil, err) go, uerr = receive_cb(nil, err)
return uerr or err return uerr or err
@ -180,7 +173,7 @@ function Private.receivebody_bychunks(sock, headers, receive_cb)
-- the server should not send trailer headers because we didn't send a -- the server should not send trailer headers because we didn't send a
-- header informing it we know how to deal with them. we do not risk -- header informing it we know how to deal with them. we do not risk
-- being caught unprepaired. -- being caught unprepaired.
headers, err = %Private.receive_headers(sock, headers) headers, err = Private.receive_headers(sock, headers)
if err then if err then
go, uerr = receive_cb(nil, err) go, uerr = receive_cb(nil, err)
return uerr or err return uerr or err
@ -202,7 +195,7 @@ end
function Private.receivebody_bylength(sock, length, receive_cb) function Private.receivebody_bylength(sock, length, receive_cb)
local uerr, go local uerr, go
while length > 0 do while length > 0 do
local size = min(%Public.BLOCKSIZE, length) local size = min(Public.BLOCKSIZE, length)
local chunk, err = sock:receive(size) local chunk, err = sock:receive(size)
if err then if err then
go, uerr = receive_cb(nil, err) go, uerr = receive_cb(nil, err)
@ -230,7 +223,7 @@ end
function Private.receivebody_untilclosed(sock, receive_cb) function Private.receivebody_untilclosed(sock, receive_cb)
local err, go, uerr local err, go, uerr
while 1 do while 1 do
local chunk, err = sock:receive(%Public.BLOCKSIZE) local chunk, err = sock:receive(Public.BLOCKSIZE)
if err == "closed" or not err then if err == "closed" or not err then
go, uerr = receive_cb(chunk) go, uerr = receive_cb(chunk)
if not go then if not go then
@ -260,14 +253,14 @@ function Private.receive_body(sock, headers, receive_cb)
local te = headers["transfer-encoding"] local te = headers["transfer-encoding"]
if te and te ~= "identity" then if te and te ~= "identity" then
-- get by chunked transfer-coding of message body -- get by chunked transfer-coding of message body
return %Private.receivebody_bychunks(sock, headers, receive_cb) return Private.receivebody_bychunks(sock, headers, receive_cb)
elseif tonumber(headers["content-length"]) then elseif tonumber(headers["content-length"]) then
-- get by content-length -- get by content-length
local length = tonumber(headers["content-length"]) local length = tonumber(headers["content-length"])
return %Private.receivebody_bylength(sock, length, receive_cb) return Private.receivebody_bylength(sock, length, receive_cb)
else else
-- get it all until connection closes -- get it all until connection closes
return %Private.receivebody_untilclosed(sock, receive_cb) return Private.receivebody_untilclosed(sock, receive_cb)
end end
end end
@ -280,7 +273,7 @@ end
-- nil if successfull or an error message in case of error -- nil if successfull or an error message in case of error
----------------------------------------------------------------------------- -----------------------------------------------------------------------------
function Private.drop_body(sock, headers) function Private.drop_body(sock, headers)
return %Private.receive_body(sock, headers, function (c, e) return 1 end) return Private.receive_body(sock, headers, function (c, e) return 1 end)
end end
----------------------------------------------------------------------------- -----------------------------------------------------------------------------
@ -325,11 +318,11 @@ function Private.send_headers(sock, headers)
headers = headers or {} headers = headers or {}
-- send request headers -- send request headers
for i, v in headers do for i, v in headers do
err = %Private.try_send(sock, i .. ": " .. v .. "\r\n") err = Private.try_send(sock, i .. ": " .. v .. "\r\n")
if err then return err end if err then return err end
end end
-- mark end of request headers -- mark end of request headers
return %Private.try_send(sock, "\r\n") return Private.try_send(sock, "\r\n")
end end
----------------------------------------------------------------------------- -----------------------------------------------------------------------------
@ -346,7 +339,7 @@ end
function Private.send_request(sock, method, uri, headers, body_cb) function Private.send_request(sock, method, uri, headers, body_cb)
local chunk, size, done, err local chunk, size, done, err
-- send request line -- send request line
err = %Private.try_send(sock, method .. " " .. uri .. " HTTP/1.1\r\n") err = Private.try_send(sock, method .. " " .. uri .. " HTTP/1.1\r\n")
if err then return err end if err then return err end
-- if there is a request message body, add content-length header -- if there is a request message body, add content-length header
if body_cb then if body_cb then
@ -360,11 +353,11 @@ function Private.send_request(sock, method, uri, headers, body_cb)
end end
end end
-- send request headers -- send request headers
err = %Private.send_headers(sock, headers) err = Private.send_headers(sock, headers)
if err then return err end if err then return err end
-- send request message body, if any -- send request message body, if any
if body_cb then if body_cb then
return %Private.send_indirect(sock, body_cb, chunk, size) return Private.send_indirect(sock, body_cb, chunk, size)
end end
end end
@ -395,7 +388,7 @@ function Private.fill_headers(headers, parsed)
local lower = {} local lower = {}
headers = headers or {} headers = headers or {}
-- set default headers -- set default headers
lower["user-agent"] = %Public.USERAGENT lower["user-agent"] = Public.USERAGENT
-- override with user values -- override with user values
for i,v in headers do for i,v in headers do
lower[strlower(i)] = v lower[strlower(i)] = v
@ -442,7 +435,7 @@ function Private.authorize(request, parsed, response)
body_cb = request.body_cb, body_cb = request.body_cb,
headers = request.headers headers = request.headers
} }
return %Public.request_cb(authorize, response) return Public.request_cb(authorize, response)
end end
----------------------------------------------------------------------------- -----------------------------------------------------------------------------
@ -482,7 +475,7 @@ function Private.redirect(request, response)
body_cb = request.body_cb, body_cb = request.body_cb,
headers = request.headers headers = request.headers
} }
local response = %Public.request_cb(redirect, response) local response = Public.request_cb(redirect, response)
-- we pass the location header as a clue we tried to redirect -- we pass the location header as a clue we tried to redirect
if response.headers then response.headers.location = redirect.url end if response.headers then response.headers.location = redirect.url end
return response return response
@ -544,7 +537,7 @@ end
function Public.request_cb(request, response) function Public.request_cb(request, response)
local parsed = URL.parse_url(request.url, { local parsed = URL.parse_url(request.url, {
host = "", host = "",
port = %Public.PORT, port = Public.PORT,
path ="/", path ="/",
scheme = "http" scheme = "http"
}) })
@ -558,35 +551,43 @@ function Public.request_cb(request, response)
-- default method -- default method
request.method = request.method or "GET" request.method = request.method or "GET"
-- fill default headers -- fill default headers
request.headers = %Private.fill_headers(request.headers, parsed) request.headers = Private.fill_headers(request.headers, parsed)
-- try to connect to server -- try to connect to server
local sock local sock
sock, response.error = connect(parsed.host, parsed.port) sock, response.error = connect(parsed.host, parsed.port)
if not sock then return response end if not sock then return response end
-- set connection timeout so that we do not hang forever -- set connection timeout so that we do not hang forever
sock:timeout(%Public.TIMEOUT) sock:timeout(Public.TIMEOUT)
-- send request message -- send request message
response.error = %Private.send_request(sock, request.method, response.error = Private.send_request(sock, request.method,
%Private.request_uri(parsed), request.headers, request.body_cb) Private.request_uri(parsed), request.headers, request.body_cb)
if response.error then return response end if response.error then return response end
-- get server response message -- get server response message
response.code, response.status, response.error = response.code, response.status, response.error =
%Private.receive_status(sock) Private.receive_status(sock)
if response.error then return response end if response.error then return response end
-- deal with 1xx status
if response.code == 100 then
response.headers, response.error = Private.receive_headers(sock, {})
if response.error then return response end
response.code, response.status, response.error =
Private.receive_status(sock)
if response.error then return response end
end
-- receive all headers -- receive all headers
response.headers, response.error = %Private.receive_headers(sock, {}) response.headers, response.error = Private.receive_headers(sock, {})
if response.error then return response end if response.error then return response end
-- decide what to do based on request and response parameters -- decide what to do based on request and response parameters
if %Private.should_redirect(request, response) then if Private.should_redirect(request, response) then
%Private.drop_body(sock, response.headers) Private.drop_body(sock, response.headers)
sock:close() sock:close()
return %Private.redirect(request, response) return Private.redirect(request, response)
elseif %Private.should_authorize(request, parsed, response) then elseif Private.should_authorize(request, parsed, response) then
%Private.drop_body(sock, response.headers) Private.drop_body(sock, response.headers)
sock:close() sock:close()
return %Private.authorize(request, parsed, response) return Private.authorize(request, parsed, response)
elseif %Private.has_body(request, response) then elseif Private.has_body(request, response) then
response.error = %Private.receive_body(sock, response.headers, response.error = Private.receive_body(sock, response.headers,
response.body_cb) response.body_cb)
if response.error then return response end if response.error then return response end
sock:close() sock:close()
@ -618,15 +619,15 @@ function Public.request(request)
local response = {} local response = {}
if request.body then if request.body then
request.body_cb = function() request.body_cb = function()
return %request.body, strlen(%request.body) return request.body, strlen(request.body)
end end
end end
local cat = Concat.create() local cat = Concat.create()
response.body_cb = function(chunk, err) response.body_cb = function(chunk, err)
if chunk then %cat:addstring(chunk) end if chunk then cat:addstring(chunk) end
return 1 return 1
end end
response = %Public.request_cb(request, response) response = Public.request_cb(request, response)
response.body = cat:getresult() response.body = cat:getresult()
response.body_cb = nil response.body_cb = nil
return response return response
@ -646,9 +647,9 @@ end
-- error: error message if any -- error: error message if any
----------------------------------------------------------------------------- -----------------------------------------------------------------------------
function Public.get(url_or_request) function Public.get(url_or_request)
local request = %Private.build_request(url_or_request) local request = Private.build_request(url_or_request)
request.method = "GET" request.method = "GET"
local response = %Public.request(request) local response = Public.request(request)
return response.body, response.headers, return response.body, response.headers,
response.code, response.error response.code, response.error
end end
@ -669,10 +670,10 @@ end
-- error: error message, or nil if successfull -- error: error message, or nil if successfull
----------------------------------------------------------------------------- -----------------------------------------------------------------------------
function Public.post(url_or_request, body) function Public.post(url_or_request, body)
local request = %Private.build_request(url_or_request) local request = Private.build_request(url_or_request)
request.method = "POST" request.method = "POST"
request.body = request.body or body request.body = request.body or body
local response = %Public.request(request) local response = Public.request(request)
return response.body, response.headers, return response.body, response.headers,
response.code, response.error response.code, response.error
end end