1 files changed, 380 insertions, 0 deletions
diff --git a/data/lighttpd/lighttpd-1.4.53/src/mod_mysql_vhost.c b/data/lighttpd/lighttpd-1.4.53/src/mod_mysql_vhost.c
new file mode 100644
index 000000000..2af5cc2ba
--- /dev/null
+++ b/data/lighttpd/lighttpd-1.4.53/src/mod_mysql_vhost.c
@@ -0,0 +1,380 @@
+#include "first.h"
+
+#include <unistd.h>
+#include <errno.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include <mysql.h>
+
+#include "base.h"
+#include "plugin.h"
+#include "fdevent.h"
+#include "log.h"
+
+#include "stat_cache.h"
+
+/*
+ * Plugin for lighttpd to use MySQL
+ *   for domain to directory lookups,
+ *   i.e virtual hosts (vhosts).
+ *
+ * /ada@riksnet.se 2004-12-06
+ */
+
+typedef struct {
+	MYSQL 	*mysql;
+	buffer  *mysql_query;
+
+	buffer  *mydb;
+	buffer  *myuser;
+	buffer  *mypass;
+	buffer  *mysock;
+
+	buffer  *hostname;
+	unsigned short port;
+} plugin_config;
+
+/* global plugin data */
+typedef struct {
+	PLUGIN_DATA;
+
+	buffer 	*tmp_buf;
+
+	plugin_config **config_storage;
+
+	plugin_config conf;
+} plugin_data;
+
+/* per connection plugin data */
+typedef struct {
+	buffer	*server_name;
+	buffer	*document_root;
+} plugin_connection_data;
+
+/* init the plugin data */
+INIT_FUNC(mod_mysql_vhost_init) {
+	plugin_data *p;
+
+	p = calloc(1, sizeof(*p));
+
+	p->tmp_buf = buffer_init();
+
+	return p;
+}
+
+/* cleanup the plugin data */
+SERVER_FUNC(mod_mysql_vhost_cleanup) {
+	plugin_data *p = p_d;
+
+	UNUSED(srv);
+
+	if (!p) return HANDLER_GO_ON;
+
+	if (p->config_storage) {
+		size_t i;
+		for (i = 0; i < srv->config_context->used; i++) {
+			plugin_config *s = p->config_storage[i];
+
+			if (!s) continue;
+
+			mysql_close(s->mysql);
+
+			buffer_free(s->mysql_query);
+			buffer_free(s->mydb);
+			buffer_free(s->myuser);
+			buffer_free(s->mypass);
+			buffer_free(s->mysock);
+			buffer_free(s->hostname);
+
+			free(s);
+		}
+		free(p->config_storage);
+	}
+	buffer_free(p->tmp_buf);
+
+	free(p);
+
+	return HANDLER_GO_ON;
+}
+
+/* handle the plugin per connection data */
+static void* mod_mysql_vhost_connection_data(server *srv, connection *con, void *p_d)
+{
+	plugin_data *p = p_d;
+	plugin_connection_data *c = con->plugin_ctx[p->id];
+
+	UNUSED(srv);
+
+	if (c) return c;
+	c = calloc(1, sizeof(*c));
+
+	c->server_name = buffer_init();
+	c->document_root = buffer_init();
+
+	return con->plugin_ctx[p->id] = c;
+}
+
+/* destroy the plugin per connection data */
+CONNECTION_FUNC(mod_mysql_vhost_handle_connection_reset) {
+	plugin_data *p = p_d;
+	plugin_connection_data *c = con->plugin_ctx[p->id];
+
+	UNUSED(srv);
+
+	if (!c) return HANDLER_GO_ON;
+
+	buffer_free(c->server_name);
+	buffer_free(c->document_root);
+
+	free(c);
+
+	con->plugin_ctx[p->id] = NULL;
+	return HANDLER_GO_ON;
+}
+
+/* set configuration values */
+SERVER_FUNC(mod_mysql_vhost_set_defaults) {
+	plugin_data *p = p_d;
+	size_t i = 0;
+
+	config_values_t cv[] = {
+		{ "mysql-vhost.db",       NULL, T_CONFIG_STRING, T_CONFIG_SCOPE_CONNECTION }, /* 0 */
+		{ "mysql-vhost.user",     NULL, T_CONFIG_STRING, T_CONFIG_SCOPE_CONNECTION }, /* 1 */
+		{ "mysql-vhost.pass",     NULL, T_CONFIG_STRING, T_CONFIG_SCOPE_CONNECTION }, /* 2 */
+		{ "mysql-vhost.sock",     NULL, T_CONFIG_STRING, T_CONFIG_SCOPE_CONNECTION }, /* 3 */
+		{ "mysql-vhost.sql",      NULL, T_CONFIG_STRING, T_CONFIG_SCOPE_CONNECTION }, /* 4 */
+		{ "mysql-vhost.hostname", NULL, T_CONFIG_STRING, T_CONFIG_SCOPE_CONNECTION }, /* 5 */
+		{ "mysql-vhost.port",     NULL, T_CONFIG_SHORT,  T_CONFIG_SCOPE_CONNECTION }, /* 6 */
+		{ NULL,                   NULL, T_CONFIG_UNSET,  T_CONFIG_SCOPE_UNSET      }
+	};
+
+	p->config_storage = calloc(1, srv->config_context->used * sizeof(plugin_config *));
+
+	for (i = 0; i < srv->config_context->used; i++) {
+		data_config const* config = (data_config const*)srv->config_context->data[i];
+		plugin_config *s;
+
+		s = calloc(1, sizeof(plugin_config));
+		s->mysql_query = buffer_init();
+		s->mydb = buffer_init();
+		s->myuser = buffer_init();
+		s->mypass = buffer_init();
+		s->mysock = buffer_init();
+		s->hostname = buffer_init();
+		s->port = 0;               /* default port for mysql */
+		s->mysql = NULL;
+
+		cv[0].destination = s->mydb;
+		cv[1].destination = s->myuser;
+		cv[2].destination = s->mypass;
+		cv[3].destination = s->mysock;
+		cv[4].destination = s->mysql_query;
+		cv[5].destination = s->hostname;
+		cv[6].destination = &(s->port);
+
+		p->config_storage[i] = s;
+
+		if (config_insert_values_global(srv, config->value, cv, i == 0 ? T_CONFIG_SCOPE_SERVER : T_CONFIG_SCOPE_CONNECTION)) {
+			return HANDLER_ERROR;
+		}
+
+		/* required:
+		 * - username
+		 * - database
+		 *
+		 * optional:
+		 * - password, default: empty
+		 * - socket, default: mysql default
+		 * - hostname, if set overrides socket
+		 * - port, default: 3306
+		 */
+
+		/* all have to be set */
+		if (!(buffer_string_is_empty(s->myuser) ||
+		      buffer_string_is_empty(s->mydb))) {
+
+			if (NULL == (s->mysql = mysql_init(NULL))) {
+				log_error_write(srv, __FILE__, __LINE__, "s", "mysql_init() failed, exiting...");
+				return HANDLER_ERROR;
+			}
+
+#if MYSQL_VERSION_ID >= 50013
+			/* in mysql versions above 5.0.3 the reconnect flag is off by default */
+			{
+				char reconnect = 1;
+				mysql_options(s->mysql, MYSQL_OPT_RECONNECT, &reconnect);
+			}
+#endif
+
+#define FOO(x) (buffer_string_is_empty(s->x) ? NULL : s->x->ptr)
+
+#if MYSQL_VERSION_ID >= 40100
+			/* CLIENT_MULTI_STATEMENTS first appeared in 4.1 */ 
+			if (!mysql_real_connect(s->mysql, FOO(hostname), FOO(myuser), FOO(mypass),
+						FOO(mydb), s->port, FOO(mysock), CLIENT_MULTI_STATEMENTS)) {
+#else
+			if (!mysql_real_connect(s->mysql, FOO(hostname), FOO(myuser), FOO(mypass),
+						FOO(mydb), s->port, FOO(mysock), 0)) {
+#endif
+				log_error_write(srv, __FILE__, __LINE__, "s", mysql_error(s->mysql));
+				return HANDLER_ERROR;
+			}
+#undef FOO
+
+			fdevent_setfd_cloexec(s->mysql->net.fd);
+		}
+	}
+
+	return HANDLER_GO_ON;
+}
+
+#define PATCH(x) \
+	p->conf.x = s->x;
+static int mod_mysql_vhost_patch_connection(server *srv, connection *con, plugin_data *p) {
+	size_t i, j;
+	plugin_config *s = p->config_storage[0];
+
+	PATCH(mysql_query);
+	PATCH(mysql);
+
+	/* skip the first, the global context */
+	for (i = 1; i < srv->config_context->used; i++) {
+		data_config *dc = (data_config *)srv->config_context->data[i];
+		s = p->config_storage[i];
+
+		/* condition didn't match */
+		if (!config_check_cond(srv, con, dc)) continue;
+
+		/* merge config */
+		for (j = 0; j < dc->value->used; j++) {
+			data_unset *du = dc->value->data[j];
+
+			if (buffer_is_equal_string(du->key, CONST_STR_LEN("mysql-vhost.sql"))) {
+				PATCH(mysql_query);
+			}
+		}
+
+		if (s->mysql) {
+			PATCH(mysql);
+		}
+	}
+
+	return 0;
+}
+#undef PATCH
+
+
+/* handle document root request */
+CONNECTION_FUNC(mod_mysql_vhost_handle_docroot) {
+	plugin_data *p = p_d;
+	plugin_connection_data *c;
+	stat_cache_entry *sce;
+
+	unsigned  cols;
+	MYSQL_ROW row;
+	MYSQL_RES *result = NULL;
+
+	/* no host specified? */
+	if (buffer_string_is_empty(con->uri.authority)) return HANDLER_GO_ON;
+
+	mod_mysql_vhost_patch_connection(srv, con, p);
+
+	if (!p->conf.mysql) return HANDLER_GO_ON;
+	if (buffer_string_is_empty(p->conf.mysql_query)) return HANDLER_GO_ON;
+
+	/* sets up connection data if not done yet */
+	c = mod_mysql_vhost_connection_data(srv, con, p_d);
+
+	/* check if cached this connection */
+	if (buffer_is_equal(c->server_name, con->uri.authority)) goto GO_ON;
+
+	/* build and run SQL query */
+	buffer_clear(p->tmp_buf);
+	for (char *b = p->conf.mysql_query->ptr, *d; *b; b = d+1) {
+		if (NULL != (d = strchr(b, '?'))) {
+			/* escape the uri.authority */
+			unsigned long to_len;
+			buffer_append_string_len(p->tmp_buf, b, (size_t)(d - b));
+			buffer_string_prepare_append(p->tmp_buf, buffer_string_length(con->uri.authority) * 2);
+			to_len = mysql_real_escape_string(p->conf.mysql,
+					p->tmp_buf->ptr + buffer_string_length(p->tmp_buf),
+					CONST_BUF_LEN(con->uri.authority));
+			if ((unsigned long)~0 == to_len) goto ERR500;
+			buffer_commit(p->tmp_buf, to_len);
+		} else {
+			d = p->conf.mysql_query->ptr + buffer_string_length(p->conf.mysql_query);
+			buffer_append_string_len(p->tmp_buf, b, (size_t)(d - b));
+			break;
+		}
+	}
+	if (mysql_real_query(p->conf.mysql, CONST_BUF_LEN(p->tmp_buf))) {
+		log_error_write(srv, __FILE__, __LINE__, "s", mysql_error(p->conf.mysql));
+		goto ERR500;
+	}
+	result = mysql_store_result(p->conf.mysql);
+	cols = mysql_num_fields(result);
+	row = mysql_fetch_row(result);
+	if (!row || cols < 1) {
+		/* no such virtual host */
+		mysql_free_result(result);
+#if MYSQL_VERSION_ID >= 40100
+		while (mysql_next_result(p->conf.mysql) == 0);
+#endif
+		return HANDLER_GO_ON;
+	}
+
+	/* sanity check that really is a directory */
+	buffer_copy_string(p->tmp_buf, row[0]);
+	buffer_append_slash(p->tmp_buf);
+
+	if (HANDLER_ERROR == stat_cache_get_entry(srv, con, p->tmp_buf, &sce)) {
+		log_error_write(srv, __FILE__, __LINE__, "sb", strerror(errno), p->tmp_buf);
+		goto ERR500;
+	}
+	if (!S_ISDIR(sce->st.st_mode)) {
+		log_error_write(srv, __FILE__, __LINE__, "sb", "Not a directory", p->tmp_buf);
+		goto ERR500;
+	}
+
+	/* cache the data */
+	buffer_copy_buffer(c->server_name, con->uri.authority);
+	buffer_copy_buffer(c->document_root, p->tmp_buf);
+
+	mysql_free_result(result);
+#if MYSQL_VERSION_ID >= 40100
+	while (mysql_next_result(p->conf.mysql) == 0);
+#endif
+
+	/* fix virtual server and docroot */
+GO_ON:
+	buffer_copy_buffer(con->server_name, c->server_name);
+	buffer_copy_buffer(con->physical.doc_root, c->document_root);
+
+	return HANDLER_GO_ON;
+
+ERR500:
+	if (result) mysql_free_result(result);
+#if MYSQL_VERSION_ID >= 40100
+	while (mysql_next_result(p->conf.mysql) == 0);
+#endif
+	con->http_status = 500; /* Internal Error */
+	con->mode = DIRECT;
+	return HANDLER_FINISHED;
+}
+
+/* this function is called at dlopen() time and inits the callbacks */
+int mod_mysql_vhost_plugin_init(plugin *p);
+int mod_mysql_vhost_plugin_init(plugin *p) {
+	p->version        = LIGHTTPD_VERSION_ID;
+	p->name           = buffer_init_string("mysql_vhost");
+
+	p->init           = mod_mysql_vhost_init;
+	p->cleanup        = mod_mysql_vhost_cleanup;
+	p->connection_reset = mod_mysql_vhost_handle_connection_reset;
+
+	p->set_defaults   = mod_mysql_vhost_set_defaults;
+	p->handle_docroot = mod_mysql_vhost_handle_docroot;
+
+	return 0;
+}