[SR-Dev] git:master: htable: execute event_route[htable:mod-init] if idefined

Module: sip-router Branch: master Commit: a18160c8f5979464bd808af4c1f565b428d2e9ac URL: http://git.sip-router.org/cgi-bin/gitweb.cgi/sip-router/?a=commit;h=a18160c… Author: Daniel-Constantin Mierla &lt;miconda(a)gmail.com&gt; Committer: Daniel-Constantin Mierla &lt;miconda(a)gmail.com&gt; Date: Tue May 19 19:44:00 2009 +0300 htable: execute event_route[htable:mod-init] if idefined - the module will execute the event route just after modules have been initialized - a typical use case could be the initialization of some values in hash tables - example: modparam("htable", "htable", "a=>size=4;") event_route[htable:mod-init] { $sht(a=>calls-to::10.10.10.10) = 0; $sht(a=>max-calls-to::10.10.10.10) = 100; $sht(a=>calls-to::10.10.10.11) = 0; $sht(a=>max-calls-to::10.10.10.11) = 10; } route { if(is_method("INVITE") && !has_totag()) { switch($rd) { case "10.10.10.10": lock("calls-to::10.10.10.10"); $sht(a=>calls-to::10.10.10.10) = $sht(a=>calls-to::10.10.10.10) + 1; unlock("calls-to::10.10.10.10"); if($sht(a=>calls-to::10.10.10.10)>$sht(a=>max-calls-to::10.10.10.10)) { sl_send_reply("500", "To many calls to .10"); exit; } break; case "10.10.10.11": ... break; } } } --- modules_k/htable/htable.c | 29 ++++++++++++++++++++++++++++- 1 files changed, 28 insertions(+), 1 deletions(-) diff --git a/modules_k/htable/htable.c b/modules_k/htable/htable.c index 4dc7f4e..bf5cbcf 100644 --- a/modules_k/htable/htable.c +++ b/modules_k/htable/htable.c @@ -30,8 +30,10 @@ #include "../../sr_module.h" #include "../../timer.h" +#include "../../route.h" #include "../../dprint.h" #include "../../lib/kmi/mi.h" +#include "../../lib/kcore/faked_msg.h" #include "../../pvar.h" #include "ht_api.h" @@ -46,6 +48,7 @@ int ht_timer_interval = 20; /** module functions */ static int ht_print(struct sip_msg*, char*, char*); static int mod_init(void); +static int child_init(int rank); void destroy(void); static int fixup_ht_rm(void** param, int param_no); @@ -115,7 +118,7 @@ struct module_exports exports= { mod_init, /* module initialization function */ 0, (destroy_function) destroy, - 0 /* per-child init function */ + child_init /* per-child init function */ }; /** @@ -157,6 +160,30 @@ static int mod_init(void) return -1; } } + return 0; +} + + +static int child_init(int rank) +{ + struct sip_msg *fmsg; + int rtb, rt; + + LM_DBG("rank is (%d)\n", rank); + if (rank!=PROC_INIT) + return 0; + + rt = route_get(&event_rt, "htable:mod-init"); + if(rt>=0) { + LM_DBG("executing event_route[htable:mod-init] (%d)\n", rt); + if(faked_msg_init()<0) + return -1; + fmsg = faked_msg_next(); + rtb = route_type; + set_route_type(REQUEST_ROUTE); + run_top_route(event_rt.rlist[rt], fmsg); + set_route_type(rtb); + } return 0; }