diff options
Diffstat (limited to 'src/lua/lua_dns_resolver.c')
| -rw-r--r-- | src/lua/lua_dns_resolver.c | 46 |
1 files changed, 46 insertions, 0 deletions
diff --git a/src/lua/lua_dns_resolver.c b/src/lua/lua_dns_resolver.c index 75f7a3b6b..7eaa54da6 100644 --- a/src/lua/lua_dns_resolver.c +++ b/src/lua/lua_dns_resolver.c @@ -51,6 +51,7 @@ LUA_FUNCTION_DEF (dns_resolver, resolve_txt); LUA_FUNCTION_DEF (dns_resolver, resolve_mx); LUA_FUNCTION_DEF (dns_resolver, resolve_ns); LUA_FUNCTION_DEF (dns_resolver, resolve); +LUA_FUNCTION_DEF (dns_resolver, idna_convert_utf8); void lua_push_dns_reply (lua_State *L, const struct rdns_reply *reply); @@ -66,6 +67,7 @@ static const struct luaL_reg dns_resolverlib_m[] = { LUA_INTERFACE_DEF (dns_resolver, resolve_mx), LUA_INTERFACE_DEF (dns_resolver, resolve_ns), LUA_INTERFACE_DEF (dns_resolver, resolve), + LUA_INTERFACE_DEF (dns_resolver, idna_convert_utf8), {"__tostring", rspamd_lua_class_tostring}, {NULL, NULL} }; @@ -688,6 +690,50 @@ lua_dns_resolver_resolve (lua_State *L) return 1; } +/*** + * @method resolver:idna_convert_utf8(hostname[, pool]) + * Converts domain name from IDN (in utf8 format) to punycode + * @return {string} new name converted + */ +static int +lua_dns_resolver_idna_convert_utf8 (lua_State *L) +{ + struct rspamd_dns_resolver *dns_resolver = lua_check_dns_resolver (L, 1); + gsize hlen; + guint conv_len = 0; + const gchar *hname = luaL_checklstring (L, 2, &hlen); + gchar *converted; + rspamd_mempool_t *pool = rspamd_lua_check_udata_maybe (L, 3, "rspamd{mempool}"); + + + if (dns_resolver && hname) { + if (!rspamd_str_has_8bit (hname, hlen)) { + /* No 8 bit, no reasons to call idna */ + lua_pushlstring (L, hname, hlen); + } + else { + converted = rspamd_dns_resolver_idna_convert_utf8 (dns_resolver, pool, + hname, hlen, &conv_len); + + if (converted == NULL) { + lua_pushnil (L); + } + else { + lua_pushlstring (L, converted, conv_len); + + if (pool == NULL) { + g_free (converted); + } + } + } + } + else { + return luaL_error (L, "invalid arguments"); + } + + return 1; +} + static gint lua_load_dns_resolver (lua_State *L) { |