From e15f8d34c5cdf83e0b3560873cd5909de99d3f97 Mon Sep 17 00:00:00 2001
From: Daniel-Constantin Mierla <miconda@gmail.com>
Date: Fri, 4 Aug 2017 15:06:09 +0200
Subject: [PATCH] websocket: new parameter event_callback

- allow setting the name of kemi function to be executed for event_route
---
 src/modules/websocket/websocket.c |  4 ++++
 src/modules/websocket/ws_conn.c   | 32 ++++++++++++++++++++++++++-----
 2 files changed, 31 insertions(+), 5 deletions(-)

diff --git a/src/modules/websocket/websocket.c b/src/modules/websocket/websocket.c
index ec2833d8398..b7c6e5e1fb4 100644
--- a/src/modules/websocket/websocket.c
+++ b/src/modules/websocket/websocket.c
@@ -70,6 +70,8 @@ static int ws_keepalive_processes = DEFAULT_KEEPALIVE_PROCESSES;
 
 int ws_verbose_list = 0;
 
+str ws_event_callback = STR_NULL;
+
 static cmd_export_t cmds[] = {
 	/* ws_frame.c */
 	{ "ws_close", (cmd_function)w_ws_close0,
@@ -105,6 +107,8 @@ static param_export_t params[] = {
 	{ "keepalive_processes",	INT_PARAM, &ws_keepalive_processes },
 
 	{ "verbose_list",		PARAM_INT, &ws_verbose_list },
+	{ "event_callback",		PARAM_STR, &ws_event_callback},
+
 	{ 0, 0, 0 }
 };
 
diff --git a/src/modules/websocket/ws_conn.c b/src/modules/websocket/ws_conn.c
index 3fd9207dcf5..9fedf33f7b0 100644
--- a/src/modules/websocket/ws_conn.c
+++ b/src/modules/websocket/ws_conn.c
@@ -29,6 +29,7 @@
 #include "../../core/tcp_conn.h"
 #include "../../core/fmsg.h"
 #include "../../core/counters.h"
+#include "../../core/kemi.h"
 #include "../../core/mem/mem.h"
 #include "ws_conn.h"
 #include "websocket.h"
@@ -37,6 +38,7 @@
 #define MAX_WS_CONNS_DUMP	50
 
 extern int ws_verbose_list;
+extern str ws_event_callback;
 
 ws_connection_t **wsconn_id_hash = NULL;
 #define wsconn_listadd	tcpconn_listadd
@@ -276,14 +278,25 @@ static void wsconn_run_route(ws_connection_t *wsc)
 	int rt, backup_rt;
 	struct run_act_ctx ctx;
 	sip_msg_t *fmsg;
+	sr_kemi_eng_t *keng = NULL;
+	str evrtname = str_init("websocket:closed");
 
 	LM_DBG("wsconn_run_route event_route[websocket:closed]\n");
 
-	rt = route_lookup(&event_rt, "websocket:closed");
+	rt = route_lookup(&event_rt, evrtname.s);
 	if (rt < 0 || event_rt.rlist[rt] == NULL)
 	{
-		LM_DBG("route does not exist");
-		return;
+		if(ws_event_callback.len<=0 || ws_event_callback.s==NULL) {
+			LM_DBG("event route does not exist");
+			return;
+		}
+		keng = sr_kemi_eng_get();
+		if(keng==NULL) {
+			LM_DBG("event route callback engine does not exist");
+			return;
+		} else {
+			rt = -1;
+		}
 	}
 
 	if (faked_msg_init() < 0)
@@ -297,9 +310,18 @@ static void wsconn_run_route(ws_connection_t *wsc)
 	fmsg->rcv = wsc->rcv;
 
 	backup_rt = get_route_type();
-	set_route_type(REQUEST_ROUTE);
+	set_route_type(EVENT_ROUTE);
 	init_run_actions_ctx(&ctx);
-	run_top_route(event_rt.rlist[rt], fmsg, 0);
+	if(rt<0) {
+		/* kemi script event route callback */
+		if(keng && keng->froute(fmsg, EVENT_ROUTE,
+					&ws_event_callback, &evrtname)<0) {
+			LM_ERR("error running event route kemi callback\n");
+		}
+	} else {
+		/* native cfg event route */
+		run_top_route(event_rt.rlist[rt], fmsg, 0);
+	}
 	set_route_type(backup_rt);
 }