mirror of
https://github.com/lunarmodules/luasocket.git
synced 2025-05-01 04:36:45 +02:00
Compare commits
3 Commits
1531e31dfb
...
8f3ed370ea
Author | SHA1 | Date | |
---|---|---|---|
|
8f3ed370ea | ||
|
c19b111bc1 | ||
|
13f2b3c663 |
112
src/http.lua
112
src/http.lua
@ -62,7 +62,7 @@ local function receiveheaders(sock, headers)
|
|||||||
-- unfold any folded values
|
-- unfold any folded values
|
||||||
while string.find(line, "^%s") do
|
while string.find(line, "^%s") do
|
||||||
value = value .. line
|
value = value .. line
|
||||||
line = sock:receive()
|
line, err = sock:receive()
|
||||||
if err then return nil, err end
|
if err then return nil, err end
|
||||||
end
|
end
|
||||||
-- save pair in table
|
-- save pair in table
|
||||||
@ -203,15 +203,8 @@ end
|
|||||||
-----------------------------------------------------------------------------
|
-----------------------------------------------------------------------------
|
||||||
local function adjusturi(reqt)
|
local function adjusturi(reqt)
|
||||||
local u = reqt
|
local u = reqt
|
||||||
local proxy
|
-- if there is a proxy, we need the full url. otherwise, just a part.
|
||||||
if reqt.proxy then
|
if not reqt.proxy and not _M.PROXY then
|
||||||
proxy = url.parse(reqt.proxy)
|
|
||||||
end
|
|
||||||
|
|
||||||
-- We just need the path if there's no proxy,
|
|
||||||
-- or if we use https over an http proxy.
|
|
||||||
-- Otherwise, we use a full url
|
|
||||||
if not proxy or (reqt.scheme == "https" and proxy.scheme == "http") then
|
|
||||||
u = {
|
u = {
|
||||||
path = socket.try(reqt.path, "invalid path 'nil'"),
|
path = socket.try(reqt.path, "invalid path 'nil'"),
|
||||||
params = reqt.params,
|
params = reqt.params,
|
||||||
@ -222,7 +215,19 @@ local function adjusturi(reqt)
|
|||||||
return url.build(u)
|
return url.build(u)
|
||||||
end
|
end
|
||||||
|
|
||||||
local function adjustheaders(reqt, https_connect)
|
local function adjustproxy(reqt)
|
||||||
|
local proxy = reqt.proxy or _M.PROXY
|
||||||
|
if proxy then
|
||||||
|
proxy = url.parse(proxy)
|
||||||
|
proxy.port = proxy.port or 3128
|
||||||
|
proxy.create = SCHEMES[proxy.scheme].create(reqt)
|
||||||
|
return proxy.host, proxy.port, proxy.create
|
||||||
|
else
|
||||||
|
return reqt.host, reqt.port, reqt.create
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
local function adjustheaders(reqt)
|
||||||
-- default headers
|
-- default headers
|
||||||
local host = reqt.host
|
local host = reqt.host
|
||||||
local port = tostring(reqt.port)
|
local port = tostring(reqt.port)
|
||||||
@ -241,8 +246,9 @@ local function adjustheaders(reqt, https_connect)
|
|||||||
url.unescape(reqt.password)))
|
url.unescape(reqt.password)))
|
||||||
end
|
end
|
||||||
-- if we have proxy authentication information, pass it along
|
-- if we have proxy authentication information, pass it along
|
||||||
if reqt.proxy and (reqt.scheme == "http" or https_connect) then
|
local proxy = reqt.proxy or _M.PROXY
|
||||||
local proxy = url.parse(reqt.proxy)
|
if proxy then
|
||||||
|
proxy = url.parse(proxy)
|
||||||
if proxy.user and proxy.password then
|
if proxy.user and proxy.password then
|
||||||
lower["proxy-authorization"] =
|
lower["proxy-authorization"] =
|
||||||
"Basic " .. (mime.b64(proxy.user .. ":" .. proxy.password))
|
"Basic " .. (mime.b64(proxy.user .. ":" .. proxy.password))
|
||||||
@ -255,80 +261,6 @@ local function adjustheaders(reqt, https_connect)
|
|||||||
return lower
|
return lower
|
||||||
end
|
end
|
||||||
|
|
||||||
local function reg(conn)
|
|
||||||
local mt = getmetatable(conn.sock).__index
|
|
||||||
for name, method in pairs(mt) do
|
|
||||||
if type(method) == "function" then
|
|
||||||
conn[name] = function (self, ...)
|
|
||||||
return method(self.sock, ...)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
local function proxy_connect_create(params, proxy)
|
|
||||||
-- Copied and adapted from luasec's https.lua
|
|
||||||
-- in function ssl.http.tcp()
|
|
||||||
|
|
||||||
local ssl = assert(
|
|
||||||
require("ssl"), 'LuaSocket: LuaSec not found')
|
|
||||||
local https = assert(
|
|
||||||
require("ssl.https"), 'LuaSocket: LuaSec not found')
|
|
||||||
|
|
||||||
-- Force client mode
|
|
||||||
params.mode = "client"
|
|
||||||
-- 'create' function
|
|
||||||
return function ()
|
|
||||||
local conn = {}
|
|
||||||
conn.proxy_sock = _M.open(proxy.host, proxy.port, proxy.create)
|
|
||||||
local try = conn.proxy_sock.try
|
|
||||||
|
|
||||||
function conn:settimeout(...)
|
|
||||||
return self.proxy_sock.c:settimeout(https.TIMEOUT)
|
|
||||||
end
|
|
||||||
|
|
||||||
-- Wrap the underlying connection function
|
|
||||||
function conn:connect(host, port)
|
|
||||||
conn.proxy_sock:sendrequestline("CONNECT", host .. ":" .. tostring(port))
|
|
||||||
conn.proxy_sock:sendheaders(adjustheaders(proxy, true))
|
|
||||||
|
|
||||||
local code, _ = conn.proxy_sock:receivestatusline()
|
|
||||||
try(code == 200 or nil)
|
|
||||||
|
|
||||||
self.sock = try(ssl.wrap(self.proxy_sock.c, params))
|
|
||||||
self.sock:sni(host)
|
|
||||||
self.sock:settimeout(https.TIMEOUT)
|
|
||||||
try(self.sock:dohandshake())
|
|
||||||
reg(self)
|
|
||||||
return 1
|
|
||||||
end
|
|
||||||
|
|
||||||
-- Close the underlying socket
|
|
||||||
function conn:close()
|
|
||||||
conn.proxy_sock:close()
|
|
||||||
end
|
|
||||||
|
|
||||||
return conn
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
local function adjustproxy(reqt)
|
|
||||||
if reqt.proxy then
|
|
||||||
local proxy = url.parse(reqt.proxy)
|
|
||||||
proxy.port = proxy.port or 3128
|
|
||||||
proxy.create = SCHEMES[proxy.scheme].create(reqt)
|
|
||||||
|
|
||||||
if reqt.scheme == "https" and proxy.scheme == "http" then
|
|
||||||
local wrapped_create = proxy_connect_create(reqt, proxy)
|
|
||||||
return reqt.host, reqt.port, wrapped_create
|
|
||||||
else
|
|
||||||
return proxy.host, proxy.port, proxy.create
|
|
||||||
end
|
|
||||||
else
|
|
||||||
return reqt.host, reqt.port, reqt.create
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
-- default url parts
|
-- default url parts
|
||||||
local default = {
|
local default = {
|
||||||
path ="/"
|
path ="/"
|
||||||
@ -338,8 +270,6 @@ local default = {
|
|||||||
local function adjustrequest(reqt)
|
local function adjustrequest(reqt)
|
||||||
-- parse url if provided
|
-- parse url if provided
|
||||||
local nreqt = reqt.url and url.parse(reqt.url, default) or {}
|
local nreqt = reqt.url and url.parse(reqt.url, default) or {}
|
||||||
-- get global proxy
|
|
||||||
nreqt.proxy = reqt.proxy or _M.PROXY
|
|
||||||
-- explicit components override url
|
-- explicit components override url
|
||||||
for i,v in base.pairs(reqt) do nreqt[i] = v end
|
for i,v in base.pairs(reqt) do nreqt[i] = v end
|
||||||
-- default to scheme particulars
|
-- default to scheme particulars
|
||||||
@ -354,7 +284,7 @@ local function adjustrequest(reqt)
|
|||||||
-- compute uri if user hasn't overriden
|
-- compute uri if user hasn't overriden
|
||||||
nreqt.uri = reqt.uri or adjusturi(nreqt)
|
nreqt.uri = reqt.uri or adjusturi(nreqt)
|
||||||
-- adjust headers in request
|
-- adjust headers in request
|
||||||
nreqt.headers = adjustheaders(nreqt, false)
|
nreqt.headers = adjustheaders(nreqt)
|
||||||
if nreqt.source
|
if nreqt.source
|
||||||
and not nreqt.headers["content-length"]
|
and not nreqt.headers["content-length"]
|
||||||
and not nreqt.headers["transfer-encoding"]
|
and not nreqt.headers["transfer-encoding"]
|
||||||
@ -362,7 +292,7 @@ local function adjustrequest(reqt)
|
|||||||
nreqt.headers["transfer-encoding"] = "chunked"
|
nreqt.headers["transfer-encoding"] = "chunked"
|
||||||
end
|
end
|
||||||
|
|
||||||
-- adjust host, port and create if there is a proxy
|
-- ajust host and port if there is a proxy
|
||||||
nreqt.host, nreqt.port, nreqt.create = adjustproxy(nreqt)
|
nreqt.host, nreqt.port, nreqt.create = adjustproxy(nreqt)
|
||||||
return nreqt
|
return nreqt
|
||||||
end
|
end
|
||||||
|
Loading…
x
Reference in New Issue
Block a user