1 /* mysqlConn.c - NeonServ v5.1
2 * Copyright (C) 2011 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 "3"
23 struct used_result *next;
26 struct escaped_string {
28 struct escaped_string *next;
31 MYSQL *mysql_conn = NULL;
32 static struct used_result *used_results;
33 static struct escaped_string *escaped_strings;
34 static int mysql_serverport;
35 static char *mysql_host, *mysql_user, *mysql_pass, *mysql_base;
39 if((errid = mysql_ping(mysql_conn))) {
40 if(mysql_errno(mysql_conn) == CR_SERVER_GONE_ERROR) {
41 if(!mysql_real_connect(mysql_conn, mysql_host, mysql_user, mysql_pass, mysql_base, mysql_serverport, NULL, 0)) {
51 MYSQL_RES *mysql_use() {
52 MYSQL_RES *res = mysql_store_result(mysql_conn);
53 struct used_result *result = malloc(sizeof(*result));
55 mysql_free_result(res);
59 result->next = used_results;
60 used_results = result;
65 struct used_result *result, *next_result;
66 for(result = used_results; result; result = next_result) {
67 next_result = result->next;
68 mysql_free_result(result->result);
72 struct escaped_string *escaped, *next_escaped;
73 for(escaped = escaped_strings; escaped; escaped = next_escaped) {
74 next_escaped = escaped->next;
75 free(escaped->string);
78 escaped_strings = NULL;
81 void init_mysql(char *host, int port, char *user, char *pass, char *base) {
82 mysql_host = strdup(host);
83 mysql_serverport = port;
84 mysql_user = strdup(user);
85 mysql_pass = strdup(pass);
86 mysql_base = strdup(base);
87 mysql_conn = mysql_init(NULL);
88 if (!mysql_real_connect(mysql_conn, mysql_host, mysql_user, mysql_pass, mysql_base, mysql_serverport, NULL, 0)) {
92 //check database version...
94 if(!mysql_query(mysql_conn, "SELECT `database_version` FROM `version`")) {
95 MYSQL_RES *res = mysql_use();
97 if((row = mysql_fetch_row(res))) {
98 version = atoi(row[0]);
101 printf("%d \n", version);
104 FILE *f = fopen("database.sql", "r");
105 mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_ON);
108 char query_buffer[8192];
109 int query_buffer_pos = 0;
110 while (fgets(line, sizeof(line), f)) {
111 query_buffer_pos += sprintf(query_buffer + query_buffer_pos, " %s", line);
112 if(line[(strlen(line) - 2)] == ';') {
113 if(mysql_query(mysql_conn, query_buffer))
115 query_buffer_pos = 0;
120 f = fopen("database.defaults.sql", "r");
123 char query_buffer[131072];
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))
130 query_buffer_pos = 0;
135 mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_OFF);
136 mysql_query(mysql_conn, "INSERT INTO `version` (`database_version`) VALUES ('" DATABASE_VERSION "')");
138 else if(version < atoi(DATABASE_VERSION)) {
140 FILE *f = fopen("database.upgrade.sql", "r");
141 mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_ON);
144 char query_buffer[8192];
145 int query_buffer_pos = 0, use_querys = 0;
146 sprintf(query_buffer, "-- version: %d", version);
147 while (fgets(line, sizeof(line), f)) {
149 query_buffer_pos += sprintf(query_buffer + query_buffer_pos, " %s", line);
150 if(line[strlen(line) - 1] == ';') {
151 mysql_query(mysql_conn, query_buffer);
152 query_buffer_pos = 0;
154 } else if(!stricmp(query_buffer, line)) {
158 if(query_buffer_pos) {
159 if(mysql_query(mysql_conn, query_buffer))
164 perror("database.sql missing!");
165 mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_OFF);
166 mysql_query(mysql_conn, "UPDATE `version` SET `database_version` = '" DATABASE_VERSION "'");
171 mysql_close(mysql_conn);
174 void show_mysql_error() {
176 printf("MySQL Error: %s\n", mysql_error(mysql_conn));
179 void printf_mysql_query(const char *text, ...) {
181 char queryBuf[MYSQLMAXLEN];
184 va_start(arg_list, text);
185 pos = vsnprintf(queryBuf, MYSQLMAXLEN - 2, text, arg_list);
187 if (pos < 0 || pos > (MYSQLMAXLEN - 2)) pos = MYSQLMAXLEN - 2;
188 queryBuf[pos] = '\0';
189 printf("MySQL: %s\n", queryBuf);
190 if(mysql_query(mysql_conn, queryBuf)) {
192 if(mysql_query(mysql_conn, queryBuf)) {
198 char* escape_string(const char *str) {
199 struct escaped_string *escapedstr = malloc(sizeof(*escapedstr));
203 char escaped[strlen(str)*2+1];
204 mysql_real_escape_string(mysql_conn, escaped, str, strlen(str));
205 escapedstr->string = strdup(escaped);
206 escapedstr->next = escaped_strings;
207 escaped_strings = escapedstr;
208 return escapedstr->string;