From 181cec4348da40331b3e8ab365732c025ec149b2 Mon Sep 17 00:00:00 2001 From: Reinhard Tartler Date: Wed, 27 Apr 2011 19:24:15 +0200 Subject: Import upstream version 0.11.0~svn1143 --- plugin_mysql.c | 243 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 243 insertions(+) create mode 100644 plugin_mysql.c (limited to 'plugin_mysql.c') diff --git a/plugin_mysql.c b/plugin_mysql.c new file mode 100644 index 0000000..ca7b2d4 --- /dev/null +++ b/plugin_mysql.c @@ -0,0 +1,243 @@ +/* $Id: plugin_mysql.c 771 2007-02-25 12:27:26Z michael $ + * $URL: https://ssl.bulix.org/svn/lcd4linux/trunk/plugin_mysql.c $ + * + * plugin for execute SQL queries into a MySQL DBSM. + * + * Copyright (C) 2004 Javier Garcia + * Copyright (C) 2004 The LCD4Linux Team + * + * This file is part of LCD4Linux. + * + * LCD4Linux is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2, or (at your option) + * any later version. + * + * LCD4Linux is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. + * + */ + +/* DOC: + * To compile the plugin remember to add -lmysqlclient option into Makefile. I.E: CC = gcc -lmysqlclient + * or run: + * $ gcc -I/usr/include/mysql -L/usr/lib/mysql plugin_mysql.c -lmysqlclient -o plugin_mysql.o + * + * exported functions: + * + * int plugin_init_mysql (void) + * + * adds various functions: + * MySQLquery(query) + * Returns the number of rows in query. + * MySQLstatus() + * Returns the current server status: + * Uptime in seconds and the number of running threads, + * questions, reloads, and open tables. + * + */ + +#include "config.h" +#include +#include +#include +#include "debug.h" +#include "plugin.h" +#include "cfg.h" + +#ifdef HAVE_MYSQL_MYSQL_H +#include +#else +#warning mysql/mysql.h not found: plugin deactivated +#endif + +#ifdef WITH_DMALLOC +#include +#endif + + +#ifdef HAVE_MYSQL_MYSQL_H +static MYSQL conex; + +static char Section[] = "Plugin:MySQL"; + + +static int configure_mysql(void) +{ + static int configured = 0; + + char server[256]; + int port; + char user[128]; + char password[256]; + char database[256]; + char *s; + + if (configured != 0) + return configured; + + s = cfg_get(Section, "server", "localhost"); + if (*s == '\0') { + info("[MySQL] empty '%s.server' entry from %s, assuming 'localhost'", Section, cfg_source()); + strcpy(server, "localhost"); + } else + strcpy(server, s); + free(s); + + if (cfg_number(Section, "port", 0, 1, 65536, &port) < 1) { + /* using 0 as default port because mysql_real_connect() will convert it to real default one */ + info("[MySQL] no '%s.port' entry from %s using MySQL's default", Section, cfg_source()); + } + + s = cfg_get(Section, "user", ""); + if (*s == '\0') { + /* If user is NULL or the empty string "", the lcd4linux Unix user is assumed. */ + info("[MySQL] empty '%s.user' entry from %s, assuming lcd4linux owner", Section, cfg_source()); + strcpy(user, ""); + } else + strcpy(user, s); + free(s); + + s = cfg_get(Section, "password", ""); + /* Do not encrypt the password because encryption is handled automatically by the MySQL client API. */ + if (*s == '\0') { + info("[MySQL] empty '%s.password' entry in %s, assuming none", Section, cfg_source()); + strcpy(password, ""); + } else + strcpy(password, s); + free(s); + + s = cfg_get(Section, "database", ""); + if (*s == '\0') { + error("[MySQL] no '%s:database' entry from %s, specify one", Section, cfg_source()); + free(s); + configured = -1; + return configured; + } + strcpy(database, s); + free(s); + + mysql_init(&conex); + if (!mysql_real_connect(&conex, server, user, password, database, port, NULL, 0)) { + error("[MySQL] conection error: %s", mysql_error(&conex)); + configured = -1; + return configured; + } + + configured = 1; + return configured; +} + +static void my_MySQLcount(RESULT * result, RESULT * query) +{ + char *q; + double value; + MYSQL_RES *res; + + if (configure_mysql() < 0) { + value = -1; + SetResult(&result, R_NUMBER, &value); + return; + } + + q = R2S(query); + + /* mysql_ping(MYSQL *mysql) checks whether the connection to the server is working. */ + /* If it has gone down, an automatic reconnection is attempted. */ + mysql_ping(&conex); + if (mysql_real_query(&conex, q, (unsigned int) strlen(q))) { + error("[MySQL] query error: %s", mysql_error(&conex)); + value = -1; + } else { + /* We don't use res=mysql_use_result(); because mysql_num_rows() will not */ + /* return the correct value until all the rows in the result set have been retrieved */ + /* with mysql_fetch_row(), so we use res=mysql_store_result(); instead */ + res = mysql_store_result(&conex); + value = (double) mysql_num_rows(res); + mysql_free_result(res); + } + + SetResult(&result, R_NUMBER, &value); +} + + +static void my_MySQLquery(RESULT * result, RESULT * query) +{ + char *q; + double value; + MYSQL_RES *res; + MYSQL_ROW row = NULL; + + if (configure_mysql() < 0) { + value = -1; + SetResult(&result, R_NUMBER, &value); + return; + } + + q = R2S(query); + + /* mysql_ping(MYSQL *mysql) checks whether the connection to the server is working. */ + /* If it has gone down, an automatic reconnection is attempted. */ + mysql_ping(&conex); + if (mysql_real_query(&conex, q, (unsigned int) strlen(q))) { + error("[MySQL] query error: %s", mysql_error(&conex)); + value = -1; + } else { + /* We don't use res=mysql_use_result(); because mysql_num_rows() will not */ + /* return the correct value until all the rows in the result set have been retrieved */ + /* with mysql_fetch_row(), so we use res=mysql_store_result(); instead */ + res = mysql_store_result(&conex); + row = mysql_fetch_row(res); + mysql_free_result(res); + } + + SetResult(&result, R_STRING, row[0]); +} + + +static void my_MySQLstatus(RESULT * result) +{ + const char *value = ""; + const char *status; + + if (configure_mysql() > 0) { + + mysql_ping(&conex); + status = mysql_stat(&conex); + if (!status) { + error("[MySQL] status error: %s", mysql_error(&conex)); + value = "error"; + } else { + value = status; + } + } + + SetResult(&result, R_STRING, value); +} + + +#endif + + +int plugin_init_mysql(void) +{ +#ifdef HAVE_MYSQL_MYSQL_H + AddFunction("MySQL::count", 1, my_MySQLcount); + AddFunction("MySQL::query", 1, my_MySQLquery); + AddFunction("MySQL::status", 0, my_MySQLstatus); +#endif + return 0; +} + +void plugin_exit_mysql(void) +{ +#ifdef HAVE_MYSQL_MYSQL_H + mysql_close(&conex); +#endif +} -- cgit v1.2.3