lighttpd 1.4.x https://www.lighttpd.net/
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.
 
 
 
 
 
 

304 lines
9.6 KiB

/*
* mod_vhostdb_mysql - virtual hosts mapping from backend MySQL/MariaDB database
*
* Copyright(c) 2017 Glenn Strauss gstrauss()gluelogic.com All rights reserved
* License: BSD 3-clause (same as lighttpd)
*/
#include "first.h"
#include <string.h>
#include <stdlib.h>
#include <mysql.h>
#include "mod_vhostdb_api.h"
#include "base.h"
#include "fdevent.h"
#include "log.h"
#include "plugin.h"
/*
* virtual host plugin using MySQL/MariaDB for domain to directory lookups
*/
typedef struct {
MYSQL *dbconn;
const buffer *sqlquery;
} vhostdb_config;
typedef struct {
void *vdata;
} plugin_config;
typedef struct {
PLUGIN_DATA;
plugin_config defaults;
plugin_config conf;
} plugin_data;
static void mod_vhostdb_dbconf_free (void *vdata)
{
vhostdb_config *dbconf = (vhostdb_config *)vdata;
if (!dbconf) return;
mysql_close(dbconf->dbconn);
free(dbconf);
}
static int mod_vhostdb_dbconf_setup (server *srv, const array *opts, void **vdata)
{
const buffer *sqlquery = NULL;
const char *dbname=NULL, *user=NULL, *pass=NULL, *host=NULL, *sock=NULL;
unsigned int port = 0;
for (size_t i = 0; i < opts->used; ++i) {
const data_string *ds = (data_string *)opts->data[i];
if (ds->type == TYPE_STRING && !buffer_is_blank(&ds->value)) {
if (buffer_is_equal_caseless_string(&ds->key, CONST_STR_LEN("sql"))) {
sqlquery = &ds->value;
} else if (buffer_is_equal_caseless_string(&ds->key, CONST_STR_LEN("dbname"))) {
dbname = ds->value.ptr;
} else if (buffer_is_equal_caseless_string(&ds->key, CONST_STR_LEN("user"))) {
user = ds->value.ptr;
} else if (buffer_is_equal_caseless_string(&ds->key, CONST_STR_LEN("password"))) {
pass = ds->value.ptr;
} else if (buffer_is_equal_caseless_string(&ds->key, CONST_STR_LEN("host"))) {
host = ds->value.ptr;
} else if (buffer_is_equal_caseless_string(&ds->key, CONST_STR_LEN("port"))) {
port = strtoul(ds->value.ptr, NULL, 10);
} else if (buffer_is_equal_caseless_string(&ds->key, CONST_STR_LEN("sock"))) {
sock = ds->value.ptr;
}
}
}
/* required:
* - sql (sql query)
* - dbname
* - user
*
* optional:
* - password, default: empty
* - socket, default: mysql default
* - hostname, if set overrides socket
* - port, default: 3306
*/
if (NULL != sqlquery && !buffer_is_blank(sqlquery)
&& dbname && *dbname && user && *user) {
vhostdb_config *dbconf;
MYSQL *dbconn = mysql_init(NULL);
if (NULL == dbconn) {
log_error(srv->errh, __FILE__, __LINE__,
"mysql_init() failed, exiting...");
return -1;
}
#if MYSQL_VERSION_ID >= 50013
/* in mysql versions above 5.0.3 the reconnect flag is off by default */
{
char reconnect = 1;
mysql_options(dbconn, MYSQL_OPT_RECONNECT, &reconnect);
}
#endif
/* CLIENT_MULTI_STATEMENTS first appeared in 4.1 */
#if MYSQL_VERSION_ID < 40100
#ifndef CLIENT_MULTI_STATEMENTS
#define CLIENT_MULTI_STATEMENTS 0
#endif
#endif
if (!mysql_real_connect(dbconn, host, user, pass, dbname, port, sock,
CLIENT_MULTI_STATEMENTS)) {
log_error(srv->errh, __FILE__, __LINE__, "%s", mysql_error(dbconn));
mysql_close(dbconn);
return -1;
}
fdevent_setfd_cloexec(dbconn->net.fd);
dbconf = (vhostdb_config *)calloc(1, sizeof(*dbconf));
dbconf->dbconn = dbconn;
dbconf->sqlquery = sqlquery;
*vdata = dbconf;
}
return 0;
}
static void mod_vhostdb_patch_config (request_st * const r, plugin_data * const p);
static int mod_vhostdb_mysql_query(request_st * const r, void *p_d, buffer *docroot)
{
plugin_data *p = (plugin_data *)p_d;
vhostdb_config *dbconf;
unsigned cols;
MYSQL_ROW row;
MYSQL_RES *result;
/*(reuse buffer for sql query before generating docroot result)*/
buffer *sqlquery = docroot;
buffer_clear(sqlquery); /*(also resets docroot (alias))*/
mod_vhostdb_patch_config(r, p);
if (NULL == p->conf.vdata) return 0; /*(after resetting docroot)*/
dbconf = (vhostdb_config *)p->conf.vdata;
for (char *b = dbconf->sqlquery->ptr, *d; *b; b = d+1) {
if (NULL != (d = strchr(b, '?'))) {
/* escape the uri.authority */
unsigned long len;
buffer_append_string_len(sqlquery, b, (size_t)(d - b));
buffer_string_prepare_append(sqlquery,
buffer_clen(&r->uri.authority) * 2);
len = mysql_real_escape_string(dbconf->dbconn,
sqlquery->ptr + buffer_clen(sqlquery),
BUF_PTR_LEN(&r->uri.authority));
if ((unsigned long)~0 == len) return -1;
buffer_commit(sqlquery, len);
} else {
d = dbconf->sqlquery->ptr + buffer_clen(dbconf->sqlquery);
buffer_append_string_len(sqlquery, b, (size_t)(d - b));
break;
}
}
if (mysql_real_query(dbconf->dbconn, BUF_PTR_LEN(sqlquery))) {
log_error(r->conf.errh, __FILE__, __LINE__, "%s",
mysql_error(dbconf->dbconn));
buffer_clear(docroot); /*(reset buffer; no result)*/
return -1;
}
buffer_clear(docroot); /*(reset buffer to store result)*/
result = mysql_store_result(dbconf->dbconn);
cols = mysql_num_fields(result);
row = mysql_fetch_row(result);
if (row && cols >= 1) {
buffer_copy_string(docroot, row[0]);
} /* else no such virtual host */
mysql_free_result(result);
#if MYSQL_VERSION_ID >= 40100
while (0 == mysql_next_result(dbconf->dbconn)) ;
#endif
return 0;
}
INIT_FUNC(mod_vhostdb_init) {
static http_vhostdb_backend_t http_vhostdb_backend_mysql =
{ "mysql", mod_vhostdb_mysql_query, NULL };
plugin_data *p = calloc(1, sizeof(*p));
/* register http_vhostdb_backend_mysql */
http_vhostdb_backend_mysql.p_d = p;
http_vhostdb_backend_set(&http_vhostdb_backend_mysql);
return p;
}
FREE_FUNC(mod_vhostdb_cleanup) {
plugin_data * const p = p_d;
if (NULL == p->cvlist) return;
/* (init i to 0 if global context; to 1 to skip empty global context) */
for (int i = !p->cvlist[0].v.u2[1], used = p->nconfig; i < used; ++i) {
config_plugin_value_t *cpv = p->cvlist + p->cvlist[i].v.u2[0];
for (; -1 != cpv->k_id; ++cpv) {
if (cpv->vtype != T_CONFIG_LOCAL || NULL == cpv->v.v) continue;
switch (cpv->k_id) {
case 0: /* vhostdb.<db> */
mod_vhostdb_dbconf_free(cpv->v.v);
break;
default:
break;
}
}
}
}
static void mod_vhostdb_merge_config_cpv(plugin_config * const pconf, const config_plugin_value_t * const cpv) {
switch (cpv->k_id) { /* index into static config_plugin_keys_t cpk[] */
case 0: /* vhostdb.<db> */
if (cpv->vtype == T_CONFIG_LOCAL)
pconf->vdata = cpv->v.v;
break;
default:/* should not happen */
return;
}
}
static void mod_vhostdb_merge_config(plugin_config * const pconf, const config_plugin_value_t *cpv) {
do {
mod_vhostdb_merge_config_cpv(pconf, cpv);
} while ((++cpv)->k_id != -1);
}
static void mod_vhostdb_patch_config(request_st * const r, plugin_data * const p) {
p->conf = p->defaults; /* copy small struct instead of memcpy() */
/*memcpy(&p->conf, &p->defaults, sizeof(plugin_config));*/
for (int i = 1, used = p->nconfig; i < used; ++i) {
if (config_check_cond(r, (uint32_t)p->cvlist[i].k_id))
mod_vhostdb_merge_config(&p->conf,p->cvlist + p->cvlist[i].v.u2[0]);
}
}
SETDEFAULTS_FUNC(mod_vhostdb_set_defaults) {
static const config_plugin_keys_t cpk[] = {
{ CONST_STR_LEN("vhostdb.mysql"),
T_CONFIG_ARRAY_KVSTRING,
T_CONFIG_SCOPE_CONNECTION }
,{ NULL, 0,
T_CONFIG_UNSET,
T_CONFIG_SCOPE_UNSET }
};
plugin_data * const p = p_d;
if (!config_plugin_values_init(srv, p, cpk, "mod_vhostdb_mysql"))
return HANDLER_ERROR;
/* process and validate config directives
* (init i to 0 if global context; to 1 to skip empty global context) */
for (int i = !p->cvlist[0].v.u2[1]; i < p->nconfig; ++i) {
config_plugin_value_t *cpv = p->cvlist + p->cvlist[i].v.u2[0];
for (; -1 != cpv->k_id; ++cpv) {
switch (cpv->k_id) {
case 0: /* vhostdb.<db> */
if (cpv->v.a->used) {
if (0 != mod_vhostdb_dbconf_setup(srv, cpv->v.a, &cpv->v.v))
return HANDLER_ERROR;
if (NULL != cpv->v.v)
cpv->vtype = T_CONFIG_LOCAL;
}
break;
default:/* should not happen */
break;
}
}
}
/* initialize p->defaults from global config context */
if (p->nconfig > 0 && p->cvlist->v.u2[1]) {
const config_plugin_value_t *cpv = p->cvlist + p->cvlist->v.u2[0];
if (-1 != cpv->k_id)
mod_vhostdb_merge_config(&p->defaults, cpv);
}
return HANDLER_GO_ON;
}
int mod_vhostdb_mysql_plugin_init (plugin *p);
int mod_vhostdb_mysql_plugin_init (plugin *p)
{
p->version = LIGHTTPD_VERSION_ID;
p->name = "vhostdb_mysql";
p->init = mod_vhostdb_init;
p->cleanup = mod_vhostdb_cleanup;
p->set_defaults = mod_vhostdb_set_defaults;
return 0;
}