mysqlnd_driver.c 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321
  1. /*
  2. +----------------------------------------------------------------------+
  3. | PHP Version 5 |
  4. +----------------------------------------------------------------------+
  5. | Copyright (c) 2006-2016 The PHP Group |
  6. +----------------------------------------------------------------------+
  7. | This source file is subject to version 3.01 of the PHP license, |
  8. | that is bundled with this package in the file LICENSE, and is |
  9. | available through the world-wide-web at the following url: |
  10. | http://www.php.net/license/3_01.txt |
  11. | If you did not receive a copy of the PHP license and are unable to |
  12. | obtain it through the world-wide-web, please send a note to |
  13. | license@php.net so we can mail you a copy immediately. |
  14. +----------------------------------------------------------------------+
  15. | Authors: Andrey Hristov <andrey@mysql.com> |
  16. | Ulf Wendel <uwendel@mysql.com> |
  17. | Georg Richter <georg@mysql.com> |
  18. +----------------------------------------------------------------------+
  19. */
  20. /* $Id: mysqlnd.c 317989 2011-10-10 20:49:28Z andrey $ */
  21. #include "php.h"
  22. #include "mysqlnd.h"
  23. #include "mysqlnd_wireprotocol.h"
  24. #include "mysqlnd_priv.h"
  25. #include "mysqlnd_result.h"
  26. #include "mysqlnd_statistics.h"
  27. #include "mysqlnd_charset.h"
  28. #include "mysqlnd_debug.h"
  29. #include "mysqlnd_reverse_api.h"
  30. #include "mysqlnd_ext_plugin.h"
  31. static zend_bool mysqlnd_library_initted = FALSE;
  32. static struct st_mysqlnd_plugin_core mysqlnd_plugin_core =
  33. {
  34. {
  35. MYSQLND_PLUGIN_API_VERSION,
  36. "mysqlnd",
  37. MYSQLND_VERSION_ID,
  38. MYSQLND_VERSION,
  39. "PHP License 3.01",
  40. "Andrey Hristov <andrey@mysql.com>, Ulf Wendel <uwendel@mysql.com>, Georg Richter <georg@mysql.com>",
  41. {
  42. NULL, /* will be filled later */
  43. mysqlnd_stats_values_names,
  44. },
  45. {
  46. NULL /* plugin shutdown */
  47. }
  48. }
  49. };
  50. /* {{{ mysqlnd_library_end */
  51. PHPAPI void mysqlnd_library_end(TSRMLS_D)
  52. {
  53. if (mysqlnd_library_initted == TRUE) {
  54. mysqlnd_plugin_subsystem_end(TSRMLS_C);
  55. mysqlnd_stats_end(mysqlnd_global_stats);
  56. mysqlnd_global_stats = NULL;
  57. mysqlnd_library_initted = FALSE;
  58. mysqlnd_reverse_api_end(TSRMLS_C);
  59. }
  60. }
  61. /* }}} */
  62. /* {{{ mysqlnd_library_init */
  63. PHPAPI void mysqlnd_library_init(TSRMLS_D)
  64. {
  65. if (mysqlnd_library_initted == FALSE) {
  66. mysqlnd_library_initted = TRUE;
  67. mysqlnd_conn_set_methods(&MYSQLND_CLASS_METHOD_TABLE_NAME(mysqlnd_conn));
  68. mysqlnd_conn_data_set_methods(&MYSQLND_CLASS_METHOD_TABLE_NAME(mysqlnd_conn_data));
  69. _mysqlnd_init_ps_subsystem();
  70. /* Should be calloc, as mnd_calloc will reference LOCK_access*/
  71. mysqlnd_stats_init(&mysqlnd_global_stats, STAT_LAST);
  72. mysqlnd_plugin_subsystem_init(TSRMLS_C);
  73. {
  74. mysqlnd_plugin_core.plugin_header.plugin_stats.values = mysqlnd_global_stats;
  75. mysqlnd_plugin_register_ex((struct st_mysqlnd_plugin_header *) &mysqlnd_plugin_core TSRMLS_CC);
  76. }
  77. #if defined(MYSQLND_DBG_ENABLED) && MYSQLND_DBG_ENABLED == 1
  78. mysqlnd_example_plugin_register(TSRMLS_C);
  79. #endif
  80. mysqlnd_debug_trace_plugin_register(TSRMLS_C);
  81. mysqlnd_register_builtin_authentication_plugins(TSRMLS_C);
  82. mysqlnd_reverse_api_init(TSRMLS_C);
  83. }
  84. }
  85. /* }}} */
  86. /* {{{ mysqlnd_error_list_pdtor */
  87. static void
  88. mysqlnd_error_list_pdtor(void * pDest)
  89. {
  90. MYSQLND_ERROR_LIST_ELEMENT * element = (MYSQLND_ERROR_LIST_ELEMENT *) pDest;
  91. #ifdef ZTS
  92. TSRMLS_FETCH();
  93. #endif
  94. DBG_ENTER("mysqlnd_error_list_pdtor");
  95. if (element->error) {
  96. mnd_pefree(element->error, TRUE);
  97. }
  98. DBG_VOID_RETURN;
  99. }
  100. /* }}} */
  101. /* {{{ mysqlnd_object_factory::get_connection */
  102. static MYSQLND *
  103. MYSQLND_METHOD(mysqlnd_object_factory, get_connection)(zend_bool persistent TSRMLS_DC)
  104. {
  105. size_t alloc_size_ret = sizeof(MYSQLND) + mysqlnd_plugin_count() * sizeof(void *);
  106. size_t alloc_size_ret_data = sizeof(MYSQLND_CONN_DATA) + mysqlnd_plugin_count() * sizeof(void *);
  107. MYSQLND * new_object;
  108. MYSQLND_CONN_DATA * data;
  109. DBG_ENTER("mysqlnd_driver::get_connection");
  110. DBG_INF_FMT("persistent=%u", persistent);
  111. new_object = mnd_pecalloc(1, alloc_size_ret, persistent);
  112. if (!new_object) {
  113. DBG_RETURN(NULL);
  114. }
  115. new_object->data = mnd_pecalloc(1, alloc_size_ret_data, persistent);
  116. if (!new_object->data) {
  117. mnd_pefree(new_object, persistent);
  118. DBG_RETURN(NULL);
  119. }
  120. new_object->persistent = persistent;
  121. new_object->m = mysqlnd_conn_get_methods();
  122. data = new_object->data;
  123. data->error_info = &(data->error_info_impl);
  124. data->options = &(data->options_impl);
  125. data->upsert_status = &(data->upsert_status_impl);
  126. data->persistent = persistent;
  127. data->m = mysqlnd_conn_data_get_methods();
  128. CONN_SET_STATE(data, CONN_ALLOCED);
  129. data->m->get_reference(data TSRMLS_CC);
  130. if (PASS != data->m->init(data TSRMLS_CC)) {
  131. new_object->m->dtor(new_object TSRMLS_CC);
  132. DBG_RETURN(NULL);
  133. }
  134. data->error_info->error_list = mnd_pecalloc(1, sizeof(zend_llist), persistent);
  135. if (!data->error_info->error_list) {
  136. new_object->m->dtor(new_object TSRMLS_CC);
  137. DBG_RETURN(NULL);
  138. } else {
  139. zend_llist_init(data->error_info->error_list, sizeof(MYSQLND_ERROR_LIST_ELEMENT), (llist_dtor_func_t)mysqlnd_error_list_pdtor, persistent);
  140. }
  141. DBG_RETURN(new_object);
  142. }
  143. /* }}} */
  144. /* {{{ mysqlnd_object_factory::clone_connection_object */
  145. static MYSQLND *
  146. MYSQLND_METHOD(mysqlnd_object_factory, clone_connection_object)(MYSQLND * to_be_cloned TSRMLS_DC)
  147. {
  148. size_t alloc_size_ret = sizeof(MYSQLND) + mysqlnd_plugin_count() * sizeof(void *);
  149. MYSQLND * new_object;
  150. DBG_ENTER("mysqlnd_driver::clone_connection_object");
  151. DBG_INF_FMT("persistent=%u", to_be_cloned->persistent);
  152. if (!to_be_cloned || !to_be_cloned->data) {
  153. DBG_RETURN(NULL);
  154. }
  155. new_object = mnd_pecalloc(1, alloc_size_ret, to_be_cloned->persistent);
  156. if (!new_object) {
  157. DBG_RETURN(NULL);
  158. }
  159. new_object->persistent = to_be_cloned->persistent;
  160. new_object->m = to_be_cloned->m;
  161. new_object->data = to_be_cloned->data->m->get_reference(to_be_cloned->data TSRMLS_CC);
  162. if (!new_object->data) {
  163. new_object->m->dtor(new_object TSRMLS_CC);
  164. new_object = NULL;
  165. }
  166. DBG_RETURN(new_object);
  167. }
  168. /* }}} */
  169. /* {{{ mysqlnd_object_factory::get_prepared_statement */
  170. static MYSQLND_STMT *
  171. MYSQLND_METHOD(mysqlnd_object_factory, get_prepared_statement)(MYSQLND_CONN_DATA * const conn TSRMLS_DC)
  172. {
  173. size_t alloc_size = sizeof(MYSQLND_STMT) + mysqlnd_plugin_count() * sizeof(void *);
  174. MYSQLND_STMT * ret = mnd_pecalloc(1, alloc_size, conn->persistent);
  175. MYSQLND_STMT_DATA * stmt = NULL;
  176. DBG_ENTER("mysqlnd_object_factory::get_prepared_statement");
  177. do {
  178. if (!ret) {
  179. break;
  180. }
  181. ret->m = mysqlnd_stmt_get_methods();
  182. ret->persistent = conn->persistent;
  183. stmt = ret->data = mnd_pecalloc(1, sizeof(MYSQLND_STMT_DATA), conn->persistent);
  184. DBG_INF_FMT("stmt=%p", stmt);
  185. if (!stmt) {
  186. break;
  187. }
  188. stmt->persistent = conn->persistent;
  189. stmt->error_info = &(stmt->error_info_impl);
  190. stmt->upsert_status = &(stmt->upsert_status_impl);
  191. stmt->state = MYSQLND_STMT_INITTED;
  192. stmt->execute_cmd_buffer.length = 4096;
  193. stmt->execute_cmd_buffer.buffer = mnd_pemalloc(stmt->execute_cmd_buffer.length, stmt->persistent);
  194. if (!stmt->execute_cmd_buffer.buffer) {
  195. break;
  196. }
  197. stmt->prefetch_rows = MYSQLND_DEFAULT_PREFETCH_ROWS;
  198. /*
  199. Mark that we reference the connection, thus it won't be
  200. be destructed till there is open statements. The last statement
  201. or normal query result will close it then.
  202. */
  203. stmt->conn = conn->m->get_reference(conn TSRMLS_CC);
  204. stmt->error_info->error_list = mnd_pecalloc(1, sizeof(zend_llist), ret->persistent);
  205. if (!stmt->error_info->error_list) {
  206. break;
  207. }
  208. zend_llist_init(stmt->error_info->error_list, sizeof(MYSQLND_ERROR_LIST_ELEMENT), (llist_dtor_func_t) mysqlnd_error_list_pdtor, conn->persistent);
  209. DBG_RETURN(ret);
  210. } while (0);
  211. SET_OOM_ERROR(*conn->error_info);
  212. if (ret) {
  213. ret->m->dtor(ret, TRUE TSRMLS_CC);
  214. ret = NULL;
  215. }
  216. DBG_RETURN(NULL);
  217. }
  218. /* }}} */
  219. /* {{{ mysqlnd_object_factory::get_io_channel */
  220. PHPAPI MYSQLND_NET *
  221. MYSQLND_METHOD(mysqlnd_object_factory, get_io_channel)(zend_bool persistent, MYSQLND_STATS * stats, MYSQLND_ERROR_INFO * error_info TSRMLS_DC)
  222. {
  223. size_t net_alloc_size = sizeof(MYSQLND_NET) + mysqlnd_plugin_count() * sizeof(void *);
  224. size_t net_data_alloc_size = sizeof(MYSQLND_NET_DATA) + mysqlnd_plugin_count() * sizeof(void *);
  225. MYSQLND_NET * net = mnd_pecalloc(1, net_alloc_size, persistent);
  226. MYSQLND_NET_DATA * net_data = mnd_pecalloc(1, net_data_alloc_size, persistent);
  227. DBG_ENTER("mysqlnd_object_factory::get_io_channel");
  228. DBG_INF_FMT("persistent=%u", persistent);
  229. if (net && net_data) {
  230. net->data = net_data;
  231. net->persistent = net->data->persistent = persistent;
  232. net->data->m = *mysqlnd_net_get_methods();
  233. if (PASS != net->data->m.init(net, stats, error_info TSRMLS_CC)) {
  234. net->data->m.dtor(net, stats, error_info TSRMLS_CC);
  235. net = NULL;
  236. }
  237. } else {
  238. if (net_data) {
  239. mnd_pefree(net_data, persistent);
  240. net_data = NULL;
  241. }
  242. if (net) {
  243. mnd_pefree(net, persistent);
  244. net = NULL;
  245. }
  246. }
  247. DBG_RETURN(net);
  248. }
  249. /* }}} */
  250. /* {{{ mysqlnd_object_factory::get_protocol_decoder */
  251. static MYSQLND_PROTOCOL *
  252. MYSQLND_METHOD(mysqlnd_object_factory, get_protocol_decoder)(zend_bool persistent TSRMLS_DC)
  253. {
  254. size_t alloc_size = sizeof(MYSQLND_PROTOCOL) + mysqlnd_plugin_count() * sizeof(void *);
  255. MYSQLND_PROTOCOL *ret = mnd_pecalloc(1, alloc_size, persistent);
  256. DBG_ENTER("mysqlnd_object_factory::get_protocol_decoder");
  257. DBG_INF_FMT("persistent=%u", persistent);
  258. if (ret) {
  259. ret->persistent = persistent;
  260. ret->m = mysqlnd_mysqlnd_protocol_methods;
  261. }
  262. DBG_RETURN(ret);
  263. }
  264. /* }}} */
  265. PHPAPI MYSQLND_CLASS_METHODS_START(mysqlnd_object_factory)
  266. MYSQLND_METHOD(mysqlnd_object_factory, get_connection),
  267. MYSQLND_METHOD(mysqlnd_object_factory, clone_connection_object),
  268. MYSQLND_METHOD(mysqlnd_object_factory, get_prepared_statement),
  269. MYSQLND_METHOD(mysqlnd_object_factory, get_io_channel),
  270. MYSQLND_METHOD(mysqlnd_object_factory, get_protocol_decoder)
  271. MYSQLND_CLASS_METHODS_END;
  272. /*
  273. * Local variables:
  274. * tab-width: 4
  275. * c-basic-offset: 4
  276. * End:
  277. * vim600: noet sw=4 ts=4 fdm=marker
  278. * vim<600: noet sw=4 ts=4
  279. */