######## NAT Traversal module - signaling functions ########
#!ifndef NATHELPER_LOADED
loadmodule "nathelper.so"
#!trydef NATHELPER_LOADED
#!endif
modparam("nathelper", "received_avp", "$avp(AVP_RECV_PARAM)")
modparam("nathelper", "sipping_from", "sip:sipcheck@MY_HOSTNAME")

#!ifdef WEBSOCKETS_ROLE
#!trydef KZ_NAT_DETECT 83
#!else
#!trydef KZ_NAT_DETECT 19
#!endif

#!trydef KZ_NAT_FIX_SDP_REQUEST 1
#!trydef KZ_NAT_FIX_SDP_REPLY 0
#!trydef KZ_NAT_SDP_TEST 8
#!trydef KZ_NAT_SDP_FIX 10


kazoo.nat_fix_sdp_request = KZ_NAT_FIX_SDP_REQUEST descr "performs request sdp replacement of private addresses"
kazoo.nat_fix_sdp_reply = KZ_NAT_FIX_SDP_REPLY descr "performs reply sdp replacement of private addresses"

####### NAT Traversal Logic ########
route[NAT_DETECT]
{
   if ($Rp == "5080") {
       setflag(FLAG_SKIP_NAT_CORRECTION);
       xlog("L_DEBUG", "$ci|log|skipping nat correction on PORT 5080\n");
   } else {
       if (is_present_hf("Record-Route")) {
           $var(i) = 0;
           $var(rr_count) = $rr_count;
           while($var(i) < $var(rr_count)) {
               $var(rr_base) = $(hdr(Record-Route)[$var(i)]);
               $var(rr_idx) = 0;
               $var(rr) = $(var(rr_base){s.select,$var(rr_idx),,});
               while($var(rr) != $null && $var(rr) != "") {
                   $var(i) = $var(i) + 1;
                   if (!is_myself("$(var(rr){nameaddr.uri})")) {
                       setflag(FLAG_SKIP_NAT_CORRECTION);
                       xlog("L_DEBUG", "$ci|log|skipping nat correction on record-route $(var(rr){nameaddr.uri})\n");
                   }
                   $var(rr_idx) = $var(rr_idx) + 1;
                   $var(rr) = $(var(rr_base){s.select,$var(rr_idx),,});
               }
           }
       }
   }

   if (isflagset(FLAG_SKIP_NAT_CORRECTION)) {
       return();
   }

   force_rport();

   if(nat_uac_test(KZ_NAT_DETECT)) {
      setflag(FLT_NATS);
      if (!is_method("REGISTER")) {
          if(is_first_hop()) set_contact_alias();
      }
   }

   if($sel(cfg_get.kazoo.nat_fix_sdp_request) == 1) {
       if (has_body("application/sdp") && nat_uac_test(KZ_NAT_SDP_TEST)) {
           fix_nated_sdp(KZ_NAT_SDP_FIX);
       }
   }

}

route[NAT_MANAGE]
{
   if ( is_request() && isflagset(FLAG_INTERNALLY_SOURCED)) {
        setbflag(FLB_NATB);
   }

   if ( is_request() && has_totag() ) {
        setbflag(FLB_NATB);
   }

   if (!(isflagset(FLT_NATS) || isbflagset(FLB_NATB))) {
        return;
   }

   if (is_reply()) {
       if(isflagset(FLT_NATS) || isbflagset(FLB_NATB)) {
           if(is_first_hop()) {
               set_contact_alias();
           }
       }
   }

   if($sel(cfg_get.kazoo.nat_fix_sdp_reply) == 1) {
       if (has_body("application/sdp") && nat_uac_test(KZ_NAT_SDP_TEST)) {
           fix_nated_sdp(KZ_NAT_SDP_FIX);
       }
   }

}

# vim: tabstop=4 softtabstop=4 shiftwidth=4 expandtab