e5b862032cdaae66ef8a75a05bcfbdc395a623be
[project/luci.git] / libs / httpd / luasrc / httpd / server.lua
1 --[[
2
3 HTTP server implementation for LuCI - helper class
4 (c) 2008 Freifunk Leipzig / Jo-Philipp Wich <xm@leipzig.freifunk.net>
5 (c) 2008 Steven Barth <steven@midlink.org>
6
7 Licensed under the Apache License, Version 2.0 (the "License");
8 you may not use this file except in compliance with the License.
9 You may obtain a copy of the License at
10
11 http://www.apache.org/licenses/LICENSE-2.0
12
13 $Id$
14
15 ]]--
16
17 module("luci.httpd.server", package.seeall)
18 require("socket")
19 require("socket.http")
20 require("luci.util")
21
22 READ_BUFSIZE = 1024
23
24
25 VHost = luci.util.class()
26
27 function VHost.__init__(self, handler)
28 self.handler = handler
29 self.dhandler = {}
30 end
31
32 function VHost.process(self, request, sourcein, sinkerr, ...)
33 local handler = self.handler
34
35 local uri = request.env.REQUEST_URI:match("^([^?]*)")
36
37 -- SCRIPT_NAME
38 request.env.SCRIPT_NAME = ""
39
40 -- Call URI part
41 request.env.PATH_INFO = uri
42
43 for k, dhandler in pairs(self.dhandler) do
44 if k == uri or k.."/" == uri:sub(1, #k+1) then
45 handler = dhandler
46 request.env.SCRIPT_NAME = k
47 request.env.PATH_INFO = uri:sub(#k+1)
48 break;
49 end
50 end
51
52 if handler then
53 return handler:process(request, sourcein, sinkerr, ...)
54 end
55 end
56
57
58 function VHost.set_default_handler(self, handler)
59 self.handler = handler
60 end
61
62
63 function VHost.set_handler(self, match, handler)
64 self.dhandler[match] = handler
65 end
66
67
68
69 Server = luci.util.class()
70
71 function Server.__init__(self, host)
72 self.host = host
73 self.vhosts = {}
74 end
75
76 function Server.set_default_vhost(self, vhost)
77 self.host = vhost
78 end
79
80 -- Sets a vhost
81 function Server.set_vhost(self, name, vhost)
82 self.vhosts[name] = vhost
83 end
84
85 function Server.create_daemon_handlers(self)
86 return function(...) return self:process(...) end,
87 function(...) return self:error_overload(...) end
88 end
89
90
91 function Server.error(self, socket, code, msg)
92 hcode = tostring(code)
93
94 socket:send( "HTTP/1.0 " .. hcode .. " " ..
95 luci.http.protocol.statusmsg[code] .. "\r\n" )
96 socket:send( "Connection: close\r\n" )
97 socket:send( "Content-Type: text/plain\r\n\r\n" )
98
99 if msg then
100 socket:send( "HTTP-Error " .. code .. ": " .. msg .. "\r\n" )
101 end
102 end
103
104 function Server.error_overload(self, socket)
105 self:error(socket, 503, "Too many simultaneous connections")
106 end
107
108
109 function Server.process( self, client )
110
111 -- Setup sockets and sources
112 local thread = {
113 receive = function(self, ...) return luci.httpd.corecv(client, ...) end
114 }
115
116 client:settimeout( 0 )
117
118 local sourcein = ltn12.source.empty()
119 local sourcehdr = luci.http.protocol.header_source( thread )
120 local sinkerr = ltn12.sink.file( io.stderr )
121
122 local close = false
123
124 local reading = { client }
125
126 local message, err
127
128 repeat
129 -- parse headers
130 message, err = luci.http.protocol.parse_message_header( sourcehdr )
131
132 if not message then
133 self:error( client, 400, err )
134 break
135 end
136
137 -- keep-alive
138 if message.http_version == 1.1 then
139 close = (message.env.HTTP_CONNECTION == "close")
140 else
141 close = not message.env.HTTP_CONNECTION or message.env.HTTP_CONNECTION == "close"
142 end
143
144 if message.request_method == "get" or message.request_method == "head" then
145 -- Be happy
146
147 elseif message.request_method == "post" then
148 -- If we have a HTTP/1.1 client and an Expect: 100-continue header then
149 -- respond with HTTP 100 Continue message
150 if message.http_version == 1.1 and message.headers['Expect'] and
151 message.headers['Expect'] == '100-continue'
152 then
153 client:send("HTTP/1.1 100 Continue\r\n\r\n")
154 end
155
156 if message.headers['Transfer-Encoding'] and
157 message.headers['Transfer-Encoding'] ~= "identity" then
158 sourcein = socket.source("http-chunked", thread)
159 elseif message.env.CONTENT_LENGTH then
160 sourcein = socket.source("by-length", thread,
161 tonumber(message.env.CONTENT_LENGTH))
162 else
163 self:error( client, 411, luci.http.protocol.statusmsg[411] )
164 break;
165 end
166
167 else
168 self:error( client, 405, luci.http.protocol.statusmsg[405] )
169 break;
170
171 end
172
173
174 local host = self.vhosts[message.env.HTTP_HOST] or self.host
175 if not host then
176 self:error( client, 500, "Unable to find matching host" )
177 break;
178 end
179
180 local response, sourceout = host:process(
181 message, sourcein, sinkerr,
182 client, io.stderr
183 )
184 if not response then
185 self:error( client, 500, "Error processing handler" )
186 end
187
188 -- Post process response
189 local sinkmode = close and "close-when-done" or "keep-open"
190
191 if sourceout then
192 if not response.headers["Content-Length"] then
193 if message.http_version == 1.1 then
194 response.headers["Transfer-Encoding"] = "chunked"
195 sinkmode = "http-chunked"
196 else
197 close = true
198 sinkmode = "close-when-done"
199 end
200 end
201 end
202
203 if close then
204 response.headers["Connection"] = "close"
205 end
206
207
208 local sinkout = socket.sink(sinkmode, client)
209
210 local header =
211 message.env.SERVER_PROTOCOL .. " " ..
212 tostring(response.status) .. " " ..
213 luci.http.protocol.statusmsg[response.status] .. "\r\n"
214
215
216 for k,v in pairs(response.headers) do
217 header = header .. k .. ": " .. v .. "\r\n"
218 end
219
220 client:send(header .. "\r\n")
221
222 if sourceout then
223 local eof = false
224 repeat
225 coroutine.yield()
226 eof = not ltn12.pump.step(sourceout, sinkout)
227 until eof
228 end
229 until close
230
231 client:close()
232 end