Module: kamailio
Branch: master
Commit: d00be11be0e5fceec5274bf3bfd02951c3752c90
URL: 
https://github.com/kamailio/kamailio/commit/d00be11be0e5fceec5274bf3bfd02951c3752c90

Author: iionita <[email protected]>
Committer: Henning Westerholt <[email protected]>
Date: 2019-04-09T21:25:53+02:00

siptrace: Add code to trace incoming CANCEL transcations

---

Modified: src/modules/siptrace/siptrace.c

---

Diff:  
https://github.com/kamailio/kamailio/commit/d00be11be0e5fceec5274bf3bfd02951c3752c90.diff
Patch: 
https://github.com/kamailio/kamailio/commit/d00be11be0e5fceec5274bf3bfd02951c3752c90.patch

---

diff --git a/src/modules/siptrace/siptrace.c b/src/modules/siptrace/siptrace.c
index ae1f53d7f3..48fa9b97b4 100644
--- a/src/modules/siptrace/siptrace.c
+++ b/src/modules/siptrace/siptrace.c
@@ -1166,11 +1166,23 @@ static void trace_onreq_out(struct cell *t, int type, 
struct tmcb_params *ps)
                        return;
                }
        }
+
+       /* for incoming cancel this is the only play(i've found) where I have 
the CANCEL transaction
+        * and can register a callback for the reply */
        memset(&sto, 0, sizeof(siptrace_data_t));
 
-       if(traced_user_avp.n != 0)
-               sto.avp = search_first_avp(traced_user_avp_type, 
traced_user_avp,
-                               &sto.avp_value, &sto.state);
+       if (unlikely(type == TMCB_E2ECANCEL_IN)) {
+               msg->msg_flags |= FL_SIPTRACE;
+               if(tmb.register_tmcb(msg, 0, TMCB_RESPONSE_READY, 
trace_onreply_out, *ps->param, 0)
+                                       <= 0) {
+                       LM_ERR("can't register trace_onreply_out\n");
+                       return;
+               }
+       } else {
+               if(traced_user_avp.n != 0)
+                       sto.avp = search_first_avp(traced_user_avp_type, 
traced_user_avp,
+                                       &sto.avp_value, &sto.state);
+       }
 
        if((sto.avp == NULL) && trace_is_off(msg)) {
                LM_DBG("trace off...\n");
@@ -1180,11 +1192,16 @@ static void trace_onreq_out(struct cell *t, int type, 
struct tmcb_params *ps)
        if(sip_trace_prepare(msg) < 0)
                return;
 
-       if(ps->send_buf.len > 0) {
-               sto.body = ps->send_buf;
+       if (unlikely(type == TMCB_E2ECANCEL_IN)) {
+               sto.body.s = msg->buf;
+               sto.body.len = msg->len;
        } else {
-               sto.body.s = "No request buffer";
-               sto.body.len = sizeof("No request buffer") - 1;
+               if(ps->send_buf.len > 0) {
+                       sto.body = ps->send_buf;
+               } else {
+                       sto.body.s = "No request buffer";
+                       sto.body.len = sizeof("No request buffer") - 1;
+               }
        }
 
        sto.callid = msg->callid->body;
@@ -1245,7 +1262,16 @@ static void trace_onreq_out(struct cell *t, int type, 
struct tmcb_params *ps)
                }
        }
 
-       sto.dir = "out";
+       /* FIXME the callback is designed for outgoing requests but this along 
with
+        * the callback registration at the begining of the function it's for a 
special
+        * case - incoming CANCEL transactions; they were not traced before; 
TMCB_E2ECANCEL_IN
+        * will throw the incoming request through this function and the 
callback in the beginning
+        * will make sure the reply for this cancel is caught */
+       if (unlikely(type == TMCB_E2ECANCEL_IN)) {
+               sto.dir = "in";
+       } else {
+               sto.dir = "out";
+       }
 
        sto.fromtag = get_from(msg)->tag_value;
        sto.totag = get_to(msg)->tag_value;
@@ -1361,13 +1387,18 @@ static void trace_onreply_out(struct cell *t, int type, 
struct tmcb_params *ps)
                return;
        }
        memset(&sto, 0, sizeof(siptrace_data_t));
-       if(traced_user_avp.n != 0)
-               sto.avp = search_first_avp(traced_user_avp_type, 
traced_user_avp,
-                               &sto.avp_value, &sto.state);
 
-       if((sto.avp == NULL) && trace_is_off(t->uas.request)) {
-               LM_DBG("trace off...\n");
-               return;
+       /* can't(don't know) set FL_SIPTRACE flag from trace_onreq_out because
+        * there no access to CANCEL transaction there */
+       if (likely(type != TMCB_RESPONSE_READY)) {
+               if(traced_user_avp.n != 0)
+                       sto.avp = search_first_avp(traced_user_avp_type, 
traced_user_avp,
+                                       &sto.avp_value, &sto.state);
+
+               if((sto.avp == NULL) && trace_is_off(t->uas.request)) {
+                       LM_DBG("trace off...\n");
+                       return;
+               }
        }
 
        req = ps->req;
@@ -1463,7 +1494,6 @@ static void trace_onreply_out(struct cell *t, int type, 
struct tmcb_params *ps)
 static void trace_tm_neg_ack_in(struct cell *t, int type, struct tmcb_params 
*ps)
 {
        LM_DBG("storing negative ack...\n");
-
        /* this condition should not exist but there seems to be a BUG in 
kamailio
         * letting requests other than the ACK inside */
        if (ps->req->first_line.u.request.method_value != METHOD_ACK) {
@@ -1623,6 +1653,11 @@ static void trace_transaction(sip_msg_t* msg, 
siptrace_info_t* info, int registe
                LM_ERR("can't register trace_onreply_in\n");
                return;
        }
+
+       if(tmb.register_tmcb(msg, 0, TMCB_E2ECANCEL_IN, trace_onreq_out, info, 
0) <= 0) {
+               LM_ERR("can't register trace_onreply_in\n");
+               return;
+       }
 }
 
 //static void trace_dialog(sip_msg_t* msg, siptrace_info_t* info)


_______________________________________________
Kamailio (SER) - Development Mailing List
[email protected]
https://lists.kamailio.org/cgi-bin/mailman/listinfo/sr-dev

Reply via email to