mirror of
https://github.com/signalwire/freeswitch.git
synced 2025-04-15 16:39:14 +00:00
469 lines
9.3 KiB
C++
469 lines
9.3 KiB
C++
|
|
#include <switch.h>
|
|
#include "freeswitch_lua.h"
|
|
using namespace LUA;
|
|
|
|
extern "C" {
|
|
int docall(lua_State * L, int narg, int nresults, int perror, int fatal);
|
|
};
|
|
|
|
Session::Session():CoreSession()
|
|
{
|
|
cb_function = cb_arg = hangup_func_str = hangup_func_arg = NULL;
|
|
hh = mark = 0;
|
|
}
|
|
|
|
Session::Session(char *nuuid, CoreSession *a_leg):CoreSession(nuuid, a_leg)
|
|
{
|
|
cb_function = cb_arg = hangup_func_str = hangup_func_arg = NULL;
|
|
hh = mark = 0;
|
|
}
|
|
|
|
Session::Session(switch_core_session_t *new_session):CoreSession(new_session)
|
|
{
|
|
cb_function = cb_arg = hangup_func_str = hangup_func_arg = NULL;
|
|
hh = mark = 0;
|
|
}
|
|
static switch_status_t lua_hanguphook(switch_core_session_t *session_hungup);
|
|
|
|
|
|
void Session::destroy(const char *err)
|
|
{
|
|
|
|
if (!allocated) {
|
|
return;
|
|
}
|
|
|
|
if (session) {
|
|
if (!channel) {
|
|
channel = switch_core_session_get_channel(session);
|
|
}
|
|
switch_channel_set_private(channel, "CoreSession", NULL);
|
|
switch_core_event_hook_remove_state_change(session, lua_hanguphook);
|
|
}
|
|
|
|
switch_safe_free(hangup_func_str);
|
|
switch_safe_free(hangup_func_arg);
|
|
switch_safe_free(cb_function);
|
|
switch_safe_free(cb_arg);
|
|
|
|
CoreSession::destroy();
|
|
|
|
|
|
if (!zstr(err)) {
|
|
lua_pushstring(L, err);
|
|
lua_error(L);
|
|
}
|
|
|
|
}
|
|
|
|
|
|
Session::~Session()
|
|
{
|
|
destroy();
|
|
}
|
|
|
|
bool Session::begin_allow_threads()
|
|
{
|
|
do_hangup_hook();
|
|
return true;
|
|
}
|
|
|
|
bool Session::end_allow_threads()
|
|
{
|
|
do_hangup_hook();
|
|
return true;
|
|
}
|
|
|
|
void Session::setLUA(lua_State * state)
|
|
{
|
|
L = state;
|
|
|
|
if (session && allocated && uuid) {
|
|
lua_setglobal(L, uuid);
|
|
lua_getglobal(L, uuid);
|
|
}
|
|
|
|
}
|
|
|
|
int Session::originate(CoreSession *a_leg_session, char *dest, int timeout)
|
|
{
|
|
int x = CoreSession::originate(a_leg_session, dest, timeout);
|
|
|
|
if (x) {
|
|
setLUA(L);
|
|
}
|
|
|
|
return x;
|
|
}
|
|
|
|
lua_State *Session::getLUA()
|
|
{
|
|
if (!L) {
|
|
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Doh!\n");
|
|
}
|
|
return L;
|
|
}
|
|
|
|
|
|
bool Session::ready()
|
|
{
|
|
bool r;
|
|
|
|
if (!session) {
|
|
return false;
|
|
}
|
|
sanity_check(false);
|
|
r = switch_channel_ready(channel) != 0;
|
|
do_hangup_hook();
|
|
|
|
return r;
|
|
}
|
|
|
|
void Session::check_hangup_hook()
|
|
{
|
|
if (hangup_func_str && (hook_state == CS_HANGUP || hook_state == CS_ROUTING)) {
|
|
hh++;
|
|
}
|
|
}
|
|
|
|
void Session::do_hangup_hook()
|
|
{
|
|
if (hh && !mark) {
|
|
int arg_count = 2;
|
|
mark++;
|
|
|
|
if (!getLUA()) {
|
|
return;
|
|
}
|
|
|
|
lua_getglobal(L, (char *) hangup_func_str);
|
|
lua_getglobal(L, uuid);
|
|
|
|
lua_pushstring(L, hook_state == CS_HANGUP ? "hangup" : "transfer");
|
|
|
|
if (hangup_func_arg) {
|
|
lua_getglobal(L, (char *) hangup_func_arg);
|
|
arg_count++;
|
|
}
|
|
|
|
docall(L, arg_count, 1, 1, 0);
|
|
|
|
const char *err = lua_tostring(L, -1);
|
|
|
|
switch_channel_set_variable(channel, "lua_hangup_hook_return_val", err);
|
|
|
|
if (!zstr(err)) {
|
|
|
|
if (!strcasecmp(err, "exit") || !strcasecmp(err, "die")) {
|
|
lua_error(L);
|
|
} else {
|
|
lua_pop(L, 1);
|
|
}
|
|
} else {
|
|
lua_pop(L, 1);
|
|
}
|
|
|
|
|
|
if (channel) {
|
|
switch_channel_set_private(channel, "CoreSession", NULL);
|
|
}
|
|
|
|
if (session) {
|
|
switch_core_event_hook_remove_state_change(session, lua_hanguphook);
|
|
}
|
|
switch_safe_free(hangup_func_str);
|
|
|
|
}
|
|
}
|
|
|
|
static switch_status_t lua_hanguphook(switch_core_session_t *session_hungup)
|
|
{
|
|
switch_channel_t *channel = switch_core_session_get_channel(session_hungup);
|
|
Session *coresession = NULL;
|
|
switch_channel_state_t state = switch_channel_get_state(channel);
|
|
|
|
if (session_hungup) {
|
|
|
|
channel = switch_core_session_get_channel(session_hungup);
|
|
|
|
if (channel) {
|
|
void *vs = switch_channel_get_private(channel, "CoreSession");
|
|
if (vs) {
|
|
coresession = (Session *) vs;
|
|
}
|
|
}
|
|
|
|
if (!(coresession && coresession->hook_state)) {
|
|
return SWITCH_STATUS_FALSE;
|
|
}
|
|
|
|
if (coresession && coresession->allocated && (state == CS_HANGUP || state == CS_ROUTING) && coresession->hook_state != state) {
|
|
coresession->hook_state = state;
|
|
coresession->check_hangup_hook();
|
|
switch_core_event_hook_remove_state_change(session_hungup, lua_hanguphook);
|
|
}
|
|
}
|
|
|
|
return SWITCH_STATUS_SUCCESS;
|
|
}
|
|
|
|
|
|
void Session::setHangupHook(char *func, char *arg)
|
|
{
|
|
|
|
sanity_check_noreturn;
|
|
|
|
switch_safe_free(hangup_func_str);
|
|
switch_safe_free(hangup_func_arg);
|
|
|
|
if (func) {
|
|
hangup_func_str = strdup(func);
|
|
if (!zstr(arg)) {
|
|
hangup_func_arg = strdup(arg);
|
|
}
|
|
switch_channel_set_private(channel, "CoreSession", this);
|
|
hook_state = switch_channel_get_state(channel);
|
|
switch_core_event_hook_add_state_change(session, lua_hanguphook);
|
|
}
|
|
}
|
|
|
|
void Session::unsetInputCallback(void)
|
|
{
|
|
sanity_check_noreturn;
|
|
switch_safe_free(cb_function);
|
|
switch_safe_free(cb_arg);
|
|
args.input_callback = NULL;
|
|
ap = NULL;
|
|
}
|
|
|
|
void Session::setInputCallback(char *cbfunc, char *funcargs)
|
|
{
|
|
|
|
sanity_check_noreturn;
|
|
|
|
switch_safe_free(cb_function);
|
|
if (cbfunc) {
|
|
cb_function = strdup(cbfunc);
|
|
}
|
|
|
|
switch_safe_free(cb_arg);
|
|
if (funcargs) {
|
|
cb_arg = strdup(funcargs);
|
|
}
|
|
|
|
args.buf = this;
|
|
switch_channel_set_private(channel, "CoreSession", this);
|
|
|
|
args.input_callback = dtmf_callback;
|
|
ap = &args;
|
|
}
|
|
|
|
switch_status_t Session::run_dtmf_callback(void *input, switch_input_type_t itype)
|
|
{
|
|
const char *ret;
|
|
|
|
if (!getLUA()) {
|
|
return SWITCH_STATUS_FALSE;;
|
|
}
|
|
|
|
switch (itype) {
|
|
case SWITCH_INPUT_TYPE_DTMF:
|
|
{
|
|
switch_dtmf_t *dtmf = (switch_dtmf_t *) input;
|
|
char str[3] = "";
|
|
int arg_count = 3;
|
|
int r;
|
|
|
|
lua_getglobal(L, (char *) cb_function);
|
|
lua_getglobal(L, uuid);
|
|
|
|
lua_pushstring(L, "dtmf");
|
|
|
|
lua_newtable(L);
|
|
lua_pushstring(L, "digit");
|
|
str[0] = dtmf->digit;
|
|
lua_pushstring(L, str);
|
|
lua_rawset(L, -3);
|
|
|
|
lua_pushstring(L, "duration");
|
|
lua_pushnumber(L, dtmf->duration);
|
|
lua_rawset(L, -3);
|
|
|
|
if (!zstr(cb_arg)) {
|
|
lua_getglobal(L, (char *) cb_arg);
|
|
arg_count++;
|
|
}
|
|
|
|
r = docall(L, arg_count, 1, 1, 0);
|
|
|
|
if (!r) {
|
|
ret = lua_tostring(L, -1);
|
|
lua_pop(L, 1);
|
|
} else {
|
|
ret = "SCRIPT_ERROR";
|
|
}
|
|
|
|
return process_callback_result((char *) ret);
|
|
}
|
|
break;
|
|
case SWITCH_INPUT_TYPE_EVENT:
|
|
{
|
|
switch_event_t *event = (switch_event_t *) input;
|
|
int arg_count = 3;
|
|
|
|
|
|
lua_getglobal(L, (char *) cb_function);
|
|
lua_getglobal(L, uuid);
|
|
lua_pushstring(L, "event");
|
|
mod_lua_conjure_event(L, event, "__Input_Event__", 1);
|
|
lua_getglobal(L, "__Input_Event__");
|
|
|
|
if (!zstr(cb_arg)) {
|
|
lua_getglobal(L, (char *) cb_arg);
|
|
arg_count++;
|
|
}
|
|
|
|
if (!docall(L, arg_count, 1, 1, 0)) {
|
|
ret = lua_tostring(L, -1);
|
|
lua_pop(L, 1);
|
|
} else {
|
|
ret = "SCRIPT_ERROR";
|
|
}
|
|
|
|
return process_callback_result((char *) ret);
|
|
}
|
|
break;
|
|
}
|
|
|
|
return SWITCH_STATUS_SUCCESS;
|
|
}
|
|
|
|
|
|
Dbh::Dbh(char *dsn, char *user, char *pass)
|
|
{
|
|
dbh = NULL;
|
|
char *tmp = NULL;
|
|
|
|
if (!zstr(user) || !zstr(pass)) {
|
|
tmp = switch_mprintf("%s%s%s%s%s", dsn,
|
|
zstr(user) ? "" : ":",
|
|
zstr(user) ? "" : user,
|
|
zstr(pass) ? "" : ":",
|
|
zstr(pass) ? "" : pass
|
|
);
|
|
|
|
dsn = tmp;
|
|
}
|
|
|
|
if (switch_cache_db_get_db_handle_dsn(&dbh, dsn) == SWITCH_STATUS_SUCCESS) {
|
|
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_DEBUG, "DBH handle %p Connected.\n", (void *) dbh);
|
|
} else {
|
|
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Connection failed. DBH NOT Connected.\n");
|
|
}
|
|
|
|
switch_safe_free(tmp);
|
|
|
|
}
|
|
|
|
Dbh::~Dbh()
|
|
{
|
|
if (dbh) release();
|
|
}
|
|
|
|
bool Dbh::release()
|
|
{
|
|
if (dbh) {
|
|
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_DEBUG, "DBH handle %p released.\n", (void *) dbh);
|
|
switch_cache_db_release_db_handle(&dbh);
|
|
return true;
|
|
}
|
|
|
|
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "DBH NOT Connected.\n");
|
|
return false;
|
|
}
|
|
|
|
bool Dbh::connected()
|
|
{
|
|
return dbh ? true : false;
|
|
}
|
|
|
|
bool Dbh::test_reactive(char *test_sql, char *drop_sql, char *reactive_sql)
|
|
{
|
|
if (dbh) {
|
|
if (switch_cache_db_test_reactive(dbh, test_sql, drop_sql, reactive_sql) == SWITCH_TRUE) {
|
|
return true;
|
|
}
|
|
}
|
|
|
|
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "DBH NOT Connected.\n");
|
|
return false;
|
|
}
|
|
|
|
int Dbh::query_callback(void *pArg, int argc, char **argv, char **cargv)
|
|
{
|
|
SWIGLUA_FN *lua_fun = (SWIGLUA_FN *)pArg;
|
|
int ret = 0;
|
|
|
|
lua_pushvalue(lua_fun->L, lua_fun->idx); /* get the lua callback function onto the stack */
|
|
|
|
lua_newtable(lua_fun->L); /* push a row (table) */
|
|
|
|
for (int i = 0; i < argc; i++) {
|
|
lua_pushstring(lua_fun->L, switch_str_nil(cargv[i]));
|
|
lua_pushstring(lua_fun->L, switch_str_nil(argv[i]));
|
|
lua_settable(lua_fun->L, -3);
|
|
}
|
|
|
|
if (docall(lua_fun->L, 1, 1, 1, 0)) {
|
|
return 1;
|
|
}
|
|
|
|
ret = lua_tonumber(lua_fun->L, -1);
|
|
lua_pop(lua_fun->L, 1);
|
|
|
|
if (ret != 0) {
|
|
return 1;
|
|
}
|
|
|
|
return 0; /* 0 to continue with next row */
|
|
}
|
|
|
|
bool Dbh::query(char *sql, SWIGLUA_FN lua_fun)
|
|
{
|
|
if (dbh) {
|
|
if (lua_fun.L) {
|
|
if (switch_cache_db_execute_sql_callback(dbh, sql, query_callback, &lua_fun, NULL) == SWITCH_STATUS_SUCCESS) {
|
|
return true;
|
|
}
|
|
} else { /* if no lua_fun arg is passed from Lua, an empty initialized struct will be sent - see freeswitch.i */
|
|
if (switch_cache_db_execute_sql(dbh, sql, NULL) == SWITCH_STATUS_SUCCESS) {
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
|
|
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "DBH NOT Connected.\n");
|
|
return false;
|
|
}
|
|
|
|
int Dbh::affected_rows()
|
|
{
|
|
if (dbh) {
|
|
return switch_cache_db_affected_rows(dbh);
|
|
}
|
|
|
|
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "DBH NOT Connected.\n");
|
|
return 0;
|
|
}
|
|
|
|
int Dbh::load_extension(const char *extension)
|
|
{
|
|
if (dbh) {
|
|
return switch_cache_db_load_extension(dbh, extension);
|
|
}
|
|
|
|
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "DBH NOT Connected.\n");
|
|
return 0;
|
|
}
|