mysqlnd_connection.c 69 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357
  1. /*
  2. +----------------------------------------------------------------------+
  3. | Copyright (c) The PHP Group |
  4. +----------------------------------------------------------------------+
  5. | This source file is subject to version 3.01 of the PHP license, |
  6. | that is bundled with this package in the file LICENSE, and is |
  7. | available through the world-wide-web at the following url: |
  8. | https://www.php.net/license/3_01.txt |
  9. | If you did not receive a copy of the PHP license and are unable to |
  10. | obtain it through the world-wide-web, please send a note to |
  11. | license@php.net so we can mail you a copy immediately. |
  12. +----------------------------------------------------------------------+
  13. | Authors: Andrey Hristov <andrey@php.net> |
  14. | Ulf Wendel <uw@php.net> |
  15. +----------------------------------------------------------------------+
  16. */
  17. #include "php.h"
  18. #include "mysqlnd.h"
  19. #include "mysqlnd_connection.h"
  20. #include "mysqlnd_vio.h"
  21. #include "mysqlnd_protocol_frame_codec.h"
  22. #include "mysqlnd_auth.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_ext_plugin.h"
  30. #include "zend_smart_str.h"
  31. extern MYSQLND_CHARSET *mysqlnd_charsets;
  32. PHPAPI const char * const mysqlnd_server_gone = "MySQL server has gone away";
  33. PHPAPI const char * const mysqlnd_out_of_sync = "Commands out of sync; you can't run this command now";
  34. PHPAPI const char * const mysqlnd_out_of_memory = "Out of memory";
  35. PHPAPI MYSQLND_STATS * mysqlnd_global_stats = NULL;
  36. /* {{{ mysqlnd_upsert_status::reset */
  37. void
  38. MYSQLND_METHOD(mysqlnd_upsert_status, reset)(MYSQLND_UPSERT_STATUS * const upsert_status)
  39. {
  40. upsert_status->warning_count = 0;
  41. upsert_status->server_status = 0;
  42. upsert_status->affected_rows = 0;
  43. upsert_status->last_insert_id = 0;
  44. }
  45. /* }}} */
  46. /* {{{ mysqlnd_upsert_status::set_affected_rows_to_error */
  47. void
  48. MYSQLND_METHOD(mysqlnd_upsert_status, set_affected_rows_to_error)(MYSQLND_UPSERT_STATUS * upsert_status)
  49. {
  50. upsert_status->affected_rows = (uint64_t) ~0;
  51. }
  52. /* }}} */
  53. MYSQLND_CLASS_METHODS_START(mysqlnd_upsert_status)
  54. MYSQLND_METHOD(mysqlnd_upsert_status, reset),
  55. MYSQLND_METHOD(mysqlnd_upsert_status, set_affected_rows_to_error),
  56. MYSQLND_CLASS_METHODS_END;
  57. /* {{{ mysqlnd_upsert_status_init */
  58. void
  59. mysqlnd_upsert_status_init(MYSQLND_UPSERT_STATUS * const upsert_status)
  60. {
  61. upsert_status->m = &MYSQLND_CLASS_METHOD_TABLE_NAME(mysqlnd_upsert_status);
  62. upsert_status->m->reset(upsert_status);
  63. }
  64. /* }}} */
  65. /* {{{ mysqlnd_error_list_pdtor */
  66. static void
  67. mysqlnd_error_list_pdtor(void * pDest)
  68. {
  69. MYSQLND_ERROR_LIST_ELEMENT * element = (MYSQLND_ERROR_LIST_ELEMENT *) pDest;
  70. DBG_ENTER("mysqlnd_error_list_pdtor");
  71. if (element->error) {
  72. mnd_pefree(element->error, TRUE);
  73. }
  74. DBG_VOID_RETURN;
  75. }
  76. /* }}} */
  77. /* {{{ mysqlnd_error_info::reset */
  78. static void
  79. MYSQLND_METHOD(mysqlnd_error_info, reset)(MYSQLND_ERROR_INFO * const info)
  80. {
  81. DBG_ENTER("mysqlnd_error_info::reset");
  82. info->error_no = 0;
  83. info->error[0] = '\0';
  84. memset(&info->sqlstate, 0, sizeof(info->sqlstate));
  85. zend_llist_clean(&info->error_list);
  86. DBG_VOID_RETURN;
  87. }
  88. /* }}} */
  89. /* {{{ mysqlnd_error_info::set_client_error */
  90. static void
  91. MYSQLND_METHOD(mysqlnd_error_info, set_client_error)(MYSQLND_ERROR_INFO * const info,
  92. const unsigned int err_no,
  93. const char * const sqlstate,
  94. const char * const error)
  95. {
  96. DBG_ENTER("mysqlnd_error_info::set_client_error");
  97. if (err_no) {
  98. MYSQLND_ERROR_LIST_ELEMENT error_for_the_list = {0};
  99. info->error_no = err_no;
  100. strlcpy(info->sqlstate, sqlstate, sizeof(info->sqlstate));
  101. strlcpy(info->error, error, sizeof(info->error));
  102. error_for_the_list.error_no = err_no;
  103. strlcpy(error_for_the_list.sqlstate, sqlstate, sizeof(error_for_the_list.sqlstate));
  104. error_for_the_list.error = mnd_pestrdup(error, TRUE);
  105. if (error_for_the_list.error) {
  106. DBG_INF_FMT("adding error [%s] to the list", error_for_the_list.error);
  107. zend_llist_add_element(&info->error_list, &error_for_the_list);
  108. }
  109. } else {
  110. info->m->reset(info);
  111. }
  112. DBG_VOID_RETURN;
  113. }
  114. /* }}} */
  115. MYSQLND_CLASS_METHODS_START(mysqlnd_error_info)
  116. MYSQLND_METHOD(mysqlnd_error_info, reset),
  117. MYSQLND_METHOD(mysqlnd_error_info, set_client_error),
  118. MYSQLND_CLASS_METHODS_END;
  119. /* {{{ mysqlnd_error_info_init */
  120. PHPAPI void
  121. mysqlnd_error_info_init(MYSQLND_ERROR_INFO * const info, const bool persistent)
  122. {
  123. DBG_ENTER("mysqlnd_error_info_init");
  124. info->m = mysqlnd_error_info_get_methods();
  125. info->m->reset(info);
  126. zend_llist_init(&info->error_list, sizeof(MYSQLND_ERROR_LIST_ELEMENT), (llist_dtor_func_t) mysqlnd_error_list_pdtor, persistent);
  127. info->persistent = persistent;
  128. DBG_VOID_RETURN;
  129. }
  130. /* }}} */
  131. /* {{{ mysqlnd_error_info_free_contents */
  132. PHPAPI void
  133. mysqlnd_error_info_free_contents(MYSQLND_ERROR_INFO * const info)
  134. {
  135. DBG_ENTER("mysqlnd_error_info_free_contents");
  136. info->m->reset(info);
  137. DBG_VOID_RETURN;
  138. }
  139. /* }}} */
  140. /* {{{ mysqlnd_connection_state::get */
  141. static enum mysqlnd_connection_state
  142. MYSQLND_METHOD(mysqlnd_connection_state, get)(const struct st_mysqlnd_connection_state * const state_struct)
  143. {
  144. DBG_ENTER("mysqlnd_connection_state::get");
  145. DBG_INF_FMT("State=%u", state_struct->state);
  146. DBG_RETURN(state_struct->state);
  147. }
  148. /* }}} */
  149. /* {{{ mysqlnd_connection_state::set */
  150. static void
  151. MYSQLND_METHOD(mysqlnd_connection_state, set)(struct st_mysqlnd_connection_state * const state_struct, const enum mysqlnd_connection_state state)
  152. {
  153. DBG_ENTER("mysqlnd_connection_state::set");
  154. DBG_INF_FMT("New state=%u", state);
  155. state_struct->state = state;
  156. DBG_VOID_RETURN;
  157. }
  158. /* }}} */
  159. MYSQLND_CLASS_METHODS_START(mysqlnd_connection_state)
  160. MYSQLND_METHOD(mysqlnd_connection_state, get),
  161. MYSQLND_METHOD(mysqlnd_connection_state, set),
  162. MYSQLND_CLASS_METHODS_END;
  163. /* {{{ mysqlnd_connection_state_init */
  164. PHPAPI void
  165. mysqlnd_connection_state_init(struct st_mysqlnd_connection_state * const state)
  166. {
  167. DBG_ENTER("mysqlnd_connection_state_init");
  168. state->m = &MYSQLND_CLASS_METHOD_TABLE_NAME(mysqlnd_connection_state);
  169. state->state = CONN_ALLOCED;
  170. DBG_VOID_RETURN;
  171. }
  172. /* }}} */
  173. /* {{{ mysqlnd_conn_data::free_options */
  174. static void
  175. MYSQLND_METHOD(mysqlnd_conn_data, free_options)(MYSQLND_CONN_DATA * conn)
  176. {
  177. bool pers = conn->persistent;
  178. if (conn->options->charset_name) {
  179. mnd_pefree(conn->options->charset_name, pers);
  180. conn->options->charset_name = NULL;
  181. }
  182. if (conn->options->auth_protocol) {
  183. mnd_pefree(conn->options->auth_protocol, pers);
  184. conn->options->auth_protocol = NULL;
  185. }
  186. if (conn->options->num_commands) {
  187. unsigned int i;
  188. for (i = 0; i < conn->options->num_commands; i++) {
  189. /* allocated with pestrdup */
  190. mnd_pefree(conn->options->init_commands[i], pers);
  191. }
  192. mnd_pefree(conn->options->init_commands, pers);
  193. conn->options->init_commands = NULL;
  194. }
  195. if (conn->options->cfg_file) {
  196. mnd_pefree(conn->options->cfg_file, pers);
  197. conn->options->cfg_file = NULL;
  198. }
  199. if (conn->options->cfg_section) {
  200. mnd_pefree(conn->options->cfg_section, pers);
  201. conn->options->cfg_section = NULL;
  202. }
  203. if (conn->options->connect_attr) {
  204. zend_hash_destroy(conn->options->connect_attr);
  205. mnd_pefree(conn->options->connect_attr, pers);
  206. conn->options->connect_attr = NULL;
  207. }
  208. if (conn->options->local_infile_directory) {
  209. mnd_pefree(conn->options->local_infile_directory, pers);
  210. conn->options->local_infile_directory = NULL;
  211. }
  212. }
  213. /* }}} */
  214. /* {{{ mysqlnd_conn_data::free_contents */
  215. static void
  216. MYSQLND_METHOD(mysqlnd_conn_data, free_contents)(MYSQLND_CONN_DATA * conn)
  217. {
  218. bool pers = conn->persistent;
  219. DBG_ENTER("mysqlnd_conn_data::free_contents");
  220. if (conn->current_result) {
  221. conn->current_result->m.free_result(conn->current_result, TRUE);
  222. conn->current_result = NULL;
  223. }
  224. if (conn->protocol_frame_codec) {
  225. conn->protocol_frame_codec->data->m.free_contents(conn->protocol_frame_codec);
  226. }
  227. if (conn->vio) {
  228. conn->vio->data->m.free_contents(conn->vio);
  229. }
  230. DBG_INF("Freeing memory of members");
  231. mysqlnd_set_persistent_string(&conn->hostname, NULL, 0, pers);
  232. mysqlnd_set_persistent_string(&conn->username, NULL, 0, pers);
  233. mysqlnd_set_persistent_string(&conn->password, NULL, 0, pers);
  234. mysqlnd_set_persistent_string(&conn->connect_or_select_db, NULL, 0, pers);
  235. mysqlnd_set_persistent_string(&conn->unix_socket, NULL, 0, pers);
  236. DBG_INF_FMT("scheme=%s", conn->scheme.s);
  237. mysqlnd_set_persistent_string(&conn->scheme, NULL, 0, pers);
  238. if (conn->server_version) {
  239. mnd_pefree(conn->server_version, pers);
  240. conn->server_version = NULL;
  241. }
  242. if (conn->host_info) {
  243. mnd_pefree(conn->host_info, pers);
  244. conn->host_info = NULL;
  245. }
  246. mysqlnd_set_persistent_string(&conn->authentication_plugin_data, NULL, 0, pers);
  247. mysqlnd_set_string(&conn->last_message, NULL, 0);
  248. conn->charset = NULL;
  249. conn->greet_charset = NULL;
  250. DBG_VOID_RETURN;
  251. }
  252. /* }}} */
  253. /* {{{ mysqlnd_conn_data::dtor */
  254. static void
  255. MYSQLND_METHOD_PRIVATE(mysqlnd_conn_data, dtor)(MYSQLND_CONN_DATA * conn)
  256. {
  257. DBG_ENTER("mysqlnd_conn_data::dtor");
  258. DBG_INF_FMT("conn=%" PRIu64, conn->thread_id);
  259. conn->m->free_contents(conn);
  260. conn->m->free_options(conn);
  261. if (conn->error_info) {
  262. mysqlnd_error_info_free_contents(conn->error_info);
  263. conn->error_info = NULL;
  264. }
  265. if (conn->protocol_frame_codec) {
  266. mysqlnd_pfc_free(conn->protocol_frame_codec, conn->stats, conn->error_info);
  267. conn->protocol_frame_codec = NULL;
  268. }
  269. if (conn->vio) {
  270. mysqlnd_vio_free(conn->vio, conn->stats, conn->error_info);
  271. conn->vio = NULL;
  272. }
  273. if (conn->payload_decoder_factory) {
  274. mysqlnd_protocol_payload_decoder_factory_free(conn->payload_decoder_factory);
  275. conn->payload_decoder_factory = NULL;
  276. }
  277. if (conn->stats) {
  278. mysqlnd_stats_end(conn->stats, conn->persistent);
  279. }
  280. mnd_pefree(conn, conn->persistent);
  281. DBG_VOID_RETURN;
  282. }
  283. /* }}} */
  284. /* {{{ mysqlnd_conn_data::set_server_option */
  285. static enum_func_status
  286. MYSQLND_METHOD(mysqlnd_conn_data, set_server_option)(MYSQLND_CONN_DATA * const conn, enum_mysqlnd_server_option option)
  287. {
  288. DBG_ENTER("mysqlnd_conn_data::set_server_option");
  289. DBG_RETURN(conn->command->set_option(conn, option));
  290. }
  291. /* }}} */
  292. /* {{{ mysqlnd_conn_data::restart_psession */
  293. static enum_func_status
  294. MYSQLND_METHOD(mysqlnd_conn_data, restart_psession)(MYSQLND_CONN_DATA * conn)
  295. {
  296. DBG_ENTER("mysqlnd_conn_data::restart_psession");
  297. MYSQLND_INC_CONN_STATISTIC(conn->stats, STAT_CONNECT_REUSED);
  298. conn->current_result = NULL;
  299. conn->last_message.s = NULL;
  300. DBG_RETURN(PASS);
  301. }
  302. /* }}} */
  303. /* {{{ mysqlnd_conn_data::end_psession */
  304. static enum_func_status
  305. MYSQLND_METHOD(mysqlnd_conn_data, end_psession)(MYSQLND_CONN_DATA * conn)
  306. {
  307. DBG_ENTER("mysqlnd_conn_data::end_psession");
  308. /* Free here what should not be seen by the next script */
  309. if (conn->current_result) {
  310. conn->current_result->m.free_result(conn->current_result, TRUE);
  311. conn->current_result = NULL;
  312. }
  313. mysqlnd_set_string(&conn->last_message, NULL, 0);
  314. conn->error_info = &conn->error_info_impl;
  315. DBG_RETURN(PASS);
  316. }
  317. /* }}} */
  318. /* {{{ mysqlnd_conn_data::fetch_auth_plugin_by_name */
  319. static struct st_mysqlnd_authentication_plugin *
  320. MYSQLND_METHOD(mysqlnd_conn_data, fetch_auth_plugin_by_name)(const char * const requested_protocol)
  321. {
  322. struct st_mysqlnd_authentication_plugin * auth_plugin;
  323. char * plugin_name = NULL;
  324. DBG_ENTER("mysqlnd_conn_data::fetch_auth_plugin_by_name");
  325. mnd_sprintf(&plugin_name, 0, "auth_plugin_%s", requested_protocol);
  326. DBG_INF_FMT("looking for %s auth plugin", plugin_name);
  327. auth_plugin = mysqlnd_plugin_find(plugin_name);
  328. mnd_sprintf_free(plugin_name);
  329. DBG_RETURN(auth_plugin);
  330. }
  331. /* }}} */
  332. /* {{{ mysqlnd_conn_data::execute_init_commands */
  333. static enum_func_status
  334. MYSQLND_METHOD(mysqlnd_conn_data, execute_init_commands)(MYSQLND_CONN_DATA * conn)
  335. {
  336. enum_func_status ret = PASS;
  337. DBG_ENTER("mysqlnd_conn_data::execute_init_commands");
  338. if (conn->options->init_commands) {
  339. unsigned int current_command = 0;
  340. for (; current_command < conn->options->num_commands; ++current_command) {
  341. const char * const command = conn->options->init_commands[current_command];
  342. if (command) {
  343. MYSQLND_INC_CONN_STATISTIC(conn->stats, STAT_INIT_COMMAND_EXECUTED_COUNT);
  344. if (PASS != conn->m->query(conn, command, strlen(command))) {
  345. MYSQLND_INC_CONN_STATISTIC(conn->stats, STAT_INIT_COMMAND_FAILED_COUNT);
  346. ret = FAIL;
  347. break;
  348. }
  349. do {
  350. if (conn->last_query_type == QUERY_SELECT) {
  351. MYSQLND_RES * result = conn->m->use_result(conn);
  352. if (result) {
  353. result->m.free_result(result, TRUE);
  354. }
  355. }
  356. } while (conn->m->next_result(conn) != FAIL);
  357. }
  358. }
  359. }
  360. DBG_RETURN(ret);
  361. }
  362. /* }}} */
  363. /* {{{ mysqlnd_conn_data::get_updated_connect_flags */
  364. static unsigned int
  365. MYSQLND_METHOD(mysqlnd_conn_data, get_updated_connect_flags)(MYSQLND_CONN_DATA * conn, unsigned int mysql_flags)
  366. {
  367. #ifdef MYSQLND_COMPRESSION_ENABLED
  368. MYSQLND_PFC * pfc = conn->protocol_frame_codec;
  369. #endif
  370. MYSQLND_VIO * vio = conn->vio;
  371. DBG_ENTER("mysqlnd_conn_data::get_updated_connect_flags");
  372. /* allow CLIENT_LOCAL_FILES capability, although extensions basing on mysqlnd
  373. shouldn't allow 'load data local infile' by default due to security issues */
  374. mysql_flags |= MYSQLND_CAPABILITIES;
  375. mysql_flags |= conn->options->flags; /* use the flags from set_client_option() */
  376. #ifndef MYSQLND_COMPRESSION_ENABLED
  377. if (mysql_flags & CLIENT_COMPRESS) {
  378. mysql_flags &= ~CLIENT_COMPRESS;
  379. }
  380. #else
  381. if (pfc && pfc->data->flags & MYSQLND_PROTOCOL_FLAG_USE_COMPRESSION) {
  382. mysql_flags |= CLIENT_COMPRESS;
  383. }
  384. #endif
  385. #ifndef MYSQLND_SSL_SUPPORTED
  386. if (mysql_flags & CLIENT_SSL) {
  387. mysql_flags &= ~CLIENT_SSL;
  388. }
  389. #else
  390. if (vio && (vio->data->options.ssl_key ||
  391. vio->data->options.ssl_cert ||
  392. vio->data->options.ssl_ca ||
  393. vio->data->options.ssl_capath ||
  394. vio->data->options.ssl_cipher))
  395. {
  396. mysql_flags |= CLIENT_SSL;
  397. }
  398. #endif
  399. if (conn->options->connect_attr && zend_hash_num_elements(conn->options->connect_attr)) {
  400. mysql_flags |= CLIENT_CONNECT_ATTRS;
  401. }
  402. DBG_RETURN(mysql_flags);
  403. }
  404. /* }}} */
  405. /* {{{ mysqlnd_conn_data::connect_handshake */
  406. static enum_func_status
  407. MYSQLND_METHOD(mysqlnd_conn_data, connect_handshake)(MYSQLND_CONN_DATA * conn,
  408. const MYSQLND_CSTRING * const scheme,
  409. const MYSQLND_CSTRING * const username,
  410. const MYSQLND_CSTRING * const password,
  411. const MYSQLND_CSTRING * const database,
  412. const unsigned int mysql_flags)
  413. {
  414. enum_func_status ret = FAIL;
  415. DBG_ENTER("mysqlnd_conn_data::connect_handshake");
  416. if (PASS == conn->vio->data->m.connect(conn->vio, *scheme, conn->persistent, conn->stats, conn->error_info) &&
  417. PASS == conn->protocol_frame_codec->data->m.reset(conn->protocol_frame_codec, conn->stats, conn->error_info))
  418. {
  419. size_t client_flags = mysql_flags;
  420. ret = conn->command->handshake(conn, *username, *password, *database, client_flags);
  421. }
  422. DBG_RETURN(ret);
  423. }
  424. /* }}} */
  425. /* {{{ mysqlnd_conn_data::get_scheme */
  426. static MYSQLND_STRING
  427. MYSQLND_METHOD(mysqlnd_conn_data, get_scheme)(MYSQLND_CONN_DATA * conn, MYSQLND_CSTRING hostname, MYSQLND_CSTRING *socket_or_pipe, unsigned int port, bool * unix_socket, bool * named_pipe)
  428. {
  429. MYSQLND_STRING transport;
  430. DBG_ENTER("mysqlnd_conn_data::get_scheme");
  431. #ifndef PHP_WIN32
  432. if (hostname.l == sizeof("localhost") - 1 && !strncasecmp(hostname.s, "localhost", hostname.l)) {
  433. DBG_INF_FMT("socket=%s", socket_or_pipe->s? socket_or_pipe->s:"n/a");
  434. if (!socket_or_pipe->s) {
  435. socket_or_pipe->s = "/tmp/mysql.sock";
  436. socket_or_pipe->l = strlen(socket_or_pipe->s);
  437. }
  438. transport.l = mnd_sprintf(&transport.s, 0, "unix://%s", socket_or_pipe->s);
  439. *unix_socket = TRUE;
  440. #else
  441. if (hostname.l == sizeof(".") - 1 && hostname.s[0] == '.') {
  442. /* named pipe in socket */
  443. if (!socket_or_pipe->s) {
  444. socket_or_pipe->s = "\\\\.\\pipe\\MySQL";
  445. socket_or_pipe->l = strlen(socket_or_pipe->s);
  446. }
  447. transport.l = mnd_sprintf(&transport.s, 0, "pipe://%s", socket_or_pipe->s);
  448. *named_pipe = TRUE;
  449. #endif
  450. } else {
  451. if (!port) {
  452. port = 3306;
  453. }
  454. transport.l = mnd_sprintf(&transport.s, 0, "tcp://%s:%u", hostname.s, port);
  455. }
  456. DBG_INF_FMT("transport=%s", transport.s? transport.s:"OOM");
  457. DBG_RETURN(transport);
  458. }
  459. /* }}} */
  460. /* {{{ mysqlnd_conn_data::connect */
  461. static enum_func_status
  462. MYSQLND_METHOD(mysqlnd_conn_data, connect)(MYSQLND_CONN_DATA * conn,
  463. MYSQLND_CSTRING hostname,
  464. MYSQLND_CSTRING username,
  465. MYSQLND_CSTRING password,
  466. MYSQLND_CSTRING database,
  467. unsigned int port,
  468. MYSQLND_CSTRING socket_or_pipe,
  469. unsigned int mysql_flags
  470. )
  471. {
  472. bool unix_socket = FALSE;
  473. bool named_pipe = FALSE;
  474. bool reconnect = FALSE;
  475. bool saved_compression = FALSE;
  476. MYSQLND_PFC * pfc = conn->protocol_frame_codec;
  477. MYSQLND_STRING transport = { NULL, 0 };
  478. DBG_ENTER("mysqlnd_conn_data::connect");
  479. DBG_INF_FMT("conn=%p", conn);
  480. SET_EMPTY_ERROR(conn->error_info);
  481. UPSERT_STATUS_SET_AFFECTED_ROWS_TO_ERROR(conn->upsert_status);
  482. DBG_INF_FMT("host=%s user=%s db=%s port=%u flags=%u persistent=%u state=%u",
  483. hostname.s?hostname.s:"", username.s?username.s:"", database.s?database.s:"", port, mysql_flags,
  484. conn? conn->persistent:0, conn? (int)GET_CONNECTION_STATE(&conn->state):-1);
  485. if (GET_CONNECTION_STATE(&conn->state) > CONN_ALLOCED) {
  486. DBG_INF("Connecting on a connected handle.");
  487. if (GET_CONNECTION_STATE(&conn->state) < CONN_QUIT_SENT) {
  488. MYSQLND_INC_CONN_STATISTIC(conn->stats, STAT_CLOSE_IMPLICIT);
  489. reconnect = TRUE;
  490. conn->m->send_close(conn);
  491. }
  492. conn->m->free_contents(conn);
  493. /* Now reconnect using the same handle */
  494. if (pfc->data->compressed) {
  495. /*
  496. we need to save the state. As we will re-connect, pfc->compressed should be off, or
  497. we will look for a compression header as part of the greet message, but there will
  498. be none.
  499. */
  500. saved_compression = TRUE;
  501. pfc->data->compressed = FALSE;
  502. }
  503. if (pfc->data->ssl) {
  504. pfc->data->ssl = FALSE;
  505. }
  506. } else {
  507. unsigned int max_allowed_size = MYSQLND_ASSEMBLED_PACKET_MAX_SIZE;
  508. conn->m->set_client_option(conn, MYSQLND_OPT_MAX_ALLOWED_PACKET, (char *)&max_allowed_size);
  509. }
  510. if (!hostname.s || !hostname.s[0]) {
  511. hostname.s = "localhost";
  512. hostname.l = strlen(hostname.s);
  513. }
  514. if (!username.s) {
  515. DBG_INF_FMT("no user given, using empty string");
  516. username.s = "";
  517. username.l = 0;
  518. }
  519. if (!password.s) {
  520. DBG_INF_FMT("no password given, using empty string");
  521. password.s = "";
  522. password.l = 0;
  523. }
  524. if (!database.s || !database.s[0]) {
  525. DBG_INF_FMT("no db given, using empty string");
  526. database.s = "";
  527. database.l = 0;
  528. } else {
  529. mysql_flags |= CLIENT_CONNECT_WITH_DB;
  530. }
  531. transport = conn->m->get_scheme(conn, hostname, &socket_or_pipe, port, &unix_socket, &named_pipe);
  532. mysql_flags = conn->m->get_updated_connect_flags(conn, mysql_flags);
  533. {
  534. const MYSQLND_CSTRING scheme = { transport.s, transport.l };
  535. if (FAIL == conn->m->connect_handshake(conn, &scheme, &username, &password, &database, mysql_flags)) {
  536. goto err;
  537. }
  538. }
  539. {
  540. SET_CONNECTION_STATE(&conn->state, CONN_READY);
  541. if (saved_compression) {
  542. pfc->data->compressed = TRUE;
  543. }
  544. /*
  545. If a connect on a existing handle is performed and mysql_flags is
  546. passed which doesn't CLIENT_COMPRESS, then we need to overwrite the value
  547. which we set based on saved_compression.
  548. */
  549. pfc->data->compressed = mysql_flags & CLIENT_COMPRESS? TRUE:FALSE;
  550. mysqlnd_set_persistent_string(&conn->scheme, transport.s, transport.l, conn->persistent);
  551. if (transport.s) {
  552. mnd_sprintf_free(transport.s);
  553. transport.s = NULL;
  554. }
  555. if (!conn->scheme.s) {
  556. goto err; /* OOM */
  557. }
  558. mysqlnd_set_persistent_string(&conn->username, username.s, username.l, conn->persistent);
  559. mysqlnd_set_persistent_string(&conn->password, username.s, password.l, conn->persistent);
  560. conn->port = port;
  561. mysqlnd_set_persistent_string(&conn->connect_or_select_db, database.s, database.l, conn->persistent);
  562. if (!unix_socket && !named_pipe) {
  563. mysqlnd_set_persistent_string(&conn->hostname, hostname.s, hostname.l, conn->persistent);
  564. {
  565. char *p;
  566. mnd_sprintf(&p, 0, "%s via TCP/IP", conn->hostname.s);
  567. if (!p) {
  568. SET_OOM_ERROR(conn->error_info);
  569. goto err; /* OOM */
  570. }
  571. conn->host_info = mnd_pestrdup(p, conn->persistent);
  572. mnd_sprintf_free(p);
  573. }
  574. } else {
  575. conn->unix_socket.s = mnd_pestrdup(socket_or_pipe.s, conn->persistent);
  576. if (unix_socket) {
  577. conn->host_info = mnd_pestrdup("Localhost via UNIX socket", conn->persistent);
  578. } else if (named_pipe) {
  579. char *p;
  580. mnd_sprintf(&p, 0, "%s via named pipe", conn->unix_socket.s);
  581. if (!p) {
  582. SET_OOM_ERROR(conn->error_info);
  583. goto err; /* OOM */
  584. }
  585. conn->host_info = mnd_pestrdup(p, conn->persistent);
  586. mnd_sprintf_free(p);
  587. } else {
  588. php_error_docref(NULL, E_WARNING, "Impossible. Should be either socket or a pipe. Report a bug!");
  589. }
  590. if (!conn->unix_socket.s || !conn->host_info) {
  591. SET_OOM_ERROR(conn->error_info);
  592. goto err; /* OOM */
  593. }
  594. conn->unix_socket.l = strlen(conn->unix_socket.s);
  595. }
  596. SET_EMPTY_ERROR(conn->error_info);
  597. mysqlnd_local_infile_default(conn);
  598. if (FAIL == conn->m->execute_init_commands(conn)) {
  599. goto err;
  600. }
  601. MYSQLND_INC_CONN_STATISTIC_W_VALUE2(conn->stats, STAT_CONNECT_SUCCESS, 1, STAT_OPENED_CONNECTIONS, 1);
  602. if (reconnect) {
  603. MYSQLND_INC_GLOBAL_STATISTIC(STAT_RECONNECT);
  604. }
  605. if (conn->persistent) {
  606. MYSQLND_INC_CONN_STATISTIC_W_VALUE2(conn->stats, STAT_PCONNECT_SUCCESS, 1, STAT_OPENED_PERSISTENT_CONNECTIONS, 1);
  607. }
  608. DBG_INF_FMT("connection_id=%" PRIu64, conn->thread_id);
  609. DBG_RETURN(PASS);
  610. }
  611. err:
  612. if (transport.s) {
  613. mnd_sprintf_free(transport.s);
  614. transport.s = NULL;
  615. }
  616. DBG_ERR_FMT("[%u] %.128s (trying to connect via %s)", conn->error_info->error_no, conn->error_info->error, conn->scheme.s);
  617. if (!conn->error_info->error_no) {
  618. char * msg;
  619. mnd_sprintf(&msg, 0, "%s (trying to connect via %s)",conn->error_info->error, conn->scheme.s);
  620. SET_CLIENT_ERROR(conn->error_info, CR_CONNECTION_ERROR, UNKNOWN_SQLSTATE, msg);
  621. mnd_sprintf_free(msg);
  622. }
  623. conn->m->free_contents(conn);
  624. MYSQLND_INC_CONN_STATISTIC(conn->stats, STAT_CONNECT_FAILURE);
  625. DBG_RETURN(FAIL);
  626. }
  627. /* }}} */
  628. /* {{{ mysqlnd_conn::connect */
  629. static enum_func_status
  630. MYSQLND_METHOD(mysqlnd_conn, connect)(MYSQLND * conn_handle,
  631. const MYSQLND_CSTRING hostname,
  632. const MYSQLND_CSTRING username,
  633. const MYSQLND_CSTRING password,
  634. const MYSQLND_CSTRING database,
  635. unsigned int port,
  636. const MYSQLND_CSTRING socket_or_pipe,
  637. unsigned int mysql_flags)
  638. {
  639. MYSQLND_CONN_DATA * conn = conn_handle->data;
  640. DBG_ENTER("mysqlnd_conn::connect");
  641. mysqlnd_options4(conn_handle, MYSQL_OPT_CONNECT_ATTR_ADD, "_client_name", "mysqlnd");
  642. if (hostname.l > 0) {
  643. mysqlnd_options4(conn_handle, MYSQL_OPT_CONNECT_ATTR_ADD, "_server_host", hostname.s);
  644. }
  645. DBG_RETURN(conn->m->connect(conn, hostname, username, password, database, port, socket_or_pipe, mysql_flags));
  646. }
  647. /* }}} */
  648. /* {{{ mysqlnd_conn_data::query */
  649. /*
  650. If conn->error_info->error_no is not zero, then we had an error.
  651. Still the result from the query is PASS
  652. */
  653. static enum_func_status
  654. MYSQLND_METHOD(mysqlnd_conn_data, query)(MYSQLND_CONN_DATA * conn, const char * const query, const size_t query_len)
  655. {
  656. enum_func_status ret = FAIL;
  657. DBG_ENTER("mysqlnd_conn_data::query");
  658. DBG_INF_FMT("conn=%p conn=%" PRIu64 " query=%s", conn, conn->thread_id, query);
  659. if (PASS == conn->m->send_query(conn, query, query_len, NULL, NULL) &&
  660. PASS == conn->m->reap_query(conn))
  661. {
  662. ret = PASS;
  663. if (conn->last_query_type == QUERY_UPSERT && UPSERT_STATUS_GET_AFFECTED_ROWS(conn->upsert_status)) {
  664. MYSQLND_INC_CONN_STATISTIC_W_VALUE(conn->stats, STAT_ROWS_AFFECTED_NORMAL, UPSERT_STATUS_GET_AFFECTED_ROWS(conn->upsert_status));
  665. }
  666. }
  667. DBG_RETURN(ret);
  668. }
  669. /* }}} */
  670. /* {{{ mysqlnd_conn_data::send_query */
  671. static enum_func_status
  672. MYSQLND_METHOD(mysqlnd_conn_data, send_query)(
  673. MYSQLND_CONN_DATA * conn, const char * const query, const size_t query_len,
  674. zval *read_cb, zval *err_cb)
  675. {
  676. DBG_ENTER("mysqlnd_conn_data::send_query");
  677. DBG_INF_FMT("conn=%" PRIu64 " query=%s", conn->thread_id, query);
  678. DBG_INF_FMT("conn->server_status=%u", UPSERT_STATUS_GET_SERVER_STATUS(conn->upsert_status));
  679. const MYSQLND_CSTRING query_string = {query, query_len};
  680. enum_func_status ret = conn->command->query(conn, query_string);
  681. DBG_INF_FMT("conn->server_status=%u", UPSERT_STATUS_GET_SERVER_STATUS(conn->upsert_status));
  682. DBG_RETURN(ret);
  683. }
  684. /* }}} */
  685. /* {{{ mysqlnd_conn_data::reap_query */
  686. static enum_func_status
  687. MYSQLND_METHOD(mysqlnd_conn_data, reap_query)(MYSQLND_CONN_DATA * conn)
  688. {
  689. DBG_ENTER("mysqlnd_conn_data::reap_query");
  690. DBG_INF_FMT("conn=%" PRIu64, conn->thread_id);
  691. DBG_INF_FMT("conn->server_status=%u", UPSERT_STATUS_GET_SERVER_STATUS(conn->upsert_status));
  692. enum_func_status ret = conn->command->reap_result(conn);
  693. DBG_INF_FMT("conn->server_status=%u", UPSERT_STATUS_GET_SERVER_STATUS(conn->upsert_status));
  694. DBG_RETURN(ret);
  695. }
  696. /* }}} */
  697. /* {{{ mysqlnd_conn_data::list_method */
  698. MYSQLND_RES *
  699. MYSQLND_METHOD(mysqlnd_conn_data, list_method)(MYSQLND_CONN_DATA * conn, const char * const query, const char * const achtung_wild, const char * const par1)
  700. {
  701. char * show_query = NULL;
  702. size_t show_query_len;
  703. MYSQLND_RES * result = NULL;
  704. DBG_ENTER("mysqlnd_conn_data::list_method");
  705. DBG_INF_FMT("conn=%" PRIu64 " query=%s wild=%p", conn->thread_id, query, achtung_wild);
  706. if (par1) {
  707. if (achtung_wild) {
  708. show_query_len = mnd_sprintf(&show_query, 0, query, par1, achtung_wild);
  709. } else {
  710. show_query_len = mnd_sprintf(&show_query, 0, query, par1);
  711. }
  712. } else {
  713. if (achtung_wild) {
  714. show_query_len = mnd_sprintf(&show_query, 0, query, achtung_wild);
  715. } else {
  716. show_query_len = strlen(show_query = (char *)query);
  717. }
  718. }
  719. if (PASS == conn->m->query(conn, show_query, show_query_len)) {
  720. result = conn->m->store_result(conn);
  721. }
  722. if (show_query != query) {
  723. mnd_sprintf_free(show_query);
  724. }
  725. DBG_RETURN(result);
  726. }
  727. /* }}} */
  728. /* {{{ mysqlnd_conn_data::err_no */
  729. static unsigned int
  730. MYSQLND_METHOD(mysqlnd_conn_data, err_no)(const MYSQLND_CONN_DATA * const conn)
  731. {
  732. return conn->error_info->error_no;
  733. }
  734. /* }}} */
  735. /* {{{ mysqlnd_conn_data::error */
  736. static const char *
  737. MYSQLND_METHOD(mysqlnd_conn_data, error)(const MYSQLND_CONN_DATA * const conn)
  738. {
  739. return conn->error_info->error;
  740. }
  741. /* }}} */
  742. /* {{{ mysqlnd_conn_data::sqlstate */
  743. static const char *
  744. MYSQLND_METHOD(mysqlnd_conn_data, sqlstate)(const MYSQLND_CONN_DATA * const conn)
  745. {
  746. return conn->error_info->sqlstate[0] ? conn->error_info->sqlstate:MYSQLND_SQLSTATE_NULL;
  747. }
  748. /* }}} */
  749. /* {{{ mysqlnd_old_escape_string */
  750. PHPAPI zend_ulong
  751. mysqlnd_old_escape_string(char * newstr, const char * escapestr, size_t escapestr_len)
  752. {
  753. DBG_ENTER("mysqlnd_old_escape_string");
  754. DBG_RETURN(mysqlnd_cset_escape_slashes(mysqlnd_find_charset_name("latin1"), newstr, escapestr, escapestr_len));
  755. }
  756. /* }}} */
  757. /* {{{ mysqlnd_conn_data::ssl_set */
  758. static enum_func_status
  759. MYSQLND_METHOD(mysqlnd_conn_data, ssl_set)(MYSQLND_CONN_DATA * const conn, const char * key, const char * const cert,
  760. const char * const ca, const char * const capath, const char * const cipher)
  761. {
  762. MYSQLND_VIO * vio = conn->vio;
  763. DBG_ENTER("mysqlnd_conn_data::ssl_set");
  764. enum_func_status ret = (
  765. PASS == vio->data->m.set_client_option(vio, MYSQLND_OPT_SSL_KEY, key) &&
  766. PASS == vio->data->m.set_client_option(vio, MYSQLND_OPT_SSL_CERT, cert) &&
  767. PASS == vio->data->m.set_client_option(vio, MYSQLND_OPT_SSL_CA, ca) &&
  768. PASS == vio->data->m.set_client_option(vio, MYSQLND_OPT_SSL_CAPATH, capath) &&
  769. PASS == vio->data->m.set_client_option(vio, MYSQLND_OPT_SSL_CIPHER, cipher)) ? PASS : FAIL;
  770. DBG_RETURN(ret);
  771. }
  772. /* }}} */
  773. /* {{{ mysqlnd_conn_data::escape_string */
  774. static zend_ulong
  775. MYSQLND_METHOD(mysqlnd_conn_data, escape_string)(MYSQLND_CONN_DATA * const conn, char * newstr, const char * escapestr, size_t escapestr_len)
  776. {
  777. zend_ulong ret = FAIL;
  778. DBG_ENTER("mysqlnd_conn_data::escape_string");
  779. DBG_INF_FMT("conn=%" PRIu64, conn->thread_id);
  780. DBG_INF_FMT("server_status=%u", UPSERT_STATUS_GET_SERVER_STATUS(conn->upsert_status));
  781. if (UPSERT_STATUS_GET_SERVER_STATUS(conn->upsert_status) & SERVER_STATUS_NO_BACKSLASH_ESCAPES) {
  782. ret = mysqlnd_cset_escape_quotes(conn->charset, newstr, escapestr, escapestr_len);
  783. } else {
  784. ret = mysqlnd_cset_escape_slashes(conn->charset, newstr, escapestr, escapestr_len);
  785. }
  786. DBG_RETURN(ret);
  787. }
  788. /* }}} */
  789. /* {{{ mysqlnd_conn_data::dump_debug_info */
  790. static enum_func_status
  791. MYSQLND_METHOD(mysqlnd_conn_data, dump_debug_info)(MYSQLND_CONN_DATA * const conn)
  792. {
  793. DBG_ENTER("mysqlnd_conn_data::dump_debug_info");
  794. DBG_INF_FMT("conn=%" PRIu64, conn->thread_id);
  795. DBG_RETURN(conn->command->debug(conn));
  796. }
  797. /* }}} */
  798. /* {{{ mysqlnd_conn_data::select_db */
  799. static enum_func_status
  800. MYSQLND_METHOD(mysqlnd_conn_data, select_db)(MYSQLND_CONN_DATA * const conn, const char * const db, const size_t db_len)
  801. {
  802. DBG_ENTER("mysqlnd_conn_data::select_db");
  803. DBG_INF_FMT("conn=%" PRIu64 " db=%s", conn->thread_id, db);
  804. const MYSQLND_CSTRING database = {db, db_len};
  805. DBG_RETURN(conn->command->init_db(conn, database));
  806. }
  807. /* }}} */
  808. /* {{{ mysqlnd_conn_data::ping */
  809. static enum_func_status
  810. MYSQLND_METHOD(mysqlnd_conn_data, ping)(MYSQLND_CONN_DATA * const conn)
  811. {
  812. DBG_ENTER("mysqlnd_conn_data::ping");
  813. DBG_INF_FMT("conn=%" PRIu64, conn->thread_id);
  814. enum_func_status ret = conn->command->ping(conn);
  815. DBG_INF_FMT("ret=%u", ret);
  816. DBG_RETURN(ret);
  817. }
  818. /* }}} */
  819. /* {{{ mysqlnd_conn_data::statistic */
  820. static enum_func_status
  821. MYSQLND_METHOD(mysqlnd_conn_data, statistic)(MYSQLND_CONN_DATA * conn, zend_string **message)
  822. {
  823. DBG_ENTER("mysqlnd_conn_data::statistic");
  824. DBG_INF_FMT("conn=%" PRIu64, conn->thread_id);
  825. DBG_RETURN(conn->command->statistics(conn, message));
  826. }
  827. /* }}} */
  828. /* {{{ mysqlnd_conn_data::kill */
  829. static enum_func_status
  830. MYSQLND_METHOD(mysqlnd_conn_data, kill)(MYSQLND_CONN_DATA * conn, unsigned int pid)
  831. {
  832. DBG_ENTER("mysqlnd_conn_data::kill");
  833. DBG_INF_FMT("conn=%" PRIu64 " pid=%u", conn->thread_id, pid);
  834. const unsigned int process_id = pid;
  835. /* 'unsigned char' is promoted to 'int' when passed through '...' */
  836. const unsigned int read_response = (pid != conn->thread_id);
  837. DBG_RETURN(conn->command->process_kill(conn, process_id, read_response));
  838. }
  839. /* }}} */
  840. /* {{{ mysqlnd_conn_data::set_charset */
  841. static enum_func_status
  842. MYSQLND_METHOD(mysqlnd_conn_data, set_charset)(MYSQLND_CONN_DATA * const conn, const char * const csname)
  843. {
  844. enum_func_status ret = FAIL;
  845. const MYSQLND_CHARSET * const charset = mysqlnd_find_charset_name(csname);
  846. DBG_ENTER("mysqlnd_conn_data::set_charset");
  847. DBG_INF_FMT("conn=%" PRIu64 " cs=%s", conn->thread_id, csname);
  848. if (!charset) {
  849. SET_CLIENT_ERROR(conn->error_info, CR_CANT_FIND_CHARSET, UNKNOWN_SQLSTATE, "Invalid character set was provided");
  850. DBG_RETURN(ret);
  851. }
  852. char * query;
  853. size_t query_len = mnd_sprintf(&query, 0, "SET NAMES %s", csname);
  854. if (FAIL == (ret = conn->m->query(conn, query, query_len)) || conn->error_info->error_no) {
  855. ret = FAIL;
  856. } else {
  857. conn->charset = charset;
  858. }
  859. mnd_sprintf_free(query);
  860. DBG_INF(ret == PASS? "PASS":"FAIL");
  861. DBG_RETURN(ret);
  862. }
  863. /* }}} */
  864. /* {{{ mysqlnd_conn_data::refresh */
  865. static enum_func_status
  866. MYSQLND_METHOD(mysqlnd_conn_data, refresh)(MYSQLND_CONN_DATA * const conn, uint8_t options)
  867. {
  868. DBG_ENTER("mysqlnd_conn_data::refresh");
  869. DBG_INF_FMT("conn=%" PRIu64 " options=%u", conn->thread_id, options);
  870. DBG_RETURN(conn->command->refresh(conn, options));
  871. }
  872. /* }}} */
  873. /* {{{ mysqlnd_conn_data::shutdown */
  874. static enum_func_status
  875. MYSQLND_METHOD(mysqlnd_conn_data, shutdown)(MYSQLND_CONN_DATA * const conn, uint8_t level)
  876. {
  877. DBG_ENTER("mysqlnd_conn_data::shutdown");
  878. DBG_INF_FMT("conn=%" PRIu64 " level=%u", conn->thread_id, level);
  879. DBG_RETURN(conn->command->shutdown(conn, level));
  880. }
  881. /* }}} */
  882. /* {{{ mysqlnd_send_close */
  883. static enum_func_status
  884. MYSQLND_METHOD(mysqlnd_conn_data, send_close)(MYSQLND_CONN_DATA * const conn)
  885. {
  886. enum_func_status ret = PASS;
  887. MYSQLND_VIO * vio = conn->vio;
  888. php_stream * net_stream = vio->data->m.get_stream(vio);
  889. enum mysqlnd_connection_state state = GET_CONNECTION_STATE(&conn->state);
  890. DBG_ENTER("mysqlnd_send_close");
  891. DBG_INF_FMT("conn=%" PRIu64 " vio->data->stream->abstract=%p", conn->thread_id, net_stream? net_stream->abstract:NULL);
  892. DBG_INF_FMT("state=%u", state);
  893. if (state >= CONN_READY) {
  894. MYSQLND_DEC_GLOBAL_STATISTIC(STAT_OPENED_CONNECTIONS);
  895. if (conn->persistent) {
  896. MYSQLND_DEC_GLOBAL_STATISTIC(STAT_OPENED_PERSISTENT_CONNECTIONS);
  897. }
  898. }
  899. switch (state) {
  900. case CONN_READY:
  901. DBG_INF("Connection clean, sending COM_QUIT");
  902. if (net_stream) {
  903. ret = conn->command->quit(conn);
  904. vio->data->m.close_stream(vio, conn->stats, conn->error_info);
  905. }
  906. SET_CONNECTION_STATE(&conn->state, CONN_QUIT_SENT);
  907. break;
  908. case CONN_SENDING_LOAD_DATA:
  909. /*
  910. Don't send COM_QUIT if we are in a middle of a LOAD DATA or we
  911. will crash (assert) a debug server.
  912. */
  913. case CONN_NEXT_RESULT_PENDING:
  914. case CONN_QUERY_SENT:
  915. case CONN_FETCHING_DATA:
  916. MYSQLND_INC_GLOBAL_STATISTIC(STAT_CLOSE_IN_MIDDLE);
  917. DBG_ERR_FMT("Brutally closing connection [%p][%s]", conn, conn->scheme.s);
  918. /*
  919. Do nothing, the connection will be brutally closed
  920. and the server will catch it and free close from its side.
  921. */
  922. ZEND_FALLTHROUGH;
  923. case CONN_ALLOCED:
  924. /*
  925. Allocated but not connected or there was failure when trying
  926. to connect with pre-allocated connect.
  927. Fall-through
  928. */
  929. SET_CONNECTION_STATE(&conn->state, CONN_QUIT_SENT);
  930. ZEND_FALLTHROUGH;
  931. case CONN_QUIT_SENT:
  932. /* The user has killed its own connection */
  933. vio->data->m.close_stream(vio, conn->stats, conn->error_info);
  934. break;
  935. }
  936. DBG_RETURN(ret);
  937. }
  938. /* }}} */
  939. /* {{{ mysqlnd_conn_data::get_reference */
  940. static MYSQLND_CONN_DATA *
  941. MYSQLND_METHOD_PRIVATE(mysqlnd_conn_data, get_reference)(MYSQLND_CONN_DATA * const conn)
  942. {
  943. DBG_ENTER("mysqlnd_conn_data::get_reference");
  944. ++conn->refcount;
  945. DBG_INF_FMT("conn=%" PRIu64 " new_refcount=%u", conn->thread_id, conn->refcount);
  946. DBG_RETURN(conn);
  947. }
  948. /* }}} */
  949. /* {{{ mysqlnd_conn_data::free_reference */
  950. static enum_func_status
  951. MYSQLND_METHOD_PRIVATE(mysqlnd_conn_data, free_reference)(MYSQLND_CONN_DATA * const conn)
  952. {
  953. enum_func_status ret = PASS;
  954. DBG_ENTER("mysqlnd_conn_data::free_reference");
  955. DBG_INF_FMT("conn=%" PRIu64 " old_refcount=%u", conn->thread_id, conn->refcount);
  956. if (!(--conn->refcount)) {
  957. /*
  958. No multithreading issues as we don't share the connection :)
  959. This will free the object too, of course because references has
  960. reached zero.
  961. */
  962. ret = conn->m->send_close(conn);
  963. conn->m->dtor(conn);
  964. }
  965. DBG_RETURN(ret);
  966. }
  967. /* }}} */
  968. /* {{{ mysqlnd_conn_data::field_count */
  969. static unsigned int
  970. MYSQLND_METHOD(mysqlnd_conn_data, field_count)(const MYSQLND_CONN_DATA * const conn)
  971. {
  972. return conn->field_count;
  973. }
  974. /* }}} */
  975. /* {{{ mysqlnd_conn_data::server_status */
  976. static unsigned int
  977. MYSQLND_METHOD(mysqlnd_conn_data, server_status)(const MYSQLND_CONN_DATA * const conn)
  978. {
  979. return UPSERT_STATUS_GET_SERVER_STATUS(conn->upsert_status);
  980. }
  981. /* }}} */
  982. /* {{{ mysqlnd_conn_data::insert_id */
  983. static uint64_t
  984. MYSQLND_METHOD(mysqlnd_conn_data, insert_id)(const MYSQLND_CONN_DATA * const conn)
  985. {
  986. return UPSERT_STATUS_GET_LAST_INSERT_ID(conn->upsert_status);
  987. }
  988. /* }}} */
  989. /* {{{ mysqlnd_conn_data::affected_rows */
  990. static uint64_t
  991. MYSQLND_METHOD(mysqlnd_conn_data, affected_rows)(const MYSQLND_CONN_DATA * const conn)
  992. {
  993. return UPSERT_STATUS_GET_AFFECTED_ROWS(conn->upsert_status);
  994. }
  995. /* }}} */
  996. /* {{{ mysqlnd_conn_data::warning_count */
  997. static unsigned int
  998. MYSQLND_METHOD(mysqlnd_conn_data, warning_count)(const MYSQLND_CONN_DATA * const conn)
  999. {
  1000. return UPSERT_STATUS_GET_WARNINGS(conn->upsert_status);
  1001. }
  1002. /* }}} */
  1003. /* {{{ mysqlnd_conn_data::info */
  1004. static const char *
  1005. MYSQLND_METHOD(mysqlnd_conn_data, info)(const MYSQLND_CONN_DATA * const conn)
  1006. {
  1007. return conn->last_message.s;
  1008. }
  1009. /* }}} */
  1010. /* {{{ mysqlnd_get_client_info */
  1011. PHPAPI const char * mysqlnd_get_client_info(void)
  1012. {
  1013. return PHP_MYSQLND_VERSION;
  1014. }
  1015. /* }}} */
  1016. /* {{{ mysqlnd_get_client_version */
  1017. PHPAPI unsigned long mysqlnd_get_client_version(void)
  1018. {
  1019. return MYSQLND_VERSION_ID;
  1020. }
  1021. /* }}} */
  1022. /* {{{ mysqlnd_conn_data::get_server_info */
  1023. static const char *
  1024. MYSQLND_METHOD(mysqlnd_conn_data, get_server_info)(const MYSQLND_CONN_DATA * const conn)
  1025. {
  1026. return conn->server_version;
  1027. }
  1028. /* }}} */
  1029. /* {{{ mysqlnd_conn_data::get_host_info */
  1030. static const char *
  1031. MYSQLND_METHOD(mysqlnd_conn_data, get_host_info)(const MYSQLND_CONN_DATA * const conn)
  1032. {
  1033. return conn->host_info;
  1034. }
  1035. /* }}} */
  1036. /* {{{ mysqlnd_conn_data::get_proto_info */
  1037. static unsigned int
  1038. MYSQLND_METHOD(mysqlnd_conn_data, get_proto_info)(const MYSQLND_CONN_DATA * const conn)
  1039. {
  1040. return conn->protocol_version;
  1041. }
  1042. /* }}} */
  1043. /* {{{ mysqlnd_conn_data::charset_name */
  1044. static const char *
  1045. MYSQLND_METHOD(mysqlnd_conn_data, charset_name)(const MYSQLND_CONN_DATA * const conn)
  1046. {
  1047. return conn->charset->name;
  1048. }
  1049. /* }}} */
  1050. /* {{{ mysqlnd_conn_data::thread_id */
  1051. static uint64_t
  1052. MYSQLND_METHOD(mysqlnd_conn_data, thread_id)(const MYSQLND_CONN_DATA * const conn)
  1053. {
  1054. return conn->thread_id;
  1055. }
  1056. /* }}} */
  1057. /* {{{ mysqlnd_conn_data::get_server_version */
  1058. static zend_ulong
  1059. MYSQLND_METHOD(mysqlnd_conn_data, get_server_version)(const MYSQLND_CONN_DATA * const conn)
  1060. {
  1061. zend_long major, minor, patch;
  1062. char *p;
  1063. if (!(p = conn->server_version)) {
  1064. return 0;
  1065. }
  1066. major = ZEND_STRTOL(p, &p, 10);
  1067. p += 1; /* consume the dot */
  1068. minor = ZEND_STRTOL(p, &p, 10);
  1069. p += 1; /* consume the dot */
  1070. patch = ZEND_STRTOL(p, &p, 10);
  1071. return (zend_ulong)(major * Z_L(10000) + (zend_ulong)(minor * Z_L(100) + patch));
  1072. }
  1073. /* }}} */
  1074. /* {{{ mysqlnd_conn_data::more_results */
  1075. static bool
  1076. MYSQLND_METHOD(mysqlnd_conn_data, more_results)(const MYSQLND_CONN_DATA * const conn)
  1077. {
  1078. DBG_ENTER("mysqlnd_conn_data::more_results");
  1079. /* (conn->state == CONN_NEXT_RESULT_PENDING) too */
  1080. DBG_RETURN(UPSERT_STATUS_GET_SERVER_STATUS(conn->upsert_status) & SERVER_MORE_RESULTS_EXISTS? TRUE:FALSE);
  1081. }
  1082. /* }}} */
  1083. /* {{{ mysqlnd_conn_data::next_result */
  1084. static enum_func_status
  1085. MYSQLND_METHOD(mysqlnd_conn_data, next_result)(MYSQLND_CONN_DATA * const conn)
  1086. {
  1087. DBG_ENTER("mysqlnd_conn_data::next_result");
  1088. DBG_INF_FMT("conn=%" PRIu64 "", conn->thread_id);
  1089. SET_EMPTY_ERROR(conn->error_info);
  1090. if (GET_CONNECTION_STATE(&conn->state) != CONN_NEXT_RESULT_PENDING) {
  1091. DBG_RETURN(FAIL);
  1092. }
  1093. UPSERT_STATUS_SET_AFFECTED_ROWS_TO_ERROR(conn->upsert_status);
  1094. /*
  1095. We are sure that there is a result set, since conn->state is set accordingly
  1096. in mysqlnd_store_result() or mysqlnd_fetch_row_unbuffered()
  1097. */
  1098. enum_func_status ret = conn->m->query_read_result_set_header(conn, NULL);
  1099. if (FAIL == ret) {
  1100. /*
  1101. There can be an error in the middle of a multi-statement, which will cancel the multi-statement.
  1102. So there are no more results and we should just return FALSE, error_no has been set
  1103. */
  1104. if (!conn->error_info->error_no) {
  1105. DBG_ERR_FMT("Serious error. %s::%u", __FILE__, __LINE__);
  1106. php_error_docref(NULL, E_WARNING, "Serious error. PID=%d", getpid());
  1107. SET_CONNECTION_STATE(&conn->state, CONN_QUIT_SENT);
  1108. conn->m->send_close(conn);
  1109. } else {
  1110. DBG_INF_FMT("Error from the server : (%u) %s", conn->error_info->error_no, conn->error_info->error);
  1111. }
  1112. DBG_RETURN(FAIL);
  1113. }
  1114. if (conn->last_query_type == QUERY_UPSERT && UPSERT_STATUS_GET_AFFECTED_ROWS(conn->upsert_status)) {
  1115. MYSQLND_INC_CONN_STATISTIC_W_VALUE(conn->stats, STAT_ROWS_AFFECTED_NORMAL, UPSERT_STATUS_GET_AFFECTED_ROWS(conn->upsert_status));
  1116. }
  1117. DBG_RETURN(PASS);
  1118. }
  1119. /* }}} */
  1120. /* {{{ mysqlnd_conn_data::change_user */
  1121. static enum_func_status
  1122. MYSQLND_METHOD(mysqlnd_conn_data, change_user)(MYSQLND_CONN_DATA * const conn,
  1123. const char * user,
  1124. const char * passwd,
  1125. const char * db,
  1126. bool silent,
  1127. size_t passwd_len
  1128. )
  1129. {
  1130. enum_func_status ret = FAIL;
  1131. DBG_ENTER("mysqlnd_conn_data::change_user");
  1132. DBG_INF_FMT("conn=%" PRIu64 " user=%s passwd=%s db=%s silent=%u",
  1133. conn->thread_id, user?user:"", passwd?"***":"null", db?db:"", silent == TRUE);
  1134. SET_EMPTY_ERROR(conn->error_info);
  1135. UPSERT_STATUS_SET_AFFECTED_ROWS_TO_ERROR(conn->upsert_status);
  1136. if (!user) {
  1137. user = "";
  1138. }
  1139. if (!passwd) {
  1140. passwd = "";
  1141. passwd_len = 0;
  1142. }
  1143. if (!db) {
  1144. db = "";
  1145. }
  1146. /* XXX: passwords that have \0 inside work during auth, but in this case won't work with change user */
  1147. ret = mysqlnd_run_authentication(conn, user, passwd, passwd_len, db, strlen(db),
  1148. conn->authentication_plugin_data, conn->options->auth_protocol,
  1149. 0 /*charset not used*/, conn->options, conn->server_capabilities, silent, TRUE/*is_change*/);
  1150. /*
  1151. Here we should close all statements. Unbuffered queries should not be a
  1152. problem as we won't allow sending COM_CHANGE_USER.
  1153. */
  1154. DBG_INF(ret == PASS? "PASS":"FAIL");
  1155. DBG_RETURN(ret);
  1156. }
  1157. /* }}} */
  1158. /* {{{ mysqlnd_conn_data::set_client_option */
  1159. static enum_func_status
  1160. MYSQLND_METHOD(mysqlnd_conn_data, set_client_option)(MYSQLND_CONN_DATA * const conn,
  1161. enum_mysqlnd_client_option option,
  1162. const char * const value
  1163. )
  1164. {
  1165. enum_func_status ret = PASS;
  1166. DBG_ENTER("mysqlnd_conn_data::set_client_option");
  1167. DBG_INF_FMT("conn=%" PRIu64 " option=%u", conn->thread_id, option);
  1168. switch (option) {
  1169. case MYSQL_OPT_READ_TIMEOUT:
  1170. case MYSQL_OPT_WRITE_TIMEOUT:
  1171. case MYSQLND_OPT_SSL_KEY:
  1172. case MYSQLND_OPT_SSL_CERT:
  1173. case MYSQLND_OPT_SSL_CA:
  1174. case MYSQLND_OPT_SSL_CAPATH:
  1175. case MYSQLND_OPT_SSL_CIPHER:
  1176. case MYSQL_OPT_SSL_VERIFY_SERVER_CERT:
  1177. case MYSQL_OPT_CONNECT_TIMEOUT:
  1178. case MYSQLND_OPT_NET_READ_BUFFER_SIZE:
  1179. ret = conn->vio->data->m.set_client_option(conn->vio, option, value);
  1180. break;
  1181. case MYSQLND_OPT_NET_CMD_BUFFER_SIZE:
  1182. case MYSQL_OPT_COMPRESS:
  1183. case MYSQL_SERVER_PUBLIC_KEY:
  1184. ret = conn->protocol_frame_codec->data->m.set_client_option(conn->protocol_frame_codec, option, value);
  1185. break;
  1186. case MYSQLND_OPT_INT_AND_FLOAT_NATIVE:
  1187. conn->options->int_and_float_native = *(unsigned int*) value;
  1188. break;
  1189. case MYSQL_OPT_LOCAL_INFILE:
  1190. if (value && (*(unsigned int*) value) ? 1 : 0) {
  1191. conn->options->flags |= CLIENT_LOCAL_FILES;
  1192. } else {
  1193. conn->options->flags &= ~CLIENT_LOCAL_FILES;
  1194. }
  1195. break;
  1196. case MYSQL_OPT_LOAD_DATA_LOCAL_DIR:
  1197. {
  1198. if (conn->options->local_infile_directory) {
  1199. mnd_pefree(conn->options->local_infile_directory, conn->persistent);
  1200. }
  1201. if (!value || (*value == '\0')) {
  1202. conn->options->local_infile_directory = NULL;
  1203. } else {
  1204. conn->options->local_infile_directory = mnd_pestrdup(value, conn->persistent);
  1205. }
  1206. break;
  1207. }
  1208. case MYSQL_INIT_COMMAND:
  1209. {
  1210. char ** new_init_commands;
  1211. char * new_command;
  1212. /* when num_commands is 0, then realloc will be effectively a malloc call, internally */
  1213. /* Don't assign to conn->options->init_commands because in case of OOM we will lose the pointer and leak */
  1214. new_init_commands = mnd_perealloc(conn->options->init_commands, sizeof(char *) * (conn->options->num_commands + 1), conn->persistent);
  1215. conn->options->init_commands = new_init_commands;
  1216. new_command = mnd_pestrdup(value, conn->persistent);
  1217. conn->options->init_commands[conn->options->num_commands] = new_command;
  1218. ++conn->options->num_commands;
  1219. break;
  1220. }
  1221. case MYSQL_READ_DEFAULT_FILE:
  1222. case MYSQL_READ_DEFAULT_GROUP:
  1223. #ifdef WHEN_SUPPORTED_BY_MYSQLI
  1224. case MYSQL_SET_CLIENT_IP:
  1225. case MYSQL_REPORT_DATA_TRUNCATION:
  1226. #endif
  1227. /* currently not supported. Todo!! */
  1228. break;
  1229. case MYSQL_SET_CHARSET_NAME:
  1230. {
  1231. char * new_charset_name;
  1232. if (!mysqlnd_find_charset_name(value)) {
  1233. SET_CLIENT_ERROR(conn->error_info, CR_CANT_FIND_CHARSET, UNKNOWN_SQLSTATE, "Unknown character set");
  1234. ret = FAIL;
  1235. break;
  1236. }
  1237. new_charset_name = mnd_pestrdup(value, conn->persistent);
  1238. if (conn->options->charset_name) {
  1239. mnd_pefree(conn->options->charset_name, conn->persistent);
  1240. }
  1241. conn->options->charset_name = new_charset_name;
  1242. DBG_INF_FMT("charset=%s", conn->options->charset_name);
  1243. break;
  1244. }
  1245. case MYSQL_OPT_NAMED_PIPE:
  1246. conn->options->protocol = MYSQL_PROTOCOL_PIPE;
  1247. break;
  1248. case MYSQL_OPT_PROTOCOL:
  1249. if (*(unsigned int*) value < MYSQL_PROTOCOL_LAST) {
  1250. conn->options->protocol = *(unsigned int*) value;
  1251. }
  1252. break;
  1253. #ifdef WHEN_SUPPORTED_BY_MYSQLI
  1254. case MYSQL_SET_CHARSET_DIR:
  1255. case MYSQL_OPT_RECONNECT:
  1256. /* we don't need external character sets, all character sets are
  1257. compiled in. For compatibility we just ignore this setting.
  1258. Same for protocol, we don't support old protocol */
  1259. case MYSQL_OPT_USE_REMOTE_CONNECTION:
  1260. case MYSQL_OPT_USE_EMBEDDED_CONNECTION:
  1261. case MYSQL_OPT_GUESS_CONNECTION:
  1262. /* todo: throw an error, we don't support embedded */
  1263. break;
  1264. #endif
  1265. case MYSQLND_OPT_MAX_ALLOWED_PACKET:
  1266. if (*(unsigned int*) value > (1<<16)) {
  1267. conn->options->max_allowed_packet = *(unsigned int*) value;
  1268. }
  1269. break;
  1270. case MYSQLND_OPT_AUTH_PROTOCOL:
  1271. {
  1272. char * new_auth_protocol = value? mnd_pestrdup(value, conn->persistent) : NULL;
  1273. if (conn->options->auth_protocol) {
  1274. mnd_pefree(conn->options->auth_protocol, conn->persistent);
  1275. }
  1276. conn->options->auth_protocol = new_auth_protocol;
  1277. DBG_INF_FMT("auth_protocol=%s", conn->options->auth_protocol);
  1278. break;
  1279. }
  1280. case MYSQL_OPT_CAN_HANDLE_EXPIRED_PASSWORDS:
  1281. if (value && (*(unsigned int*) value) ? 1 : 0) {
  1282. conn->options->flags |= CLIENT_CAN_HANDLE_EXPIRED_PASSWORDS;
  1283. } else {
  1284. conn->options->flags &= ~CLIENT_CAN_HANDLE_EXPIRED_PASSWORDS;
  1285. }
  1286. break;
  1287. case MYSQL_OPT_CONNECT_ATTR_RESET:
  1288. if (conn->options->connect_attr) {
  1289. DBG_INF_FMT("Before reset %d attribute(s)", zend_hash_num_elements(conn->options->connect_attr));
  1290. zend_hash_clean(conn->options->connect_attr);
  1291. }
  1292. break;
  1293. case MYSQL_OPT_CONNECT_ATTR_DELETE:
  1294. if (conn->options->connect_attr && value) {
  1295. DBG_INF_FMT("Before delete %d attribute(s)", zend_hash_num_elements(conn->options->connect_attr));
  1296. zend_hash_str_del(conn->options->connect_attr, value, strlen(value));
  1297. DBG_INF_FMT("%d left", zend_hash_num_elements(conn->options->connect_attr));
  1298. }
  1299. break;
  1300. #ifdef WHEN_SUPPORTED_BY_MYSQLI
  1301. case MYSQL_SHARED_MEMORY_BASE_NAME:
  1302. case MYSQL_OPT_USE_RESULT:
  1303. case MYSQL_SECURE_AUTH:
  1304. /* not sure, todo ? */
  1305. #endif
  1306. default:
  1307. ret = FAIL;
  1308. }
  1309. DBG_RETURN(ret);
  1310. }
  1311. /* }}} */
  1312. /* {{{ mysqlnd_conn_data::set_client_option_2d */
  1313. static enum_func_status
  1314. MYSQLND_METHOD(mysqlnd_conn_data, set_client_option_2d)(MYSQLND_CONN_DATA * const conn,
  1315. const enum_mysqlnd_client_option option,
  1316. const char * const key,
  1317. const char * const value
  1318. )
  1319. {
  1320. enum_func_status ret = PASS;
  1321. DBG_ENTER("mysqlnd_conn_data::set_client_option_2d");
  1322. DBG_INF_FMT("conn=%" PRIu64 " option=%u", conn->thread_id, option);
  1323. switch (option) {
  1324. case MYSQL_OPT_CONNECT_ATTR_ADD:
  1325. if (!conn->options->connect_attr) {
  1326. DBG_INF("Initializing connect_attr hash");
  1327. conn->options->connect_attr = mnd_pemalloc(sizeof(HashTable), conn->persistent);
  1328. zend_hash_init(conn->options->connect_attr, 0, NULL, conn->persistent ? ZVAL_INTERNAL_PTR_DTOR : ZVAL_PTR_DTOR, conn->persistent);
  1329. }
  1330. DBG_INF_FMT("Adding [%s][%s]", key, value);
  1331. {
  1332. zval attrz;
  1333. zend_string *str;
  1334. if (conn->persistent) {
  1335. str = zend_string_init(key, strlen(key), 1);
  1336. GC_MAKE_PERSISTENT_LOCAL(str);
  1337. ZVAL_NEW_STR(&attrz, zend_string_init(value, strlen(value), 1));
  1338. GC_MAKE_PERSISTENT_LOCAL(Z_COUNTED(attrz));
  1339. } else {
  1340. str = zend_string_init(key, strlen(key), 0);
  1341. ZVAL_NEW_STR(&attrz, zend_string_init(value, strlen(value), 0));
  1342. }
  1343. zend_hash_update(conn->options->connect_attr, str, &attrz);
  1344. zend_string_release_ex(str, 1);
  1345. }
  1346. break;
  1347. default:
  1348. ret = FAIL;
  1349. }
  1350. DBG_RETURN(ret);
  1351. }
  1352. /* }}} */
  1353. /* {{{ mysqlnd_conn_data::use_result */
  1354. static MYSQLND_RES *
  1355. MYSQLND_METHOD(mysqlnd_conn_data, use_result)(MYSQLND_CONN_DATA * const conn)
  1356. {
  1357. DBG_ENTER("mysqlnd_conn_data::use_result");
  1358. DBG_INF_FMT("conn=%" PRIu64, conn->thread_id);
  1359. if (!conn->current_result) {
  1360. DBG_RETURN(NULL);
  1361. }
  1362. /* Nothing to store for UPSERT/LOAD DATA */
  1363. if (conn->last_query_type != QUERY_SELECT || GET_CONNECTION_STATE(&conn->state) != CONN_FETCHING_DATA) {
  1364. SET_CLIENT_ERROR(conn->error_info, CR_COMMANDS_OUT_OF_SYNC, UNKNOWN_SQLSTATE, mysqlnd_out_of_sync);
  1365. DBG_ERR("Command out of sync");
  1366. DBG_RETURN(NULL);
  1367. }
  1368. MYSQLND_INC_CONN_STATISTIC(conn->stats, STAT_UNBUFFERED_SETS);
  1369. conn->current_result->conn = conn->m->get_reference(conn);
  1370. MYSQLND_RES *result = conn->current_result->m.use_result(conn->current_result, FALSE);
  1371. if (!result) {
  1372. conn->current_result->m.free_result(conn->current_result, TRUE);
  1373. }
  1374. conn->current_result = NULL;
  1375. DBG_RETURN(result);
  1376. }
  1377. /* }}} */
  1378. /* {{{ mysqlnd_conn_data::store_result */
  1379. static MYSQLND_RES *
  1380. MYSQLND_METHOD(mysqlnd_conn_data, store_result)(MYSQLND_CONN_DATA * const conn)
  1381. {
  1382. DBG_ENTER("mysqlnd_conn_data::store_result");
  1383. DBG_INF_FMT("conn=%" PRIu64 " conn=%p", conn->thread_id, conn);
  1384. if (!conn->current_result) {
  1385. DBG_RETURN(NULL);
  1386. }
  1387. /* Nothing to store for UPSERT/LOAD DATA*/
  1388. if (conn->last_query_type != QUERY_SELECT || GET_CONNECTION_STATE(&conn->state) != CONN_FETCHING_DATA) {
  1389. SET_CLIENT_ERROR(conn->error_info, CR_COMMANDS_OUT_OF_SYNC, UNKNOWN_SQLSTATE, mysqlnd_out_of_sync);
  1390. DBG_ERR("Command out of sync");
  1391. DBG_RETURN(NULL);
  1392. }
  1393. MYSQLND_INC_CONN_STATISTIC(conn->stats, STAT_BUFFERED_SETS);
  1394. MYSQLND_RES *result = conn->current_result->m.store_result(conn->current_result, conn, NULL);
  1395. if (!result) {
  1396. conn->current_result->m.free_result(conn->current_result, TRUE);
  1397. }
  1398. conn->current_result = NULL;
  1399. DBG_RETURN(result);
  1400. }
  1401. /* }}} */
  1402. /* {{{ mysqlnd_conn_data::get_connection_stats */
  1403. static void
  1404. MYSQLND_METHOD(mysqlnd_conn_data, get_connection_stats)(const MYSQLND_CONN_DATA * const conn,
  1405. zval * return_value ZEND_FILE_LINE_DC)
  1406. {
  1407. DBG_ENTER("mysqlnd_conn_data::get_connection_stats");
  1408. mysqlnd_fill_stats_hash(conn->stats, mysqlnd_stats_values_names, return_value ZEND_FILE_LINE_CC);
  1409. DBG_VOID_RETURN;
  1410. }
  1411. /* }}} */
  1412. /* {{{ mysqlnd_conn_data::set_autocommit */
  1413. static enum_func_status
  1414. MYSQLND_METHOD(mysqlnd_conn_data, set_autocommit)(MYSQLND_CONN_DATA * conn, unsigned int mode)
  1415. {
  1416. DBG_ENTER("mysqlnd_conn_data::set_autocommit");
  1417. DBG_RETURN(conn->m->query(conn, (mode) ? "SET AUTOCOMMIT=1":"SET AUTOCOMMIT=0", sizeof("SET AUTOCOMMIT=1") - 1));
  1418. }
  1419. /* }}} */
  1420. /* {{{ mysqlnd_conn_data::tx_commit */
  1421. static enum_func_status
  1422. MYSQLND_METHOD(mysqlnd_conn_data, tx_commit)(MYSQLND_CONN_DATA * conn)
  1423. {
  1424. return conn->m->tx_commit_or_rollback(conn, TRUE, TRANS_COR_NO_OPT, NULL);
  1425. }
  1426. /* }}} */
  1427. /* {{{ mysqlnd_conn_data::tx_rollback */
  1428. static enum_func_status
  1429. MYSQLND_METHOD(mysqlnd_conn_data, tx_rollback)(MYSQLND_CONN_DATA * conn)
  1430. {
  1431. return conn->m->tx_commit_or_rollback(conn, FALSE, TRANS_COR_NO_OPT, NULL);
  1432. }
  1433. /* }}} */
  1434. /* {{{ mysqlnd_tx_cor_options_to_string */
  1435. static void
  1436. MYSQLND_METHOD(mysqlnd_conn_data, tx_cor_options_to_string)(const MYSQLND_CONN_DATA * const conn, smart_str * str, const unsigned int mode)
  1437. {
  1438. if ((mode & TRANS_COR_AND_CHAIN) && !(mode & TRANS_COR_AND_NO_CHAIN)) {
  1439. if (str->s && ZSTR_LEN(str->s)) {
  1440. smart_str_appendl(str, " ", sizeof(" ") - 1);
  1441. }
  1442. smart_str_appendl(str, "AND CHAIN", sizeof("AND CHAIN") - 1);
  1443. } else if ((mode & TRANS_COR_AND_NO_CHAIN) && !(mode & TRANS_COR_AND_CHAIN)) {
  1444. if (str->s && ZSTR_LEN(str->s)) {
  1445. smart_str_appendl(str, " ", sizeof(" ") - 1);
  1446. }
  1447. smart_str_appendl(str, "AND NO CHAIN", sizeof("AND NO CHAIN") - 1);
  1448. }
  1449. if ((mode & TRANS_COR_RELEASE) && !(mode & TRANS_COR_NO_RELEASE)) {
  1450. if (str->s && ZSTR_LEN(str->s)) {
  1451. smart_str_appendl(str, " ", sizeof(" ") - 1);
  1452. }
  1453. smart_str_appendl(str, "RELEASE", sizeof("RELEASE") - 1);
  1454. } else if ((mode & TRANS_COR_NO_RELEASE) && !(mode & TRANS_COR_RELEASE)) {
  1455. if (str->s && ZSTR_LEN(str->s)) {
  1456. smart_str_appendl(str, " ", sizeof(" ") - 1);
  1457. }
  1458. smart_str_appendl(str, "NO RELEASE", sizeof("NO RELEASE") - 1);
  1459. }
  1460. smart_str_0(str);
  1461. }
  1462. /* }}} */
  1463. /* {{{ mysqlnd_escape_string_for_tx_name_in_comment */
  1464. static char *
  1465. mysqlnd_escape_string_for_tx_name_in_comment(const char * const name)
  1466. {
  1467. char * ret = NULL;
  1468. DBG_ENTER("mysqlnd_escape_string_for_tx_name_in_comment");
  1469. if (name) {
  1470. bool warned = FALSE;
  1471. const char * p_orig = name;
  1472. char * p_copy;
  1473. p_copy = ret = mnd_emalloc(strlen(name) + 1 + 2 + 2 + 1); /* space, open, close, NullS */
  1474. *p_copy++ = ' ';
  1475. *p_copy++ = '/';
  1476. *p_copy++ = '*';
  1477. while (1) {
  1478. register char v = *p_orig;
  1479. if (v == 0) {
  1480. break;
  1481. }
  1482. if ((v >= '0' && v <= '9') ||
  1483. (v >= 'a' && v <= 'z') ||
  1484. (v >= 'A' && v <= 'Z') ||
  1485. v == '-' ||
  1486. v == '_' ||
  1487. v == ' ' ||
  1488. v == '=')
  1489. {
  1490. *p_copy++ = v;
  1491. } else if (warned == FALSE) {
  1492. php_error_docref(NULL, E_WARNING, "Transaction name has been truncated, since it can only contain the A-Z, a-z, 0-9, \"\\\", \"-\", \"_\", and \"=\" characters");
  1493. warned = TRUE;
  1494. }
  1495. ++p_orig;
  1496. }
  1497. *p_copy++ = '*';
  1498. *p_copy++ = '/';
  1499. *p_copy++ = 0;
  1500. }
  1501. DBG_RETURN(ret);
  1502. }
  1503. /* }}} */
  1504. /* {{{ mysqlnd_conn_data::tx_commit_ex */
  1505. static enum_func_status
  1506. MYSQLND_METHOD(mysqlnd_conn_data, tx_commit_or_rollback)(MYSQLND_CONN_DATA * conn, const bool commit, const unsigned int flags, const char * const name)
  1507. {
  1508. DBG_ENTER("mysqlnd_conn_data::tx_commit_or_rollback");
  1509. smart_str tmp_str = {0, 0};
  1510. conn->m->tx_cor_options_to_string(conn, &tmp_str, flags);
  1511. smart_str_0(&tmp_str);
  1512. char * query;
  1513. size_t query_len;
  1514. char * name_esc = mysqlnd_escape_string_for_tx_name_in_comment(name);
  1515. query_len = mnd_sprintf(&query, 0, (commit? "COMMIT%s %s":"ROLLBACK%s %s"),
  1516. name_esc? name_esc:"", tmp_str.s? ZSTR_VAL(tmp_str.s):"");
  1517. smart_str_free(&tmp_str);
  1518. if (name_esc) {
  1519. mnd_efree(name_esc);
  1520. name_esc = NULL;
  1521. }
  1522. if (!query) {
  1523. SET_OOM_ERROR(conn->error_info);
  1524. DBG_RETURN(FAIL);
  1525. }
  1526. enum_func_status ret = conn->m->query(conn, query, query_len);
  1527. mnd_sprintf_free(query);
  1528. DBG_RETURN(ret);
  1529. }
  1530. /* }}} */
  1531. /* {{{ mysqlnd_conn_data::tx_begin */
  1532. static enum_func_status
  1533. MYSQLND_METHOD(mysqlnd_conn_data, tx_begin)(MYSQLND_CONN_DATA * conn, const unsigned int mode, const char * const name)
  1534. {
  1535. DBG_ENTER("mysqlnd_conn_data::tx_begin");
  1536. smart_str tmp_str = {0, 0};
  1537. if (mode & TRANS_START_WITH_CONSISTENT_SNAPSHOT) {
  1538. if (tmp_str.s) {
  1539. smart_str_appendl(&tmp_str, ", ", sizeof(", ") - 1);
  1540. }
  1541. smart_str_appendl(&tmp_str, "WITH CONSISTENT SNAPSHOT", sizeof("WITH CONSISTENT SNAPSHOT") - 1);
  1542. }
  1543. if (mode & TRANS_START_READ_WRITE) {
  1544. if (tmp_str.s && ZSTR_LEN(tmp_str.s)) {
  1545. smart_str_appendl(&tmp_str, ", ", sizeof(", ") - 1);
  1546. }
  1547. smart_str_appendl(&tmp_str, "READ WRITE", sizeof("READ WRITE") - 1);
  1548. } else if (mode & TRANS_START_READ_ONLY) {
  1549. if (tmp_str.s && ZSTR_LEN(tmp_str.s)) {
  1550. smart_str_appendl(&tmp_str, ", ", sizeof(", ") - 1);
  1551. }
  1552. smart_str_appendl(&tmp_str, "READ ONLY", sizeof("READ ONLY") - 1);
  1553. }
  1554. smart_str_0(&tmp_str);
  1555. char * name_esc = mysqlnd_escape_string_for_tx_name_in_comment(name);
  1556. char * query;
  1557. unsigned int query_len = mnd_sprintf(&query, 0, "START TRANSACTION%s %s", name_esc? name_esc:"", tmp_str.s? ZSTR_VAL(tmp_str.s):"");
  1558. smart_str_free(&tmp_str);
  1559. if (name_esc) {
  1560. mnd_efree(name_esc);
  1561. name_esc = NULL;
  1562. }
  1563. if (!query) {
  1564. SET_OOM_ERROR(conn->error_info);
  1565. DBG_RETURN(FAIL);
  1566. }
  1567. enum_func_status ret = conn->m->query(conn, query, query_len);
  1568. mnd_sprintf_free(query);
  1569. if (ret && mode & (TRANS_START_READ_WRITE | TRANS_START_READ_ONLY) && mysqlnd_stmt_errno(conn) == 1064) {
  1570. SET_CLIENT_ERROR(conn->error_info, CR_NOT_IMPLEMENTED, UNKNOWN_SQLSTATE,
  1571. "This server version doesn't support 'READ WRITE' and 'READ ONLY'. Minimum 5.6.5 is required");
  1572. }
  1573. DBG_RETURN(ret);
  1574. }
  1575. /* }}} */
  1576. /* {{{ mysqlnd_conn_data::tx_savepoint */
  1577. static enum_func_status
  1578. MYSQLND_METHOD(mysqlnd_conn_data, tx_savepoint)(MYSQLND_CONN_DATA * conn, const char * const name)
  1579. {
  1580. DBG_ENTER("mysqlnd_conn_data::tx_savepoint");
  1581. if (!name) {
  1582. SET_CLIENT_ERROR(conn->error_info, CR_UNKNOWN_ERROR, UNKNOWN_SQLSTATE, "Savepoint name not provided");
  1583. DBG_RETURN(FAIL);
  1584. }
  1585. char *query;
  1586. size_t query_len = mnd_sprintf(&query, 0, "SAVEPOINT `%s`", name);
  1587. if (!query) {
  1588. SET_OOM_ERROR(conn->error_info);
  1589. DBG_RETURN(FAIL);
  1590. }
  1591. enum_func_status ret = conn->m->query(conn, query, query_len);
  1592. mnd_sprintf_free(query);
  1593. DBG_RETURN(ret);
  1594. }
  1595. /* }}} */
  1596. /* {{{ mysqlnd_conn_data::tx_savepoint_release */
  1597. static enum_func_status
  1598. MYSQLND_METHOD(mysqlnd_conn_data, tx_savepoint_release)(MYSQLND_CONN_DATA * conn, const char * const name)
  1599. {
  1600. DBG_ENTER("mysqlnd_conn_data::tx_savepoint_release");
  1601. if (!name) {
  1602. SET_CLIENT_ERROR(conn->error_info, CR_UNKNOWN_ERROR, UNKNOWN_SQLSTATE, "Savepoint name not provided");
  1603. DBG_RETURN(FAIL);
  1604. }
  1605. char *query;
  1606. size_t query_len = mnd_sprintf(&query, 0, "RELEASE SAVEPOINT `%s`", name);
  1607. if (!query) {
  1608. SET_OOM_ERROR(conn->error_info);
  1609. DBG_RETURN(FAIL);
  1610. }
  1611. enum_func_status ret = conn->m->query(conn, query, query_len);
  1612. mnd_sprintf_free(query);
  1613. DBG_RETURN(ret);
  1614. }
  1615. /* }}} */
  1616. /* {{{ mysqlnd_conn_data::negotiate_client_api_capabilities */
  1617. static size_t
  1618. MYSQLND_METHOD(mysqlnd_conn_data, negotiate_client_api_capabilities)(MYSQLND_CONN_DATA * const conn, const size_t flags)
  1619. {
  1620. unsigned int ret = 0;
  1621. DBG_ENTER("mysqlnd_conn_data::negotiate_client_api_capabilities");
  1622. if (conn) {
  1623. ret = conn->client_api_capabilities;
  1624. conn->client_api_capabilities = flags;
  1625. }
  1626. DBG_RETURN(ret);
  1627. }
  1628. /* }}} */
  1629. /* {{{ mysqlnd_conn_data::get_client_api_capabilities */
  1630. static size_t
  1631. MYSQLND_METHOD(mysqlnd_conn_data, get_client_api_capabilities)(const MYSQLND_CONN_DATA * const conn)
  1632. {
  1633. DBG_ENTER("mysqlnd_conn_data::get_client_api_capabilities");
  1634. DBG_RETURN(conn? conn->client_api_capabilities : 0);
  1635. }
  1636. /* }}} */
  1637. /* {{{ _mysqlnd_stmt_init */
  1638. MYSQLND_STMT *
  1639. MYSQLND_METHOD(mysqlnd_conn_data, stmt_init)(MYSQLND_CONN_DATA * const conn)
  1640. {
  1641. MYSQLND_STMT * ret;
  1642. DBG_ENTER("mysqlnd_conn_data::stmt_init");
  1643. ret = conn->object_factory.get_prepared_statement(conn);
  1644. DBG_RETURN(ret);
  1645. }
  1646. /* }}} */
  1647. MYSQLND_CLASS_METHODS_START(mysqlnd_conn_data)
  1648. MYSQLND_METHOD(mysqlnd_conn_data, connect),
  1649. MYSQLND_METHOD(mysqlnd_conn_data, escape_string),
  1650. MYSQLND_METHOD(mysqlnd_conn_data, set_charset),
  1651. MYSQLND_METHOD(mysqlnd_conn_data, query),
  1652. MYSQLND_METHOD(mysqlnd_conn_data, send_query),
  1653. MYSQLND_METHOD(mysqlnd_conn_data, reap_query),
  1654. MYSQLND_METHOD(mysqlnd_conn_data, use_result),
  1655. MYSQLND_METHOD(mysqlnd_conn_data, store_result),
  1656. MYSQLND_METHOD(mysqlnd_conn_data, next_result),
  1657. MYSQLND_METHOD(mysqlnd_conn_data, more_results),
  1658. MYSQLND_METHOD(mysqlnd_conn_data, stmt_init),
  1659. MYSQLND_METHOD(mysqlnd_conn_data, shutdown),
  1660. MYSQLND_METHOD(mysqlnd_conn_data, refresh),
  1661. MYSQLND_METHOD(mysqlnd_conn_data, ping),
  1662. MYSQLND_METHOD(mysqlnd_conn_data, kill),
  1663. MYSQLND_METHOD(mysqlnd_conn_data, select_db),
  1664. MYSQLND_METHOD(mysqlnd_conn_data, dump_debug_info),
  1665. MYSQLND_METHOD(mysqlnd_conn_data, change_user),
  1666. MYSQLND_METHOD(mysqlnd_conn_data, err_no),
  1667. MYSQLND_METHOD(mysqlnd_conn_data, error),
  1668. MYSQLND_METHOD(mysqlnd_conn_data, sqlstate),
  1669. MYSQLND_METHOD(mysqlnd_conn_data, thread_id),
  1670. MYSQLND_METHOD(mysqlnd_conn_data, get_connection_stats),
  1671. MYSQLND_METHOD(mysqlnd_conn_data, get_server_version),
  1672. MYSQLND_METHOD(mysqlnd_conn_data, get_server_info),
  1673. MYSQLND_METHOD(mysqlnd_conn_data, statistic),
  1674. MYSQLND_METHOD(mysqlnd_conn_data, get_host_info),
  1675. MYSQLND_METHOD(mysqlnd_conn_data, get_proto_info),
  1676. MYSQLND_METHOD(mysqlnd_conn_data, info),
  1677. MYSQLND_METHOD(mysqlnd_conn_data, charset_name),
  1678. MYSQLND_METHOD(mysqlnd_conn_data, list_method),
  1679. MYSQLND_METHOD(mysqlnd_conn_data, insert_id),
  1680. MYSQLND_METHOD(mysqlnd_conn_data, affected_rows),
  1681. MYSQLND_METHOD(mysqlnd_conn_data, warning_count),
  1682. MYSQLND_METHOD(mysqlnd_conn_data, field_count),
  1683. MYSQLND_METHOD(mysqlnd_conn_data, server_status),
  1684. MYSQLND_METHOD(mysqlnd_conn_data, set_server_option),
  1685. MYSQLND_METHOD(mysqlnd_conn_data, set_client_option),
  1686. MYSQLND_METHOD(mysqlnd_conn_data, free_contents),
  1687. MYSQLND_METHOD(mysqlnd_conn_data, free_options),
  1688. MYSQLND_METHOD_PRIVATE(mysqlnd_conn_data, dtor),
  1689. mysqlnd_query_read_result_set_header,
  1690. MYSQLND_METHOD_PRIVATE(mysqlnd_conn_data, get_reference),
  1691. MYSQLND_METHOD_PRIVATE(mysqlnd_conn_data, free_reference),
  1692. MYSQLND_METHOD(mysqlnd_conn_data, restart_psession),
  1693. MYSQLND_METHOD(mysqlnd_conn_data, end_psession),
  1694. MYSQLND_METHOD(mysqlnd_conn_data, send_close),
  1695. MYSQLND_METHOD(mysqlnd_conn_data, ssl_set),
  1696. mysqlnd_result_init,
  1697. MYSQLND_METHOD(mysqlnd_conn_data, set_autocommit),
  1698. MYSQLND_METHOD(mysqlnd_conn_data, tx_commit),
  1699. MYSQLND_METHOD(mysqlnd_conn_data, tx_rollback),
  1700. MYSQLND_METHOD(mysqlnd_conn_data, tx_begin),
  1701. MYSQLND_METHOD(mysqlnd_conn_data, tx_commit_or_rollback),
  1702. MYSQLND_METHOD(mysqlnd_conn_data, tx_cor_options_to_string),
  1703. MYSQLND_METHOD(mysqlnd_conn_data, tx_savepoint),
  1704. MYSQLND_METHOD(mysqlnd_conn_data, tx_savepoint_release),
  1705. MYSQLND_METHOD(mysqlnd_conn_data, execute_init_commands),
  1706. MYSQLND_METHOD(mysqlnd_conn_data, get_updated_connect_flags),
  1707. MYSQLND_METHOD(mysqlnd_conn_data, connect_handshake),
  1708. MYSQLND_METHOD(mysqlnd_conn_data, fetch_auth_plugin_by_name),
  1709. MYSQLND_METHOD(mysqlnd_conn_data, set_client_option_2d),
  1710. MYSQLND_METHOD(mysqlnd_conn_data, negotiate_client_api_capabilities),
  1711. MYSQLND_METHOD(mysqlnd_conn_data, get_client_api_capabilities),
  1712. MYSQLND_METHOD(mysqlnd_conn_data, get_scheme)
  1713. MYSQLND_CLASS_METHODS_END;
  1714. /* {{{ mysqlnd_conn::get_reference */
  1715. static MYSQLND *
  1716. MYSQLND_METHOD(mysqlnd_conn, clone_object)(MYSQLND * const conn)
  1717. {
  1718. MYSQLND * ret;
  1719. DBG_ENTER("mysqlnd_conn::get_reference");
  1720. ret = conn->data->object_factory.clone_connection_object(conn);
  1721. DBG_RETURN(ret);
  1722. }
  1723. /* }}} */
  1724. /* {{{ mysqlnd_conn_data::dtor */
  1725. static void
  1726. MYSQLND_METHOD_PRIVATE(mysqlnd_conn, dtor)(MYSQLND * conn)
  1727. {
  1728. DBG_ENTER("mysqlnd_conn::dtor");
  1729. DBG_INF_FMT("conn=%" PRIu64, conn->data->thread_id);
  1730. conn->data->m->free_reference(conn->data);
  1731. mnd_pefree(conn, conn->persistent);
  1732. DBG_VOID_RETURN;
  1733. }
  1734. /* }}} */
  1735. /* {{{ mysqlnd_conn_data::close */
  1736. static enum_func_status
  1737. MYSQLND_METHOD(mysqlnd_conn, close)(MYSQLND * conn_handle, const enum_connection_close_type close_type)
  1738. {
  1739. MYSQLND_CONN_DATA * conn = conn_handle->data;
  1740. DBG_ENTER("mysqlnd_conn::close");
  1741. DBG_INF_FMT("conn=%" PRIu64, conn->thread_id);
  1742. if (GET_CONNECTION_STATE(&conn->state) >= CONN_READY) {
  1743. static enum_mysqlnd_collected_stats close_type_to_stat_map[MYSQLND_CLOSE_LAST] = {
  1744. STAT_CLOSE_EXPLICIT,
  1745. STAT_CLOSE_IMPLICIT,
  1746. STAT_CLOSE_DISCONNECT
  1747. };
  1748. MYSQLND_INC_CONN_STATISTIC(conn->stats, close_type_to_stat_map[close_type]);
  1749. }
  1750. /*
  1751. Close now, free_reference will try,
  1752. if we are last, but that's not a problem.
  1753. */
  1754. enum_func_status ret = conn->m->send_close(conn);
  1755. conn_handle->m->dtor(conn_handle);
  1756. DBG_RETURN(ret);
  1757. }
  1758. /* }}} */
  1759. MYSQLND_CLASS_METHODS_START(mysqlnd_conn)
  1760. MYSQLND_METHOD(mysqlnd_conn, connect),
  1761. MYSQLND_METHOD(mysqlnd_conn, clone_object),
  1762. MYSQLND_METHOD_PRIVATE(mysqlnd_conn, dtor),
  1763. MYSQLND_METHOD(mysqlnd_conn, close)
  1764. MYSQLND_CLASS_METHODS_END;
  1765. #include "php_network.h"
  1766. /* {{{ mysqlnd_stream_array_to_fd_set */
  1767. MYSQLND **
  1768. mysqlnd_stream_array_check_for_readiness(MYSQLND ** conn_array)
  1769. {
  1770. unsigned int cnt = 0;
  1771. MYSQLND **p = conn_array, **p_p;
  1772. MYSQLND **ret = NULL;
  1773. while (*p) {
  1774. const enum mysqlnd_connection_state conn_state = GET_CONNECTION_STATE(&((*p)->data->state));
  1775. if (conn_state <= CONN_READY || conn_state == CONN_QUIT_SENT) {
  1776. cnt++;
  1777. }
  1778. p++;
  1779. }
  1780. if (cnt) {
  1781. MYSQLND **ret_p = ret = ecalloc(cnt + 1, sizeof(MYSQLND *));
  1782. p_p = p = conn_array;
  1783. while (*p) {
  1784. const enum mysqlnd_connection_state conn_state = GET_CONNECTION_STATE(&((*p)->data->state));
  1785. if (conn_state <= CONN_READY || conn_state == CONN_QUIT_SENT) {
  1786. *ret_p = *p;
  1787. *p = NULL;
  1788. ret_p++;
  1789. } else {
  1790. *p_p = *p;
  1791. p_p++;
  1792. }
  1793. p++;
  1794. }
  1795. *ret_p = NULL;
  1796. }
  1797. return ret;
  1798. }
  1799. /* }}} */
  1800. /* {{{ mysqlnd_stream_array_to_fd_set */
  1801. static unsigned int
  1802. mysqlnd_stream_array_to_fd_set(MYSQLND ** conn_array, fd_set * fds, php_socket_t * max_fd)
  1803. {
  1804. php_socket_t this_fd;
  1805. php_stream *stream = NULL;
  1806. unsigned int cnt = 0;
  1807. MYSQLND **p = conn_array;
  1808. DBG_ENTER("mysqlnd_stream_array_to_fd_set");
  1809. while (*p) {
  1810. /* get the fd.
  1811. * NB: Most other code will NOT use the PHP_STREAM_CAST_INTERNAL flag
  1812. * when casting. It is only used here so that the buffered data warning
  1813. * is not displayed.
  1814. * */
  1815. stream = (*p)->data->vio->data->m.get_stream((*p)->data->vio);
  1816. DBG_INF_FMT("conn=%" PRIu64 " stream=%p", (*p)->data->thread_id, stream);
  1817. if (stream != NULL &&
  1818. SUCCESS == php_stream_cast(stream, PHP_STREAM_AS_FD_FOR_SELECT | PHP_STREAM_CAST_INTERNAL, (void*)&this_fd, 1) &&
  1819. ZEND_VALID_SOCKET(this_fd))
  1820. {
  1821. PHP_SAFE_FD_SET(this_fd, fds);
  1822. if (this_fd > *max_fd) {
  1823. *max_fd = this_fd;
  1824. }
  1825. ++cnt;
  1826. }
  1827. ++p;
  1828. }
  1829. DBG_RETURN(cnt ? 1 : 0);
  1830. }
  1831. /* }}} */
  1832. /* {{{ mysqlnd_stream_array_from_fd_set */
  1833. static unsigned int
  1834. mysqlnd_stream_array_from_fd_set(MYSQLND ** conn_array, fd_set * fds)
  1835. {
  1836. php_socket_t this_fd;
  1837. php_stream *stream = NULL;
  1838. unsigned int ret = 0;
  1839. bool disproportion = FALSE;
  1840. MYSQLND **fwd = conn_array, **bckwd = conn_array;
  1841. DBG_ENTER("mysqlnd_stream_array_from_fd_set");
  1842. while (*fwd) {
  1843. stream = (*fwd)->data->vio->data->m.get_stream((*fwd)->data->vio);
  1844. DBG_INF_FMT("conn=%" PRIu64 " stream=%p", (*fwd)->data->thread_id, stream);
  1845. if (stream != NULL && SUCCESS == php_stream_cast(stream, PHP_STREAM_AS_FD_FOR_SELECT | PHP_STREAM_CAST_INTERNAL,
  1846. (void*)&this_fd, 1) && ZEND_VALID_SOCKET(this_fd)) {
  1847. if (PHP_SAFE_FD_ISSET(this_fd, fds)) {
  1848. if (disproportion) {
  1849. *bckwd = *fwd;
  1850. }
  1851. ++bckwd;
  1852. ++fwd;
  1853. ++ret;
  1854. continue;
  1855. }
  1856. }
  1857. disproportion = TRUE;
  1858. ++fwd;
  1859. }
  1860. *bckwd = NULL;/* NULL-terminate the list */
  1861. DBG_RETURN(ret);
  1862. }
  1863. /* }}} */
  1864. #ifndef PHP_WIN32
  1865. #define php_select(m, r, w, e, t) select(m, r, w, e, t)
  1866. #else
  1867. #include "win32/select.h"
  1868. #endif
  1869. /* {{{ mysqlnd_poll */
  1870. PHPAPI enum_func_status
  1871. mysqlnd_poll(MYSQLND **r_array, MYSQLND **e_array, MYSQLND ***dont_poll, long sec, long usec, int * desc_num)
  1872. {
  1873. struct timeval tv;
  1874. struct timeval *tv_p = NULL;
  1875. fd_set rfds, wfds, efds;
  1876. php_socket_t max_fd = 0;
  1877. int retval, sets = 0;
  1878. int set_count, max_set_count = 0;
  1879. DBG_ENTER("_mysqlnd_poll");
  1880. if (sec < 0 || usec < 0) {
  1881. php_error_docref(NULL, E_WARNING, "Negative values passed for sec and/or usec");
  1882. DBG_RETURN(FAIL);
  1883. }
  1884. FD_ZERO(&rfds);
  1885. FD_ZERO(&wfds);
  1886. FD_ZERO(&efds);
  1887. if (r_array != NULL) {
  1888. *dont_poll = mysqlnd_stream_array_check_for_readiness(r_array);
  1889. set_count = mysqlnd_stream_array_to_fd_set(r_array, &rfds, &max_fd);
  1890. if (set_count > max_set_count) {
  1891. max_set_count = set_count;
  1892. }
  1893. sets += set_count;
  1894. }
  1895. if (e_array != NULL) {
  1896. set_count = mysqlnd_stream_array_to_fd_set(e_array, &efds, &max_fd);
  1897. if (set_count > max_set_count) {
  1898. max_set_count = set_count;
  1899. }
  1900. sets += set_count;
  1901. }
  1902. if (!sets) {
  1903. php_error_docref(NULL, E_WARNING, *dont_poll ? "All arrays passed are clear":"No stream arrays were passed");
  1904. DBG_ERR_FMT(*dont_poll ? "All arrays passed are clear":"No stream arrays were passed");
  1905. DBG_RETURN(FAIL);
  1906. }
  1907. if (!PHP_SAFE_MAX_FD(max_fd, max_set_count)) {
  1908. DBG_RETURN(FAIL);
  1909. }
  1910. /* Solaris + BSD do not like microsecond values which are >= 1 sec */
  1911. if (usec > 999999) {
  1912. tv.tv_sec = sec + (usec / 1000000);
  1913. tv.tv_usec = usec % 1000000;
  1914. } else {
  1915. tv.tv_sec = sec;
  1916. tv.tv_usec = usec;
  1917. }
  1918. tv_p = &tv;
  1919. retval = php_select(max_fd + 1, &rfds, &wfds, &efds, tv_p);
  1920. if (retval == -1) {
  1921. php_error_docref(NULL, E_WARNING, "Unable to select [%d]: %s (max_fd=%d)",
  1922. errno, strerror(errno), max_fd);
  1923. DBG_RETURN(FAIL);
  1924. }
  1925. if (r_array != NULL) {
  1926. mysqlnd_stream_array_from_fd_set(r_array, &rfds);
  1927. }
  1928. if (e_array != NULL) {
  1929. mysqlnd_stream_array_from_fd_set(e_array, &efds);
  1930. }
  1931. *desc_num = retval;
  1932. DBG_RETURN(PASS);
  1933. }
  1934. /* }}} */
  1935. /* {{{ mysqlnd_connect */
  1936. PHPAPI MYSQLND * mysqlnd_connection_connect(MYSQLND * conn_handle,
  1937. const char * const host,
  1938. const char * const user,
  1939. const char * const passwd, unsigned int passwd_len,
  1940. const char * const db, unsigned int db_len,
  1941. unsigned int port,
  1942. const char * const sock_or_pipe,
  1943. unsigned int mysql_flags,
  1944. unsigned int client_api_flags
  1945. )
  1946. {
  1947. enum_func_status ret = FAIL;
  1948. bool self_alloced = FALSE;
  1949. MYSQLND_CSTRING hostname = { host, host? strlen(host) : 0 };
  1950. MYSQLND_CSTRING username = { user, user? strlen(user) : 0 };
  1951. MYSQLND_CSTRING password = { passwd, passwd_len };
  1952. MYSQLND_CSTRING database = { db, db_len };
  1953. MYSQLND_CSTRING socket_or_pipe = { sock_or_pipe, sock_or_pipe? strlen(sock_or_pipe) : 0 };
  1954. DBG_ENTER("mysqlnd_connect");
  1955. DBG_INF_FMT("host=%s user=%s db=%s port=%u flags=%u", host? host:"", user? user:"", db? db:"", port, mysql_flags);
  1956. if (!conn_handle) {
  1957. self_alloced = TRUE;
  1958. if (!(conn_handle = mysqlnd_connection_init(client_api_flags, FALSE, NULL))) {
  1959. /* OOM */
  1960. DBG_RETURN(NULL);
  1961. }
  1962. }
  1963. ret = conn_handle->m->connect(conn_handle, hostname, username, password, database, port, socket_or_pipe, mysql_flags);
  1964. if (ret == FAIL) {
  1965. if (self_alloced) {
  1966. /*
  1967. We have alloced, thus there are no references to this
  1968. object - we are free to kill it!
  1969. */
  1970. conn_handle->m->dtor(conn_handle);
  1971. }
  1972. DBG_RETURN(NULL);
  1973. }
  1974. DBG_RETURN(conn_handle);
  1975. }
  1976. /* }}} */
  1977. /* {{{ mysqlnd_connection_init */
  1978. PHPAPI MYSQLND *
  1979. mysqlnd_connection_init(const size_t client_flags, const bool persistent, MYSQLND_CLASS_METHODS_TYPE(mysqlnd_object_factory) *object_factory)
  1980. {
  1981. MYSQLND_CLASS_METHODS_TYPE(mysqlnd_object_factory) *factory = object_factory? object_factory : &MYSQLND_CLASS_METHOD_TABLE_NAME(mysqlnd_object_factory);
  1982. MYSQLND * ret;
  1983. DBG_ENTER("mysqlnd_connection_init");
  1984. ret = factory->get_connection(factory, persistent);
  1985. if (ret && ret->data) {
  1986. ret->data->m->negotiate_client_api_capabilities(ret->data, client_flags);
  1987. }
  1988. DBG_RETURN(ret);
  1989. }
  1990. /* }}} */