func_frame_drop: New function

Adds function to selectively drop specified frames
in the TX or RX direction on a channel, including
control frames.

ASTERISK-29478

Change-Id: I8147c9d55d74e2e48861edba6b22f930920541ec
This commit is contained in:
Naveen Albert 2021-06-16 16:30:05 -04:00 committed by Friendly Automation
parent 835ab50724
commit 7383f74dfc
2 changed files with 296 additions and 0 deletions

View File

@ -0,0 +1,5 @@
Subject: func_framedrop
New function to selectively drop specified frames
in either direction on a channel.

291
funcs/func_frame_drop.c Normal file
View File

@ -0,0 +1,291 @@
/*
* Asterisk -- An open source telephony toolkit.
*
* Copyright (C) 2021, Naveen Albert
*
* Naveen Albert <asterisk@phreaknet.org>
*
* See http://www.asterisk.org for more information about
* the Asterisk project. Please do not directly contact
* any of the maintainers of this project for assistance;
* the project provides a web site, mailing lists and IRC
* channels for your use.
*
* This program is free software, distributed under the terms of
* the GNU General Public License Version 2. See the LICENSE file
* at the top of the source tree.
*/
/*! \file
*
* \brief Function that drops specified frames from channels
*
* \author Naveen Albert <asterisk@phreaknet.org>
*
* \ingroup functions
*/
/*** MODULEINFO
<support_level>extended</support_level>
***/
#include "asterisk.h"
#include "asterisk/module.h"
#include "asterisk/channel.h"
#include "asterisk/pbx.h"
#include "asterisk/framehook.h"
/*** DOCUMENTATION
<function name="FRAME_DROP" language="en_US">
<synopsis>
Drops specific frame types in the TX or RX direction on a channel.
</synopsis>
<syntax>
<parameter name="direction" required="true">
<para>List of frame types to be dropped for the specified direction. Direction can be <literal>TX</literal> or <literal>RX</literal>. The <literal>TX</literal> direction will prevent Asterisk from sending frames to a channel, and the <literal>RX</literal> direction will prevent Asterisk from receiving frames from a channel.</para>
<para>Subsequent calls to this function will replace previous settings, allowing certain frames to be dropped only temporarily, for instance.</para>
<para>Below are the different types of frames that can be dropped. Other actions may need to be taken in conjunction with use of this function:
for instance, if you drop ANSWER control frames, you should explicitly use <literal>Progress()</literal> for your call or undesired behavior
may occur.</para>
<enumlist>
<enum name = "DTMF_BEGIN" />
<enum name = "DTMF_END" />
<enum name = "VOICE" />
<enum name = "VIDEO" />
<enum name = "CONTROL" />
<enum name = "NULL" />
<enum name = "IAX" />
<enum name = "TEXT" />
<enum name = "TEXT_DATA" />
<enum name = "IMAGE" />
<enum name = "HTML" />
<enum name = "CNG" />
<enum name = "MODEM" />
</enumlist>
<para>The following CONTROL frames can also be dropped:</para>
<enumlist>
<enum name = "RING" />
<enum name = "RINGING" />
<enum name = "ANSWER" />
<enum name = "BUSY" />
<enum name = "TAKEOFFHOOK" />
<enum name = "OFFHOOK" />
<enum name = "CONGESTION" />
<enum name = "FLASH" />
<enum name = "WINK" />
<enum name = "PROGRESS" />
<enum name = "PROCEEDING" />
<enum name = "HOLD" />
<enum name = "UNHOLD" />
<enum name = "VIDUPDATE" />
<enum name = "CONNECTED_LINE" />
<enum name = "REDIRECTING" />
</enumlist>
</parameter>
</syntax>
<description>
<para>Examples:</para>
<para>exten => 1,1,Set(FRAME_DROP(TX)=DTMF_BEGIN,DTMF_END); drop only DTMF frames towards this channel.</para>
<para>exten => 1,1,Set(FRAME_DROP(TX)=ANSWER); drop only ANSWER CONTROL frames towards this channel.</para>
<para>exten => 1,1,Set(FRAME_DROP(RX)=DTMF_BEGIN,DTMF_END); drop only DTMF frames received on this channel.</para>
</description>
</function>
***/
static struct {
enum ast_frame_type type;
const char *str;
} frametype2str[] = {
{ AST_FRAME_DTMF_BEGIN, ",DTMF_BEGIN," },
{ AST_FRAME_DTMF_END, ",DTMF_END," },
{ AST_FRAME_VOICE, ",VOICE," },
{ AST_FRAME_VIDEO, ",VIDEO," },
{ AST_FRAME_CONTROL, ",CONTROL," },
{ AST_FRAME_NULL, ",NULL," },
{ AST_FRAME_IAX, ",IAX," },
{ AST_FRAME_TEXT, ",TEXT," },
{ AST_FRAME_TEXT_DATA, ",TEXT_DATA," },
{ AST_FRAME_IMAGE, ",IMAGE," },
{ AST_FRAME_HTML, ",HTML," },
{ AST_FRAME_CNG, ",CNG," },
{ AST_FRAME_MODEM, ",MODEM," },
};
static struct {
int type;
const char *str;
} controlframetype2str[] = {
{ AST_CONTROL_RING, ",RING," },
{ AST_CONTROL_RINGING, ",RINGING," },
{ AST_CONTROL_ANSWER, ",ANSWER," },
{ AST_CONTROL_BUSY, ",BUSY," },
{ AST_CONTROL_TAKEOFFHOOK, ",TAKEOFFHOOK," },
{ AST_CONTROL_OFFHOOK, ",OFFHOOK," },
{ AST_CONTROL_CONGESTION, ",CONGESTION," },
{ AST_CONTROL_FLASH, ",FLASH," },
{ AST_CONTROL_WINK, ",WINK," },
{ AST_CONTROL_PROGRESS, ",PROGRESS," },
{ AST_CONTROL_PROCEEDING, ",PROCEEDING," },
{ AST_CONTROL_HOLD, ",HOLD," },
{ AST_CONTROL_UNHOLD, ",UNHOLD," },
{ AST_CONTROL_VIDUPDATE, ",VIDUPDATE," },
{ AST_CONTROL_CONNECTED_LINE, ",CONNECTED_LINE," },
{ AST_CONTROL_REDIRECTING, ",REDIRECTING," },
};
enum direction {
TX = 0,
RX,
};
struct frame_drop_data {
enum direction list_type;
int values[ARRAY_LEN(frametype2str)];
int controlvalues[ARRAY_LEN(controlframetype2str)];
};
static void datastore_destroy_cb(void *data) {
ast_free(data);
}
static const struct ast_datastore_info frame_drop_datastore = {
.type = "framedrop",
.destroy = datastore_destroy_cb
};
static void hook_destroy_cb(void *framedata)
{
ast_free(framedata);
}
static struct ast_frame *hook_event_cb(struct ast_channel *chan, struct ast_frame *frame, enum ast_framehook_event event, void *data)
{
int i;
int drop_frame = 0;
struct frame_drop_data *framedata = data;
if (!frame) {
return frame;
}
if (!((event == AST_FRAMEHOOK_EVENT_WRITE && framedata->list_type == TX) ||
(event == AST_FRAMEHOOK_EVENT_READ && framedata->list_type == RX))) {
return frame;
}
if (frame->frametype == AST_FRAME_CONTROL) {
for (i = 0; i < ARRAY_LEN(controlframetype2str); i++) {
if (frame->subclass.integer == controlframetype2str[i].type) {
if (framedata->controlvalues[i]) {
drop_frame = 1;
}
break;
}
}
} else {
for (i = 0; i < ARRAY_LEN(frametype2str); i++) {
if (frame->frametype == frametype2str[i].type) {
if (framedata->values[i]) {
drop_frame = 1;
}
break;
}
}
}
if (drop_frame) {
ast_frfree(frame);
frame = &ast_null_frame;
}
return frame;
}
static int frame_drop_helper(struct ast_channel *chan, const char *cmd, char *data, const char *value)
{
char *buffer;
struct frame_drop_data *framedata;
struct ast_datastore *datastore = NULL;
struct ast_framehook_interface interface = {
.version = AST_FRAMEHOOK_INTERFACE_VERSION,
.event_cb = hook_event_cb,
.destroy_cb = hook_destroy_cb,
};
int i = 0;
if (!(framedata = ast_calloc(1, sizeof(*framedata)))) {
return 0;
}
interface.data = framedata;
if (!strcasecmp(data, "RX")) {
framedata->list_type = RX;
} else {
framedata->list_type = TX;
}
buffer = ast_malloc(sizeof(value) + 3); /* leading and trailing comma and null terminator */
snprintf(buffer, sizeof(value) + 2, ",%s,", value);
for (i = 0; i < ARRAY_LEN(frametype2str); i++) {
if (strcasestr(value, frametype2str[i].str)) {
framedata->values[i] = 1;
}
}
for (i = 0; i < ARRAY_LEN(controlframetype2str); i++) {
if (strcasestr(value, controlframetype2str[i].str)) {
framedata->controlvalues[i] = 1;
}
}
ast_free(buffer);
ast_channel_lock(chan);
i = ast_framehook_attach(chan, &interface);
if (i >= 0) {
int *id;
if ((datastore = ast_channel_datastore_find(chan, &frame_drop_datastore, NULL))) {
id = datastore->data;
ast_framehook_detach(chan, *id);
ast_channel_datastore_remove(chan, datastore);
ast_datastore_free(datastore);
}
if (!(datastore = ast_datastore_alloc(&frame_drop_datastore, NULL))) {
ast_framehook_detach(chan, i);
ast_channel_unlock(chan);
return 0;
}
if (!(id = ast_calloc(1, sizeof(int)))) {
ast_datastore_free(datastore);
ast_framehook_detach(chan, i);
ast_channel_unlock(chan);
return 0;
}
*id = i; /* Store off the id. The channel is still locked so it is safe to access this ptr. */
datastore->data = id;
ast_channel_datastore_add(chan, datastore);
}
ast_channel_unlock(chan);
return 0;
}
static struct ast_custom_function frame_drop_function = {
.name = "FRAME_DROP",
.write = frame_drop_helper,
};
static int unload_module(void)
{
return ast_custom_function_unregister(&frame_drop_function);
}
static int load_module(void)
{
int res = ast_custom_function_register(&frame_drop_function);
return res ? AST_MODULE_LOAD_DECLINE : AST_MODULE_LOAD_SUCCESS;
}
AST_MODULE_INFO_STANDARD_EXTENDED(ASTERISK_GPL_KEY, "Function to drop frames on a channel.");