فهرست منبع

corex: new function via_add_xavp_params(flags)

- control the flag for adding xavp fields to via params
Daniel-Constantin Mierla 7 سال پیش
والد
کامیت
4655167442
1فایلهای تغییر یافته به همراه36 افزوده شده و 1 حذف شده
  1. 36 1
      src/modules/corex/corex_mod.c

+ 36 - 1
src/modules/corex/corex_mod.c

@@ -55,6 +55,7 @@ static int w_set_send_socket(sip_msg_t *msg, char *psock, char *p2);
 static int w_set_recv_socket(sip_msg_t *msg, char *psock, char *p2);
 static int w_set_source_address(sip_msg_t *msg, char *paddr, char *p2);
 static int w_via_add_srvid(sip_msg_t *msg, char *pflags, char *p2);
+static int w_via_add_xavp_params(sip_msg_t *msg, char *pflags, char *p2);
 
 static int fixup_file_op(void** param, int param_no);
 
@@ -114,6 +115,8 @@ static cmd_export_t cmds[]={
 		0, ANY_ROUTE },
 	{"via_add_srvid", (cmd_function)w_via_add_srvid, 1, fixup_igp_null,
 		0, ANY_ROUTE },
+	{"via_add_xavp_params", (cmd_function)w_via_add_xavp_params, 1, fixup_igp_null,
+		0, ANY_ROUTE },
 
 	{0, 0, 0, 0, 0, 0}
 };
@@ -718,6 +721,34 @@ static int w_via_add_srvid(sip_msg_t *msg, char *pflags, char *s2)
 	return ki_via_add_srvid(msg, fval);
 }
 
+/**
+ *
+ */
+static int ki_via_add_xavp_params(sip_msg_t *msg, int fval)
+{
+	if(msg==NULL)
+		return -1;
+	if(fval) {
+		msg->msg_flags |= FL_ADD_XAVP_VIA;
+	} else {
+		msg->msg_flags &= ~(FL_ADD_XAVP_VIA);
+	}
+	return 1;
+}
+
+/**
+ *
+ */
+static int w_via_add_xavp_params(sip_msg_t *msg, char *pflags, char *s2)
+{
+	int fval=0;
+	if(fixup_get_ivalue(msg, (gparam_t*)pflags, &fval)!=0) {
+		LM_ERR("no flag value\n");
+		return -1;
+	}
+	return ki_via_add_xavp_params(msg, fval);
+}
+
 /**
  *
  */
@@ -773,7 +804,11 @@ static sr_kemi_t sr_kemi_corex_exports[] = {
 		{ SR_KEMIP_INT, SR_KEMIP_NONE, SR_KEMIP_NONE,
 			SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE }
 	},
-
+	{ str_init("corex"), str_init("via_add_xavp_params"),
+		SR_KEMIP_INT, ki_via_add_xavp_params,
+		{ SR_KEMIP_INT, SR_KEMIP_NONE, SR_KEMIP_NONE,
+			SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE }
+	},
 	{ {0, 0}, {0, 0}, 0, NULL, { 0, 0, 0, 0, 0, 0 } }
 };
 /* clang-format on */