summaryrefslogtreecommitdiffstatshomepage
path: root/plugins/http-codec/init.lua
diff options
context:
space:
mode:
Diffstat (limited to 'plugins/http-codec/init.lua')
-rw-r--r--plugins/http-codec/init.lua291
1 files changed, 291 insertions, 0 deletions
diff --git a/plugins/http-codec/init.lua b/plugins/http-codec/init.lua
new file mode 100644
index 00000000000..11a32945729
--- /dev/null
+++ b/plugins/http-codec/init.lua
@@ -0,0 +1,291 @@
+--[[
+
+Copyright 2014-2015 The Luvit Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+ http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS-IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+
+--]]
+
+local exports = {}
+
+exports.name = "luvit/http-codec"
+exports.version = "1.0.0-1"
+exports.homepage = "https://github.com/luvit/luvit/blob/master/deps/http-codec.lua"
+exports.description = "A simple pair of functions for converting between hex and raw strings."
+exports.tags = {"codec", "http"}
+exports.license = "Apache 2"
+exports.author = { name = "Tim Caswell" }
+
+local sub = string.sub
+local gsub = string.gsub
+local lower = string.lower
+local find = string.find
+local format = string.format
+local concat = table.concat
+local match = string.match
+
+local STATUS_CODES = {
+ [100] = 'Continue',
+ [101] = 'Switching Protocols',
+ [102] = 'Processing', -- RFC 2518, obsoleted by RFC 4918
+ [200] = 'OK',
+ [201] = 'Created',
+ [202] = 'Accepted',
+ [203] = 'Non-Authoritative Information',
+ [204] = 'No Content',
+ [205] = 'Reset Content',
+ [206] = 'Partial Content',
+ [207] = 'Multi-Status', -- RFC 4918
+ [300] = 'Multiple Choices',
+ [301] = 'Moved Permanently',
+ [302] = 'Moved Temporarily',
+ [303] = 'See Other',
+ [304] = 'Not Modified',
+ [305] = 'Use Proxy',
+ [307] = 'Temporary Redirect',
+ [400] = 'Bad Request',
+ [401] = 'Unauthorized',
+ [402] = 'Payment Required',
+ [403] = 'Forbidden',
+ [404] = 'Not Found',
+ [405] = 'Method Not Allowed',
+ [406] = 'Not Acceptable',
+ [407] = 'Proxy Authentication Required',
+ [408] = 'Request Time-out',
+ [409] = 'Conflict',
+ [410] = 'Gone',
+ [411] = 'Length Required',
+ [412] = 'Precondition Failed',
+ [413] = 'Request Entity Too Large',
+ [414] = 'Request-URI Too Large',
+ [415] = 'Unsupported Media Type',
+ [416] = 'Requested Range Not Satisfiable',
+ [417] = 'Expectation Failed',
+ [418] = "I'm a teapot", -- RFC 2324
+ [422] = 'Unprocessable Entity', -- RFC 4918
+ [423] = 'Locked', -- RFC 4918
+ [424] = 'Failed Dependency', -- RFC 4918
+ [425] = 'Unordered Collection', -- RFC 4918
+ [426] = 'Upgrade Required', -- RFC 2817
+ [500] = 'Internal Server Error',
+ [501] = 'Not Implemented',
+ [502] = 'Bad Gateway',
+ [503] = 'Service Unavailable',
+ [504] = 'Gateway Time-out',
+ [505] = 'HTTP Version not supported',
+ [506] = 'Variant Also Negotiates', -- RFC 2295
+ [507] = 'Insufficient Storage', -- RFC 4918
+ [509] = 'Bandwidth Limit Exceeded',
+ [510] = 'Not Extended' -- RFC 2774
+}
+
+exports.encoder = function ()
+
+ local mode
+ local encodeHead, encodeRaw, encodeChunked
+
+ function encodeHead(item)
+ if not item or item == "" then
+ return item
+ elseif not (type(item) == "table") then
+ error("expected a table but got a " .. type(item) .. " when encoding data")
+ end
+ local head, chunkedEncoding
+ local version = item.version or 1.1
+ if item.method then
+ local path = item.path
+ assert(path and #path > 0, "expected non-empty path")
+ head = { item.method .. ' ' .. item.path .. ' HTTP/' .. version .. '\r\n' }
+ else
+ local reason = item.reason or STATUS_CODES[item.code]
+ head = { 'HTTP/' .. version .. ' ' .. item.code .. ' ' .. reason .. '\r\n' }
+ end
+ for i = 1, #item do
+ local key, value = unpack(item[i])
+ local lowerKey = lower(key)
+ if lowerKey == "transfer-encoding" then
+ chunkedEncoding = lower(value) == "chunked"
+ end
+ value = gsub(tostring(value), "[\r\n]+", " ")
+ head[#head + 1] = key .. ': ' .. tostring(value) .. '\r\n'
+ end
+ head[#head + 1] = '\r\n'
+
+ mode = chunkedEncoding and encodeChunked or encodeRaw
+ return concat(head)
+ end
+
+ function encodeRaw(item)
+ if type(item) ~= "string" then
+ mode = encodeHead
+ return encodeHead(item)
+ end
+ return item
+ end
+
+ function encodeChunked(item)
+ if type(item) ~= "string" then
+ mode = encodeHead
+ local extra = encodeHead(item)
+ if extra then
+ return "0\r\n\r\n" .. extra
+ else
+ return "0\r\n\r\n"
+ end
+ end
+ if #item == 0 then
+ mode = encodeHead
+ end
+ return format("%x", #item) .. "\r\n" .. item .. "\r\n"
+ end
+
+ mode = encodeHead
+ return function (item)
+ return mode(item)
+ end
+end
+
+exports.decoder = function ()
+
+ -- This decoder is somewhat stateful with 5 different parsing states.
+ local decodeHead, decodeEmpty, decodeRaw, decodeChunked, decodeCounted
+ local mode -- state variable that points to various decoders
+ local bytesLeft -- For counted decoder
+
+ -- This state is for decoding the status line and headers.
+ function decodeHead(chunk)
+ if not chunk then return end
+
+ local _, length = find(chunk, "\r?\n\r?\n", 1)
+ -- First make sure we have all the head before continuing
+ if not length then
+ if #chunk < 8 * 1024 then return end
+ -- But protect against evil clients by refusing heads over 8K long.
+ error("entity too large")
+ end
+
+ -- Parse the status/request line
+ local head = {}
+ local _, offset
+ local version
+ _, offset, version, head.code, head.reason =
+ find(chunk, "^HTTP/(%d%.%d) (%d+) ([^\r\n]+)\r?\n")
+ if offset then
+ head.code = tonumber(head.code)
+ else
+ _, offset, head.method, head.path, version =
+ find(chunk, "^(%u+) ([^ ]+) HTTP/(%d%.%d)\r?\n")
+ if not offset then
+ error("expected HTTP data")
+ end
+ end
+ version = tonumber(version)
+ head.version = version
+ head.keepAlive = version > 1.0
+
+ -- We need to inspect some headers to know how to parse the body.
+ local contentLength
+ local chunkedEncoding
+
+ -- Parse the header lines
+ while true do
+ local key, value
+ _, offset, key, value = find(chunk, "^([^:\r\n]+): *([^\r\n]+)\r?\n", offset + 1)
+ if not offset then break end
+ local lowerKey = lower(key)
+
+ -- Inspect a few headers and remember the values
+ if lowerKey == "content-length" then
+ contentLength = tonumber(value)
+ elseif lowerKey == "transfer-encoding" then
+ chunkedEncoding = lower(value) == "chunked"
+ elseif lowerKey == "connection" then
+ head.keepAlive = lower(value) == "keep-alive"
+ end
+ head[#head + 1] = {key, value}
+ end
+
+ if head.keepAlive and (not (chunkedEncoding or (contentLength and contentLength > 0)))
+ or (head.method == "GET" or head.method == "HEAD") then
+ mode = decodeEmpty
+ elseif chunkedEncoding then
+ mode = decodeChunked
+ elseif contentLength then
+ bytesLeft = contentLength
+ mode = decodeCounted
+ elseif not head.keepAlive then
+ mode = decodeRaw
+ end
+
+ return head, sub(chunk, length + 1)
+
+ end
+
+ -- This is used for inserting a single empty string into the output string for known empty bodies
+ function decodeEmpty(chunk)
+ mode = decodeHead
+ return "", chunk or ""
+ end
+
+ function decodeRaw(chunk)
+ if not chunk then return "", "" end
+ if #chunk == 0 then return end
+ return chunk, ""
+ end
+
+ function decodeChunked(chunk)
+ local len, term
+ len, term = match(chunk, "^(%x+)(..)")
+ if not len then return end
+ assert(term == "\r\n")
+ local length = tonumber(len, 16)
+ if #chunk < length + 4 + #len then return end
+ if length == 0 then
+ mode = decodeHead
+ end
+ chunk = sub(chunk, #len + 3)
+ assert(sub(chunk, length + 1, length + 2) == "\r\n")
+ return sub(chunk, 1, length), sub(chunk, length + 3)
+ end
+
+ function decodeCounted(chunk)
+ if bytesLeft == 0 then
+ mode = decodeEmpty
+ return mode(chunk)
+ end
+ local length = #chunk
+ -- Make sure we have at least one byte to process
+ if length == 0 then return end
+
+ if length >= bytesLeft then
+ mode = decodeEmpty
+ end
+
+ -- If the entire chunk fits, pass it all through
+ if length <= bytesLeft then
+ bytesLeft = bytesLeft - length
+ return chunk, ""
+ end
+
+ return sub(chunk, 1, bytesLeft), sub(chunk, bytesLeft + 1)
+ end
+
+ -- Switch between states by changing which decoder mode points to
+ mode = decodeHead
+ return function (chunk)
+ return mode(chunk)
+ end
+
+end
+
+return exports