Module: sip-router Branch: master Commit: 55280b28321d735babbdb529c78a0f0a458e524f URL: http://git.sip-router.org/cgi-bin/gitweb.cgi/sip-router/?a=commit;h=55280b2… Author: Victor Seva &lt;linuxmaniac(a)torreviejawireless.org&gt; Committer: Victor Seva &lt;linuxmaniac(a)torreviejawireless.org&gt; Date: Fri Jan 3 10:02:30 2014 +0100 app_lua: sr.xavp.get() allow get all values of a var and <0 indx --- modules/app_lua/app_lua_sr.c | 75 +++++++++++++++++++++++++++++++++--------- 1 files changed, 59 insertions(+), 16 deletions(-) diff --git a/modules/app_lua/app_lua_sr.c b/modules/app_lua/app_lua_sr.c index 14ea838..b7abb33 100644 --- a/modules/app_lua/app_lua_sr.c +++ b/modules/app_lua/app_lua_sr.c @@ -1299,7 +1299,10 @@ static int lua_sr_xavp_get(lua_State *L) sr_xavp_t *avp; int num_param = 0; int param = -1; + int all_flag = 0; int simple_flag = 0; + lua_Number elem = 1; + int xavp_size = 0; env_L = sr_lua_env_get(); num_param = lua_gettop(L); @@ -1322,32 +1325,72 @@ static int lua_sr_xavp_get(lua_State *L) if(!lua_isnumber(L, param)) { - LM_ERR("invalid int parameter\n"); - return 0; + if(lua_isnil(L, param)) + { + all_flag = 1; + } + else + { + LM_ERR("invalid parameter, must be int or nil\n"); + return 0; + } + } + else + { + indx = lua_tointeger(L, param); } - indx = lua_tointeger(L, param); param = param - 1; - xavp_name.s = (char*)lua_tostring(L, param); if(xavp_name.s==NULL || env_L->msg==NULL) + { + LM_ERR("No xavp name in %d param\n", param); return 0; + } xavp_name.len = strlen(xavp_name.s); - - avp = xavp_get_by_index(&xavp_name, indx, NULL); - if(avp==NULL){ - LM_ERR("can't get xavp:%.*s index:%d\n", xavp_name.len, xavp_name.s, indx); - lua_pushnil(L); - return 1; + if(all_flag>0) { + indx = 0; + lua_newtable(L); } - - if (simple_flag != 0) + xavp_size = xavp_count(&xavp_name, NULL); + if(indx<0) { - lua_sr_push_xavp_table_simple(L, avp); + if((indx*-1)>xavp_size) + { + LM_ERR("can't get xavp:%.*s index:%d\n", xavp_name.len, xavp_name.s, indx); + lua_pushnil(L); + return 1; + } + indx = xavp_size + indx; } - else + + do { - lua_sr_push_xavp_table(L, avp); - } + avp = xavp_get_by_index(&xavp_name, indx, NULL); + if(avp==NULL){ + LM_ERR("can't get xavp:%.*s index:%d\n", xavp_name.len, xavp_name.s, indx); + lua_pushnil(L); + return 1; + } + if(all_flag!=0) { + lua_pushnumber(L, elem); + elem = elem + 1; + } + if (simple_flag != 0) + { + lua_sr_push_xavp_table_simple(L, avp); + } + else + { + lua_sr_push_xavp_table(L, avp); + } + if(all_flag==0) return 1; + else { + lua_rawset(L, -3); + indx = indx + 1; + avp = xavp_get_by_index(&xavp_name, indx, NULL); + } + }while(avp!=NULL); + return 1; }