modules/database/mod_dbd.c

	mod_dbd.c revision b85a3d8639df8d6dfc52cceb8f45dba021e86c89
2N/A/* Copyright 2003-5 The Apache Software Foundation or its licensors, as
2N/A * applicable.
2N/A *
2N/A * Licensed under the Apache License, Version 2.0 (the "License");
2N/A * you may not use this file except in compliance with the License.
2N/A * You may obtain a copy of the License at
2N/A *
2N/A *     http://www.apache.org/licenses/LICENSE-2.0
2N/A *
2N/A * Unless required by applicable law or agreed to in writing, software
2N/A * distributed under the License is distributed on an "AS IS" BASIS,
2N/A * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
2N/A * See the License for the specific language governing permissions and
2N/A * limitations under the License.
2N/A */
2N/A
2N/A/* Overview of what this is and does:
2N/A * http://www.apache.org/~niq/dbd.html
2N/A * or
2N/A * http://apache.webthing.com/database/
2N/A */
2N/A
2N/A#include <ctype.h>
2N/A
2N/A#include "http_protocol.h"
2N/A#include "http_config.h"
2N/A#include "http_log.h"
2N/A#include "apr_reslist.h"
2N/A#include "apr_strings.h"
2N/A
2N/A#include "apr_dbd.h"
12N/A#include "mod_dbd.h"
12N/Aextern module AP_MODULE_DECLARE_DATA dbd_module;
2N/A
2N/A/************ svr cfg: manage db connection pool ****************/
2N/A
2N/Atypedef struct dbd_prepared {
26N/A    const char *label;
26N/A    const char *query;
2N/A    struct dbd_prepared *next;
26N/A} dbd_prepared;
26N/Atypedef struct svr_cfg {
26N/A    const char *name;
26N/A    const char *params;
26N/A    int persist;
26N/A    dbd_prepared *prepared;
26N/A#if APR_HAS_THREADS
26N/A    apr_reslist_t *dbpool;
26N/A    int nmin;
26N/A    int nkeep;
26N/A    int nmax;
26N/A    int exptime;
26N/A#else
26N/A    ap_dbd_t *conn;
26N/A#endif
26N/A} svr_cfg;
26N/A
26N/Atypedef enum { cmd_name, cmd_params, cmd_persist,
26N/A               cmd_min, cmd_keep, cmd_max, cmd_exp
26N/A} cmd_parts;
26N/A
2N/A
26N/A#define ISINT(val) \
26N/A        for (p = val; *p; ++p)        \
26N/A                if (!isdigit(*p))        \
26N/A                        return "Argument must be numeric!"
26N/Astatic const char *dbd_param(cmd_parms *cmd, void *cfg, const char *val)
26N/A{
26N/A    const char *p;
26N/A    const apr_dbd_driver_t *driver = NULL;
26N/A    svr_cfg *svr = (svr_cfg*) ap_get_module_config
26N/A        (cmd->server->module_config, &dbd_module);
26N/A
26N/A    switch ((int) cmd->info) {
26N/A    case cmd_name:
2N/A        svr->name = val;
26N/A        /* loading the driver involves once-only dlloading that is
26N/A         * best done at server startup.  This also guarantees that
26N/A         * load_driver won't return an error later.
26N/A         */
26N/A        switch (apr_dbd_get_driver(cmd->pool, svr->name, &driver)) {
26N/A        case APR_ENOTIMPL:
26N/A            return apr_psprintf(cmd->pool, "DBD: No driver for %s", svr->name);
12N/A        case APR_EDSOOPEN:
26N/A            return apr_psprintf(cmd->pool,
26N/A                                "DBD: Can't load driver file apr_dbd_%s.so",
26N/A                                svr->name);
26N/A        case APR_ESYMNOTFOUND:
2N/A            return apr_psprintf(cmd->pool,
26N/A                                "DBD: Failed to load driver apr_dbd_%s_driver",
26N/A                                svr->name);
26N/A        }
2N/A        break;
7N/A    case cmd_params:
26N/A        svr->params = val;
21N/A        break;
7N/A    case cmd_persist:
26N/A        ISINT(val);
26N/A        svr->persist = atoi(val);
26N/A        break;
26N/A#if APR_HAS_THREADS
26N/A    case cmd_min:
26N/A        ISINT(val);
26N/A        svr->nmin = atoi(val);
26N/A        break;
26N/A    case cmd_keep:
26N/A        ISINT(val);
26N/A        svr->nkeep = atoi(val);
26N/A        break;
26N/A    case cmd_max:
26N/A        ISINT(val);
26N/A        svr->nmax = atoi(val);
26N/A        break;
7N/A    case cmd_exp:
26N/A        ISINT(val);
26N/A        svr->exptime = atoi(val);
26N/A        break;
26N/A#endif
26N/A    }
26N/A    return NULL;
26N/A}
26N/Avoid ap_dbd_prepare(server_rec *s, const char *query, const char *label)
26N/A{
26N/A    svr_cfg *svr = ap_get_module_config(s->module_config, &dbd_module);
26N/A    dbd_prepared *prepared = apr_pcalloc(s->process->pool, sizeof(dbd_prepared));
26N/A    prepared->label = label;
26N/A    prepared->query = query;
26N/A    prepared->next = svr->prepared;
26N/A    svr->prepared = prepared;
26N/A}
26N/Astatic const char *dbd_prepare(cmd_parms *cmd, void *cfg, const char *query,
26N/A                               const char *label)
26N/A{
26N/A    ap_dbd_prepare(cmd->server, query, label);
2N/A    return NULL;
26N/A}
26N/Astatic const command_rec dbd_cmds[] = {
26N/A    AP_INIT_TAKE1("DBDriver", dbd_param, (void*)cmd_name, RSRC_CONF,
26N/A                  "SQL Driver"),
26N/A    AP_INIT_TAKE1("DBDParams", dbd_param, (void*)cmd_params, RSRC_CONF,
26N/A                  "SQL Driver Params"),
26N/A    AP_INIT_TAKE1("DBDPersist", dbd_param, (void*)cmd_persist, RSRC_CONF,
26N/A                  "Use persistent connection/pool"),
26N/A    AP_INIT_TAKE2("DBDPrepareSQL", dbd_prepare, NULL, RSRC_CONF,
26N/A                  "Prepared SQL statement, label"),
26N/A#if APR_HAS_THREADS
26N/A    AP_INIT_TAKE1("DBDMin", dbd_param, (void*)cmd_min, RSRC_CONF,
26N/A                  "Minimum number of connections"),
26N/A    AP_INIT_TAKE1("DBDKeep", dbd_param, (void*)cmd_keep, RSRC_CONF,
26N/A                  "Maximum number of sustained connections"),
26N/A    AP_INIT_TAKE1("DBDMax", dbd_param, (void*)cmd_max, RSRC_CONF,
26N/A                  "Maximum number of connections"),
26N/A    AP_INIT_TAKE1("DBDExptime", dbd_param, (void*)cmd_exp, RSRC_CONF,
26N/A                  "Keepalive time for idle connections"),
26N/A#endif
26N/A    {NULL}
26N/A};
26N/A#define COND_PARAM(x,val) \
2N/A    if (cfg->x == val) {              \
26N/A        cfg->x = ((svr_cfg*)base)->x; \
26N/A    }
26N/A#define COND_PARAM0(x) COND_PARAM(x,0)
26N/A#define COND_PARAM1(x) COND_PARAM(x,-1)
26N/Astatic void *dbd_merge(apr_pool_t *pool, void *base, void *add) {
26N/A    svr_cfg *cfg = apr_pmemdup(pool, add, sizeof(svr_cfg));
26N/A    COND_PARAM0(name);
26N/A    COND_PARAM0(params);
26N/A    COND_PARAM1(persist);
2N/A#if APR_HAS_THREADS
26N/A    COND_PARAM0(nmin);
26N/A    COND_PARAM0(nkeep);
2N/A    COND_PARAM0(nmax);
26N/A    COND_PARAM0(exptime);
26N/A#endif
26N/A    return cfg;
26N/A}
26N/A#undef COND_PARAM
26N/A#undef COND_PARAM0
26N/A#undef COND_PARAM1
26N/Astatic void *dbd_cfg(apr_pool_t *p, server_rec *x)
26N/A{
26N/A    svr_cfg *svr = (svr_cfg*) apr_pcalloc(p, sizeof(svr_cfg));
    svr->persist = -1;
    return svr;
}
static apr_status_t dbd_prepared_init(apr_pool_t *pool, svr_cfg *svr,
                                      ap_dbd_t *dbd)
{
    dbd_prepared *p;
    apr_status_t ret = APR_SUCCESS;
    apr_dbd_prepared_t *stmt = NULL;
    dbd->prepared = apr_hash_make(pool);

    for (p = svr->prepared; p; p = p->next) {
        if (apr_dbd_prepare(dbd->driver, pool, dbd->handle, p->query,
                            p->label, &stmt) == 0) {
            apr_hash_set(dbd->prepared, p->label, APR_HASH_KEY_STRING, stmt);
        }
        else {
            ret = APR_EGENERAL;
        }
    }
    return ret;
}
/************ svr cfg: manage db connection pool ****************/
/* an apr_reslist_constructor for SQL connections
 * Also use this for opening in non-reslist modes, since it gives
 * us all the error-handling in one place.
 */
static apr_status_t dbd_construct(void **db, void *params, apr_pool_t *pool)
{
    svr_cfg *svr = (svr_cfg*) params;
    ap_dbd_t *rec = apr_pcalloc(pool, sizeof(ap_dbd_t));
    apr_status_t rv = apr_dbd_get_driver(pool, svr->name, &rec->driver);

/* Error-checking get_driver isn't necessary now (because it's done at
 * config-time).  But in case that changes in future ...
 */
    switch (rv) {
    case APR_ENOTIMPL:
        ap_log_perror(APLOG_MARK, APLOG_CRIT, 0, pool,
                      "DBD: driver for %s not available", svr->name);
        return rv;
    case APR_EDSOOPEN:
        ap_log_perror(APLOG_MARK, APLOG_CRIT, 0, pool,
                      "DBD: can't find driver for %s", svr->name);
        return rv;
    case APR_ESYMNOTFOUND:
        ap_log_perror(APLOG_MARK, APLOG_CRIT, 0, pool,
                      "DBD: driver for %s is invalid or corrupted", svr->name);
        return rv;
    default:
        ap_log_perror(APLOG_MARK, APLOG_CRIT, 0, pool,
                      "DBD: mod_dbd not compatible with apr in get_driver");
        return rv;
    case APR_SUCCESS:
        break;
    }

    rv = apr_dbd_open(rec->driver, pool, svr->params, &rec->handle);
    switch (rv) {
    case APR_EGENERAL:
        ap_log_perror(APLOG_MARK, APLOG_CRIT, 0, pool,
                      "DBD: Can't connect to %s[%s]", svr->name, svr->params);
        return rv;
    default:
        ap_log_perror(APLOG_MARK, APLOG_CRIT, 0, pool,
                      "DBD: mod_dbd not compatible with apr in open");
        return rv;
    case APR_SUCCESS:
        break;
    }
    *db = rec;
    rv = dbd_prepared_init(pool, svr, rec);
    return rv;
}
#if APR_HAS_THREADS
static apr_status_t dbd_destruct(void *sql, void *params, apr_pool_t *pool)
{
    ap_dbd_t *rec = sql;
    return apr_dbd_close(rec->driver, rec->handle);
}

static apr_status_t dbd_setup(apr_pool_t *pool, server_rec *s)
{
    svr_cfg *svr = ap_get_module_config(s->module_config, &dbd_module);
    apr_status_t rv = apr_reslist_create(&svr->dbpool, svr->nmin, svr->nkeep,
                                         svr->nmax, svr->exptime,
                                         dbd_construct, dbd_destruct,
                                         svr, pool);
    if (rv == APR_SUCCESS) {
        apr_pool_cleanup_register(pool, svr->dbpool,
                                  (void*)apr_reslist_destroy,
                                  apr_pool_cleanup_null);
    }
    else {
        ap_log_perror(APLOG_MARK, APLOG_CRIT, 0, pool,
                      "DBD Pool: failed to initialise");
    }
    return rv;
}

#endif


/* Functions we export for modules to use:
        - open acquires a connection from the pool (opens one if necessary)
        - close releases it back in to the pool
*/

#if APR_HAS_THREADS
ap_dbd_t* ap_dbd_open(apr_pool_t *pool, server_rec *s)
{
    ap_dbd_t *rec = NULL;
    svr_cfg *svr = ap_get_module_config(s->module_config, &dbd_module);
    apr_status_t rv;
    const char *errmsg;

    if (!svr->persist) {
        /* Return a once-only connection */
        rv = dbd_construct((void**)&rec, svr, pool);
        return (rv == APR_SUCCESS) ? rec : NULL;
    }

    if (!svr->dbpool) {
        if (dbd_setup(s->process->pool, s) != APR_SUCCESS) {
            return NULL;
        }
    }
    if (apr_reslist_acquire(svr->dbpool, (void**)&rec) != APR_SUCCESS) {
        ap_log_perror(APLOG_MARK, APLOG_ERR, 0, pool,
                      "Failed to acquire DBD connection from pool!");
        return NULL;
    }
    if (apr_dbd_check_conn(rec->driver, pool, rec->handle) != APR_SUCCESS) {
        errmsg = apr_dbd_error(rec->driver, rec->handle, rv);
        if (!errmsg) {
            errmsg = "(unknown)";
        }
        ap_log_perror(APLOG_MARK, APLOG_ERR, 0, pool,
                      "DBD[%s] Error: %s", svr->name, errmsg );
        apr_reslist_invalidate(svr->dbpool, rec);
        return NULL;
    }
    return rec;
}
#else
ap_dbd_t* ap_dbd_open(apr_pool_t *pool, server_rec *s)
{
    apr_status_t rv;
    const char *errmsg;
    ap_dbd_t *rec = NULL;
    svr_cfg *svr = ap_get_module_config(s->module_config, &dbd_module);

    if (!svr->persist) {
        /* Return a once-only connection */
        rv = dbd_construct((void**)&rec, svr, pool);
        return (rv == APR_SUCCESS) ? rec : NULL;
    }

/* since we're in nothread-land, we can mess with svr->conn with impunity */
/* If we have a persistent connection and it's good, we'll use it */
    if (svr->conn) {
        if (apr_dbd_check_conn(svr->conn->driver, pool, svr->conn->handle) != 0){
            errmsg = apr_dbd_error(rec->driver, rec->handle, rv);
            if (!errmsg) {
                errmsg = "(unknown)";
            }
            ap_log_perror(APLOG_MARK, APLOG_ERR, 0, pool,
                          "DBD[%s] Error: %s", svr->name, errmsg);
            svr->conn = NULL;
        }
    }
/* We don't have a connection right now, so we'll open one */
    if (!svr->conn) {
        rv = dbd_construct((void**)&rec, svr, pool);
        svr->conn = (rv == APR_SUCCESS) ? rec : NULL;
    }
    return svr->conn;
}
#endif
void ap_dbd_close(server_rec *s, ap_dbd_t *sql)
{
    svr_cfg *svr = ap_get_module_config(s->module_config, &dbd_module);
    if (!svr->persist) {
        apr_dbd_close(sql->driver, sql->handle);
    }
#if APR_HAS_THREADS
    else {
        apr_reslist_release(svr->dbpool, sql);
    }
#endif
}
static apr_status_t dbd_close(void *CONN)
{
    ap_dbd_t *conn = CONN;
    return apr_dbd_close(conn->driver, conn->handle);
}
#if APR_HAS_THREADS
typedef struct {
    ap_dbd_t *conn;
    apr_reslist_t *dbpool;
} dbd_pool_rec;
static apr_status_t dbd_release(void *REQ)
{
    dbd_pool_rec *req = REQ;
    apr_reslist_release(req->dbpool, req->conn);
    return APR_SUCCESS;
}
ap_dbd_t *ap_dbd_acquire(request_rec *r)
{
    svr_cfg *svr;
    dbd_pool_rec *req = ap_get_module_config(r->request_config, &dbd_module);
    if (!req) {
        req = apr_palloc(r->pool, sizeof(dbd_pool_rec));
        req->conn = ap_dbd_open(r->pool, r->server);
        if (req->conn) {
            svr = ap_get_module_config(r->server->module_config, &dbd_module);
            ap_set_module_config(r->request_config, &dbd_module, req);
            if (svr->persist) {
                req->dbpool = svr->dbpool;
                apr_pool_cleanup_register(r->pool, req, dbd_release,
                                          apr_pool_cleanup_null);
            }
            else {
                apr_pool_cleanup_register(r->pool, req->conn, dbd_close,
                                          apr_pool_cleanup_null);
            }
        }
    }
    return req->conn;
}
#else
ap_dbd_t *ap_dbd_acquire(request_rec *r)
{
    svr_cfg *svr;
    ap_dbd_t *ret = ap_get_module_config(r->request_config, &dbd_module);
    if (!ret) {
        svr = ap_get_module_config(r->server->module_config, &dbd_module);
        ret = ap_dbd_open(r->pool, r->server);
        if ( ret ) {
            ap_set_module_config(r->request_config, &dbd_module, ret);
            if (!svr->persist) {
                apr_pool_cleanup_register(r->pool, svr->conn, dbd_close,
                                          apr_pool_cleanup_null);
            }
            /* if persist then dbd_open registered cleanup on proc pool */
        }
    }
    return ret;
}
#endif

static void dbd_hooks(apr_pool_t *pool)
{
#if APR_HAS_THREADS
    ap_hook_child_init((void*)dbd_setup, NULL, NULL, APR_HOOK_MIDDLE);
#endif
    APR_REGISTER_OPTIONAL_FN(ap_dbd_open);
    APR_REGISTER_OPTIONAL_FN(ap_dbd_close);
    APR_REGISTER_OPTIONAL_FN(ap_dbd_acquire);
    APR_REGISTER_OPTIONAL_FN(ap_dbd_prepare);
    apr_dbd_init(pool);
}

module AP_MODULE_DECLARE_DATA dbd_module = {
    STANDARD20_MODULE_STUFF,
    NULL,
    NULL,
    dbd_cfg,
    dbd_merge,
    dbd_cmds,
    dbd_hooks
};