add odbc support to javascript
git-svn-id: http://svn.freeswitch.org/svn/freeswitch/trunk@3514 d0543943-73ff-0310-b7d9-9358b9ac24b2
This commit is contained in:
Anthony Minessale
parent
2986cee258
commit
af5336563a
|
|
@ -50,7 +50,7 @@
|
|||
<!-- Applications -->
|
||||
<load module="mod_bridgecall"/>
|
||||
<load module="mod_commands"/>
|
||||
<!--<load module="mod_conference"/>-->
|
||||
<load module="mod_conference"/>
|
||||
<load module="mod_dptools"/>
|
||||
<load module="mod_echo"/>
|
||||
<!--<load module="mod_park"/>-->
|
||||
|
|
@ -88,6 +88,7 @@
|
|||
<modules>
|
||||
<load module="mod_spidermonkey_teletone"/>
|
||||
<load module="mod_spidermonkey_core_db"/>
|
||||
<!--<load module="mod_spidermonkey_odbc"/>-->
|
||||
</modules>
|
||||
</configuration>
|
||||
|
||||
|
|
|
|||
|
|
@ -43,5 +43,6 @@ formats/mod_sndfile
|
|||
#languages/mod_spidermonkey
|
||||
#languages/mod_spidermonkey_teletone
|
||||
#languages/mod_spidermonkey_core_db
|
||||
#languages/mod_spidermonkey_odbc
|
||||
timers/mod_softtimer
|
||||
#xml_int/mod_xml_rpc
|
||||
|
|
|
|||
|
|
@ -1,13 +1,13 @@
|
|||
include ../mod_spidermonkey/sm.mak
|
||||
|
||||
OBJS=#$(BASE)/libs/mozilla/js/src/$(OS_CONFIG)_$(VER).OBJ/libjs.a $(BASE)/libs/mozilla/nsprpub/dist/lib/libnspr4.a -lunixodbc
|
||||
OBJS=#$(BASE)/libs/mozilla/js/src/$(OS_CONFIG)_$(VER).OBJ/libjs.a $(BASE)/libs/mozilla/nsprpub/dist/lib/libnspr4.a
|
||||
LINKER=$(CC)
|
||||
|
||||
LDFLAGS=-lodbc
|
||||
|
||||
all: depends $(MODNAME).$(DYNAMIC_LIB_EXTEN)
|
||||
|
||||
depends:
|
||||
MAKE=$(MAKE) $(BASE)/build/buildlib.sh $(BASE) install unixODBC-2.2.12.tar.gz --prefix=$(PREFIX)
|
||||
MAKE=$(MAKE) $(BASE)/build/buildlib.sh $(BASE) install unixODBC-2.2.12.tar.gz --prefix=$(PREFIX) --disable-gui --without-x --with-pic
|
||||
|
||||
$(MODNAME).$(DYNAMIC_LIB_EXTEN): $(MODNAME).c
|
||||
$(CC) $(CFLAGS) -fPIC -c $(MODNAME).c -o $(MODNAME).o
|
||||
|
|
|
|||
|
|
@ -31,6 +31,7 @@
|
|||
*/
|
||||
#include "mod_spidermonkey.h"
|
||||
|
||||
|
||||
#include <sql.h>
|
||||
#ifdef _MSC_VER
|
||||
#pragma warning(push)
|
||||
|
|
@ -42,22 +43,438 @@
|
|||
#endif
|
||||
#include <sqltypes.h>
|
||||
|
||||
static const char modname[] = "odbc";
|
||||
static const char modname[] = "ODBC";
|
||||
|
||||
struct odbc_obj {
|
||||
char *dsn;
|
||||
char *username;
|
||||
char *password;
|
||||
SQLHENV env;
|
||||
SQLHDBC con;
|
||||
SQLHSTMT stmt;
|
||||
uint32_t state;
|
||||
SQLCHAR *colbuf;
|
||||
int32 cblen;
|
||||
SQLCHAR *code;
|
||||
int32 codelen;
|
||||
};
|
||||
|
||||
typedef enum {
|
||||
ODBC_STATE_INIT,
|
||||
ODBC_STATE_DOWN,
|
||||
ODBC_STATE_CONNECTED,
|
||||
ODBC_STATE_ERROR
|
||||
} odbc_state_t;
|
||||
typedef struct odbc_obj odbc_obj_t;
|
||||
|
||||
typedef enum {
|
||||
ODBC_SUCCESS = 0,
|
||||
ODBC_FAIL = -1
|
||||
} odbc_status_t;
|
||||
|
||||
|
||||
static odbc_obj_t *new_odbc_obj(char *dsn, char *username, char *password)
|
||||
{
|
||||
odbc_obj_t *new_obj;
|
||||
|
||||
if (!(new_obj = malloc(sizeof(*new_obj)))) {
|
||||
goto err;
|
||||
}
|
||||
|
||||
if (!(new_obj->dsn = strdup(dsn))) {
|
||||
goto err;
|
||||
}
|
||||
|
||||
if (!(new_obj->username = strdup(username))) {
|
||||
goto err;
|
||||
}
|
||||
|
||||
if (!(new_obj->password = strdup(password))) {
|
||||
goto err;
|
||||
}
|
||||
|
||||
new_obj->env = SQL_NULL_HANDLE;
|
||||
new_obj->state = ODBC_STATE_INIT;
|
||||
|
||||
return new_obj;
|
||||
|
||||
err:
|
||||
if (new_obj) {
|
||||
switch_safe_free(new_obj->dsn);
|
||||
switch_safe_free(new_obj->username);
|
||||
switch_safe_free(new_obj->password);
|
||||
switch_safe_free(new_obj);
|
||||
}
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
odbc_status_t odbc_obj_disconnect(odbc_obj_t *obj)
|
||||
{
|
||||
int result;
|
||||
|
||||
if (obj->state == ODBC_STATE_CONNECTED) {
|
||||
result = SQLDisconnect(obj->con);
|
||||
if (result == ODBC_SUCCESS) {
|
||||
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_DEBUG, "Disconnected %d from [%s]\n", result, obj->dsn);
|
||||
} else {
|
||||
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Error Disconnectiong [%s]\n", obj->dsn);
|
||||
}
|
||||
} else {
|
||||
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_WARNING, "[%s] already disconnected\n", obj->dsn);
|
||||
}
|
||||
|
||||
obj->state = ODBC_STATE_DOWN;
|
||||
|
||||
return ODBC_SUCCESS;
|
||||
}
|
||||
|
||||
odbc_status_t odbc_obj_connect(odbc_obj_t *obj)
|
||||
{
|
||||
int result;
|
||||
SQLINTEGER err;
|
||||
int16_t mlen;
|
||||
unsigned char msg[200], stat[10];
|
||||
|
||||
if (obj->env == SQL_NULL_HANDLE) {
|
||||
result = SQLAllocHandle(SQL_HANDLE_ENV, SQL_NULL_HANDLE, &obj->env);
|
||||
|
||||
if ((result != SQL_SUCCESS) && (result != SQL_SUCCESS_WITH_INFO)) {
|
||||
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_DEBUG, "Error AllocHandle\n");
|
||||
return ODBC_FAIL;
|
||||
}
|
||||
|
||||
result = SQLSetEnvAttr(obj->env, SQL_ATTR_ODBC_VERSION, (void *) SQL_OV_ODBC3, 0);
|
||||
|
||||
if ((result != SQL_SUCCESS) && (result != SQL_SUCCESS_WITH_INFO)) {
|
||||
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_DEBUG, "Error SetEnv\n");
|
||||
SQLFreeHandle(SQL_HANDLE_ENV, obj->env);
|
||||
return ODBC_FAIL;
|
||||
}
|
||||
|
||||
result = SQLAllocHandle(SQL_HANDLE_DBC, obj->env, &obj->con);
|
||||
|
||||
if ((result != SQL_SUCCESS) && (result != SQL_SUCCESS_WITH_INFO)) {
|
||||
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_DEBUG, "Error AllocHDB %d\n", result);
|
||||
SQLFreeHandle(SQL_HANDLE_ENV, obj->env);
|
||||
return ODBC_FAIL;
|
||||
}
|
||||
SQLSetConnectAttr(obj->con, SQL_LOGIN_TIMEOUT, (SQLPOINTER *) 10, 0);
|
||||
}
|
||||
if (obj->state == ODBC_STATE_CONNECTED) {
|
||||
odbc_obj_disconnect(obj);
|
||||
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_DEBUG, "Re-connecting %s\n", obj->dsn);
|
||||
}
|
||||
|
||||
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_DEBUG, "Connecting %s\n", obj->dsn);
|
||||
|
||||
result = SQLConnect(obj->con,
|
||||
(SQLCHAR *) obj->dsn, SQL_NTS,
|
||||
(SQLCHAR *) obj->username, SQL_NTS,
|
||||
(SQLCHAR *) obj->password, SQL_NTS);
|
||||
|
||||
if ((result != SQL_SUCCESS) && (result != SQL_SUCCESS_WITH_INFO)) {
|
||||
SQLGetDiagRec(SQL_HANDLE_DBC, obj->con, 1, stat, &err, msg, 100, &mlen);
|
||||
SQLFreeHandle(SQL_HANDLE_ENV, obj->env);
|
||||
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_DEBUG, "Error SQLConnect=%d errno=%d %s\n", result, (int)err, msg);
|
||||
return ODBC_FAIL;
|
||||
} else {
|
||||
|
||||
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_DEBUG, "Connected to [%s]\n", obj->dsn);
|
||||
obj->state = ODBC_STATE_CONNECTED;
|
||||
}
|
||||
|
||||
return ODBC_SUCCESS;
|
||||
}
|
||||
|
||||
static void destroy_odbc_obj(odbc_obj_t **objp)
|
||||
{
|
||||
odbc_obj_t *obj = *objp;
|
||||
|
||||
odbc_obj_disconnect(obj);
|
||||
|
||||
SQLFreeHandle(SQL_HANDLE_STMT, obj->stmt);
|
||||
SQLFreeHandle(SQL_HANDLE_DBC, obj->con);
|
||||
SQLFreeHandle(SQL_HANDLE_ENV, obj->env);
|
||||
|
||||
switch_safe_free(obj->dsn);
|
||||
switch_safe_free(obj->username);
|
||||
switch_safe_free(obj->password);
|
||||
switch_safe_free(obj->colbuf);
|
||||
switch_safe_free(obj->code);
|
||||
switch_safe_free(obj);
|
||||
}
|
||||
|
||||
|
||||
/* ODBC Object */
|
||||
/*********************************************************************************/
|
||||
static JSBool odbc_construct(JSContext *cx, JSObject *obj, uintN argc, jsval *argv, jsval *rval)
|
||||
{
|
||||
odbc_obj_t *odbc_obj = NULL;
|
||||
char *dsn, *username, *password;
|
||||
int32 blen = 1024;
|
||||
|
||||
if (argc < 3) {
|
||||
return JS_FALSE;
|
||||
}
|
||||
|
||||
dsn = JS_GetStringBytes(JS_ValueToString(cx, argv[0]));
|
||||
username = JS_GetStringBytes(JS_ValueToString(cx, argv[1]));
|
||||
password = JS_GetStringBytes(JS_ValueToString(cx, argv[2]));
|
||||
|
||||
if (argc > 3) {
|
||||
int32 len;
|
||||
JS_ValueToInt32(cx, argv[3], &len);
|
||||
|
||||
if (len > 0) {
|
||||
blen = len;
|
||||
}
|
||||
}
|
||||
|
||||
if (dsn && username && password) {
|
||||
odbc_obj = new_odbc_obj(dsn, username, password);
|
||||
}
|
||||
|
||||
if (!odbc_obj) {
|
||||
return JS_FALSE;
|
||||
}
|
||||
|
||||
if (!(odbc_obj->colbuf = (SQLCHAR *) malloc(blen))) {
|
||||
destroy_odbc_obj(&odbc_obj);
|
||||
return JS_FALSE;
|
||||
}
|
||||
|
||||
odbc_obj->cblen = blen;
|
||||
|
||||
blen += 1536;
|
||||
|
||||
if (!(odbc_obj->code = (SQLCHAR *) malloc(blen))) {
|
||||
destroy_odbc_obj(&odbc_obj);
|
||||
return JS_FALSE;
|
||||
}
|
||||
|
||||
odbc_obj->codelen = blen;
|
||||
|
||||
JS_SetPrivate(cx, obj, odbc_obj);
|
||||
|
||||
return JS_TRUE;
|
||||
}
|
||||
|
||||
static void odbc_destroy(JSContext *cx, JSObject *obj)
|
||||
{
|
||||
odbc_obj_t *odbc_obj = (odbc_obj_t *) JS_GetPrivate(cx, obj);
|
||||
|
||||
if (odbc_obj) {
|
||||
destroy_odbc_obj(&odbc_obj);
|
||||
}
|
||||
}
|
||||
|
||||
static JSBool odbc_my_method(JSContext *cx, JSObject *obj, uintN argc, jsval *argv, jsval *rval)
|
||||
static JSBool odbc_connect(JSContext *cx, JSObject *obj, uintN argc, jsval *argv, jsval *rval)
|
||||
{
|
||||
return JS_FALSE;
|
||||
odbc_obj_t *odbc_obj = (odbc_obj_t *) JS_GetPrivate(cx, obj);
|
||||
JSBool tf = JS_TRUE;
|
||||
|
||||
if (odbc_obj) {
|
||||
if (odbc_obj_connect(odbc_obj) == ODBC_SUCCESS) {
|
||||
tf = JS_TRUE;
|
||||
} else {
|
||||
tf = JS_FALSE;
|
||||
}
|
||||
}
|
||||
|
||||
*rval = BOOLEAN_TO_JSVAL( tf );
|
||||
|
||||
return JS_TRUE;
|
||||
}
|
||||
|
||||
static JSBool odbc_exec(JSContext *cx, JSObject *obj, uintN argc, jsval *argv, jsval *rval)
|
||||
{
|
||||
odbc_obj_t *odbc_obj = (odbc_obj_t *) JS_GetPrivate(cx, obj);
|
||||
char *sql;
|
||||
JSBool tf = JS_FALSE;
|
||||
int result;
|
||||
|
||||
if (argc < 1) {
|
||||
goto done;
|
||||
}
|
||||
|
||||
if (odbc_obj->state != ODBC_STATE_CONNECTED) {
|
||||
goto done;
|
||||
}
|
||||
|
||||
if (odbc_obj->stmt) {
|
||||
SQLFreeHandle (SQL_HANDLE_STMT, odbc_obj->stmt);
|
||||
odbc_obj->stmt = NULL;
|
||||
}
|
||||
|
||||
sql = JS_GetStringBytes(JS_ValueToString(cx, argv[0]));
|
||||
|
||||
if (SQLAllocHandle(SQL_HANDLE_STMT, odbc_obj->con, &odbc_obj->stmt) != SQL_SUCCESS) {
|
||||
goto done;
|
||||
}
|
||||
|
||||
if (SQLPrepare(odbc_obj->stmt, (unsigned char *)sql, SQL_NTS) != SQL_SUCCESS) {
|
||||
goto done;
|
||||
}
|
||||
|
||||
result = SQLExecute(odbc_obj->stmt);
|
||||
|
||||
if (result != SQL_SUCCESS && result != SQL_SUCCESS_WITH_INFO) {
|
||||
goto done;
|
||||
}
|
||||
|
||||
tf = JS_TRUE;
|
||||
|
||||
done:
|
||||
|
||||
*rval = BOOLEAN_TO_JSVAL( tf );
|
||||
|
||||
return JS_TRUE;
|
||||
}
|
||||
|
||||
static JSBool odbc_num_rows(JSContext *cx, JSObject *obj, uintN argc, jsval *argv, jsval *rval)
|
||||
{
|
||||
odbc_obj_t *odbc_obj = (odbc_obj_t *) JS_GetPrivate(cx, obj);
|
||||
|
||||
SQLSMALLINT rows = 0;
|
||||
|
||||
if (odbc_obj->state != ODBC_STATE_CONNECTED) {
|
||||
goto done;
|
||||
}
|
||||
|
||||
if (odbc_obj->stmt) {
|
||||
SQLNumResultCols(odbc_obj->stmt, &rows);
|
||||
}
|
||||
|
||||
done:
|
||||
|
||||
*rval = INT_TO_JSVAL( rows );
|
||||
|
||||
return JS_TRUE;
|
||||
|
||||
}
|
||||
|
||||
|
||||
static JSBool odbc_next_row(JSContext *cx, JSObject *obj, uintN argc, jsval *argv, jsval *rval)
|
||||
{
|
||||
odbc_obj_t *odbc_obj = (odbc_obj_t *) JS_GetPrivate(cx, obj);
|
||||
int result = 0;
|
||||
JSBool tf = JS_FALSE;
|
||||
|
||||
if (odbc_obj->state != ODBC_STATE_CONNECTED) {
|
||||
goto done;
|
||||
}
|
||||
|
||||
if (odbc_obj->stmt) {
|
||||
if ((result = SQLFetch(odbc_obj->stmt) == SQL_SUCCESS)) {
|
||||
tf = JS_TRUE;
|
||||
}
|
||||
}
|
||||
|
||||
done:
|
||||
|
||||
*rval = BOOLEAN_TO_JSVAL( tf );
|
||||
|
||||
return JS_TRUE;
|
||||
}
|
||||
|
||||
static char *escape_data(char *in)
|
||||
{
|
||||
switch_size_t nlen = strlen(in);
|
||||
uint32_t qc = 0;
|
||||
char *p, *q, *r;
|
||||
|
||||
for(p = in; p && *p; p++) {
|
||||
if (*p == '"') {
|
||||
qc += 2;
|
||||
}
|
||||
}
|
||||
|
||||
nlen += qc + 1;
|
||||
|
||||
if (!(q = (char *) malloc(nlen))) {
|
||||
return NULL;
|
||||
}
|
||||
|
||||
r = q;
|
||||
qc = 0;
|
||||
for(p = in; p && *p; p++) {
|
||||
if (*p == '"') {
|
||||
*r++ = '\\';
|
||||
}
|
||||
*r++ = *p;
|
||||
if (++qc > nlen) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
*r++ = '\0';
|
||||
|
||||
return q;
|
||||
|
||||
}
|
||||
|
||||
|
||||
static JSBool odbc_get_data(JSContext *cx, JSObject *obj, uintN argc, jsval *argv, jsval *rval)
|
||||
{
|
||||
|
||||
odbc_obj_t *odbc_obj = (odbc_obj_t *) JS_GetPrivate(cx, obj);
|
||||
JSBool tf = JS_FALSE;
|
||||
|
||||
if (odbc_obj->state != ODBC_STATE_CONNECTED) {
|
||||
goto done;
|
||||
}
|
||||
|
||||
if (odbc_obj->stmt) {
|
||||
SQLSMALLINT c = 0, x = 0;
|
||||
int result;
|
||||
char code[66560];
|
||||
|
||||
snprintf(code, sizeof(code), "~var _oDbC_dB_RoW_DaTa_ = {}");
|
||||
eval_some_js(code, cx, obj, rval);
|
||||
if (*rval == JS_FALSE) {
|
||||
return JS_TRUE;
|
||||
}
|
||||
|
||||
result = SQLNumResultCols(odbc_obj->stmt, &c);
|
||||
if (result == SQL_SUCCESS || result == SQL_SUCCESS_WITH_INFO) {
|
||||
for (x = 1; x <= c; x++) {
|
||||
SQLSMALLINT NameLength, DataType, DecimalDigits, Nullable;
|
||||
SQLUINTEGER ColumnSize;
|
||||
SQLCHAR name[1024] = "";
|
||||
SQLCHAR *data = odbc_obj->colbuf;
|
||||
SQLCHAR *esc = NULL;
|
||||
|
||||
SQLDescribeCol(odbc_obj->stmt, x, name, sizeof(name), &NameLength, &DataType, &ColumnSize, &DecimalDigits, &Nullable);
|
||||
SQLGetData(odbc_obj->stmt, x, SQL_C_CHAR, odbc_obj->colbuf, odbc_obj->cblen, NULL);
|
||||
|
||||
if (strchr((char *)odbc_obj->colbuf, '"')) { /* please don't */
|
||||
esc = (SQLCHAR *) escape_data((char *)odbc_obj->colbuf);
|
||||
data = esc;
|
||||
}
|
||||
|
||||
snprintf((char *)odbc_obj->code, odbc_obj->codelen, "~_oDbC_dB_RoW_DaTa_[\"%s\"] = \"%s\"", name, data);
|
||||
switch_safe_free(esc);
|
||||
|
||||
eval_some_js((char *)odbc_obj->code, cx, obj, rval);
|
||||
|
||||
if (*rval == JS_FALSE) {
|
||||
return JS_TRUE;
|
||||
}
|
||||
}
|
||||
|
||||
JS_GetProperty(cx, obj, "_oDbC_dB_RoW_DaTa_", rval);
|
||||
return JS_TRUE;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
done:
|
||||
|
||||
*rval = BOOLEAN_TO_JSVAL( tf );
|
||||
|
||||
return JS_TRUE;
|
||||
|
||||
}
|
||||
|
||||
enum odbc_tinyid {
|
||||
|
|
@ -65,7 +482,11 @@ enum odbc_tinyid {
|
|||
};
|
||||
|
||||
static JSFunctionSpec odbc_methods[] = {
|
||||
// {"myMethod", odbc_my_method, 1},
|
||||
{"connect", odbc_connect, 1},
|
||||
{"exec", odbc_exec, 1},
|
||||
{"numRows", odbc_num_rows, 1},
|
||||
{"nextRow", odbc_next_row, 1},
|
||||
{"getData", odbc_get_data, 1},
|
||||
{0}
|
||||
};
|
||||
|
||||
|
|
@ -79,7 +500,7 @@ static JSPropertySpec odbc_props[] = {
|
|||
static JSBool odbc_getProperty(JSContext *cx, JSObject *obj, jsval id, jsval *vp)
|
||||
{
|
||||
JSBool res = JS_TRUE;
|
||||
|
||||
|
||||
return res;
|
||||
}
|
||||
|
||||
|
|
@ -91,7 +512,7 @@ JSClass odbc_class = {
|
|||
};
|
||||
|
||||
|
||||
switch_status_t spidermonkey_load(JSContext *cx, JSObject *obj)
|
||||
switch_status_t odbc_load(JSContext *cx, JSObject *obj)
|
||||
{
|
||||
JS_InitClass(cx,
|
||||
obj,
|
||||
|
|
@ -110,7 +531,7 @@ switch_status_t spidermonkey_load(JSContext *cx, JSObject *obj)
|
|||
|
||||
const sm_module_interface_t odbc_module_interface = {
|
||||
/*.name = */ modname,
|
||||
/*.spidermonkey_load*/ spidermonkey_load,
|
||||
/*.spidermonkey_load*/ odbc_load,
|
||||
/*.next*/ NULL
|
||||
};
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue