[mod_verto] Limit websocket payload size to 1000000.

diff --git a/src/mod/endpoints/mod_verto/mod_verto.c b/src/mod/endpoints/mod_verto/mod_verto.c
index 2245c33a5641e8a1350653acc3478043e9bfb5ae..d98e1abdef5ab298d5d58da9e81f7d69ed307a98 100644 (file)
--- a/src/mod/endpoints/mod_verto/mod_verto.c
+++ b/src/mod/endpoints/mod_verto/mod_verto.c
@@ -1970,6 +1970,7 @@ static void client_run(jsock_t *jsock)
 {
        int flags = KWS_BLOCK;
        int idle = 0;
+       ks_json_t *params = NULL;
        
        if (jsock->profile->vhosts) {
                flags |= KWS_STAY_OPEN;
@@ -1977,7 +1978,14 @@ static void client_run(jsock_t *jsock)
        }
 
        ks_pool_open(&jsock->kpool);
+
+#if defined(KS_VERSION_NUM) && KS_VERSION_NUM >= 20000
+       params = ks_json_create_object();
+       ks_json_add_number_to_object(params, "payload_size_max", 1000000);
+       if (kws_init_ex(&jsock->ws, jsock->client_socket, (jsock->ptype & PTYPE_CLIENT_SSL) ? jsock->profile->ssl_ctx : NULL, 0, flags, jsock->kpool, params) != KS_STATUS_SUCCESS) {
+#else
        if (kws_init(&jsock->ws, jsock->client_socket, (jsock->ptype & PTYPE_CLIENT_SSL) ? jsock->profile->ssl_ctx : NULL, 0, flags, jsock->kpool) != KS_STATUS_SUCCESS) {
+#endif
                log_and_exit(SWITCH_LOG_NOTICE, "%s WS SETUP FAILED\n", jsock->name);
        }
 
@@ -2123,6 +2131,8 @@ static void client_run(jsock_t *jsock)
        detach_jsock(jsock);
        kws_destroy(&jsock->ws);
        ks_pool_close(&jsock->kpool);
+       ks_json_delete(&params);
+
        return;
 }
 
@@ -6769,6 +6779,10 @@ SWITCH_MODULE_LOAD_FUNCTION(mod_verto_load)
        ks_ssl_init_skip(KS_TRUE);
        ks_init();
 
+#if defined(KS_VERSION_NUM) && KS_VERSION_NUM < 20000
+       kws_set_global_payload_size_max(1000000);
+#endif
+
        if (switch_event_reserve_subclass(MY_EVENT_LOGIN) != SWITCH_STATUS_SUCCESS) {
                ks_shutdown();
                switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Couldn't register subclass %s!\n", MY_EVENT_LOGIN);