diff options
| author | Vsevolod Stakhov <vsevolod@highsecure.ru> | 2021-02-19 16:51:47 +0000 |
|---|---|---|
| committer | Vsevolod Stakhov <vsevolod@highsecure.ru> | 2021-02-19 16:51:47 +0000 |
| commit | e2a6741de65902d65b84f8f62df5114674ae336a (patch) | |
| tree | 26e9102008ea78379cbf2412e8684bb1642b0f9f /src/lua/lua_url.c | |
| parent | dbc9ff655dfb459eb8af328a82a5b8c848cda480 (diff) | |
| download | rspamd-e2a6741de65902d65b84f8f62df5114674ae336a.tar.gz rspamd-e2a6741de65902d65b84f8f62df5114674ae336a.zip | |
[Minor] Lua_url: Allow to get numeric flags from an url
Diffstat (limited to 'src/lua/lua_url.c')
| -rw-r--r-- | src/lua/lua_url.c | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/src/lua/lua_url.c b/src/lua/lua_url.c index 29f75995b..945f3dbf4 100644 --- a/src/lua/lua_url.c +++ b/src/lua/lua_url.c @@ -55,6 +55,7 @@ LUA_FUNCTION_DEF (url, tostring); LUA_FUNCTION_DEF (url, get_raw); LUA_FUNCTION_DEF (url, get_tld); LUA_FUNCTION_DEF (url, get_flags); +LUA_FUNCTION_DEF (url, get_flags_num); LUA_FUNCTION_DEF (url, get_protocol); LUA_FUNCTION_DEF (url, to_table); LUA_FUNCTION_DEF (url, is_phished); @@ -93,6 +94,7 @@ static const struct luaL_reg urllib_m[] = { LUA_INTERFACE_DEF (url, get_visible), LUA_INTERFACE_DEF (url, get_count), LUA_INTERFACE_DEF (url, get_flags), + LUA_INTERFACE_DEF (url, get_flags_num), {"get_redirected", lua_url_get_phished}, LUA_INTERFACE_DEF (url, set_redirected), {"__tostring", lua_url_tostring}, @@ -926,6 +928,22 @@ lua_url_get_flags (lua_State *L) #undef PUSH_FLAG +static gint +lua_url_get_flags_num (lua_State *L) +{ + LUA_TRACE_POINT; + struct rspamd_lua_url *url = lua_check_url (L, 1); + + if (url) { + lua_pushinteger (L, url->url->flags); + } + else { + return luaL_error (L, "invalid arguments"); + } + + return 1; +} + void lua_tree_url_callback (gpointer key, gpointer value, gpointer ud) { |