lua: add vlc.strings.iconv to use vlc_iconv, shouldn't eat kittens

[vlc] / modules / misc / lua / libs / strings.c

diff --git a/modules/misc/lua/libs/strings.c b/modules/misc/lua/libs/strings.c
index b9a56a48414e5870ecfcbfcd0ddae296041e5aa4..a43e972a5112b3f15e5218b8e2384d079507cec1 100644 (file)
--- a/modules/misc/lua/libs/strings.c
+++ b/modules/misc/lua/libs/strings.c
@@ -36,7 +36,6 @@
 #include <vlc_common.h>
 #include <vlc_plugin.h>
 #include <vlc_meta.h>
-#include <vlc_charset.h>
 #include <vlc_aout.h>
 
 #include <lua.h>        /* Low level lua C API */
@@ -113,6 +112,34 @@ static int vlclua_convert_xml_special_chars( lua_State *L )
     return i_top;
 }
 
+static int vlclua_iconv( lua_State *L )
+{
+    int i_ret;
+    size_t i_out_bytes, i_in_bytes;
+    char *psz_output, *psz_original;
+
+    if( lua_gettop( L ) < 3 ) return vlclua_error( L );
+
+    const char *psz_input = luaL_checklstring( L, 3, &i_in_bytes );
+
+    if( i_in_bytes == 0 ) return vlclua_error( L );
+    vlc_iconv_t iconv_handle = vlc_iconv_open( luaL_checkstring(L, 1),
+                                             luaL_checkstring(L, 2) );
+
+    if( iconv_handle == (vlc_iconv_t)-1 )
+       return vlclua_error( L );
+    psz_output = psz_original = malloc( 4 * i_in_bytes );
+    i_out_bytes = 4 * i_in_bytes;
+    i_ret = vlc_iconv( iconv_handle, &psz_input ,
+                       &i_in_bytes, &psz_output, &i_out_bytes );
+    *psz_output = '\0';
+
+    lua_pushstring( L, psz_original );
+    vlc_iconv_close( iconv_handle );
+    free( psz_original );
+    return 1;
+}
+
 /*****************************************************************************
  *
  *****************************************************************************/
@@ -121,6 +148,7 @@ static const luaL_Reg vlclua_strings_reg[] = {
     { "encode_uri_component", vlclua_encode_uri_component },
     { "resolve_xml_special_chars", vlclua_resolve_xml_special_chars },
     { "convert_xml_special_chars", vlclua_convert_xml_special_chars },
+    { "iconv", vlclua_iconv },
     { NULL, NULL }
 };