core: kemi - exported KSR.add_tcp_alias() and KSR.add_tcp_alias_via()

src/core/kemi.c

@@ -933,6 +933,60 @@ static int sr_kemi_core_set_advertised_port(sip_msg_t *msg, str *port)
 	return SR_KEMI_TRUE;
 }
 
+/**
+ *
+ */
+static int sr_kemi_core_add_tcp_alias(sip_msg_t *msg, int port)
+{
+	if(msg==NULL) {
+		LM_WARN("invalid msg parameter\n");
+		return SR_KEMI_FALSE;
+	}
+
+#ifdef USE_TCP
+	if ( msg->rcv.proto==PROTO_TCP
+#ifdef USE_TLS
+				|| msg->rcv.proto==PROTO_TLS
+#endif
+			) {
+		if (tcpconn_add_alias(msg->rcv.proto_reserved1, port,
+							msg->rcv.proto)!=0){
+			LM_ERR("adding tcp alias failed\n");
+			return SR_KEMI_FALSE;
+		}
+	}
+#endif
+
+	return SR_KEMI_TRUE;
+}
+
+/**
+ *
+ */
+static int sr_kemi_core_add_tcp_alias_via(sip_msg_t *msg)
+{
+	if(msg==NULL) {
+		LM_WARN("invalid msg parameter\n");
+		return SR_KEMI_FALSE;
+	}
+
+#ifdef USE_TCP
+	if ( msg->rcv.proto==PROTO_TCP
+#ifdef USE_TLS
+				|| msg->rcv.proto==PROTO_TLS
+#endif
+			) {
+		if (tcpconn_add_alias(msg->rcv.proto_reserved1, msg->via1->port,
+							msg->rcv.proto)!=0){
+			LM_ERR("adding tcp alias failed\n");
+			return SR_KEMI_FALSE;
+		}
+	}
+#endif
+	return SR_KEMI_TRUE;
+}
+
+
 /**
  *
  */
@@ -1132,6 +1186,16 @@ static sr_kemi_t _sr_kemi_core[] = {
 		{ SR_KEMIP_STR, SR_KEMIP_NONE, SR_KEMIP_NONE,
 			SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE }
 	},
+	{ str_init(""), str_init("add_tcp_alias"),
+		SR_KEMIP_INT, sr_kemi_core_add_tcp_alias,
+		{ SR_KEMIP_INT, SR_KEMIP_NONE, SR_KEMIP_NONE,
+			SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE }
+	},
+	{ str_init(""), str_init("add_tcp_alias_via"),
+		SR_KEMIP_INT, sr_kemi_core_add_tcp_alias_via,
+		{ SR_KEMIP_NONE, 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 } }
 };