~philho/+junk/Lua : contents of TestCGI.lua at revision 41

~philho/+junk/Lua : (revision 41)
#!Lua-5.0.exe
-- Test of CGI capability.
-- by Philippe Lhoste <PhiLho(a)GMX.net> http://Phi.Lho.free.fr
-- v. 1.2 -- 2003/07/07 -- Converted to Lua 5.0, improved split and encode entities
-- v. 1.1 -- 2002/05/17 -- Added some other environment variables, encoded output
-- v. 1.0 -- 2001/08/07

-- Maximum length of input (to avoid overflooding)
maxInput = 5 * 1024	-- 5KB

programName = "TestCGI.lua"
programVersion = "1.1"

-- io.write HTML header
io.write[[
Content-Type: text/html

<html>
]]

-- Page to ouput
htmlPageHead = [[
<head>
	<title>Lua CGI Test</title>
	<meta HTTP-EQUIV="Pragma" CONTENT="no-cache">
	<meta name="GENERATOR" content="%s %s">
</head>
<body>
<center><h1>Test of CGI capability using Lua</h1></center>
]]

io.write(format(htmlPageHead, programName, programVersion))

function Split(separator, toSplit)
	local splitted = {}
	local i = 0
	local re = "([^" .. separator .. "]*)" .. separator .. "?"
	for part in string.gfind(toSplit .. separator, re) do
		i = i + 1
		splitted[i] = part
	end
	splitted[i] = nil
	return splitted
end

function OutputError(message)
	io.write('<p><font color="red">', message, '</font></p>')
end

function DecodeString(encodedString)
	if encodedString == nil or type(encodedString) ~= "string" then
		return ''
	end
	-- Convert plus to space
	local decodedString = string.gsub(encodedString, '+', ' ')
	-- Convert %XX from hex numbers to alphanumeric
	decodedString = string.gsub(decodedString, "%%(%x%x)", function (hexa) return string.char(tonumber(hexa, 16)) end)
	return decodedString
end

entities =
{
	['&'] = "&amp;",
	['<'] = "&lt;",
	['>'] = "&gt;",
	-- French entities (the most common ones)
	['à'] = "&agrave;",
	['â'] = "&acirc;",
	['é'] = "&eacute;",
	['è'] = "&egrave;",
	['ê'] = "&ecirc;",
	['ë'] = "&euml;",
	['î'] = "&icirc;",
	['ï'] = "&iuml;",
	['ô'] = "&ocirc;",
	['ù'] = "&ugrave;",
	['û'] = "&ucirc;",
	['À'] = "&Agrave;",
	['Â'] = "&Acirc;",
	['É'] = "&Eacute;",
	['È'] = "&Egrave;",
	['Ê'] = "&Ecirc;",
	['Ë'] = "&Euml;",
	['Î'] = "&Icirc;",
	['Ï'] = "&Iuml;",
	['Ô'] = "&Ocirc;",
	['Ù'] = "&Ugrave;",
	['Û'] = "&Ucirc;",
	['ç'] = "&ccedil;",
	['Ç'] = "&Ccedil;",
	['«'] = "&laquo;",
	['»'] = "&raquo;",
	['©'] = "&copy;",
	['®'] = "&reg;",
	[''] = "&OElig;", -- Not understood by all browsers
	[''] = "&oelig;", -- Not understood by all browsers
}

function EncodeEntities(toEncode)
	if toEncode == nil or type(toEncode) ~= "string" then
		return ''
	end

	local EncodeToEntities = function (char)
		local entity = entities[char]
		if entity == nil then
			local code = string.byte(char)
			if code > 127 then
				entity = string.format("&#%d;", code)
			end
		end
		return entity or char
	end

	encodedString = string.gsub(toEncode, '(.)', EncodeToEntities)
	return encodedString
end

-- Discard the input (POST request) in case it is too long, to avoid attacks.
-- Idea and code from CGILua.
function FlushInput(dataLength)
	local blockSize = 8192
	local dump
	while dataLength > 0 do
		-- Must avoid to read _STDIN with too much size, as it will lock waiting for more data...
		blockSize = math.min(dataLength, blockSize)
		dump = io.read(blockSize)
		if dump == nil then break end
		dataLength = dataLength - string.len(dump)
	end
end

function GetFormInput()
	local requestMethod = os.getenv("REQUEST_METHOD")
	local queryString = ''
	local query = {}
	local resultQuery = {}
	local contentLength

	if requestMethod == "GET" then
		queryString = os.getenv("QUERY_STRING")
	elseif requestMethod == "POST" then
		contentLength = tonumber(os.getenv("CONTENT_LENGTH"))
		if contentLength > maxInput then
			FlushInput(contentLength)
			OutputError(string.format("Your request exceeded the allowed maximum length: %s characters!", maxInput))
			return nil
		end
		queryString = io.read(contentLength)
	elseif requestMethod == "" then
		-- Command line parameters
	else
		OutputError("Use *only* GET or POST method!")
		return nil
	end
	if queryString and queryString ~= '' then
		query = Split('&', queryString)
		for i, val in query do
			-- Split into name and value: name=value
			local pair = Split('=', val)
			resultQuery[DecodeString(pair[1])] = DecodeString(pair[2])
		end
	end
	return resultQuery, queryString
end

function DumpInputData()

	local DumpVal = function (name, value)
		if value == nil or value == '' then
			value = "&nbsp;"	-- Preserve table cell
		end
		io.write("<tr><td>", name, "</td><td>", EncodeEntities(value), "</td>\n")
	end

	local inputData, query = GetFormInput()
	if query then
		io.write("<p><b>Query String:</b> ", EncodeEntities(query), "</p>\n")
	else
		io.write("<p>No query string</p>\n")
	end
	io.write[[
<table border="1">
<tr>
<td colspan="2" align="center"><font size="+2"><b>Input Data</b></font></td>
</tr>
<tr>
<td><b>Name</b></td><td><b>Value</b></td>
</tr>
]]
	local itemNb = 0
	if inputData then
		for i, val in inputData do
			DumpVal(i, val)
			itemNb = itemNb + 1
		end
	end
	if itemNb == 0 then
		io.write[[
<tr>
<td colspan='2'>No data to display</td>
</tr>
]]
	end
	io.write[[
</table>
]]
end

function DumpCGIVariables()

	local DumpVar = function (varName)
		local value = os.getenv(varName)
		if value == nil then
			value = "(not defined)"
		end
		io.write("<tr><td>", varName, "</td><td>", EncodeEntities(value), "</td>\n")
	end

	io.write[[
<table border="1">
<tr>
<td colspan="2" align="center"><font size="+2"><b>CGI Variables</b></font></td>
</tr>
]]

	io.write[[
<tr>
<td colspan="2"><font size="+1"><b>CGI environment variables:</b></font></td>
</tr><tr>
<td><b>Name</b></td><td><b>Value</b></td>
</tr>
]]
	DumpVar("AUTH_TYPE")
	DumpVar("CONTENT_LENGTH")
	DumpVar("GATEWAY_INTERFACE")
	DumpVar("PATH_INFO")
	DumpVar("PATH_TRANSLATED")
	DumpVar("QUERY_STRING")
	DumpVar("REMOTE_ADDR")
	DumpVar("REMOTE_HOST")
	DumpVar("REMOTE_IDENT")
	DumpVar("REMOTE_USER")
	DumpVar("REQUEST_METHOD")
	DumpVar("SCRIPT_NAME")
	DumpVar("SERVER_NAME")
	DumpVar("SERVER_PORT")
	DumpVar("SERVER_PROTOCOL")
	DumpVar("SERVER_SOFTWARE")

	io.write[[
<tr>
<td colspan="2"><font size="+1"><b>HTTP environment variables:</b></font></td>
</tr><tr>
<td><b>Name</b></td><td><b>Value</b></td>
</tr>
]]
	DumpVar("HTTP_ACCEPT")
	DumpVar("HTTP_ACCEPT_ENCODING")
	DumpVar("HTTP_ACCEPT_LANGUAGE")
	DumpVar("HTTP_CACHE_CONTROL")
	DumpVar("HTTP_HOST")
	DumpVar("HTTP_REFERER")
	DumpVar("HTTP_USER_AGENT")
	DumpVar("HTTP_VIA")
	DumpVar("HTTP_X_FORWARDED_FOR")

	io.write[[
</table>
]]
end

DumpInputData()
DumpCGIVariables()

-- io.write HTML tail
io.write[[
</body>
</html>
]]