aboutsummaryrefslogtreecommitdiffstats
path: root/util/httpstream.lua
blob: 9c3c9ae6105803c76d230b5da92e210211436c3a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84

local coroutine = coroutine;
local tonumber = tonumber;

local deadroutine = coroutine.create(function() end);
coroutine.resume(deadroutine);

module("httpstream")

local function parser(data, success_cb)
	local function readline()
		if not data then coroutine.yield("Unexpected EOF"); end
		local pos, line = (data:find("\r\n", nil, true));
		if not pos then
			local newdata = coroutine.yield();
			if not newdata then data = nil; coroutine.yield("Unexpected EOF"); end
			data = data..newdata;
			return readline();
		end
		line, data = data:sub(1, pos-1), data:sub(pos+2);
		return line;
	end
	local function readlength(n)
		if not data then coroutine.yield("Unexpected EOF"); end
		while #data < n do
			local newdata = coroutine.yield();
			if not newdata then data = nil; coroutine.yield("Unexpected EOF"); end
			data = data..newdata;
		end
		local r = data:sub(1, n);
		data = data:sub(n + 1);
		return r;
	end
	
	while true do
		-- read status line
		local status_line = readline();
		local method, path, httpversion = status_line:match("^(%S+)%s+(%S+)%s+HTTP/(%S+)$");
		if not method then coroutine.yield("invalid-status-line"); end
		-- TODO parse url
		
		local headers = {}; -- read headers
		while true do
			local line = readline();
			if line == "" then break; end -- headers done
			local key, val = line:match("^([^%s:]+): *(.*)$");
			if not key then coroutine.yield("invalid-header-line"); end -- TODO handle multi-line and invalid headers
			key = key:lower();
			headers[key] = headers[key] and headers[key]..","..val or val;
		end
		
		-- read body
		local len = tonumber(headers["content-length"]);
		len = len or 0; -- TODO check for invalid len
		local body = readlength(len);
		
		success_cb({
			method = method;
			path = path;
			httpversion = httpversion;
			headers = headers;
			body = body;
		});
	end
end

function new(success_cb, error_cb)
	local co = coroutine.create(parser);
	return {
		feed = function(self, data)
			local success, result = coroutine.resume(co, data, success_cb);
			if result then
				if result.method then
					success_cb(result);
				else -- error
					error_cb(result);
					co = deadroutine;
				end
			end
		end;
	};
end

return _M;