1 /* mysqlConn.c - NeonServ v5.6
2 * Copyright (C) 2011-2012 Philipp Kreil (pk910)
4 * This program is free software: you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation, either version 3 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program. If not, see <http://www.gnu.org/licenses/>.
18 #include "mysqlConn.h"
19 #include "ConfigParser.h"
22 #define DATABASE_VERSION "20"
24 static void show_mysql_error();
26 struct mysql_conn_struct {
29 struct used_result *used_results;
30 struct escaped_string *escaped_strings;
31 struct mysql_conn_struct *next;
36 struct used_result *next;
39 struct escaped_string {
41 struct escaped_string *next;
44 struct mysql_conn_struct *get_mysql_conn_struct();
46 struct mysql_conn_struct *mysql_conns = NULL;
47 static int mysql_serverport;
48 static char *mysql_host, *mysql_user, *mysql_pass, *mysql_base;
51 static pthread_mutex_t synchronized;
54 static void check_mysql() {
55 MYSQL *mysql_conn = get_mysql_conn();
57 if((errid = mysql_ping(mysql_conn))) {
58 if(mysql_errno(mysql_conn) == CR_SERVER_GONE_ERROR) {
59 if(!mysql_real_connect(mysql_conn, mysql_host, mysql_user, mysql_pass, mysql_base, mysql_serverport, NULL, 0)) {
69 MYSQL_RES *mysql_use() {
70 struct mysql_conn_struct *mysql_conn = get_mysql_conn_struct();
71 MYSQL_RES *res = mysql_store_result(mysql_conn->mysql_conn);
72 struct used_result *result = malloc(sizeof(*result));
74 mysql_free_result(res);
78 result->next = mysql_conn->used_results;
79 mysql_conn->used_results = result;
84 struct mysql_conn_struct *mysql_conn = get_mysql_conn_struct();
85 if(!mysql_conn) return;
86 struct used_result *result, *next_result;
87 for(result = mysql_conn->used_results; result; result = next_result) {
88 next_result = result->next;
89 mysql_free_result(result->result);
92 mysql_conn->used_results = NULL;
93 struct escaped_string *escaped, *next_escaped;
94 for(escaped = mysql_conn->escaped_strings; escaped; escaped = next_escaped) {
95 next_escaped = escaped->next;
96 free(escaped->string);
99 mysql_conn->escaped_strings = NULL;
103 char *new_mysql_host = get_string_field("MySQL.host");
104 char *new_mysql_user = get_string_field("MySQL.user");
105 char *new_mysql_pass = get_string_field("MySQL.pass");
106 char *new_mysql_base = get_string_field("MySQL.base");
107 if(!(new_mysql_host && new_mysql_user && new_mysql_pass && new_mysql_base))
113 mysql_host = strdup(new_mysql_host);
117 mysql_user = strdup(new_mysql_user);
121 mysql_pass = strdup(new_mysql_pass);
125 mysql_base = strdup(new_mysql_base);
127 mysql_serverport = get_int_field("MySQL.port");
128 if(!mysql_serverport)
129 mysql_serverport = 3306;
134 THREAD_MUTEX_INIT(synchronized);
136 MYSQL *mysql_conn = get_mysql_conn();
138 //check database version...
140 if(!mysql_query(mysql_conn, "SELECT `database_version` FROM `version`")) {
141 MYSQL_RES *res = mysql_use();
143 if((row = mysql_fetch_row(res))) {
144 version = atoi(row[0]);
149 FILE *f = fopen("database.sql", "r");
150 mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_ON);
153 char query_buffer[8192];
154 int query_buffer_pos = 0;
155 while (fgets(line, sizeof(line), f)) {
156 query_buffer_pos += sprintf(query_buffer + query_buffer_pos, " %s", line);
157 if(line[(strlen(line) - 2)] == ';') {
158 if(mysql_query(mysql_conn, query_buffer))
160 query_buffer_pos = 0;
165 printf_log("main", LOG_ERROR | LOG_MYSQL, "File not found: database.sql");
166 f = fopen("database.defaults.sql", "r");
169 char query_buffer[131072];
170 int query_buffer_pos = 0;
171 while (fgets(line, sizeof(line), f)) {
172 query_buffer_pos += sprintf(query_buffer + query_buffer_pos, " %s", line);
173 if(line[(strlen(line) - 2)] == ';') {
174 if(mysql_query(mysql_conn, query_buffer))
176 query_buffer_pos = 0;
181 printf_log("main", LOG_ERROR | LOG_MYSQL, "File not found: database.defaults.sql");
183 MYSQL_RES *res = mysql_store_result(mysql_conn);
184 mysql_free_result(res);
185 } while(!mysql_next_result(mysql_conn));
186 mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_OFF);
187 mysql_query(mysql_conn, "INSERT INTO `version` (`database_version`) VALUES ('" DATABASE_VERSION "')");
189 else if(version < atoi(DATABASE_VERSION)) {
191 FILE *f = fopen("database.upgrade.sql", "r");
192 mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_ON);
195 char query_buffer[8192];
196 int query_buffer_pos = 0, use_querys = 0;
197 sprintf(query_buffer, "-- version: %d", version);
198 while (fgets(line, sizeof(line), f)) {
200 query_buffer_pos += sprintf(query_buffer + query_buffer_pos, " %s", line);
201 if(line[strlen(line) - 1] == ';') {
202 mysql_query(mysql_conn, query_buffer);
203 query_buffer_pos = 0;
205 } else if(!stricmplen(query_buffer, line, strlen(query_buffer))) {
209 if(query_buffer_pos) {
210 if(mysql_query(mysql_conn, query_buffer))
215 printf_log("main", LOG_ERROR | LOG_MYSQL, "File not found: database.upgrade.sql");
217 MYSQL_RES *res = mysql_store_result(mysql_conn);
218 mysql_free_result(res);
219 } while(!mysql_next_result(mysql_conn));
220 mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_OFF);
221 mysql_query(mysql_conn, "UPDATE `version` SET `database_version` = '" DATABASE_VERSION "'");
226 struct mysql_conn_struct *mysql_conn, *next;
227 for(mysql_conn = mysql_conns; mysql_conn; mysql_conn = next) {
228 next = mysql_conn->next;
229 mysql_close(mysql_conn->mysql_conn);
235 static void show_mysql_error() {
236 MYSQL *mysql_conn = get_mysql_conn();
238 printf_log("main", LOG_ERROR | LOG_MYSQL, "Error: %s\n", mysql_error(mysql_conn));
241 void printf_mysql_query(const char *text, ...) {
242 MYSQL *mysql_conn = get_mysql_conn();
244 char queryBuf[MYSQLMAXLEN];
247 va_start(arg_list, text);
248 pos = vsnprintf(queryBuf, MYSQLMAXLEN - 2, text, arg_list);
250 if (pos < 0 || pos > (MYSQLMAXLEN - 2)) pos = MYSQLMAXLEN - 2;
251 queryBuf[pos] = '\0';
252 printf_log("main", LOG_MYSQL, "%s\n", queryBuf);
253 if(mysql_query(mysql_conn, queryBuf)) {
255 if(mysql_query(mysql_conn, queryBuf)) {
261 void printf_long_mysql_query(int len, const char *text, ...) {
262 MYSQL *mysql_conn = get_mysql_conn();
267 va_start(arg_list, text);
268 pos = vsnprintf(queryBuf, len - 2, text, arg_list);
270 if (pos < 0 || pos > (len - 2)) pos = len - 2;
271 queryBuf[pos] = '\0';
272 printf_log("main", LOG_MYSQL, "%s\n", queryBuf);
273 if(mysql_query(mysql_conn, queryBuf)) {
275 if(mysql_query(mysql_conn, queryBuf)) {
281 char* escape_string(const char *str) {
282 struct mysql_conn_struct *mysql_conn = get_mysql_conn_struct();
283 struct escaped_string *escapedstr = malloc(sizeof(*escapedstr));
287 char escaped[strlen(str)*2+1];
288 mysql_real_escape_string(mysql_conn->mysql_conn, escaped, str, strlen(str));
289 escapedstr->string = strdup(escaped);
290 escapedstr->next = mysql_conn->escaped_strings;
291 mysql_conn->escaped_strings = escapedstr;
292 return escapedstr->string;
295 struct mysql_conn_struct *get_mysql_conn_struct() {
296 SYNCHRONIZE(synchronized);
297 struct mysql_conn_struct *mysql_conn;
300 tid = (unsigned int) pthread_self_tid();
304 for(mysql_conn = mysql_conns; mysql_conn; mysql_conn = mysql_conn->next) {
305 if(mysql_conn->tid == tid) {
306 DESYNCHRONIZE(synchronized);
310 mysql_conn = malloc(sizeof(*mysql_conn));
311 mysql_conn->mysql_conn = mysql_init(NULL);
312 mysql_conn->tid = tid;
313 mysql_conn->used_results = NULL;
314 mysql_conn->escaped_strings = NULL;
315 mysql_conn->next = mysql_conns;
316 mysql_conns = mysql_conn;
317 if (!mysql_real_connect(mysql_conn->mysql_conn, mysql_host, mysql_user, mysql_pass, mysql_base, mysql_serverport, NULL, 0)) {
321 DESYNCHRONIZE(synchronized);
325 MYSQL *get_mysql_conn() {
326 struct mysql_conn_struct *mysql_conn = get_mysql_conn_struct();
327 return mysql_conn->mysql_conn;