3 #define DATABASE_VERSION "1"
7 struct used_result *next;
10 struct escaped_string {
12 struct escaped_string *next;
15 MYSQL *mysql_conn = NULL;
16 static struct used_result *used_results;
17 static struct escaped_string *escaped_strings;
18 static int mysql_serverport;
19 static char *mysql_host, *mysql_user, *mysql_pass, *mysql_base;
23 if((errid = mysql_ping(mysql_conn))) {
24 if(mysql_errno(mysql_conn) == CR_SERVER_GONE_ERROR) {
25 if(!mysql_real_connect(mysql_conn, mysql_host, mysql_user, mysql_pass, mysql_base, mysql_serverport, NULL, 0)) {
35 MYSQL_RES *mysql_use() {
36 MYSQL_RES *res = mysql_store_result(mysql_conn);
37 struct used_result *result = malloc(sizeof(*result));
39 mysql_free_result(res);
43 result->next = used_results;
44 used_results = result;
49 struct used_result *result, *next_result;
50 for(result = used_results; result; result = next_result) {
51 next_result = result->next;
52 mysql_free_result(result->result);
56 struct escaped_string *escaped, *next_escaped;
57 for(escaped = escaped_strings; escaped; escaped = next_escaped) {
58 next_escaped = escaped->next;
59 free(escaped->string);
62 escaped_strings = NULL;
65 void init_mysql(char *host, int port, char *user, char *pass, char *base) {
66 mysql_host = strdup(host);
67 mysql_serverport = port;
68 mysql_user = strdup(user);
69 mysql_pass = strdup(pass);
70 mysql_base = strdup(base);
71 mysql_conn = mysql_init(NULL);
72 if (!mysql_real_connect(mysql_conn, mysql_host, mysql_user, mysql_pass, mysql_base, mysql_serverport, NULL, 0)) {
76 //check database version...
78 if(!mysql_query(mysql_conn, "SELECT `database_version` FROM `version`")) {
79 MYSQL_RES *res = mysql_use();
81 if((row = mysql_fetch_row(res))) {
82 version = atoi(row[0]);
85 printf("%d \n", version);
88 FILE *f = fopen("database.sql", "r");
89 mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_ON);
92 char query_buffer[8192];
93 int query_buffer_pos = 0;
94 while (fgets(line, sizeof(line), f)) {
95 query_buffer_pos += sprintf(query_buffer + query_buffer_pos, " %s", line);
96 if(line[(strlen(line) - 2)] == ';') {
97 if(mysql_query(mysql_conn, query_buffer))
104 f = fopen("database.defaults.sql", "r");
107 char query_buffer[131072];
108 int query_buffer_pos = 0;
109 while (fgets(line, sizeof(line), f)) {
110 query_buffer_pos += sprintf(query_buffer + query_buffer_pos, " %s", line);
111 if(line[(strlen(line) - 2)] == ';') {
112 if(mysql_query(mysql_conn, query_buffer))
114 query_buffer_pos = 0;
119 mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_OFF);
120 mysql_query(mysql_conn, "INSERT INTO `version` (`database_version`) VALUES ('" DATABASE_VERSION "')");
122 else if(version < atoi(DATABASE_VERSION)) {
124 FILE *f = fopen("database.upgrade.sql", "r");
125 mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_ON);
128 char query_buffer[8192];
129 int query_buffer_pos = 0, use_querys = 0;
130 sprintf(query_buffer, "-- version: %d", version);
131 while (fgets(line, sizeof(line), f)) {
133 query_buffer_pos += sprintf(query_buffer + query_buffer_pos, " %s", line);
134 if(line[strlen(line) - 1] == ';') {
135 mysql_query(mysql_conn, query_buffer);
136 query_buffer_pos = 0;
138 } else if(!stricmp(query_buffer, line)) {
142 if(query_buffer_pos) {
143 if(mysql_query(mysql_conn, query_buffer))
148 perror("database.sql missing!");
149 mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_OFF);
150 mysql_query(mysql_conn, "UPDATE `version` SET `database_version` = '" DATABASE_VERSION "'");
155 mysql_close(mysql_conn);
158 void show_mysql_error() {
160 printf("MySQL Error: %s\n", mysql_error(mysql_conn));
163 void printf_mysql_query(const char *text, ...) {
165 char queryBuf[MYSQLMAXLEN];
168 va_start(arg_list, text);
169 pos = vsnprintf(queryBuf, MYSQLMAXLEN - 2, text, arg_list);
171 if (pos < 0 || pos > (MYSQLMAXLEN - 2)) pos = MYSQLMAXLEN - 2;
172 queryBuf[pos] = '\0';
173 printf("MySQL: %s\n", queryBuf);
174 if(mysql_query(mysql_conn, queryBuf)) {
176 if(mysql_query(mysql_conn, queryBuf)) {
182 char* escape_string(const char *str) {
183 struct escaped_string *escapedstr = malloc(sizeof(*escapedstr));
187 char escaped[strlen(str)*2+1];
188 mysql_real_escape_string(mysql_conn, escaped, str, strlen(str));
189 escapedstr->string = strdup(escaped);
190 escapedstr->next = escaped_strings;
191 escaped_strings = escapedstr;
192 return escapedstr->string;