src/mysqlConn.c

   1 /* mysqlConn.c - NeonServ v5.3
   2  * Copyright (C) 2011-2012  Philipp Kreil (pk910)
   3  *
   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.
   8  *
   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.
  13  *
  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/>.
  16  */
  17
  18 #include "mysqlConn.h"
  19 #define DATABASE_VERSION "15"
  20
  21 struct mysql_conn_struct {
  22     unsigned int tid;
  23     MYSQL *mysql_conn;
  24     struct used_result *used_results;
  25     struct escaped_string *escaped_strings;
  26     struct mysql_conn_struct *next;
  27 };
  28
  29 struct used_result {
  30     MYSQL_RES *result;
  31     struct used_result *next;
  32 };
  33
  34 struct escaped_string {
  35     char *string;
  36     struct escaped_string *next;
  37 };
  38
  39 struct mysql_conn_struct *get_mysql_conn_struct();
  40
  41 struct mysql_conn_struct *mysql_conns = NULL;
  42 static int mysql_serverport;
  43 static char *mysql_host, *mysql_user, *mysql_pass, *mysql_base;
  44
  45 #ifdef HAVE_THREADS
  46 static pthread_mutex_t synchronized;
  47 #endif
  48
  49 void check_mysql() {
  50     MYSQL *mysql_conn = get_mysql_conn();
  51     int errid;
  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)) {
  55                 show_mysql_error();
  56             }
  57         } else {
  58             //mysql error
  59             show_mysql_error();
  60         }
  61     }
  62 }
  63
  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));
  68     if (!result) {
  69         mysql_free_result(res);
  70         return NULL;
  71     }
  72     result->result = res;
  73     result->next = mysql_conn->used_results;
  74     mysql_conn->used_results = result;
  75     return res;
  76 }
  77
  78 void mysql_free() {
  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);
  85         free(result);
  86     }
  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);
  92         free(escaped);
  93     }
  94     mysql_conn->escaped_strings = NULL;
  95 }
  96
  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);
 104
 105
 106     MYSQL *mysql_conn = get_mysql_conn();
 107
 108     //check database version...
 109     int version = 0;
 110     if(!mysql_query(mysql_conn, "SELECT `database_version` FROM `version`")) {
 111         MYSQL_RES *res = mysql_use();
 112         MYSQL_ROW row;
 113         if((row = mysql_fetch_row(res))) {
 114             version = atoi(row[0]);
 115         }
 116     }
 117     if(!version) {
 118         //CREATE DATABASE
 119         FILE *f = fopen("database.sql", "r");
 120         mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_ON);
 121         if (f) {
 122             char line[512];
 123             char query_buffer[8192];
 124             int query_buffer_pos = 0;
 125             while (fgets(line, sizeof(line), f)) {
 126                 query_buffer_pos += sprintf(query_buffer + query_buffer_pos, " %s", line);
 127                 if(line[(strlen(line) - 2)] == ';') {
 128                     if(mysql_query(mysql_conn, query_buffer))
 129                         show_mysql_error();
 130                     query_buffer_pos = 0;
 131                 }
 132             }
 133             fclose(f);
 134         }
 135         f = fopen("database.defaults.sql", "r");
 136         if (f) {
 137             char line[4096];
 138             char query_buffer[131072];
 139             int query_buffer_pos = 0;
 140             while (fgets(line, sizeof(line), f)) {
 141                 query_buffer_pos += sprintf(query_buffer + query_buffer_pos, " %s", line);
 142                 if(line[(strlen(line) - 2)] == ';') {
 143                     if(mysql_query(mysql_conn, query_buffer))
 144                         show_mysql_error();
 145                     query_buffer_pos = 0;
 146                 }
 147             }
 148             fclose(f);
 149         }
 150         do {
 151             MYSQL_RES *res = mysql_store_result(mysql_conn);
 152             mysql_free_result(res);
 153         } while(!mysql_next_result(mysql_conn));
 154         mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_OFF);
 155         mysql_query(mysql_conn, "INSERT INTO `version` (`database_version`) VALUES ('" DATABASE_VERSION "')");
 156     }
 157     else if(version < atoi(DATABASE_VERSION)) {
 158         //UPDATE DATABASE
 159         FILE *f = fopen("database.upgrade.sql", "r");
 160         mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_ON);
 161         if (f) {
 162             char line[512];
 163             char query_buffer[8192];
 164             int query_buffer_pos = 0, use_querys = 0;
 165             sprintf(query_buffer, "-- version: %d", version);
 166             while (fgets(line, sizeof(line), f)) {
 167                 if(use_querys) {
 168                     query_buffer_pos += sprintf(query_buffer + query_buffer_pos, " %s", line);
 169                     if(line[strlen(line) - 1] == ';') {
 170                         mysql_query(mysql_conn, query_buffer);
 171                         query_buffer_pos = 0;
 172                     }
 173                 } else if(!stricmplen(query_buffer, line, strlen(query_buffer))) {
 174                     use_querys = 1;
 175                 }
 176             }
 177             if(query_buffer_pos) {
 178                 if(mysql_query(mysql_conn, query_buffer))
 179                     show_mysql_error();
 180             }
 181             fclose(f);
 182         } else
 183             perror("database.sql missing!");
 184         do {
 185             MYSQL_RES *res = mysql_store_result(mysql_conn);
 186             mysql_free_result(res);
 187         } while(!mysql_next_result(mysql_conn));
 188         mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_OFF);
 189         mysql_query(mysql_conn, "UPDATE `version` SET `database_version` = '" DATABASE_VERSION "'");
 190     }
 191 }
 192
 193 void free_mysql() {
 194     struct mysql_conn_struct *mysql_conn, *next;
 195     for(mysql_conn = mysql_conns; mysql_conn; mysql_conn = next) {
 196         next = mysql_conn->next;
 197         mysql_close(mysql_conn->mysql_conn);
 198         free(mysql_conn);
 199     }
 200     mysql_conns = NULL;
 201 }
 202
 203 void show_mysql_error() {
 204     MYSQL *mysql_conn = get_mysql_conn();
 205     //show mysql_error()
 206     putlog(LOGLEVEL_ERROR, "MySQL Error: %s\n", mysql_error(mysql_conn));
 207 }
 208
 209 void printf_mysql_query(const char *text, ...) {
 210     MYSQL *mysql_conn = get_mysql_conn();
 211     va_list arg_list;
 212     char queryBuf[MYSQLMAXLEN];
 213     int pos;
 214     queryBuf[0] = '\0';
 215     va_start(arg_list, text);
 216     pos = vsnprintf(queryBuf, MYSQLMAXLEN - 2, text, arg_list);
 217     va_end(arg_list);
 218     if (pos < 0 || pos > (MYSQLMAXLEN - 2)) pos = MYSQLMAXLEN - 2;
 219     queryBuf[pos] = '\0';
 220     putlog(LOGLEVEL_MYSQL, "MySQL: %s\n", queryBuf);
 221     if(mysql_query(mysql_conn, queryBuf)) {
 222         check_mysql();
 223         if(mysql_query(mysql_conn, queryBuf)) {
 224             show_mysql_error();
 225         }
 226     }
 227 }
 228
 229 void printf_long_mysql_query(int len, const char *text, ...) {
 230     MYSQL *mysql_conn = get_mysql_conn();
 231     va_list arg_list;
 232     char queryBuf[len];
 233     int pos;
 234     queryBuf[0] = '\0';
 235     va_start(arg_list, text);
 236     pos = vsnprintf(queryBuf, len - 2, text, arg_list);
 237     va_end(arg_list);
 238     if (pos < 0 || pos > (len - 2)) pos = len - 2;
 239     queryBuf[pos] = '\0';
 240     putlog(LOGLEVEL_MYSQL, "MySQL: %s\n", queryBuf);
 241     if(mysql_query(mysql_conn, queryBuf)) {
 242         check_mysql();
 243         if(mysql_query(mysql_conn, queryBuf)) {
 244             show_mysql_error();
 245         }
 246     }
 247 }
 248
 249 char* escape_string(const char *str) {
 250     struct mysql_conn_struct *mysql_conn = get_mysql_conn_struct();
 251     struct escaped_string *escapedstr = malloc(sizeof(*escapedstr));
 252     if (!escapedstr) {
 253         return NULL;
 254     }
 255     char escaped[strlen(str)*2+1];
 256     mysql_real_escape_string(mysql_conn->mysql_conn, escaped, str, strlen(str));
 257     escapedstr->string = strdup(escaped);
 258     escapedstr->next = mysql_conn->escaped_strings;
 259     mysql_conn->escaped_strings = escapedstr;
 260     return escapedstr->string;
 261 }
 262
 263 struct mysql_conn_struct *get_mysql_conn_struct() {
 264     SYNCHRONIZE(synchronized);
 265     struct mysql_conn_struct *mysql_conn;
 266     unsigned int tid;
 267     #ifdef HAVE_THREADS
 268     tid = (unsigned int) pthread_self_tid();
 269     #else
 270     tid = 1;
 271     #endif
 272     for(mysql_conn = mysql_conns; mysql_conn; mysql_conn = mysql_conn->next) {
 273         if(mysql_conn->tid == tid) {
 274             DESYNCHRONIZE(synchronized);
 275             return mysql_conn;
 276         }
 277     }
 278     mysql_conn = malloc(sizeof(*mysql_conn));
 279     mysql_conn->mysql_conn = mysql_init(NULL);
 280     mysql_conn->tid = tid;
 281     mysql_conn->used_results = NULL;
 282     mysql_conn->escaped_strings = NULL;
 283     mysql_conn->next = mysql_conns;
 284     mysql_conns = mysql_conn;
 285     if (!mysql_real_connect(mysql_conn->mysql_conn, mysql_host, mysql_user, mysql_pass, mysql_base, mysql_serverport, NULL, 0)) {
 286         //error
 287         show_mysql_error();
 288     }
 289     DESYNCHRONIZE(synchronized);
 290     return mysql_conn;
 291 }
 292
 293 MYSQL *get_mysql_conn() {
 294     struct mysql_conn_struct *mysql_conn = get_mysql_conn_struct();
 295     return mysql_conn->mysql_conn;
 296 }