123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814 |
- /*
- * $Id$
- *
- * usrloc module
- *
- * Copyright (C) 2009 Daniel-Constantin Mierla (asipto.com).
- *
- * Permission to use, copy, modify, and distribute this software for any
- * purpose with or without fee is hereby granted, provided that the above
- * copyright notice and this permission notice appear in all copies.
- *
- * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
- * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
- * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
- * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
- * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
- * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
- * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
- */
- #include "../../ip_addr.h"
- #include "../../dprint.h"
- #include "../../lib/srutils/sruid.h"
- #include "ul_rpc.h"
- #include "dlist.h"
- #include "ucontact.h"
- #include "udomain.h"
- #include "ul_mod.h"
- #include "utime.h"
- /*! CSEQ nr used */
- #define RPC_UL_CSEQ 1
- /*! call-id used for ul_add and ul_rm_contact */
- static str rpc_ul_cid = str_init("[email protected]");
- /*! path used for ul_add and ul_rm_contact */
- static str rpc_ul_path = str_init("dummypath");
- /*! user agent used for ul_add */
- static str rpc_ul_ua = str_init("SIP Router MI Server");
- extern sruid_t _ul_sruid;
- static const char* ul_rpc_dump_doc[2] = {
- "Dump user location tables",
- 0
- };
- int rpc_dump_contact(rpc_t* rpc, void* ctx, void *ih, ucontact_t* c)
- {
- void* vh;
- str empty_str = {"[not set]", 9};
- str state_str = {"[not set]", 9};
- str socket_str = {"[not set]", 9};
- time_t t;
- t = time(0);
- if(rpc->struct_add(ih, "{", "Contact", &vh)<0)
- {
- rpc->fault(ctx, 500, "Internal error creating contact struct");
- return -1;
- }
- if(rpc->struct_add(vh, "S", "Address", &c->c)<0)
- {
- rpc->fault(ctx, 500, "Internal error adding addr");
- return -1;
- }
- if (c->expires == 0) { if(rpc->struct_add(vh, "s", "Expires", "permanent")<0)
- {
- rpc->fault(ctx, 500, "Internal error adding expire");
- return -1;
- }
- } else if (c->expires == UL_EXPIRED_TIME) {
- if(rpc->struct_add(vh, "s", "Expires", "deleted")<0)
- {
- rpc->fault(ctx, 500, "Internal error adding expire");
- return -1;
- }
- } else if (t > c->expires) {
- if(rpc->struct_add(vh, "s", "Expires", "expired")<0)
- {
- rpc->fault(ctx, 500, "Internal error adding expire");
- return -1;
- }
- } else {
- if(rpc->struct_add(vh, "d", "Expires", (int)(c->expires - t))<0)
- {
- rpc->fault(ctx, 500, "Internal error adding expire");
- return -1;
- }
- }
- if (c->state == CS_NEW) {
- state_str.s = "CS_NEW";
- state_str.len = 6;
- } else if (c->state == CS_SYNC) {
- state_str.s = "CS_SYNC";
- state_str.len = 7;
- } else if (c->state== CS_DIRTY) {
- state_str.s = "CS_DIRTY";
- state_str.len = 8;
- } else {
- state_str.s = "CS_UNKNOWN";
- state_str.len = 10;
- }
- if(c->sock)
- {
- socket_str.s = c->sock->sock_str.s;
- socket_str.len = c->sock->sock_str.len;
- }
- if(rpc->struct_add(vh, "f", "Q", q2double(c->q))<0)
- {
- rpc->fault(ctx, 500, "Internal error adding q");
- return -1;
- }
- if(rpc->struct_add(vh, "S", "Call-ID", &c->callid)<0)
- {
- rpc->fault(ctx, 500, "Internal error adding callid");
- return -1;
- }
- if(rpc->struct_add(vh, "d", "CSeq", c->cseq)<0)
- {
- rpc->fault(ctx, 500, "Internal error adding cseq");
- return -1;
- }
- if(rpc->struct_add(vh, "S", "User-Agent",
- (c->user_agent.len)?&c->user_agent: &empty_str)<0)
- {
- rpc->fault(ctx, 500, "Internal error adding user-agent");
- return -1;
- }
- if(rpc->struct_add(vh, "S", "Received",
- (c->received.len)?&c->received: &empty_str)<0)
- {
- rpc->fault(ctx, 500, "Internal error adding received");
- return -1;
- }
- if(rpc->struct_add(vh, "S", "Path",
- (c->path.len)?&c->path: &empty_str)<0)
- {
- rpc->fault(ctx, 500, "Internal error adding path");
- return -1;
- }
- if(rpc->struct_add(vh, "S", "State", &state_str)<0)
- {
- rpc->fault(ctx, 500, "Internal error adding state");
- return -1;
- }
- if(rpc->struct_add(vh, "d", "Flags", c->flags)<0)
- {
- rpc->fault(ctx, 500, "Internal error adding flags");
- return -1;
- }
- if(rpc->struct_add(vh, "d", "CFlags", c->cflags)<0)
- {
- rpc->fault(ctx, 500, "Internal error adding cflags");
- return -1;
- }
- if(rpc->struct_add(vh, "S", "Socket", &socket_str)<0)
- {
- rpc->fault(ctx, 500, "Internal error adding socket");
- return -1;
- }
- if(rpc->struct_add(vh, "d", "Methods", c->methods)<0)
- {
- rpc->fault(ctx, 500, "Internal error adding methods");
- return -1;
- }
- if(rpc->struct_add(vh, "S", "Ruid", (c->ruid.len)?&c->ruid: &empty_str)<0)
- {
- rpc->fault(ctx, 500, "Internal error adding ruid");
- return -1;
- }
- if(rpc->struct_add(vh, "S", "Instance",
- (c->instance.len)?&c->instance: &empty_str)<0)
- {
- rpc->fault(ctx, 500, "Internal error adding instance");
- return -1;
- }
- if(rpc->struct_add(vh, "d", "Reg-Id", c->reg_id)<0)
- {
- rpc->fault(ctx, 500, "Internal error adding reg_id");
- return -1;
- }
- if(rpc->struct_add(vh, "d", "Last-Keepalive", (int)c->last_keepalive)<0)
- {
- rpc->fault(ctx, 500, "Internal error adding reg_id");
- return -1;
- }
- if(rpc->struct_add(vh, "d", "Last-Modified", (int)c->last_modified)<0)
- {
- rpc->fault(ctx, 500, "Internal error adding reg_id");
- return -1;
- }
- return 0;
- }
- static void ul_rpc_dump(rpc_t* rpc, void* ctx)
- {
- struct urecord* r;
- dlist_t* dl;
- udomain_t* dom;
- str brief = {0, 0};
- int summary = 0;
- ucontact_t* c;
- void* th;
- void* ah;
- void* ih;
- void* sh;
- int max, n, i;
- rpc->scan(ctx, "*S", &brief);
- if(brief.len==5 && (strncmp(brief.s, "brief", 5)==0))
- summary = 1;
-
- for( dl=root ; dl ; dl=dl->next ) {
- dom = dl->d;
- if (rpc->add(ctx, "{", &th) < 0)
- {
- rpc->fault(ctx, 500, "Internal error creating top rpc");
- return;
- }
- if(rpc->struct_add(th, "Sd{",
- "Domain", &dl->name,
- "Size", (int)dom->size,
- "AoRs", &ah)<0)
- {
- rpc->fault(ctx, 500, "Internal error creating inner struct");
- return;
- }
- for(i=0,n=0,max=0; i<dom->size; i++) {
- lock_ulslot( dom, i);
- n += dom->table[i].n;
- if(max<dom->table[i].n)
- max= dom->table[i].n;
- for( r = dom->table[i].first ; r ; r=r->next ) {
- if(summary==1)
- {
- if(rpc->struct_add(ah, "S",
- "AoR", &r->aor)<0)
- {
- rpc->fault(ctx, 500, "Internal error creating aor struct");
- return;
- }
- } else {
- if(rpc->struct_add(ah, "Sd{",
- "AoR", &r->aor,
- "HashID", r->aorhash,
- "Contacts", &ih)<0)
- {
- unlock_ulslot( dom, i);
- rpc->fault(ctx, 500, "Internal error creating aor struct");
- return;
- }
- for( c=r->contacts ; c ; c=c->next)
- {
- if (rpc_dump_contact(rpc, ctx, ih, c) == -1) {
- unlock_ulslot(dom, i);
- return;
- }
- }
- }
- }
- unlock_ulslot( dom, i);
- }
- /* extra attributes node */
- if(rpc->struct_add(th, "{", "Stats", &sh)<0)
- {
- rpc->fault(ctx, 500, "Internal error creating stats struct");
- return;
- }
- if(rpc->struct_add(sh, "dd",
- "Records", n,
- "Max-Slots", max)<0)
- {
- rpc->fault(ctx, 500, "Internal error adding stats");
- return;
- }
- }
- }
- static const char* ul_rpc_lookup_doc[2] = {
- "Lookup one AOR in the usrloc location table",
- 0
- };
- /*!
- * \brief Search a domain in the global domain list
- * \param table domain (table) name
- * \return pointer to domain if found, 0 if not found
- */
- static inline udomain_t* rpc_find_domain(str* table)
- {
- dlist_t* dom;
- for( dom=root ; dom ; dom=dom->next ) {
- if ((dom->name.len == table->len) &&
- !memcmp(dom->name.s, table->s, table->len))
- return dom->d;
- }
- return 0;
- }
- /*!
- * \brief Convert address of record
- *
- * Convert an address of record string to lower case, and truncate
- * it when use_domain is not set.
- * \param aor address of record
- * \return 0 on success, -1 on error
- */
- static inline int rpc_fix_aor(str *aor)
- {
- char *p;
- p = memchr( aor->s, '@', aor->len);
- if (use_domain) {
- if (p==NULL)
- return -1;
- } else {
- if (p)
- aor->len = p - aor->s;
- }
- strlower(aor);
- return 0;
- }
- /*!
- * \brief Dumps the contacts of an AOR
- * \param rpc Handle to RPC structure
- * \param ctx not used
- * \note expects 2 arguments: the table name and the AOR
- */
- static void ul_rpc_lookup(rpc_t* rpc, void* ctx)
- {
- udomain_t* dom;
- str table = {0, 0};
- str aor = {0, 0};
- void* th;
- urecord_t *rec;
- ucontact_t* con;
- int ret;
- int rpl_tree;
- if (rpc->scan(ctx, "S", &table) != 1) {
- rpc->fault(ctx, 500, "Not enough parameters (table and AOR to lookup)");
- return;
- }
- if (rpc->scan(ctx, "S", &aor) != 1) {
- rpc->fault(ctx, 500, "Not enough parameters (table and AOR to lookup)");
- return;
- }
- /* look for table */
- dom = rpc_find_domain( &table );
- if (dom == NULL) {
- rpc->fault(ctx, 500, "Domain not found");
- return;
- }
- /* process the aor */
- if ( rpc_fix_aor(&aor) != 0 ) {
- rpc->fault(ctx, 500, "Domain missing in AOR");
- return;
- }
- lock_udomain( dom, &aor);
- ret = get_urecord( dom, &aor, &rec);
- if (ret == 1) {
- unlock_udomain( dom, &aor);
- rpc->fault(ctx, 500, "AOR not found in location table");
- return;
- }
- get_act_time();
- rpl_tree = 0;
- if (rpc->add(ctx, "{", &th) < 0)
- {
- rpc->fault(ctx, 500, "Internal error creating outer rpc");
- return;
- }
- /* We have contacts, list them */
- for( con=rec->contacts ; con ; con=con->next) {
- if (VALID_CONTACT( con, act_time)) {
- rpl_tree++;
- if (rpc_dump_contact(rpc, ctx, th, con) == -1) {
- unlock_udomain(dom, &aor);
- return;
- }
- }
- }
- unlock_udomain( dom, &aor);
- if (rpl_tree==0) {
- rpc->fault(ctx, 500, "AOR has no contacts");
- return;
- }
- return;
- }
- static void ul_rpc_rm_aor(rpc_t* rpc, void* ctx)
- {
- udomain_t* dom;
- str table = {0, 0};
- str aor = {0, 0};
- if (rpc->scan(ctx, "SS", &table, &aor) != 2) {
- rpc->fault(ctx, 500, "Not enough parameters (table and AOR to lookup)");
- return;
- }
- /* look for table */
- dom = rpc_find_domain( &table );
- if (dom == NULL) {
- rpc->fault(ctx, 500, "Domain not found");
- return;
- }
- /* process the aor */
- if ( rpc_fix_aor(&aor) != 0 ) {
- rpc->fault(ctx, 500, "Domain missing in AOR");
- return;
- }
- lock_udomain( dom, &aor);
- if (delete_urecord( dom, &aor, 0) < 0) {
- unlock_udomain( dom, &aor);
- rpc->fault(ctx, 500, "Failed to delete AOR");
- return;
- }
- unlock_udomain( dom, &aor);
- return;
- }
- static const char* ul_rpc_rm_aor_doc[2] = {
- "Delete a address of record including its contacts",
- 0
- };
- static void ul_rpc_rm_contact(rpc_t* rpc, void* ctx)
- {
- udomain_t* dom;
- str table = {0, 0};
- str aor = {0, 0};
- str contact = {0, 0};
- urecord_t *rec;
- ucontact_t* con;
- int ret;
- if (rpc->scan(ctx, "SSS", &table, &aor, &contact) != 3) {
- rpc->fault(ctx, 500, "Not enough parameters (table, AOR and contact)");
- return;
- }
- /* look for table */
- dom = rpc_find_domain( &table );
- if (dom == NULL) {
- rpc->fault(ctx, 500, "Domain not found");
- return;
- }
- /* process the aor */
- if ( rpc_fix_aor(&aor) != 0 ) {
- rpc->fault(ctx, 500, "Domain missing in AOR");
- return;
- }
- lock_udomain( dom, &aor);
- ret = get_urecord( dom, &aor, &rec);
- if (ret == 1) {
- unlock_udomain( dom, &aor);
- rpc->fault(ctx, 404, "AOR not found");
- return;
- }
- ret = get_ucontact( rec, &contact, &rpc_ul_cid, &rpc_ul_path, RPC_UL_CSEQ+1, &con);
- if (ret < 0) {
- unlock_udomain( dom, &aor);
- rpc->fault(ctx, 500, "Internal error (can't get contact)");
- return;
- }
- if (ret > 0) {
- unlock_udomain( dom, &aor);
- rpc->fault(ctx, 404, "Contact not found");
- return;
- }
- if (delete_ucontact(rec, con) < 0) {
- unlock_udomain( dom, &aor);
- rpc->fault(ctx, 500, "Internal error (can't delete contact)");
- return;
- }
- release_urecord(rec);
- unlock_udomain( dom, &aor);
- return;
- }
- static const char* ul_rpc_rm_contact_doc[2] = {
- "Delete a contact from an AOR record",
- 0
- };
- static void ul_rpc_flush(rpc_t* rpc, void* ctx)
- {
- synchronize_all_udomains(0, 1);
- return;
- }
- static const char* ul_rpc_flush_doc[2] = {
- "Flush the usrloc memory cache to DB",
- 0
- };
- /*!
- * \brief Add a new contact for an address of record
- * \note Expects 9 parameters: table name, AOR, contact, expires, Q,
- * path, flags, cflags, methods
- */
- static void ul_rpc_add(rpc_t* rpc, void* ctx)
- {
- str table = {0, 0};
- str aor = {0, 0};
- str contact = {0, 0};
- str path = {0, 0};
- str temp = {0, 0};
- double dtemp;
- ucontact_info_t ci;
- urecord_t* r;
- ucontact_t* c;
- udomain_t *dom;
- int ret;
- memset( &ci, 0, sizeof(ucontact_info_t));
- ret = rpc->scan(ctx, "SSSdfSddd", &table, &aor, &contact, &ci.expires,
- &dtemp, &path, &ci.flags, &ci.cflags, &ci.methods);
- if(path.len==1 && (strncmp(path.s, "0", 1)==0)) {
- LM_DBG("path == 0 -> unset\n");
- }
- else {
- ci.path = &path;
- }
- LM_DBG("ret: %d table:%.*s aor:%.*s contact:%.*s expires:%d dtemp:%f path:%.*s flags:%d bflags:%d methods:%d\n",
- ret, table.len, table.s, aor.len, aor.s, contact.len, contact.s,
- (int) ci.expires, dtemp, (ci.path)?ci.path->len:0, (ci.path && ci.path->s)?ci.path->s:"", ci.flags, ci.cflags, (int) ci.methods);
- if ( ret != 9) {
- rpc->fault(ctx, 500, "Not enough parameters or wrong format");
- return;
- }
- ci.q = double2q(dtemp);
- temp.s = q2str(ci.q, (unsigned int*)&temp.len);
- LM_DBG("q:%.*s\n", temp.len, temp.s);
- /* look for table */
- dom = rpc_find_domain( &table );
- if (dom == NULL) {
- rpc->fault(ctx, 500, "Domain not found");
- return;
- }
- /* process the aor */
- if ( rpc_fix_aor(&aor) != 0 ) {
- rpc->fault(ctx, 500, "Domain missing in AOR");
- return;
- }
- if(sruid_next(&_ul_sruid)<0)
- {
- rpc->fault(ctx, 500, "Can't obtain next uid");
- return;
- }
- ci.ruid = _ul_sruid.uid;
- lock_udomain( dom, &aor);
- ret = get_urecord( dom, &aor, &r);
- if(ret==1) {
- if (insert_urecord( dom, &aor, &r) < 0)
- {
- unlock_udomain( dom, &aor);
- rpc->fault(ctx, 500, "Can't insert record");
- return;
- }
- c = 0;
- } else {
- if (get_ucontact( r, &contact, &rpc_ul_cid, &rpc_ul_path, RPC_UL_CSEQ+1, &c) < 0)
- {
- unlock_udomain( dom, &aor);
- rpc->fault(ctx, 500, "Can't get record");
- return;
- }
- }
- get_act_time();
- ci.callid = &rpc_ul_cid;
- ci.user_agent = &rpc_ul_ua;
- ci.cseq = RPC_UL_CSEQ;
- /* 0 expires means permanent contact */
- if (ci.expires!=0)
- ci.expires += act_time;
- if (c) {
- if (update_ucontact( r, c, &ci) < 0)
- {
- release_urecord(r);
- unlock_udomain( dom, &aor);
- rpc->fault(ctx, 500, "Can't update contact");
- return;
- }
- } else {
- if ( insert_ucontact( r, &contact, &ci, &c) < 0 )
- {
- release_urecord(r);
- unlock_udomain( dom, &aor);
- rpc->fault(ctx, 500, "Can't insert contact");
- return;
- }
- }
- release_urecord(r);
- unlock_udomain( dom, &aor);
- return;
- }
- static const char* ul_rpc_add_doc[2] = {
- "Add a new contact for an address of record",
- 0
- };
- #define QUERY_LEN 256
- static void ul_rpc_db_users(rpc_t* rpc, void* ctx)
- {
- str table = {0, 0};
- char query[QUERY_LEN];
- str query_str;
- db1_res_t* res;
- int count;
- if (db_mode == NO_DB) {
- rpc->fault(ctx, 500, "Command is not supported in db_mode=0");
- return;
- }
- if (rpc->scan(ctx, "S", &table) != 1) {
- rpc->fault(ctx, 500, "Not enough parameters (table to lookup)");
- return;
- }
- if (user_col.len + domain_col.len + table.len + 32 > QUERY_LEN) {
- rpc->fault(ctx, 500, "Too long database query");
- return;
- }
- if (!DB_CAPABILITY(ul_dbf, DB_CAP_RAW_QUERY)) {
- rpc->fault(ctx, 500, "Database does not support raw queries");
- return;
- }
- if (ul_dbf.use_table(ul_dbh, &table) < 0) {
- rpc->fault(ctx, 500, "Failed to use table");
- return;
- }
-
- memset(query, 0, QUERY_LEN);
- query_str.len = snprintf(query, QUERY_LEN,
- "SELECT COUNT(DISTINCT %.*s, %.*s) FROM %.*s WHERE (UNIX_TIMESTAMP(expires) = 0) OR (expires > NOW())",
- user_col.len, user_col.s,
- domain_col.len, domain_col.s,
- table.len, table.s);
- query_str.s = query;
- if (ul_dbf.raw_query(ul_dbh, &query_str, &res) < 0) {
- rpc->fault(ctx, 500, "Failed to query AoR count");
- return;
- }
- count = (int)VAL_INT(ROW_VALUES(RES_ROWS(res)));
- ul_dbf.free_result(ul_dbh, res);
- rpc->add(ctx, "d", count);
- }
- static const char* ul_rpc_db_users_doc[2] = {
- "Tell number of different unexpired users (AoRs) in database table (db_mode!=0 only)",
- 0
- };
- static void ul_rpc_db_contacts(rpc_t* rpc, void* ctx)
- {
- str table = {0, 0};
- char query[QUERY_LEN];
- str query_str;
- db1_res_t* res;
- int count;
- if (db_mode == NO_DB) {
- rpc->fault(ctx, 500, "Command is not supported in db_mode=0");
- return;
- }
- if (rpc->scan(ctx, "S", &table) != 1) {
- rpc->fault(ctx, 500, "Not enough parameters (table to lookup)");
- return;
- }
- if (table.len + 22 > QUERY_LEN) {
- rpc->fault(ctx, 500, "Too long database query");
- return;
- }
- if (!DB_CAPABILITY(ul_dbf, DB_CAP_RAW_QUERY)) {
- rpc->fault(ctx, 500, "Database does not support raw queries");
- return;
- }
- if (ul_dbf.use_table(ul_dbh, &table) < 0) {
- rpc->fault(ctx, 500, "Failed to use table");
- return;
- }
-
- memset(query, 0, QUERY_LEN);
- query_str.len = snprintf(query, QUERY_LEN, "SELECT COUNT(*) FROM %.*s WHERE (UNIX_TIMESTAMP(expires) = 0) OR (expires > NOW())",
- table.len, table.s);
- query_str.s = query;
- if (ul_dbf.raw_query(ul_dbh, &query_str, &res) < 0) {
- rpc->fault(ctx, 500, "Failed to query contact count");
- return;
- }
- count = (int)VAL_INT(ROW_VALUES(RES_ROWS(res)));
- ul_dbf.free_result(ul_dbh, res);
- rpc->add(ctx, "d", count);
- }
- static const char* ul_rpc_db_contacts_doc[2] = {
- "Tell number of unexpired contacts in database table (db_mode=3 only)",
- 0
- };
- static void ul_rpc_db_expired_contacts(rpc_t* rpc, void* ctx)
- {
- str table = {0, 0};
- char query[QUERY_LEN];
- str query_str;
- db1_res_t* res;
- int count;
- if (db_mode == NO_DB) {
- rpc->fault(ctx, 500, "Command is not supported in db_mode=0");
- return;
- }
- if (rpc->scan(ctx, "S", &table) != 1) {
- rpc->fault(ctx, 500, "Not enough parameters (table to lookup)");
- return;
- }
- if (table.len + 22 > QUERY_LEN) {
- rpc->fault(ctx, 500, "Too long database query");
- return;
- }
- if (!DB_CAPABILITY(ul_dbf, DB_CAP_RAW_QUERY)) {
- rpc->fault(ctx, 500, "Database does not support raw queries");
- return;
- }
- if (ul_dbf.use_table(ul_dbh, &table) < 0) {
- rpc->fault(ctx, 500, "Failed to use table");
- return;
- }
-
- memset(query, 0, QUERY_LEN);
- query_str.len = snprintf(query, QUERY_LEN, "SELECT COUNT(*) FROM %.*s WHERE (UNIX_TIMESTAMP(expires) > 0) AND (expires <= NOW())",
- table.len, table.s);
- query_str.s = query;
- if (ul_dbf.raw_query(ul_dbh, &query_str, &res) < 0) {
- rpc->fault(ctx, 500, "Failed to query contact count");
- return;
- }
- count = (int)VAL_INT(ROW_VALUES(RES_ROWS(res)));
- ul_dbf.free_result(ul_dbh, res);
- rpc->add(ctx, "d", count);
- }
- static const char* ul_rpc_db_expired_contacts_doc[2] = {
- "Tell number of expired contacts in database table (db_mode=3 only)",
- 0
- };
- rpc_export_t ul_rpc[] = {
- {"ul.dump", ul_rpc_dump, ul_rpc_dump_doc, 0},
- {"ul.lookup", ul_rpc_lookup, ul_rpc_lookup_doc, 0},
- {"ul.rm", ul_rpc_rm_aor, ul_rpc_rm_aor_doc, 0},
- {"ul.rm_contact", ul_rpc_rm_contact, ul_rpc_rm_contact_doc, 0},
- {"ul.flush", ul_rpc_flush, ul_rpc_flush_doc, 0},
- {"ul.add", ul_rpc_add, ul_rpc_add_doc, 0},
- {"ul.db_users", ul_rpc_db_users, ul_rpc_db_users_doc, 0},
- {"ul.db_contacts", ul_rpc_db_contacts, ul_rpc_db_contacts_doc, 0},
- {"ul.db_expired_contacts", ul_rpc_db_expired_contacts, ul_rpc_db_expired_contacts_doc, 0},
- {0, 0, 0, 0}
- };
|