| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476 |
- /**
- * $Id$
- *
- * Copyright (C) 2010 Daniel-Constantin Mierla (asipto.com)
- *
- * This file is part of Kamailio, a free SIP server.
- *
- * Kamailio is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version
- *
- * Kamailio is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
- *
- */
- #include <stdio.h>
- #include <unistd.h>
- #include <stdlib.h>
- #include <sys/stat.h>
- #include "../../sr_module.h"
- #include "../../dprint.h"
- #include "../../route.h"
- #include "../../ut.h"
- #include "../../modules/sl/sl.h"
- #include "../../modules/tm/tm_load.h"
- #include "../../modules_k/sqlops/sql_api.h"
- #include "../../modules_k/rr/api.h"
- #include "../../modules/auth/api.h"
- #include "../../modules_k/auth_db/api.h"
- #include "../../modules_k/maxfwd/api.h"
- #include "../../modules_k/registrar/api.h"
- #include "../../modules_k/dispatcher/api.h"
- #include "../../modules/xhttp/api.h"
- #include "app_lua_api.h"
- #define SR_LUA_EXP_MOD_SL (1<<0)
- #define SR_LUA_EXP_MOD_TM (1<<1)
- #define SR_LUA_EXP_MOD_SQLOPS (1<<2)
- #define SR_LUA_EXP_MOD_RR (1<<3)
- #define SR_LUA_EXP_MOD_AUTH (1<<4)
- #define SR_LUA_EXP_MOD_AUTH_DB (1<<5)
- #define SR_LUA_EXP_MOD_MAXFWD (1<<6)
- #define SR_LUA_EXP_MOD_REGISTRAR (1<<7)
- #define SR_LUA_EXP_MOD_DISPATCHER (1<<8)
- #define SR_LUA_EXP_MOD_XHTTP (1<<9)
- /**
- *
- */
- static unsigned int _sr_lua_exp_reg_mods = 0;
- /**
- * auth
- */
- static auth_api_s_t _lua_authb;
- /**
- * auth_db
- */
- static auth_db_api_t _lua_auth_dbb;
- /**
- * dispatcher
- */
- static dispatcher_api_t _lua_dispatcherb;
- /**
- * maxfwd
- */
- static maxfwd_api_t _lua_maxfwdb;
- /**
- * registrar
- */
- static registrar_api_t _lua_registrarb;
- /**
- * rr
- */
- static rr_api_t _lua_rrb;
- /**
- * sqlops
- */
- static sqlops_api_t _lua_sqlopsb;
- /**
- * sl
- */
- static sl_api_t _lua_slb;
- /**
- * tm
- */
- static tm_api_t _lua_tmb;
- static tm_xapi_t _lua_xtmb;
- /**
- * xhttp
- */
- static xhttp_api_t _lua_xhttpb;
- /**
- *
- */
- static int lua_sr_sl_send_reply (lua_State *L)
- {
- str txt;
- int code;
- int ret;
- sr_lua_env_t *env_L;
- env_L = sr_lua_env_get();
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_SL))
- {
- LM_WARN("weird: sl function executed but module not registered\n");
- return app_lua_return_error(L);
- }
- if(env_L->msg==NULL)
- {
- LM_WARN("invalid parameters from Lua env\n");
- return app_lua_return_error(L);
- }
- code = lua_tointeger(L, -2);
- if(code<100 || code>=800)
- return app_lua_return_error(L);
-
- txt.s = (char*)lua_tostring(L, -1);
- if(txt.s==NULL || env_L->msg==NULL)
- return app_lua_return_error(L);
- txt.len = strlen(txt.s);
- ret = _lua_slb.freply(env_L->msg, code, &txt);
- return app_lua_return_int(L, ret);
- }
- /**
- *
- */
- static int lua_sr_sl_get_reply_totag (lua_State *L)
- {
- str txt;
- int ret;
- sr_lua_env_t *env_L;
- env_L = sr_lua_env_get();
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_SL))
- {
- LM_WARN("weird: sl function executed but module not registered\n");
- return app_lua_return_false(L);
- }
- if(env_L->msg==NULL)
- {
- LM_WARN("invalid parameters from Lua env\n");
- return app_lua_return_error(L);
- }
- ret = _lua_slb.get_reply_totag(env_L->msg, &txt);
- if(ret<0)
- {
- LM_WARN("sl get_reply_totag returned false\n");
- return app_lua_return_false(L);
- }
- lua_pushlstring(L, txt.s, txt.len);
- return 1;
- }
- /**
- *
- */
- static const luaL_reg _sr_sl_Map [] = {
- {"send_reply", lua_sr_sl_send_reply},
- {"get_reply_totag", lua_sr_sl_get_reply_totag},
- {NULL, NULL}
- };
- /**
- *
- */
- static int lua_sr_tm_t_reply(lua_State *L)
- {
- char *txt;
- int code;
- int ret;
- sr_lua_env_t *env_L;
- env_L = sr_lua_env_get();
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_TM))
- {
- LM_WARN("weird: tm function executed but module not registered\n");
- return app_lua_return_error(L);
- }
- if(env_L->msg==NULL)
- {
- LM_WARN("invalid parameters from Lua env\n");
- return app_lua_return_error(L);
- }
- code = lua_tointeger(L, -2);
- if(code<100 || code>=800)
- return app_lua_return_error(L);
- txt = (char*)lua_tostring(L, -1);
- if(txt!=NULL && env_L->msg!=NULL)
- {
- ret = _lua_tmb.t_reply(env_L->msg, code, txt);
- return app_lua_return_int(L, ret);
- }
- return app_lua_return_error(L);
- }
- /**
- *
- */
- static int lua_sr_tm_t_relay(lua_State *L)
- {
- int ret;
- sr_lua_env_t *env_L;
- env_L = sr_lua_env_get();
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_TM))
- {
- LM_WARN("weird: tm function executed but module not registered\n");
- return app_lua_return_error(L);
- }
- if(env_L->msg==NULL)
- {
- LM_WARN("invalid parameters from Lua env\n");
- return app_lua_return_error(L);
- }
- ret = _lua_tmb.t_relay(env_L->msg, NULL, NULL);
- return app_lua_return_int(L, ret);
- }
- /**
- *
- */
- static int lua_sr_tm_t_on_failure(lua_State *L)
- {
- char *name;
- int i;
- sr_lua_env_t *env_L;
- env_L = sr_lua_env_get();
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_TM))
- {
- LM_WARN("weird: tm function executed but module not registered\n");
- return app_lua_return_error(L);
- }
- if(env_L->msg==NULL)
- {
- LM_WARN("invalid parameters from Lua env\n");
- return app_lua_return_error(L);
- }
- name = (char*)lua_tostring(L, -1);
- if(name==NULL)
- {
- LM_WARN("invalid parameters from Lua\n");
- return app_lua_return_error(L);
- }
- i = route_get(&failure_rt, name);
- if(failure_rt.rlist[i]==0)
- {
- LM_WARN("no actions in failure_route[%s]\n", name);
- return app_lua_return_error(L);
- }
- _lua_xtmb.t_on_failure((unsigned int)i);
- return app_lua_return_int(L, 1);
- }
- /**
- *
- */
- static int lua_sr_tm_t_on_branch(lua_State *L)
- {
- char *name;
- int i;
- sr_lua_env_t *env_L;
- env_L = sr_lua_env_get();
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_TM))
- {
- LM_WARN("weird: tm function executed but module not registered\n");
- return app_lua_return_error(L);
- }
- if(env_L->msg==NULL)
- {
- LM_WARN("invalid parameters from Lua env\n");
- return app_lua_return_error(L);
- }
- name = (char*)lua_tostring(L, -1);
- if(name==NULL)
- {
- LM_WARN("invalid parameters from Lua\n");
- return app_lua_return_error(L);
- }
- i = route_get(&branch_rt, name);
- if(branch_rt.rlist[i]==0)
- {
- LM_WARN("no actions in branch_route[%s]\n", name);
- return app_lua_return_error(L);
- }
- _lua_xtmb.t_on_branch((unsigned int)i);
- return app_lua_return_int(L, 1);
- }
- /**
- *
- */
- static int lua_sr_tm_t_on_reply(lua_State *L)
- {
- char *name;
- int i;
- sr_lua_env_t *env_L;
- env_L = sr_lua_env_get();
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_TM))
- {
- LM_WARN("weird: tm function executed but module not registered\n");
- return app_lua_return_error(L);
- }
- if(env_L->msg==NULL)
- {
- LM_WARN("invalid parameters from Lua env\n");
- return app_lua_return_error(L);
- }
- name = (char*)lua_tostring(L, -1);
- if(name==NULL)
- {
- LM_WARN("invalid parameters from Lua\n");
- return app_lua_return_error(L);
- }
- i = route_get(&onreply_rt, name);
- if(onreply_rt.rlist[i]==0)
- {
- LM_WARN("no actions in onreply_route[%s]\n", name);
- return app_lua_return_error(L);
- }
- _lua_xtmb.t_on_reply((unsigned int)i);
- return app_lua_return_int(L, 1);
- }
- /**
- *
- */
- static int lua_sr_tm_t_check_trans(lua_State *L)
- {
- int ret;
- sr_lua_env_t *env_L;
- env_L = sr_lua_env_get();
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_TM))
- {
- LM_WARN("weird: tm function executed but module not registered\n");
- return app_lua_return_error(L);
- }
- if(env_L->msg==NULL)
- {
- LM_WARN("invalid parameters from Lua env\n");
- return app_lua_return_error(L);
- }
- ret = _lua_xtmb.t_check_trans(env_L->msg);
- return app_lua_return_int(L, ret);
- }
- /**
- *
- */
- static int lua_sr_tm_t_is_canceled(lua_State *L)
- {
- int ret;
- sr_lua_env_t *env_L;
- env_L = sr_lua_env_get();
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_TM))
- {
- LM_WARN("weird: tm function executed but module not registered\n");
- return app_lua_return_error(L);
- }
- if(env_L->msg==NULL)
- {
- LM_WARN("invalid parameters from Lua env\n");
- return app_lua_return_error(L);
- }
- ret = _lua_xtmb.t_is_canceled(env_L->msg);
- return app_lua_return_int(L, ret);
- }
- /**
- *
- */
- static const luaL_reg _sr_tm_Map [] = {
- {"t_reply", lua_sr_tm_t_reply},
- {"t_relay", lua_sr_tm_t_relay},
- {"t_on_failure", lua_sr_tm_t_on_failure},
- {"t_on_branch", lua_sr_tm_t_on_branch},
- {"t_on_reply", lua_sr_tm_t_on_reply},
- {"t_check_trans", lua_sr_tm_t_check_trans},
- {"t_is_canceled", lua_sr_tm_t_is_canceled},
- {NULL, NULL}
- };
- /**
- *
- */
- static int lua_sr_sqlops_query(lua_State *L)
- {
- str scon;
- str squery;
- str sres;
- int ret;
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_SQLOPS))
- {
- LM_WARN("weird: sqlops function executed but module not registered\n");
- return app_lua_return_error(L);
- }
- scon.s = (char*)lua_tostring(L, -3);
- squery.s = (char*)lua_tostring(L, -2);
- sres.s = (char*)lua_tostring(L, -1);
- if(scon.s == NULL || squery.s == NULL || sres.s == NULL)
- {
- LM_WARN("invalid parameters from Lua\n");
- return app_lua_return_error(L);
- }
- scon.len = strlen(scon.s);
- squery.len = strlen(squery.s);
- sres.len = strlen(sres.s);
- ret = _lua_sqlopsb.query(&scon, &squery, &sres);
- return app_lua_return_int(L, ret);
- }
- /**
- *
- */
- static int lua_sr_sqlops_value(lua_State *L)
- {
- str sres;
- int col;
- int row;
- sql_val_t *val;
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_SQLOPS))
- {
- LM_WARN("weird: sqlops function executed but module not registered\n");
- return app_lua_return_false(L);
- }
- sres.s = (char*)lua_tostring(L, -3);
- row = lua_tointeger(L, -2);
- col = lua_tointeger(L, -1);
- if(row<0 || col<0 || sres.s==NULL)
- {
- LM_WARN("invalid parameters from Lua\n");
- return app_lua_return_false(L);
- }
- sres.len = strlen(sres.s);
- if(_lua_sqlopsb.value(&sres, row, col, &val)<0)
- return app_lua_return_false(L);
- if(val->flags&PV_VAL_NULL)
- {
- lua_pushinteger(L, 0);
- return 1;
- }
- if(val->flags&PV_VAL_INT)
- {
- lua_pushinteger(L, val->value.n);
- return 1;
- }
- lua_pushlstring(L, val->value.s.s, val->value.s.len);
- return 1;
- }
- /**
- *
- */
- static int lua_sr_sqlops_is_null(lua_State *L)
- {
- str sres;
- int col;
- int row;
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_SQLOPS))
- {
- LM_WARN("weird: sqlops function executed but module not registered\n");
- return app_lua_return_false(L);
- }
- sres.s = (char*)lua_tostring(L, -3);
- row = lua_tointeger(L, -2);
- col = lua_tointeger(L, -1);
- if(row<0 || col<0 || sres.s==NULL)
- {
- LM_WARN("invalid parameters from Lua\n");
- return app_lua_return_false(L);
- }
- sres.len = strlen(sres.s);
- if(_lua_sqlopsb.is_null(&sres, row, col)==1)
- return app_lua_return_true(L);
- return app_lua_return_false(L);
- }
- /**
- *
- */
- static int lua_sr_sqlops_column(lua_State *L)
- {
- str sres;
- int col;
- str name = {0, 0};
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_SQLOPS))
- {
- LM_WARN("weird: sqlops function executed but module not registered\n");
- return app_lua_return_false(L);
- }
- sres.s = (char*)lua_tostring(L, -2);
- col = lua_tointeger(L, -1);
- if(col<0 || sres.s==NULL)
- {
- LM_WARN("invalid parameters from Lua\n");
- return app_lua_return_false(L);
- }
- sres.len = strlen(sres.s);
- if(_lua_sqlopsb.column(&sres, col, &name)<0)
- return app_lua_return_false(L);
- lua_pushlstring(L, name.s, name.len);
- return 1;
- }
- /**
- *
- */
- static int lua_sr_sqlops_nrows(lua_State *L)
- {
- str sres;
- int rows;
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_SQLOPS))
- {
- LM_WARN("weird: sqlops function executed but module not registered\n");
- return app_lua_return_false(L);
- }
- sres.s = (char*)lua_tostring(L, -1);
- if(sres.s==NULL)
- {
- LM_WARN("invalid parameters from Lua\n");
- return app_lua_return_false(L);
- }
- sres.len = strlen(sres.s);
- rows = _lua_sqlopsb.nrows(&sres);
- if(rows<0)
- return app_lua_return_false(L);
- lua_pushinteger(L, rows);
- return 1;
- }
- /**
- *
- */
- static int lua_sr_sqlops_ncols(lua_State *L)
- {
- str sres;
- int cols;
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_SQLOPS))
- {
- LM_WARN("weird: sqlops function executed but module not registered\n");
- return app_lua_return_false(L);
- }
- sres.s = (char*)lua_tostring(L, -1);
- if(sres.s==NULL)
- {
- LM_WARN("invalid parameters from Lua\n");
- return app_lua_return_false(L);
- }
- sres.len = strlen(sres.s);
- cols = _lua_sqlopsb.ncols(&sres);
- if(cols<0)
- return app_lua_return_false(L);
- lua_pushinteger(L, cols);
- return 1;
- }
- /**
- *
- */
- static int lua_sr_sqlops_reset(lua_State *L)
- {
- str sres;
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_SQLOPS))
- {
- LM_WARN("weird: sqlops function executed but module not registered\n");
- return app_lua_return_false(L);
- }
- sres.s = (char*)lua_tostring(L, -1);
- if(sres.s==NULL)
- {
- LM_WARN("invalid parameters from Lua\n");
- return app_lua_return_false(L);
- }
- sres.len = strlen(sres.s);
- _lua_sqlopsb.reset(&sres);
- return app_lua_return_true(L);
- }
- /**
- *
- */
- static const luaL_reg _sr_sqlops_Map [] = {
- {"query", lua_sr_sqlops_query},
- {"value", lua_sr_sqlops_value},
- {"is_null", lua_sr_sqlops_is_null},
- {"column", lua_sr_sqlops_column},
- {"nrows", lua_sr_sqlops_nrows},
- {"ncols", lua_sr_sqlops_ncols},
- {"reset", lua_sr_sqlops_reset},
- {NULL, NULL}
- };
- /**
- *
- */
- static int lua_sr_rr_record_route(lua_State *L)
- {
- int ret;
- str sv = {0, 0};
- sr_lua_env_t *env_L;
- env_L = sr_lua_env_get();
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_RR))
- {
- LM_WARN("weird: rr function executed but module not registered\n");
- return app_lua_return_error(L);
- }
- if(env_L->msg==NULL)
- {
- LM_WARN("invalid parameters from Lua env\n");
- return app_lua_return_error(L);
- }
- if(lua_gettop(L)==1)
- {
- sv.s = (char*)lua_tostring(L, -1);
- if(sv.s!=NULL)
- sv.len = strlen(sv.s);
- }
- ret = _lua_rrb.record_route(env_L->msg, (sv.len>0)?&sv:NULL);
- return app_lua_return_int(L, ret);
- }
- /**
- *
- */
- static int lua_sr_rr_loose_route(lua_State *L)
- {
- int ret;
- sr_lua_env_t *env_L;
- env_L = sr_lua_env_get();
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_RR))
- {
- LM_WARN("weird: rr function executed but module not registered\n");
- return app_lua_return_error(L);
- }
- if(env_L->msg==NULL)
- {
- LM_WARN("invalid parameters from Lua env\n");
- return app_lua_return_error(L);
- }
- ret = _lua_rrb.loose_route(env_L->msg);
- return app_lua_return_int(L, ret);
- }
- /**
- *
- */
- static const luaL_reg _sr_rr_Map [] = {
- {"record_route", lua_sr_rr_record_route},
- {"loose_route", lua_sr_rr_loose_route},
- {NULL, NULL}
- };
- static int lua_sr_auth_challenge(lua_State *L, int hftype)
- {
- int ret;
- str realm = {0, 0};
- int flags;
- sr_lua_env_t *env_L;
- env_L = sr_lua_env_get();
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_AUTH))
- {
- LM_WARN("weird: auth function executed but module not registered\n");
- return app_lua_return_error(L);
- }
- if(env_L->msg==NULL)
- {
- LM_WARN("invalid parameters from Lua env\n");
- return app_lua_return_error(L);
- }
- if(lua_gettop(L)!=2)
- {
- LM_WARN("invalid number of parameters from Lua\n");
- return app_lua_return_error(L);
- }
- realm.s = (char*)lua_tostring(L, -2);
- flags = lua_tointeger(L, -1);
- if(flags<0 || realm.s==NULL)
- {
- LM_WARN("invalid parameters from Lua\n");
- return app_lua_return_error(L);
- }
- realm.len = strlen(realm.s);
- ret = _lua_authb.auth_challenge(env_L->msg, &realm, flags, hftype);
- return app_lua_return_int(L, ret);
- }
- /**
- *
- */
- static int lua_sr_auth_www_challenge(lua_State *L)
- {
- return lua_sr_auth_challenge(L, HDR_AUTHORIZATION_T);
- }
- /**
- *
- */
- static int lua_sr_auth_proxy_challenge(lua_State *L)
- {
- return lua_sr_auth_challenge(L, HDR_PROXYAUTH_T);
- }
- /**
- *
- */
- static int lua_sr_auth_pv_authenticate(lua_State *L, int hftype)
- {
- int ret;
- str realm = {0, 0};
- str passwd = {0, 0};
- int flags;
- sr_lua_env_t *env_L;
- env_L = sr_lua_env_get();
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_AUTH))
- {
- LM_WARN("weird: auth function executed but module not registered\n");
- return app_lua_return_error(L);
- }
- if(env_L->msg==NULL)
- {
- LM_WARN("invalid parameters from Lua env\n");
- return app_lua_return_error(L);
- }
- if(lua_gettop(L)!=3)
- {
- LM_WARN("invalid number of parameters from Lua\n");
- return app_lua_return_error(L);
- }
- realm.s = (char*)lua_tostring(L, -3);
- passwd.s = (char*)lua_tostring(L, -2);
- flags = lua_tointeger(L, -1);
- if(flags<0 || realm.s==NULL || passwd.s==NULL)
- {
- LM_WARN("invalid parameters from Lua\n");
- return app_lua_return_error(L);
- }
- realm.len = strlen(realm.s);
- passwd.len = strlen(passwd.s);
- ret = _lua_authb.pv_authenticate(env_L->msg, &realm, &passwd, flags,
- hftype);
- return app_lua_return_int(L, ret);
- }
- /**
- *
- */
- static int lua_sr_auth_pv_www_authenticate(lua_State *L)
- {
- return lua_sr_auth_pv_authenticate(L, HDR_AUTHORIZATION_T);
- }
- /**
- *
- */
- static int lua_sr_auth_pv_proxy_authenticate(lua_State *L)
- {
- return lua_sr_auth_pv_authenticate(L, HDR_PROXYAUTH_T);
- }
- /**
- *
- */
- static int lua_sr_auth_consume_credentials(lua_State *L)
- {
- int ret;
- sr_lua_env_t *env_L;
- env_L = sr_lua_env_get();
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_AUTH))
- {
- LM_WARN("weird: auth function executed but module not registered\n");
- return app_lua_return_error(L);
- }
- if(env_L->msg==NULL)
- {
- LM_WARN("invalid parameters from Lua env\n");
- return app_lua_return_error(L);
- }
- ret = _lua_authb.consume_credentials(env_L->msg);
- return app_lua_return_int(L, ret);
- }
- /**
- *
- */
- static const luaL_reg _sr_auth_Map [] = {
- {"www_challenge", lua_sr_auth_www_challenge},
- {"proxy_challenge", lua_sr_auth_proxy_challenge},
- {"pv_www_authenticate", lua_sr_auth_pv_www_authenticate},
- {"pv_proxy_authenticate", lua_sr_auth_pv_proxy_authenticate},
- {"consume_credentials", lua_sr_auth_consume_credentials},
- {NULL, NULL}
- };
- /**
- *
- */
- static int lua_sr_auth_db_authenticate(lua_State *L, hdr_types_t hftype)
- {
- int ret;
- str realm = {0, 0};
- str table = {0, 0};
- sr_lua_env_t *env_L;
- env_L = sr_lua_env_get();
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_AUTH_DB))
- {
- LM_WARN("weird: auth function executed but module not registered\n");
- return app_lua_return_error(L);
- }
- if(env_L->msg==NULL)
- {
- LM_WARN("invalid parameters from Lua env\n");
- return app_lua_return_error(L);
- }
- if(lua_gettop(L)!=2)
- {
- LM_WARN("invalid number of parameters from Lua\n");
- return app_lua_return_error(L);
- }
- realm.s = (char*)lua_tostring(L, -2);
- table.s = (char*)lua_tostring(L, -1);
- if(realm.s==NULL || table.s==NULL)
- {
- LM_WARN("invalid parameters from Lua\n");
- return app_lua_return_error(L);
- }
- realm.len = strlen(realm.s);
- table.len = strlen(table.s);
- ret = _lua_auth_dbb.digest_authenticate(env_L->msg, &realm, &table,
- hftype);
- return app_lua_return_int(L, ret);
- }
- /**
- *
- */
- static int lua_sr_auth_db_www_authenticate(lua_State *L)
- {
- return lua_sr_auth_db_authenticate(L, HDR_AUTHORIZATION_T);
- }
- /**
- *
- */
- static int lua_sr_auth_db_proxy_authenticate(lua_State *L)
- {
- return lua_sr_auth_db_authenticate(L, HDR_PROXYAUTH_T);
- }
- /**
- *
- */
- static const luaL_reg _sr_auth_db_Map [] = {
- {"www_authenticate", lua_sr_auth_db_www_authenticate},
- {"proxy_authenticate", lua_sr_auth_db_proxy_authenticate},
- {NULL, NULL}
- };
- /**
- *
- */
- static int lua_sr_maxfwd_process_maxfwd(lua_State *L)
- {
- int ret;
- int limit;
- sr_lua_env_t *env_L;
- env_L = sr_lua_env_get();
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_MAXFWD))
- {
- LM_WARN("weird: maxfwd function executed but module not registered\n");
- return app_lua_return_error(L);
- }
- if(env_L->msg==NULL)
- {
- LM_WARN("invalid parameters from Lua env\n");
- return app_lua_return_error(L);
- }
- if(lua_gettop(L)!=1)
- {
- LM_WARN("invalid number of parameters from Lua\n");
- return app_lua_return_error(L);
- }
- limit = lua_tointeger(L, -1);
- if(limit<0)
- {
- LM_WARN("invalid parameters from Lua\n");
- return app_lua_return_error(L);
- }
- ret = _lua_maxfwdb.process_maxfwd(env_L->msg, limit);
- return app_lua_return_int(L, ret);
- }
- /**
- *
- */
- static const luaL_reg _sr_maxfwd_Map [] = {
- {"process_maxfwd", lua_sr_maxfwd_process_maxfwd},
- {NULL, NULL}
- };
- /**
- *
- */
- static int lua_sr_registrar_save(lua_State *L)
- {
- int ret;
- int flags;
- char *table;
- sr_lua_env_t *env_L;
- flags = 0;
- env_L = sr_lua_env_get();
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_MAXFWD))
- {
- LM_WARN("weird: maxfwd function executed but module not registered\n");
- return app_lua_return_error(L);
- }
- if(env_L->msg==NULL)
- {
- LM_WARN("invalid parameters from Lua env\n");
- return app_lua_return_error(L);
- }
- if(lua_gettop(L)==1)
- {
- table = (char*)lua_tostring(L, -1);
- } else if(lua_gettop(L)==2) {
- table = (char*)lua_tostring(L, -2);
- flags = lua_tointeger(L, -1);
- } else {
- LM_WARN("invalid number of parameters from Lua\n");
- return app_lua_return_error(L);
- }
- if(table==NULL || strlen(table)==0)
- {
- LM_WARN("invalid parameters from Lua\n");
- return app_lua_return_error(L);
- }
- ret = _lua_registrarb.save(env_L->msg, table, flags);
- return app_lua_return_int(L, ret);
- }
- /**
- *
- */
- static int lua_sr_registrar_lookup(lua_State *L)
- {
- int ret;
- char *table;
- sr_lua_env_t *env_L;
- env_L = sr_lua_env_get();
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_MAXFWD))
- {
- LM_WARN("weird: maxfwd function executed but module not registered\n");
- return app_lua_return_error(L);
- }
- if(env_L->msg==NULL)
- {
- LM_WARN("invalid parameters from Lua env\n");
- return app_lua_return_error(L);
- }
- if(lua_gettop(L)!=1)
- {
- LM_WARN("invalid number of parameters from Lua\n");
- return app_lua_return_error(L);
- }
- table = (char*)lua_tostring(L, -1);
- if(table==NULL || strlen(table)==0)
- {
- LM_WARN("invalid parameters from Lua\n");
- return app_lua_return_error(L);
- }
- ret = _lua_registrarb.lookup(env_L->msg, table);
- return app_lua_return_int(L, ret);
- }
- /**
- *
- */
- static const luaL_reg _sr_registrar_Map [] = {
- {"save", lua_sr_registrar_save},
- {"lookup", lua_sr_registrar_lookup},
- {NULL, NULL}
- };
- /**
- *
- */
- static int lua_sr_dispatcher_select(lua_State *L)
- {
- int ret;
- int setid;
- int algid;
- int mode;
- sr_lua_env_t *env_L;
- env_L = sr_lua_env_get();
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_DISPATCHER))
- {
- LM_WARN("weird: dispatcher function executed but module"
- " not registered\n");
- return app_lua_return_error(L);
- }
- if(env_L->msg==NULL)
- {
- LM_WARN("invalid parameters from Lua env\n");
- return app_lua_return_error(L);
- }
- if(lua_gettop(L)==3)
- {
- setid = lua_tointeger(L, -3);
- algid = lua_tointeger(L, -2);
- mode = lua_tointeger(L, -1);
- } else if(lua_gettop(L)==2) {
- setid = lua_tointeger(L, -2);
- algid = lua_tointeger(L, -1);
- mode = 0;
- } else {
- LM_WARN("invalid number of parameters from Lua\n");
- return app_lua_return_error(L);
- }
- ret = _lua_dispatcherb.select(env_L->msg, setid, algid, mode);
- return app_lua_return_int(L, ret);
- }
- /**
- *
- */
- static int lua_sr_dispatcher_next(lua_State *L)
- {
- int ret;
- int mode;
- sr_lua_env_t *env_L;
- env_L = sr_lua_env_get();
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_DISPATCHER))
- {
- LM_WARN("weird: dispatcher function executed but module"
- " not registered\n");
- return app_lua_return_error(L);
- }
- if(env_L->msg==NULL)
- {
- LM_WARN("invalid parameters from Lua env\n");
- return app_lua_return_error(L);
- }
- mode = 0;
- if(lua_gettop(L)==1)
- {
- /* mode given as parameter */
- mode = lua_tointeger(L, -1);
- }
- ret = _lua_dispatcherb.next(env_L->msg, mode);
- return app_lua_return_int(L, ret);
- }
- /**
- *
- */
- static int lua_sr_dispatcher_mark(lua_State *L)
- {
- int ret;
- int mode;
- sr_lua_env_t *env_L;
- env_L = sr_lua_env_get();
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_DISPATCHER))
- {
- LM_WARN("weird: dispatcher function executed but module"
- " not registered\n");
- return app_lua_return_error(L);
- }
- if(env_L->msg==NULL)
- {
- LM_WARN("invalid parameters from Lua env\n");
- return app_lua_return_error(L);
- }
- mode = 0;
- if(lua_gettop(L)==1)
- {
- /* mode given as parameter */
- mode = lua_tointeger(L, -1);
- }
- ret = _lua_dispatcherb.mark(env_L->msg, mode);
- return app_lua_return_int(L, ret);
- }
- /**
- *
- */
- static int lua_sr_dispatcher_is_from(lua_State *L)
- {
- int ret;
- int mode;
- sr_lua_env_t *env_L;
- env_L = sr_lua_env_get();
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_DISPATCHER))
- {
- LM_WARN("weird: dispatcher function executed but module"
- " not registered\n");
- return app_lua_return_error(L);
- }
- if(env_L->msg==NULL)
- {
- LM_WARN("invalid parameters from Lua env\n");
- return app_lua_return_error(L);
- }
- mode = -1;
- if(lua_gettop(L)==1)
- {
- /* mode given as parameter */
- mode = lua_tointeger(L, -1);
- }
- ret = _lua_dispatcherb.is_from(env_L->msg, mode);
- return app_lua_return_int(L, ret);
- }
- /**
- *
- */
- static int lua_sr_xhttp_reply(lua_State *L)
- {
- int rcode;
- str reason;
- str ctype;
- str mbody;
- int ret;
- sr_lua_env_t *env_L;
- env_L = sr_lua_env_get();
- if(!(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_XHTTP))
- {
- LM_WARN("weird: xhttp function executed but module not registered\n");
- return app_lua_return_error(L);
- }
- if(env_L->msg==NULL)
- {
- LM_WARN("invalid parameters from Lua env\n");
- return app_lua_return_error(L);
- }
- rcode = lua_tointeger(L, -4);
- reason.s = (char*)lua_tostring(L, -3);
- ctype.s = (char*)lua_tostring(L, -2);
- mbody.s = (char*)lua_tostring(L, -1);
- if(reason.s == NULL || ctype.s == NULL || mbody.s == NULL)
- {
- LM_WARN("invalid parameters from Lua\n");
- return app_lua_return_error(L);
- }
- reason.len = strlen(reason.s);
- ctype.len = strlen(ctype.s);
- mbody.len = strlen(mbody.s);
- ret = _lua_xhttpb.reply(env_L->msg, rcode, &reason, &ctype, &mbody);
- return app_lua_return_int(L, ret);
- }
- /**
- *
- */
- static const luaL_reg _sr_xhttp_Map [] = {
- {"reply", lua_sr_xhttp_reply},
- {NULL, NULL}
- };
- /**
- *
- */
- static const luaL_reg _sr_dispatcher_Map [] = {
- {"select", lua_sr_dispatcher_select},
- {"next", lua_sr_dispatcher_next},
- {"mark", lua_sr_dispatcher_mark},
- {"is_from", lua_sr_dispatcher_is_from},
- {NULL, NULL}
- };
- /**
- *
- */
- int lua_sr_exp_init_mod(void)
- {
- if(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_SL)
- {
- /* bind the SL API */
- if (sl_load_api(&_lua_slb) < 0) {
- LM_ERR("cannot bind to SL API\n");
- return -1;
- }
- LM_DBG("loaded sl api\n");
- }
- if(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_TM)
- {
- /* bind the TM API */
- if (tm_load_api(&_lua_tmb) < 0)
- {
- LM_ERR("cannot bind to TM API\n");
- return -1;
- }
- LM_DBG("loaded tm api\n");
- /* bind the TM XAPI */
- if (tm_load_xapi(&_lua_xtmb) < 0)
- {
- LM_ERR("cannot bind to TM XAPI\n");
- return -1;
- }
- LM_DBG("loaded tm xapi\n");
- }
- if(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_SQLOPS)
- {
- /* bind the SQLOPS API */
- if (sqlops_load_api(&_lua_sqlopsb) < 0)
- {
- LM_ERR("cannot bind to SQLOPS API\n");
- return -1;
- }
- LM_DBG("loaded sqlops api\n");
- }
- if(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_RR)
- {
- /* bind the RR API */
- if (rr_load_api(&_lua_rrb) < 0)
- {
- LM_ERR("cannot bind to RR API\n");
- return -1;
- }
- LM_DBG("loaded rr api\n");
- }
- if(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_AUTH)
- {
- /* bind the AUTH API */
- if (auth_load_api(&_lua_authb) < 0)
- {
- LM_ERR("cannot bind to AUTH API\n");
- return -1;
- }
- LM_DBG("loaded auth api\n");
- }
- if(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_AUTH_DB)
- {
- /* bind the AUTH_DB API */
- if (auth_db_load_api(&_lua_auth_dbb) < 0)
- {
- LM_ERR("cannot bind to AUTH_DB API\n");
- return -1;
- }
- LM_DBG("loaded auth_db api\n");
- }
- if(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_MAXFWD)
- {
- /* bind the MAXFWD API */
- if (maxfwd_load_api(&_lua_maxfwdb) < 0)
- {
- LM_ERR("cannot bind to MAXFWD API\n");
- return -1;
- }
- LM_DBG("loaded maxfwd api\n");
- }
- if(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_REGISTRAR)
- {
- /* bind the REGISTRAR API */
- if (registrar_load_api(&_lua_registrarb) < 0)
- {
- LM_ERR("cannot bind to REGISTRAR API\n");
- return -1;
- }
- LM_DBG("loaded registrar api\n");
- }
- if(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_DISPATCHER)
- {
- /* bind the DISPATCHER API */
- if (dispatcher_load_api(&_lua_dispatcherb) < 0)
- {
- LM_ERR("cannot bind to DISPATCHER API\n");
- return -1;
- }
- LM_DBG("loaded dispatcher api\n");
- }
- if(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_XHTTP)
- {
- /* bind the XHTTP API */
- if (xhttp_load_api(&_lua_xhttpb) < 0)
- {
- LM_ERR("cannot bind to XHTTP API\n");
- return -1;
- }
- LM_DBG("loaded xhttp api\n");
- }
- return 0;
- }
- /**
- *
- */
- int lua_sr_exp_register_mod(char *mname)
- {
- int len;
- len = strlen(mname);
- if(len==2 && strcmp(mname, "sl")==0)
- {
- _sr_lua_exp_reg_mods |= SR_LUA_EXP_MOD_SL;
- return 0;
- } else if(len==2 && strcmp(mname, "tm")==0) {
- _sr_lua_exp_reg_mods |= SR_LUA_EXP_MOD_TM;
- return 0;
- } else if(len==6 && strcmp(mname, "sqlops")==0) {
- _sr_lua_exp_reg_mods |= SR_LUA_EXP_MOD_SQLOPS;
- return 0;
- } else if(len==2 && strcmp(mname, "rr")==0) {
- _sr_lua_exp_reg_mods |= SR_LUA_EXP_MOD_RR;
- return 0;
- } else if(len==4 && strcmp(mname, "auth")==0) {
- _sr_lua_exp_reg_mods |= SR_LUA_EXP_MOD_AUTH;
- return 0;
- } else if(len==7 && strcmp(mname, "auth_db")==0) {
- _sr_lua_exp_reg_mods |= SR_LUA_EXP_MOD_AUTH_DB;
- return 0;
- } else if(len==6 && strcmp(mname, "maxfwd")==0) {
- _sr_lua_exp_reg_mods |= SR_LUA_EXP_MOD_MAXFWD;
- return 0;
- } else if(len==9 && strcmp(mname, "registrar")==0) {
- _sr_lua_exp_reg_mods |= SR_LUA_EXP_MOD_REGISTRAR;
- return 0;
- } else if(len==10 && strcmp(mname, "dispatcher")==0) {
- _sr_lua_exp_reg_mods |= SR_LUA_EXP_MOD_DISPATCHER;
- return 0;
- } else if(len==5 && strcmp(mname, "xhttp")==0) {
- _sr_lua_exp_reg_mods |= SR_LUA_EXP_MOD_XHTTP;
- return 0;
- }
- return -1;
- }
- /**
- *
- */
- void lua_sr_exp_openlibs(lua_State *L)
- {
- if(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_SL)
- luaL_openlib(L, "sr.sl", _sr_sl_Map, 0);
- if(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_TM)
- luaL_openlib(L, "sr.tm", _sr_tm_Map, 0);
- if(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_SQLOPS)
- luaL_openlib(L, "sr.sqlops", _sr_sqlops_Map, 0);
- if(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_RR)
- luaL_openlib(L, "sr.rr", _sr_rr_Map, 0);
- if(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_AUTH)
- luaL_openlib(L, "sr.auth", _sr_auth_Map, 0);
- if(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_AUTH_DB)
- luaL_openlib(L, "sr.auth_db", _sr_auth_db_Map, 0);
- if(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_MAXFWD)
- luaL_openlib(L, "sr.maxfwd", _sr_maxfwd_Map, 0);
- if(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_REGISTRAR)
- luaL_openlib(L, "sr.registrar", _sr_registrar_Map, 0);
- if(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_DISPATCHER)
- luaL_openlib(L, "sr.dispatcher", _sr_dispatcher_Map, 0);
- if(_sr_lua_exp_reg_mods&SR_LUA_EXP_MOD_XHTTP)
- luaL_openlib(L, "sr.xhttp", _sr_xhttp_Map, 0);
- }
|