X-Git-Url: http://git.pk910.de/?p=NeonServV5.git;a=blobdiff_plain;f=src%2FmysqlConn.c;h=b82d317cc4a5770ffd6a3c8af6812bd9e8ec6ecc;hp=66494e7f5055cf333db9d500c9bde2a605c7f2cd;hb=b013fcf166b6b84f7b946412dacfe84ba5cfe6b6;hpb=0f1dc61921eef1db8e404a5a82372e2d1cd55daa diff --git a/src/mysqlConn.c b/src/mysqlConn.c index 66494e7..b82d317 100644 --- a/src/mysqlConn.c +++ b/src/mysqlConn.c @@ -1,5 +1,22 @@ +/* mysqlConn.c - NeonServ v5.3 + * Copyright (C) 2011-2012 Philipp Kreil (pk910) + * + * This program 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 3 of the License, or + * (at your option) any later version. + * + * This program 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, see . + */ #include "mysqlConn.h" +#define DATABASE_VERSION "14" struct used_result { MYSQL_RES *result; @@ -14,12 +31,14 @@ struct escaped_string { MYSQL *mysql_conn = NULL; static struct used_result *used_results; static struct escaped_string *escaped_strings; +static int mysql_serverport; +static char *mysql_host, *mysql_user, *mysql_pass, *mysql_base; void check_mysql() { int errid; if((errid = mysql_ping(mysql_conn))) { if(mysql_errno(mysql_conn) == CR_SERVER_GONE_ERROR) { - if(!mysql_real_connect(mysql_conn, MYSQL_HOST, MYSQL_USER, MYSQL_PASS, MYSQL_BASE, MYSQL_PORT, NULL, 0)) { + if(!mysql_real_connect(mysql_conn, mysql_host, mysql_user, mysql_pass, mysql_base, mysql_serverport, NULL, 0)) { show_mysql_error(); } } else { @@ -59,12 +78,101 @@ void mysql_free() { escaped_strings = NULL; } -void init_mysql() { +void init_mysql(char *host, int port, char *user, char *pass, char *base) { + mysql_host = strdup(host); + mysql_serverport = port; + mysql_user = strdup(user); + mysql_pass = strdup(pass); + mysql_base = strdup(base); mysql_conn = mysql_init(NULL); - if (!mysql_real_connect(mysql_conn, MYSQL_HOST, MYSQL_USER, MYSQL_PASS, MYSQL_BASE, MYSQL_PORT, NULL, 0)) { + if (!mysql_real_connect(mysql_conn, mysql_host, mysql_user, mysql_pass, mysql_base, mysql_serverport, NULL, 0)) { //error show_mysql_error(); } + //check database version... + int version = 0; + if(!mysql_query(mysql_conn, "SELECT `database_version` FROM `version`")) { + MYSQL_RES *res = mysql_use(); + MYSQL_ROW row; + if((row = mysql_fetch_row(res))) { + version = atoi(row[0]); + } + } + printf("%d \n", version); + if(!version) { + //CREATE DATABASE + FILE *f = fopen("database.sql", "r"); + mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_ON); + if (f) { + char line[512]; + char query_buffer[8192]; + int query_buffer_pos = 0; + while (fgets(line, sizeof(line), f)) { + query_buffer_pos += sprintf(query_buffer + query_buffer_pos, " %s", line); + if(line[(strlen(line) - 2)] == ';') { + if(mysql_query(mysql_conn, query_buffer)) + show_mysql_error(); + query_buffer_pos = 0; + } + } + fclose(f); + } + f = fopen("database.defaults.sql", "r"); + if (f) { + char line[4096]; + char query_buffer[131072]; + int query_buffer_pos = 0; + while (fgets(line, sizeof(line), f)) { + query_buffer_pos += sprintf(query_buffer + query_buffer_pos, " %s", line); + if(line[(strlen(line) - 2)] == ';') { + if(mysql_query(mysql_conn, query_buffer)) + show_mysql_error(); + query_buffer_pos = 0; + } + } + fclose(f); + } + do { + MYSQL_RES *res = mysql_store_result(mysql_conn); + mysql_free_result(res); + } while(!mysql_next_result(mysql_conn)); + mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_OFF); + mysql_query(mysql_conn, "INSERT INTO `version` (`database_version`) VALUES ('" DATABASE_VERSION "')"); + } + else if(version < atoi(DATABASE_VERSION)) { + //UPDATE DATABASE + FILE *f = fopen("database.upgrade.sql", "r"); + mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_ON); + if (f) { + char line[512]; + char query_buffer[8192]; + int query_buffer_pos = 0, use_querys = 0; + sprintf(query_buffer, "-- version: %d", version); + while (fgets(line, sizeof(line), f)) { + if(use_querys) { + query_buffer_pos += sprintf(query_buffer + query_buffer_pos, " %s", line); + if(line[strlen(line) - 1] == ';') { + mysql_query(mysql_conn, query_buffer); + query_buffer_pos = 0; + } + } else if(!stricmplen(query_buffer, line, strlen(query_buffer))) { + use_querys = 1; + } + } + if(query_buffer_pos) { + if(mysql_query(mysql_conn, query_buffer)) + show_mysql_error(); + } + fclose(f); + } else + perror("database.sql missing!"); + do { + MYSQL_RES *res = mysql_store_result(mysql_conn); + mysql_free_result(res); + } while(!mysql_next_result(mysql_conn)); + mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_OFF); + mysql_query(mysql_conn, "UPDATE `version` SET `database_version` = '" DATABASE_VERSION "'"); + } } void free_mysql() { @@ -95,6 +203,25 @@ void printf_mysql_query(const char *text, ...) { } } +void printf_long_mysql_query(int len, const char *text, ...) { + va_list arg_list; + char queryBuf[len]; + int pos; + queryBuf[0] = '\0'; + va_start(arg_list, text); + pos = vsnprintf(queryBuf, len - 2, text, arg_list); + va_end(arg_list); + if (pos < 0 || pos > (len - 2)) pos = len - 2; + queryBuf[pos] = '\0'; + printf("MySQL: %s\n", queryBuf); + if(mysql_query(mysql_conn, queryBuf)) { + check_mysql(); + if(mysql_query(mysql_conn, queryBuf)) { + show_mysql_error(); + } + } +} + char* escape_string(const char *str) { struct escaped_string *escapedstr = malloc(sizeof(*escapedstr)); if (!escapedstr) {