b7f78481a2
If a filename contains dots, extension was parsed as everything after the first dot. File extension must be what it is after the last dot. Also includes a rewrited workaround for mimetypes if the requested file gzip compressed.
124 lines
4.8 KiB
Lua
124 lines
4.8 KiB
Lua
-- httpserver
|
|
-- Author: Marcos Kirsch
|
|
|
|
-- Starts web server in the specified port.
|
|
return function (port)
|
|
|
|
local s = net.createServer(net.TCP, 10) -- 10 seconds client timeout
|
|
s:listen(
|
|
port,
|
|
function (connection)
|
|
|
|
-- This variable holds the thread used for sending data back to the user.
|
|
-- We do it in a separate thread because we need to yield when sending lots
|
|
-- of data in order to avoid overflowing the mcu's buffer.
|
|
local connectionThread
|
|
|
|
local allowStatic = {GET=true, HEAD=true, POST=false, PUT=false, DELETE=false, TRACE=false, OPTIONS=false, CONNECT=false, PATCH=false}
|
|
|
|
local function onRequest(connection, req)
|
|
collectgarbage()
|
|
local method = req.method
|
|
local uri = req.uri
|
|
local fileServeFunction = nil
|
|
|
|
print("Method: " .. method);
|
|
|
|
if #(uri.file) > 32 then
|
|
-- nodemcu-firmware cannot handle long filenames.
|
|
uri.args = {code = 400, errorString = "Bad Request"}
|
|
fileServeFunction = dofile("httpserver-error.lc")
|
|
else
|
|
local fileExists = file.open(uri.file, "r")
|
|
file.close()
|
|
|
|
if not fileExists then
|
|
-- gzip check
|
|
fileExists = file.open(uri.file .. ".gz", "r")
|
|
file.close()
|
|
|
|
if fileExists then
|
|
print("gzip variant exists, serving that one")
|
|
uri.file = uri.file .. ".gz"
|
|
uri.isGzipped = true
|
|
end
|
|
end
|
|
|
|
if not fileExists then
|
|
uri.args = {code = 404, errorString = "Not Found"}
|
|
fileServeFunction = dofile("httpserver-error.lc")
|
|
elseif uri.isScript then
|
|
fileServeFunction = dofile(uri.file)
|
|
else
|
|
if allowStatic[method] then
|
|
uri.args = {file = uri.file, ext = uri.ext, gzipped = uri.isGzipped}
|
|
fileServeFunction = dofile("httpserver-static.lc")
|
|
else
|
|
uri.args = {code = 405, errorString = "Method not supported"}
|
|
fileServeFunction = dofile("httpserver-error.lc")
|
|
end
|
|
end
|
|
end
|
|
connectionThread = coroutine.create(fileServeFunction)
|
|
coroutine.resume(connectionThread, connection, req, uri.args)
|
|
end
|
|
|
|
local function onReceive(connection, payload)
|
|
collectgarbage()
|
|
local conf = dofile("httpserver-conf.lc")
|
|
local auth
|
|
local user = "Anonymous"
|
|
|
|
-- parse payload and decide what to serve.
|
|
local req = dofile("httpserver-request.lc")(payload)
|
|
print("Requested URI: " .. req.request)
|
|
if conf.auth.enabled then
|
|
auth = dofile("httpserver-basicauth.lc")
|
|
user = auth.authenticate(payload) -- authenticate returns nil on failed auth
|
|
end
|
|
|
|
if user and req.methodIsValid and (req.method == "GET" or req.method == "POST" or req.method == "PUT") then
|
|
onRequest(connection, req)
|
|
else
|
|
local args = {}
|
|
local fileServeFunction = dofile("httpserver-error.lc")
|
|
if not user then
|
|
args = {code = 401, errorString = "Not Authorized", headers = {auth.authErrorHeader()}}
|
|
elseif req.methodIsValid then
|
|
args = {code = 501, errorString = "Not Implemented"}
|
|
else
|
|
args = {code = 400, errorString = "Bad Request"}
|
|
end
|
|
connectionThread = coroutine.create(fileServeFunction)
|
|
coroutine.resume(connectionThread, connection, req, args)
|
|
end
|
|
end
|
|
|
|
local function onSent(connection, payload)
|
|
collectgarbage()
|
|
if connectionThread then
|
|
local connectionThreadStatus = coroutine.status(connectionThread)
|
|
if connectionThreadStatus == "suspended" then
|
|
-- Not finished sending file, resume.
|
|
coroutine.resume(connectionThread)
|
|
elseif connectionThreadStatus == "dead" then
|
|
-- We're done sending file.
|
|
connection:close()
|
|
connectionThread = nil
|
|
end
|
|
end
|
|
end
|
|
|
|
connection:on("receive", onReceive)
|
|
connection:on("sent", onSent)
|
|
|
|
end
|
|
)
|
|
-- false and nil evaluate as false
|
|
local ip = wifi.sta.getip()
|
|
if not ip then ip = wifi.ap.getip() end
|
|
print("nodemcu-httpserver running at http://" .. ip .. ":" .. port)
|
|
return s
|
|
|
|
end
|