mirror of https://github.com/sipwise/kamailio.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
300 lines
8.1 KiB
300 lines
8.1 KiB
/* $Id$
|
|
*
|
|
* Copyright (C) 2009 Sippy Software, Inc., http://www.sippysoft.com
|
|
*
|
|
* This file is part of SIP-Router, a free SIP server.
|
|
*
|
|
* SIP-Router is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version
|
|
*
|
|
* SIP-Router is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*
|
|
*/
|
|
|
|
#include "../../str.h"
|
|
#include "../../sr_module.h"
|
|
|
|
#include "python_exec.h"
|
|
#include "python_iface.h"
|
|
#include "python_msgobj.h"
|
|
#include "python_support.h"
|
|
|
|
#include <Python.h>
|
|
#include <libgen.h>
|
|
|
|
MODULE_VERSION
|
|
|
|
static int mod_init(void);
|
|
static int child_init(int rank);
|
|
static void mod_destroy(void);
|
|
|
|
static str script_name = {.s = "/usr/local/etc/sip-router/handler.py", .len = 0};
|
|
static str mod_init_fname = { .s = "mod_init", .len = 0};
|
|
static str child_init_mname = { .s = "child_init", .len = 0};
|
|
PyObject *handler_obj;
|
|
PyObject *format_exc_obj;
|
|
|
|
PyThreadState *myThreadState;
|
|
|
|
/** module parameters */
|
|
static param_export_t params[]={
|
|
{"script_name", STR_PARAM, &script_name },
|
|
{"mod_init_function", STR_PARAM, &mod_init_fname },
|
|
{"child_init_method", STR_PARAM, &child_init_mname },
|
|
{0,0,0}
|
|
};
|
|
|
|
/*
|
|
* Exported functions
|
|
*/
|
|
static cmd_export_t cmds[] = {
|
|
{ "python_exec", (cmd_function)python_exec1, 1, NULL, 0,
|
|
REQUEST_ROUTE | FAILURE_ROUTE
|
|
| ONREPLY_ROUTE | BRANCH_ROUTE },
|
|
{ "python_exec", (cmd_function)python_exec2, 2, NULL, 0,
|
|
REQUEST_ROUTE | FAILURE_ROUTE
|
|
| ONREPLY_ROUTE | BRANCH_ROUTE },
|
|
{ 0, 0, 0, 0, 0, 0 }
|
|
};
|
|
|
|
/** module exports */
|
|
struct module_exports exports = {
|
|
"app_python", /* module name */
|
|
RTLD_NOW | RTLD_GLOBAL, /* dlopen flags */
|
|
cmds, /* exported functions */
|
|
params, /* exported parameters */
|
|
0, /* exported statistics */
|
|
0, /* exported MI functions */
|
|
0, /* exported pseudo-variables */
|
|
0, /* extra processes */
|
|
mod_init, /* module initialization function */
|
|
(response_function) NULL, /* response handling function */
|
|
(destroy_function) mod_destroy, /* destroy function */
|
|
child_init /* per-child init function */
|
|
};
|
|
|
|
static int
|
|
mod_init(void)
|
|
{
|
|
char *dname, *bname, *tname;
|
|
int i;
|
|
PyObject *sys_path, *pDir, *pModule, *pFunc, *pArgs;
|
|
PyThreadState *mainThreadState;
|
|
|
|
if (script_name.len == 0) {
|
|
script_name.len = strlen(script_name.s);
|
|
}
|
|
if (mod_init_fname.len == 0) {
|
|
mod_init_fname.len = strlen(mod_init_fname.s);
|
|
}
|
|
if (child_init_mname.len == 0) {
|
|
child_init_mname.len = strlen(child_init_mname.s);
|
|
}
|
|
|
|
tname = as_asciiz(&script_name);
|
|
if(tname==NULL)
|
|
{
|
|
LM_ERR("no more pkg memory\n");
|
|
return -1;
|
|
}
|
|
dname = dirname(tname);
|
|
if (strlen(dname) == 0)
|
|
dname = ".";
|
|
memcpy(tname, script_name.s, script_name.len);
|
|
bname = basename(tname);
|
|
i = strlen(bname);
|
|
pkg_free(tname);
|
|
if (bname[i - 1] == 'c' || bname[i - 1] == 'o')
|
|
i -= 1;
|
|
if (bname[i - 3] == '.' && bname[i - 2] == 'p' && bname[i - 1] == 'y') {
|
|
bname[i - 3] = '\0';
|
|
} else {
|
|
LM_ERR("%s: script_name doesn't look like a python script\n",
|
|
script_name.s);
|
|
return -1;
|
|
}
|
|
|
|
Py_Initialize();
|
|
PyEval_InitThreads();
|
|
mainThreadState = PyThreadState_Get();
|
|
|
|
Py_InitModule("Router", RouterMethods);
|
|
|
|
if (python_msgobj_init() != 0) {
|
|
LM_ERR("python_msgobj_init() has failed\n");
|
|
PyEval_ReleaseLock();
|
|
return -1;
|
|
}
|
|
|
|
sys_path = PySys_GetObject("path");
|
|
/* PySys_GetObject doesn't pass reference! No need to DEREF */
|
|
if (sys_path == NULL) {
|
|
LM_ERR("cannot import sys.path\n");
|
|
PyEval_ReleaseLock();
|
|
return -1;
|
|
}
|
|
|
|
pDir = PyString_FromString(dname);
|
|
if (pDir == NULL) {
|
|
LM_ERR("PyString_FromString() has filed\n");
|
|
PyEval_ReleaseLock();
|
|
return -1;
|
|
}
|
|
PyList_Insert(sys_path, 0, pDir);
|
|
Py_DECREF(pDir);
|
|
|
|
pModule = PyImport_ImportModule(bname);
|
|
if (pModule == NULL) {
|
|
LM_ERR("cannot import %s\n", bname);
|
|
PyEval_ReleaseLock();
|
|
return -1;
|
|
}
|
|
|
|
pFunc = PyObject_GetAttrString(pModule, mod_init_fname.s);
|
|
Py_DECREF(pModule);
|
|
/* pFunc is a new reference */
|
|
if (pFunc == NULL || !PyCallable_Check(pFunc)) {
|
|
LM_ERR("cannot locate %s function in %s module\n",
|
|
mod_init_fname.s, script_name.s);
|
|
Py_XDECREF(pFunc);
|
|
PyEval_ReleaseLock();
|
|
return -1;
|
|
}
|
|
|
|
pModule = PyImport_ImportModule("traceback");
|
|
if (pModule == NULL) {
|
|
LM_ERR("cannot import traceback module\n");
|
|
Py_DECREF(pFunc);
|
|
PyEval_ReleaseLock();
|
|
return -1;
|
|
}
|
|
|
|
format_exc_obj = PyObject_GetAttrString(pModule, "format_exception");
|
|
Py_DECREF(pModule);
|
|
if (format_exc_obj == NULL || !PyCallable_Check(format_exc_obj)) {
|
|
LM_ERR("cannot locate format_exception function in" \
|
|
" traceback module\n");
|
|
Py_XDECREF(format_exc_obj);
|
|
Py_DECREF(pFunc);
|
|
PyEval_ReleaseLock();
|
|
return -1;
|
|
}
|
|
|
|
pArgs = PyTuple_New(0);
|
|
if (pArgs == NULL) {
|
|
LM_ERR("PyTuple_New() has failed\n");
|
|
Py_DECREF(pFunc);
|
|
Py_DECREF(format_exc_obj);
|
|
PyEval_ReleaseLock();
|
|
return -1;
|
|
}
|
|
|
|
handler_obj = PyObject_CallObject(pFunc, pArgs);
|
|
Py_DECREF(pFunc);
|
|
Py_DECREF(pArgs);
|
|
|
|
if (PyErr_Occurred()) {
|
|
python_handle_exception("mod_init");
|
|
Py_XDECREF(handler_obj);
|
|
Py_DECREF(format_exc_obj);
|
|
PyEval_ReleaseLock();
|
|
return -1;
|
|
}
|
|
|
|
if (handler_obj == NULL) {
|
|
LM_ERR("%s function has not returned object\n",
|
|
mod_init_fname.s);
|
|
Py_DECREF(format_exc_obj);
|
|
PyEval_ReleaseLock();
|
|
return -1;
|
|
}
|
|
|
|
myThreadState = PyThreadState_New(mainThreadState->interp);
|
|
PyEval_ReleaseLock();
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
child_init(int rank)
|
|
{
|
|
PyObject *pFunc, *pArgs, *pValue, *pResult;
|
|
int rval;
|
|
|
|
PyEval_AcquireLock();
|
|
PyThreadState_Swap(myThreadState);
|
|
|
|
pFunc = PyObject_GetAttrString(handler_obj, child_init_mname.s);
|
|
if (pFunc == NULL || !PyCallable_Check(pFunc)) {
|
|
LM_ERR("cannot locate %s function\n", child_init_mname.s);
|
|
if (pFunc != NULL) {
|
|
Py_DECREF(pFunc);
|
|
}
|
|
PyThreadState_Swap(NULL);
|
|
PyEval_ReleaseLock();
|
|
return -1;
|
|
}
|
|
|
|
pArgs = PyTuple_New(1);
|
|
if (pArgs == NULL) {
|
|
LM_ERR("PyTuple_New() has failed\n");
|
|
Py_DECREF(pFunc);
|
|
PyThreadState_Swap(NULL);
|
|
PyEval_ReleaseLock();
|
|
return -1;
|
|
}
|
|
|
|
pValue = PyInt_FromLong(rank);
|
|
if (pValue == NULL) {
|
|
LM_ERR("PyInt_FromLong() has failed\n");
|
|
Py_DECREF(pArgs);
|
|
Py_DECREF(pFunc);
|
|
PyThreadState_Swap(NULL);
|
|
PyEval_ReleaseLock();
|
|
return -1;
|
|
}
|
|
PyTuple_SetItem(pArgs, 0, pValue);
|
|
/* pValue has been stolen */
|
|
|
|
pResult = PyObject_CallObject(pFunc, pArgs);
|
|
Py_DECREF(pFunc);
|
|
Py_DECREF(pArgs);
|
|
|
|
if (PyErr_Occurred()) {
|
|
python_handle_exception("child_init");
|
|
Py_XDECREF(pResult);
|
|
PyThreadState_Swap(NULL);
|
|
PyEval_ReleaseLock();
|
|
return -1;
|
|
}
|
|
|
|
if (pResult == NULL) {
|
|
LM_ERR("PyObject_CallObject() returned NULL but no exception!\n");
|
|
PyThreadState_Swap(NULL);
|
|
PyEval_ReleaseLock();
|
|
return -1;
|
|
}
|
|
|
|
rval = PyInt_AsLong(pResult);
|
|
Py_DECREF(pResult);
|
|
PyThreadState_Swap(NULL);
|
|
PyEval_ReleaseLock();
|
|
return rval;
|
|
}
|
|
|
|
static void
|
|
mod_destroy(void)
|
|
{
|
|
|
|
return;
|
|
}
|