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
85
86
87
88
89
90
91
92
93
94
95
96
|
#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <lua.h>
#include <lauxlib.h>
#include <lualib.h>
#include "conn.h"
#include "lua-privsep.h"
#ifndef PRIVSEP_PATH
#define PRIVSEP_PATH "./"
#endif
static int traceback (lua_State *L) {
if (!lua_isstring(L, 1)) /* 'message' not a string? */
return 1; /* keep it intact */
fprintf(stderr, "traceback\n");
lua_getfield(L, LUA_GLOBALSINDEX, "debug");
if (!lua_istable(L, -1)) {
fprintf(stderr, "traceback: debug\n");
lua_pop(L, 1);
return 1;
}
lua_getfield(L, -1, "traceback");
if (!lua_isfunction(L, -1)) {
fprintf(stderr, "traceback: traceback\n");
lua_pop(L, 2);
return 1;
}
lua_pushvalue(L, 1); /* pass error message */
lua_pushinteger(L, 2); /* skip this function and traceback */
lua_call(L, 2, 1); /* call debug.traceback */
return 1;
}
static void lerror(lua_State *L, const char *fmt, ...)
{
va_list argp;
va_start(argp, fmt);
vfprintf(stderr, fmt, argp);
va_end(argp);
lua_close(L);
exit(EXIT_FAILURE);
}
int call_lua(struct conn *c, const char *msg, size_t msglen)
{
const char *retbuf;
size_t retsize;
lua_State *L = c->L;
lua_getglobal(L, "handler");
lua_pushlstring(L, msg, msglen);
if (lua_pcall(L, 1, 1, 0))
return luaL_error(L, lua_tostring(L, -1));
if (!lua_isstring(L, -1))
lerror(L, "handler function must return string\n");
retbuf = lua_tolstring(L, -1, &retsize);
return write(c->io.fd, retbuf, retsize);
}
int init_lua(struct conn *c)
{
const char *luamain = PRIVSEP_PATH "handler.lua";
int traceback_index;
const char *retbuf;
size_t retsize;
c->L = luaL_newstate();
if (c == NULL)
return -1;
luaL_openlibs(c->L);
if (luaL_loadfile(c->L, luamain) || lua_pcall(c->L, 0, 0, 0)) {
lerror(c->L, "%s: %s", luamain, lua_tostring(c->L, -1));
lua_close(c->L);
c->L = NULL;
return -1;
}
return 0;
}
void close_lua(struct conn *c)
{
lua_close(c->L);
c->L = NULL;
}
|