sql.c

   1 #include "ndbneb.h"
   2 #include <stdarg.h>
   3 #include <stdio.h>
   4 #include "sql.h"
   5 #include "logging.h"
   6
   7 static struct {
   8         char *host;
   9         char *name;
  10         char *user;
  11         char *pass;
  12         char *table;
  13         unsigned int port;
  14         MYSQL *link;
  15 } db;
  16
  17 #undef ESC_BUFS
  18 #define ESC_BUFS 8 /* must be a power of 2 */
  19 #define MAX_ESC_STRING 65536
  20
  21 typedef unsigned int uint;
  22
  23 #define esc(s) sql_escape(s)
  24 char *sql_escape(const char *str)
  25 {
  26         static int idx = 0;
  27         static char *buf_ary[ESC_BUFS] = { NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL };
  28         static uint bufsize[ESC_BUFS] = { 0, 0, 0, 0, 0, 0, 0, 0 };
  29         char *buf;
  30         int len;
  31
  32         if (!str || !*str)
  33                 return "";
  34
  35         if (!db.link)
  36                 return NULL;
  37
  38         len = strlen(str);
  39         if (len > MAX_ESC_STRING) {
  40                 lwarn("Unescaped string too large (%d bytes, %d is allowed)",
  41                           len, MAX_ESC_STRING);
  42                 return "";
  43         }
  44
  45         if (bufsize[idx] < len * 2) {
  46                 uint size = len * 2;
  47                 if (size < 16384)
  48                         size = 16384;
  49
  50                 buf_ary[idx] = realloc(buf_ary[idx], size);
  51                 if (!buf_ary[idx]) {
  52                         bufsize[idx] = 0;
  53                         lwarn("Failed to allocate %d bytes for sql escape buffer", size);
  54                         return "";
  55                 }
  56                 bufsize[idx] = size;
  57         }
  58         buf = buf_ary[idx];
  59         idx = (idx + 1) & (ESC_BUFS - 1);
  60         mysql_real_escape_string(db.link, buf, str, len);
  61
  62         return buf;
  63 }
  64
  65 /*
  66  * these two functions are only here to allow callers
  67  * access to error reporting without having to expose
  68  * the db-link to theh callers. It's also nifty as we
  69  * want to remain database layer agnostic
  70  */
  71 const char *sql_error()
  72 {
  73         if (db.link)
  74                 return mysql_error(db.link);
  75
  76         return "unknown error (no link)";
  77 }
  78
  79 int sql_errno(void)
  80 {
  81         if (db.link)
  82                 return mysql_errno(db.link);
  83
  84         return -1;
  85 }
  86
  87 SQL_RESULT *sql_get_result(void)
  88 {
  89         if (db.link)
  90                 return mysql_use_result(db.link);
  91
  92         return NULL;
  93 }
  94
  95 SQL_ROW sql_fetch_row(MYSQL_RES *result)
  96 {
  97         if (result)
  98                 return mysql_fetch_row(result);
  99
 100         return NULL;
 101 }
 102
 103 void sql_free_result(SQL_RESULT *result)
 104 {
 105         if (result)
 106                 mysql_free_result(result);
 107 }
 108
 109 int sql_query(const char *fmt, ...)
 110 {
 111         char *query;
 112         int len, result = 0;
 113         va_list ap;
 114
 115         if (!db.link)
 116                 return -1;
 117
 118         va_start(ap, fmt);
 119         len = vasprintf(&query, fmt, ap);
 120         va_end(ap);
 121
 122         if (len == -1 || !query) {
 123                 lerr("sql_query: Failed to build query from format-string '%s'", fmt);
 124                 return -1;
 125         }
 126         if ((result = mysql_real_query(db.link, query, len))) {
 127                 lerr("mysql_query(): Retrying failed query [%s]: %s",
 128                      query, sql_error());
 129
 130                 sql_close();
 131                 if (sql_init() < 0 || (result = mysql_real_query(db.link, query, len)))
 132                         lerr("mysql_query(): Dropping failed query [%s]: %s",
 133                              query, sql_error());
 134         }
 135
 136         free(query);
 137
 138         return result;
 139 }
 140
 141 int sql_init(void)
 142 {
 143         if (!(db.link = mysql_init(NULL)))
 144                 return -1;
 145
 146         if (!db.host) {
 147                 db.host = "";
 148                 db.port = 0;
 149         }
 150
 151         if (!(mysql_real_connect(db.link, db.host, db.user, db.pass,
 152                                                          db.name, db.port, NULL, 0)))
 153         {
 154                 lerr("Failed to connect to '%s' at '%s':'%d' using %s:%s as credentials: %s",
 155                      db.name, db.host, db.port, db.user, db.pass, sql_error());
 156
 157                 db.link = NULL;
 158                 return -1;
 159         }
 160
 161         return 0;
 162 }
 163
 164 int sql_close(void)
 165 {
 166         if (db.link)
 167                 mysql_close(db.link);
 168
 169         return 0;
 170 }
 171
 172 const char *sql_db_name(void)
 173 {
 174         if (!db.name)
 175                 return "monitor_reports";
 176
 177         return db.name;
 178 }
 179
 180 const char *sql_table_name(void)
 181 {
 182         if (!db.table)
 183                 return "report_data";
 184
 185         return db.table;
 186 }
 187
 188 int sql_config(const char *key, const char *value)
 189 {
 190         if (!prefixcmp(key, "db_database"))
 191                 db.name = strdup(value);
 192         else if (!prefixcmp(key, "db_user"))
 193                 db.user = strdup(value);
 194         else if (!prefixcmp(key, "db_pass"))
 195                 db.pass = strdup(value);
 196         else if (!prefixcmp(key, "db_host"))
 197                 db.host = strdup(value);
 198         else if (!prefixcmp(key, "db_table")) {
 199                 db.table = strdup(value);
 200         }
 201         else if (!prefixcmp(key, "db_port")) {
 202                 char *endp;
 203
 204                 db.port = (unsigned int)strtoul(value, &endp, 0);
 205
 206                 if (endp == value || *endp != 0)
 207                         return -1;
 208         }
 209         else
 210                 return -1; /* config error */
 211
 212         return 0;
 213 }