1 /* mysqlConn.c - NeonServ v5.3
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 #define DATABASE_VERSION "15"
21 struct mysql_conn_struct {
24 struct used_result *used_results;
25 struct escaped_string *escaped_strings;
26 struct mysql_conn_struct *next;
31 struct used_result *next;
34 struct escaped_string {
36 struct escaped_string *next;
39 struct mysql_conn_struct *get_mysql_conn_struct();
41 struct mysql_conn_struct *mysql_conns = NULL;
42 static int mysql_serverport;
43 static char *mysql_host, *mysql_user, *mysql_pass, *mysql_base;
46 static pthread_mutex_t synchronized;
50 MYSQL *mysql_conn = get_mysql_conn();
52 if((errid = mysql_ping(mysql_conn))) {
53 if(mysql_errno(mysql_conn) == CR_SERVER_GONE_ERROR) {
54 if(!mysql_real_connect(mysql_conn, mysql_host, mysql_user, mysql_pass, mysql_base, mysql_serverport, NULL, 0)) {
64 MYSQL_RES *mysql_use() {
65 struct mysql_conn_struct *mysql_conn = get_mysql_conn_struct();
66 MYSQL_RES *res = mysql_store_result(mysql_conn->mysql_conn);
67 struct used_result *result = malloc(sizeof(*result));
69 mysql_free_result(res);
73 result->next = mysql_conn->used_results;
74 mysql_conn->used_results = result;
79 struct mysql_conn_struct *mysql_conn = get_mysql_conn_struct();
80 if(!mysql_conn) return;
81 struct used_result *result, *next_result;
82 for(result = mysql_conn->used_results; result; result = next_result) {
83 next_result = result->next;
84 mysql_free_result(result->result);
87 mysql_conn->used_results = NULL;
88 struct escaped_string *escaped, *next_escaped;
89 for(escaped = mysql_conn->escaped_strings; escaped; escaped = next_escaped) {
90 next_escaped = escaped->next;
91 free(escaped->string);
94 mysql_conn->escaped_strings = NULL;
97 void init_mysql(char *host, int port, char *user, char *pass, char *base) {
98 THREAD_MUTEX_INIT(synchronized);
99 mysql_host = strdup(host);
100 mysql_serverport = port;
101 mysql_user = strdup(user);
102 mysql_pass = strdup(pass);
103 mysql_base = strdup(base);
106 MYSQL *mysql_conn = get_mysql_conn();
108 //check database version...
110 if(!mysql_query(mysql_conn, "SELECT `database_version` FROM `version`")) {
111 MYSQL_RES *res = mysql_use();
113 if((row = mysql_fetch_row(res))) {
114 version = atoi(row[0]);
117 printf("%d \n", version);
120 FILE *f = fopen("database.sql", "r");
121 mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_ON);
124 char query_buffer[8192];
125 int query_buffer_pos = 0;
126 while (fgets(line, sizeof(line), f)) {
127 query_buffer_pos += sprintf(query_buffer + query_buffer_pos, " %s", line);
128 if(line[(strlen(line) - 2)] == ';') {
129 if(mysql_query(mysql_conn, query_buffer))
131 query_buffer_pos = 0;
136 f = fopen("database.defaults.sql", "r");
139 char query_buffer[131072];
140 int query_buffer_pos = 0;
141 while (fgets(line, sizeof(line), f)) {
142 query_buffer_pos += sprintf(query_buffer + query_buffer_pos, " %s", line);
143 if(line[(strlen(line) - 2)] == ';') {
144 if(mysql_query(mysql_conn, query_buffer))
146 query_buffer_pos = 0;
152 MYSQL_RES *res = mysql_store_result(mysql_conn);
153 mysql_free_result(res);
154 } while(!mysql_next_result(mysql_conn));
155 mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_OFF);
156 mysql_query(mysql_conn, "INSERT INTO `version` (`database_version`) VALUES ('" DATABASE_VERSION "')");
158 else if(version < atoi(DATABASE_VERSION)) {
160 FILE *f = fopen("database.upgrade.sql", "r");
161 mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_ON);
164 char query_buffer[8192];
165 int query_buffer_pos = 0, use_querys = 0;
166 sprintf(query_buffer, "-- version: %d", version);
167 while (fgets(line, sizeof(line), f)) {
169 query_buffer_pos += sprintf(query_buffer + query_buffer_pos, " %s", line);
170 if(line[strlen(line) - 1] == ';') {
171 mysql_query(mysql_conn, query_buffer);
172 query_buffer_pos = 0;
174 } else if(!stricmplen(query_buffer, line, strlen(query_buffer))) {
178 if(query_buffer_pos) {
179 if(mysql_query(mysql_conn, query_buffer))
184 perror("database.sql missing!");
186 MYSQL_RES *res = mysql_store_result(mysql_conn);
187 mysql_free_result(res);
188 } while(!mysql_next_result(mysql_conn));
189 mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_OFF);
190 mysql_query(mysql_conn, "UPDATE `version` SET `database_version` = '" DATABASE_VERSION "'");
195 struct mysql_conn_struct *mysql_conn, *next;
196 for(mysql_conn = mysql_conns; mysql_conn; mysql_conn = next) {
197 next = mysql_conn->next;
198 mysql_close(mysql_conn->mysql_conn);
204 void show_mysql_error() {
205 MYSQL *mysql_conn = get_mysql_conn();
207 printf("MySQL Error: %s\n", mysql_error(mysql_conn));
210 void printf_mysql_query(const char *text, ...) {
211 MYSQL *mysql_conn = get_mysql_conn();
213 char queryBuf[MYSQLMAXLEN];
216 va_start(arg_list, text);
217 pos = vsnprintf(queryBuf, MYSQLMAXLEN - 2, text, arg_list);
219 if (pos < 0 || pos > (MYSQLMAXLEN - 2)) pos = MYSQLMAXLEN - 2;
220 queryBuf[pos] = '\0';
221 printf("MySQL: %s\n", queryBuf);
222 if(mysql_query(mysql_conn, queryBuf)) {
224 if(mysql_query(mysql_conn, queryBuf)) {
230 void printf_long_mysql_query(int len, const char *text, ...) {
231 MYSQL *mysql_conn = get_mysql_conn();
236 va_start(arg_list, text);
237 pos = vsnprintf(queryBuf, len - 2, text, arg_list);
239 if (pos < 0 || pos > (len - 2)) pos = len - 2;
240 queryBuf[pos] = '\0';
241 printf("MySQL: %s\n", queryBuf);
242 if(mysql_query(mysql_conn, queryBuf)) {
244 if(mysql_query(mysql_conn, queryBuf)) {
250 char* escape_string(const char *str) {
251 struct mysql_conn_struct *mysql_conn = get_mysql_conn_struct();
252 struct escaped_string *escapedstr = malloc(sizeof(*escapedstr));
256 char escaped[strlen(str)*2+1];
257 mysql_real_escape_string(mysql_conn->mysql_conn, escaped, str, strlen(str));
258 escapedstr->string = strdup(escaped);
259 escapedstr->next = mysql_conn->escaped_strings;
260 mysql_conn->escaped_strings = escapedstr;
261 return escapedstr->string;
264 struct mysql_conn_struct *get_mysql_conn_struct() {
265 SYNCHRONIZE(synchronized);
266 struct mysql_conn_struct *mysql_conn;
269 tid = (unsigned int) pthread_self_tid();
273 for(mysql_conn = mysql_conns; mysql_conn; mysql_conn = mysql_conn->next) {
274 if(mysql_conn->tid == tid) {
275 DESYNCHRONIZE(synchronized);
279 mysql_conn = malloc(sizeof(*mysql_conn));
280 mysql_conn->mysql_conn = mysql_init(NULL);
281 mysql_conn->tid = tid;
282 mysql_conn->used_results = NULL;
283 mysql_conn->escaped_strings = NULL;
284 mysql_conn->next = mysql_conns;
285 mysql_conns = mysql_conn;
286 if (!mysql_real_connect(mysql_conn->mysql_conn, mysql_host, mysql_user, mysql_pass, mysql_base, mysql_serverport, NULL, 0)) {
290 DESYNCHRONIZE(synchronized);
294 MYSQL *get_mysql_conn() {
295 struct mysql_conn_struct *mysql_conn = get_mysql_conn_struct();
296 return mysql_conn->mysql_conn;