mirror of
https://gitlab.nic.cz/labs/bird.git
synced 2024-12-23 02:01:55 +00:00
136 lines
3.3 KiB
C
136 lines
3.3 KiB
C
#include "nest/bird.h"
|
|
#include "conf/conf.h"
|
|
#include "filter/filter.h"
|
|
#include "lua.h"
|
|
|
|
#include <lua.h>
|
|
#include <lualib.h>
|
|
#include <lauxlib.h>
|
|
|
|
/* Docs: http://pgl.yoyo.org/luai/i/luaL_dostring */
|
|
|
|
struct lua_new_filter_writer_data {
|
|
struct lua_filter_chunk *first, *last;
|
|
};
|
|
|
|
static int lua_new_filter_writer(lua_State *L UNUSED, const void *p, size_t sz, void *ud) {
|
|
struct lua_new_filter_writer_data *d = ud;
|
|
struct lua_filter_chunk *cur = cfg_allocz(sizeof(struct lua_filter_chunk));
|
|
|
|
cur->size = sz;
|
|
cur->chunk = cfg_alloc(sz);
|
|
memcpy(cur->chunk, p, sz);
|
|
|
|
if (d->last)
|
|
d->last = d->last->next = cur;
|
|
else
|
|
d->last = d->first = cur;
|
|
|
|
return 0;
|
|
}
|
|
|
|
struct filter * lua_new_filter(struct f_inst *inst) {
|
|
struct filter *f = cfg_alloc(sizeof(struct filter));
|
|
f->name = NULL;
|
|
f->type = FILTER_LUA;
|
|
|
|
struct f_val string = f_eval(inst, cfg_mem);
|
|
if (string.type != T_STRING) {
|
|
cf_error("Lua filter must be a string");
|
|
return NULL;
|
|
}
|
|
|
|
lua_State *L = luaL_newstate();
|
|
luaL_openlibs(L);
|
|
int loadres = luaL_loadstring(L, string.val.s);
|
|
switch (loadres) {
|
|
case LUA_ERRMEM:
|
|
lua_close(L);
|
|
cf_error("Memory allocation error occured when loading lua chunk");
|
|
return NULL;
|
|
case LUA_ERRSYNTAX:
|
|
{
|
|
const char *e = lua_tostring(L, -1);
|
|
char *ec = cfg_alloc(strlen(e) + 1);
|
|
strcpy(ec, e);
|
|
lua_close(L);
|
|
cf_error("Lua syntax error: %s", ec);
|
|
return NULL;
|
|
}
|
|
case 0: /* Everything OK */
|
|
break;
|
|
}
|
|
|
|
struct lua_new_filter_writer_data lnfwd = {};
|
|
lua_dump(L, lua_new_filter_writer, &lnfwd, 0); /* No error to handle */
|
|
lua_close(L);
|
|
|
|
f->lua_chunk = lnfwd.first;
|
|
return f;
|
|
}
|
|
|
|
static const char *lua_interpret_reader(lua_State *L UNUSED, void *ud, size_t *sz) {
|
|
struct lua_filter_chunk **cptr = ud;
|
|
if ((*cptr) == NULL)
|
|
return NULL;
|
|
|
|
*sz = (*cptr)->size;
|
|
void *out = (*cptr)->chunk;
|
|
*cptr = (*cptr)->next;
|
|
return out;
|
|
}
|
|
|
|
struct f_val lua_interpret(struct lua_filter_chunk *chunk, struct rte **e, struct rta **a, struct ea_list **ea, struct linpool *lp, int flags) {
|
|
lua_State *L = luaL_newstate();
|
|
luaL_openlibs(L);
|
|
lua_bird_state *lbs = luaB_init(L, lp);
|
|
luaB_push_route(L, *e);
|
|
struct lua_filter_chunk **rptr = &chunk;
|
|
lua_load(L, lua_interpret_reader, rptr, "", "b");
|
|
int le = lua_pcall(L, 0, LUA_MULTRET, 0);
|
|
struct f_val out = F_VAL_VOID;
|
|
if (le && lbs->exception) {
|
|
out = F_VAL(T_RETURN, i, lbs->exception);
|
|
} else if (le) {
|
|
log(L_ERR "bad lua: %s", lua_tostring(L, -1));
|
|
out = F_VAL(T_RETURN, i, F_ERROR);
|
|
} else if (lua_isnumber(L, -1)) {
|
|
out = F_VAL(T_INT, i, lua_tonumber(L, -1));
|
|
} else {
|
|
log(L_WARN "lua return value is not a number (unimplemented): %s", lua_tostring(L, -1));
|
|
out = F_VAL(T_RETURN, i, F_ERROR);
|
|
}
|
|
|
|
lua_close(L);
|
|
return out;
|
|
}
|
|
|
|
int lua_filter_same(struct lua_filter_chunk *new, struct lua_filter_chunk *old) {
|
|
size_t npos = 0, opos = 0;
|
|
while (new && old) {
|
|
size_t nrem = new->size - npos;
|
|
size_t orem = old->size - opos;
|
|
size_t rem = MIN(nrem, orem);
|
|
if (memcmp(new->chunk + npos, old->chunk + opos, rem))
|
|
return 0;
|
|
|
|
npos += rem;
|
|
opos += rem;
|
|
|
|
if (npos == new->size) {
|
|
new = new->next;
|
|
npos = 0;
|
|
}
|
|
|
|
if (opos == old->size) {
|
|
old = old->next;
|
|
opos = 0;
|
|
}
|
|
}
|
|
|
|
if (!new && !old)
|
|
return 1;
|
|
else
|
|
return 0;
|
|
}
|