mysqlConn.c

   1
   2 #include "mysqlConn.h"
   3
   4 struct used_result {
   5     MYSQL_RES *result;
   6     struct used_result *next;
   7 };
   8
   9 struct escaped_string {
  10     char *string;
  11     struct escaped_string *next;
  12 };
  13
  14 MYSQL *mysql_conn = NULL;
  15 static struct used_result *used_results;
  16 static struct escaped_string *escaped_strings;
  17
  18 void check_mysql() {
  19     if(mysql_ping(mysql_conn)) {
  20         //mysql error
  21     }
  22 }
  23
  24 MYSQL_RES *mysql_use() {
  25     MYSQL_RES *res = mysql_use_result(mysql_conn);
  26     struct used_result *result = malloc(sizeof(*result));
  27     if (!result) {
  28         mysql_free_result(res);
  29         return NULL;
  30     }
  31     result->result = res;
  32     result->next = used_results;
  33     used_results = result;
  34     return res;
  35 }
  36
  37 void mysql_free() {
  38     struct used_result *result, *next_result;
  39     for(result = used_results; result; result = next_result) {
  40         next_result = result->next;
  41         mysql_free_result(result->result);
  42         free(result);
  43     }
  44     struct escaped_string *escaped, *next_escaped;
  45     for(escaped = escaped_strings; escaped; escaped = next_escaped) {
  46         next_escaped = escaped->next;
  47         free(escaped->string);
  48         free(escaped);
  49     }
  50 }
  51
  52 void init_mysql() {
  53     mysql_conn = mysql_init(NULL);
  54     if (!mysql_real_connect(mysql_conn, MYSQL_HOST, MYSQL_USER, MYSQL_PASS, MYSQL_BASE, MYSQL_PORT, NULL, 0)) {
  55         //error
  56         show_mysql_error();
  57     }
  58 }
  59
  60 void free_mysql() {
  61     mysql_close(mysql_conn);
  62 }
  63
  64 void show_mysql_error() {
  65     //show mysql_error()
  66     printf("MySQL Error: %s\n", mysql_error(mysql_conn));
  67 }
  68
  69 void printf_mysql_query(const char *text, ...) {
  70     va_list arg_list;
  71     char queryBuf[MYSQLMAXLEN];
  72     int pos;
  73     queryBuf[0] = '\0';
  74     va_start(arg_list, text);
  75     pos = vsnprintf(queryBuf, MYSQLMAXLEN - 2, text, arg_list);
  76     va_end(arg_list);
  77     if (pos < 0 || pos > (MYSQLMAXLEN - 2)) pos = MYSQLMAXLEN - 2;
  78     queryBuf[pos] = '\0';
  79     printf("MySQL: %s\n", queryBuf);
  80     if(mysql_query(mysql_conn, queryBuf)) {
  81         show_mysql_error();
  82     }
  83 }
  84
  85 char* escape_string(const char *str) {
  86     struct escaped_string *escapedstr = malloc(sizeof(*escapedstr));
  87     if (!escapedstr) {
  88         return NULL;
  89     }
  90     char escaped[strlen(str)*2+1];
  91     mysql_real_escape_string(mysql_conn, escaped, str, strlen(str));
  92     escapedstr->string = strdup(escaped);
  93     escapedstr->next = escaped_strings;
  94     escaped_strings = escapedstr;
  95     return escapedstr->string;
  96 }