git.m6w6.name Git - m6w6/ext-pq/blob - src/php_pqstm.c

   1 /*
   2     +--------------------------------------------------------------------+
   3     | PECL :: pq                                                         |
   4     +--------------------------------------------------------------------+
   5     | Redistribution and use in source and binary forms, with or without |
   6     | modification, are permitted provided that the conditions mentioned |
   7     | in the accompanying LICENSE file are met.                          |
   8     +--------------------------------------------------------------------+
   9     | Copyright (c) 2013, Michael Wallner <mike@php.net>                 |
  10     +--------------------------------------------------------------------+
  11 */
  12
  13 #ifdef HAVE_CONFIG_H
  14 #       include "config.h"
  15 #endif
  16
  17 #include <php.h>
  18 #include <ext/standard/php_smart_str.h>
  19
  20 #include "php_pq.h"
  21 #include "php_pq_misc.h"
  22 #include "php_pq_object.h"
  23 #include "php_pqexc.h"
  24 #include "php_pqconn.h"
  25 #include "php_pqres.h"
  26 #include "php_pqstm.h"
  27
  28 zend_class_entry *php_pqstm_class_entry;
  29 static zend_object_handlers php_pqstm_object_handlers;
  30 static HashTable php_pqstm_object_prophandlers;
  31
  32 static void php_pqstm_object_free(void *o TSRMLS_DC)
  33 {
  34         php_pqstm_object_t *obj = o;
  35 #if DBG_GC
  36         fprintf(stderr, "FREE stm(#%d) %p (conn(#%d): %p)\n", obj->zv.handle, obj, obj->intern->conn->zv.handle, obj->intern->conn);
  37 #endif
  38         if (obj->intern) {
  39                 char *quoted_name = PQescapeIdentifier(obj->intern->conn->intern->conn, obj->intern->name, strlen(obj->intern->name));
  40
  41                 php_pq_callback_dtor(&obj->intern->conn->intern->onevent);
  42
  43                 if (quoted_name) {
  44                         PGresult *res;
  45                         smart_str cmd = {0};
  46
  47                         smart_str_appends(&cmd, "DEALLOCATE ");
  48                         smart_str_appends(&cmd, quoted_name);
  49                         smart_str_0(&cmd);
  50                         PQfreemem(quoted_name);
  51
  52                         if ((res = PQexec(obj->intern->conn->intern->conn, cmd.c))) {
  53                                 PHP_PQclear(res);
  54                         }
  55                         smart_str_free(&cmd);
  56                 }
  57
  58                 php_pq_object_delref(obj->intern->conn TSRMLS_CC);
  59                 efree(obj->intern->name);
  60                 zend_hash_destroy(&obj->intern->bound);
  61                 efree(obj->intern);
  62                 obj->intern = NULL;
  63         }
  64         zend_object_std_dtor((zend_object *) o TSRMLS_CC);
  65         efree(obj);
  66 }
  67
  68 zend_object_value php_pqstm_create_object_ex(zend_class_entry *ce, php_pqstm_t *intern, php_pqstm_object_t **ptr TSRMLS_DC)
  69 {
  70         php_pqstm_object_t *o;
  71
  72         o = ecalloc(1, sizeof(*o));
  73         zend_object_std_init((zend_object *) o, ce TSRMLS_CC);
  74         object_properties_init((zend_object *) o, ce);
  75         o->prophandler = &php_pqstm_object_prophandlers;
  76
  77         if (ptr) {
  78                 *ptr = o;
  79         }
  80
  81         if (intern) {
  82                 o->intern = intern;
  83         }
  84
  85         o->zv.handle = zend_objects_store_put((zend_object *) o, NULL, php_pqstm_object_free, NULL TSRMLS_CC);
  86         o->zv.handlers = &php_pqstm_object_handlers;
  87
  88         return o->zv;
  89 }
  90
  91 static zend_object_value php_pqstm_create_object(zend_class_entry *class_type TSRMLS_DC)
  92 {
  93         return php_pqstm_create_object_ex(class_type, NULL, NULL TSRMLS_CC);
  94 }
  95
  96 static void php_pqstm_object_read_name(zval *object, void *o, zval *return_value TSRMLS_DC)
  97 {
  98         php_pqstm_object_t *obj = o;
  99
 100         RETVAL_STRING(obj->intern->name, 1);
 101 }
 102
 103 static void php_pqstm_object_read_connection(zval *object, void *o, zval *return_value TSRMLS_DC)
 104 {
 105         php_pqstm_object_t *obj = o;
 106
 107         php_pq_object_to_zval(obj->intern->conn, &return_value TSRMLS_CC);
 108 }
 109
 110
 111 ZEND_BEGIN_ARG_INFO_EX(ai_pqstm_construct, 0, 0, 3)
 112         ZEND_ARG_OBJ_INFO(0, Connection, pq\\Connection, 0)
 113         ZEND_ARG_INFO(0, type)
 114         ZEND_ARG_INFO(0, query)
 115         ZEND_ARG_ARRAY_INFO(0, types, 1)
 116         ZEND_ARG_INFO(0, async)
 117 ZEND_END_ARG_INFO();
 118 static PHP_METHOD(pqstm, __construct) {
 119         zend_error_handling zeh;
 120         zval *zconn, *ztypes = NULL;
 121         char *name_str, *query_str;
 122         int name_len, *query_len;
 123         zend_bool async = 0;
 124         STATUS rv;
 125
 126         zend_replace_error_handling(EH_THROW, exce(EX_INVALID_ARGUMENT), &zeh TSRMLS_CC);
 127         rv = zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "Oss|a/!b", &zconn, php_pqconn_class_entry, &name_str, &name_len, &query_str, &query_len, &ztypes, &async);
 128         zend_restore_error_handling(&zeh TSRMLS_CC);
 129
 130         if (SUCCESS == rv) {
 131                 php_pqstm_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 132                 php_pqconn_object_t *conn_obj = zend_object_store_get_object(zconn TSRMLS_CC);
 133
 134                 if (!conn_obj->intern) {
 135                         throw_exce(EX_UNINITIALIZED TSRMLS_CC, "pq\\Connection not initialized");
 136                 } else {
 137                         if (async) {
 138                                 rv = php_pqconn_prepare_async(zconn, conn_obj, name_str, query_str, ztypes ? Z_ARRVAL_P(ztypes) : NULL TSRMLS_CC);
 139                         } else {
 140                                 rv = php_pqconn_prepare(zconn, conn_obj, name_str, query_str, ztypes ? Z_ARRVAL_P(ztypes) : NULL TSRMLS_CC);
 141                         }
 142
 143                         if (SUCCESS == rv) {
 144                                 php_pqstm_t *stm = ecalloc(1, sizeof(*stm));
 145
 146                                 php_pq_object_addref(conn_obj TSRMLS_CC);
 147                                 stm->conn = conn_obj;
 148                                 stm->name = estrdup(name_str);
 149                                 ZEND_INIT_SYMTABLE(&stm->bound);
 150                                 obj->intern = stm;
 151                         }
 152                 }
 153         }
 154 }
 155 ZEND_BEGIN_ARG_INFO_EX(ai_pqstm_bind, 0, 0, 2)
 156         ZEND_ARG_INFO(0, param_no)
 157         ZEND_ARG_INFO(1, param_ref)
 158 ZEND_END_ARG_INFO();
 159 static PHP_METHOD(pqstm, bind) {
 160         long param_no;
 161         zval *param_ref;
 162
 163         if (SUCCESS == zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "lz", &param_no, &param_ref)) {
 164                 php_pqstm_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 165
 166                 if (!obj->intern) {
 167                         throw_exce(EX_UNINITIALIZED TSRMLS_CC, "pq\\Statement not initialized");
 168                 } else {
 169                         Z_ADDREF_P(param_ref);
 170                         zend_hash_index_update(&obj->intern->bound, param_no, (void *) &param_ref, sizeof(zval *), NULL);
 171                         zend_hash_sort(&obj->intern->bound, zend_qsort, compare_index, 0 TSRMLS_CC);
 172                 }
 173         }
 174 }
 175
 176 ZEND_BEGIN_ARG_INFO_EX(ai_pqstm_exec, 0, 0, 0)
 177         ZEND_ARG_ARRAY_INFO(0, params, 1)
 178 ZEND_END_ARG_INFO();
 179 static PHP_METHOD(pqstm, exec) {
 180         zend_error_handling zeh;
 181         zval *zparams = NULL;
 182         STATUS rv;
 183
 184         zend_replace_error_handling(EH_THROW, exce(EX_INVALID_ARGUMENT), &zeh TSRMLS_CC);
 185         rv = zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "|a/!", &zparams);
 186         zend_restore_error_handling(&zeh TSRMLS_CC);
 187
 188         if (SUCCESS == rv) {
 189                 php_pqstm_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 190
 191                 if (!obj->intern) {
 192                         throw_exce(EX_UNINITIALIZED TSRMLS_CC, "pq\\Statement not initialized");
 193                 } else {
 194                         int count = 0;
 195                         char **params = NULL;
 196                         HashTable zdtor;
 197                         PGresult *res;
 198
 199                         ZEND_INIT_SYMTABLE(&zdtor);
 200
 201                         if (zparams) {
 202                                 count = php_pq_params_to_array(Z_ARRVAL_P(zparams), &params, &zdtor TSRMLS_CC);
 203                         } else {
 204                                 count = php_pq_params_to_array(&obj->intern->bound, &params, &zdtor TSRMLS_CC);
 205                         }
 206
 207                         res = PQexecPrepared(obj->intern->conn->intern->conn, obj->intern->name, count, (const char *const*) params, NULL, NULL, 0);
 208
 209                         if (params) {
 210                                 efree(params);
 211                         }
 212                         zend_hash_destroy(&zdtor);
 213
 214                         if (!res) {
 215                                 throw_exce(EX_RUNTIME TSRMLS_CC, "Failed to execute statement (%s)", PHP_PQerrorMessage(obj->intern->conn->intern->conn));
 216                         } else if (SUCCESS == php_pqres_success(res TSRMLS_CC)) {
 217                                 php_pq_object_to_zval_no_addref(PQresultInstanceData(res, php_pqconn_event), &return_value TSRMLS_CC);
 218                                 php_pqconn_notify_listeners(obj->intern->conn TSRMLS_CC);
 219                         }
 220                 }
 221         }
 222 }
 223
 224 ZEND_BEGIN_ARG_INFO_EX(ai_pqstm_exec_async, 0, 0, 0)
 225         ZEND_ARG_ARRAY_INFO(0, params, 1)
 226         ZEND_ARG_INFO(0, callable)
 227 ZEND_END_ARG_INFO();
 228 static PHP_METHOD(pqstm, execAsync) {
 229         zend_error_handling zeh;
 230         zval *zparams = NULL;
 231         php_pq_callback_t resolver = {{0}};
 232         STATUS rv;
 233
 234         zend_replace_error_handling(EH_THROW, exce(EX_INVALID_ARGUMENT), &zeh TSRMLS_CC);
 235         rv = zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "|a/!f", &zparams, &resolver.fci, &resolver.fcc);
 236         zend_restore_error_handling(&zeh TSRMLS_CC);
 237
 238         if (SUCCESS == rv) {
 239                 php_pqstm_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 240
 241                 if (!obj->intern) {
 242                         throw_exce(EX_UNINITIALIZED TSRMLS_CC, "pq\\Statement not initialized");
 243                 } else {
 244                         int count;
 245                         char **params = NULL;
 246                         HashTable zdtor;
 247
 248                         if (zparams) {
 249                                 ZEND_INIT_SYMTABLE(&zdtor);
 250                                 count = php_pq_params_to_array(Z_ARRVAL_P(zparams), &params, &zdtor TSRMLS_CC);
 251                         }
 252
 253                         if (!PQsendQueryPrepared(obj->intern->conn->intern->conn, obj->intern->name, count, (const char *const*) params, NULL, NULL, 0)) {
 254                                 throw_exce(EX_IO TSRMLS_CC, "Failed to execute statement (%s)", PHP_PQerrorMessage(obj->intern->conn->intern->conn));
 255                         } else if (obj->intern->conn->intern->unbuffered && !PQsetSingleRowMode(obj->intern->conn->intern->conn)) {
 256                                 throw_exce(EX_RUNTIME TSRMLS_CC, "Failed to enable unbuffered mode (%s)", PHP_PQerrorMessage(obj->intern->conn->intern->conn));
 257                         } else {
 258                                 php_pq_callback_dtor(&obj->intern->conn->intern->onevent);
 259                                 if (resolver.fci.size > 0) {
 260                                         obj->intern->conn->intern->onevent = resolver;
 261                                         php_pq_callback_addref(&obj->intern->conn->intern->onevent);
 262                                 }
 263                                 obj->intern->conn->intern->poller = PQconsumeInput;
 264                         }
 265
 266                         if (params) {
 267                                 efree(params);
 268                         }
 269                         if (zparams) {
 270                                 zend_hash_destroy(&zdtor);
 271                         }
 272
 273                         php_pqconn_notify_listeners(obj->intern->conn TSRMLS_CC);
 274                 }
 275         }
 276 }
 277
 278 ZEND_BEGIN_ARG_INFO_EX(ai_pqstm_desc, 0, 0, 0)
 279 ZEND_END_ARG_INFO();
 280 static PHP_METHOD(pqstm, desc) {
 281         zend_error_handling zeh;
 282         STATUS rv;
 283
 284         zend_replace_error_handling(EH_THROW, exce(EX_INVALID_ARGUMENT), &zeh TSRMLS_CC);
 285         rv = zend_parse_parameters_none();
 286         zend_restore_error_handling(&zeh TSRMLS_CC);
 287
 288         if (SUCCESS == rv) {
 289                 php_pqstm_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 290
 291                 if (!obj->intern) {
 292                         throw_exce(EX_UNINITIALIZED TSRMLS_CC, "pq\\Statement not initialized");
 293                 } else {
 294                         PGresult *res = PQdescribePrepared(obj->intern->conn->intern->conn, obj->intern->name);
 295
 296                         if (!res) {
 297                                 throw_exce(EX_RUNTIME TSRMLS_CC, "Failed to describe statement (%s)", PHP_PQerrorMessage(obj->intern->conn->intern->conn));
 298                         } else {
 299                                 if (SUCCESS == php_pqres_success(res TSRMLS_CC)) {
 300                                         int p, params;
 301
 302                                         array_init(return_value);
 303                                         for (p = 0, params = PQnparams(res); p < params; ++p) {
 304                                                 add_next_index_long(return_value, PQparamtype(res, p));
 305                                         }
 306                                 }
 307                                 PHP_PQclear(res);
 308                                 php_pqconn_notify_listeners(obj->intern->conn TSRMLS_CC);
 309                         }
 310                 }
 311         }
 312 }
 313
 314 ZEND_BEGIN_ARG_INFO_EX(ai_pqstm_desc_async, 0, 0, 0)
 315 ZEND_END_ARG_INFO();
 316 static PHP_METHOD(pqstm, descAsync) {
 317         zend_error_handling zeh;
 318         STATUS rv;
 319
 320         zend_replace_error_handling(EH_THROW, exce(EX_INVALID_ARGUMENT), &zeh TSRMLS_CC);
 321         rv = zend_parse_parameters_none();
 322         zend_restore_error_handling(&zeh TSRMLS_CC);
 323
 324         if (SUCCESS == rv) {
 325                 php_pqstm_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 326
 327                 if (!obj->intern) {
 328                         throw_exce(EX_UNINITIALIZED TSRMLS_CC, "pq\\Statement not initialized");
 329                 } else if (!PQsendDescribePrepared(obj->intern->conn->intern->conn, obj->intern->name)) {
 330                         throw_exce(EX_IO TSRMLS_CC, "Failed to describe statement: %s", PHP_PQerrorMessage(obj->intern->conn->intern->conn));
 331                 } else {
 332                         obj->intern->conn->intern->poller = PQconsumeInput;
 333                         php_pqconn_notify_listeners(obj->intern->conn TSRMLS_CC);
 334                 }
 335         }
 336 }
 337
 338 static zend_function_entry php_pqstm_methods[] = {
 339         PHP_ME(pqstm, __construct, ai_pqstm_construct, ZEND_ACC_PUBLIC|ZEND_ACC_CTOR)
 340         PHP_ME(pqstm, bind, ai_pqstm_bind, ZEND_ACC_PUBLIC)
 341         PHP_ME(pqstm, exec, ai_pqstm_exec, ZEND_ACC_PUBLIC)
 342         PHP_ME(pqstm, desc, ai_pqstm_desc, ZEND_ACC_PUBLIC)
 343         PHP_ME(pqstm, execAsync, ai_pqstm_exec_async, ZEND_ACC_PUBLIC)
 344         PHP_ME(pqstm, descAsync, ai_pqstm_desc_async, ZEND_ACC_PUBLIC)
 345         {0}
 346 };
 347
 348 PHP_MINIT_FUNCTION(pqstm)
 349 {
 350         zend_class_entry ce = {0};
 351         php_pq_object_prophandler_t ph = {0};
 352
 353         INIT_NS_CLASS_ENTRY(ce, "pq", "Statement", php_pqstm_methods);
 354         php_pqstm_class_entry = zend_register_internal_class_ex(&ce, NULL, NULL TSRMLS_CC);
 355         php_pqstm_class_entry->create_object = php_pqstm_create_object;
 356
 357         memcpy(&php_pqstm_object_handlers, zend_get_std_object_handlers(), sizeof(zend_object_handlers));
 358         php_pqstm_object_handlers.read_property = php_pq_object_read_prop;
 359         php_pqstm_object_handlers.write_property = php_pq_object_write_prop;
 360         php_pqstm_object_handlers.clone_obj = NULL;
 361         php_pqstm_object_handlers.get_property_ptr_ptr = NULL;
 362         php_pqstm_object_handlers.get_gc = NULL;
 363         php_pqstm_object_handlers.get_properties = php_pq_object_properties;
 364         php_pqstm_object_handlers.get_debug_info = php_pq_object_debug_info;
 365
 366         zend_hash_init(&php_pqstm_object_prophandlers, 2, NULL, NULL, 1);
 367
 368         zend_declare_property_null(php_pqstm_class_entry, ZEND_STRL("name"), ZEND_ACC_PUBLIC TSRMLS_CC);
 369         ph.read = php_pqstm_object_read_name;
 370         zend_hash_add(&php_pqstm_object_prophandlers, "name", sizeof("name"), (void *) &ph, sizeof(ph), NULL);
 371
 372         zend_declare_property_null(php_pqstm_class_entry, ZEND_STRL("connection"), ZEND_ACC_PUBLIC TSRMLS_CC);
 373         ph.read = php_pqstm_object_read_connection;
 374         zend_hash_add(&php_pqstm_object_prophandlers, "connection", sizeof("connection"), (void *) &ph, sizeof(ph), NULL);
 375
 376         return SUCCESS;
 377 }
 378
 379 /*
 380  * Local variables:
 381  * tab-width: 4
 382  * c-basic-offset: 4
 383  * End:
 384  * vim600: noet sw=4 ts=4 fdm=marker
 385  * vim<600: noet sw=4 ts=4
 386  */