47feede6777f217fb2e2dff71635da04898e0077nd

47feede6777f217fb2e2dff71635da04898e0077nd#include "mod_lua.h"

47feede6777f217fb2e2dff71635da04898e0077nd#include "lua_apr.h"

af33a4994ae2ff15bc67d19ff1a7feb906745bf8rbowen#include "lua_dbd.h"

3f08db06526d6901aa08c110b5bc7dde6bc39905nd

47feede6777f217fb2e2dff71635da04898e0077ndAPLOG_USE_MODULE(lua);

47feede6777f217fb2e2dff71635da04898e0077ndstatic APR_OPTIONAL_FN_TYPE(ap_dbd_close) *lua_ap_dbd_close = NULL;

47feede6777f217fb2e2dff71635da04898e0077ndstatic APR_OPTIONAL_FN_TYPE(ap_dbd_open) *lua_ap_dbd_open = NULL;

3f08db06526d6901aa08c110b5bc7dde6bc39905nd

47feede6777f217fb2e2dff71635da04898e0077nd

47feede6777f217fb2e2dff71635da04898e0077nd

47feede6777f217fb2e2dff71635da04898e0077nd

b43f840409794ed298e8634f6284741f193b6c4ftakashistatic request_rec *ap_lua_check_request_rec(lua_State *L, int index)

47feede6777f217fb2e2dff71635da04898e0077nd{

864d6d55a72bdb982ebabbc95cf8f051c43fa6ddrbowen request_rec *r;

707c2713ba8f1aa11c1f22f69d3ec73522054b9fcovener luaL_checkudata(L, index, "Apache2.Request");

47feede6777f217fb2e2dff71635da04898e0077nd r = lua_unboxpointer(L, index);

b43f840409794ed298e8634f6284741f193b6c4ftakashi return r;

707c2713ba8f1aa11c1f22f69d3ec73522054b9fcovener}

47feede6777f217fb2e2dff71635da04898e0077nd

47feede6777f217fb2e2dff71635da04898e0077ndstatic lua_db_handle *lua_get_db_handle(lua_State *L)

707c2713ba8f1aa11c1f22f69d3ec73522054b9fcovener{

a99c5d4cc3cab6a62b04d52000dbc22ce1fa2d94coar luaL_checktype(L, 1, LUA_TTABLE);

a99c5d4cc3cab6a62b04d52000dbc22ce1fa2d94coar lua_rawgeti(L, 1, 0);

a99c5d4cc3cab6a62b04d52000dbc22ce1fa2d94coar luaL_checktype(L, -1, LUA_TUSERDATA);

a99c5d4cc3cab6a62b04d52000dbc22ce1fa2d94coar return (lua_db_handle *) lua_topointer(L, -1);

707c2713ba8f1aa11c1f22f69d3ec73522054b9fcovener}

707c2713ba8f1aa11c1f22f69d3ec73522054b9fcovener

707c2713ba8f1aa11c1f22f69d3ec73522054b9fcovenerstatic lua_db_result_set *lua_get_result_set(lua_State *L)

707c2713ba8f1aa11c1f22f69d3ec73522054b9fcovener{

707c2713ba8f1aa11c1f22f69d3ec73522054b9fcovener luaL_checktype(L, 1, LUA_TTABLE);

707c2713ba8f1aa11c1f22f69d3ec73522054b9fcovener lua_rawgeti(L, 1, 0);

b43f840409794ed298e8634f6284741f193b6c4ftakashi luaL_checktype(L, -1, LUA_TUSERDATA);

b43f840409794ed298e8634f6284741f193b6c4ftakashi return (lua_db_result_set *) lua_topointer(L, -1);

b43f840409794ed298e8634f6284741f193b6c4ftakashi}

b43f840409794ed298e8634f6284741f193b6c4ftakashi

b43f840409794ed298e8634f6284741f193b6c4ftakashi

47feede6777f217fb2e2dff71635da04898e0077ndint lua_db_close(lua_State *L)

b43f840409794ed298e8634f6284741f193b6c4ftakashi{

b43f840409794ed298e8634f6284741f193b6c4ftakashi /*~~~~~~~~~~~~~~~~~~~~*/

b43f840409794ed298e8634f6284741f193b6c4ftakashi lua_db_handle *db;

47feede6777f217fb2e2dff71635da04898e0077nd apr_status_t rc = 0;

30471a4650391f57975f60bbb6e4a90be7b284bfhumbedooh /*~~~~~~~~~~~~~~~~~~~~*/

1f1b6bf13313fdd14a45e52e553d3ff28689b717coar

47feede6777f217fb2e2dff71635da04898e0077nd db = lua_get_db_handle(L);

b43f840409794ed298e8634f6284741f193b6c4ftakashi if (db && db->alive) {

47feede6777f217fb2e2dff71635da04898e0077nd if (db->type == LUA_DBTYPE_APR_DBD) {

b43f840409794ed298e8634f6284741f193b6c4ftakashi rc = apr_dbd_close(db->driver, db->handle);

b43f840409794ed298e8634f6284741f193b6c4ftakashi if (db->pool) apr_pool_destroy(db->pool);

707c2713ba8f1aa11c1f22f69d3ec73522054b9fcovener }

47feede6777f217fb2e2dff71635da04898e0077nd else {

707c2713ba8f1aa11c1f22f69d3ec73522054b9fcovener lua_ap_dbd_close = APR_RETRIEVE_OPTIONAL_FN(ap_dbd_close);

b43f840409794ed298e8634f6284741f193b6c4ftakashi if (lua_ap_dbd_close != NULL)

b43f840409794ed298e8634f6284741f193b6c4ftakashi if (db->dbdhandle) lua_ap_dbd_close(db->server, db->dbdhandle);

b43f840409794ed298e8634f6284741f193b6c4ftakashi }

a99c5d4cc3cab6a62b04d52000dbc22ce1fa2d94coar

b43f840409794ed298e8634f6284741f193b6c4ftakashi db->driver = NULL;

b43f840409794ed298e8634f6284741f193b6c4ftakashi db->handle = NULL;

b43f840409794ed298e8634f6284741f193b6c4ftakashi db->alive = 0;

a99c5d4cc3cab6a62b04d52000dbc22ce1fa2d94coar db->pool = NULL;

20f499565e77defe9dab24dd85c02f38a1175855nd }

645cf915f6bc22be17750bc5bb34ade8de6744dfnd

b43f840409794ed298e8634f6284741f193b6c4ftakashi lua_settop(L, 0);

b43f840409794ed298e8634f6284741f193b6c4ftakashi lua_pushnumber(L, rc);

b43f840409794ed298e8634f6284741f193b6c4ftakashi return 1;

b43f840409794ed298e8634f6284741f193b6c4ftakashi}

b43f840409794ed298e8634f6284741f193b6c4ftakashi

b43f840409794ed298e8634f6284741f193b6c4ftakashiint lua_db_gc(lua_State *L)

b43f840409794ed298e8634f6284741f193b6c4ftakashi{

b43f840409794ed298e8634f6284741f193b6c4ftakashi /*~~~~~~~~~~~~~~~~*/

b43f840409794ed298e8634f6284741f193b6c4ftakashi lua_db_handle *db;

b43f840409794ed298e8634f6284741f193b6c4ftakashi /*~~~~~~~~~~~~~~~~~~~~*/

b43f840409794ed298e8634f6284741f193b6c4ftakashi

b43f840409794ed298e8634f6284741f193b6c4ftakashi db = lua_touserdata(L, 1);

b43f840409794ed298e8634f6284741f193b6c4ftakashi if (db && db->alive) {

b43f840409794ed298e8634f6284741f193b6c4ftakashi if (db->type == LUA_DBTYPE_APR_DBD) {

b43f840409794ed298e8634f6284741f193b6c4ftakashi apr_dbd_close(db->driver, db->handle);

b43f840409794ed298e8634f6284741f193b6c4ftakashi if (db->pool) apr_pool_destroy(db->pool);

b43f840409794ed298e8634f6284741f193b6c4ftakashi }

b43f840409794ed298e8634f6284741f193b6c4ftakashi else {

b43f840409794ed298e8634f6284741f193b6c4ftakashi lua_ap_dbd_close = APR_RETRIEVE_OPTIONAL_FN(ap_dbd_close);

b43f840409794ed298e8634f6284741f193b6c4ftakashi if (lua_ap_dbd_close != NULL)

b43f840409794ed298e8634f6284741f193b6c4ftakashi if (db->dbdhandle) lua_ap_dbd_close(db->server, db->dbdhandle);

b43f840409794ed298e8634f6284741f193b6c4ftakashi }

b43f840409794ed298e8634f6284741f193b6c4ftakashi db->driver = NULL;

b43f840409794ed298e8634f6284741f193b6c4ftakashi db->handle = NULL;

b43f840409794ed298e8634f6284741f193b6c4ftakashi db->alive = 0;

b43f840409794ed298e8634f6284741f193b6c4ftakashi db->pool = NULL;

b43f840409794ed298e8634f6284741f193b6c4ftakashi }

b43f840409794ed298e8634f6284741f193b6c4ftakashi lua_settop(L, 0);

b43f840409794ed298e8634f6284741f193b6c4ftakashi return 0;

b43f840409794ed298e8634f6284741f193b6c4ftakashi}

b43f840409794ed298e8634f6284741f193b6c4ftakashi

727872d18412fc021f03969b8641810d8896820bhumbedoohint lua_db_active(lua_State *L)

cc7e1025de9ac63bd4db6fe7f71c158b2cf09fe4humbedooh{

0d0ba3a410038e179b695446bb149cce6264e0abnd /*~~~~~~~~~~~~~~~~~~~~*/

cc7e1025de9ac63bd4db6fe7f71c158b2cf09fe4humbedooh lua_db_handle *db = 0;

727872d18412fc021f03969b8641810d8896820bhumbedooh apr_status_t rc = 0;

0d0ba3a410038e179b695446bb149cce6264e0abnd /*~~~~~~~~~~~~~~~~~~~~*/

0d0ba3a410038e179b695446bb149cce6264e0abnd

0d0ba3a410038e179b695446bb149cce6264e0abnd db = lua_get_db_handle(L);

ac082aefa89416cbdc9a1836eaf3bed9698201c8humbedooh if (db && db->alive) {

0d0ba3a410038e179b695446bb149cce6264e0abnd rc = apr_dbd_check_conn(db->driver, db->pool, db->handle);

0d0ba3a410038e179b695446bb149cce6264e0abnd if (rc == APR_SUCCESS) {

0d0ba3a410038e179b695446bb149cce6264e0abnd lua_pushboolean(L, 1);

727872d18412fc021f03969b8641810d8896820bhumbedooh return 1;

0d0ba3a410038e179b695446bb149cce6264e0abnd }

0d0ba3a410038e179b695446bb149cce6264e0abnd }

30471a4650391f57975f60bbb6e4a90be7b284bfhumbedooh

205f749042ed530040a4f0080dbcb47ceae8a374rjung lua_pushboolean(L, 0);

af33a4994ae2ff15bc67d19ff1a7feb906745bf8rbowen return 1;

0d0ba3a410038e179b695446bb149cce6264e0abnd}

7fec19672a491661b2fe4b29f685bc7f4efa64d4nd

int lua_db_query(lua_State *L)

{

/*~~~~~~~~~~~~~~~~~~~~~~~*/

lua_db_handle *db = 0;

apr_status_t rc = 0;

int x = 0;

const char *statement;

/*~~~~~~~~~~~~~~~~~~~~~~~*/

luaL_checktype(L, 3, LUA_TSTRING);

statement = lua_tostring(L, 3);

db = lua_get_db_handle(L);

if (db && db->alive)

rc = apr_dbd_query(db->driver, db->handle, &x, statement);

else {

rc = 0;

x = -1;

}

if (rc == APR_SUCCESS)

lua_pushnumber(L, x);

else {

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/

const char *err = apr_dbd_error(db->driver, db->handle, rc);

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/

lua_pushnil(L);

if (err) {

lua_pushstring(L, err);

return 2;

}

}

return 1;

}

int lua_db_escape(lua_State *L)

{

/*~~~~~~~~~~~~~~~~~~~~~*/

lua_db_handle *db = 0;

const char *statement;

const char *escaped = 0;

request_rec *r;

/*~~~~~~~~~~~~~~~~~~~~~*/

r = ap_lua_check_request_rec(L, 2);

if (r) {

luaL_checktype(L, 3, LUA_TSTRING);

statement = lua_tostring(L, 3);

db = lua_get_db_handle(L);

if (db && db->alive) {

apr_dbd_init(r->pool);

escaped = apr_dbd_escape(db->driver, r->pool, statement,

db->handle);

if (escaped) {

lua_pushstring(L, escaped);

return 1;

}

}

else {

lua_pushnil(L);

}

return (1);

}

return 0;

}

int lua_db_get_row(lua_State *L)

{

int row_no,x;

const char *entry;

apr_dbd_row_t *row = 0;

lua_db_result_set *res = lua_get_result_set(L);

row_no = luaL_optinteger(L, 2, 0);

lua_settop(L,0);

/* Fetch all rows at once? */

if (row_no == 0) {

row_no = 1;

lua_newtable(L);

while (apr_dbd_get_row(res->driver, res->pool, res->results,

&row, -1) != -1)

{

lua_pushinteger(L, row_no);

lua_newtable(L);

for (x = 0; x < res->cols; x++) {

entry = apr_dbd_get_entry(res->driver, row, x);

if (entry) {

lua_pushinteger(L, x + 1);

lua_pushstring(L, entry);

lua_rawset(L, -3);

}

}

lua_rawset(L, -3);

row_no++;

}

return 1;

}

/* Just fetch a single row */

if (apr_dbd_get_row(res->driver, res->pool, res->results,

&row, row_no) != -1)

{

lua_newtable(L);

for (x = 0; x < res->cols; x++) {

entry = apr_dbd_get_entry(res->driver, row, x);

if (entry) {

lua_pushinteger(L, x + 1);

lua_pushstring(L, entry);

lua_rawset(L, -3);

}

}

return 1;

}

return 0;

}

int lua_db_select(lua_State *L)

{

/*~~~~~~~~~~~~~~~~~~~~~~~*/

lua_db_handle *db = 0;

apr_status_t rc = 0;

const char *statement;

request_rec *r;

/*~~~~~~~~~~~~~~~~~~~~~~~*/

r = ap_lua_check_request_rec(L, 2);

if (r) {

luaL_checktype(L, 3, LUA_TSTRING);

statement = lua_tostring(L, 3);

db = lua_get_db_handle(L);

if (db && db->alive) {

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/

int cols;

apr_dbd_results_t *results = 0;

lua_db_result_set* resultset = NULL;

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/

rc = apr_dbd_select(db->driver, db->pool, db->handle,

&results, statement, 0);

if (rc == APR_SUCCESS) {

cols = apr_dbd_num_cols(db->driver, results);

if (cols > 0) {

lua_newtable(L);

resultset = lua_newuserdata(L, sizeof(lua_db_result_set));

resultset->cols = cols;

resultset->driver = db->driver;

resultset->pool = db->pool;

resultset->rows = apr_dbd_num_tuples(db->driver, results);

resultset->results = results;

luaL_newmetatable(L, "lua_apr.dbselect");

lua_pushliteral(L, "__call");

lua_pushcfunction(L, lua_db_get_row);

lua_rawset(L, -3);

lua_setmetatable(L, -3);

lua_rawseti(L, -2, 0);

return 1;

}

return 0;

}

else {

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/

const char *err = apr_dbd_error(db->driver, db->handle, rc);

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/

lua_pushnil(L);

if (err) {

lua_pushstring(L, err);

return 2;

}

}

}

lua_pushboolean(L, 0);

return 1;

}

return 0;

}

int lua_db_prepared_select(lua_State *L)

{

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/

lua_db_prepared_statement *st = 0;

apr_status_t rc = 0;

const char **vars;

int x, have;

/*~~~~~~~~~~~~~~~~~~~~~~~*/

/* Fetch the prepared statement and the vars passed */

luaL_checktype(L, 1, LUA_TTABLE);

lua_rawgeti(L, 1, 0);

luaL_checktype(L, -1, LUA_TUSERDATA);

st = (lua_db_prepared_statement*) lua_topointer(L, -1);

/* Check if we got enough variables passed on to us.

have = lua_gettop(L) - 2;

if (st->variables != -1 && have < st->variables ) {

lua_pushboolean(L, 0);

lua_pushfstring(L,

"Error in executing prepared statement: Expected %d arguments, got %d.",

st->variables, have);

return 2;

}

vars = apr_pcalloc(st->db->pool, have*sizeof(char *));

for (x = 0; x < have; x++) {

vars[x] = lua_tostring(L, x + 2);

}

/* Fire off the query */

if (st->db && st->db->alive) {

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/

int cols;

apr_dbd_results_t *results = 0;

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/

rc = apr_dbd_pselect(st->db->driver, st->db->pool, st->db->handle,

&results, st->statement, 0, have, vars);

if (rc == APR_SUCCESS) {

/*~~~~~~~~~~~~~~~~~~~~~*/

lua_db_result_set *resultset;

/*~~~~~~~~~~~~~~~~~~~~~*/

cols = apr_dbd_num_cols(st->db->driver, results);

lua_newtable(L);

resultset = lua_newuserdata(L, sizeof(lua_db_result_set));

resultset->cols = cols;

resultset->driver = st->db->driver;

resultset->pool = st->db->pool;

resultset->rows = apr_dbd_num_tuples(st->db->driver, results);

resultset->results = results;

luaL_newmetatable(L, "lua_apr.dbselect");

lua_pushliteral(L, "__call");

lua_pushcfunction(L, lua_db_get_row);

lua_rawset(L, -3);

lua_setmetatable(L, -3);

lua_rawseti(L, -2, 0);

return 1;

}

else {

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/

const char *err = apr_dbd_error(st->db->driver, st->db->handle, rc);

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/

lua_pushnil(L);

if (err) {

lua_pushstring(L, err);

return 2;

}

return 1;

}

}

lua_pushboolean(L, 0);

lua_pushliteral(L,

"Database connection seems to be closed, please reacquire it.");

return (2);

}

int lua_db_prepared_query(lua_State *L)

{

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/

lua_db_prepared_statement *st = 0;

apr_status_t rc = 0;

const char **vars;

int x, have;

/*~~~~~~~~~~~~~~~~~~~~~~~*/

/* Fetch the prepared statement and the vars passed */

luaL_checktype(L, 1, LUA_TTABLE);

lua_rawgeti(L, 1, 0);

luaL_checktype(L, -1, LUA_TUSERDATA);

st = (lua_db_prepared_statement*) lua_topointer(L, -1);

/* Check if we got enough variables passed on to us.

have = lua_gettop(L) - 2;

if (st->variables != -1 && have < st->variables ) {

lua_pushboolean(L, 0);

lua_pushfstring(L,

"Error in executing prepared statement: Expected %d arguments, got %d.",

st->variables, have);

return 2;

}

vars = apr_pcalloc(st->db->pool, have*sizeof(char *));

for (x = 0; x < have; x++) {

vars[x] = lua_tostring(L, x + 2);

}

/* Fire off the query */

if (st->db && st->db->alive) {

/*~~~~~~~~~~~~~~*/

int affected = 0;

/*~~~~~~~~~~~~~~*/

rc = apr_dbd_pquery(st->db->driver, st->db->pool, st->db->handle,

&affected, st->statement, have, vars);

if (rc == APR_SUCCESS) {

lua_pushinteger(L, affected);

return 1;

}

else {

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/

const char *err = apr_dbd_error(st->db->driver, st->db->handle, rc);

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/

lua_pushnil(L);

if (err) {

lua_pushstring(L, err);

return 2;

}

return 1;

}

}

lua_pushboolean(L, 0);

lua_pushliteral(L,

"Database connection seems to be closed, please reacquire it.");

return (2);

}

int lua_db_prepare(lua_State* L)

{

/*~~~~~~~~~~~~~~~~~~~~~~~~~~*/

lua_db_handle *db = 0;

apr_status_t rc = 0;

const char *statement, *at;

request_rec *r;

lua_db_prepared_statement* st;

int need = 0;

/*~~~~~~~~~~~~~~~~~~~~~~~~~~*/

r = ap_lua_check_request_rec(L, 2);

if (r) {

apr_dbd_prepared_t *pstatement = NULL;

luaL_checktype(L, 3, LUA_TSTRING);

statement = lua_tostring(L, 3);

/* Count number of variables in statement */

at = ap_strchr_c(statement,'%');

while (at != NULL) {

if (at[1] == '%') {

at++;

}

else {

need++;

}

at = ap_strchr_c(at+1,'%');

}

db = lua_get_db_handle(L);

rc = apr_dbd_prepare(db->driver, r->pool, db->handle, statement,

NULL, &pstatement);

if (rc != APR_SUCCESS) {

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/

const char *err = apr_dbd_error(db->driver, db->handle, rc);

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/

lua_pushnil(L);

if (err) {

lua_pushstring(L, err);

return 2;

}

return 1;

}

/* Push the prepared statement table */

lua_newtable(L);

st = lua_newuserdata(L, sizeof(lua_db_prepared_statement));

st->statement = pstatement;

st->variables = need;

st->db = db;

lua_pushliteral(L, "select");

lua_pushcfunction(L, lua_db_prepared_select);

lua_rawset(L, -4);

lua_pushliteral(L, "query");

lua_pushcfunction(L, lua_db_prepared_query);

lua_rawset(L, -4);

lua_rawseti(L, -2, 0);

return 1;

}

return 0;

}

int lua_db_prepared(lua_State* L)

{

/*~~~~~~~~~~~~~~~~~~~~~~~~~~*/

lua_db_handle *db = 0;

const char *tag;

request_rec *r;

lua_db_prepared_statement* st;

/*~~~~~~~~~~~~~~~~~~~~~~~~~~*/

r = ap_lua_check_request_rec(L, 2);

if (r) {

apr_dbd_prepared_t *pstatement = NULL;

db = lua_get_db_handle(L);

luaL_checktype(L, 3, LUA_TSTRING);

tag = lua_tostring(L, 3);

/* Look for the statement */

pstatement = apr_hash_get(db->dbdhandle->prepared, tag,

APR_HASH_KEY_STRING);

if (pstatement == NULL) {

lua_pushnil(L);

lua_pushfstring(L,

"Could not find any prepared statement called %s!", tag);

return 2;

}

/* Push the prepared statement table */

lua_newtable(L);

st = lua_newuserdata(L, sizeof(lua_db_prepared_statement));

st->statement = pstatement;

st->variables = -1; /* we don't know :( */

st->db = db;

lua_pushliteral(L, "select");

lua_pushcfunction(L, lua_db_prepared_select);

lua_rawset(L, -4);

lua_pushliteral(L, "query");

lua_pushcfunction(L, lua_db_prepared_query);

lua_rawset(L, -4);

lua_rawseti(L, -2, 0);

return 1;

}

return 0;

}

static lua_db_handle* lua_push_db_handle(lua_State *L, request_rec* r, int type,

apr_pool_t* pool)

{

lua_db_handle* db;

lua_newtable(L);

db = lua_newuserdata(L, sizeof(lua_db_handle));

db->alive = 1;

db->pool = pool;

db->type = type;

db->dbdhandle = 0;

db->server = r->server;

luaL_newmetatable(L, "lua_apr.dbacquire");

lua_pushliteral(L, "__gc");

lua_pushcfunction(L, lua_db_gc);

lua_rawset(L, -3);

lua_setmetatable(L, -2);

lua_rawseti(L, -2, 0);

lua_pushliteral(L, "escape");

lua_pushcfunction(L, lua_db_escape);

lua_rawset(L, -3);

lua_pushliteral(L, "close");

lua_pushcfunction(L, lua_db_close);

lua_rawset(L, -3);

lua_pushliteral(L, "select");

lua_pushcfunction(L, lua_db_select);

lua_rawset(L, -3);

lua_pushliteral(L, "query");

lua_pushcfunction(L, lua_db_query);

lua_rawset(L, -3);

lua_pushliteral(L, "active");

lua_pushcfunction(L, lua_db_active);

lua_rawset(L, -3);

lua_pushliteral(L, "prepare");

lua_pushcfunction(L, lua_db_prepare);

lua_rawset(L, -3);

lua_pushliteral(L, "prepared");

lua_pushcfunction(L, lua_db_prepared);

lua_rawset(L, -3);

return db;

}

AP_LUA_DECLARE(int) lua_db_acquire(lua_State *L)

{

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~*/

const char *type;

const char *arguments;

const char *error = 0;

request_rec *r;

lua_db_handle *db = 0;

apr_status_t rc = 0;

ap_dbd_t *dbdhandle = NULL;

apr_pool_t *pool = NULL;

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~*/

r = ap_lua_check_request_rec(L, 1);

if (r) {

type = luaL_optstring(L, 2, "mod_dbd"); /* Defaults to mod_dbd */

if (!strcmp(type, "mod_dbd")) {

lua_settop(L, 0);

lua_ap_dbd_open = APR_RETRIEVE_OPTIONAL_FN(ap_dbd_open);

if (lua_ap_dbd_open)

dbdhandle = (ap_dbd_t *) lua_ap_dbd_open(

r->server->process->pool, r->server);

if (dbdhandle) {

db = lua_push_db_handle(L, r, LUA_DBTYPE_MOD_DBD, dbdhandle->pool);

db->driver = dbdhandle->driver;

db->handle = dbdhandle->handle;

db->dbdhandle = dbdhandle;

return 1;

}

else {

lua_pushnil(L);

if ( lua_ap_dbd_open == NULL )

lua_pushliteral(L,

"mod_dbd doesn't seem to have been loaded.");

else

lua_pushliteral(

L,

"Could not acquire connection from mod_dbd. If your database is running, this may indicate a permission problem.");

return 2;

}

}

else {

rc = apr_pool_create(&pool, NULL);

if (rc != APR_SUCCESS) {

lua_pushnil(L);

lua_pushliteral(L, "Could not allocate memory for database!");

return 2;

}

apr_pool_tag(pool, "lua_dbd_pool");

apr_dbd_init(pool);

dbdhandle = apr_pcalloc(pool, sizeof(ap_dbd_t));

rc = apr_dbd_get_driver(pool, type, &dbdhandle->driver);

if (rc == APR_SUCCESS) {

luaL_checktype(L, 3, LUA_TSTRING);

arguments = lua_tostring(L, 3);

lua_settop(L, 0);

if (strlen(arguments)) {

rc = apr_dbd_open_ex(dbdhandle->driver, pool,

arguments, &dbdhandle->handle, &error);

if (rc == APR_SUCCESS) {

db = lua_push_db_handle(L, r, LUA_DBTYPE_APR_DBD, pool);

db->driver = dbdhandle->driver;

db->handle = dbdhandle->handle;

db->dbdhandle = dbdhandle;

return 1;

}

else {

lua_pushnil(L);

if (error) {

lua_pushstring(L, error);

return 2;

}

return 1;

}

}

lua_pushnil(L);

lua_pushliteral(L,

"No database connection string was specified.");

apr_pool_destroy(pool);

return (2);

}

else {

lua_pushnil(L);

if (APR_STATUS_IS_ENOTIMPL(rc)) {

lua_pushfstring(L,

"driver for %s not available", type);

}

else if (APR_STATUS_IS_EDSOOPEN(rc)) {

lua_pushfstring(L,

"can't find driver for %s", type);

}

else if (APR_STATUS_IS_ESYMNOTFOUND(rc)) {

lua_pushfstring(L,

"driver for %s is invalid or corrupted",

type);

}

else {

lua_pushliteral(L,

"mod_lua not compatible with APR in get_driver");

}

lua_pushinteger(L, rc);

apr_pool_destroy(pool);

return 3;

}

}

lua_pushnil(L);

return 1;

}

return 0;

}