2 * uhttpd - Tiny single-threaded httpd
4 * Copyright (C) 2010-2012 Jo-Philipp Wich <xm@subsignal.org>
5 * Copyright (C) 2012 Felix Fietkau <nbd@openwrt.org>
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
11 * http://www.apache.org/licenses/LICENSE-2.0
13 * Unless required by applicable law or agreed to in writing, software
14 * distributed under the License is distributed on an "AS IS" BASIS,
15 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 * See the License for the specific language governing permissions and
17 * limitations under the License.
21 #include <libubox/blobmsg.h>
31 #define UH_LUA_CB "handle_request"
33 static const struct uhttpd_ops
*ops
;
34 static struct config
*_conf
;
39 static int uh_lua_recv(lua_State
*L
)
41 static struct pollfd pfd
= {
50 len
= luaL_checknumber(L
, 1);
55 buf
= luaL_prepbuffer(&B
);
56 r
= read(STDIN_FILENO
, buf
, LUAL_BUFFERSIZE
);
58 if (errno
== EWOULDBLOCK
|| errno
== EAGAIN
) {
61 if (pfd
.revents
& POLLIN
)
76 if (r
!= LUAL_BUFFERSIZE
)
81 lua_pushnumber(L
, data_len
);
92 static int uh_lua_send(lua_State
*L
)
97 buf
= luaL_checklstring(L
, 1, &len
);
99 len
= write(STDOUT_FILENO
, buf
, len
);
101 lua_pushnumber(L
, len
);
106 uh_lua_strconvert(lua_State
*L
, int (*convert
)(char *, int, const char *, int))
109 static char out_buf
[4096];
113 in_buf
= luaL_checklstring(L
, 1, &in_len
);
114 out_len
= convert(out_buf
, sizeof(out_buf
), in_buf
, in_len
);
120 error
= "buffer overflow";
122 error
= "malformed string";
124 luaL_error(L
, "%s on URL conversion\n", error
);
127 lua_pushlstring(L
, out_buf
, out_len
);
131 static int uh_lua_urldecode(lua_State
*L
)
133 return uh_lua_strconvert(L
, ops
->urldecode
);
136 static int uh_lua_urlencode(lua_State
*L
)
138 return uh_lua_strconvert(L
, ops
->urlencode
);
141 static lua_State
*uh_lua_state_init(void)
143 const char *msg
= "(unknown error)";
151 /* build uhttpd api table */
154 lua_pushcfunction(L
, uh_lua_send
);
155 lua_setfield(L
, -2, "send");
157 lua_pushcfunction(L
, uh_lua_send
);
158 lua_setfield(L
, -2, "sendc");
160 lua_pushcfunction(L
, uh_lua_recv
);
161 lua_setfield(L
, -2, "recv");
163 lua_pushcfunction(L
, uh_lua_urldecode
);
164 lua_setfield(L
, -2, "urldecode");
166 lua_pushcfunction(L
, uh_lua_urlencode
);
167 lua_setfield(L
, -2, "urlencode");
169 lua_pushstring(L
, conf
.docroot
);
170 lua_setfield(L
, -2, "docroot");
172 lua_setglobal(L
, "uhttpd");
174 ret
= luaL_loadfile(L
, conf
.lua_handler
);
180 ret
= lua_pcall(L
, 0, 0, 0);
182 status
= "initializing";
186 lua_getglobal(L
, UH_LUA_CB
);
187 if (!lua_isfunction(L
, -1)) {
188 fprintf(stderr
, "Error: Lua handler provides no " UH_LUA_CB
"() callback.\n");
195 if (!lua_isnil(L
, -1))
196 msg
= lua_tostring(L
, -1);
198 fprintf(stderr
, "Error %s Lua handler: %s\n", status
, msg
);
203 static void lua_main(struct client
*cl
, struct path_info
*pi
, char *url
)
205 struct blob_attr
*cur
;
209 int path_len
, prefix_len
;
213 lua_getglobal(L
, UH_LUA_CB
);
215 /* new env table for this request */
218 prefix_len
= strlen(conf
.lua_prefix
);
219 path_len
= strlen(url
);
220 str
= strchr(url
, '?');
223 path_len
= str
- url
;
225 if (path_len
> prefix_len
) {
226 lua_pushlstring(L
, url
+ prefix_len
,
227 path_len
- prefix_len
);
228 lua_setfield(L
, -2, "PATH_INFO");
231 for (var
= ops
->get_process_vars(cl
, pi
); var
->name
; var
++) {
235 lua_pushstring(L
, var
->value
);
236 lua_setfield(L
, -2, var
->name
);
239 lua_pushnumber(L
, 0.9 + (cl
->request
.version
/ 10.0));
240 lua_setfield(L
, -2, "HTTP_VERSION");
243 blob_for_each_attr(cur
, cl
->hdr
.head
, rem
) {
244 lua_pushstring(L
, blobmsg_data(cur
));
245 lua_setfield(L
, -2, blobmsg_name(cur
));
247 lua_setfield(L
, -2, "headers");
249 switch(lua_pcall(L
, 1, 0, 0)) {
252 error
= luaL_checkstring(L
, -1);
254 error
= "(unknown error)";
256 printf("Status: 500 Internal Server Error\r\n\r\n"
257 "Unable to launch the requested Lua program:\n"
258 " %s: %s\n", pi
->phys
, strerror(errno
));
264 static void lua_handle_request(struct client
*cl
, char *url
, struct path_info
*pi
)
266 static struct path_info _pi
;
269 pi
->name
= conf
.lua_prefix
;
270 pi
->phys
= conf
.lua_handler
;
272 if (!ops
->create_process(cl
, pi
, url
, lua_main
)) {
273 ops
->client_error(cl
, 500, "Internal Server Error",
274 "Failed to create CGI process: %s", strerror(errno
));
278 static bool check_lua_url(const char *url
)
280 return ops
->path_match(conf
.lua_prefix
, url
);
283 static struct dispatch_handler lua_dispatch
= {
284 .check_url
= check_lua_url
,
285 .handle_request
= lua_handle_request
,
288 static int lua_plugin_init(const struct uhttpd_ops
*o
, struct config
*c
)
292 _L
= uh_lua_state_init();
293 ops
->dispatch_add(&lua_dispatch
);
297 const struct uhttpd_plugin uhttpd_plugin
= {
298 .init
= lua_plugin_init
,