diff options
| author | Vsevolod Stakhov <vsevolod@rspamd.com> | 2022-07-02 13:32:56 +0100 |
|---|---|---|
| committer | Vsevolod Stakhov <vsevolod@rspamd.com> | 2022-07-02 13:32:56 +0100 |
| commit | dccf92aa82ccca974a280c22708047ea2594e15a (patch) | |
| tree | 2b09463eee0bf3c2c019bad664e48cac582c2130 /src/lua/lua_upstream.c | |
| parent | d87f4e9f5759b4cb85072b422d353cea1437df54 (diff) | |
| download | rspamd-dccf92aa82ccca974a280c22708047ea2594e15a.tar.gz rspamd-dccf92aa82ccca974a280c22708047ea2594e15a.zip | |
[Feature] Allow lua_http module to accept upstreams
Diffstat (limited to 'src/lua/lua_upstream.c')
| -rw-r--r-- | src/lua/lua_upstream.c | 23 |
1 files changed, 9 insertions, 14 deletions
diff --git a/src/lua/lua_upstream.c b/src/lua/lua_upstream.c index 5019f28d3..94e251a95 100644 --- a/src/lua/lua_upstream.c +++ b/src/lua/lua_upstream.c @@ -95,15 +95,10 @@ static const struct luaL_reg upstream_m[] = { /* Upstream class */ -struct rspamd_lua_upstream { - struct upstream *up; - gint upref; -}; - -static struct rspamd_lua_upstream * -lua_check_upstream (lua_State * L) +struct rspamd_lua_upstream * +lua_check_upstream(lua_State *L, int pos) { - void *ud = rspamd_lua_check_udata (L, 1, "rspamd{upstream}"); + void *ud = rspamd_lua_check_udata (L, pos, "rspamd{upstream}"); luaL_argcheck (L, ud != NULL, 1, "'upstream' expected"); return ud ? (struct rspamd_lua_upstream *)ud : NULL; @@ -118,7 +113,7 @@ static gint lua_upstream_get_addr (lua_State *L) { LUA_TRACE_POINT; - struct rspamd_lua_upstream *up = lua_check_upstream (L); + struct rspamd_lua_upstream *up = lua_check_upstream(L, 1); if (up) { rspamd_lua_ip_push (L, rspamd_upstream_addr_next (up->up)); @@ -139,7 +134,7 @@ static gint lua_upstream_get_name (lua_State *L) { LUA_TRACE_POINT; - struct rspamd_lua_upstream *up = lua_check_upstream (L); + struct rspamd_lua_upstream *up = lua_check_upstream(L, 1); if (up) { lua_pushstring (L, rspamd_upstream_name (up->up)); @@ -160,7 +155,7 @@ static gint lua_upstream_get_port (lua_State *L) { LUA_TRACE_POINT; - struct rspamd_lua_upstream *up = lua_check_upstream (L); + struct rspamd_lua_upstream *up = lua_check_upstream(L, 1); if (up) { lua_pushinteger (L, rspamd_upstream_port (up->up)); @@ -180,7 +175,7 @@ static gint lua_upstream_fail (lua_State *L) { LUA_TRACE_POINT; - struct rspamd_lua_upstream *up = lua_check_upstream (L); + struct rspamd_lua_upstream *up = lua_check_upstream(L, 1); gboolean fail_addr = FALSE; const gchar *reason = "unknown"; @@ -211,7 +206,7 @@ static gint lua_upstream_ok (lua_State *L) { LUA_TRACE_POINT; - struct rspamd_lua_upstream *up = lua_check_upstream (L); + struct rspamd_lua_upstream *up = lua_check_upstream(L, 1); if (up) { rspamd_upstream_ok (up->up); @@ -224,7 +219,7 @@ static gint lua_upstream_destroy (lua_State *L) { LUA_TRACE_POINT; - struct rspamd_lua_upstream *up = lua_check_upstream (L); + struct rspamd_lua_upstream *up = lua_check_upstream(L, 1); if (up) { /* Remove reference to the parent */ |