sql.cpp 26 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078
  1. // Copyright (c) rAthena Dev Teams - Licensed under GNU GPL
  2. // For more information, see LICENCE in the main folder
  3. #include "sql.hpp"
  4. #ifdef WIN32
  5. #include "winapi.hpp"
  6. #endif
  7. #include <mysql.h>
  8. #include <stdlib.h>// strtoul
  9. #include "cbasetypes.hpp"
  10. #include "malloc.hpp"
  11. #include "showmsg.hpp"
  12. #include "strlib.hpp"
  13. #include "timer.hpp"
  14. // MySQL 8.0 or later removed my_bool typedef.
  15. // Reintroduce it as a bandaid fix.
  16. // See https://bugs.mysql.com/?id=87337
  17. #if !defined(MARIADB_BASE_VERSION) && !defined(MARIADB_VERSION_ID) && MYSQL_VERSION_ID >= 80001 && MYSQL_VERSION_ID != 80002
  18. #define my_bool bool
  19. #endif
  20. #define SQL_CONF_NAME "conf/inter_athena.conf"
  21. void ra_mysql_error_handler(unsigned int ecode);
  22. int mysql_reconnect_type;
  23. unsigned int mysql_reconnect_count;
  24. /// Sql handle
  25. struct Sql
  26. {
  27. StringBuf buf;
  28. MYSQL handle;
  29. MYSQL_RES* result;
  30. MYSQL_ROW row;
  31. unsigned long* lengths;
  32. int keepalive;
  33. };
  34. // Column length receiver.
  35. // Takes care of the possible size missmatch between uint32 and unsigned long.
  36. struct s_column_length
  37. {
  38. uint32* out_length;
  39. unsigned long length;
  40. };
  41. typedef struct s_column_length s_column_length;
  42. /// Sql statement
  43. struct SqlStmt
  44. {
  45. StringBuf buf;
  46. MYSQL_STMT* stmt;
  47. MYSQL_BIND* params;
  48. MYSQL_BIND* columns;
  49. s_column_length* column_lengths;
  50. size_t max_params;
  51. size_t max_columns;
  52. bool bind_params;
  53. bool bind_columns;
  54. };
  55. ///////////////////////////////////////////////////////////////////////////////
  56. // Sql Handle
  57. ///////////////////////////////////////////////////////////////////////////////
  58. /// Allocates and initializes a new Sql handle.
  59. Sql* Sql_Malloc(void)
  60. {
  61. Sql* self;
  62. CREATE(self, Sql, 1);
  63. mysql_init(&self->handle);
  64. StringBuf_Init(&self->buf);
  65. self->lengths = NULL;
  66. self->result = NULL;
  67. self->keepalive = INVALID_TIMER;
  68. my_bool reconnect = 1;
  69. mysql_options(&self->handle, MYSQL_OPT_RECONNECT, &reconnect);
  70. return self;
  71. }
  72. /**
  73. * Retrieves the last error number.
  74. * @param self : sql handle
  75. * @return last error number
  76. */
  77. unsigned int Sql_GetError( Sql* self ){
  78. return mysql_errno( &self->handle );
  79. }
  80. static int Sql_P_Keepalive(Sql* self);
  81. /**
  82. * Establishes a connection to schema
  83. * @param self : sql handle
  84. * @param user : username to access
  85. * @param passwd : password
  86. * @param host : hostname
  87. * @param port : port
  88. * @param db : schema name
  89. * @return
  90. */
  91. int Sql_Connect(Sql* self, const char* user, const char* passwd, const char* host, uint16 port, const char* db)
  92. {
  93. if( self == NULL )
  94. return SQL_ERROR;
  95. StringBuf_Clear(&self->buf);
  96. #if !defined(MARIADB_BASE_VERSION) && !defined(MARIADB_VERSION_ID) && MYSQL_VERSION_ID >= 50710
  97. unsigned int md = SSL_MODE_DISABLED;
  98. if( mysql_options( &self->handle, MYSQL_OPT_SSL_MODE, &md ) ){
  99. ShowSQL( "Your MySQL version does not understand \"MYSQL_OPT_SSL_MODE\" yet. Please consider upgrading - especially if you encounter SSL related error messages from your MySQL server.\n" );
  100. }
  101. #endif
  102. if( !mysql_real_connect(&self->handle, host, user, passwd, db, (unsigned int)port, NULL/*unix_socket*/, 0/*clientflag*/) )
  103. {
  104. ShowSQL("%s\n", mysql_error(&self->handle));
  105. return SQL_ERROR;
  106. }
  107. self->keepalive = Sql_P_Keepalive(self);
  108. if( self->keepalive == INVALID_TIMER )
  109. {
  110. ShowSQL("Failed to establish keepalive for DB connection!\n");
  111. return SQL_ERROR;
  112. }
  113. return SQL_SUCCESS;
  114. }
  115. /// Retrieves the timeout of the connection.
  116. int Sql_GetTimeout(Sql* self, uint32* out_timeout)
  117. {
  118. if( self && out_timeout && SQL_SUCCESS == Sql_Query(self, "SHOW VARIABLES LIKE 'wait_timeout'") )
  119. {
  120. char* data;
  121. size_t len;
  122. if( SQL_SUCCESS == Sql_NextRow(self) &&
  123. SQL_SUCCESS == Sql_GetData(self, 1, &data, &len) )
  124. {
  125. *out_timeout = (uint32)strtoul(data, NULL, 10);
  126. Sql_FreeResult(self);
  127. return SQL_SUCCESS;
  128. }
  129. Sql_FreeResult(self);
  130. }
  131. return SQL_ERROR;
  132. }
  133. /// Retrieves the name of the columns of a table into out_buf, with the separator after each name.
  134. int Sql_GetColumnNames(Sql* self, const char* table, char* out_buf, size_t buf_len, char sep)
  135. {
  136. char* data;
  137. size_t len;
  138. size_t off = 0;
  139. if( self == NULL || SQL_ERROR == Sql_Query(self, "EXPLAIN `%s`", table) )
  140. return SQL_ERROR;
  141. out_buf[off] = '\0';
  142. while( SQL_SUCCESS == Sql_NextRow(self) && SQL_SUCCESS == Sql_GetData(self, 0, &data, &len) )
  143. {
  144. len = strnlen(data, len);
  145. if( off + len + 2 > buf_len )
  146. {
  147. ShowDebug("Sql_GetColumns: output buffer is too small\n");
  148. *out_buf = '\0';
  149. return SQL_ERROR;
  150. }
  151. memcpy(out_buf+off, data, len);
  152. off += len;
  153. out_buf[off++] = sep;
  154. }
  155. out_buf[off] = '\0';
  156. Sql_FreeResult(self);
  157. return SQL_SUCCESS;
  158. }
  159. /// Changes the encoding of the connection.
  160. int Sql_SetEncoding(Sql* self, const char* encoding)
  161. {
  162. if( self && Sql_Query(self, "SET NAMES %s", encoding) == 0 )
  163. return SQL_SUCCESS;
  164. return SQL_ERROR;
  165. }
  166. /// Pings the connection.
  167. int Sql_Ping(Sql* self)
  168. {
  169. if( self && mysql_ping(&self->handle) == 0 )
  170. return SQL_SUCCESS;
  171. return SQL_ERROR;
  172. }
  173. /// Wrapper function for Sql_Ping.
  174. ///
  175. /// @private
  176. static TIMER_FUNC(Sql_P_KeepaliveTimer){
  177. Sql* self = (Sql*)data;
  178. ShowInfo("Pinging SQL server to keep connection alive...\n");
  179. Sql_Ping(self);
  180. return 0;
  181. }
  182. /// Establishes keepalive (periodic ping) on the connection.
  183. ///
  184. /// @return the keepalive timer id, or INVALID_TIMER
  185. /// @private
  186. static int Sql_P_Keepalive(Sql* self)
  187. {
  188. uint32 timeout, ping_interval;
  189. // set a default value first
  190. timeout = 28800; // 8 hours
  191. // request the timeout value from the mysql server
  192. Sql_GetTimeout(self, &timeout);
  193. if( timeout < 60 )
  194. timeout = 60;
  195. // establish keepalive
  196. ping_interval = timeout - 30; // 30-second reserve
  197. //add_timer_func_list(Sql_P_KeepaliveTimer, "Sql_P_KeepaliveTimer");
  198. return add_timer_interval(gettick() + ping_interval*1000, Sql_P_KeepaliveTimer, 0, (intptr_t)self, ping_interval*1000);
  199. }
  200. /// Escapes a string.
  201. size_t Sql_EscapeString(Sql* self, char *out_to, const char *from)
  202. {
  203. if( self )
  204. return (size_t)mysql_real_escape_string(&self->handle, out_to, from, (unsigned long)strlen(from));
  205. else
  206. return (size_t)mysql_escape_string(out_to, from, (unsigned long)strlen(from));
  207. }
  208. /// Escapes a string.
  209. size_t Sql_EscapeStringLen(Sql* self, char *out_to, const char *from, size_t from_len)
  210. {
  211. if( self )
  212. return (size_t)mysql_real_escape_string(&self->handle, out_to, from, (unsigned long)from_len);
  213. else
  214. return (size_t)mysql_escape_string(out_to, from, (unsigned long)from_len);
  215. }
  216. /// Executes a query.
  217. int Sql_Query(Sql* self, const char* query, ...)
  218. {
  219. int res;
  220. va_list args;
  221. va_start(args, query);
  222. res = Sql_QueryV(self, query, args);
  223. va_end(args);
  224. return res;
  225. }
  226. /// Executes a query.
  227. int Sql_QueryV(Sql* self, const char* query, va_list args)
  228. {
  229. if( self == NULL )
  230. return SQL_ERROR;
  231. Sql_FreeResult(self);
  232. StringBuf_Clear(&self->buf);
  233. StringBuf_Vprintf(&self->buf, query, args);
  234. if( mysql_real_query(&self->handle, StringBuf_Value(&self->buf), (unsigned long)StringBuf_Length(&self->buf)) )
  235. {
  236. ShowSQL("DB error - %s\n", mysql_error(&self->handle));
  237. ra_mysql_error_handler(mysql_errno(&self->handle));
  238. return SQL_ERROR;
  239. }
  240. self->result = mysql_store_result(&self->handle);
  241. if( mysql_errno(&self->handle) != 0 )
  242. {
  243. ShowSQL("DB error - %s\n", mysql_error(&self->handle));
  244. ra_mysql_error_handler(mysql_errno(&self->handle));
  245. return SQL_ERROR;
  246. }
  247. return SQL_SUCCESS;
  248. }
  249. /// Executes a query.
  250. int Sql_QueryStr(Sql* self, const char* query)
  251. {
  252. if( self == NULL )
  253. return SQL_ERROR;
  254. Sql_FreeResult(self);
  255. StringBuf_Clear(&self->buf);
  256. StringBuf_AppendStr(&self->buf, query);
  257. if( mysql_real_query(&self->handle, StringBuf_Value(&self->buf), (unsigned long)StringBuf_Length(&self->buf)) )
  258. {
  259. ShowSQL("DB error - %s\n", mysql_error(&self->handle));
  260. ra_mysql_error_handler(mysql_errno(&self->handle));
  261. return SQL_ERROR;
  262. }
  263. self->result = mysql_store_result(&self->handle);
  264. if( mysql_errno(&self->handle) != 0 )
  265. {
  266. ShowSQL("DB error - %s\n", mysql_error(&self->handle));
  267. ra_mysql_error_handler(mysql_errno(&self->handle));
  268. return SQL_ERROR;
  269. }
  270. return SQL_SUCCESS;
  271. }
  272. /// Returns the number of the AUTO_INCREMENT column of the last INSERT/UPDATE query.
  273. uint64 Sql_LastInsertId(Sql* self)
  274. {
  275. if( self )
  276. return (uint64)mysql_insert_id(&self->handle);
  277. else
  278. return 0;
  279. }
  280. /// Returns the number of columns in each row of the result.
  281. uint32 Sql_NumColumns(Sql* self)
  282. {
  283. if( self && self->result )
  284. return (uint32)mysql_num_fields(self->result);
  285. return 0;
  286. }
  287. /// Returns the number of rows in the result.
  288. uint64 Sql_NumRows(Sql* self)
  289. {
  290. if( self && self->result )
  291. return (uint64)mysql_num_rows(self->result);
  292. return 0;
  293. }
  294. /// Returns the number of rows affected by the last query
  295. uint64 Sql_NumRowsAffected(Sql* self)
  296. {
  297. if( self )
  298. return (uint64)mysql_affected_rows(&self->handle);
  299. return 0;
  300. }
  301. /// Fetches the next row.
  302. int Sql_NextRow(Sql* self)
  303. {
  304. if( self && self->result )
  305. {
  306. self->row = mysql_fetch_row(self->result);
  307. if( self->row )
  308. {
  309. self->lengths = mysql_fetch_lengths(self->result);
  310. return SQL_SUCCESS;
  311. }
  312. self->lengths = NULL;
  313. if( mysql_errno(&self->handle) == 0 )
  314. return SQL_NO_DATA;
  315. }
  316. return SQL_ERROR;
  317. }
  318. /// Gets the data of a column.
  319. int Sql_GetData(Sql* self, size_t col, char** out_buf, size_t* out_len)
  320. {
  321. if( self && self->row )
  322. {
  323. if( col < Sql_NumColumns(self) )
  324. {
  325. if( out_buf ) *out_buf = self->row[col];
  326. if( out_len ) *out_len = (size_t)self->lengths[col];
  327. }
  328. else
  329. {// out of range - ignore
  330. if( out_buf ) *out_buf = NULL;
  331. if( out_len ) *out_len = 0;
  332. }
  333. return SQL_SUCCESS;
  334. }
  335. return SQL_ERROR;
  336. }
  337. /// Frees the result of the query.
  338. void Sql_FreeResult(Sql* self)
  339. {
  340. if( self && self->result )
  341. {
  342. mysql_free_result(self->result);
  343. self->result = NULL;
  344. self->row = NULL;
  345. self->lengths = NULL;
  346. }
  347. }
  348. /// Closes the handle
  349. void Sql_Close(Sql* self) {
  350. if (self) {
  351. mysql_close(&self->handle);
  352. }
  353. }
  354. /// Shows debug information (last query).
  355. void Sql_ShowDebug_(Sql* self, const char* debug_file, const unsigned long debug_line)
  356. {
  357. if( self == NULL )
  358. ShowDebug("at %s:%lu - self is NULL\n", debug_file, debug_line);
  359. else if( StringBuf_Length(&self->buf) > 0 )
  360. ShowDebug("at %s:%lu - %s\n", debug_file, debug_line, StringBuf_Value(&self->buf));
  361. else
  362. ShowDebug("at %s:%lu\n", debug_file, debug_line);
  363. }
  364. /// Frees a Sql handle returned by Sql_Malloc.
  365. void Sql_Free(Sql* self)
  366. {
  367. if( self )
  368. {
  369. Sql_FreeResult(self);
  370. StringBuf_Destroy(&self->buf);
  371. if( self->keepalive != INVALID_TIMER ) delete_timer(self->keepalive, Sql_P_KeepaliveTimer);
  372. Sql_Close(self);
  373. aFree(self);
  374. }
  375. }
  376. ///////////////////////////////////////////////////////////////////////////////
  377. // Prepared Statements
  378. ///////////////////////////////////////////////////////////////////////////////
  379. /// Returns the mysql integer type for the target size.
  380. ///
  381. /// @private
  382. static enum enum_field_types Sql_P_SizeToMysqlIntType(int sz)
  383. {
  384. switch( sz )
  385. {
  386. case 1: return MYSQL_TYPE_TINY;
  387. case 2: return MYSQL_TYPE_SHORT;
  388. case 4: return MYSQL_TYPE_LONG;
  389. case 8: return MYSQL_TYPE_LONGLONG;
  390. default:
  391. ShowDebug("SizeToMysqlIntType: unsupported size (%d)\n", sz);
  392. return MYSQL_TYPE_NULL;
  393. }
  394. }
  395. /// Binds a parameter/result.
  396. ///
  397. /// @private
  398. static int Sql_P_BindSqlDataType(MYSQL_BIND* bind, enum SqlDataType buffer_type, void* buffer, size_t buffer_len, unsigned long* out_length, int8* out_is_null)
  399. {
  400. memset(bind, 0, sizeof(MYSQL_BIND));
  401. switch( buffer_type )
  402. {
  403. case SQLDT_NULL: bind->buffer_type = MYSQL_TYPE_NULL;
  404. buffer_len = 0;// FIXME length = ? [FlavioJS]
  405. break;
  406. // fixed size
  407. case SQLDT_UINT8: bind->is_unsigned = 1;
  408. case SQLDT_INT8: bind->buffer_type = MYSQL_TYPE_TINY;
  409. buffer_len = 1;
  410. break;
  411. case SQLDT_UINT16: bind->is_unsigned = 1;
  412. case SQLDT_INT16: bind->buffer_type = MYSQL_TYPE_SHORT;
  413. buffer_len = 2;
  414. break;
  415. case SQLDT_UINT32: bind->is_unsigned = 1;
  416. case SQLDT_INT32: bind->buffer_type = MYSQL_TYPE_LONG;
  417. buffer_len = 4;
  418. break;
  419. case SQLDT_UINT64: bind->is_unsigned = 1;
  420. case SQLDT_INT64: bind->buffer_type = MYSQL_TYPE_LONGLONG;
  421. buffer_len = 8;
  422. break;
  423. // platform dependent size
  424. case SQLDT_UCHAR: bind->is_unsigned = 1;
  425. case SQLDT_CHAR: bind->buffer_type = Sql_P_SizeToMysqlIntType(sizeof(char));
  426. buffer_len = sizeof(char);
  427. break;
  428. case SQLDT_USHORT: bind->is_unsigned = 1;
  429. case SQLDT_SHORT: bind->buffer_type = Sql_P_SizeToMysqlIntType(sizeof(short));
  430. buffer_len = sizeof(short);
  431. break;
  432. case SQLDT_UINT: bind->is_unsigned = 1;
  433. case SQLDT_INT: bind->buffer_type = Sql_P_SizeToMysqlIntType(sizeof(int));
  434. buffer_len = sizeof(int);
  435. break;
  436. case SQLDT_ULONG: bind->is_unsigned = 1;
  437. case SQLDT_LONG: bind->buffer_type = Sql_P_SizeToMysqlIntType(sizeof(long));
  438. buffer_len = sizeof(long);
  439. break;
  440. case SQLDT_ULONGLONG: bind->is_unsigned = 1;
  441. case SQLDT_LONGLONG: bind->buffer_type = Sql_P_SizeToMysqlIntType(sizeof(int64));
  442. buffer_len = sizeof(int64);
  443. break;
  444. // floating point
  445. case SQLDT_FLOAT: bind->buffer_type = MYSQL_TYPE_FLOAT;
  446. buffer_len = 4;
  447. break;
  448. case SQLDT_DOUBLE: bind->buffer_type = MYSQL_TYPE_DOUBLE;
  449. buffer_len = 8;
  450. break;
  451. // other
  452. case SQLDT_STRING:
  453. case SQLDT_ENUM: bind->buffer_type = MYSQL_TYPE_STRING;
  454. break;
  455. case SQLDT_BLOB: bind->buffer_type = MYSQL_TYPE_BLOB;
  456. break;
  457. default:
  458. ShowDebug("Sql_P_BindSqlDataType: unsupported buffer type (%d)\n", buffer_type);
  459. return SQL_ERROR;
  460. }
  461. bind->buffer = buffer;
  462. bind->buffer_length = (unsigned long)buffer_len;
  463. bind->length = out_length;
  464. bind->is_null = (my_bool*)out_is_null;
  465. return SQL_SUCCESS;
  466. }
  467. /// Prints debug information about a field (type and length).
  468. ///
  469. /// @private
  470. static void Sql_P_ShowDebugMysqlFieldInfo(const char* prefix, enum enum_field_types type, int is_unsigned, unsigned long length, const char* length_postfix)
  471. {
  472. const char* sign = (is_unsigned ? "UNSIGNED " : "");
  473. const char* type_string;
  474. switch( type )
  475. {
  476. default:
  477. ShowDebug("%stype=%s%u, length=%lu\n", prefix, sign, type, length);
  478. return;
  479. #define SHOW_DEBUG_OF(x) case x: type_string = #x; break
  480. SHOW_DEBUG_OF(MYSQL_TYPE_TINY);
  481. SHOW_DEBUG_OF(MYSQL_TYPE_SHORT);
  482. SHOW_DEBUG_OF(MYSQL_TYPE_LONG);
  483. SHOW_DEBUG_OF(MYSQL_TYPE_INT24);
  484. SHOW_DEBUG_OF(MYSQL_TYPE_LONGLONG);
  485. SHOW_DEBUG_OF(MYSQL_TYPE_DECIMAL);
  486. SHOW_DEBUG_OF(MYSQL_TYPE_FLOAT);
  487. SHOW_DEBUG_OF(MYSQL_TYPE_DOUBLE);
  488. SHOW_DEBUG_OF(MYSQL_TYPE_TIMESTAMP);
  489. SHOW_DEBUG_OF(MYSQL_TYPE_DATE);
  490. SHOW_DEBUG_OF(MYSQL_TYPE_TIME);
  491. SHOW_DEBUG_OF(MYSQL_TYPE_DATETIME);
  492. SHOW_DEBUG_OF(MYSQL_TYPE_YEAR);
  493. SHOW_DEBUG_OF(MYSQL_TYPE_STRING);
  494. SHOW_DEBUG_OF(MYSQL_TYPE_VAR_STRING);
  495. SHOW_DEBUG_OF(MYSQL_TYPE_BLOB);
  496. SHOW_DEBUG_OF(MYSQL_TYPE_SET);
  497. SHOW_DEBUG_OF(MYSQL_TYPE_ENUM);
  498. SHOW_DEBUG_OF(MYSQL_TYPE_NULL);
  499. #undef SHOW_DEBUG_TYPE_OF
  500. }
  501. ShowDebug("%stype=%s%s, length=%lu%s\n", prefix, sign, type_string, length, length_postfix);
  502. }
  503. /// Reports debug information about a truncated column.
  504. ///
  505. /// @private
  506. static void SqlStmt_P_ShowDebugTruncatedColumn(SqlStmt* self, size_t i)
  507. {
  508. MYSQL_RES* meta;
  509. MYSQL_FIELD* field;
  510. MYSQL_BIND* column;
  511. meta = mysql_stmt_result_metadata(self->stmt);
  512. field = mysql_fetch_field_direct(meta, (unsigned int)i);
  513. ShowSQL("DB error - data of field '%s' was truncated.\n", field->name);
  514. ShowDebug("column - %lu\n", (unsigned long)i);
  515. Sql_P_ShowDebugMysqlFieldInfo("data - ", field->type, field->flags&UNSIGNED_FLAG, self->column_lengths[i].length, "");
  516. column = &self->columns[i];
  517. if( column->buffer_type == MYSQL_TYPE_STRING )
  518. Sql_P_ShowDebugMysqlFieldInfo("buffer - ", column->buffer_type, column->is_unsigned, column->buffer_length, "+1(nul-terminator)");
  519. else
  520. Sql_P_ShowDebugMysqlFieldInfo("buffer - ", column->buffer_type, column->is_unsigned, column->buffer_length, "");
  521. mysql_free_result(meta);
  522. }
  523. /// Allocates and initializes a new SqlStmt handle.
  524. SqlStmt* SqlStmt_Malloc(Sql* sql)
  525. {
  526. SqlStmt* self;
  527. MYSQL_STMT* stmt;
  528. if( sql == NULL )
  529. return NULL;
  530. stmt = mysql_stmt_init(&sql->handle);
  531. if( stmt == NULL )
  532. {
  533. ShowSQL("DB error - %s\n", mysql_error(&sql->handle));
  534. return NULL;
  535. }
  536. CREATE(self, SqlStmt, 1);
  537. StringBuf_Init(&self->buf);
  538. self->stmt = stmt;
  539. self->params = NULL;
  540. self->columns = NULL;
  541. self->column_lengths = NULL;
  542. self->max_params = 0;
  543. self->max_columns = 0;
  544. self->bind_params = false;
  545. self->bind_columns = false;
  546. return self;
  547. }
  548. /// Prepares the statement.
  549. int SqlStmt_Prepare(SqlStmt* self, const char* query, ...)
  550. {
  551. int res;
  552. va_list args;
  553. va_start(args, query);
  554. res = SqlStmt_PrepareV(self, query, args);
  555. va_end(args);
  556. return res;
  557. }
  558. /// Prepares the statement.
  559. int SqlStmt_PrepareV(SqlStmt* self, const char* query, va_list args)
  560. {
  561. if( self == NULL )
  562. return SQL_ERROR;
  563. SqlStmt_FreeResult(self);
  564. StringBuf_Clear(&self->buf);
  565. StringBuf_Vprintf(&self->buf, query, args);
  566. if( mysql_stmt_prepare(self->stmt, StringBuf_Value(&self->buf), (unsigned long)StringBuf_Length(&self->buf)) )
  567. {
  568. ShowSQL("DB error - %s\n", mysql_stmt_error(self->stmt));
  569. ra_mysql_error_handler(mysql_stmt_errno(self->stmt));
  570. return SQL_ERROR;
  571. }
  572. self->bind_params = false;
  573. return SQL_SUCCESS;
  574. }
  575. /// Prepares the statement.
  576. int SqlStmt_PrepareStr(SqlStmt* self, const char* query)
  577. {
  578. if( self == NULL )
  579. return SQL_ERROR;
  580. SqlStmt_FreeResult(self);
  581. StringBuf_Clear(&self->buf);
  582. StringBuf_AppendStr(&self->buf, query);
  583. if( mysql_stmt_prepare(self->stmt, StringBuf_Value(&self->buf), (unsigned long)StringBuf_Length(&self->buf)) )
  584. {
  585. ShowSQL("DB error - %s\n", mysql_stmt_error(self->stmt));
  586. ra_mysql_error_handler(mysql_stmt_errno(self->stmt));
  587. return SQL_ERROR;
  588. }
  589. self->bind_params = false;
  590. return SQL_SUCCESS;
  591. }
  592. /// Returns the number of parameters in the prepared statement.
  593. size_t SqlStmt_NumParams(SqlStmt* self)
  594. {
  595. if( self )
  596. return (size_t)mysql_stmt_param_count(self->stmt);
  597. else
  598. return 0;
  599. }
  600. /// Binds a parameter to a buffer.
  601. int SqlStmt_BindParam(SqlStmt* self, size_t idx, enum SqlDataType buffer_type, void* buffer, size_t buffer_len)
  602. {
  603. if( self == NULL )
  604. return SQL_ERROR;
  605. if( !self->bind_params )
  606. {// initialize the bindings
  607. size_t i;
  608. size_t count;
  609. count = SqlStmt_NumParams(self);
  610. if( self->max_params < count )
  611. {
  612. self->max_params = count;
  613. RECREATE(self->params, MYSQL_BIND, count);
  614. }
  615. memset(self->params, 0, count*sizeof(MYSQL_BIND));
  616. for( i = 0; i < count; ++i )
  617. self->params[i].buffer_type = MYSQL_TYPE_NULL;
  618. self->bind_params = true;
  619. }
  620. if( idx < self->max_params )
  621. return Sql_P_BindSqlDataType(self->params+idx, buffer_type, buffer, buffer_len, NULL, NULL);
  622. else
  623. return SQL_SUCCESS;// out of range - ignore
  624. }
  625. /// Executes the prepared statement.
  626. int SqlStmt_Execute(SqlStmt* self)
  627. {
  628. if( self == NULL )
  629. return SQL_ERROR;
  630. SqlStmt_FreeResult(self);
  631. if( (self->bind_params && mysql_stmt_bind_param(self->stmt, self->params)) ||
  632. mysql_stmt_execute(self->stmt) )
  633. {
  634. ShowSQL("DB error - %s\n", mysql_stmt_error(self->stmt));
  635. ra_mysql_error_handler(mysql_stmt_errno(self->stmt));
  636. return SQL_ERROR;
  637. }
  638. self->bind_columns = false;
  639. if( mysql_stmt_store_result(self->stmt) )// store all the data
  640. {
  641. ShowSQL("DB error - %s\n", mysql_stmt_error(self->stmt));
  642. ra_mysql_error_handler(mysql_stmt_errno(self->stmt));
  643. return SQL_ERROR;
  644. }
  645. return SQL_SUCCESS;
  646. }
  647. /// Returns the number of the AUTO_INCREMENT column of the last INSERT/UPDATE statement.
  648. uint64 SqlStmt_LastInsertId(SqlStmt* self)
  649. {
  650. if( self )
  651. return (uint64)mysql_stmt_insert_id(self->stmt);
  652. else
  653. return 0;
  654. }
  655. /// Returns the number of columns in each row of the result.
  656. size_t SqlStmt_NumColumns(SqlStmt* self)
  657. {
  658. if( self )
  659. return (size_t)mysql_stmt_field_count(self->stmt);
  660. else
  661. return 0;
  662. }
  663. /// Binds the result of a column to a buffer.
  664. int SqlStmt_BindColumn(SqlStmt* self, size_t idx, enum SqlDataType buffer_type, void* buffer, size_t buffer_len, uint32* out_length, int8* out_is_null)
  665. {
  666. if( self == NULL )
  667. return SQL_ERROR;
  668. if( buffer_type == SQLDT_STRING || buffer_type == SQLDT_ENUM )
  669. {
  670. if( buffer_len < 1 )
  671. {
  672. ShowDebug("SqlStmt_BindColumn: buffer_len(%" PRIuPTR ") is too small, no room for the nul-terminator\n", buffer_len);
  673. return SQL_ERROR;
  674. }
  675. --buffer_len;// nul-terminator
  676. }
  677. if( !self->bind_columns )
  678. {// initialize the bindings
  679. size_t i;
  680. size_t cols;
  681. cols = SqlStmt_NumColumns(self);
  682. if( self->max_columns < cols )
  683. {
  684. self->max_columns = cols;
  685. RECREATE(self->columns, MYSQL_BIND, cols);
  686. RECREATE(self->column_lengths, s_column_length, cols);
  687. }
  688. memset(self->columns, 0, cols*sizeof(MYSQL_BIND));
  689. memset(self->column_lengths, 0, cols*sizeof(s_column_length));
  690. for( i = 0; i < cols; ++i )
  691. self->columns[i].buffer_type = MYSQL_TYPE_NULL;
  692. self->bind_columns = true;
  693. }
  694. if( idx < self->max_columns )
  695. {
  696. self->column_lengths[idx].out_length = out_length;
  697. return Sql_P_BindSqlDataType(self->columns+idx, buffer_type, buffer, buffer_len, &self->column_lengths[idx].length, out_is_null);
  698. }
  699. else
  700. {
  701. return SQL_SUCCESS;// out of range - ignore
  702. }
  703. }
  704. /// Returns the number of rows in the result.
  705. uint64 SqlStmt_NumRows(SqlStmt* self)
  706. {
  707. if( self )
  708. return (uint64)mysql_stmt_num_rows(self->stmt);
  709. else
  710. return 0;
  711. }
  712. /// Fetches the next row.
  713. int SqlStmt_NextRow(SqlStmt* self)
  714. {
  715. int err;
  716. size_t i;
  717. size_t cols;
  718. if( self == NULL )
  719. return SQL_ERROR;
  720. // bind columns
  721. if( self->bind_columns && mysql_stmt_bind_result(self->stmt, self->columns) )
  722. err = 1;// error binding columns
  723. else
  724. err = mysql_stmt_fetch(self->stmt);// fetch row
  725. // check for errors
  726. if( err == MYSQL_NO_DATA )
  727. return SQL_NO_DATA;
  728. #if defined(MYSQL_DATA_TRUNCATED)
  729. // MySQL 5.0/5.1 defines and returns MYSQL_DATA_TRUNCATED [FlavioJS]
  730. if( err == MYSQL_DATA_TRUNCATED )
  731. {
  732. my_bool truncated;
  733. if( !self->bind_columns )
  734. {
  735. ShowSQL("DB error - data truncated (unknown source, columns are not bound)\n");
  736. return SQL_ERROR;
  737. }
  738. // find truncated column
  739. cols = SqlStmt_NumColumns(self);
  740. for( i = 0; i < cols; ++i )
  741. {
  742. MYSQL_BIND* column = &self->columns[i];
  743. column->error = &truncated;
  744. mysql_stmt_fetch_column(self->stmt, column, (unsigned int)i, 0);
  745. column->error = NULL;
  746. if( truncated )
  747. {// report truncated column
  748. SqlStmt_P_ShowDebugTruncatedColumn(self, i);
  749. return SQL_ERROR;
  750. }
  751. }
  752. ShowSQL("DB error - data truncated (unknown source)\n");
  753. return SQL_ERROR;
  754. }
  755. #endif
  756. if( err )
  757. {
  758. ShowSQL("DB error - %s\n", mysql_stmt_error(self->stmt));
  759. ra_mysql_error_handler(mysql_stmt_errno(self->stmt));
  760. return SQL_ERROR;
  761. }
  762. // propagate column lengths and clear unused parts of string/enum/blob buffers
  763. cols = SqlStmt_NumColumns(self);
  764. for( i = 0; i < cols; ++i )
  765. {
  766. unsigned long length = self->column_lengths[i].length;
  767. MYSQL_BIND* column = &self->columns[i];
  768. #if !defined(MYSQL_DATA_TRUNCATED)
  769. // MySQL 4.1/(below?) returns success even if data is truncated, so we test truncation manually [FlavioJS]
  770. if( column->buffer_length < length )
  771. {// report truncated column
  772. if( column->buffer_type == MYSQL_TYPE_STRING || column->buffer_type == MYSQL_TYPE_BLOB )
  773. {// string/enum/blob column
  774. SqlStmt_P_ShowDebugTruncatedColumn(self, i);
  775. return SQL_ERROR;
  776. }
  777. // FIXME numeric types and null [FlavioJS]
  778. }
  779. #endif
  780. if( self->column_lengths[i].out_length )
  781. *self->column_lengths[i].out_length = (uint32)length;
  782. if( column->buffer_type == MYSQL_TYPE_STRING )
  783. {// clear unused part of the string/enum buffer (and nul-terminate)
  784. memset((char*)column->buffer + length, 0, column->buffer_length - length + 1);
  785. }
  786. else if( column->buffer_type == MYSQL_TYPE_BLOB && length < column->buffer_length )
  787. {// clear unused part of the blob buffer
  788. memset((char*)column->buffer + length, 0, column->buffer_length - length);
  789. }
  790. }
  791. return SQL_SUCCESS;
  792. }
  793. /// Frees the result of the statement execution.
  794. void SqlStmt_FreeResult(SqlStmt* self)
  795. {
  796. if( self )
  797. mysql_stmt_free_result(self->stmt);
  798. }
  799. /// Shows debug information (with statement).
  800. void SqlStmt_ShowDebug_(SqlStmt* self, const char* debug_file, const unsigned long debug_line)
  801. {
  802. if( self == NULL )
  803. ShowDebug("at %s:%lu - self is NULL\n", debug_file, debug_line);
  804. else if( StringBuf_Length(&self->buf) > 0 )
  805. ShowDebug("at %s:%lu - %s\n", debug_file, debug_line, StringBuf_Value(&self->buf));
  806. else
  807. ShowDebug("at %s:%lu\n", debug_file, debug_line);
  808. }
  809. /// Frees a SqlStmt returned by SqlStmt_Malloc.
  810. void SqlStmt_Free(SqlStmt* self)
  811. {
  812. if( self )
  813. {
  814. SqlStmt_FreeResult(self);
  815. StringBuf_Destroy(&self->buf);
  816. mysql_stmt_close(self->stmt);
  817. if( self->params )
  818. aFree(self->params);
  819. if( self->columns )
  820. {
  821. aFree(self->columns);
  822. aFree(self->column_lengths);
  823. }
  824. aFree(self);
  825. }
  826. }
  827. /// Receives MySQL error codes during runtime (not on first-time-connects).
  828. void ra_mysql_error_handler(unsigned int ecode) {
  829. switch( ecode ) {
  830. case 2003:// Can't connect to MySQL (this error only happens here when failing to reconnect)
  831. if( mysql_reconnect_type == 1 ) {
  832. static unsigned int retry = 1;
  833. if( ++retry > mysql_reconnect_count ) {
  834. ShowFatalError("MySQL has been unreachable for too long, %d reconnects were attempted. Shutting Down\n", retry);
  835. exit(EXIT_FAILURE);
  836. }
  837. }
  838. break;
  839. }
  840. }
  841. void Sql_inter_server_read(const char* cfgName, bool first) {
  842. char line[1024], w1[1024], w2[1024];
  843. FILE* fp;
  844. fp = fopen(cfgName, "r");
  845. if(fp == NULL) {
  846. if( first ) {
  847. ShowFatalError("File not found: %s\n", cfgName);
  848. exit(EXIT_FAILURE);
  849. } else
  850. ShowError("File not found: %s\n", cfgName);
  851. return;
  852. }
  853. while(fgets(line, sizeof(line), fp)) {
  854. int i = sscanf(line, "%1023[^:]: %1023[^\r\n]", w1, w2);
  855. if(i != 2)
  856. continue;
  857. if(!strcmpi(w1,"mysql_reconnect_type")) {
  858. mysql_reconnect_type = atoi(w2);
  859. switch( mysql_reconnect_type ) {
  860. case 1:
  861. case 2:
  862. break;
  863. default:
  864. ShowError("%s::mysql_reconnect_type is set to %d which is not valid, defaulting to 1...\n", cfgName, mysql_reconnect_type);
  865. mysql_reconnect_type = 1;
  866. break;
  867. }
  868. } else if(!strcmpi(w1,"mysql_reconnect_count")) {
  869. mysql_reconnect_count = atoi(w2);
  870. if( mysql_reconnect_count < 1 )
  871. mysql_reconnect_count = 1;
  872. } else if(!strcmpi(w1,"import"))
  873. Sql_inter_server_read(w2,false);
  874. }
  875. fclose(fp);
  876. return;
  877. }
  878. void Sql_Init(void) {
  879. Sql_inter_server_read(SQL_CONF_NAME,true);
  880. }
  881. #ifdef my_bool
  882. #undef my_bool
  883. #endif